mirror of
https://github.com/protocolbuffers/protobuf-go.git
synced 2025-01-25 06:35:21 +00:00
f26a9e7e30
An Is prefix implies it returns a boolean. A Check prefix better suggests that it could return an error. Change-Id: I6ffcb32099a944c656c07654c294a0980efb2d0e Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/220338 Reviewed-by: Damien Neil <dneil@google.com>
65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
// Copyright 2019 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// Package wirefuzz includes a fuzzer for the wire marshaler and unmarshaler.
|
|
package wirefuzz
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"google.golang.org/protobuf/internal/impl"
|
|
"google.golang.org/protobuf/proto"
|
|
piface "google.golang.org/protobuf/runtime/protoiface"
|
|
|
|
fuzzpb "google.golang.org/protobuf/internal/testprotos/fuzz"
|
|
)
|
|
|
|
// Fuzz is a fuzzer for proto.Marshal and proto.Unmarshal.
|
|
func Fuzz(data []byte) (score int) {
|
|
m1 := &fuzzpb.Fuzz{}
|
|
vout, valid := impl.Validate(m1.ProtoReflect().Type(), piface.UnmarshalInput{
|
|
Buf: data,
|
|
})
|
|
vinit := vout.Flags&piface.UnmarshalInitialized != 0
|
|
if err := (proto.UnmarshalOptions{
|
|
AllowPartial: true,
|
|
}).Unmarshal(data, m1); err != nil {
|
|
switch valid {
|
|
case impl.ValidationUnknown:
|
|
case impl.ValidationInvalid:
|
|
default:
|
|
panic("unmarshal error with validation status: " + valid.String())
|
|
}
|
|
return 0
|
|
}
|
|
switch valid {
|
|
case impl.ValidationUnknown:
|
|
case impl.ValidationValid:
|
|
default:
|
|
panic("unmarshal ok with validation status: " + valid.String())
|
|
}
|
|
if proto.CheckInitialized(m1) != nil && vinit {
|
|
panic("validation reports partial message is initialized")
|
|
}
|
|
data1, err := proto.MarshalOptions{
|
|
AllowPartial: true,
|
|
}.Marshal(m1)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if proto.Size(m1) != len(data1) {
|
|
panic(fmt.Errorf("size does not match output %v", m1))
|
|
}
|
|
m2 := &fuzzpb.Fuzz{}
|
|
if err := (proto.UnmarshalOptions{
|
|
AllowPartial: true,
|
|
}).Unmarshal(data1, m2); err != nil {
|
|
panic(err)
|
|
}
|
|
if !proto.Equal(m1, m2) {
|
|
panic("not equal")
|
|
}
|
|
return 1
|
|
}
|