fix(share/p2p/peer-manager) fix race for hasPeerCh pointer read (cele…

…stiaorg#2835)
walldiss · Oct 11, 2023 · 937eb43 · 937eb43
1 parent 3e56a04
commit 937eb43
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/share/p2p/peers/pool.go b/share/p2p/peers/pool.go
@@ -89,8 +89,11 @@ func (p *pool) next(ctx context.Context) <-chan peer.ID {
 				return
 			}
 
+			p.m.RLock()
+			hasPeerCh := p.hasPeerCh
+			p.m.RUnlock()
 			select {
-			case <-p.hasPeerCh:
+			case <-hasPeerCh:
 			case <-ctx.Done():
 				return
 			}