aboutsummaryrefslogtreecommitdiffstats
path: root/accounts/account_manager.go
diff options
context:
space:
mode:
authorFelix Lange <fjl@twurst.com>2015-03-10 07:09:39 +0800
committerFelix Lange <fjl@twurst.com>2015-03-10 07:19:01 +0800
commit487f68ec4892794cb994cffd95d5bc2bf3052d3e (patch)
tree85c8fa9c2bef1e671ceeb782c9e706986c064d94 /accounts/account_manager.go
parent9bf513e99305af733110cf23a0b47c8e73359010 (diff)
downloadgo-tangerine-487f68ec4892794cb994cffd95d5bc2bf3052d3e.tar.gz
go-tangerine-487f68ec4892794cb994cffd95d5bc2bf3052d3e.tar.zst
go-tangerine-487f68ec4892794cb994cffd95d5bc2bf3052d3e.zip
accounts: add {Timed,}Unlock, remove SignLocked
Diffstat (limited to 'accounts/account_manager.go')
-rw-r--r--accounts/account_manager.go48
1 files changed, 31 insertions, 17 deletions
diff --git a/accounts/account_manager.go b/accounts/account_manager.go
index 4575334bf..fdd7d83e9 100644
--- a/accounts/account_manager.go
+++ b/accounts/account_manager.go
@@ -54,10 +54,9 @@ type Account struct {
}
type Manager struct {
- keyStore crypto.KeyStore2
- unlocked map[string]*unlocked
- unlockTime time.Duration
- mutex sync.RWMutex
+ keyStore crypto.KeyStore2
+ unlocked map[string]*unlocked
+ mutex sync.RWMutex
}
type unlocked struct {
@@ -65,11 +64,10 @@ type unlocked struct {
abort chan struct{}
}
-func NewManager(keyStore crypto.KeyStore2, unlockTime time.Duration) *Manager {
+func NewManager(keyStore crypto.KeyStore2) *Manager {
return &Manager{
- keyStore: keyStore,
- unlocked: make(map[string]*unlocked),
- unlockTime: unlockTime,
+ keyStore: keyStore,
+ unlocked: make(map[string]*unlocked),
}
}
@@ -115,15 +113,28 @@ func (am *Manager) Sign(a Account, toSign []byte) (signature []byte, err error)
return signature, err
}
-func (am *Manager) SignLocked(a Account, keyAuth string, toSign []byte) (signature []byte, err error) {
- key, err := am.keyStore.GetKey(a.Address, keyAuth)
+// TimedUnlock unlocks the account with the given address.
+// When timeout has passed, the account will be locked again.
+func (am *Manager) TimedUnlock(addr []byte, keyAuth string, timeout time.Duration) error {
+ key, err := am.keyStore.GetKey(addr, keyAuth)
if err != nil {
- return nil, err
+ return err
}
- u := am.addUnlocked(a.Address, key)
- go am.dropLater(a.Address, u)
- signature, err = crypto.Sign(toSign, key.PrivateKey)
- return signature, err
+ u := am.addUnlocked(addr, key)
+ go am.dropLater(addr, u, timeout)
+ return nil
+}
+
+// Unlock unlocks the account with the given address. The account
+// stays unlocked until the program exits or until a TimedUnlock
+// timeout (started after the call to Unlock) expires.
+func (am *Manager) Unlock(addr []byte, keyAuth string) error {
+ key, err := am.keyStore.GetKey(addr, keyAuth)
+ if err != nil {
+ return err
+ }
+ am.addUnlocked(addr, key)
+ return nil
}
func (am *Manager) NewAccount(auth string) (Account, error) {
@@ -155,6 +166,9 @@ func (am *Manager) addUnlocked(addr []byte, key *crypto.Key) *unlocked {
if found {
// terminate dropLater for this key to avoid unexpected drops.
close(prev.abort)
+ // the key is zeroed here instead of in dropLater because
+ // there might not actually be a dropLater running for this
+ // key, i.e. when Unlock was used.
zeroKey(prev.PrivateKey)
}
am.unlocked[string(addr)] = u
@@ -162,8 +176,8 @@ func (am *Manager) addUnlocked(addr []byte, key *crypto.Key) *unlocked {
return u
}
-func (am *Manager) dropLater(addr []byte, u *unlocked) {
- t := time.NewTimer(am.unlockTime)
+func (am *Manager) dropLater(addr []byte, u *unlocked, timeout time.Duration) {
+ t := time.NewTimer(timeout)
defer t.Stop()
select {
case <-u.abort: