/
rsa.go
108 lines (98 loc) · 2.26 KB
/
rsa.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
package cipher
import (
"crypto/rand"
"crypto/rsa"
"crypto/x509"
"encoding/pem"
"log"
"os"
"runtime"
)
const (
privateKeyPrefix = "PRIVATE KEY"
publicKeyPrefix = "PUBLIC KEY"
privateFileName = "private.pem"
publicFileName = "public.pem"
)
func init() {
log.SetFlags(log.Ldate | log.Lshortfile)
}
func RsaEncrypt(plainText, key []byte) (cryptText []byte, err error) {
block, _ := pem.Decode(key)
defer func() {
if err := recover(); err != nil {
switch err.(type) {
case runtime.Error:
log.Println("runtime err:", err, "Check that the key is correct")
default:
log.Println("error:", err)
}
}
}()
publicKeyInterface, err := x509.ParsePKIXPublicKey(block.Bytes)
if err != nil {
return nil, err
}
publicKey := publicKeyInterface.(*rsa.PublicKey)
cipherText, err := rsa.EncryptPKCS1v15(rand.Reader, publicKey, plainText)
if err != nil {
return nil, err
}
return cipherText, nil
}
func RsaDecrypt(cryptText, key []byte) (plainText []byte, err error) {
block, _ := pem.Decode(key)
defer func() {
if err := recover(); err != nil {
switch err.(type) {
case runtime.Error:
log.Println("runtime err:", err, "Check that the key is correct")
default:
log.Println("error:", err)
}
}
}()
privateKey, err := x509.ParsePKCS1PrivateKey(block.Bytes)
if err != nil {
return []byte{}, err
}
plainText, err = rsa.DecryptPKCS1v15(rand.Reader, privateKey, cryptText)
if err != nil {
return []byte{}, err
}
return plainText, nil
}
func GetRsaKey() error {
privateKey, err := rsa.GenerateKey(rand.Reader, 2048)
if err != nil {
return err
}
x509PrivateKey := x509.MarshalPKCS1PrivateKey(privateKey)
privateFile, err := os.Create(privateFileName)
if err != nil {
return err
}
defer privateFile.Close()
privateBlock := pem.Block{
Type: privateKeyPrefix,
Bytes: x509PrivateKey,
}
if err = pem.Encode(privateFile, &privateBlock); err != nil {
return err
}
publicKey := privateKey.PublicKey
x509PublicKey, err := x509.MarshalPKIXPublicKey(&publicKey)
if err != nil {
panic(err)
}
publicFile, _ := os.Create(publicFileName)
defer publicFile.Close()
publicBlock := pem.Block{
Type: publicKeyPrefix,
Bytes: x509PublicKey,
}
if err = pem.Encode(publicFile, &publicBlock); err != nil {
return err
}
return nil
}