(address string, useTLS, testCA bool, tlsServerName string)
| 288 | } |
| 289 | |
| 290 | func newConn(address string, useTLS, testCA bool, tlsServerName string) (*grpc.ClientConn, error) { |
| 291 | var opts []grpc.DialOption |
| 292 | if *customCredentialsType != "" { |
| 293 | if *customCredentialsType == googleDefaultCredsName { |
| 294 | opts = append(opts, grpc.WithCredentialsBundle(google.NewDefaultCredentials())) |
| 295 | } else if *customCredentialsType == computeEngineCredsName { |
| 296 | opts = append(opts, grpc.WithCredentialsBundle(google.NewComputeEngineCredentials())) |
| 297 | } else { |
| 298 | logger.Fatalf("Unknown custom credentials: %v", *customCredentialsType) |
| 299 | } |
| 300 | } else if useTLS { |
| 301 | var sn string |
| 302 | if tlsServerName != "" { |
| 303 | sn = tlsServerName |
| 304 | } |
| 305 | var creds credentials.TransportCredentials |
| 306 | if testCA { |
| 307 | var err error |
| 308 | if *caFile == "" { |
| 309 | *caFile = testdata.Path("x509/server_ca_cert.pem") |
| 310 | } |
| 311 | creds, err = credentials.NewClientTLSFromFile(*caFile, sn) |
| 312 | if err != nil { |
| 313 | logger.Fatalf("Failed to create TLS credentials: %v", err) |
| 314 | } |
| 315 | } else { |
| 316 | creds = credentials.NewClientTLSFromCert(nil, sn) |
| 317 | } |
| 318 | opts = append(opts, grpc.WithTransportCredentials(creds)) |
| 319 | } else { |
| 320 | opts = append(opts, grpc.WithTransportCredentials(insecure.NewCredentials())) |
| 321 | } |
| 322 | return grpc.NewClient(address, opts...) |
| 323 | } |
| 324 | |
| 325 | func main() { |
| 326 | flag.Parse() |
no test coverage detected