From b9c27e56832db5faf40d54aaee0b42eaea34ace4 Mon Sep 17 00:00:00 2001 From: Deirdre Connolly Date: Tue, 26 Nov 2019 02:03:45 -0500 Subject: [PATCH] Handle Response::Error and send Message::Reject generated from a PeerError::Rejected --- zebra-network/src/peer/server.rs | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/zebra-network/src/peer/server.rs b/zebra-network/src/peer/server.rs index e0d820a7b..b7dca3d6b 100644 --- a/zebra-network/src/peer/server.rs +++ b/zebra-network/src/peer/server.rs @@ -374,6 +374,11 @@ where match rsp { Response::Ok => { /* generic success, do nothing */ } + Response::Error => { + if let Err(e) = self.peer_tx.send(Message::from(PeerError::Rejected)).await { + self.fail_with(e.into()); + } + } Response::Peers(addrs) => { if let Err(e) = self.peer_tx.send(Message::Addr(addrs)).await { self.fail_with(e.into());