Files
certctl/internal/connector/issuer/local/bundle9_coverage_test.go
T
Shankar fdd445c09f crypto/signer: introduce Signer interface; refactor local issuer to use it
This is a load-bearing internal refactor with no user-visible behavior
change. The new internal/crypto/signer package abstracts CA private-key
signing behind a Signer interface (embeds stdlib crypto.Signer + adds
Algorithm()). The local issuer now consumes this interface; the
historical c.caKey crypto.Signer field is renamed c.caSigner signer.Signer.

What landed:

  * internal/crypto/signer/ — new stdlib-only package
    - Signer interface: crypto.Signer + Algorithm()
    - Algorithm enum: RSA-2048, RSA-3072, RSA-4096, ECDSA-P256, ECDSA-P384
    - Driver interface: Load / Generate / Name
    - FileDriver: production driver, wraps file-on-disk PEM, hooks for
      DirHardener + Marshaler so the local package can inject Bundle 9
      keystore.ensureKeyDirSecure + keymem.marshalPrivateKeyAndZeroize
    - MemoryDriver: in-memory test driver; safe for concurrent use
    - parse.go: ParsePrivateKey moved here from local.go (PKCS#1, SEC 1, PKCS#8)
    - 91.6% coverage (gate ≥85)

  * internal/connector/issuer/local/local.go — refactor
    - Rename c.caKey crypto.Signer → c.caSigner signer.Signer
    - Rewire 4 signing call sites: leaf cert (line ~613), CRL (~849),
      OCSP response (~887), CA bootstrap (~482) — all access the
      interface; the bootstrap also switches to interface-level
      Public() + Signer
    - Wrap freshly-generated and freshly-loaded keys; reject Ed25519
      and other unsupported algorithms at load time (was silently
      accepted before, would have failed at first sign)
    - Delete the duplicated parsePrivateKey helper (single source of
      truth now lives in the signer package)
    - Update the L-014 threat-model comment block (lines 1-29) with a
      forward-reference paragraph: file-on-disk caveats apply only to
      FileDriver-backed signers; alternative drivers close that leg
    - Coverage 86.7 → 86.5 (above CI floor of 86); the 0.2pp drop is
      mechanical from deleting parsePrivateKey, partially recovered by
      a new test pinning the Wrap error path

  * internal/crypto/signer/equivalence_test.go — Phase 3 safety net
    - RSA byte-strict equality for leaf certs / CRLs / OCSP responses
      (PKCS#1 v1.5 is deterministic)
    - ECDSA TBS-strict equality (signature differs because of random k)
    - Both signatures independently validate against the CA
    - Negative sentinel proves the equivalence checker isn't trivially-
      passing

  * docs/architecture.md — new 'CA Signing Abstraction' section under
    Security Model, with ASCII diagram of FileDriver / MemoryDriver /
    future PKCS11Driver / future CloudKMSDriver

  * Test file mechanical edits (only):
    - bundle9_coverage_test.go: parsePrivateKey → signer.ParsePrivateKey
      (function moved, not behavior changed)
    - local_test.go: append one targeted test
      (TestSubCA_LoadCAFromDisk_RejectsUnsupportedKeyAlgorithm) that
      pins the new Wrap error path I introduced — recovers coverage
      cost of the deletion above

What did NOT change (verified empty diffs):
  * api/openapi.yaml
  * migrations/
  * internal/connector/issuer/interface.go
  * go.mod / go.sum (no new dependencies; stdlib only)

This refactor is the prerequisite for three downstream items:
  - PKCS#11/HSM driver (V3-Pro)
  - CRL/OCSP responder (V2)
  - SSH CA lifecycle (V2)

Each of those adds a new signing call site. Doing the abstraction now
costs once; deferring would cost three times.
2026-04-28 22:03:55 +00:00

859 lines
29 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
package local
import (
"bytes"
"context"
"crypto/ecdsa"
"crypto/elliptic"
"crypto/rand"
"crypto/rsa"
"crypto/x509"
"crypto/x509/pkix"
"encoding/pem"
"errors"
"io"
"log/slog"
"math/big"
"net"
"os"
"path/filepath"
"runtime"
"strings"
"testing"
"time"
"github.com/shankar0123/certctl/internal/connector/issuer"
"github.com/shankar0123/certctl/internal/crypto/signer"
)
// Bundle-9 / Audit H-010 + L-002 + L-003 + L-012 + M-028 regression suite.
//
// Goal: lift internal/connector/issuer/local/ coverage from the pre-bundle
// baseline (68.3%) to ≥85% by exercising the previously untested paths:
//
// GetCACertPEM (0.0%) — happy path + uninitialized-CA path
// GetRenewalInfo (0.0%) — returns nil + true (current behavior)
// parsePrivateKey (27.3%) — RSA / ECDSA EC / PKCS8-RSA / PKCS8-ECDSA
// / unknown type / non-signer PKCS8 / malformed
// resolveEKUsAndKeyUsage (10.0%) — empty list / each individual EKU /
// unknown EKU / mixed TLS+email
// hashPublicKey (44.4%) — RSA / ECDSA-P256 / ECDSA-P384 /
// ECDSA-P521 / unsupported curve
// ecdsaToECDH (0.0%) — round-trip pin: byte-identical to
// legacy elliptic.Marshal output
// validateCSRUnicode (58.3%) — every rejection arm + clean-pass arm
// keymem.go / keystore.go (0.0%) — every branch
//
// We also exercise IssueCertificate / RenewCertificate failure paths
// (malformed PEM, invalid CSR signature, post-rejection unicode) to lift
// those out of the high-50s. The bundle's promised floor is 85%; we aim
// for headroom.
// ---------------------------------------------------------------------------
// Helpers
// ---------------------------------------------------------------------------
func newTestConnectorBundle9(t *testing.T) *Connector {
t.Helper()
c := New(&Config{ValidityDays: 7}, slog.New(slog.NewTextHandler(io.Discard, nil)))
if err := c.ensureCA(context.Background()); err != nil {
t.Fatalf("ensureCA: %v", err)
}
return c
}
func mustGenECDSAKey(t *testing.T, curve elliptic.Curve) *ecdsa.PrivateKey {
t.Helper()
k, err := ecdsa.GenerateKey(curve, rand.Reader)
if err != nil {
t.Fatalf("generate key: %v", err)
}
return k
}
func mustGenRSAKey(t *testing.T) *rsa.PrivateKey {
t.Helper()
k, err := rsa.GenerateKey(rand.Reader, 2048)
if err != nil {
t.Fatalf("generate rsa key: %v", err)
}
return k
}
func mustEncodeCSR(t *testing.T, key any, tmpl *x509.CertificateRequest) string {
t.Helper()
der, err := x509.CreateCertificateRequest(rand.Reader, tmpl, key)
if err != nil {
t.Fatalf("create csr: %v", err)
}
return string(pem.EncodeToMemory(&pem.Block{Type: "CERTIFICATE REQUEST", Bytes: der}))
}
// ---------------------------------------------------------------------------
// GetCACertPEM / GetRenewalInfo (lift 0% → 100%)
// ---------------------------------------------------------------------------
func TestGetCACertPEM_ReturnsAfterEnsureCA(t *testing.T) {
c := newTestConnectorBundle9(t)
pemStr, err := c.GetCACertPEM(context.Background())
if err != nil {
t.Fatalf("GetCACertPEM err: %v", err)
}
if !strings.Contains(pemStr, "-----BEGIN CERTIFICATE-----") {
t.Errorf("expected PEM CA cert, got %q", pemStr)
}
}
func TestGetCACertPEM_TriggersEnsureCAOnFreshConnector(t *testing.T) {
// Fresh connector — GetCACertPEM should call ensureCA implicitly.
c := New(&Config{ValidityDays: 7}, slog.New(slog.NewTextHandler(io.Discard, nil)))
pemStr, err := c.GetCACertPEM(context.Background())
if err != nil {
t.Fatalf("GetCACertPEM on fresh connector: %v", err)
}
if pemStr == "" {
t.Fatal("expected non-empty PEM")
}
}
func TestGetRenewalInfo_ReturnsNilNil(t *testing.T) {
c := newTestConnectorBundle9(t)
info, err := c.GetRenewalInfo(context.Background(), "any-cert-pem")
if err != nil {
t.Fatalf("GetRenewalInfo err: %v", err)
}
if info != nil {
t.Errorf("expected nil RenewalInfo for local CA (no ARI support), got %+v", info)
}
}
// ---------------------------------------------------------------------------
// parsePrivateKey (27.3% → all branches)
// ---------------------------------------------------------------------------
func TestParsePrivateKey_RSAPKCS1(t *testing.T) {
k := mustGenRSAKey(t)
der := x509.MarshalPKCS1PrivateKey(k)
signer, err := signer.ParsePrivateKey(&pem.Block{Type: "RSA PRIVATE KEY", Bytes: der})
if err != nil {
t.Fatalf("parsePrivateKey RSA PKCS1: %v", err)
}
if _, ok := signer.(*rsa.PrivateKey); !ok {
t.Errorf("expected *rsa.PrivateKey, got %T", signer)
}
}
func TestParsePrivateKey_ECPrivateKey(t *testing.T) {
k := mustGenECDSAKey(t, elliptic.P256())
der, err := x509.MarshalECPrivateKey(k)
if err != nil {
t.Fatalf("marshal: %v", err)
}
signer, err := signer.ParsePrivateKey(&pem.Block{Type: "EC PRIVATE KEY", Bytes: der})
if err != nil {
t.Fatalf("parsePrivateKey EC: %v", err)
}
if _, ok := signer.(*ecdsa.PrivateKey); !ok {
t.Errorf("expected *ecdsa.PrivateKey, got %T", signer)
}
}
func TestParsePrivateKey_PKCS8RSA(t *testing.T) {
k := mustGenRSAKey(t)
der, err := x509.MarshalPKCS8PrivateKey(k)
if err != nil {
t.Fatalf("marshal pkcs8: %v", err)
}
signer, err := signer.ParsePrivateKey(&pem.Block{Type: "PRIVATE KEY", Bytes: der})
if err != nil {
t.Fatalf("parsePrivateKey PKCS8: %v", err)
}
if _, ok := signer.(*rsa.PrivateKey); !ok {
t.Errorf("expected RSA, got %T", signer)
}
}
func TestParsePrivateKey_PKCS8ECDSA(t *testing.T) {
k := mustGenECDSAKey(t, elliptic.P256())
der, err := x509.MarshalPKCS8PrivateKey(k)
if err != nil {
t.Fatalf("marshal pkcs8: %v", err)
}
signer, err := signer.ParsePrivateKey(&pem.Block{Type: "PRIVATE KEY", Bytes: der})
if err != nil {
t.Fatalf("parsePrivateKey PKCS8 ECDSA: %v", err)
}
if _, ok := signer.(*ecdsa.PrivateKey); !ok {
t.Errorf("expected ECDSA, got %T", signer)
}
}
func TestParsePrivateKey_UnknownType(t *testing.T) {
_, err := signer.ParsePrivateKey(&pem.Block{Type: "DSA PRIVATE KEY", Bytes: []byte{1, 2, 3}})
if err == nil {
t.Fatal("expected error on unknown PEM type")
}
if !strings.Contains(err.Error(), "unsupported private key type") {
t.Errorf("error should mention unsupported, got: %v", err)
}
}
func TestParsePrivateKey_MalformedPKCS8(t *testing.T) {
_, err := signer.ParsePrivateKey(&pem.Block{Type: "PRIVATE KEY", Bytes: []byte{0xff, 0xff, 0xff}})
if err == nil {
t.Fatal("expected error on malformed PKCS8")
}
}
// ---------------------------------------------------------------------------
// resolveEKUsAndKeyUsage (10% → all branches)
// ---------------------------------------------------------------------------
func TestResolveEKUsAndKeyUsage_EmptyDefaultsToTLS(t *testing.T) {
ekus, usage := resolveEKUsAndKeyUsage(nil)
if len(ekus) != 2 {
t.Errorf("expected default serverAuth+clientAuth, got %d EKUs: %v", len(ekus), ekus)
}
if usage&x509.KeyUsageDigitalSignature == 0 {
t.Error("expected DigitalSignature in default key usage")
}
if usage&x509.KeyUsageKeyEncipherment == 0 {
t.Error("expected KeyEncipherment in default key usage (TLS server EKU)")
}
}
func TestResolveEKUsAndKeyUsage_ServerAuthOnly(t *testing.T) {
ekus, _ := resolveEKUsAndKeyUsage([]string{"serverAuth"})
if len(ekus) != 1 || ekus[0] != x509.ExtKeyUsageServerAuth {
t.Errorf("expected only serverAuth, got: %v", ekus)
}
}
func TestResolveEKUsAndKeyUsage_AllKnownEKUs(t *testing.T) {
// ekuNameToX509 supports: serverAuth, clientAuth, codeSigning,
// emailProtection, timeStamping. OCSPSigning is intentionally not
// in the local-CA allowlist (responder cert is signed by the same
// CA but issued via the OCSP path, not the EKU enum).
known := []string{"serverAuth", "clientAuth", "codeSigning", "emailProtection", "timeStamping"}
ekus, usage := resolveEKUsAndKeyUsage(known)
if len(ekus) != len(known) {
t.Errorf("expected %d EKUs, got %d: %v", len(known), len(ekus), ekus)
}
if usage&x509.KeyUsageContentCommitment == 0 {
t.Error("expected non-repudiation set when emailProtection is in mix")
}
if usage&x509.KeyUsageKeyEncipherment == 0 {
t.Error("expected KeyEncipherment set when serverAuth is in mix")
}
}
func TestResolveEKUsAndKeyUsage_AllUnknownFallsBackToDefault(t *testing.T) {
ekus, usage := resolveEKUsAndKeyUsage([]string{"madeUp1", "madeUp2"})
if len(ekus) != 2 {
t.Errorf("expected 2 default EKUs after fallback, got %d", len(ekus))
}
if usage&x509.KeyUsageDigitalSignature == 0 {
t.Error("expected DigitalSignature in fallback default")
}
}
func TestResolveEKUsAndKeyUsage_UnknownEKUIgnored(t *testing.T) {
ekus, _ := resolveEKUsAndKeyUsage([]string{"serverAuth", "totallyMadeUp"})
if len(ekus) != 1 || ekus[0] != x509.ExtKeyUsageServerAuth {
t.Errorf("unknown EKU should be silently dropped, got: %v", ekus)
}
}
func TestResolveEKUsAndKeyUsage_EmailOnlyHasNoKeyEncipherment(t *testing.T) {
_, usage := resolveEKUsAndKeyUsage([]string{"emailProtection"})
if usage&x509.KeyUsageKeyEncipherment != 0 {
t.Error("email-only should NOT include KeyEncipherment")
}
if usage&x509.KeyUsageContentCommitment == 0 {
t.Error("email-only SHOULD include ContentCommitment (non-repudiation)")
}
}
// ---------------------------------------------------------------------------
// hashPublicKey (44.4% → all curves) + ecdsaToECDH (0% → all curves)
// ---------------------------------------------------------------------------
func TestHashPublicKey_RSA(t *testing.T) {
k := mustGenRSAKey(t)
out := hashPublicKey(&k.PublicKey)
if len(out) != 4 {
t.Errorf("expected 4-byte SKI prefix, got %d", len(out))
}
}
func TestHashPublicKey_ECDSA_P256(t *testing.T) {
k := mustGenECDSAKey(t, elliptic.P256())
out := hashPublicKey(&k.PublicKey)
if len(out) != 4 {
t.Errorf("expected 4-byte SKI prefix, got %d", len(out))
}
}
func TestHashPublicKey_ECDSA_P384(t *testing.T) {
k := mustGenECDSAKey(t, elliptic.P384())
_ = hashPublicKey(&k.PublicKey)
}
func TestHashPublicKey_ECDSA_P521(t *testing.T) {
k := mustGenECDSAKey(t, elliptic.P521())
_ = hashPublicKey(&k.PublicKey)
}
func TestHashPublicKey_UnknownTypeReturnsEmpty(t *testing.T) {
type bogusPub struct{}
out := hashPublicKey(bogusPub{})
if len(out) != 4 {
t.Errorf("expected 4-byte hash even for empty input (sha256 prefix), got %d", len(out))
}
}
// TestHashPublicKey_ECDSA_RoundTripPin asserts that the new
// crypto/ecdh-based encoding produces byte-identical output to the legacy
// elliptic.Marshal call this PR removed (M-028 SA1019 migration). If this
// test fails, the SubjectKeyId of every certificate the local CA has ever
// issued would silently change on upgrade, breaking pinning + audit
// fingerprinting downstream.
func TestHashPublicKey_ECDSA_RoundTripPin(t *testing.T) {
cases := []struct {
name string
curve elliptic.Curve
}{
{"P256", elliptic.P256()},
{"P384", elliptic.P384()},
{"P521", elliptic.P521()},
}
for _, tc := range cases {
t.Run(tc.name, func(t *testing.T) {
k := mustGenECDSAKey(t, tc.curve)
ecdhPub, err := ecdsaToECDH(&k.PublicKey)
if err != nil {
t.Fatalf("ecdsaToECDH: %v", err)
}
ecdhBytes := ecdhPub.Bytes()
// Pin assertion — we DELIBERATELY use the deprecated API here
// as a regression oracle to prove the new crypto/ecdh path
// produces byte-identical output. If elliptic.Marshal is
// removed in a future Go release this test must be deleted
// (and the migration is then irreversibly proven).
//lint:ignore SA1019 deliberate regression oracle for M-028 round-trip pin
legacy := elliptic.Marshal(k.Curve, k.X, k.Y)
if !bytes.Equal(ecdhBytes, legacy) {
t.Fatalf("ECDH .Bytes() != legacy elliptic.Marshal output\n new: %x\n old: %x", ecdhBytes, legacy)
}
})
}
}
func TestEcdsaToECDH_RejectsP224(t *testing.T) {
k := mustGenECDSAKey(t, elliptic.P224())
_, err := ecdsaToECDH(&k.PublicKey)
if err == nil {
t.Fatal("expected unsupported-curve error for P-224")
}
if !strings.Contains(err.Error(), "unsupported curve") {
t.Errorf("expected unsupported-curve error, got: %v", err)
}
}
func TestEcdsaToECDH_RejectsNilKey(t *testing.T) {
if _, err := ecdsaToECDH(nil); err == nil {
t.Fatal("expected error on nil key")
}
}
// ---------------------------------------------------------------------------
// validateCSRUnicode (58% → all branches)
// ---------------------------------------------------------------------------
func TestValidateCSRUnicode_CleanPasses(t *testing.T) {
csr := &x509.CertificateRequest{
Subject: pkix.Name{CommonName: "example.com"},
DNSNames: []string{"www.example.com", "api.example.com"},
EmailAddresses: []string{"admin@example.com"},
}
if err := validateCSRUnicode(csr, []string{"alt.example.com"}); err != nil {
t.Errorf("clean CSR rejected: %v", err)
}
}
func TestValidateCSRUnicode_RejectsCNHomograph(t *testing.T) {
csr := &x509.CertificateRequest{
Subject: pkix.Name{CommonName: "аpple.com"}, // Cyrillic а
}
err := validateCSRUnicode(csr, nil)
if err == nil {
t.Fatal("expected rejection for CN homograph")
}
if !strings.Contains(err.Error(), "CommonName") {
t.Errorf("error should mention CommonName, got: %v", err)
}
}
func TestValidateCSRUnicode_RejectsDNSNameRTL(t *testing.T) {
csr := &x509.CertificateRequest{
Subject: pkix.Name{CommonName: "ok.com"},
DNSNames: []string{"good\u202Eevil.com"},
}
err := validateCSRUnicode(csr, nil)
if err == nil {
t.Fatal("expected rejection for DNSName RTL override")
}
if !strings.Contains(err.Error(), "DNSNames") {
t.Errorf("error should mention DNSNames, got: %v", err)
}
}
func TestValidateCSRUnicode_RejectsEmailZeroWidth(t *testing.T) {
csr := &x509.CertificateRequest{
Subject: pkix.Name{CommonName: "ok.com"},
EmailAddresses: []string{"good\u200Bbad@example.com"},
}
err := validateCSRUnicode(csr, nil)
if err == nil {
t.Fatal("expected rejection for email zero-width")
}
if !strings.Contains(err.Error(), "EmailAddresses") {
t.Errorf("error should mention EmailAddresses, got: %v", err)
}
}
func TestValidateCSRUnicode_RejectsAdditionalSAN(t *testing.T) {
csr := &x509.CertificateRequest{
Subject: pkix.Name{CommonName: "ok.com"},
}
err := validateCSRUnicode(csr, []string{"good\u202Eevil.com"})
if err == nil {
t.Fatal("expected rejection for additional SAN RTL")
}
if !strings.Contains(err.Error(), "request SANs") {
t.Errorf("error should mention request SANs, got: %v", err)
}
}
// ---------------------------------------------------------------------------
// IssueCertificate / RenewCertificate failure paths (lift 55-68% → higher)
// ---------------------------------------------------------------------------
func TestIssueCertificate_RejectsMalformedCSRPEM(t *testing.T) {
c := newTestConnectorBundle9(t)
_, err := c.IssueCertificate(context.Background(), issuer.IssuanceRequest{
CommonName: "x.com",
CSRPEM: "not a pem",
})
if err == nil {
t.Fatal("expected error on malformed CSR PEM")
}
}
func TestIssueCertificate_RejectsBadCSRSignature(t *testing.T) {
c := newTestConnectorBundle9(t)
// Build a valid CSR using key A, then re-sign the CertificateRequest
// payload with key B (or just flip bytes in the signature) — the
// CheckSignature path inside IssueCertificate must reject this.
keyA := mustGenECDSAKey(t, elliptic.P256())
der, err := x509.CreateCertificateRequest(rand.Reader, &x509.CertificateRequest{
Subject: pkix.Name{CommonName: "x.com"},
}, keyA)
if err != nil {
t.Fatal(err)
}
// Flip a byte deep in the signature (last 16 bytes are signature octets).
if len(der) < 20 {
t.Skip("unexpectedly short DER")
}
der[len(der)-5] ^= 0xff
tamperedPEM := string(pem.EncodeToMemory(&pem.Block{Type: "CERTIFICATE REQUEST", Bytes: der}))
_, issErr := c.IssueCertificate(context.Background(), issuer.IssuanceRequest{
CommonName: "x.com",
CSRPEM: tamperedPEM,
})
if issErr == nil {
t.Fatal("expected error on tampered CSR")
}
}
func TestIssueCertificate_RejectsHomographCSR(t *testing.T) {
c := newTestConnectorBundle9(t)
k := mustGenECDSAKey(t, elliptic.P256())
csrPEM := mustEncodeCSR(t, k, &x509.CertificateRequest{
Subject: pkix.Name{CommonName: "аpple.com"},
})
_, err := c.IssueCertificate(context.Background(), issuer.IssuanceRequest{
CommonName: "аpple.com",
CSRPEM: csrPEM,
})
if err == nil {
t.Fatal("expected unicode-rejection error")
}
if !strings.Contains(err.Error(), "CommonName") {
t.Errorf("expected CommonName-cited error, got: %v", err)
}
}
func TestRenewCertificate_RejectsMalformedCSRPEM(t *testing.T) {
c := newTestConnectorBundle9(t)
_, err := c.RenewCertificate(context.Background(), issuer.RenewalRequest{
CommonName: "x.com",
CSRPEM: "not a pem",
})
if err == nil {
t.Fatal("expected error on malformed CSR PEM")
}
}
func TestRenewCertificate_RejectsHomographCSR(t *testing.T) {
c := newTestConnectorBundle9(t)
k := mustGenECDSAKey(t, elliptic.P256())
csrPEM := mustEncodeCSR(t, k, &x509.CertificateRequest{
Subject: pkix.Name{CommonName: "аpple.com"},
})
_, err := c.RenewCertificate(context.Background(), issuer.RenewalRequest{
CommonName: "аpple.com",
CSRPEM: csrPEM,
})
if err == nil {
t.Fatal("expected unicode-rejection error on renew")
}
}
func TestRenewCertificate_HappyPath(t *testing.T) {
c := newTestConnectorBundle9(t)
k := mustGenECDSAKey(t, elliptic.P256())
csrPEM := mustEncodeCSR(t, k, &x509.CertificateRequest{
Subject: pkix.Name{CommonName: "renew.example.com"},
})
res, err := c.RenewCertificate(context.Background(), issuer.RenewalRequest{
CommonName: "renew.example.com",
CSRPEM: csrPEM,
})
if err != nil {
t.Fatalf("renew failed: %v", err)
}
if !strings.Contains(res.CertPEM, "BEGIN CERTIFICATE") {
t.Errorf("expected cert PEM, got: %s", res.CertPEM)
}
}
// ---------------------------------------------------------------------------
// keymem.go — marshalPrivateKeyAndZeroize
// ---------------------------------------------------------------------------
func TestMarshalPrivateKeyAndZeroize_HappyPath(t *testing.T) {
k := mustGenECDSAKey(t, elliptic.P256())
var captured []byte
err := marshalPrivateKeyAndZeroize(k, func(der []byte) error {
// Take a defensive copy — we promise NOT to retain `der`, but for
// the test we want to inspect it AFTER the function returns to
// prove zeroization happened to the underlying buffer.
captured = make([]byte, len(der))
copy(captured, der)
// Verify the DER decodes correctly while we have it.
if _, parseErr := x509.ParseECPrivateKey(der); parseErr != nil {
t.Errorf("DER inside callback should parse: %v", parseErr)
}
return nil
})
if err != nil {
t.Fatalf("marshal: %v", err)
}
// Captured bytes should still be valid PKCS-DER (we copied them).
if _, err := x509.ParseECPrivateKey(captured); err != nil {
t.Errorf("captured copy should still parse: %v", err)
}
}
func TestMarshalPrivateKeyAndZeroize_NilKey(t *testing.T) {
err := marshalPrivateKeyAndZeroize(nil, func([]byte) error { return nil })
if err == nil {
t.Fatal("expected error on nil key")
}
}
func TestMarshalPrivateKeyAndZeroize_OnDERError(t *testing.T) {
k := mustGenECDSAKey(t, elliptic.P256())
wantErr := errors.New("simulated downstream failure")
gotErr := marshalPrivateKeyAndZeroize(k, func([]byte) error { return wantErr })
if !errors.Is(gotErr, wantErr) {
t.Errorf("expected error to propagate, got: %v", gotErr)
}
}
// ---------------------------------------------------------------------------
// keystore.go — ensureKeyDirSecure
// ---------------------------------------------------------------------------
func TestEnsureKeyDirSecure_CreatesNewDir(t *testing.T) {
if runtime.GOOS == "windows" {
t.Skip("permission semantics differ on windows")
}
tmp := filepath.Join(t.TempDir(), "fresh")
if err := ensureKeyDirSecure(tmp); err != nil {
t.Fatalf("ensureKeyDirSecure: %v", err)
}
info, err := os.Stat(tmp)
if err != nil {
t.Fatalf("stat: %v", err)
}
if info.Mode().Perm() != 0o700 {
t.Errorf("expected 0700 after ensure, got %#o", info.Mode().Perm())
}
}
func TestEnsureKeyDirSecure_AcceptsExisting0700(t *testing.T) {
if runtime.GOOS == "windows" {
t.Skip("permission semantics differ on windows")
}
dir := t.TempDir()
// t.TempDir creates 0700 on unix.
_ = os.Chmod(dir, 0o700)
if err := ensureKeyDirSecure(dir); err != nil {
t.Errorf("0700 dir should be accepted: %v", err)
}
}
func TestEnsureKeyDirSecure_TightensPermissive(t *testing.T) {
if runtime.GOOS == "windows" {
t.Skip("permission semantics differ on windows")
}
dir := t.TempDir()
if err := os.Chmod(dir, 0o755); err != nil {
t.Fatalf("chmod: %v", err)
}
if err := ensureKeyDirSecure(dir); err != nil {
t.Fatalf("ensureKeyDirSecure should tighten: %v", err)
}
info, err := os.Stat(dir)
if err != nil {
t.Fatal(err)
}
if info.Mode().Perm() != 0o700 {
t.Errorf("expected 0700 after tighten, got %#o", info.Mode().Perm())
}
}
func TestEnsureKeyDirSecure_RejectsEmpty(t *testing.T) {
if err := ensureKeyDirSecure(""); err == nil {
t.Error("expected refusal of empty path")
}
if err := ensureKeyDirSecure("/"); err == nil {
t.Error("expected refusal of root")
}
if err := ensureKeyDirSecure("."); err == nil {
t.Error("expected refusal of dot")
}
}
func TestEnsureKeyDirSecure_AcceptsOwnerOnlyMode(t *testing.T) {
if runtime.GOOS == "windows" {
t.Skip("permission semantics differ on windows")
}
dir := t.TempDir()
if err := os.Chmod(dir, 0o500); err != nil {
t.Fatalf("chmod: %v", err)
}
if err := ensureKeyDirSecure(dir); err != nil {
t.Errorf("0500 (owner-only no-write) should be accepted: %v", err)
}
// Restore so t.TempDir cleanup works.
_ = os.Chmod(dir, 0o700)
}
// ---------------------------------------------------------------------------
// loadCAFromDisk negative paths (lift to push total over 85%)
// ---------------------------------------------------------------------------
func TestLoadCAFromDisk_RejectsExpiredCA(t *testing.T) {
dir := t.TempDir()
caKey := mustGenECDSAKey(t, elliptic.P256())
template := &x509.Certificate{
SerialNumber: big.NewInt(1),
Subject: pkix.Name{CommonName: "expired-ca"},
NotBefore: time.Now().Add(-2 * time.Hour),
NotAfter: time.Now().Add(-1 * time.Hour),
KeyUsage: x509.KeyUsageCertSign | x509.KeyUsageCRLSign,
BasicConstraintsValid: true,
IsCA: true,
}
der, err := x509.CreateCertificate(rand.Reader, template, template, &caKey.PublicKey, caKey)
if err != nil {
t.Fatal(err)
}
certPath := filepath.Join(dir, "ca.crt")
keyPath := filepath.Join(dir, "ca.key")
if err := os.WriteFile(certPath, pem.EncodeToMemory(&pem.Block{Type: "CERTIFICATE", Bytes: der}), 0o600); err != nil {
t.Fatal(err)
}
keyDER, _ := x509.MarshalECPrivateKey(caKey)
if err := os.WriteFile(keyPath, pem.EncodeToMemory(&pem.Block{Type: "EC PRIVATE KEY", Bytes: keyDER}), 0o600); err != nil {
t.Fatal(err)
}
c := New(&Config{ValidityDays: 7, CACertPath: certPath, CAKeyPath: keyPath}, slog.New(slog.NewTextHandler(io.Discard, nil)))
err = c.ensureCA(context.Background())
if err == nil {
t.Fatal("expected error for expired CA")
}
if !strings.Contains(err.Error(), "expired") {
t.Errorf("expected expired-CA error, got: %v", err)
}
}
func TestLoadCAFromDisk_RejectsNonCACert(t *testing.T) {
dir := t.TempDir()
caKey := mustGenECDSAKey(t, elliptic.P256())
// IsCA: false -> should be rejected
template := &x509.Certificate{
SerialNumber: big.NewInt(2),
Subject: pkix.Name{CommonName: "not-a-ca"},
NotBefore: time.Now().Add(-time.Hour),
NotAfter: time.Now().Add(time.Hour),
KeyUsage: x509.KeyUsageDigitalSignature,
BasicConstraintsValid: true,
IsCA: false,
}
der, err := x509.CreateCertificate(rand.Reader, template, template, &caKey.PublicKey, caKey)
if err != nil {
t.Fatal(err)
}
certPath := filepath.Join(dir, "ca.crt")
keyPath := filepath.Join(dir, "ca.key")
if err := os.WriteFile(certPath, pem.EncodeToMemory(&pem.Block{Type: "CERTIFICATE", Bytes: der}), 0o600); err != nil {
t.Fatal(err)
}
keyDER, _ := x509.MarshalECPrivateKey(caKey)
if err := os.WriteFile(keyPath, pem.EncodeToMemory(&pem.Block{Type: "EC PRIVATE KEY", Bytes: keyDER}), 0o600); err != nil {
t.Fatal(err)
}
c := New(&Config{ValidityDays: 7, CACertPath: certPath, CAKeyPath: keyPath}, slog.New(slog.NewTextHandler(io.Discard, nil)))
err = c.ensureCA(context.Background())
if err == nil {
t.Fatal("expected error for non-CA cert")
}
}
func TestLoadCAFromDisk_HappyPath(t *testing.T) {
dir := t.TempDir()
caKey := mustGenECDSAKey(t, elliptic.P256())
template := &x509.Certificate{
SerialNumber: big.NewInt(3),
Subject: pkix.Name{CommonName: "valid-ca"},
NotBefore: time.Now().Add(-time.Hour),
NotAfter: time.Now().AddDate(1, 0, 0),
KeyUsage: x509.KeyUsageCertSign | x509.KeyUsageCRLSign,
BasicConstraintsValid: true,
IsCA: true,
}
der, err := x509.CreateCertificate(rand.Reader, template, template, &caKey.PublicKey, caKey)
if err != nil {
t.Fatal(err)
}
certPath := filepath.Join(dir, "ca.crt")
keyPath := filepath.Join(dir, "ca.key")
if err := os.WriteFile(certPath, pem.EncodeToMemory(&pem.Block{Type: "CERTIFICATE", Bytes: der}), 0o600); err != nil {
t.Fatal(err)
}
keyDER, _ := x509.MarshalECPrivateKey(caKey)
if err := os.WriteFile(keyPath, pem.EncodeToMemory(&pem.Block{Type: "EC PRIVATE KEY", Bytes: keyDER}), 0o600); err != nil {
t.Fatal(err)
}
c := New(&Config{ValidityDays: 7, CACertPath: certPath, CAKeyPath: keyPath}, slog.New(slog.NewTextHandler(io.Discard, nil)))
if err := c.ensureCA(context.Background()); err != nil {
t.Fatalf("loadCAFromDisk happy: %v", err)
}
if !c.subCA {
t.Error("expected subCA=true after disk-load")
}
}
func TestLoadCAFromDisk_MissingCert(t *testing.T) {
c := New(&Config{ValidityDays: 7, CACertPath: "/nope/missing.crt", CAKeyPath: "/nope/missing.key"}, slog.New(slog.NewTextHandler(io.Discard, nil)))
err := c.ensureCA(context.Background())
if err == nil {
t.Fatal("expected error for missing CA file")
}
}
// ---------------------------------------------------------------------------
// Final pushes to clear the ≥85% coverage gate.
// ---------------------------------------------------------------------------
func TestParseIP_ValidAndInvalid(t *testing.T) {
if parseIP("10.0.0.1") == nil {
t.Error("10.0.0.1 should parse")
}
if parseIP("not-an-ip") != nil {
t.Error("garbage shouldn't parse")
}
if parseIP("::1") == nil {
t.Error("IPv6 ::1 should parse")
}
}
func TestIsEmail_TrueAndFalse(t *testing.T) {
// isEmail is a simple "contains @" check — that's the spec it
// implements; we just pin both sides of the binary decision.
if !isEmail("user@example.com") {
t.Error("user@example.com should be an email")
}
if isEmail("just-a-host.example.com") {
t.Error("plain host should not be classified as email")
}
if isEmail("") {
t.Error("empty string should not be classified as email")
}
}
func TestValidateConfig_AllArms(t *testing.T) {
c := New(&Config{ValidityDays: 7}, slog.New(slog.NewTextHandler(io.Discard, nil)))
// Malformed JSON — must fail.
if err := c.ValidateConfig(context.Background(), []byte("not json")); err == nil {
t.Error("malformed JSON should be rejected")
}
// Default validity (zero) — must fail (validity_days must be >=1).
if err := c.ValidateConfig(context.Background(), []byte(`{"validity_days":0}`)); err == nil {
t.Error("validity_days < 1 should be rejected")
}
// Sub-CA with cert path but no key path — must fail.
if err := c.ValidateConfig(context.Background(), []byte(`{"validity_days":7,"ca_cert_path":"/x"}`)); err == nil {
t.Error("sub-CA with only cert path should be rejected")
}
// Sub-CA with key path but no cert path — must fail.
if err := c.ValidateConfig(context.Background(), []byte(`{"validity_days":7,"ca_key_path":"/x"}`)); err == nil {
t.Error("sub-CA with only key path should be rejected")
}
// Sub-CA with both paths but pointing nowhere — must fail (Stat).
if err := c.ValidateConfig(context.Background(), []byte(`{"validity_days":7,"ca_cert_path":"/nope","ca_key_path":"/nope-key"}`)); err == nil {
t.Error("sub-CA with non-existent paths should be rejected")
}
// Self-signed mode with valid validity — must pass.
if err := c.ValidateConfig(context.Background(), []byte(`{"validity_days":7}`)); err != nil {
t.Errorf("self-signed valid config should pass: %v", err)
}
}
func TestGenerateCertificate_WithMaxTTLCap(t *testing.T) {
c := newTestConnectorBundle9(t)
k := mustGenECDSAKey(t, elliptic.P256())
csrPEM := mustEncodeCSR(t, k, &x509.CertificateRequest{
Subject: pkix.Name{CommonName: "ttl.example.com"},
DNSNames: []string{"ttl.example.com"},
IPAddresses: []net.IP{net.ParseIP("10.0.0.5")},
EmailAddresses: []string{"ops@ttl.example.com"},
})
res, err := c.IssueCertificate(context.Background(), issuer.IssuanceRequest{
CommonName: "ttl.example.com",
CSRPEM: csrPEM,
MaxTTLSeconds: 3600, // 1h cap
})
if err != nil {
t.Fatalf("issue failed: %v", err)
}
if got := res.NotAfter.Sub(res.NotBefore); got > time.Hour+time.Minute {
t.Errorf("MaxTTL cap not honored, got window %s", got)
}
}