add lock for new address

filecoin-project · Dec 21, 2021 · f7c167b · f7c167b
1 parent 876c949
commit f7c167b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pkg/wallet/dsbackend.go b/pkg/wallet/dsbackend.go
@@ -120,6 +120,9 @@ func (backend *DSBackend) HasAddress(addr address.Address) bool {
 // NewAddress creates a new address and stores it.
 // Safe for concurrent access.
 func (backend *DSBackend) NewAddress(protocol address.Protocol) (address.Address, error) {
+	backend.lk.Lock()
+	defer backend.lk.Unlock()
+
 	switch protocol {
 	case address.BLS:
 		return backend.newBLSAddress()
@@ -166,9 +169,6 @@ func (backend *DSBackend) putKeyInfo(ki *crypto.KeyInfo) error {
 		KeyInfo: ki,
 	}
 
-	backend.lk.Lock()
-	defer backend.lk.Unlock()
-
 	var keyJSON []byte
 	err = backend.UsePassword(func(password []byte) error {
 		var err error