Alessandro RosandGitHub
363ac206f7
Check for limits before allocating memory by using sizes passed from the remote peer. Also add fuzzing to all MoQ primitives.
78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
package parameter
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
var cases = []struct {
|
|
name string
|
|
count int
|
|
enc []byte
|
|
dec Parameters
|
|
}{
|
|
{
|
|
name: "no parameters",
|
|
count: 0,
|
|
enc: []byte{},
|
|
dec: nil,
|
|
},
|
|
{
|
|
name: "authorization token",
|
|
count: 1,
|
|
enc: []byte{
|
|
0x03, // type delta = 3 (AuthorizationToken)
|
|
0x08, // inner length = 8
|
|
0x03, // alias type = UseValue
|
|
0x01, // token type = 1
|
|
0x73, 0x65, 0x63, 0x72, 0x65, 0x74, // token value = "secret"
|
|
},
|
|
dec: Parameters{
|
|
&AuthorizationToken{
|
|
AliasType: AuthorizationTokenAliasTypeUseValue,
|
|
TokenType: 1,
|
|
TokenValue: []byte("secret"),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
func TestUnmarshal(t *testing.T) {
|
|
for _, ca := range cases {
|
|
t.Run(ca.name, func(t *testing.T) {
|
|
var params Parameters
|
|
_, err := params.Unmarshal(ca.count, ca.enc)
|
|
require.NoError(t, err)
|
|
require.Equal(t, ca.dec, params)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestMarshal(t *testing.T) {
|
|
for _, ca := range cases {
|
|
t.Run(ca.name, func(t *testing.T) {
|
|
buf := make([]byte, ca.dec.MarshalSize())
|
|
ca.dec.MarshalTo(buf)
|
|
require.Equal(t, ca.enc, buf)
|
|
})
|
|
}
|
|
}
|
|
|
|
func FuzzUnmarshal(f *testing.F) {
|
|
for _, ca := range cases {
|
|
f.Add(ca.count, ca.enc)
|
|
}
|
|
|
|
f.Fuzz(func(_ *testing.T, count int, buf []byte) {
|
|
var params Parameters
|
|
_, err := params.Unmarshal(count, buf)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
buf = make([]byte, params.MarshalSize())
|
|
params.MarshalTo(buf)
|
|
})
|
|
}
|