You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
1.6 KiB
Go
69 lines
1.6 KiB
Go
package starcrypto
|
|
|
|
import (
|
|
"crypto"
|
|
"crypto/ecdsa"
|
|
"crypto/rsa"
|
|
"crypto/x509"
|
|
"encoding/pem"
|
|
"errors"
|
|
"golang.org/x/crypto/ssh"
|
|
)
|
|
|
|
func EncodePrivateKey(private crypto.PrivateKey, secret string) ([]byte, error) {
|
|
switch private.(type) {
|
|
case *rsa.PrivateKey:
|
|
return EncodeRsaPrivateKey(private.(*rsa.PrivateKey), secret)
|
|
case *ecdsa.PrivateKey:
|
|
return EncodeEcdsaPrivateKey(private.(*ecdsa.PrivateKey), secret)
|
|
default:
|
|
return nil, errors.New("private key type error")
|
|
}
|
|
}
|
|
|
|
func EncodePublicKey(public crypto.PublicKey) ([]byte, error) {
|
|
switch public.(type) {
|
|
case *rsa.PublicKey:
|
|
return EncodeRsaPublicKey(public.(*rsa.PublicKey))
|
|
case *ecdsa.PublicKey:
|
|
return EncodeEcdsaPublicKey(public.(*ecdsa.PublicKey))
|
|
default:
|
|
return nil, errors.New("public key type error")
|
|
}
|
|
}
|
|
|
|
func DecodePrivateKey(private []byte, password string) (crypto.PrivateKey, error) {
|
|
blk, _ := pem.Decode(private)
|
|
if blk == nil {
|
|
return nil, errors.New("private key error")
|
|
}
|
|
switch blk.Type {
|
|
case "RSA PRIVATE KEY":
|
|
return DecodeRsaPrivateKey(private, password)
|
|
case "EC PRIVATE KEY":
|
|
return DecodeEcdsaPrivateKey(private, password)
|
|
default:
|
|
return nil, errors.New("private key type error")
|
|
}
|
|
}
|
|
|
|
func DecodePublicKey(pubStr []byte) (crypto.PublicKey, error) {
|
|
blk, _ := pem.Decode(pubStr)
|
|
if blk == nil {
|
|
return nil, errors.New("public key error")
|
|
}
|
|
pub, err := x509.ParsePKIXPublicKey(blk.Bytes)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return pub, nil
|
|
}
|
|
|
|
func EncodeSSHPublicKey(public crypto.PublicKey) ([]byte, error) {
|
|
publicKey, err := ssh.NewPublicKey(public)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return ssh.MarshalAuthorizedKey(publicKey), nil
|
|
}
|