cerc-io/lighthouse
16
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

Improve logging around peer scoring (#5325)

Browse Source 
* Improve logging around score states

* Improve logs also
This commit is contained in:
Age Manning 2024-03-05 06:39:23 +11:00 committed by GitHub
parent ee6f667702
commit f919f82e4f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
beacon_node/lighthouse_network/src/peer_manager/peerdb.rs
View File

@ -168,7 +168,7 @@ impl<TSpec: EthSpec> PeerDB<TSpec> {
fn score_state_banned_or_disconnected(&self, peer_id: &PeerId) -> bool {
if let Some(peer) = self.peers.get(peer_id) {
match peer.score_state() {
ScoreState::Banned | ScoreState::Disconnected => true,
ScoreState::Banned | ScoreState::ForcedDisconnect => true,
_ => self.ip_is_banned(peer).is_some(),
}
} else {
@ -1062,12 +1062,12 @@ impl<TSpec: EthSpec> PeerDB<TSpec> {
log: &slog::Logger,
) -> ScoreTransitionResult {
match (info.score_state(), previous_state) {
(ScoreState::Banned, ScoreState::Healthy | ScoreState::Disconnected) => {
(ScoreState::Banned, ScoreState::Healthy | ScoreState::ForcedDisconnect) => {
debug!(log, "Peer has been banned"; "peer_id" => %peer_id, "score" => %info.score());
ScoreTransitionResult::Banned
}
(ScoreState::Disconnected, ScoreState::Banned | ScoreState::Healthy) => {
debug!(log, "Peer transitioned to disconnect state"; "peer_id" => %peer_id, "score" => %info.score(), "past_state" => %previous_state);
(ScoreState::ForcedDisconnect, ScoreState::Banned | ScoreState::Healthy) => {
debug!(log, "Peer transitioned to forced disconnect score state"; "peer_id" => %peer_id, "score" => %info.score(), "past_score_state" => %previous_state);
// disconnect the peer if it's currently connected or dialing
if info.is_connected_or_dialing() {
ScoreTransitionResult::Disconnected
@ -1079,18 +1079,20 @@ impl<TSpec: EthSpec> PeerDB<TSpec> {
ScoreTransitionResult::NoAction
}
}
(ScoreState::Healthy, ScoreState::Disconnected) => {
debug!(log, "Peer transitioned to healthy state"; "peer_id" => %peer_id, "score" => %info.score(), "past_state" => %previous_state);
(ScoreState::Healthy, ScoreState::ForcedDisconnect) => {
debug!(log, "Peer transitioned to healthy score state"; "peer_id" => %peer_id, "score" => %info.score(), "past_score_state" => %previous_state);
ScoreTransitionResult::NoAction
}
(ScoreState::Healthy, ScoreState::Banned) => {
debug!(log, "Peer transitioned to healthy state"; "peer_id" => %peer_id, "score" => %info.score(), "past_state" => %previous_state);
debug!(log, "Peer transitioned to healthy score state"; "peer_id" => %peer_id, "score" => %info.score(), "past_score_state" => %previous_state);
// unban the peer if it was previously banned.
ScoreTransitionResult::Unbanned
}
// Explicitly ignore states that haven't transitioned.
(ScoreState::Healthy, ScoreState::Healthy) => ScoreTransitionResult::NoAction,
(ScoreState::Disconnected, ScoreState::Disconnected) => ScoreTransitionResult::NoAction,
(ScoreState::ForcedDisconnect, ScoreState::ForcedDisconnect) => {
ScoreTransitionResult::NoAction
}
(ScoreState::Banned, ScoreState::Banned) => ScoreTransitionResult::NoAction,
}

8
beacon_node/lighthouse_network/src/peer_manager/peerdb/score.rs
View File

@ -104,7 +104,7 @@ pub(crate) enum ScoreState {
/// We are content with the peers performance. We permit connections and messages.
Healthy,
/// The peer should be disconnected. We allow re-connections if the peer is persistent.
Disconnected,
ForcedDisconnect,
/// The peer is banned. We disallow new connections until it's score has decayed into a
/// tolerable threshold.
Banned,
@ -115,7 +115,7 @@ impl std::fmt::Display for ScoreState {
match self {
ScoreState::Healthy => write!(f, "Healthy"),
ScoreState::Banned => write!(f, "Banned"),
ScoreState::Disconnected => write!(f, "Disconnected"),
ScoreState::ForcedDisconnect => write!(f, "Disconnected"),
}
}
}
@ -313,7 +313,7 @@ impl Score {
pub(crate) fn state(&self) -> ScoreState {
match self.score() {
x if x <= MIN_SCORE_BEFORE_BAN => ScoreState::Banned,
x if x <= MIN_SCORE_BEFORE_DISCONNECT => ScoreState::Disconnected,
x if x <= MIN_SCORE_BEFORE_DISCONNECT => ScoreState::ForcedDisconnect,
_ => ScoreState::Healthy,
}
}
@ -407,7 +407,7 @@ mod tests {
assert!(score.score() < 0.0);
assert_eq!(score.state(), ScoreState::Healthy);
score.test_add(-1.0001);
assert_eq!(score.state(), ScoreState::Disconnected);
assert_eq!(score.state(), ScoreState::ForcedDisconnect);
}
#[test]