kopia/auth/credentials.go

package auth

import (
	"crypto/sha256"
	"io/ioutil"
	"sync"

	"golang.org/x/crypto/pbkdf2"
)

type Credentials interface {
	Username() string
	PrivateKey() *UserPrivateKey
}

type credentials struct {
	sync.Mutex
	once sync.Once

	username       string
	privateKey     *UserPrivateKey
	passwordPrompt func() string
}

func (pc *credentials) Username() string {
	return pc.username
}

func (pc *credentials) PrivateKey() *UserPrivateKey {
	pc.once.Do(pc.deriveKeyFromPassword)

	return pc.privateKey
}

func (pc *credentials) deriveKeyFromPassword() {
	if pc.privateKey != nil {
		return
	}

	password := pc.passwordPrompt()
	k := pbkdf2.Key([]byte(password), []byte(pc.username), pbkdf2Rounds, 32, sha256.New)
	pk, err := newPrivateKey(k)
	if err != nil {
		panic("should not happen")
	}
	pc.privateKey = pk
}

func Password(username, password string) Credentials {
	return &credentials{
		username: username,
		passwordPrompt: func() string {
			return password
		},
	}
}

func PasswordPrompt(username string, prompt func() string) Credentials {
	return &credentials{
		username:       username,
		passwordPrompt: prompt,
	}
}

func Key(username string, key []byte) (Credentials, error) {
	pk, err := newPrivateKey(key)
	if err != nil {
		return nil, err
	}

	return &credentials{
		username:   username,
		privateKey: pk,
	}, nil
}

func KeyFromFile(username string, fileName string) (Credentials, error) {
	k, err := ioutil.ReadFile(fileName)
	if err != nil {
		return nil, err
	}
	return Key(username, k)
}