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

NiluPlatform · Aug 19, 2018 · bb958d1 · bb958d1
1 parent d90f879
commit bb958d1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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