allowances/db.go

64 lines
1.1 KiB
Go
Raw Normal View History

package allowances
2013-02-21 00:19:21 -08:00
import (
2013-02-26 22:30:23 -08:00
"encoding/json"
"os"
2013-03-04 21:56:39 -08:00
"sync"
2013-02-21 00:19:21 -08:00
"golang.org/x/crypto/bcrypt"
"mcquay.me/fs"
)
2013-03-04 21:56:39 -08:00
var dbMutex sync.RWMutex
func GetHashes(filename string) (Passes, bool, error) {
r := []string{}
exists := false
if !fs.Exists(filename) {
return r, exists, nil
}
exists = true
f, err := os.Open(filename)
2013-02-26 22:30:23 -08:00
if err != nil {
return nil, exists, err
2013-02-26 22:30:23 -08:00
}
err = json.NewDecoder(f).Decode(&r)
2013-02-26 22:30:23 -08:00
if err != nil {
return nil, exists, err
2013-02-26 22:30:23 -08:00
}
return r, exists, nil
2013-02-26 22:30:23 -08:00
}
func AddPassword(filename, pw string) error {
curPasses, _, err := GetHashes(filename)
2013-02-26 22:30:23 -08:00
if err != nil {
return err
2013-02-26 22:30:23 -08:00
}
hpass, err := bcrypt.GenerateFromPassword(
[]byte(pw), bcrypt.DefaultCost)
curPasses = append(curPasses, string(hpass))
f, err := os.Create(filename)
2013-02-26 22:30:23 -08:00
if err != nil {
return err
2013-02-26 22:30:23 -08:00
}
if err := json.NewEncoder(f).Encode(curPasses); err != nil {
return err
}
return nil
2013-02-26 22:30:23 -08:00
}
2013-02-21 00:19:21 -08:00
type Passes []string
func (p Passes) Check(attempt string) (bool, error) {
// TODO: parallelize
for _, hash := range p {
2013-02-21 00:36:32 -08:00
err := bcrypt.CompareHashAndPassword([]byte(hash), []byte(attempt))
2013-02-21 00:19:21 -08:00
if err == nil {
return true, err
2013-02-21 00:19:21 -08:00
}
}
return false, nil
2013-02-21 00:19:21 -08:00
}