@ -3557,32 +3557,49 @@ void ProcessMessage(
return ;
return ;
}
}
/** Maybe disconnect a peer and discourage future connections from its address.
*
* @ param [ in ] pnode The node to check .
* @ return True if the peer was marked for disconnection in this function
*/
bool PeerLogicValidation : : MaybeDiscourageAndDisconnect ( CNode & pnode )
bool PeerLogicValidation : : MaybeDiscourageAndDisconnect ( CNode & pnode )
{
{
LOCK ( cs_main ) ;
NodeId peer_id { pnode . GetId ( ) } ;
CNodeState & state = * State ( pnode . GetId ( ) ) ;
{
LOCK ( cs_main ) ;
CNodeState & state = * State ( peer_id ) ;
if ( state . m_should_discourage ) {
// There's nothing to do if the m_should_discourage flag isn't set
if ( ! state . m_should_discourage ) return false ;
// Reset m_should_discourage
state . m_should_discourage = false ;
state . m_should_discourage = false ;
if ( pnode . HasPermission ( PF_NOBAN ) ) {
} // cs_main
LogPrintf ( " Warning: not punishing whitelisted peer %s! \n " , pnode . addr . ToString ( ) ) ;
} else if ( pnode . m_manual_connection ) {
if ( pnode . HasPermission ( PF_NOBAN ) ) {
LogPrintf ( " Warning: not punishing manually-connected peer %s! \n " , pnode . addr . ToString ( ) ) ;
// Peer has the NOBAN permission flag - log but don't disconnect
} else if ( pnode . addr . IsLocal ( ) ) {
LogPrintf ( " Warning: not punishing noban peer %d! \n " , peer_id ) ;
// Disconnect but don't discourage this local node
return false ;
LogPrintf ( " Warning: disconnecting but not discouraging local peer %s! \n " , pnode . addr . ToString ( ) ) ;
}
pnode . fDisconnect = true ;
} else {
if ( pnode . m_manual_connection ) {
// Disconnect and discourage all nodes sharing the address
// Peer is a manual connection - log but don't disconnect
LogPrintf ( " Disconnecting and discouraging peer %s! \n " , pnode . addr . ToString ( ) ) ;
LogPrintf ( " Warning: not punishing manually connected peer %d! \n " , peer_id ) ;
if ( m_banman ) {
return false ;
m_banman - > Discourage ( pnode . addr ) ;
}
}
connman - > DisconnectNode ( pnode . addr ) ;
if ( pnode . addr . IsLocal ( ) ) {
}
// Peer is on a local address. Disconnect this peer, but don't discourage the local address
LogPrintf ( " Warning: disconnecting but not discouraging local peer %d! \n " , peer_id ) ;
pnode . fDisconnect = true ;
return true ;
return true ;
}
}
return false ;
// Normal case: Disconnect the peer and discourage all nodes sharing the address
LogPrintf ( " Disconnecting and discouraging peer %d! \n " , peer_id ) ;
if ( m_banman ) m_banman - > Discourage ( pnode . addr ) ;
connman - > DisconnectNode ( pnode . addr ) ;
return true ;
}
}
bool PeerLogicValidation : : ProcessMessages ( CNode * pfrom , std : : atomic < bool > & interruptMsgProc )
bool PeerLogicValidation : : ProcessMessages ( CNode * pfrom , std : : atomic < bool > & interruptMsgProc )