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

e7cc5b41 · cpusoft · Péter Szilágyi · 34ecb495 · e7cc5b41
Commit e7cc5b41 authored Apr 11, 2018 by cpusoft Committed by Péter Szilágyi Apr 11, 2018
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

peer.go les/peer.go +2 -0

No files found.
--- 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