move invalid id check to pre-load

worldcoin · Jan 10, 2025 · 8852eac · 8852eac
1 parent 6972e3a
commit 8852eac
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/iris-mpc/src/bin/server.rs b/iris-mpc/src/bin/server.rs
@@ -1082,6 +1082,10 @@ async fn server_main(config: Config) -> eyre::Result<()> {
                             time_waiting_for_stream += now_load_summary.elapsed();
                             now_load_summary = Instant::now();
                             let index = result.index();
+                            if index == 0 || index > store_len {
+                                tracing::error!("Invalid iris index {}", index);
+                                return Err(eyre!("Invalid iris index {}", index));
+                            }
                             match result {
                                 StoredIris::DB(iris) => {
                                     n_loaded_from_db += 1;
@@ -1128,11 +1132,6 @@ async fn server_main(config: Config) -> eyre::Result<()> {
                                 );
                             }
 
-                            if index == 0 || index > store_len {
-                                tracing::error!("Invalid iris index {}", index);
-                                return Err(eyre!("Invalid iris index {}", index));
-                            }
-
                             // if the serial id hasn't been loaded before, count is as unique record
                             if all_serial_ids.contains(&(index as i64)) {
                                 actor.increment_db_size(index - 1);