Skip to content

Commit 09e07cf

Browse files
noottimwu20
authored andcommitted
fix(dot/network): add nil checks in connManager (ChainSafe#2069)
1 parent d635ed1 commit 09e07cf

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

dot/network/connmgr.go

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,9 @@ func (cm *ConnManager) unprotectedPeers(peers []peer.ID) []peer.ID {
130130
func (cm *ConnManager) Connected(n network.Network, c network.Conn) {
131131
logger.Tracef(
132132
"Host %s connected to peer %s", n.LocalPeer(), c.RemotePeer())
133-
cm.connectHandler(c.RemotePeer())
133+
if cm.connectHandler != nil {
134+
cm.connectHandler(c.RemotePeer())
135+
}
134136

135137
cm.Lock()
136138
defer cm.Unlock()
@@ -169,7 +171,9 @@ func (cm *ConnManager) Disconnected(_ network.Network, c network.Conn) {
169171
logger.Tracef("Host %s disconnected from peer %s", c.LocalPeer(), c.RemotePeer())
170172

171173
cm.Unprotect(c.RemotePeer(), "")
172-
cm.disconnectHandler(c.RemotePeer())
174+
if cm.disconnectHandler != nil {
175+
cm.disconnectHandler(c.RemotePeer())
176+
}
173177
}
174178

175179
// OpenedStream is called when a stream is opened

0 commit comments

Comments
 (0)