feat: 新增生成SSHKey接口

Author: rehiy

api/keypair/keygen.go

@@ -0,0 +1,36 @@
+package keypair
+
+import (
+	"github.com/gin-gonic/gin"
+
+	"tdp-cloud/helper/secure"
+)
+
+func keygen(c *gin.Context) {
+
+	var rq struct {
+		KeyType string
+	}
+
+	if err := c.ShouldBind(&rq); err != nil {
+		c.Set("Error", err)
+		return
+	}
+
+	var (
+		err                   error
+		privateKey, publicKey string
+	)
+
+	switch rq.KeyType {
+	case "ssh":
+		privateKey, publicKey, err = secure.NewSSHKeypair()
+	}
+
+	if err == nil {
+		c.Set("Payload", gin.H{"PrivateKey": privateKey, "PublicKey": publicKey})
+	} else {
+		c.Set("Error", err)
+	}
+
+}

api/keypair/router.go

@@ -18,6 +18,8 @@ func Router(api *gin.RouterGroup) {
 		rg.POST("/detail", detail)
 		rg.POST("/update", update)
 		rg.POST("/delete", delete)
+
+		rg.POST("/keygen", keygen)
 	}
 
 }

helper/secure/ssh.go

@@ -0,0 +1,44 @@
+package secure
+
+import (
+	"crypto/rand"
+	"crypto/rsa"
+	"crypto/x509"
+	"encoding/pem"
+
+	"golang.org/x/crypto/ssh"
+)
+
+func NewSSHKeypair() (string, string, error) {
+
+	privateKey, err := rsa.GenerateKey(rand.Reader, 4096)
+	if err != nil {
+		return "", "", err
+	}
+
+	privateKeyBytes := pem.EncodeToMemory(&pem.Block{
+		Type:  "RSA PRIVATE KEY",
+		Bytes: x509.MarshalPKCS1PrivateKey(privateKey),
+	})
+
+	publicKeyBytes, err := NewSSHPublicKey(&privateKey.PublicKey)
+	if err != nil {
+		return "", "", err
+	}
+
+	return string(privateKeyBytes), string(publicKeyBytes), nil
+
+}
+
+func NewSSHPublicKey(privatekey *rsa.PublicKey) ([]byte, error) {
+
+	publicKey, err := ssh.NewPublicKey(privatekey)
+	if err != nil {
+		return nil, err
+	}
+
+	publicKeyBytes := ssh.MarshalAuthorizedKey(publicKey)
+
+	return publicKeyBytes, nil
+
+}