Merge "Add function to generate SSH KeyPair via templater"

go.mod

@@ -32,6 +32,7 @@ require (
 	github.com/spf13/cobra v1.0.0
 	github.com/spf13/pflag v1.0.5
 	github.com/stretchr/testify v1.6.1
+	golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9
 	golang.org/x/net v0.0.0-20201021035429-f5854403a974 // indirect
 	golang.org/x/sys v0.0.0-20210119212857-b64e53b001e4 // indirect
 	k8s.io/api v0.17.9

pkg/document/plugin/templater/extlib/crypto.go

@@ -29,6 +29,8 @@ import (
 	"encoding/pem"
 	"math/big"
 	"time"
+
+	"golang.org/x/crypto/ssh"
 )
 
 func toUint32(i int) uint32 { return uint32(i) }
@@ -41,6 +43,11 @@ type dnParser struct {
 	dn    []string
 }
 
+type sshKey struct {
+	Private string
+	Public  string
+}
+
 func (p *dnParser) startOver() {
 	p.dn = append(p.dn, p.cur.String())
 	p.cur = bytes.Buffer{}
@@ -218,6 +225,34 @@ func generateCertificateAuthorityEx(
 	return generateCertificateAuthorityWithKeyInternalEx(subj, daysValid, priv)
 }
 
+// genSSHKeyPair make a pair of public and private keys for SSH access.
+// Public key is encoded in the format for inclusion in an OpenSSH authorized_keys file.
+// Private Key generated is PEM encoded
+func genSSHKeyPair(encryptionBit int) (sshKey, error) {
+	key := sshKey{}
+	privateKey, err := rsa.GenerateKey(rand.Reader, encryptionBit)
+	if err != nil {
+		return key, err
+	}
+
+	privateKeyPEM := &pem.Block{Type: "RSA PRIVATE KEY", Bytes: x509.MarshalPKCS1PrivateKey(privateKey)}
+	var private bytes.Buffer
+	if err = pem.Encode(&private, privateKeyPEM); err != nil {
+		return key, err
+	}
+
+	// generate public key
+	pub, err := ssh.NewPublicKey(&privateKey.PublicKey)
+	if err != nil {
+		return key, err
+	}
+
+	public := ssh.MarshalAuthorizedKey(pub)
+	key.Public = string(public)
+	key.Private = private.String()
+	return key, nil
+}
+
 func generateCertificateAuthorityWithPEMKeyEx(
 	subj string,
 	daysValid int,

pkg/document/plugin/templater/extlib/crypto_test.go

@@ -168,3 +168,12 @@ func TestNameFromString(t *testing.T) {
 		assert.Equal(t, tc.expectedOut, *r)
 	}
 }
+
+func TestGenSSHKeyPair(t *testing.T) {
+	key, err := genSSHKeyPair(2048)
+	assert.Nil(t, err)
+	assert.NotNil(t, key.Private)
+	assert.NotNil(t, key.Public)
+	assert.Contains(t, key.Private, "RSA PRIVATE KEY")
+	assert.Contains(t, key.Public, "ssh-rsa")
+}

pkg/document/plugin/templater/extlib/funcmap.go

@@ -32,6 +32,7 @@ var genericMap = map[string]interface{}{
 	"genCAWithKeyEx":         generateCertificateAuthorityWithPEMKeyEx,
 	"genSignedCertEx":        generateSignedCertificateEx,
 	"genSignedCertWithKeyEx": generateSignedCertificateWithPEMKeyEx,
+	"genSSHKeyPair":          genSSHKeyPair,
 	"regexGen":               regexGen,
 	"toYaml":                 toYaml,
 	"toUint32":               toUint32,