plumtree

1 files changed, 12 insertions, 0 deletions

diff --git a/src/swarm/peer/inbound.rs b/src/swarm/peer/inbound.rs
index 7154357..ddcc376 100644
--- a/src/swarm/peer/inbound.rs
+++ b/src/swarm/peer/inbound.rs
@@ -254,6 +254,18 @@ pub(crate) async fn handle_message(
         transport::ParsedMessage::ProbeReply(reply) => {
             crate::swarm::working_links::answered(state, &peer, reply);
         }
+        transport::ParsedMessage::Prune(link) => {
+            state
+                .dissemination
+                .linked(state, &peer, link.writer, false)
+                .await;
+        }
+        transport::ParsedMessage::Graft(link) => {
+            state
+                .dissemination
+                .linked(state, &peer, link.writer, true)
+                .await;
+        }
         transport::ParsedMessage::Who(_)
         | transport::ParsedMessage::Me(_)
         | transport::ParsedMessage::Ping(_)