tiktak: защита от decompression bomb в /create

Приходят 100 КБ webm с VP8-кадром гигантского разрешения (boundary у них
буквально ----bomb*). Проверку f.Size > 5MB он проходит, validateWebm смотрит
только первые 512 байт, а ffmpeg разворачивает такой кадр в ~1 ГБ сырых
пикселей. При семафоре на 75 параллельных это укладывает box, чекер не
укладывается в свои 7 секунд => MUMBLE.

- video/probe.go: ffprobe читает geometry из заголовка (кадр не декодируется),
  режем > 4096 по стороне и > 8 Mpx.
- Blur: image.DecodeConfig до image.Decode, тот же лимит по пикселям.
  imaging.Blur держит несколько копий битмапа, без лимита это OOM сам по себе.
  Заодно f.Close(), которого не было.
- ffmpeg -threads 1, семафор 75 -> 16.
- BodyLimit 8M: размер видео проверялся, а форма нет. Огромные субтитры
  превращались в слайс на миллионы строк в GenerateVtt. Плюс явный
  MaxSubtitlesSize 64K.

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
2026-08-26 12:16:56 +03:00
co-authored by Claude Opus 5
parent c0089b2c11
commit 576baa3949
4 changed files with 70 additions and 2 deletions
+7
View File
@@ -23,6 +23,10 @@ func NewServer(db *db.Storage, vs *video.Storage, c Config) *Server {
t := Template{template.Must(template.ParseGlob(c.TemplatesFolder + "/*.html"))}
s.e.Renderer = &t
s.e.Use(middleware.Recover())
// Hard cap on the whole request. The video file is size-checked separately,
// but nothing bounded the form fields: a huge subtitles blob becomes one
// slice entry per line inside GenerateVtt.
s.e.Use(middleware.BodyLimit("8M"))
s.e.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
return func(c echo.Context) error {
cc := &auth.Context{Context: c}
@@ -111,6 +115,9 @@ func (s *Server) handleCreate(c echo.Context) error {
descr := c.FormValue("description")
subt := c.FormValue("subtitles")
if len(subt) > video.MaxSubtitlesSize {
return c.Render(http.StatusUnprocessableEntity, "create", ErrorResponse{"subtitles are too long"})
}
isPrivate := c.FormValue("private")
v := db.Video{UserID: uid, Description: descr, Private: isPrivate == "on", Link: link}
+24 -2
View File
@@ -6,6 +6,7 @@ import (
"github.com/disintegration/imaging"
"image"
"image/png"
"io"
"os"
"os/exec"
"strings"
@@ -14,11 +15,17 @@ import (
var (
ProcessingDeadline = 2 * time.Second
semaphore = make(chan struct{}, 75)
// 75 concurrent ffmpeg processes are enough to swap the box out on their
// own, bomb or no bomb.
semaphore = make(chan struct{}, 16)
)
func generatePreview(ctx context.Context, inp string, out string) error {
cmd := exec.CommandContext(ctx, "ffmpeg", "-y", "-i", inp, "-vframes", "1", out)
// Reject decompression bombs before ffmpeg expands a frame into memory.
if err := checkGeometry(ctx, inp); err != nil {
return err
}
cmd := exec.CommandContext(ctx, "ffmpeg", "-y", "-threads", "1", "-i", inp, "-vframes", "1", out)
s := strings.Builder{}
cmd.Stdout = &s
cmd.Stderr = &s
@@ -48,6 +55,21 @@ func Blur(inp string, out string) error {
if err != nil {
return err
}
defer f.Close()
// Same guard on the decode side: imaging.Blur keeps several copies of the
// bitmap alive, so an oversized preview is an OOM all by itself.
cfg, _, err := image.DecodeConfig(f)
if err != nil {
return err
}
if cfg.Width*cfg.Height > MaxPixels {
return fmt.Errorf("preview %dx%d is too large to blur", cfg.Width, cfg.Height)
}
if _, err := f.Seek(0, io.SeekStart); err != nil {
return err
}
img, _, err := image.Decode(f)
if err != nil {
return err
+36
View File
@@ -0,0 +1,36 @@
package video
import (
"context"
"fmt"
"os/exec"
"strings"
)
// Decompression-bomb guard. A 100 KB webm can carry a 16383x16383 VP8 frame
// (the format's maximum): tiny on disk, ~1 GB once ffmpeg expands it to raw
// pixels. Probing the container header is cheap -- ffprobe reads metadata and
// never decodes a frame -- so we reject absurd geometry before any decoding.
const (
MaxDimension = 4096
MaxPixels = 8 << 20 // 8 Mpx, ~4x a 1080p frame
)
func checkGeometry(ctx context.Context, inp string) error {
args := []string{"-v", "error", "-select_streams", "v:0",
"-show_entries", "stream=width,height", "-of", "csv=s=x:p=0", inp}
out, err := exec.CommandContext(ctx, "ffprobe", args...).Output()
if err != nil {
return fmt.Errorf("cannot probe video: %v", err)
}
var w, h int
if _, err := fmt.Sscanf(strings.TrimSpace(string(out)), "%dx%d", &w, &h); err != nil {
return fmt.Errorf("cannot read video geometry")
}
if w <= 0 || h <= 0 || w > MaxDimension || h > MaxDimension || w*h > MaxPixels {
return fmt.Errorf("video resolution %dx%d is not supported", w, h)
}
return nil
}
+3
View File
@@ -12,6 +12,9 @@ import (
const (
MaxSize = 5 << 20
// GenerateVtt splits this on \n and allocates one slice entry per line.
MaxSubtitlesSize = 64 << 10
)
var (