delete rsa impl
This commit is contained in:
@@ -1,147 +0,0 @@
|
||||
package rsa
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"errors"
|
||||
"os"
|
||||
"log"
|
||||
|
||||
"crypto/rand"
|
||||
"crypto/rsa"
|
||||
"crypto/x509"
|
||||
"encoding/pem"
|
||||
)
|
||||
|
||||
|
||||
var (
|
||||
privateKeyName = "rsa_private_key.pem"
|
||||
publicKeyName = "rsa_public_key.pem"
|
||||
)
|
||||
|
||||
func GenerateKeys(path string) (*rsa.PrivateKey, *rsa.PublicKey, error) {
|
||||
// Generate a new RSA key pair
|
||||
privateKey, err := rsa.GenerateKey(rand.Reader, 2048)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
// Save the private key to a file
|
||||
privateKeyPath := path + privateKeyName
|
||||
privateKeyFile, err := os.Create(privateKeyPath)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
defer privateKeyFile.Close()
|
||||
|
||||
privateKeyPEM := pem.EncodeToMemory(&pem.Block{
|
||||
Type: "RSA PRIVATE KEY",
|
||||
Bytes: x509.MarshalPKCS1PrivateKey(privateKey),
|
||||
})
|
||||
_, err = privateKeyFile.Write(privateKeyPEM)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
fmt.Printf("Private key saved to %s\n", privateKeyPath)
|
||||
|
||||
// Extract and save the public key
|
||||
publicKeyPath := path + publicKeyName
|
||||
publicKeyFile, err := os.Create(publicKeyPath)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
defer publicKeyFile.Close()
|
||||
|
||||
publicKey := &privateKey.PublicKey
|
||||
publicKeyBytes, err := x509.MarshalPKIXPublicKey(publicKey)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
publicKeyPEM := pem.EncodeToMemory(&pem.Block{
|
||||
Type: "RSA PUBLIC KEY",
|
||||
Bytes: publicKeyBytes,
|
||||
})
|
||||
_, err = publicKeyFile.Write(publicKeyPEM)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
fmt.Printf("Public key saved to %s\n", publicKeyPath)
|
||||
|
||||
return privateKey, publicKey, nil
|
||||
}
|
||||
|
||||
func LoadKeys(path string) (*rsa.PrivateKey, *rsa.PublicKey, error) {
|
||||
// Load private key from file
|
||||
privateKeyPath := path + privateKeyName
|
||||
privateKeyData, err := os.ReadFile(privateKeyPath)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to read private key: %v", err)
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
block, _ := pem.Decode(privateKeyData)
|
||||
if block == nil || block.Type != "RSA PRIVATE KEY" {
|
||||
log.Fatalf("Failed to decode PEM block containing private key")
|
||||
return nil, nil, err
|
||||
}
|
||||
privateKey, err := x509.ParsePKCS1PrivateKey(block.Bytes)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to parse private key: %v", err)
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
// Load public key from file
|
||||
publicKeyPath := path + publicKeyName
|
||||
publicKeyData, err := os.ReadFile(publicKeyPath)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to read public key: %v", err)
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
block, _ = pem.Decode(publicKeyData)
|
||||
if block == nil || block.Type != "RSA PUBLIC KEY" {
|
||||
log.Fatalf("Failed to decode PEM block containing public key")
|
||||
return nil, nil, err
|
||||
}
|
||||
publicKeyInterface, err := x509.ParsePKIXPublicKey(block.Bytes)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to parse public key: %v", err)
|
||||
return nil, nil, err
|
||||
}
|
||||
publicKey := publicKeyInterface.(*rsa.PublicKey)
|
||||
|
||||
return privateKey, publicKey, nil
|
||||
}
|
||||
|
||||
|
||||
func LoadOrGenerateKeys(path string) (*rsa.PrivateKey, *rsa.PublicKey, error) {
|
||||
_, err := os.Open(path + privateKeyName)
|
||||
if errors.Is(err, os.ErrNotExist) {
|
||||
fmt.Println("Not Exists - Generate Keys")
|
||||
return GenerateKeys(path)
|
||||
}
|
||||
|
||||
return LoadKeys(path)
|
||||
}
|
||||
|
||||
|
||||
func Encrypt(data []byte, key *rsa.PublicKey) ([]byte, error) {
|
||||
ciphertext, err := rsa.EncryptPKCS1v15(rand.Reader, key, data)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to encrypt: %v", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return ciphertext, nil
|
||||
}
|
||||
|
||||
func Decrypt(data []byte, key *rsa.PrivateKey) ([]byte, error) {
|
||||
decrypted, err := rsa.DecryptPKCS1v15(rand.Reader, key, data)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to decrypt: %v", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return decrypted, nil
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user