Skip to content

Commit

Permalink
introduce certopts
Browse files Browse the repository at this point in the history
  • Loading branch information
hanshasselberg committed Jan 21, 2021
1 parent 97e54f1 commit b2678b8
Show file tree
Hide file tree
Showing 7 changed files with 54 additions and 74 deletions.
18 changes: 5 additions & 13 deletions agent/consul/server_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -53,20 +53,12 @@ func testTLSCertificates(serverName string) (cert string, key string, cacert str
return "", "", "", err
}

serial, err := tlsutil.GenerateSerialNumber()
if err != nil {
return "", "", "", err
}

cert, privateKey, err := tlsutil.GenerateCert(
signer,
ca,
serial,
"Test Cert Name",
365,
[]string{serverName},
nil,
[]x509.ExtKeyUsage{x509.ExtKeyUsageServerAuth, x509.ExtKeyUsageClientAuth},
tlsutil.CertOpts{
Signer: signer, CA: ca, Name: "Test Cert Name",
Days: 365, DNSNames: []string{serverName},
ExtKeyUsage: []x509.ExtKeyUsage{x509.ExtKeyUsageServerAuth, x509.ExtKeyUsageClientAuth},
},
)
if err != nil {
return "", "", "", err
Expand Down
17 changes: 3 additions & 14 deletions agent/pool/peek_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -201,21 +201,10 @@ func generateTestCert(serverName string) (cert tls.Certificate, caPEM []byte, er
return tls.Certificate{}, nil, err
}

// generate leaf
serial, err := tlsutil.GenerateSerialNumber()
if err != nil {
return tls.Certificate{}, nil, err
}

certificate, privateKey, err := tlsutil.GenerateCert(
signer,
ca,
serial,
"Test Cert Name",
365,
[]string{serverName},
nil,
[]x509.ExtKeyUsage{x509.ExtKeyUsageServerAuth},
tlsutil.CertOpts{Signer: signer, CA: ca, Name: "Test Cert Name", Days: 365, DNSNames: []string{serverName},
ExtKeyUsage: []x509.ExtKeyUsage{x509.ExtKeyUsageServerAuth},
},
)
if err != nil {
return tls.Certificate{}, nil, err
Expand Down
21 changes: 5 additions & 16 deletions agent/routine-leak-checker/leak_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -15,31 +15,20 @@ import (
)

func testTLSCertificates(serverName string) (cert string, key string, cacert string, err error) {
ca, _, err := tlsutil.GenerateCA(tlsutil.CAOpts{})
if err != nil {
return "", "", "", err
}

// generate leaf
serial, err := tlsutil.GenerateSerialNumber()
signer, _, err := tlsutil.GeneratePrivateKey()
if err != nil {
return "", "", "", err
}

signer, _, err := tlsutil.GeneratePrivateKey()
ca, _, err := tlsutil.GenerateCA(tlsutil.CAOpts{Signer: signer})
if err != nil {
return "", "", "", err
}

cert, privateKey, err := tlsutil.GenerateCert(
signer,
ca,
serial,
"Test Cert Name",
365,
[]string{serverName},
nil,
[]x509.ExtKeyUsage{x509.ExtKeyUsageServerAuth, x509.ExtKeyUsageClientAuth},
tlsutil.CertOpts{Signer: signer, CA: ca, Name: "Test Cert Name", Days: 365, DNSNames: []string{serverName},
ExtKeyUsage: []x509.ExtKeyUsage{x509.ExtKeyUsageServerAuth, x509.ExtKeyUsageClientAuth},
},
)
if err != nil {
return "", "", "", err
Expand Down
18 changes: 5 additions & 13 deletions agent/testagent.go
Original file line number Diff line number Diff line change
Expand Up @@ -577,20 +577,12 @@ func testTLSCertificates(serverName string) (cert string, key string, cacert str
return "", "", "", err
}

serial, err := tlsutil.GenerateSerialNumber()
if err != nil {
return "", "", "", err
}

cert, privateKey, err := tlsutil.GenerateCert(
signer,
ca,
serial,
"Test Cert Name",
365,
[]string{serverName},
nil,
[]x509.ExtKeyUsage{x509.ExtKeyUsageServerAuth, x509.ExtKeyUsageClientAuth},
tlsutil.CertOpts{
Signer: signer, CA: ca, Name: "Test Cert Name",
Days: 365, DNSNames: []string{serverName},
ExtKeyUsage: []x509.ExtKeyUsage{x509.ExtKeyUsageServerAuth, x509.ExtKeyUsageClientAuth},
},
)
if err != nil {
return "", "", "", err
Expand Down
11 changes: 4 additions & 7 deletions command/tls/cert/create/tls_cert_create.go
Original file line number Diff line number Diff line change
Expand Up @@ -176,13 +176,10 @@ func (c *cmd) Run(args []string) int {
return 1
}

sn, err := tlsutil.GenerateSerialNumber()
if err != nil {
c.UI.Error(err.Error())
return 1
}

pub, priv, err := tlsutil.GenerateCert(signer, string(cert), sn, name, c.days, DNSNames, IPAddresses, extKeyUsage)
pub, priv, err := tlsutil.GenerateCert(tlsutil.CertOpts{
Signer: signer, CA: string(cert), Name: name, Days: c.days,
DNSNames: DNSNames, IPAddresses: IPAddresses, ExtKeyUsage: extKeyUsage,
})
if err != nil {
c.UI.Error(err.Error())
return 1
Expand Down
36 changes: 28 additions & 8 deletions tlsutil/generate.go
Original file line number Diff line number Diff line change
Expand Up @@ -125,9 +125,20 @@ func GenerateCA(opts CAOpts) (string, string, error) {
return buf.String(), pk, nil
}

type CertOpts struct {
Signer crypto.Signer
Serial *big.Int
CA string
Name string
Days int
DNSNames []string
IPAddresses []net.IP
ExtKeyUsage []x509.ExtKeyUsage
}

// GenerateCert generates a new certificate for agent TLS (not to be confused with Connect TLS)
func GenerateCert(signer crypto.Signer, ca string, sn *big.Int, name string, days int, DNSNames []string, IPAddresses []net.IP, extKeyUsage []x509.ExtKeyUsage) (string, string, error) {
parent, err := parseCert(ca)
func GenerateCert(opts CertOpts) (string, string, error) {
parent, err := parseCert(opts.CA)
if err != nil {
return "", "", err
}
Expand All @@ -142,21 +153,30 @@ func GenerateCert(signer crypto.Signer, ca string, sn *big.Int, name string, day
return "", "", err
}

sn := opts.Serial
if sn == nil {
var err error
sn, err = GenerateSerialNumber()
if err != nil {
return "", "", err
}
}

template := x509.Certificate{
SerialNumber: sn,
Subject: pkix.Name{CommonName: name},
Subject: pkix.Name{CommonName: opts.Name},
BasicConstraintsValid: true,
KeyUsage: x509.KeyUsageDigitalSignature | x509.KeyUsageKeyEncipherment,
ExtKeyUsage: extKeyUsage,
ExtKeyUsage: opts.ExtKeyUsage,
IsCA: false,
NotAfter: time.Now().AddDate(0, 0, days),
NotAfter: time.Now().AddDate(0, 0, opts.Days),
NotBefore: time.Now(),
SubjectKeyId: id,
DNSNames: DNSNames,
IPAddresses: IPAddresses,
DNSNames: opts.DNSNames,
IPAddresses: opts.IPAddresses,
}

bs, err := x509.CreateCertificate(rand.Reader, &template, parent, signee.Public(), signer)
bs, err := x509.CreateCertificate(rand.Reader, &template, parent, signee.Public(), opts.Signer)
if err != nil {
return "", "", err
}
Expand Down
7 changes: 4 additions & 3 deletions tlsutil/generate_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -98,13 +98,14 @@ func TestGenerateCert(t *testing.T) {
ca, _, err := GenerateCA(CAOpts{Signer: signer})
require.Nil(t, err)

sn, err := GenerateSerialNumber()
require.Nil(t, err)
DNSNames := []string{"server.dc1.consul"}
IPAddresses := []net.IP{net.ParseIP("123.234.243.213")}
extKeyUsage := []x509.ExtKeyUsage{x509.ExtKeyUsageServerAuth}
name := "Cert Name"
certificate, pk, err := GenerateCert(signer, ca, sn, name, 365, DNSNames, IPAddresses, extKeyUsage)
certificate, pk, err := GenerateCert(CertOpts{
Signer: signer, CA: ca, Name: name, Days: 365, DNSNames: DNSNames,
IPAddresses: IPAddresses, ExtKeyUsage: extKeyUsage,
})
require.Nil(t, err)
require.NotEmpty(t, certificate)
require.NotEmpty(t, pk)
Expand Down

0 comments on commit b2678b8

Please sign in to comment.