-
Notifications
You must be signed in to change notification settings - Fork 6
/
secret_credentials.go
123 lines (103 loc) · 2.63 KB
/
secret_credentials.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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package btc
import (
"context"
"errors"
"time"
uuid "github.com/hashicorp/go-uuid"
"github.com/hashicorp/vault/helper/salt"
"github.com/hashicorp/vault/logical"
"github.com/hashicorp/vault/logical/framework"
)
func secretCredentials(b *backend) *framework.Secret {
return &framework.Secret{
Type: SecretCredsType,
Fields: map[string]*framework.FieldSchema{
"token": &framework.FieldSchema{
Type: framework.TypeString,
Description: "Access token for wallets",
},
},
DefaultDuration: time.Duration(5 * time.Minute),
Revoke: b.secretCredsRevoke,
}
}
func (b *backend) secretCredsRevoke(ctx context.Context, req *logical.Request, d *framework.FieldData) (*logical.Response, error) {
id, ok := req.Secret.InternalData["token"].(string)
if !ok {
return nil, errors.New(MissingInternalDataError)
}
s, err := salt.NewSalt(ctx, req.Storage, nil)
if err != nil {
return nil, err
}
err = req.Storage.Delete(ctx, PathCreds+s.SaltID(id))
if err != nil {
return nil, err
}
return nil, nil
}
func newToken(ctx context.Context, s logical.Storage, config *salt.Config) (string, string, error) {
token, err := uuid.GenerateUUID()
if err != nil {
return "", "", err
}
newSalt, err := salt.NewSalt(ctx, s, config)
if err != nil {
return "", "", err
}
return token, newSalt.SaltID(token), nil
}
func (b *backend) GetToken(ctx context.Context, s logical.Storage, token string, walletType int) (*credential, error) {
newSalt, err := salt.NewSalt(ctx, s, nil)
if err != nil {
return nil, err
}
leaseID := newSalt.SaltID(token)
var path string
switch walletType {
case StandardType:
path = PathCreds
case MultiSigType:
path = PathMultiSigCreds
case SegWitType:
path = PathSegWitCreds
}
path = path + leaseID
entry, err := s.Get(ctx, path)
if err != nil {
return nil, err
}
if entry == nil {
return nil, nil
}
var cred credential
if err := entry.DecodeJSON(&cred); err != nil {
return nil, err
}
return &cred, nil
}
func (b *backend) RevokeToken(ctx context.Context, store logical.Storage, token *credential, walletType int) error {
var secretType string
switch walletType {
case StandardType:
secretType = SecretCredsType
case MultiSigType:
secretType = MultiSigSecretCredsType
case SegWitType:
secretType = SegWitSecretCredsType
}
secret := b.Secret(secretType)
request := &logical.Request{
Storage: store,
Operation: logical.RevokeOperation,
Secret: &logical.Secret{
InternalData: map[string]interface{}{"token": token.Token},
LeaseID: token.LeaseID,
},
}
_, err := secret.HandleRevoke(ctx, request)
if err != nil {
return err
}
return nil
}