route/vendor/github.com/bifurcation/mint/fuzz_test.go

90 lines
2.1 KiB
Go
Raw Normal View History

2018-01-03 19:19:49 +00:00
package mint
import (
"math/rand"
"testing"
)
var structs = []interface{}{
// Handshake messages
&ClientHelloBody{},
&ServerHelloBody{},
&FinishedBody{VerifyDataLen: 32},
&EncryptedExtensionsBody{},
&CertificateBody{},
&CertificateVerifyBody{},
// Extensions
&Extension{},
&ExtensionList{},
new(ServerNameExtension),
&ALPNExtension{},
&KeyShareExtension{HandshakeType: HandshakeTypeClientHello},
&KeyShareExtension{HandshakeType: HandshakeTypeHelloRetryRequest},
&KeyShareExtension{HandshakeType: HandshakeTypeServerHello},
&SupportedGroupsExtension{},
&SignatureAlgorithmsExtension{},
&PreSharedKeyExtension{HandshakeType: HandshakeTypeClientHello},
&PreSharedKeyExtension{HandshakeType: HandshakeTypeServerHello},
&SupportedVersionsExtension{},
}
var validHex = []string{
// Handshake messages
chValidHex,
shValidHex,
finValidHex,
encExtValidHex,
certValidHex,
certVerifyValidHex,
// Extensions
extValidHex,
extListValidHex,
validExtensionTestCases[ExtensionTypeServerName].marshaledHex,
validExtensionTestCases[ExtensionTypeALPN].marshaledHex,
keyShareClientHex,
keyShareHelloRetryHex,
keyShareServerHex,
validExtensionTestCases[ExtensionTypeSupportedGroups].marshaledHex,
validExtensionTestCases[ExtensionTypeSignatureAlgorithms].marshaledHex,
pskClientHex,
pskServerHex,
validExtensionTestCases[ExtensionTypeSupportedVersions].marshaledHex,
}
func randomBytes(n int, rand *rand.Rand) []byte {
r := make([]byte, n)
for i := 0; i < n; i++ {
r[i] = byte(rand.Int31())
}
return r
}
type unmarshaler interface {
Unmarshal([]byte) (int, error)
}
// This just looks for crashes due to bounds errors etc.
func TestFuzz(t *testing.T) {
rand := rand.New(rand.NewSource(0))
for i, iface := range structs {
m := iface.(unmarshaler)
// Provide random data
for j := 0; j < 100; j++ {
len := rand.Intn(1024)
bytes := randomBytes(len, rand)
m.Unmarshal(bytes)
}
// Provide partially valid data
valid := unhex(validHex[i])
random := randomBytes(10*len(valid), rand)
for cut := 0; cut < len(valid)-1; cut++ {
testCase := append(valid[:cut], random...)
m.Unmarshal(testCase)
}
}
}