Add flags for tls file path (#1419)

benchmark/worker/benchmark_client.go

@@ -19,6 +19,7 @@
 package main
 
 import (
+	"flag"
 	"math"
 	"runtime"
 	"sync"
@@ -36,6 +37,10 @@ import (
 	"google.golang.org/grpc/testdata"
 )
 
+var (
+	caFile = flag.String("ca_file", "", "The file containning the CA root cert file")
+)
+
 type lockingHistogram struct {
 	mu        sync.Mutex
 	histogram *stats.Histogram
@@ -119,7 +124,10 @@ func createConns(config *testpb.ClientConfig) ([]*grpc.ClientConn, func(), error
 
 	// Check and set security options.
 	if config.SecurityParams != nil {
-		creds, err := credentials.NewClientTLSFromFile(testdata.Path("ca.pem"), config.SecurityParams.ServerHostOverride)
+		if *caFile == "" {
+			*caFile = testdata.Path("ca.pem")
+		}
+		creds, err := credentials.NewClientTLSFromFile(*caFile, config.SecurityParams.ServerHostOverride)
 		if err != nil {
 			return nil, nil, grpc.Errorf(codes.InvalidArgument, "failed to create TLS credentials %v", err)
 		}

benchmark/worker/benchmark_server.go

@@ -19,6 +19,7 @@
 package main
 
 import (
+	"flag"
 	"runtime"
 	"strconv"
 	"strings"
@@ -35,6 +36,11 @@ import (
 	"google.golang.org/grpc/testdata"
 )
 
+var (
+	certFile = flag.String("tls_cert_file", "", "The TLS cert file")
+	keyFile  = flag.String("tls_key_file", "", "The TLS key file")
+)
+
 type benchmarkServer struct {
 	port            int
 	cores           int
@@ -85,7 +91,13 @@ func startBenchmarkServer(config *testpb.ServerConfig, serverPort int) (*benchma
 
 	// Set security options.
 	if config.SecurityParams != nil {
-		creds, err := credentials.NewServerTLSFromFile(testdata.Path("server1.pem"), testdata.Path("server1.key"))
+		if *certFile == "" {
+			*certFile = testdata.Path("server1.pem")
+		}
+		if *keyFile == "" {
+			*keyFile = testdata.Path("server1.key")
+		}
+		creds, err := credentials.NewServerTLSFromFile(*certFile, *keyFile)
 		if err != nil {
 			grpclog.Fatalf("failed to generate credentials %v", err)
 		}