VerifyStage: headers validation using preferified hashes.

akula-bft · Aug 29, 2021 · aba9202 · aba9202
1 parent 00d54dc
commit aba9202
Show file tree

Hide file tree

Showing 6 changed files with 120,907 additions and 8 deletions.
diff --git a/src/downloader/downloader_impl.rs b/src/downloader/downloader_impl.rs
@@ -2,8 +2,9 @@ use crate::downloader::{
     chain_config::{ChainConfig, ChainsConfig},
     headers::{
         fetch_receive_stage::FetchReceiveStage, fetch_request_stage::FetchRequestStage,
-        header_slices::HeaderSlices, refill_stage::RefillStage, retry_stage::RetryStage,
-        save_stage::SaveStage, verify_stage::VerifyStage,
+        header_slices::HeaderSlices, preverified_hashes_config::PreverifiedHashesConfig,
+        refill_stage::RefillStage, retry_stage::RetryStage, save_stage::SaveStage,
+        verify_stage::VerifyStage,
     },
     opts::Opts,
     sentry_client,
@@ -79,7 +80,10 @@ impl Downloader {
 
         let retry_stage = RetryStage::new(Arc::clone(&header_slices));
 
-        let verify_stage = VerifyStage::new(Arc::clone(&header_slices));
+        let verify_stage = VerifyStage::new(
+            Arc::clone(&header_slices),
+            PreverifiedHashesConfig::new(&self.opts.chain_name)?,
+        );
 
         let save_stage = SaveStage::new(Arc::clone(&header_slices));
 

diff --git a/src/downloader/headers/mod.rs b/src/downloader/headers/mod.rs
@@ -2,6 +2,7 @@ pub mod header_slices;
 
 pub mod fetch_receive_stage;
 pub mod fetch_request_stage;
+pub mod preverified_hashes_config;
 pub mod refill_stage;
 pub mod retry_stage;
 pub mod save_stage;
@@ -14,5 +15,6 @@ pub use ui_crossterm::HeaderSlicesView;
 
 #[cfg(not(feature = "crossterm"))]
 pub mod ui_tracing;
+
 #[cfg(not(feature = "crossterm"))]
 pub use ui_tracing::HeaderSlicesView;