You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
1.6 KiB
Go
76 lines
1.6 KiB
Go
package internal
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"crypto/x509"
|
|
"io/ioutil"
|
|
"sync"
|
|
)
|
|
|
|
var (
|
|
accounts = make(map[string]Account)
|
|
tlsConfigs = make(map[string]*tls.Config)
|
|
lock sync.RWMutex
|
|
)
|
|
|
|
// Account holds the username/password for an etcd cluster.
|
|
type Account struct {
|
|
User string
|
|
Pass string
|
|
}
|
|
|
|
// AddAccount adds the username/password for the given etcd cluster.
|
|
func AddAccount(endpoints []string, user, pass string) {
|
|
lock.Lock()
|
|
defer lock.Unlock()
|
|
|
|
accounts[getClusterKey(endpoints)] = Account{
|
|
User: user,
|
|
Pass: pass,
|
|
}
|
|
}
|
|
|
|
// AddTLS adds the tls cert files for the given etcd cluster.
|
|
func AddTLS(endpoints []string, certFile, certKeyFile, caFile string, insecureSkipVerify bool) error {
|
|
cert, err := tls.LoadX509KeyPair(certFile, certKeyFile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
caData, err := ioutil.ReadFile(caFile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
pool := x509.NewCertPool()
|
|
pool.AppendCertsFromPEM(caData)
|
|
|
|
lock.Lock()
|
|
defer lock.Unlock()
|
|
tlsConfigs[getClusterKey(endpoints)] = &tls.Config{
|
|
Certificates: []tls.Certificate{cert},
|
|
RootCAs: pool,
|
|
InsecureSkipVerify: insecureSkipVerify,
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetAccount gets the username/password for the given etcd cluster.
|
|
func GetAccount(endpoints []string) (Account, bool) {
|
|
lock.RLock()
|
|
defer lock.RUnlock()
|
|
|
|
account, ok := accounts[getClusterKey(endpoints)]
|
|
return account, ok
|
|
}
|
|
|
|
// GetTLS gets the tls config for the given etcd cluster.
|
|
func GetTLS(endpoints []string) (*tls.Config, bool) {
|
|
lock.RLock()
|
|
defer lock.RUnlock()
|
|
|
|
cfg, ok := tlsConfigs[getClusterKey(endpoints)]
|
|
return cfg, ok
|
|
}
|