import (
"errors"
"fmt"
+ "internal/saferio"
"math"
"math/big"
"reflect"
offset += t.length
numElements++
}
- ret = reflect.MakeSlice(sliceType, numElements, numElements)
+ elemSize := uint64(elemType.Size())
+ safeCap := saferio.SliceCapWithSize(elemSize, uint64(numElements))
+ if safeCap < 0 {
+ err = SyntaxError{fmt.Sprintf("%s slice too big: %d elements of %d bytes", elemType.Kind(), numElements, elemSize)}
+ return
+ }
+ ret = reflect.MakeSlice(sliceType, 0, safeCap)
params := fieldParameters{}
offset := 0
for i := 0; i < numElements; i++ {
+ ret = reflect.Append(ret, reflect.Zero(elemType))
offset, err = parseField(ret.Index(i), bytes, offset, params)
if err != nil {
return
import (
"bytes"
"encoding/hex"
+ "errors"
"fmt"
"math"
"math/big"
"reflect"
+ "runtime"
"strings"
"testing"
"time"
_ = oidPublicKeyRSA.String()
}
}
+
+func TestParsingMemoryConsumption(t *testing.T) {
+ // Craft a syntatically valid, but empty, ~10 MB DER bomb. A successful
+ // unmarshal of this bomb should yield ~280 MB. However, the parsing should
+ // fail due to the empty content; and, in such cases, we want to make sure
+ // that we do not unnecessarily allocate memories.
+ derBomb := make([]byte, 10_000_000)
+ for i := range derBomb {
+ derBomb[i] = 0x30
+ }
+ derBomb = append([]byte{0x30, 0x83, 0x98, 0x96, 0x80}, derBomb...)
+
+ var m runtime.MemStats
+ runtime.GC()
+ runtime.ReadMemStats(&m)
+ memBefore := m.TotalAlloc
+
+ var out []struct {
+ Id []int
+ Critical bool `asn1:"optional"`
+ Value []byte
+ }
+ _, err := Unmarshal(derBomb, &out)
+ if !errors.As(err, &SyntaxError{}) {
+ t.Fatalf("Incorrect error result: want (%v), but got (%v) instead", &SyntaxError{}, err)
+ }
+
+ runtime.ReadMemStats(&m)
+ memDiff := m.TotalAlloc - memBefore
+
+ // Ensure that the memory allocated does not exceed 10<<21 (~20 MB) when
+ // the parsing fails.
+ if memDiff > 10<<21 {
+ t.Errorf("Too much memory allocated while parsing DER: %v MiB", memDiff/1024/1024)
+ }
+}
# CRYPTO-MATH is crypto that exposes math/big APIs - no cgo, net; fmt now ok.
- CRYPTO, FMT, math/big
+ CRYPTO, FMT, math/big, internal/saferio
< crypto/internal/boring/bbig
< crypto/rand
< crypto/ed25519 # depends on crypto/rand.Reader