75 lines
1.6 KiB
Go
75 lines
1.6 KiB
Go
package models
|
|
|
|
import (
|
|
"crypto/rsa"
|
|
"crypto/x509"
|
|
"encoding/pem"
|
|
"errors"
|
|
"io/ioutil"
|
|
|
|
"github.com/go-redis/redis"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func ReadPublicKeyRSAFromString(pemString string) (*rsa.PublicKey, error) {
|
|
pemByte := []byte(pemString)
|
|
decoded, _ := pem.Decode(pemByte)
|
|
defer func() {
|
|
recover()
|
|
}()
|
|
keyInterface, err := x509.ParsePKIXPublicKey(decoded.Bytes)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
return nil, err
|
|
}
|
|
pub := keyInterface.(*rsa.PublicKey)
|
|
return pub, nil
|
|
}
|
|
|
|
func redisHGetOrCreateWithDefault(redisClient *redis.Client, key string, field string, defaultValue string) (string, error) {
|
|
keyExist, err := redisClient.HExists(key, field).Result()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if keyExist {
|
|
value, err := redisClient.HGet(key, field).Result()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return value, nil
|
|
} else {
|
|
_, err := redisClient.HSet(key, field, defaultValue).Result()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return defaultValue, nil
|
|
}
|
|
}
|
|
|
|
func readPrivateKeyRSA(keyPath string) (*rsa.PrivateKey, error) {
|
|
file, err := ioutil.ReadFile(keyPath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
decoded, _ := pem.Decode(file)
|
|
if decoded == nil {
|
|
return nil, errors.New("ACTOR_PEM IS INVALID. FAILED TO READ")
|
|
}
|
|
privateKey, err := x509.ParsePKCS1PrivateKey(decoded.Bytes)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return privateKey, nil
|
|
}
|
|
|
|
func generatePublicKeyPEMString(publicKey *rsa.PublicKey) string {
|
|
publicKeyByte := x509.MarshalPKCS1PublicKey(publicKey)
|
|
publicKeyPem := pem.EncodeToMemory(
|
|
&pem.Block{
|
|
Type: "RSA PUBLIC KEY",
|
|
Bytes: publicKeyByte,
|
|
},
|
|
)
|
|
return string(publicKeyPem)
|
|
}
|