-
Notifications
You must be signed in to change notification settings - Fork 2
/
key.go
94 lines (84 loc) · 1.66 KB
/
key.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
package pki
import (
"crypto/ecdsa"
"crypto/ed25519"
"crypto/elliptic"
"crypto/rand"
"crypto/rsa"
"crypto/x509"
"encoding/pem"
"errors"
"os"
)
type Key struct {
Value any
Algo string
}
const (
DefaultCrtAlgo = RSA
UnsupportedAlgorithm = ""
EDDSA = "eddsa"
ECDSA = "ecdsa"
RSA = "rsa"
)
func ParseKey(path string) (*Key, error) {
data, err := os.ReadFile(path)
if err != nil {
return nil, err
}
block, _ := pem.Decode(data)
keyData, err := x509.ParsePKCS8PrivateKey(block.Bytes)
if err != nil {
keyData, err = x509.ParsePKCS1PrivateKey(block.Bytes)
if err != nil {
keyData, err = x509.ParseECPrivateKey(block.Bytes)
if err != nil {
return nil, err
}
}
}
key := Key{keyData, UnsupportedAlgorithm}
switch keyData.(type) {
case *rsa.PrivateKey:
key.Algo = RSA
case *ecdsa.PrivateKey:
key.Algo = ECDSA
case ed25519.PrivateKey:
key.Algo = EDDSA
default:
key.Algo = UnsupportedAlgorithm
}
return &key, nil
}
func newKey(algo string) (*Key, error) {
var (
key any
err error
)
switch algo {
case EDDSA:
_, key, err = ed25519.GenerateKey(rand.Reader)
case ECDSA:
key, err = ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
case RSA:
key, err = rsa.GenerateKey(rand.Reader, 4096)
default:
return nil, errors.New("requested algorithm not supported")
}
if err != nil {
return nil, err
}
return &Key{key, algo}, nil
}
func (key *Key) Public() any {
switch k := key.Value.(type) {
case *rsa.PrivateKey:
return &k.PublicKey
case *ecdsa.PrivateKey:
return &k.PublicKey
case ed25519.PrivateKey:
return k.Public().(ed25519.PublicKey)
default:
return nil
}
}