mirror of
https://github.com/hanzoai/deploy.git
synced 2026-10-07 08:44:50 +00:00
go.mod still declared `module github.com/hanzoai/deploy` after the repo was renamed. A Go module path is not cosmetic — it is the resolvable location. So `go get github.com/hanzoai/cd` could not work, and anything importing this module had to spell a name that no longer exists at that address. The build log of v3.6.0 is where it surfaced: the compiler was still emitting github.com/hanzoai/deploy/cmd/cd/commands. 3541 import paths across 690 .go, 19 .proto, 4 .yaml, 3 .sh, 2 .mod, Procfile and Makefile. Plus the 62 ghcr.io/hanzoai/deploy image references, moved to ghcr.io/hanzoai/cd to match hanzo.yml's build target — a distinct string from the module path, replaced separately so neither could clobber the other. go mod tidy clean, CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build ./... clean. Co-authored-by: Hanzo Dev <[email protected]>
640 lines
21 KiB
Go
640 lines
21 KiB
Go
package apiclient_test
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"crypto/x509"
|
|
"net"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/connectivity"
|
|
"google.golang.org/grpc/credentials"
|
|
"google.golang.org/grpc/health"
|
|
"google.golang.org/grpc/health/grpc_health_v1"
|
|
"google.golang.org/grpc/status"
|
|
|
|
utilstls "github.com/hanzoai/cd/util/tls"
|
|
|
|
"github.com/hanzoai/cd/reposerver/apiclient"
|
|
"github.com/hanzoai/cd/reposerver/apiclient/mocks"
|
|
)
|
|
|
|
func TestNewRepoServerClient_CorrectClientReturned(t *testing.T) {
|
|
t.Parallel()
|
|
mockClientset := &mocks.Clientset{
|
|
RepoServerServiceClient: &mocks.RepoServerServiceClient{},
|
|
}
|
|
|
|
closer, client, err := mockClientset.NewRepoServerClient()
|
|
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, closer)
|
|
assert.NotNil(t, client)
|
|
assert.Equal(t, mockClientset.RepoServerServiceClient, client)
|
|
}
|
|
|
|
func TestNewRepoServerClientset_InvalidInput(t *testing.T) {
|
|
t.Parallel()
|
|
// Call the function with invalid inputs
|
|
invalidClientset := apiclient.NewRepoServerClientset("", -1, utilstls.Configuration{})
|
|
|
|
assert.NotNil(t, invalidClientset)
|
|
assert.Implements(t, (*apiclient.Clientset)(nil), invalidClientset)
|
|
}
|
|
|
|
func TestNewRepoServerClientset_SuccessfulConnection(t *testing.T) {
|
|
t.Parallel()
|
|
// Call the function with valid inputs
|
|
clientset := apiclient.NewRepoServerClientset("localhost:8080", 1, utilstls.Configuration{})
|
|
|
|
assert.NotNil(t, clientset)
|
|
assert.Implements(t, (*apiclient.Clientset)(nil), clientset)
|
|
}
|
|
|
|
func TestNewRepoServerClientset_SuccessfulConnectionWithTLS(t *testing.T) {
|
|
t.Parallel()
|
|
// Call the function with valid inputs
|
|
clientset := apiclient.NewRepoServerClientset("localhost:8080", 1, utilstls.Configuration{
|
|
DisableTLS: false,
|
|
StrictValidation: true,
|
|
Certificates: nil,
|
|
})
|
|
|
|
assert.NotNil(t, clientset)
|
|
assert.Implements(t, (*apiclient.Clientset)(nil), clientset)
|
|
}
|
|
|
|
func TestNewConnection_TLSWithStrictValidation(t *testing.T) {
|
|
t.Parallel()
|
|
tlsConfig := utilstls.Configuration{
|
|
DisableTLS: false,
|
|
StrictValidation: true,
|
|
Certificates: nil,
|
|
}
|
|
|
|
conn, err := apiclient.NewConnection("example.com:443", 10, &tlsConfig)
|
|
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, conn)
|
|
}
|
|
|
|
func TestNewConnection_TLSWithStrictValidationAndCertificates(t *testing.T) {
|
|
t.Parallel()
|
|
tlsConfig := utilstls.Configuration{
|
|
DisableTLS: false,
|
|
StrictValidation: true,
|
|
Certificates: nil,
|
|
}
|
|
|
|
conn, err := apiclient.NewConnection("example.com:443", 10, &tlsConfig)
|
|
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, conn)
|
|
}
|
|
|
|
func TestNewConnection_InsecureConnection(t *testing.T) {
|
|
t.Parallel()
|
|
// Create a TLS configuration with TLS disabled
|
|
tlsConfig := utilstls.Configuration{
|
|
DisableTLS: true,
|
|
StrictValidation: false,
|
|
Certificates: nil,
|
|
}
|
|
|
|
conn, err := apiclient.NewConnection("example.com:80", 10, &tlsConfig)
|
|
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, conn)
|
|
}
|
|
|
|
func TestNewConnection_InMemoryCertificates(t *testing.T) {
|
|
tlsConfig := utilstls.Configuration{
|
|
DisableTLS: false,
|
|
StrictValidation: false,
|
|
ClientCertificates: []tls.Certificate{{}},
|
|
}
|
|
|
|
conn, err := apiclient.NewConnection("example.com:443", 10, &tlsConfig)
|
|
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, conn)
|
|
}
|
|
|
|
func TestNewConnection_ServesFromCacheWhenFilesUnchanged(t *testing.T) {
|
|
t.Cleanup(apiclient.ResetClientCertCache)
|
|
apiclient.ResetClientCertCache()
|
|
|
|
certFile := filepath.Join("..", "..", "util", "tls", "testdata", "valid_tls.crt")
|
|
keyFile := filepath.Join("..", "..", "util", "tls", "testdata", "valid_tls.key")
|
|
|
|
tempCertFile := filepath.Join(t.TempDir(), "client.crt")
|
|
tempKeyFile := filepath.Join(t.TempDir(), "client.key")
|
|
|
|
certBytes, err := os.ReadFile(certFile)
|
|
require.NoError(t, err)
|
|
require.NoError(t, os.WriteFile(tempCertFile, certBytes, 0o600))
|
|
|
|
keyBytes, err := os.ReadFile(keyFile)
|
|
require.NoError(t, err)
|
|
require.NoError(t, os.WriteFile(tempKeyFile, keyBytes, 0o600))
|
|
|
|
tlsConfig := utilstls.Configuration{
|
|
StrictValidation: false,
|
|
ClientCertFile: tempCertFile,
|
|
ClientCertKeyFile: tempKeyFile,
|
|
}
|
|
|
|
// First call: loads from disk and populates the cache.
|
|
conn, err := apiclient.NewConnection("example.com:443", 10, &tlsConfig)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, conn)
|
|
|
|
// Second call: files are untouched, mtime is unchanged — cert must be served from the cache without re-reading disk.
|
|
conn, err = apiclient.NewConnection("example.com:443", 10, &tlsConfig)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, conn)
|
|
}
|
|
|
|
func TestNewConnection_ReloadsClientCertificateWhenFileChanges(t *testing.T) {
|
|
t.Cleanup(apiclient.ResetClientCertCache)
|
|
apiclient.ResetClientCertCache()
|
|
|
|
certFile := filepath.Join("..", "..", "util", "tls", "testdata", "valid_tls.crt")
|
|
keyFile := filepath.Join("..", "..", "util", "tls", "testdata", "valid_tls.key")
|
|
|
|
tempCertFile := filepath.Join(t.TempDir(), "client.crt")
|
|
tempKeyFile := filepath.Join(t.TempDir(), "client.key")
|
|
|
|
certBytes, err := os.ReadFile(certFile)
|
|
require.NoError(t, err)
|
|
require.NoError(t, os.WriteFile(tempCertFile, certBytes, 0o600))
|
|
|
|
keyBytes, err := os.ReadFile(keyFile)
|
|
require.NoError(t, err)
|
|
require.NoError(t, os.WriteFile(tempKeyFile, keyBytes, 0o600))
|
|
|
|
tlsConfig := utilstls.Configuration{
|
|
StrictValidation: false,
|
|
ClientCertFile: tempCertFile,
|
|
ClientCertKeyFile: tempKeyFile,
|
|
}
|
|
|
|
// First call: loads and caches the cert.
|
|
conn, err := apiclient.NewConnection("example.com:443", 10, &tlsConfig)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, conn)
|
|
|
|
// Simulate cert rotation: overwrite with fresh (still valid) cert bytes.
|
|
// Writing updates the mtime, which is enough to trigger a reload.
|
|
require.NoError(t, os.WriteFile(tempCertFile, certBytes, 0o600))
|
|
require.NoError(t, os.WriteFile(tempKeyFile, keyBytes, 0o600))
|
|
|
|
// Second call: mtime changed → cache invalidated → reloads from disk.
|
|
// The new cert is valid, so the call must succeed.
|
|
conn, err = apiclient.NewConnection("example.com:443", 10, &tlsConfig)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, conn)
|
|
}
|
|
|
|
func TestNewConnection_ErrorWhenRotatedCertIsInvalid(t *testing.T) {
|
|
t.Cleanup(apiclient.ResetClientCertCache)
|
|
apiclient.ResetClientCertCache()
|
|
|
|
certFile := filepath.Join("..", "..", "util", "tls", "testdata", "valid_tls.crt")
|
|
keyFile := filepath.Join("..", "..", "util", "tls", "testdata", "valid_tls.key")
|
|
|
|
tempCertFile := filepath.Join(t.TempDir(), "client.crt")
|
|
tempKeyFile := filepath.Join(t.TempDir(), "client.key")
|
|
|
|
certBytes, err := os.ReadFile(certFile)
|
|
require.NoError(t, err)
|
|
require.NoError(t, os.WriteFile(tempCertFile, certBytes, 0o600))
|
|
|
|
keyBytes, err := os.ReadFile(keyFile)
|
|
require.NoError(t, err)
|
|
require.NoError(t, os.WriteFile(tempKeyFile, keyBytes, 0o600))
|
|
|
|
tlsConfig := utilstls.Configuration{
|
|
StrictValidation: false,
|
|
ClientCertFile: tempCertFile,
|
|
ClientCertKeyFile: tempKeyFile,
|
|
}
|
|
|
|
// First call: succeeds and populates the cache.
|
|
conn, err := apiclient.NewConnection("example.com:443", 10, &tlsConfig)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, conn)
|
|
|
|
require.NoError(t, os.WriteFile(tempCertFile, []byte("not a valid certificate"), 0o600))
|
|
require.NoError(t, os.WriteFile(tempKeyFile, []byte("not a valid key"), 0o600))
|
|
|
|
// Second call: mtime changed → reload attempted → parse fails → error returned.
|
|
// The stale (valid) cached cert must NOT be silently served.
|
|
_, err = apiclient.NewConnection("example.com:443", 10, &tlsConfig)
|
|
require.Error(t, err,
|
|
"a corrupt cert file must cause an error after cache invalidation, "+
|
|
"not silently serve the stale cached cert")
|
|
assert.Contains(t, err.Error(), "failed to load client certificate",
|
|
"error must identify the cert load as the failure site")
|
|
}
|
|
|
|
func TestNewConnection_DoesNotCacheClientCertificateLoadError(t *testing.T) {
|
|
t.Cleanup(apiclient.ResetClientCertCache)
|
|
apiclient.ResetClientCertCache()
|
|
|
|
tempDir := t.TempDir()
|
|
tempCertFile := filepath.Join(tempDir, "client.crt")
|
|
tempKeyFile := filepath.Join(tempDir, "client.key")
|
|
|
|
tlsConfig := utilstls.Configuration{
|
|
StrictValidation: false,
|
|
ClientCertFile: tempCertFile,
|
|
ClientCertKeyFile: tempKeyFile,
|
|
}
|
|
|
|
conn, err := apiclient.NewConnection("example.com:443", 10, &tlsConfig)
|
|
require.NoError(t, err, "missing cert file must not cause an error — mTLS is skipped gracefully")
|
|
assert.NotNil(t, conn)
|
|
|
|
certFile := filepath.Join("..", "..", "util", "tls", "testdata", "valid_tls.crt")
|
|
keyFile := filepath.Join("..", "..", "util", "tls", "testdata", "valid_tls.key")
|
|
|
|
certBytes, err := os.ReadFile(certFile)
|
|
require.NoError(t, err)
|
|
require.NoError(t, os.WriteFile(tempCertFile, certBytes, 0o600))
|
|
|
|
keyBytes, err := os.ReadFile(keyFile)
|
|
require.NoError(t, err)
|
|
require.NoError(t, os.WriteFile(tempKeyFile, keyBytes, 0o600))
|
|
|
|
conn, err = apiclient.NewConnection("example.com:443", 10, &tlsConfig)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, conn)
|
|
}
|
|
|
|
func TestMTLSIntegration_NoClientCert_IsRejected(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
_, parsedClientCert := generateClientCA(t, "test-client")
|
|
|
|
pool := x509.NewCertPool()
|
|
pool.AddCert(parsedClientCert)
|
|
fixture := newMTLSServer(t, pool)
|
|
|
|
conn, err := apiclient.NewConnection(fixture.addr, 5, &utilstls.Configuration{
|
|
StrictValidation: false,
|
|
})
|
|
require.NoError(t, err, "NewConnection is lazy — dial error surfaces on first RPC")
|
|
defer func(conn *grpc.ClientConn) {
|
|
err := conn.Close()
|
|
if err != nil {
|
|
t.Logf("failed to close connection: %v", err)
|
|
}
|
|
}(conn)
|
|
|
|
ctx, cancel := context.WithTimeout(t.Context(), 5*time.Second)
|
|
defer cancel()
|
|
conn.Connect()
|
|
for {
|
|
st := conn.GetState()
|
|
if st == connectivity.Ready || st == connectivity.TransientFailure || st == connectivity.Shutdown {
|
|
break
|
|
}
|
|
if !conn.WaitForStateChange(ctx, st) {
|
|
break
|
|
}
|
|
}
|
|
|
|
err = healthCheck(t, conn)
|
|
require.Error(t, err, "the server enforces mTLS; a client without a certificate must be rejected")
|
|
// gRPC may surface the TLS alert as "certificate required", as a transport-level
|
|
// "broken pipe" when the server closes the connection after detecting the missing cert,
|
|
// or as "connection reset by peer" when the server sends a TCP RST before completing
|
|
// the TLS handshake. All are valid indicators of an mTLS rejection.
|
|
assert.True(t,
|
|
strings.Contains(err.Error(), "certificate") || strings.Contains(err.Error(), "broken pipe") || strings.Contains(err.Error(), "connection reset by peer"),
|
|
"rejection must be a TLS-related error (certificate, broken pipe, or connection reset by peer), got: %v", err,
|
|
)
|
|
}
|
|
|
|
func TestMTLSIntegration_ValidClientCert_IsAccepted(t *testing.T) {
|
|
t.Parallel()
|
|
t.Cleanup(apiclient.ResetClientCertCache)
|
|
|
|
clientCert, parsedClientCert := generateClientCA(t, "test-client")
|
|
|
|
pool := x509.NewCertPool()
|
|
pool.AddCert(parsedClientCert)
|
|
fixture := newMTLSServer(t, pool)
|
|
|
|
conn, err := apiclient.NewConnection(fixture.addr, 5, &utilstls.Configuration{
|
|
StrictValidation: false,
|
|
ClientCertificates: []tls.Certificate{*clientCert},
|
|
})
|
|
require.NoError(t, err)
|
|
defer func(conn *grpc.ClientConn) {
|
|
err := conn.Close()
|
|
if err != nil {
|
|
t.Logf("failed to close connection: %v", err)
|
|
}
|
|
}(conn)
|
|
|
|
require.NoError(t, healthCheck(t, conn), "client presenting a cert trusted by the server's ClientCAs must succeed")
|
|
}
|
|
|
|
func TestMTLSIntegration_UntrustedClientCert_IsRejected(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
_, trustedParsed := generateClientCA(t, "trusted-ca")
|
|
trustedPool := x509.NewCertPool()
|
|
trustedPool.AddCert(trustedParsed)
|
|
fixture := newMTLSServer(t, trustedPool)
|
|
|
|
untrustedCert, _ := generateClientCA(t, "untrusted-client")
|
|
|
|
conn, err := apiclient.NewConnection(fixture.addr, 5, &utilstls.Configuration{
|
|
StrictValidation: false,
|
|
ClientCertificates: []tls.Certificate{*untrustedCert},
|
|
})
|
|
require.NoError(t, err, "NewConnection is lazy")
|
|
defer func(conn *grpc.ClientConn) {
|
|
err := conn.Close()
|
|
if err != nil {
|
|
t.Logf("failed to close connection: %v", err)
|
|
}
|
|
}(conn)
|
|
|
|
require.Error(t, healthCheck(t, conn), "a cert signed by an unknown CA must be rejected by the server")
|
|
}
|
|
|
|
func TestMTLSIntegration_HealthCheckEphemeralCert_IsAccepted(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
hcCert, err := utilstls.GenerateHealthCheckClientCert()
|
|
require.NoError(t, err, "GenerateHealthCheckClientCert")
|
|
|
|
parsedHCCert, err := x509.ParseCertificate(hcCert.Certificate[0])
|
|
require.NoError(t, err)
|
|
|
|
pool := x509.NewCertPool()
|
|
pool.AddCert(parsedHCCert)
|
|
fixture := newMTLSServer(t, pool)
|
|
|
|
conn, err := apiclient.NewConnection(fixture.addr, 5, &utilstls.Configuration{
|
|
StrictValidation: false, // liveness probe skips server cert verification
|
|
ClientCertificates: []tls.Certificate{*hcCert},
|
|
})
|
|
require.NoError(t, err)
|
|
defer func(conn *grpc.ClientConn) {
|
|
err := conn.Close()
|
|
if err != nil {
|
|
t.Logf("failed to close connection: %v", err)
|
|
}
|
|
}(conn)
|
|
|
|
require.NoError(t, healthCheck(t, conn), "liveness probe using the ephemeral health-check cert must be accepted")
|
|
}
|
|
|
|
// TestMTLSIntegration_DisableTLS_PlaintextConnection verifies that a server
|
|
// running without TLS accepts a plaintext client, establishing the baseline
|
|
// that DisableTLS=true on the client side still works after this PR.
|
|
func TestMTLSIntegration_DisableTLS_PlaintextConnection(t *testing.T) {
|
|
t.Parallel()
|
|
lis, err := new(net.ListenConfig).Listen(t.Context(), "tcp", "127.0.0.1:0")
|
|
require.NoError(t, err)
|
|
|
|
srv := grpc.NewServer() // no TLS credentials
|
|
hsvc := health.NewServer()
|
|
hsvc.SetServingStatus("", grpc_health_v1.HealthCheckResponse_SERVING)
|
|
grpc_health_v1.RegisterHealthServer(srv, hsvc)
|
|
go func() { _ = srv.Serve(lis) }()
|
|
t.Cleanup(srv.Stop)
|
|
|
|
conn, err := apiclient.NewConnection(lis.Addr().String(), 5, &utilstls.Configuration{
|
|
DisableTLS: true,
|
|
})
|
|
require.NoError(t, err)
|
|
defer conn.Close()
|
|
|
|
err = healthCheck(t, conn)
|
|
require.NoError(t, err, "plaintext connection to a plaintext server must succeed")
|
|
}
|
|
|
|
// TestMTLSIntegration_StrictValidation_CertificatesPoolTakesPrecedence
|
|
// exercises the buildTLSClientConfig fix: when Certificates is non-nil,
|
|
// InsecureSkipVerify must be false even if StrictValidation=false in the
|
|
// struct. This is the `strictValidation := tlsConfig.StrictValidation ||
|
|
// tlsConfig.Certificates != nil` guard in buildTLSClientConfig.
|
|
//
|
|
// The test verifies observable behaviour: the connection succeeds only when
|
|
// the client presents the correct CA pool (i.e. server cert verification is
|
|
// actually being performed, not skipped).
|
|
func TestMTLSIntegration_StrictValidation_CertificatesPoolTakesPrecedence(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
clientCert, parsedClientCert := generateClientCA(t, "test-client-strict")
|
|
clientPool := x509.NewCertPool()
|
|
clientPool.AddCert(parsedClientCert)
|
|
fixture := newMTLSServer(t, clientPool)
|
|
|
|
// We need the client to verify the server's certificate, so we must tell
|
|
// it which CA signed the server cert. The server is self-signed, so we
|
|
// extract its cert and build a server-CA pool from it.
|
|
serverTLSState, err := extractServerCert(t, fixture.addr, clientCert)
|
|
require.NoError(t, err)
|
|
serverCAParsed, err := x509.ParseCertificate(serverTLSState.Certificate[0])
|
|
require.NoError(t, err)
|
|
serverCAPool := x509.NewCertPool()
|
|
serverCAPool.AddCert(serverCAParsed)
|
|
|
|
// StrictValidation=false BUT Certificates is non-nil.
|
|
// buildTLSClientConfig must treat this as strict=true (the fix).
|
|
conn, err := apiclient.NewConnection(fixture.addr, 5, &utilstls.Configuration{
|
|
StrictValidation: false, // struct field says "not strict" …
|
|
Certificates: serverCAPool, // … but a pool is present, so strict wins
|
|
ClientCertificates: []tls.Certificate{*clientCert},
|
|
})
|
|
require.NoError(t, err)
|
|
defer conn.Close()
|
|
|
|
err = healthCheck(t, conn)
|
|
require.NoError(t, err,
|
|
"when Certificates pool is set, server cert must be verified and the "+
|
|
"connection must succeed when the pool contains the server's CA")
|
|
}
|
|
|
|
func TestNewConnection_ClientCertFileDoesNotExist_SkipsMTLS(t *testing.T) {
|
|
t.Cleanup(apiclient.ResetClientCertCache)
|
|
apiclient.ResetClientCertCache()
|
|
|
|
tlsConfig := utilstls.Configuration{
|
|
StrictValidation: false,
|
|
ClientCertFile: "/nonexistent/path/client.crt",
|
|
ClientCertKeyFile: "/nonexistent/path/client.key",
|
|
}
|
|
|
|
conn, err := apiclient.NewConnection("example.com:443", 10, &tlsConfig)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, conn)
|
|
}
|
|
|
|
func TestNewConnection_ClientCertFileStatNonErrNotExist_ReturnsError(t *testing.T) {
|
|
t.Cleanup(apiclient.ResetClientCertCache)
|
|
apiclient.ResetClientCertCache()
|
|
|
|
dir := t.TempDir()
|
|
require.NoError(t, os.Chmod(dir, 0o000))
|
|
t.Cleanup(func() { _ = os.Chmod(dir, 0o700) })
|
|
|
|
tlsConfig := utilstls.Configuration{
|
|
StrictValidation: false,
|
|
ClientCertFile: filepath.Join(dir, "client.crt"),
|
|
ClientCertKeyFile: filepath.Join(dir, "client.key"),
|
|
}
|
|
|
|
_, err := apiclient.NewConnection("example.com:443", 10, &tlsConfig)
|
|
require.Error(t, err)
|
|
assert.Contains(t, err.Error(), "could not stat client cert file")
|
|
}
|
|
|
|
func TestNewConnection_ClientCertCallbackReturnsEmptyCertWhenFileDisappears(t *testing.T) {
|
|
t.Cleanup(apiclient.ResetClientCertCache)
|
|
apiclient.ResetClientCertCache()
|
|
|
|
srcCert := filepath.Join("..", "..", "util", "tls", "testdata", "valid_tls.crt")
|
|
srcKey := filepath.Join("..", "..", "util", "tls", "testdata", "valid_tls.key")
|
|
|
|
dir := t.TempDir()
|
|
tempCert := filepath.Join(dir, "client.crt")
|
|
tempKey := filepath.Join(dir, "client.key")
|
|
|
|
certBytes, err := os.ReadFile(srcCert)
|
|
require.NoError(t, err)
|
|
require.NoError(t, os.WriteFile(tempCert, certBytes, 0o600))
|
|
|
|
keyBytes, err := os.ReadFile(srcKey)
|
|
require.NoError(t, err)
|
|
require.NoError(t, os.WriteFile(tempKey, keyBytes, 0o600))
|
|
|
|
tlsConfig := utilstls.Configuration{
|
|
StrictValidation: false,
|
|
ClientCertFile: tempCert,
|
|
ClientCertKeyFile: tempKey,
|
|
}
|
|
|
|
conn, err := apiclient.NewConnection("example.com:443", 10, &tlsConfig)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, conn)
|
|
|
|
require.NoError(t, os.Remove(tempCert))
|
|
require.NoError(t, os.Remove(tempKey))
|
|
|
|
apiclient.ResetClientCertCache()
|
|
conn, err = apiclient.NewConnection("example.com:443", 10, &tlsConfig)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, conn)
|
|
}
|
|
|
|
func extractServerCert(t *testing.T, addr string, clientCert *tls.Certificate) (*tls.Certificate, error) {
|
|
t.Helper()
|
|
cfg := &tls.Config{
|
|
InsecureSkipVerify: true, //nolint:gosec // intentional: only used to fetch the cert bytes
|
|
Certificates: []tls.Certificate{*clientCert},
|
|
}
|
|
d := &tls.Dialer{Config: cfg}
|
|
ctx, cancel := context.WithTimeout(t.Context(), 5*time.Second)
|
|
defer cancel()
|
|
conn, err := d.DialContext(ctx, "tcp", addr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer conn.Close()
|
|
tlsConnection, ok := conn.(*tls.Conn)
|
|
if !ok {
|
|
t.Fatal("expected *tls.Conn from tls.Dialer")
|
|
}
|
|
state := tlsConnection.ConnectionState()
|
|
if len(state.PeerCertificates) == 0 {
|
|
t.Fatal("server sent no certificate")
|
|
}
|
|
raw := state.PeerCertificates[0].Raw
|
|
tlsCert := &tls.Certificate{Certificate: [][]byte{raw}}
|
|
return tlsCert, nil
|
|
}
|
|
|
|
type mTLSServerFixture struct {
|
|
addr string
|
|
server *grpc.Server //nolint:unused // for some reason this linter complains
|
|
}
|
|
|
|
func newMTLSServer(t *testing.T, clientCAs *x509.CertPool) *mTLSServerFixture {
|
|
t.Helper()
|
|
|
|
serverCert, err := utilstls.GenerateX509KeyPair(utilstls.CertOptions{
|
|
Hosts: []string{"localhost", "127.0.0.1"},
|
|
Organization: "Argo CD Integration Test Server",
|
|
IsCA: false,
|
|
ECDSACurve: "P256",
|
|
ValidFor: time.Hour,
|
|
})
|
|
require.NoError(t, err, "generating server certificate")
|
|
|
|
serverTLSCfg := &tls.Config{
|
|
Certificates: []tls.Certificate{*serverCert},
|
|
ClientCAs: clientCAs,
|
|
ClientAuth: tls.RequireAndVerifyClientCert,
|
|
MinVersion: tls.VersionTLS12,
|
|
}
|
|
|
|
lis, err := new(net.ListenConfig).Listen(t.Context(), "tcp", "127.0.0.1:0")
|
|
require.NoError(t, err, "binding test listener")
|
|
|
|
srv := grpc.NewServer(grpc.Creds(credentials.NewTLS(serverTLSCfg)))
|
|
hsvc := health.NewServer()
|
|
hsvc.SetServingStatus("", grpc_health_v1.HealthCheckResponse_SERVING)
|
|
grpc_health_v1.RegisterHealthServer(srv, hsvc)
|
|
|
|
go func() { _ = srv.Serve(lis) }()
|
|
t.Cleanup(srv.Stop)
|
|
|
|
return &mTLSServerFixture{addr: lis.Addr().String(), server: srv}
|
|
}
|
|
|
|
func healthCheck(t *testing.T, conn *grpc.ClientConn) error {
|
|
t.Helper()
|
|
ctx, cancel := context.WithTimeout(t.Context(), 5*time.Second)
|
|
defer cancel()
|
|
resp, err := grpc_health_v1.NewHealthClient(conn).Check(ctx, &grpc_health_v1.HealthCheckRequest{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if resp.GetStatus() != grpc_health_v1.HealthCheckResponse_SERVING {
|
|
return status.Errorf(codes.Unavailable, "server status: %v", resp.GetStatus())
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func generateClientCA(t *testing.T, cn string) (*tls.Certificate, *x509.Certificate) {
|
|
t.Helper()
|
|
cert, err := utilstls.GenerateX509KeyPair(utilstls.CertOptions{
|
|
Hosts: []string{cn},
|
|
Organization: "Argo CD Integration Test",
|
|
IsCA: true,
|
|
ECDSACurve: "P256",
|
|
ValidFor: time.Hour,
|
|
ExtKeyUsage: []x509.ExtKeyUsage{x509.ExtKeyUsageClientAuth},
|
|
})
|
|
require.NoError(t, err)
|
|
parsed, err := x509.ParseCertificate(cert.Certificate[0])
|
|
require.NoError(t, err)
|
|
return cert, parsed
|
|
}
|