fix(dot/network): add nil checks in connManager (#2069)

ChainSafe · Nov 25, 2021 · 7f9c042 · 7f9c042
1 parent 2f62832
commit 7f9c042
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/dot/network/connmgr.go b/dot/network/connmgr.go
@@ -130,7 +130,9 @@ func (cm *ConnManager) unprotectedPeers(peers []peer.ID) []peer.ID {
 func (cm *ConnManager) Connected(n network.Network, c network.Conn) {
 	logger.Tracef(
 		"Host %s connected to peer %s", n.LocalPeer(), c.RemotePeer())
-	cm.connectHandler(c.RemotePeer())
+	if cm.connectHandler != nil {
+		cm.connectHandler(c.RemotePeer())
+	}
 
 	cm.Lock()
 	defer cm.Unlock()
@@ -169,7 +171,9 @@ func (cm *ConnManager) Disconnected(_ network.Network, c network.Conn) {
 	logger.Tracef("Host %s disconnected from peer %s", c.LocalPeer(), c.RemotePeer())
 
 	cm.Unprotect(c.RemotePeer(), "")
-	cm.disconnectHandler(c.RemotePeer())
+	if cm.disconnectHandler != nil {
+		cm.disconnectHandler(c.RemotePeer())
+	}
 }
 
 // OpenedStream is called when a stream is opened