release-21.2: rpc,security: use the tenant client cert for pod-pod communication

pkg/ccl/serverccl/admin_test.go

@@ -19,6 +19,7 @@ import (
 	"github.com/cockroachdb/cockroach/pkg/testutils/serverutils"
 	"github.com/cockroachdb/cockroach/pkg/testutils/sqlutils"
 	"github.com/cockroachdb/cockroach/pkg/util/leaktest"
+	"github.com/cockroachdb/cockroach/pkg/util/log"
 	"github.com/stretchr/testify/require"
 )
 
@@ -28,6 +29,7 @@ var adminPrefix = "/_admin/v1/"
 // that we see all zone configs (#27718).
 func TestAdminAPIDataDistributionPartitioning(t *testing.T) {
 	defer leaktest.AfterTest(t)()
+	defer log.Scope(t).Close(t)
 
 	testCluster := serverutils.StartNewTestCluster(t, 3, base.TestClusterArgs{})
 	defer testCluster.Stopper().Stop(context.Background())
@@ -80,6 +82,7 @@ func TestAdminAPIDataDistributionPartitioning(t *testing.T) {
 // TestAdminAPIChartCatalog verifies that an error doesn't happen.
 func TestAdminAPIChartCatalog(t *testing.T) {
 	defer leaktest.AfterTest(t)()
+	defer log.Scope(t).Close(t)
 
 	testCluster := serverutils.StartNewTestCluster(t, 3, base.TestClusterArgs{})
 	defer testCluster.Stopper().Stop(context.Background())

pkg/ccl/serverccl/role_authentication_test.go

@@ -21,13 +21,15 @@ import (
 	"github.com/cockroachdb/cockroach/pkg/testutils/serverutils"
 	"github.com/cockroachdb/cockroach/pkg/util"
 	"github.com/cockroachdb/cockroach/pkg/util/leaktest"
+	"github.com/cockroachdb/cockroach/pkg/util/log"
 	"github.com/cockroachdb/cockroach/pkg/util/timeutil"
 	"github.com/stretchr/testify/require"
 	"golang.org/x/crypto/bcrypt"
 )
 
 func TestVerifyPassword(t *testing.T) {
 	defer leaktest.AfterTest(t)()
+	defer log.Scope(t).Close(t)
 
 	ctx := context.Background()
 	s, db, _ := serverutils.StartServer(t, base.TestServerArgs{})

pkg/ccl/serverccl/server_sql_test.go

@@ -140,14 +140,15 @@ func TestTenantHTTP(t *testing.T) {
 	tc := serverutils.StartNewTestCluster(t, 1, base.TestClusterArgs{})
 	defer tc.Stopper().Stop(ctx)
 
-	httpClient, err := tc.Server(0).RPCContext().GetHTTPClient()
-	require.NoError(t, err)
-
 	tenant, err := tc.Server(0).StartTenant(ctx,
 		base.TestTenantArgs{
 			TenantID: serverutils.TestTenantID(),
 		})
 	require.NoError(t, err)
+
+	httpClient, err := tenant.RPCContext().GetHTTPClient()
+	require.NoError(t, err)
+
 	t.Run("prometheus", func(t *testing.T) {
 		resp, err := httpClient.Get("https://" + tenant.HTTPAddr() + "/_status/vars")
 		defer http.DefaultClient.CloseIdleConnections()

pkg/ccl/serverccl/tenant_grpc_test.go

@@ -45,9 +45,6 @@ func TestTenantGRPCServices(t *testing.T) {
 
 	server := testCluster.Server(0)
 
-	httpClient, err := server.RPCContext().GetHTTPClient()
-	require.NoError(t, err)
-
 	tenantID := serverutils.TestTenantID()
 	tenant, connTenant := serverutils.StartTenant(t, server, base.TestTenantArgs{
 		TenantID: tenantID,
@@ -59,6 +56,8 @@ func TestTenantGRPCServices(t *testing.T) {
 	})
 	defer connTenant.Close()
 
+	t.Logf("subtests starting")
+
 	t.Run("gRPC is running", func(t *testing.T) {
 		grpcAddr := tenant.SQLAddr()
 		rpcCtx := tenant.RPCContext()
@@ -74,6 +73,9 @@ func TestTenantGRPCServices(t *testing.T) {
 		require.NotEmpty(t, resp.Statements)
 	})
 
+	httpClient, err := tenant.RPCContext().GetHTTPClient()
+	require.NoError(t, err)
+
 	t.Run("gRPC Gateway is running", func(t *testing.T) {
 		resp, err := httpClient.Get("https://" + tenant.HTTPAddr() + "/_status/statements")
 		defer http.DefaultClient.CloseIdleConnections()

pkg/ccl/serverccl/tenant_status_test.go

@@ -131,7 +131,8 @@ func TestTenantCannotSeeNonTenantStats(t *testing.T) {
 	err = serverutils.GetJSONProto(nonTenant, path, &nonTenantCombinedStats)
 	require.NoError(t, err)
 
-	checkStatements := func(tc []testCase, actual *serverpb.StatementsResponse) {
+	checkStatements := func(t *testing.T, tc []testCase, actual *serverpb.StatementsResponse) {
+		t.Helper()
 		var expectedStatements []string
 		for _, stmt := range tc {
 			var expectedStmt = stmt.stmt
@@ -162,38 +163,44 @@ func TestTenantCannotSeeNonTenantStats(t *testing.T) {
 		require.Equal(t, expectedStatements, actualStatements)
 	}
 
-	// First we verify that we have expected stats from tenants
-	checkStatements(testCaseTenant, tenantStats)
-	checkStatements(testCaseTenant, tenantCombinedStats)
+	// First we verify that we have expected stats from tenants.
+	t.Run("tenant-stats", func(t *testing.T) {
+		checkStatements(t, testCaseTenant, tenantStats)
+		checkStatements(t, testCaseTenant, tenantCombinedStats)
+	})
 
 	// Now we verify the non tenant stats are what we expected.
-	checkStatements(testCaseNonTenant, &nonTenantStats)
-	checkStatements(testCaseNonTenant, &nonTenantCombinedStats)
+	t.Run("non-tenant-stats", func(t *testing.T) {
+		checkStatements(t, testCaseNonTenant, &nonTenantStats)
+		checkStatements(t, testCaseNonTenant, &nonTenantCombinedStats)
+	})
 
 	// Now we verify that tenant and non-tenant have no visibility into each other's stats.
-	for _, tenantStmt := range tenantStats.Statements {
-		for _, nonTenantStmt := range nonTenantStats.Statements {
-			require.NotEqual(t, tenantStmt, nonTenantStmt, "expected tenant to have no visibility to non-tenant's statement stats, but found:", nonTenantStmt)
+	t.Run("overlap", func(t *testing.T) {
+		for _, tenantStmt := range tenantStats.Statements {
+			for _, nonTenantStmt := range nonTenantStats.Statements {
+				require.NotEqual(t, tenantStmt, nonTenantStmt, "expected tenant to have no visibility to non-tenant's statement stats, but found:", nonTenantStmt)
+			}
 		}
-	}
 
-	for _, tenantTxn := range tenantStats.Transactions {
-		for _, nonTenantTxn := range nonTenantStats.Transactions {
-			require.NotEqual(t, tenantTxn, nonTenantTxn, "expected tenant to have no visibility to non-tenant's transaction stats, but found:", nonTenantTxn)
+		for _, tenantTxn := range tenantStats.Transactions {
+			for _, nonTenantTxn := range nonTenantStats.Transactions {
+				require.NotEqual(t, tenantTxn, nonTenantTxn, "expected tenant to have no visibility to non-tenant's transaction stats, but found:", nonTenantTxn)
+			}
 		}
-	}
 
-	for _, tenantStmt := range tenantCombinedStats.Statements {
-		for _, nonTenantStmt := range nonTenantCombinedStats.Statements {
-			require.NotEqual(t, tenantStmt, nonTenantStmt, "expected tenant to have no visibility to non-tenant's statement stats, but found:", nonTenantStmt)
+		for _, tenantStmt := range tenantCombinedStats.Statements {
+			for _, nonTenantStmt := range nonTenantCombinedStats.Statements {
+				require.NotEqual(t, tenantStmt, nonTenantStmt, "expected tenant to have no visibility to non-tenant's statement stats, but found:", nonTenantStmt)
+			}
 		}
-	}
 
-	for _, tenantTxn := range tenantCombinedStats.Transactions {
-		for _, nonTenantTxn := range nonTenantCombinedStats.Transactions {
-			require.NotEqual(t, tenantTxn, nonTenantTxn, "expected tenant to have no visibility to non-tenant's transaction stats, but found:", nonTenantTxn)
+		for _, tenantTxn := range tenantCombinedStats.Transactions {
+			for _, nonTenantTxn := range nonTenantCombinedStats.Transactions {
+				require.NotEqual(t, tenantTxn, nonTenantTxn, "expected tenant to have no visibility to non-tenant's transaction stats, but found:", nonTenantTxn)
+			}
 		}
-	}
+	})
 }
 
 func TestResetSQLStatsRPCForTenant(t *testing.T) {

pkg/cli/cert.go

@@ -290,10 +290,10 @@ func runListCerts(cmd *cobra.Command, args []string) error {
 var certCmds = []*cobra.Command{
 	createCACertCmd,
 	createClientCACertCmd,
-	mtCreateTenantClientCACertCmd,
+	mtCreateTenantCACertCmd,
 	createNodeCertCmd,
 	createClientCertCmd,
-	mtCreateTenantClientCertCmd,
+	mtCreateTenantCertCmd,
 	listCertsCmd,
 }
 

pkg/cli/mt.go

@@ -19,8 +19,8 @@ func init() {
 	mtCmd.AddCommand(mtTestDirectorySvr)
 
 	mtCertsCmd.AddCommand(
-		mtCreateTenantClientCACertCmd,
-		mtCreateTenantClientCertCmd,
+		mtCreateTenantCACertCmd,
+		mtCreateTenantCertCmd,
 	)
 
 	mtCmd.AddCommand(mtCertsCmd)

pkg/cli/mt_cert.go

@@ -11,6 +11,7 @@
 package cli
 
 import (
+	"fmt"
 	"strconv"
 
 	"github.com/cockroachdb/cockroach/pkg/cli/clierrorplus"
@@ -19,9 +20,9 @@ import (
 	"github.com/spf13/cobra"
 )
 
-// mtCreateTenantClientCACertCmd generates a tenant CA certificate and stores it
+// mtCreateTenantCACertCmd generates a tenant CA certificate and stores it
 // in the cert directory.
-var mtCreateTenantClientCACertCmd = &cobra.Command{
+var mtCreateTenantCACertCmd = &cobra.Command{
 	Use:   "create-tenant-client-ca --certs-dir=<path to cockroach certs dir> --ca-key=<path>",
 	Short: "create tenant client CA certificate and key",
 	Long: `
@@ -34,7 +35,7 @@ If the CA certificate exists and --overwrite is true, the new CA certificate is
 	Args: cobra.NoArgs,
 	RunE: clierrorplus.MaybeDecorateError(func(cmd *cobra.Command, args []string) error {
 		return errors.Wrap(
-			security.CreateTenantClientCAPair(
+			security.CreateTenantCAPair(
 				certCtx.certsDir,
 				certCtx.caKey,
 				certCtx.keySize,
@@ -47,8 +48,8 @@ If the CA certificate exists and --overwrite is true, the new CA certificate is
 
 // A createClientCert command generates a client certificate and stores it
 // in the cert directory under <username>.crt and key under <username>.key.
-var mtCreateTenantClientCertCmd = &cobra.Command{
-	Use:   "create-tenant-client --certs-dir=<path to cockroach certs dir> --ca-key=<path-to-ca-key> <tenant-id>",
+var mtCreateTenantCertCmd = &cobra.Command{
+	Use:   "create-tenant-client --certs-dir=<path to cockroach certs dir> --ca-key=<path-to-ca-key> <tenant-id> <host 1> <host 2> ... <host N>",
 	Short: "create tenant client certificate and key",
 	Long: `
 Generate a tenant client certificate "<certs-dir>/client-tenant.<tenant-id>.crt" and key
@@ -59,28 +60,53 @@ If --overwrite is true, any existing files are overwritten.
 Requires a CA cert in "<certs-dir>/ca-client-tenant.crt" and matching key in "--ca-key".
 If "ca-client-tenant.crt" contains more than one certificate, the first is used.
 Creation fails if the CA expiration time is before the desired certificate expiration.
+
+If no server addresses are passed, then a default list containing 127.0.0.1, ::1, localhost and *.local is used.
 `,
-	Args: cobra.ExactArgs(1),
+	Args: cobra.MinimumNArgs(1),
 	RunE: clierrorplus.MaybeDecorateError(
 		func(cmd *cobra.Command, args []string) error {
-			tenantID, err := strconv.ParseUint(args[0], 10, 64)
+			tenantIDs := args[0]
+
+			var hostAddrs []string
+			if len(args) > 1 {
+				hostAddrs = args[1:]
+			} else {
+				// Default list.
+				// We need this default because of this CI problem:
+				// https://github.com/cockroachdb/cockroach/issues/71387
+				//
+				// If/when this issue is fixed, the command can be updated to
+				// not provide a default any more (which would be less error
+				// prone.)
+				hostAddrs = []string{
+					"127.0.0.1",
+					"::1",
+					"localhost",
+					"*.local",
+				}
+				fmt.Fprintf(stderr, "Warning: no server address specified. Using %+v.\n", hostAddrs)
+			}
+
+			tenantID, err := strconv.ParseUint(tenantIDs, 10, 64)
 			if err != nil {
-				return errors.Wrapf(err, "%s is invalid uint64", args[0])
+				return errors.Wrapf(err, "%s is invalid uint64", tenantIDs)
 			}
-			cp, err := security.CreateTenantClientPair(
+			cp, err := security.CreateTenantPair(
 				certCtx.certsDir,
 				certCtx.caKey,
 				certCtx.keySize,
 				certCtx.certificateLifetime,
 				tenantID,
+				hostAddrs,
 			)
 			if err != nil {
 				return errors.Wrap(
 					err,
 					"failed to generate tenant client certificate and key")
 			}
 			return errors.Wrap(
-				security.WriteTenantClientPair(certCtx.certsDir, cp, certCtx.overwriteFiles),
+				security.WriteTenantPair(certCtx.certsDir, cp, certCtx.overwriteFiles),
 				"failed to write tenant client certificate and key")
 		}),
 }

pkg/rpc/auth.go

@@ -121,13 +121,39 @@ func (a kvAuth) authenticate(ctx context.Context) (roachpb.TenantID, error) {
 		return roachpb.TenantID{}, err
 	}
 
-	subj := tlsInfo.State.PeerCertificates[0].Subject
-	if security.Contains(subj.OrganizationalUnit, security.TenantsOU) {
-		// Tenant authentication.
-		return tenantFromCommonName(subj.CommonName)
+	clientCert := tlsInfo.State.PeerCertificates[0]
+	if a.tenant.tenantID == roachpb.SystemTenantID {
+		// This node is a KV node.
+		//
+		// Is this a connection from a SQL tenant server?
+		if security.IsTenantCertificate(clientCert) {
+			// Incoming connection originating from a tenant SQL server,
+			// into a KV node.
+			// We extract the tenant ID to perform authorization
+			// of the RPC for this particular tenant.
+			return tenantFromCommonName(clientCert.Subject.CommonName)
+		}
+	} else {
+		// This node is a SQL tenant server.
+		//
+		// Is this a connection from another SQL tenant server?
+		if security.IsTenantCertificate(clientCert) {
+			// Incoming connection originating from a tenant SQL server,
+			// into a KV node. Let through. The other server
+			// is able to use any of this server's RPCs.
+			return roachpb.TenantID{}, nil
+		}
 	}
 
-	// KV auth.
+	// Here we handle the following cases:
+	//
+	// - incoming connection from a RPC admin client into either a KV
+	//   node or a SQL server, using a valid root or node client cert.
+	// - incoming connections from another KV node into a KV node, using
+	//   a node client cert.
+	//
+	// In both cases, we must check that the client cert is either root
+	// or node.
 
 	// TODO(benesch): the vast majority of RPCs should be limited to just
 	// NodeUser. This is not a security concern, as RootUser has access to
@@ -137,5 +163,6 @@ func (a kvAuth) authenticate(ctx context.Context) (roachpb.TenantID, error) {
 		!security.Contains(certUsers, security.RootUser) {
 		return roachpb.TenantID{}, authErrorf("user %s is not allowed to perform this RPC", certUsers)
 	}
+
 	return roachpb.TenantID{}, nil
 }

pkg/rpc/auth_tenant.go

@@ -24,6 +24,8 @@ import (
 // server, that is, it ensures that the request only accesses resources
 // available to the tenant.
 type tenantAuthorizer struct {
+	// tenantID is the tenant ID for the current node.
+	// Equals SystemTenantID when running a KV node.
 	tenantID roachpb.TenantID
 }
 

pkg/rpc/auth_test.go

@@ -127,7 +127,7 @@ func TestTenantFromCert(t *testing.T) {
 			p := peer.Peer{AuthInfo: tlsInfo}
 			ctx := peer.NewContext(context.Background(), &p)
 
-			tenID, err := kvAuth{}.authenticate(ctx)
+			tenID, err := kvAuth{tenant: tenantAuthorizer{tenantID: roachpb.SystemTenantID}}.authenticate(ctx)
 
 			if tc.expErr == "" {
 				require.Equal(t, tc.expTenID, tenID)

pkg/rpc/context.go

@@ -715,7 +715,7 @@ func (ctx *Context) grpcDialOptions(
 		if ctx.tenID == roachpb.SystemTenantID {
 			tlsConfig, err = ctx.GetClientTLSConfig()
 		} else {
-			tlsConfig, err = ctx.GetTenantClientTLSConfig()
+			tlsConfig, err = ctx.GetTenantTLSConfig()
 		}
 		if err != nil {
 			return nil, err