Merge pull request #7902 from AdamKorcz/fuzz5

Planbuilder: Add fuzzer

go/vt/vtgate/planbuilder/fuzz.go

@@ -0,0 +1,51 @@
+/*
+Copyright 2021 The Vitess Authors.
+
+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 planbuilder
+
+import (
+ "vitess.io/vitess/go/vt/key"
+ topodatapb "vitess.io/vitess/go/vt/proto/topodata"
+ "vitess.io/vitess/go/vt/sqlparser"
+ "vitess.io/vitess/go/vt/vtgate/semantics"
+ "vitess.io/vitess/go/vt/vtgate/vindexes"
+)
+
+var _ semantics.SchemaInformation = (*fakeFuzzSI)(nil)
+
+type fakeFuzzSI struct {
+ tables map[string]*vindexes.Table
+}
+
+// Helper func:
+func (s *fakeFuzzSI) FindTableOrVindex(tablename sqlparser.TableName) (*vindexes.Table, vindexes.Vindex, string, topodatapb.TabletType, key.Destination, error) {
+ return s.tables[sqlparser.String(tablename)], nil, "", 0, nil, nil
+}
+
+// FuzzAnalyse implements the fuzzer
+func FuzzAnalyse(data []byte) int {
+ tree, err := sqlparser.Parse(string(data))
+ if err != nil {
+ return -1
+ }
+ semTable, err := semantics.Analyse(tree, &fakeFuzzSI{})
+ if err != nil {
+ return 0
+ }
+ _, _ = createQGFromSelect(tree.(*sqlparser.Select), semTable)
+ return 1
+}