mirror of
https://github.com/protocolbuffers/protobuf-go.git
synced 2025-01-04 02:38:50 +00:00
75f53c59e1
All the fuzzers have the same test, which runs the fuzzer against every entry in the corpus. Move the test logic into a separate package. Change-Id: I3a7e2ca75d20a5ff6d51ed9e6151629e6667684b Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/212258 Reviewed-by: Joe Tsai <joetsai@google.com>
40 lines
965 B
Go
40 lines
965 B
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 jsonfuzz includes fuzzers for protojson.Marshal and protojson.Unmarshal.
|
|
package jsonfuzz
|
|
|
|
import (
|
|
"google.golang.org/protobuf/encoding/protojson"
|
|
"google.golang.org/protobuf/proto"
|
|
|
|
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{}
|
|
if err := (protojson.UnmarshalOptions{
|
|
AllowPartial: true,
|
|
}).Unmarshal(data, m1); err != nil {
|
|
return 0
|
|
}
|
|
data1, err := protojson.MarshalOptions{
|
|
AllowPartial: true,
|
|
}.Marshal(m1)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
m2 := &fuzzpb.Fuzz{}
|
|
if err := (protojson.UnmarshalOptions{
|
|
AllowPartial: true,
|
|
}).Unmarshal(data1, m2); err != nil {
|
|
return 0
|
|
}
|
|
if !proto.Equal(m1, m2) {
|
|
panic("not equal")
|
|
}
|
|
return 1
|
|
}
|