78 lines
2.3 KiB
Go
78 lines
2.3 KiB
Go
|
package httpapi
|
||
|
|
|
||
|
|
import (
|
||
|
|
"archive/zip"
|
||
|
|
"bytes"
|
||
|
|
"io"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
// En zip-bombe skal ikke kunne skrive gigabyte ned på den disk, databasen ligger
|
||
|
|
// på.
|
||
|
|
//
|
||
|
|
// Uploaden er begrænset til 64 MB, og det siger ingenting om, hvad de 64 MB
|
||
|
|
// bliver til: deflate når omkring 1032:1, så de kan pakkes ud til seksogtres
|
||
|
|
// gigabyte. Prøven bygger en rigtig lille bombe — fire megabyte nuller, som
|
||
|
|
// pakker til nogle få kilobyte — og ser efter, at loftet på det udpakkede holder.
|
||
|
|
//
|
||
|
|
// Skrevet imod tallene frem for imod endepunktet, fordi det, der skal holde, er
|
||
|
|
// forholdet: en fil, hvis udpakkede størrelse overstiger loftet, må ikke læses
|
||
|
|
// færdig.
|
||
|
|
func TestAZipBombIsRefusedByItsUnpackedSize(t *testing.T) {
|
||
|
|
var buf bytes.Buffer
|
||
|
|
zw := zip.NewWriter(&buf)
|
||
|
|
w, err := zw.Create("bombe.bin")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
// Nuller pakker næsten helt væk. Fire megabyte er nok til at vise forholdet
|
||
|
|
// uden at prøven selv bruger en gigabyte.
|
||
|
|
if _, err := w.Write(make([]byte, 4<<20)); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if err := zw.Close(); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
packed := int64(buf.Len())
|
||
|
|
if packed > 64<<10 {
|
||
|
|
t.Fatalf("prøvens egen bombe pakkede kun til %d bytes; den viser ikke noget", packed)
|
||
|
|
}
|
||
|
|
|
||
|
|
zr, err := zip.NewReader(bytes.NewReader(buf.Bytes()), packed)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
f := zr.File[0]
|
||
|
|
if f.UncompressedSize64 != 4<<20 {
|
||
|
|
t.Fatalf("udpakket størrelse = %d", f.UncompressedSize64)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Det, importen gør: læs højst det, budgettet tillader, plus én byte — så en
|
||
|
|
// fil, der er præcis på grænsen, kan skelnes fra en, der er over.
|
||
|
|
rc, err := f.Open()
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer rc.Close()
|
||
|
|
budget := int64(1 << 20)
|
||
|
|
n, err := io.Copy(io.Discard, io.LimitReader(rc, budget+1))
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if n <= budget {
|
||
|
|
t.Fatalf("læste %d bytes af en fil på 4 MB med et budget på %d — loftet fanger den ikke", n, budget)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Tallene skal stå i et forhold, der giver mening: en enkelt fil må aldrig kunne
|
||
|
|
// bruge hele arkivets budget, og arkivet må ikke være mindre end én fil.
|
||
|
|
func TestTheImportCeilingsAgreeWithEachOther(t *testing.T) {
|
||
|
|
if maxNoteImportUnpacked <= maxUploadBytes {
|
||
|
|
t.Error("arkivets budget er ikke større end én enkelt fil")
|
||
|
|
}
|
||
|
|
if maxNoteImportFiles < 2000 {
|
||
|
|
t.Error("filloftet er under det, en rigtig Apple Noter-eksport indeholder")
|
||
|
|
}
|
||
|
|
}
|