forked from TrueCloudLab/lego
134 lines
3.4 KiB
Go
134 lines
3.4 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/xenolf/lego/acme"
|
|
"github.com/xenolf/lego/log"
|
|
)
|
|
|
|
// Account represents a users local saved credentials
|
|
type Account struct {
|
|
Email string `json:"email"`
|
|
key crypto.PrivateKey
|
|
Registration *acme.RegistrationResource `json:"registration"`
|
|
|
|
conf *Configuration
|
|
}
|
|
|
|
// NewAccount creates a new account for an email address
|
|
func NewAccount(email string, conf *Configuration) *Account {
|
|
accKeysPath := conf.AccountKeysPath(email)
|
|
// TODO: move to function in configuration?
|
|
accKeyPath := accKeysPath + string(os.PathSeparator) + email + ".key"
|
|
if err := checkFolder(accKeysPath); err != nil {
|
|
log.Fatalf("Could not check/create directory for account %s: %v", email, err)
|
|
}
|
|
|
|
var privKey crypto.PrivateKey
|
|
if _, err := os.Stat(accKeyPath); os.IsNotExist(err) {
|
|
|
|
log.Printf("No key found for account %s. Generating a curve P384 EC key.", email)
|
|
privKey, err = generatePrivateKey(accKeyPath)
|
|
if err != nil {
|
|
log.Fatalf("Could not generate RSA private account key for account %s: %v", email, err)
|
|
}
|
|
|
|
log.Printf("Saved key to %s", accKeyPath)
|
|
} else {
|
|
privKey, err = loadPrivateKey(accKeyPath)
|
|
if err != nil {
|
|
log.Fatalf("Could not load RSA private key from file %s: %v", accKeyPath, err)
|
|
}
|
|
}
|
|
|
|
accountFile := path.Join(conf.AccountPath(email), "account.json")
|
|
if _, err := os.Stat(accountFile); os.IsNotExist(err) {
|
|
return &Account{Email: email, key: privKey, conf: conf}
|
|
}
|
|
|
|
fileBytes, err := ioutil.ReadFile(accountFile)
|
|
if err != nil {
|
|
log.Fatalf("Could not load file for account %s -> %v", email, err)
|
|
}
|
|
|
|
var acc Account
|
|
err = json.Unmarshal(fileBytes, &acc)
|
|
if err != nil {
|
|
log.Fatalf("Could not parse file for account %s -> %v", email, err)
|
|
}
|
|
|
|
acc.key = privKey
|
|
acc.conf = conf
|
|
|
|
if acc.Registration == nil || acc.Registration.Body.Status == "" {
|
|
reg, err := tryRecoverAccount(privKey, conf)
|
|
if err != nil {
|
|
log.Fatalf("Could not load account for %s. Registration is nil -> %#v", email, err)
|
|
}
|
|
|
|
acc.Registration = reg
|
|
err = acc.Save()
|
|
if err != nil {
|
|
log.Fatalf("Could not save account for %s. Registration is nil -> %#v", email, err)
|
|
}
|
|
}
|
|
|
|
if acc.conf == nil {
|
|
log.Fatalf("Could not load account for %s. Configuration is nil.", email)
|
|
}
|
|
|
|
return &acc
|
|
}
|
|
|
|
func tryRecoverAccount(privKey crypto.PrivateKey, conf *Configuration) (*acme.RegistrationResource, error) {
|
|
// couldn't load account but got a key. Try to look the account up.
|
|
serverURL := conf.context.GlobalString("server")
|
|
client, err := acme.NewClient(serverURL, &Account{key: privKey, conf: conf}, acme.RSA2048)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
reg, err := client.ResolveAccountByKey()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return reg, nil
|
|
}
|
|
|
|
/** Implementation of the acme.User interface **/
|
|
|
|
// GetEmail returns the email address for the account
|
|
func (a *Account) GetEmail() string {
|
|
return a.Email
|
|
}
|
|
|
|
// GetPrivateKey returns the private RSA account key.
|
|
func (a *Account) GetPrivateKey() crypto.PrivateKey {
|
|
return a.key
|
|
}
|
|
|
|
// GetRegistration returns the server registration
|
|
func (a *Account) GetRegistration() *acme.RegistrationResource {
|
|
return a.Registration
|
|
}
|
|
|
|
/** End **/
|
|
|
|
// Save the account to disk
|
|
func (a *Account) Save() error {
|
|
jsonBytes, err := json.MarshalIndent(a, "", "\t")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return ioutil.WriteFile(
|
|
path.Join(a.conf.AccountPath(a.Email), "account.json"),
|
|
jsonBytes,
|
|
0600,
|
|
)
|
|
}
|