675e0ef683
* small identity refactor: + Optimize? iterative cert chain methods to use array instead of slice + Add `ToChain` helper for converting 1d to 2d cert chain TODO: replace literal declarations with this + rename `ChainRaw/RestChainRaw` to `RawChain/RawRestChain` (adjective noun, instead of nound adjective) * add regression tests for V3-1320 * fix V3-1320 * separate `DialUnverifiedIDOption` from `DialOption` * separate `PingNode` and `DialNode` from `PingAddress` and `DialAddress` * update node ID while bootstrapping * goimports & fix comment * add test case
94 lines
2.6 KiB
Go
94 lines
2.6 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package tlsopts
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"crypto/x509"
|
|
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/credentials"
|
|
|
|
"storj.io/storj/pkg/identity"
|
|
"storj.io/storj/pkg/peertls"
|
|
"storj.io/storj/pkg/storj"
|
|
)
|
|
|
|
// ServerOption returns a grpc `ServerOption` for incoming connections
|
|
// to the node with this full identity.
|
|
func (opts *Options) ServerOption() grpc.ServerOption {
|
|
pcvFuncs := append(
|
|
[]peertls.PeerCertVerificationFunc{
|
|
peertls.VerifyPeerCertChains,
|
|
},
|
|
opts.PCVFuncs...,
|
|
)
|
|
tlsConfig := &tls.Config{
|
|
Certificates: []tls.Certificate{*opts.Cert},
|
|
InsecureSkipVerify: true,
|
|
ClientAuth: tls.RequireAnyClientCert,
|
|
VerifyPeerCertificate: peertls.VerifyPeerFunc(
|
|
pcvFuncs...,
|
|
),
|
|
}
|
|
|
|
return grpc.Creds(credentials.NewTLS(tlsConfig))
|
|
}
|
|
|
|
// DialOption returns a grpc `DialOption` for making outgoing connections
|
|
// to the node with this peer identity.
|
|
func (opts *Options) DialOption(id storj.NodeID) (grpc.DialOption, error) {
|
|
if id.IsZero() {
|
|
return nil, Error.New("no ID specified for DialOption")
|
|
}
|
|
return grpc.WithTransportCredentials(opts.TransportCredentials(id)), nil
|
|
}
|
|
|
|
// DialUnverifiedIDOption returns a grpc `DialUnverifiedIDOption`
|
|
func (opts *Options) DialUnverifiedIDOption() grpc.DialOption {
|
|
return grpc.WithTransportCredentials(opts.TransportCredentials(storj.NodeID{}))
|
|
}
|
|
|
|
// TransportCredentials returns a grpc `credentials.TransportCredentials`
|
|
// implementation for use within peertls.
|
|
func (opts *Options) TransportCredentials(id storj.NodeID) credentials.TransportCredentials {
|
|
return credentials.NewTLS(opts.TLSConfig(id))
|
|
}
|
|
|
|
// TLSConfig returns a TSLConfig for use in handshaking with a peer.
|
|
func (opts *Options) TLSConfig(id storj.NodeID) *tls.Config {
|
|
pcvFuncs := append(
|
|
[]peertls.PeerCertVerificationFunc{
|
|
peertls.VerifyPeerCertChains,
|
|
},
|
|
opts.PCVFuncs...,
|
|
)
|
|
if !id.IsZero() {
|
|
pcvFuncs = append(pcvFuncs, verifyIdentity(id))
|
|
}
|
|
return &tls.Config{
|
|
Certificates: []tls.Certificate{*opts.Cert},
|
|
InsecureSkipVerify: true,
|
|
VerifyPeerCertificate: peertls.VerifyPeerFunc(
|
|
pcvFuncs...,
|
|
),
|
|
}
|
|
}
|
|
|
|
func verifyIdentity(id storj.NodeID) peertls.PeerCertVerificationFunc {
|
|
return func(_ [][]byte, parsedChains [][]*x509.Certificate) (err error) {
|
|
defer mon.TaskNamed("verifyIdentity")(nil)(&err)
|
|
peer, err := identity.PeerIdentityFromCerts(parsedChains[0][0], parsedChains[0][1], parsedChains[0][2:])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if peer.ID.String() != id.String() {
|
|
return Error.New("peer ID did not match requested ID")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|