golang生成ssh 密钥方法

不多说 直接上代码

/*
 * Genarate rsa keys.
 */
package secure

import (
    "crypto/rsa"
    "crypto/rand"
    "crypto/x509"
    "encoding/pem"
    "golang.org/x/crypto/ssh"
    //"strings"
    //"github.com/golang/glog"
)




func GenerateKey(bits int) (*rsa.PrivateKey, *rsa.PublicKey, error) {
    private, err := rsa.GenerateKey(rand.Reader, bits)
    if err != nil {
        return nil, nil, err
    }
    return private, &private.PublicKey, nil

}

func EncodePrivateKey(private *rsa.PrivateKey) []byte {
    return pem.EncodeToMemory(&pem.Block{
        Bytes: x509.MarshalPKCS1PrivateKey(private),
        Type:  "RSA PRIVATE KEY",
    })
}

func EncodePublicKey(public *rsa.PublicKey) ([]byte, error) {
    publicBytes, err := x509.MarshalPKIXPublicKey(public)
    if err != nil {
        return nil, err
    }
    return pem.EncodeToMemory(&pem.Block{
        Bytes: publicBytes,
        Type:  "PUBLIC KEY",
    }), nil
}

//EncodeSSHKey
func EncodeSSHKey(public *rsa.PublicKey) ([]byte, error) {
    publicKey, err := ssh.NewPublicKey(public)
    if err != nil {
        return nil, err
    }
    return ssh.MarshalAuthorizedKey(publicKey), nil
}

func MakeSSHKeyPair() (string, string, error) {

    pkey, pubkey, err := GenerateKey(2048)
    if err != nil {
        return "", "", err
    }

    pub, err := EncodeSSHKey(pubkey)
    if err != nil {
        return "", "", err
    }

    //glog.Info("privateKey=[%s]\n pubKey=[%s]",string(EncodePrivateKey(pkey)),string(pub))
    return string(EncodePrivateKey(pkey)), string(pub), nil
}