Merge pull request #7689 from AdamKorcz/fuzz2

grpcvtgateconn: Add fuzzer

go/vt/vtgate/grpcvtgateconn/fuzz_flaky_test.go

@@ -0,0 +1,119 @@
+// +build gofuzz
+
+/*
+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.
+*/
+
+package grpcvtgateconn
+
+import (
+	"context"
+	"flag"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"net"
+	"os"
+	"testing"
+
+	"google.golang.org/grpc"
+
+	"vitess.io/vitess/go/vt/servenv"
+	"vitess.io/vitess/go/vt/vtgate/grpcvtgateservice"
+	"vitess.io/vitess/go/vt/vtgate/vtgateconn"
+)
+
+func init() {
+	testing.Init()
+}
+
+func IsDivisibleBy(n int, divisibleby int) bool {
+	return (n % divisibleby) == 0
+}
+
+func Fuzz(data []byte) int {
+	t := &testing.T{}
+	if len(data) < 20 {
+		return -1
+	}
+	if IsDivisibleBy(len(data), 10) == false {
+		return -1
+	}
+
+	var opts []grpc.ServerOption
+	// fake service
+	service := CreateFakeServer(t)
+
+	// listen on a random port
+	listener, err := net.Listen("tcp", ":0")
+	if err != nil {
+		fmt.Println("Cannot listen: %v", err)
+		return -1
+	}
+	defer listener.Close()
+
+	// add auth interceptors
+	opts = append(opts, grpc.StreamInterceptor(servenv.FakeAuthStreamInterceptor))
+	opts = append(opts, grpc.UnaryInterceptor(servenv.FakeAuthUnaryInterceptor))
+
+	// Create a gRPC server and listen on the port
+	server := grpc.NewServer(opts...)
+	grpcvtgateservice.RegisterForTest(server, service)
+	go server.Serve(listener)
+	defer server.GracefulStop()
+
+	authJSON := `{
+         "Username": "valid",
+         "Password": "valid"
+        }`
+
+	f, err := ioutil.TempFile("", "static_auth_creds.json")
+	if err != nil {
+		return -1
+	}
+	defer os.Remove(f.Name())
+	if _, err := io.WriteString(f, authJSON); err != nil {
+		return -1
+	}
+	if err := f.Close(); err != nil {
+		return -1
+	}
+
+	// Create a Go RPC client connecting to the server
+	ctx := context.Background()
+	flag.Set("grpc_auth_static_client_creds", f.Name())
+	client, err := dial(ctx, listener.Addr().String())
+	if err != nil {
+		fmt.Println("dial failed: %v", err)
+		return -1
+	}
+	defer client.Close()
+
+	RegisterTestDialProtocol(client)
+	conn, err := vtgateconn.DialProtocol(context.Background(), "test", "")
+	if err != nil {
+		fmt.Println("Got err: %v from vtgateconn.DialProtocol", err)
+		return -1
+	}
+	session := conn.Session("connection_ks@rdonly", testExecuteOptions)
+
+	// Do the actual fuzzing:
+	// 10 executions per fuzz run
+	ctx = newContext()
+	chunkSize := len(data) / 10
+	for i := 0; i < len(data); i = i + chunkSize {
+		from := i           //lower
+		to := i + chunkSize //upper
+		_, _ = session.Execute(ctx, string(data[from:to]), nil)
+	}
+	return 1
+}