From fd43eddb19f5c3ed6608b60f90ffcfbe0d51bd46 Mon Sep 17 00:00:00 2001
From: Cody Oss <codyoss@google.com>
Date: Tue, 22 Sep 2020 14:24:43 -0600
Subject: [PATCH] test(spanner/spansql): add fuzz target for ParseQuery

This target will be called from a runner managed by the oss-fuzz
project. This target is an experimental test integration with the
oss-fuzz project.
---
 spanner/spansql/fuzz.go | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 spanner/spansql/fuzz.go

diff --git a/spanner/spansql/fuzz.go b/spanner/spansql/fuzz.go
new file mode 100644
index 00000000000..e95624be714
--- /dev/null
+++ b/spanner/spansql/fuzz.go
@@ -0,0 +1,28 @@
+// Copyright 2020 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// +build gofuzz
+
+package spansql
+
+func FuzzParseQuery(data []byte) int {
+	if _, err := ParseQuery(string(data)); err != nil {
+		// The value 0 signals data is an invalid query that should be
+		// added to the corpus.
+		return 0
+	}
+	// The value 1 signals the input was lexically corrent and the
+	// fuzzer should increase the priority of the given input.
+	return 1
+}