add ecdsa and ed25519

Author: TheFox0x7

modules/ssh/ssh.go

@@ -6,6 +6,9 @@ package ssh
 import (
 	"bytes"
 	"context"
+	"crypto/ecdsa"
+	"crypto/ed25519"
+	"crypto/elliptic"
 	"crypto/rand"
 	"crypto/rsa"
 	"crypto/x509"
@@ -58,7 +61,9 @@ const giteaPermissionExtensionKeyID = "gitea-perm-ext-key-id"
 type KeyType string
 
 const (
-	RSA KeyType = "rsa"
+	RSA     KeyType = "rsa"
+	ECDSA   KeyType = "ecdsa"
+	ED25519 KeyType = "ed25519"
 )
 
 func getExitStatusFromError(err error) int {
@@ -458,6 +463,15 @@ func keyGen(keytype KeyType) (any, any, error) {
 			return nil, nil, err
 		}
 		return privateKey, &privateKey.PublicKey, nil
+	case ED25519:
+		pub, priv, err := ed25519.GenerateKey(rand.Reader)
+		return priv, pub, err
+	case ECDSA:
+		priv, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
+		if err != nil {
+			return nil, nil, err
+		}
+		return priv, &priv.PublicKey, nil
 	default:
 		return nil, nil, errors.New("unknown keyType")
 	}

modules/ssh/ssh_test.go

@@ -1,6 +1,8 @@
 package ssh_test
 
 import (
+	"crypto/ecdsa"
+	"crypto/ed25519"
 	"crypto/rsa"
 	"crypto/x509"
 	"encoding/pem"
@@ -22,6 +24,14 @@ func TestGenKeyPair(t *testing.T) {
 			keyType:      ssh.RSA,
 			expectedType: &rsa.PrivateKey{},
 		},
+		{
+			keyType:      ssh.ED25519,
+			expectedType: ed25519.PrivateKey{},
+		},
+		{
+			keyType:      ssh.ECDSA,
+			expectedType: &ecdsa.PrivateKey{},
+		},
 	}
 	for _, tC := range testCases {
 		t.Run("Generate"+string(tC.keyType), func(t *testing.T) {