From 591f09ab7ccf5332e66d6e6d24e1e8fb08167b89 Mon Sep 17 00:00:00 2001
From: Damien Neil <dneil@google.com>
Date: Wed, 10 Jul 2019 16:11:19 -0700
Subject: [PATCH] internal/fuzz: add basic fuzzing target for Marshal/Unmarshal

Change-Id: Ibc4e1d53958abd0973fd568e9dafd5be9a8bdc0d
Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/200537
Reviewed-by: Joe Tsai <thebrokentoaster@gmail.com>
---
 .gitignore                 |  1 +
 internal/fuzz/README.md    | 12 ++++++++++++
 internal/fuzz/wire/fuzz.go | 40 ++++++++++++++++++++++++++++++++++++++
 3 files changed, 53 insertions(+)
 create mode 100644 internal/fuzz/README.md
 create mode 100644 internal/fuzz/wire/fuzz.go

diff --git a/.gitignore b/.gitignore
index fd09e450..6e3f595c 100644
--- a/.gitignore
+++ b/.gitignore
@@ -6,6 +6,7 @@
 /cmd/protoc-gen-go-grpc/protoc-gen-go-grpc
 /cmd/protoc-gen-go-grpc/testdata/go.sum
 /vendor
+/internal/fuzz/wire/fuzz-fuzz.zip
 
 # This file includes artifacts of the system test that should not be checked in.
 # For files created by specific development environment (e.g. editor),
diff --git a/internal/fuzz/README.md b/internal/fuzz/README.md
new file mode 100644
index 00000000..9ec65863
--- /dev/null
+++ b/internal/fuzz/README.md
@@ -0,0 +1,12 @@
+# Fuzzing
+
+Fuzzing support using [go-fuzz](https://github.com/dvyukov/go-fuzz).
+
+Basic operation:
+
+```sh
+$ go install github.com/dvyukov/go-fuzz/go-fuzz github.com/dvyukov/go-fuzz/go-fuzz-build
+$ cd internal/fuzz/{fuzzer}
+$ GOFUZZ111MODULE=on go-fuzz-build .
+$ go-fuzz
+```
diff --git a/internal/fuzz/wire/fuzz.go b/internal/fuzz/wire/fuzz.go
new file mode 100644
index 00000000..c0da64ac
--- /dev/null
+++ b/internal/fuzz/wire/fuzz.go
@@ -0,0 +1,40 @@
+// 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.
+
+// +build gofuzz
+
+// Package wire includes a fuzzer for the wire marshaler and unmarshaler.
+package wire
+
+import (
+	"google.golang.org/protobuf/proto"
+
+	testpb "google.golang.org/protobuf/internal/testprotos/test"
+)
+
+// Fuzz is a fuzzer for proto.Marshal and proto.Unmarshal.
+func Fuzz(data []byte) int {
+	score := 0
+	for _, newf := range []func() proto.Message{
+		func() proto.Message { return &testpb.TestAllTypes{} },
+	} {
+		m1 := newf()
+		if err := proto.Unmarshal(data, m1); err != nil {
+			continue
+		}
+		score = 1
+		data1, err := proto.Marshal(m1)
+		if err != nil {
+			panic(err)
+		}
+		m2 := newf()
+		if err := proto.Unmarshal(data1, m2); err != nil {
+			panic(err)
+		}
+		if !proto.Equal(m1, m2) {
+			panic("not equal")
+		}
+	}
+	return score
+}