les: add ps.lock.Unlock() before return (#16360)

7 years ago · e7cc5b4160
parent 34ecb495b6
commit e7cc5b4160
1 changed files with 2 additions and 0 deletions
--- a/les/peer.go
+++ b/les/peer.go
@ -545,9 +545,11 @@ func (ps *peerSet) notify(n peerSetNotify) {
 func (ps *peerSet) Register(p *peer) error {
 	ps.lock.Lock()
 	if ps.closed {
+		ps.lock.Unlock()
 		return errClosed
 	}
 	if _, ok := ps.peers[p.id]; ok {
+		ps.lock.Unlock()
 		return errAlreadyRegistered
 	}
 	ps.peers[p.id] = p