-
Notifications
You must be signed in to change notification settings - Fork 617
/
vault_source.go
186 lines (163 loc) · 4.21 KB
/
vault_source.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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
package cert
import (
"crypto/tls"
"crypto/x509"
"fmt"
"log"
"path"
"strings"
"time"
"github.com/hashicorp/vault/api"
)
// VaultSource implements a certificate source which loads
// TLS and client authorization certificates from a Vault server.
// The Vault token should be set through the VAULT_TOKEN environment
// variable.
//
// The TLS certificates are updated automatically when Refresh
// is not zero. Refresh cannot be less than one second to prevent
// busy loops.
type VaultSource struct {
Client *vaultClient
CertPath string
ClientCAPath string
CAUpgradeCN string
Refresh time.Duration
}
func (s *VaultSource) LoadClientCAs() (*x509.CertPool, error) {
if s.ClientCAPath == "" {
return nil, nil
}
return newCertPool(s.ClientCAPath, s.CAUpgradeCN, s.load)
}
func (s *VaultSource) Certificates() chan []tls.Certificate {
ch := make(chan []tls.Certificate, 1)
go watch(ch, s.Refresh, s.CertPath, s.load)
return ch
}
func (s *VaultSource) load(path string) (pemBlocks map[string][]byte, err error) {
pemBlocks = map[string][]byte{}
// get will read a key=value pair from the secret
// and store it as <name>-{cert,key}.pem so that
// they are recognized by the post-processing function
// which assembles the certificates.
// The value can be stored either as string or []byte.
get := func(name, typ string, secret *api.Secret, v2 bool) {
data := secret.Data
if v2 {
x, ok := secret.Data["data"]
if !ok {
return
}
data, ok = x.(map[string]interface{})
if !ok {
return
}
}
v := data[typ]
if v == nil {
return
}
var b []byte
switch v.(type) {
case string:
b = []byte(v.(string))
case []byte:
b = v.([]byte)
default:
log.Printf("[WARN] cert: key %s has type %T", name, v)
return
}
pemBlocks[name+"-"+typ+".pem"] = b
}
c, err := s.Client.Get()
if err != nil {
return nil, fmt.Errorf("vault: client: %s", err)
}
mountPath, v2, err := s.isKVv2(path, c)
if err != nil {
return nil, fmt.Errorf("vault: query mount path: %s", err)
}
// get the subkeys under 'path'.
// Each subkey refers to a certificate.
p := path
if v2 {
p = s.addPrefixToVKVPath(p, mountPath, "metadata")
}
certs, err := c.Logical().List(p)
if err != nil {
return nil, fmt.Errorf("vault: list: %s", err)
}
if certs == nil || certs.Data["keys"] == nil {
return nil, nil
}
for _, x := range certs.Data["keys"].([]interface{}) {
name := x.(string)
p := path + "/" + name
if v2 {
p = s.addPrefixToVKVPath(p, mountPath, "data")
}
secret, err := c.Logical().Read(p)
if err != nil {
log.Printf("[WARN] cert: Failed to read %s from Vault: %s", p, err)
continue
}
if secret == nil {
log.Printf("[WARN] cert: Failed to find %s in Vault: %s", p, err)
continue
}
get(name, "cert", secret, v2)
get(name, "key", secret, v2)
}
return pemBlocks, nil
}
func (s *VaultSource) addPrefixToVKVPath(p, mountPath, apiPrefix string) string {
p = strings.TrimPrefix(p, mountPath)
return path.Join(mountPath, apiPrefix, p)
}
func (s *VaultSource) isKVv2(path string, client *api.Client) (string, bool, error) {
mountPath, version, err := s.kvPreflightVersionRequest(client, path)
if err != nil {
return "", false, err
}
return mountPath, version == 2, nil
}
func (s *VaultSource) kvPreflightVersionRequest(client *api.Client, path string) (string, int, error) {
r := client.NewRequest("GET", "/v1/sys/internal/ui/mounts/"+path)
resp, err := client.RawRequest(r)
if resp != nil {
defer resp.Body.Close()
}
if err != nil {
// If we get a 404 we are using an older version of vault, default to
// version 1
if resp != nil && resp.StatusCode == 404 {
return "", 1, nil
}
return "", 0, err
}
secret, err := api.ParseSecret(resp.Body)
if err != nil {
return "", 0, err
}
var mountPath string
if mountPathRaw, ok := secret.Data["path"]; ok {
mountPath = mountPathRaw.(string)
}
options := secret.Data["options"]
if options == nil {
return mountPath, 1, nil
}
versionRaw := options.(map[string]interface{})["version"]
if versionRaw == nil {
return mountPath, 1, nil
}
version := versionRaw.(string)
switch version {
case "", "1":
return mountPath, 1, nil
case "2":
return mountPath, 2, nil
}
return mountPath, 1, nil
}