mirror of https://github.com/grpc/grpc-go.git
credentials.TLSInfo: export State
This commit is contained in:
parent
1a700f0ee9
commit
a6fe03ae2e
|
@ -119,7 +119,7 @@ type TransportAuthenticator interface {
|
|||
// TLSInfo contains the auth information for a TLS authenticated connection.
|
||||
// It implements the AuthInfo interface.
|
||||
type TLSInfo struct {
|
||||
state tls.ConnectionState
|
||||
State tls.ConnectionState
|
||||
}
|
||||
|
||||
func (t TLSInfo) AuthType() string {
|
||||
|
|
|
@ -117,16 +117,20 @@ func (s *testServer) UnaryCall(ctx context.Context, in *testpb.SimpleRequest) (*
|
|||
if !ok {
|
||||
grpclog.Fatalf("Failed to get AuthInfo from ctx.")
|
||||
}
|
||||
var authType string
|
||||
var authType, serverName string
|
||||
switch info := authInfo.(type) {
|
||||
case credentials.TLSInfo:
|
||||
authType = info.AuthType()
|
||||
serverName = info.State.ServerName
|
||||
default:
|
||||
grpclog.Fatalf("Unknown AuthInfo type")
|
||||
}
|
||||
if authType != s.security {
|
||||
grpclog.Fatalf("Wrong auth type: got %q, want %q", authType, s.security)
|
||||
}
|
||||
if serverName != "x.test.youtube.com" {
|
||||
return nil, fmt.Errorf("Unknown server name %q", serverName)
|
||||
}
|
||||
}
|
||||
|
||||
// Simulate some service delay.
|
||||
|
|
Loading…
Reference in New Issue