From 592b3afb9f51e600c79eb8caa1c0f8635becbee0 Mon Sep 17 00:00:00 2001
From: Youngjoon Lee <5462944+youngjoon-lee@users.noreply.github.com>
Date: Thu, 26 Sep 2024 18:27:53 +0900
Subject: [PATCH] more realistic proposer

---
 mixnet/ordering/src/iteration.rs | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/mixnet/ordering/src/iteration.rs b/mixnet/ordering/src/iteration.rs
index 04afc99..4a60386 100644
--- a/mixnet/ordering/src/iteration.rs
+++ b/mixnet/ordering/src/iteration.rs
@@ -220,11 +220,14 @@ impl Iteration {
             }
 
             // If all senders finally emitted all data+noise messages,
+            // (or if all senders finally emitted all data messages)
             // and If all data messages have been received by the receiver,
             // stop the iteration.
-            if all_sent_count == all_sent_count_target
-                && sent_data_msgs.len() == recv_data_msgs.len()
-            {
+            let sent_count_target_reached = match data_sent_count_target {
+                Some(target) => sent_data_msgs.len() == target as usize,
+                None => all_sent_count == all_sent_count_target,
+            };
+            if sent_count_target_reached && sent_data_msgs.len() == recv_data_msgs.len() {
                 break;
             }