From 225b6a2d47acb96e7269896658630c9516dfafdf Mon Sep 17 00:00:00 2001 From: Carlo Mazzaferro Date: Tue, 15 Oct 2024 15:54:23 +0200 Subject: [PATCH] u8 => u64 --- iris-mpc-upgrade/src/bin/tcp_ssl_upgrade_client.rs | 6 +++--- iris-mpc-upgrade/src/bin/tcp_upgrade_server.rs | 6 +++--- iris-mpc-upgrade/src/config.rs | 2 +- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/iris-mpc-upgrade/src/bin/tcp_ssl_upgrade_client.rs b/iris-mpc-upgrade/src/bin/tcp_ssl_upgrade_client.rs index 537a7546e..669e47ac4 100644 --- a/iris-mpc-upgrade/src/bin/tcp_ssl_upgrade_client.rs +++ b/iris-mpc-upgrade/src/bin/tcp_ssl_upgrade_client.rs @@ -92,9 +92,9 @@ async fn main() -> eyre::Result<()> { server2.write_u64(end).await?; server3.write_u64(end).await?; - server1.write_u8(args.batch_size).await?; - server2.write_u8(args.batch_size).await?; - server3.write_u8(args.batch_size).await?; + server1.write_u64(args.batch_size).await?; + server2.write_u64(args.batch_size).await?; + server3.write_u64(args.batch_size).await?; server1.flush().await?; server2.flush().await?; diff --git a/iris-mpc-upgrade/src/bin/tcp_upgrade_server.rs b/iris-mpc-upgrade/src/bin/tcp_upgrade_server.rs index 92311b868..757e67b03 100644 --- a/iris-mpc-upgrade/src/bin/tcp_upgrade_server.rs +++ b/iris-mpc-upgrade/src/bin/tcp_upgrade_server.rs @@ -108,8 +108,8 @@ async fn main() -> eyre::Result<()> { ); } - let batch_size1 = client_stream1.read_u8().await?; - let batch_size2 = client_stream2.read_u8().await?; + let batch_size1 = client_stream1.read_u64().await?; + let batch_size2 = client_stream2.read_u64().await?; if batch_size1 != batch_size2 { bail!( @@ -120,7 +120,7 @@ async fn main() -> eyre::Result<()> { } let num_elements = end1.checked_sub(start1).unwrap(); - let num_batches = num_elements / u64::from(batch_size1); + let num_batches = num_elements / batch_size1; tracing::info!("Batch size: {}, num batches: {}", batch_size1, num_batches); let mut batch = Vec::new(); diff --git a/iris-mpc-upgrade/src/config.rs b/iris-mpc-upgrade/src/config.rs index 453f7ba81..cafaf8d71 100644 --- a/iris-mpc-upgrade/src/config.rs +++ b/iris-mpc-upgrade/src/config.rs @@ -88,7 +88,7 @@ pub struct UpgradeClientConfig { pub party_id: u8, #[clap(long)] - pub batch_size: u8, + pub batch_size: u64, #[clap(long)] pub eye: Eye,