error refactoring

beacon_node/network/src/sync/block_lookups/mod.rs

@@ -26,7 +26,6 @@ use super::{
 };
 use crate::beacon_processor::{ChainSegmentProcessId, WorkEvent};
 use crate::metrics;
-use crate::sync::block_lookups::single_block_lookup::LookupVerifyError;
 
 mod parent_lookup;
 mod single_block_lookup;
@@ -315,9 +314,9 @@ impl<T: BeaconChainTypes> BlockLookups<T> {
         let should_remove = match request_ref.verify_block(block) {
             Ok(Some((root, block))) => {
                 if triggered_parent_request {
-                    // The lookup status here is irrelevant because we wait until the parent chain
-                    // is complete before processing the block.
-                    if let Err(e) = request_ref.add_block(root, block) {
+                    if let LookupDownloadStatus::AvailabilityCheck(e) =
+                        request_ref.add_block(root, block)
+                    {
                         handle_block_lookup_verify_error(
                             request_id_ref,
                             request_ref,
@@ -387,9 +386,9 @@ impl<T: BeaconChainTypes> BlockLookups<T> {
         let should_remove = match request_ref.verify_blob(blob) {
             Ok(Some((block_root, blobs))) => {
                 if triggered_parent_request {
-                    // The lookup status here is irrelevant because we wait until the parent chain
-                    // is complete before processing the block.
-                    if let Err(e) = request_ref.add_blobs(block_root, blobs) {
+                    if let LookupDownloadStatus::AvailabilityCheck(e) =
+                        request_ref.add_blobs(block_root, blobs)
+                    {
                         handle_block_lookup_verify_error(
                             request_id_ref,
                             request_ref,
@@ -514,9 +513,9 @@ impl<T: BeaconChainTypes> BlockLookups<T> {
 
         match parent_lookup.verify_block(block, &mut self.failed_chains) {
             Ok(Some((block_root, block))) => {
-                let process_or_search = parent_lookup.add_block(block_root, block); //TODO(sean) fix
+                let process_or_search = parent_lookup.add_block(block_root, block);
                 match process_or_search {
-                    Ok(LookupDownloadStatus::Process(wrapper)) => {
+                    LookupDownloadStatus::Process(wrapper) => {
                         let chain_hash = parent_lookup.chain_hash();
                         if self
                             .send_block_for_processing(
@@ -531,7 +530,7 @@ impl<T: BeaconChainTypes> BlockLookups<T> {
                             self.parent_lookups.push(parent_lookup)
                         }
                     }
-                    Ok(LookupDownloadStatus::SearchBlock(block_root)) => {
+                    LookupDownloadStatus::SearchBlock(block_root) => {
                         if let Some(peer_source) =
                             parent_lookup.peer_source(ResponseType::Block, peer_id)
                         {
@@ -541,7 +540,7 @@ impl<T: BeaconChainTypes> BlockLookups<T> {
                             warn!(self.log, "Response from untracked peer"; "peer_id" => %peer_id, "block_root" => ?block_root);
                         }
                     }
-                    Err(e) => {}
+                    LookupDownloadStatus::AvailabilityCheck(e) => {}
                 }
             }
             Ok(None) => {
@@ -555,9 +554,7 @@ impl<T: BeaconChainTypes> BlockLookups<T> {
                 | ParentVerifyError::ExtraBlocksReturned
                 | ParentVerifyError::UnrequestedBlobId
                 | ParentVerifyError::ExtraBlobsReturned
-                | ParentVerifyError::InvalidIndex(_)
-                //TODO(sean) treat this differntly?
-                | ParentVerifyError::AvailabilityCheck(_) => {
+                | ParentVerifyError::InvalidIndex(_) => {
                     let e = e.into();
                     warn!(self.log, "Peer sent invalid response to parent request.";
                         "peer_id" => %peer_id, "reason" => %e);
@@ -616,7 +613,7 @@ impl<T: BeaconChainTypes> BlockLookups<T> {
 
         match parent_lookup.verify_blob(blob, &mut self.failed_chains) {
             Ok(Some((block_root, blobs))) => {
-                let processed_or_search = parent_lookup.add_blobs(block_root, blobs).unwrap(); //TODO(sean) fix
+                let processed_or_search = parent_lookup.add_blobs(block_root, blobs);
                 match processed_or_search {
                     LookupDownloadStatus::Process(wrapper) => {
                         let chain_hash = parent_lookup.chain_hash();
@@ -643,6 +640,7 @@ impl<T: BeaconChainTypes> BlockLookups<T> {
                             warn!(self.log, "Response from untracked peer"; "peer_id" => %peer_id, "block_root" => ?block_root);
                         }
                     }
+                    LookupDownloadStatus::AvailabilityCheck(e) => {}
                 }
             }
             Ok(None) => {
@@ -656,9 +654,7 @@ impl<T: BeaconChainTypes> BlockLookups<T> {
                 | ParentVerifyError::ExtraBlocksReturned
                 | ParentVerifyError::UnrequestedBlobId
                 | ParentVerifyError::ExtraBlobsReturned
-                | ParentVerifyError::InvalidIndex(_)
-                //TODO(sean) treat differently?
-                | ParentVerifyError::AvailabilityCheck(_) => {
+                | ParentVerifyError::InvalidIndex(_) => {
                     let e = e.into();
                     warn!(self.log, "Peer sent invalid response to parent request.";
                         "peer_id" => %peer_id, "reason" => %e);
@@ -1328,16 +1324,16 @@ impl<T: BeaconChainTypes> BlockLookups<T> {
     }
 }
 
-fn handle_block_lookup_verify_error<T: BeaconChainTypes>(
+fn handle_block_lookup_verify_error<T: BeaconChainTypes, Err: Into<&'static str>>(
     request_id_ref: &mut u32,
     request_ref: &mut SingleBlockLookup<SINGLE_BLOCK_LOOKUP_MAX_ATTEMPTS, T>,
     response_type: ResponseType,
     peer_id: PeerId,
-    error: LookupVerifyError,
+    e: Err,
     cx: &mut SyncNetworkContext<T>,
     log: &Logger,
 ) -> ShouldRemoveLookup {
-    let msg: &str = error.into();
+    let msg = e.into();
     cx.report_peer(peer_id, PeerAction::LowToleranceError, msg);
 
     debug!(log, "Single block lookup failed";

beacon_node/network/src/sync/block_lookups/parent_lookup.rs

@@ -7,7 +7,7 @@ use crate::sync::{
 };
 use beacon_chain::blob_verification::AsBlock;
 use beacon_chain::blob_verification::BlockWrapper;
-use beacon_chain::data_availability_checker::DataAvailabilityChecker;
+use beacon_chain::data_availability_checker::{AvailabilityCheckError, DataAvailabilityChecker};
 use beacon_chain::BeaconChainTypes;
 use lighthouse_network::PeerId;
 use std::sync::Arc;
@@ -45,7 +45,6 @@ pub enum ParentVerifyError {
     ExtraBlobsReturned,
     InvalidIndex(u64),
     PreviousFailure { parent_root: Hash256 },
-    AvailabilityCheck(String),
 }
 
 #[derive(Debug, PartialEq, Eq)]
@@ -64,6 +63,19 @@ pub enum RequestError {
 pub enum LookupDownloadStatus<T: EthSpec> {
     Process(BlockWrapper<T>),
     SearchBlock(Hash256),
+    AvailabilityCheck(AvailabilityCheckError),
+}
+
+impl<T: EthSpec> From<Result<BlockWrapper<T>, AvailabilityCheckError>> for LookupDownloadStatus<T> {
+    fn from(value: Result<BlockWrapper<T>, AvailabilityCheckError>) -> Self {
+        match value {
+            Ok(wrapper) => LookupDownloadStatus::Process(wrapper),
+            Err(AvailabilityCheckError::MissingBlobs(block_root)) => {
+                LookupDownloadStatus::SearchBlock(block_root)
+            }
+            Err(e) => LookupDownloadStatus::AvailabilityCheck(e),
+        }
+    }
 }
 
 impl<T: BeaconChainTypes> ParentLookup<T> {
@@ -172,22 +184,18 @@ impl<T: BeaconChainTypes> ParentLookup<T> {
         &mut self,
         block_root: Hash256,
         block: Arc<SignedBeaconBlock<T::EthSpec>>,
-    ) -> Result<LookupDownloadStatus<T::EthSpec>, ParentVerifyError> {
+    ) -> LookupDownloadStatus<T::EthSpec> {
         self.current_parent_request_id = None;
-        self.current_parent_request
-            .add_block(block_root, block)
-            .map_err(Into::into)
+        self.current_parent_request.add_block(block_root, block)
     }
 
     pub fn add_blobs(
         &mut self,
         block_root: Hash256,
         blobs: FixedBlobSidecarList<T::EthSpec>,
-    ) -> Result<LookupDownloadStatus<T::EthSpec>, ParentVerifyError> {
+    ) -> LookupDownloadStatus<T::EthSpec> {
         self.current_parent_blob_request_id = None;
-        self.current_parent_request
-            .add_blobs(block_root, blobs)
-            .map_err(Into::into)
+        self.current_parent_request.add_blobs(block_root, blobs)
     }
 
     pub fn pending_block_response(&self, req_id: Id) -> bool {
@@ -361,7 +369,6 @@ impl From<LookupVerifyError> for ParentVerifyError {
             E::UnrequestedBlobId => ParentVerifyError::UnrequestedBlobId,
             E::ExtraBlobsReturned => ParentVerifyError::ExtraBlobsReturned,
             E::InvalidIndex(index) => ParentVerifyError::InvalidIndex(index),
-            E::AvailabilityCheck(e) => ParentVerifyError::AvailabilityCheck(e),
         }
     }
 }

beacon_node/network/src/sync/block_lookups/single_block_lookup.rs

@@ -109,21 +109,15 @@ impl<const MAX_ATTEMPTS: u8, T: BeaconChainTypes> SingleBlockLookup<MAX_ATTEMPTS
         let block_root = blob.block_root;
 
         if let Some(blob_opt) = self.downloaded_blobs.get_mut(blob.index as usize) {
-            //TODO(sean) should we log a warn if there is already a downloaded blob?
             *blob_opt = Some(blob.clone());
 
             if let Some(block) = self.downloaded_block.as_ref() {
-                match self.da_checker.wrap_block(
+                let result = self.da_checker.wrap_block(
                     block_root,
                     block.clone(),
                     self.downloaded_blobs.clone(),
-                ) {
-                    Ok(wrapper) => Ok(LookupDownloadStatus::Process(wrapper)),
-                    Err(AvailabilityCheckError::MissingBlobs) => {
-                        Ok(LookupDownloadStatus::SearchBlock(block_root))
-                    }
-                    Err(e) => Err(LookupVerifyError::AvailabilityCheck(format!("{e:?}"))),
-                }
+                );
+                Ok(LookupDownloadStatus::from(result))
             } else {
                 Ok(LookupDownloadStatus::SearchBlock(block_root))
             }
@@ -136,24 +130,19 @@ impl<const MAX_ATTEMPTS: u8, T: BeaconChainTypes> SingleBlockLookup<MAX_ATTEMPTS
         &mut self,
         block_root: Hash256,
         blobs: FixedBlobSidecarList<T::EthSpec>,
-    ) -> Result<LookupDownloadStatus<T::EthSpec>, LookupVerifyError> {
+    ) -> LookupDownloadStatus<T::EthSpec> {
         for (index, blob_opt) in self.downloaded_blobs.iter_mut().enumerate() {
             if let Some(Some(downloaded_blob)) = blobs.get(index) {
-                //TODO(sean) should we log a warn if there is already a downloaded blob?
                 *blob_opt = Some(downloaded_blob.clone());
             }
         }
 
         if let Some(block) = self.downloaded_block.as_ref() {
-            match self.da_checker.wrap_block(block_root, block.clone(), blobs) {
-                Ok(wrapper) => Ok(LookupDownloadStatus::Process(wrapper)),
-                Err(AvailabilityCheckError::MissingBlobs) => {
-                    Ok(LookupDownloadStatus::SearchBlock(block_root))
-                }
-                Err(e) => Err(LookupVerifyError::AvailabilityCheck(format!("{e:?}"))),
-            }
+            self.da_checker
+                .wrap_block(block_root, block.clone(), blobs)
+                .into()
         } else {
-            Ok(LookupDownloadStatus::SearchBlock(block_root))
+            LookupDownloadStatus::SearchBlock(block_root)
         }
     }
 
@@ -161,31 +150,22 @@ impl<const MAX_ATTEMPTS: u8, T: BeaconChainTypes> SingleBlockLookup<MAX_ATTEMPTS
         &mut self,
         block_root: Hash256,
         block: Arc<SignedBeaconBlock<T::EthSpec>>,
-    ) -> Result<LookupDownloadStatus<T::EthSpec>, LookupVerifyError> {
-        //TODO(sean) check for existing block?
+    ) -> LookupDownloadStatus<T::EthSpec> {
         self.downloaded_block = Some(block.clone());
 
-        match self
-            .da_checker
+        self.da_checker
             .wrap_block(block_root, block, self.downloaded_blobs.clone())
-        {
-            Ok(wrapper) => Ok(LookupDownloadStatus::Process(wrapper)),
-            Err(AvailabilityCheckError::MissingBlobs) => {
-                Ok(LookupDownloadStatus::SearchBlock(block_root))
-            }
-            Err(e) => Err(LookupVerifyError::AvailabilityCheck(format!("{e:?}"))),
-        }
+            .into()
     }
 
     pub fn add_block_wrapper(
         &mut self,
         block_root: Hash256,
         block: BlockWrapper<T::EthSpec>,
-    ) -> Result<LookupDownloadStatus<T::EthSpec>, LookupVerifyError> {
+    ) -> LookupDownloadStatus<T::EthSpec> {
         match block {
             BlockWrapper::Block(block) => self.add_block(block_root, block),
             BlockWrapper::BlockAndBlobs(block, blobs) => {
-                //TODO(sean) check for existing block?
                 self.downloaded_block = Some(block);
                 self.add_blobs(block_root, blobs)
             }