HyParView (slim)

1 files changed, 7 insertions, 2 deletions

diff --git a/src/swarm/peer/inbound.rs b/src/swarm/peer/inbound.rs
index ddcc376..0c41466 100644
--- a/src/swarm/peer/inbound.rs
+++ b/src/swarm/peer/inbound.rs
@@ -256,16 +256,21 @@ pub(crate) async fn handle_message(
         }
         transport::ParsedMessage::Prune(link) => {
             state
-                .dissemination
+                .dissemination()
                 .linked(state, &peer, link.writer, false)
                 .await;
         }
         transport::ParsedMessage::Graft(link) => {
             state
-                .dissemination
+                .dissemination()
                 .linked(state, &peer, link.writer, true)
                 .await;
         }
+        message @ (transport::ParsedMessage::Neighbour(_)
+        | transport::ParsedMessage::NeighbourReply(_)
+        | transport::ParsedMessage::Disconnect(_)) => {
+            state.topology().received(state, &peer, &message).await;
+        }
         transport::ParsedMessage::Who(_)
         | transport::ParsedMessage::Me(_)
         | transport::ParsedMessage::Ping(_)