storj/pkg/rpc/rpcpeer/peer_grpc.go

// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.

package rpcpeer

import (
	"context"

	"google.golang.org/grpc/credentials"
	"google.golang.org/grpc/peer"
)

// grpcInternalFromContext returns a peer from the context using grpc.
func grpcInternalFromContext(ctx context.Context) (*Peer, error) {
	peer, ok := peer.FromContext(ctx)
	if !ok {
		return nil, Error.New("unable to get grpc peer from context")
	}

	tlsInfo, ok := peer.AuthInfo.(credentials.TLSInfo)
	if !ok {
		return nil, Error.New("peer AuthInfo is not credentials.TLSInfo")
	}

	return &Peer{
		Addr:  peer.Addr,
		State: tlsInfo.State,
	}, nil
}