early return

EspressoSystems · Oct 9, 2024 · 38d714e · 38d714e
1 parent 5dedc9f
commit 38d714e
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/sequencer/src/catchup.rs b/sequencer/src/catchup.rs
@@ -136,13 +136,14 @@ impl<ApiVer: StaticVersionType> StateCatchup for StatePeers<ApiVer> {
                 .await
             {
                 Ok(res) => {
-                    if res.proof.account == account.into() {
-                        match res.proof.verify(&fee_merkle_tree_root) {
-                            Ok(_) => return Ok(res),
-                            Err(err) => tracing::warn!("Error verifying account proof: {}", err),
-                        }
-                    } else {
+                    if res.proof.account != account.into() {
                         tracing::warn!("Invalid proof received from peer {:?}", client.url);
+                        continue;
+                    }
+
+                    match res.proof.verify(&fee_merkle_tree_root) {
+                        Ok(_) => return Ok(res),
+                        Err(err) => tracing::warn!("Error verifying account proof: {}", err),
                     }
                 }
                 Err(err) => {