Ask for headers only once (mit-dci#40)

In some cases, we might ask for headers twice, downloading duplicated headers during ibd. This commit fixes it
Davidson-Souza · May 3, 2023 · 2f6f7e5 · 2f6f7e5
1 parent f518a30
commit 2f6f7e5
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/src/blockchain/p2p_blockchain/node.rs b/src/blockchain/p2p_blockchain/node.rs
@@ -445,16 +445,20 @@ impl UtreexoNode {
 
                     if version.services.has(ServiceFlags::NODE_UTREEXO) {
                         if let NodeState::WaitingPeer = self.state {
-                            try_and_log!(
-                                self.send_to_random_peer(
-                                    NodeRequest::GetHeaders(
-                                        self.chain.get_block_locator().expect("Can get locators"),
-                                    ),
-                                    ServiceFlags::NONE
-                                )
-                                .await
-                            );
-                            self.state = NodeState::DownloadHeaders;
+                            if !self.inflight.contains_key(&InflightRequests::Headers) {
+                                try_and_log!(
+                                    self.send_to_random_peer(
+                                        NodeRequest::GetHeaders(
+                                            self.chain
+                                                .get_block_locator()
+                                                .expect("Can get locators"),
+                                        ),
+                                        ServiceFlags::NONE
+                                    )
+                                    .await
+                                );
+                                self.state = NodeState::DownloadHeaders;
+                            }
                         }
                     }
                 }