constellation/joinservice/internal/kms/kms.go

74 lines
2.0 KiB
Go
Raw Permalink Normal View History

/*
Copyright (c) Edgeless Systems GmbH
SPDX-License-Identifier: AGPL-3.0-only
*/
// Package kms handles communication with Constellation's key service to request data encryption keys for new or rejoining nodes.
package kms
import (
"context"
"fmt"
2024-02-08 09:20:01 -05:00
"log/slog"
"github.com/edgelesssys/constellation/v2/keyservice/keyserviceproto"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials/insecure"
)
// Client interacts with Constellation's keyservice.
type Client struct {
2024-02-08 09:20:01 -05:00
log *slog.Logger
endpoint string
grpc grpcClient
}
// New creates a new KMS.
2024-02-08 09:20:01 -05:00
func New(log *slog.Logger, endpoint string) Client {
return Client{
log: log,
endpoint: endpoint,
grpc: client{},
}
}
// GetDataKey returns a data encryption key for the given UUID.
func (c Client) GetDataKey(ctx context.Context, keyID string, length int) ([]byte, error) {
2024-02-08 09:20:01 -05:00
log := c.log.With(slog.String("keyID", keyID), slog.String("endpoint", c.endpoint))
// the KMS does not use aTLS since traffic is only routed through the Constellation cluster
// cluster internal connections are considered trustworthy
2024-02-08 09:20:01 -05:00
log.Info(fmt.Sprintf("Connecting to KMS at %s", c.endpoint))
conn, err := grpc.DialContext(ctx, c.endpoint, grpc.WithTransportCredentials(insecure.NewCredentials()))
if err != nil {
return nil, err
}
defer conn.Close()
2024-02-08 09:20:01 -05:00
log.Info("Requesting data key")
res, err := c.grpc.GetDataKey(
ctx,
&keyserviceproto.GetDataKeyRequest{
DataKeyId: keyID,
Length: uint32(length),
},
conn,
)
if err != nil {
return nil, fmt.Errorf("fetching data encryption key from Constellation KMS: %w", err)
}
2024-02-08 09:20:01 -05:00
log.Info("Data key request successful")
return res.DataKey, nil
}
type grpcClient interface {
GetDataKey(context.Context, *keyserviceproto.GetDataKeyRequest, *grpc.ClientConn) (*keyserviceproto.GetDataKeyResponse, error)
}
type client struct{}
func (c client) GetDataKey(ctx context.Context, req *keyserviceproto.GetDataKeyRequest, conn *grpc.ClientConn) (*keyserviceproto.GetDataKeyResponse, error) {
return keyserviceproto.NewAPIClient(conn).GetDataKey(ctx, req)
}