open-consul/agent/connect/generate.go

99 lines
2.3 KiB
Go
Raw Normal View History

2018-05-09 22:12:31 +00:00
package connect
import (
"bytes"
2018-05-12 10:58:14 +00:00
"crypto"
2018-05-09 22:12:31 +00:00
"crypto/ecdsa"
"crypto/elliptic"
"crypto/rand"
"crypto/rsa"
2018-05-09 22:12:31 +00:00
"crypto/x509"
"encoding/pem"
"fmt"
"strings"
2018-05-09 22:12:31 +00:00
)
const (
DefaultPrivateKeyType = "ec"
DefaultPrivateKeyBits = 256
)
func pemEncodeKey(key []byte, blockType string) (string, error) {
var buf bytes.Buffer
if err := pem.Encode(&buf, &pem.Block{Type: blockType, Bytes: key}); err != nil {
return "", fmt.Errorf("error encoding private key: %s", err)
}
return buf.String(), nil
}
func generateRSAKey(keyBits int) (crypto.Signer, string, error) {
var pk *rsa.PrivateKey
pk, err := rsa.GenerateKey(rand.Reader, keyBits)
if err != nil {
return nil, "", fmt.Errorf("error generating RSA private key: %s", err)
}
bs := x509.MarshalPKCS1PrivateKey(pk)
pemBlock, err := pemEncodeKey(bs, "RSA PRIVATE KEY")
if err != nil {
return nil, "", err
}
return pk, pemBlock, nil
}
func generateECDSAKey(keyBits int) (crypto.Signer, string, error) {
2018-05-09 22:12:31 +00:00
var pk *ecdsa.PrivateKey
var curve elliptic.Curve
2018-05-09 22:12:31 +00:00
switch keyBits {
case 224:
curve = elliptic.P224()
case 256:
curve = elliptic.P256()
case 384:
curve = elliptic.P384()
case 521:
curve = elliptic.P521()
default:
return nil, "", fmt.Errorf("error generating ECDSA private key: unknown curve length %d", keyBits)
}
pk, err := ecdsa.GenerateKey(curve, rand.Reader)
2018-05-09 22:12:31 +00:00
if err != nil {
return nil, "", fmt.Errorf("error generating ECDSA private key: %s", err)
2018-05-09 22:12:31 +00:00
}
bs, err := x509.MarshalECPrivateKey(pk)
if err != nil {
return nil, "", fmt.Errorf("error marshaling ECDSA private key: %s", err)
2018-05-09 22:12:31 +00:00
}
pemBlock, err := pemEncodeKey(bs, "EC PRIVATE KEY")
2018-05-09 22:12:31 +00:00
if err != nil {
return nil, "", err
2018-05-09 22:12:31 +00:00
}
return pk, pemBlock, nil
}
// GeneratePrivateKey generates a new Private key
func GeneratePrivateKeyWithConfig(keyType string, keyBits int) (crypto.Signer, string, error) {
switch strings.ToLower(keyType) {
case "rsa":
return generateRSAKey(keyBits)
case "ec":
return generateECDSAKey(keyBits)
default:
return nil, "", fmt.Errorf("unknown private key type requested: %s", keyType)
}
}
func GeneratePrivateKey() (crypto.Signer, string, error) {
// TODO: find any calls to this func, replace with calls to GeneratePrivateKeyWithConfig()
// using prefs `private_key_type` and `private_key_bits`
return GeneratePrivateKeyWithConfig(DefaultPrivateKeyType, DefaultPrivateKeyBits)
2018-05-09 22:12:31 +00:00
}