mirror of
https://github.com/gravitational/teleport.git
synced 2026-09-24 16:17:11 +08:00
105 lines
3.2 KiB
Go
105 lines
3.2 KiB
Go
/*
|
|
Copyright 2021 Gravitational, Inc.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
// Package tlsutils contains utilities for TLS configuration and formats.
|
|
package tlsutils
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"crypto/x509"
|
|
"encoding/pem"
|
|
"net"
|
|
"strings"
|
|
|
|
"github.com/gravitational/trace"
|
|
)
|
|
|
|
// ParseCertificatePEMStrict parses a PEM-encoded x509 certificate.
|
|
//
|
|
// It is a strict variant of [ParseCertificatePEM], disallowing trailing data in
|
|
// the PEM block and requiring the block type to be CERTIFICATE.
|
|
func ParseCertificatePEMStrict(bytes []byte) (*x509.Certificate, error) {
|
|
return parseCertificatePEM(bytes, true /* strict */)
|
|
}
|
|
|
|
// ParseCertificatePEM parses a PEM-encoded x509 certificate.
|
|
func ParseCertificatePEM(bytes []byte) (*x509.Certificate, error) {
|
|
return parseCertificatePEM(bytes, false /* strict */)
|
|
}
|
|
|
|
func parseCertificatePEM(bytes []byte, strict bool) (*x509.Certificate, error) {
|
|
block, rest := pem.Decode(bytes)
|
|
const wantBlockType = "CERTIFICATE"
|
|
switch {
|
|
case block == nil:
|
|
return nil, trace.BadParameter("expected PEM-encoded block")
|
|
case strict && block.Type != wantBlockType:
|
|
return nil, trace.BadParameter("certificate PEM has unexpected block type %q (want %q)", block.Type, wantBlockType)
|
|
case strict && len(rest) > 0:
|
|
return nil, trace.BadParameter("certificate PEM has unexpected trailing data")
|
|
}
|
|
|
|
cert, err := x509.ParseCertificate(block.Bytes)
|
|
if err != nil {
|
|
return nil, trace.BadParameter("%s", err)
|
|
}
|
|
return cert, nil
|
|
}
|
|
|
|
// ContextDialer represents network dialer interface that uses context
|
|
type ContextDialer interface {
|
|
// DialContext is a function that dials the specified address
|
|
DialContext(ctx context.Context, network, addr string) (net.Conn, error)
|
|
}
|
|
|
|
// TLSDial dials and establishes TLS connection using custom dialer
|
|
// is similar to tls.DialWithDialer
|
|
// Note: function taken from lib/utils/tlsdial.go
|
|
func TLSDial(ctx context.Context, dialer ContextDialer, network, addr string, tlsConfig *tls.Config) (*tls.Conn, error) {
|
|
if tlsConfig == nil {
|
|
return nil, trace.BadParameter("tls config must be specified")
|
|
}
|
|
|
|
plainConn, err := dialer.DialContext(ctx, network, addr)
|
|
if err != nil {
|
|
return nil, trace.Wrap(err)
|
|
}
|
|
|
|
colonPos := strings.LastIndex(addr, ":")
|
|
if colonPos == -1 {
|
|
colonPos = len(addr)
|
|
}
|
|
hostname := addr[:colonPos]
|
|
|
|
// If no ServerName is set, infer the ServerName
|
|
// from the hostname we're connecting to.
|
|
if tlsConfig.ServerName == "" {
|
|
// Make a copy to avoid polluting argument or default.
|
|
tlsConfig = tlsConfig.Clone()
|
|
tlsConfig.ServerName = hostname
|
|
}
|
|
|
|
conn := tls.Client(plainConn, tlsConfig)
|
|
err = conn.HandshakeContext(ctx)
|
|
if err != nil {
|
|
plainConn.Close()
|
|
return nil, trace.Wrap(err)
|
|
}
|
|
|
|
return conn, nil
|
|
}
|