Sharded decrypt and uniqueness verification (#1358)

* refactor UniqueBytesValidator to UniqueTagValidator with fixed width array * add implementation of Serializable for UniqueTag * tmp working commit * working commit, all tests passing * remove type aliases because clippy doesn't like them * Update ipa-core/src/query/runner/hybrid.rs Co-authored-by: Alex Koshelev <[email protected]> * fix formatting error * add a ReshardByTag step and narrow to it * do modulo on u128, update compile check for TAG_SIZE * update type signature and name of flatten3v * update test to generate_random_tag directly * reorganize ShardIndex implementations and remove conflicting implementation --------- Co-authored-by: Alex Koshelev <[email protected]>
private-attribution · Oct 18, 2024 · 0c276b9 · 0c276b9
1 parent 4307265
commit 0c276b9
Show file tree

Hide file tree

Showing 10 changed files with 419 additions and 223 deletions.
diff --git a/ipa-core/Cargo.toml b/ipa-core/Cargo.toml
@@ -82,6 +82,7 @@ ipa-step = { version = "*", path = "../ipa-step" }
 ipa-step-derive = { version = "*", path = "../ipa-step-derive" }
 
 aes = "0.8.3"
+assertions = "0.1.0"
 async-trait = "0.1.79"
 async-scoped = { version = "0.9.0", features = ["use-tokio"], optional = true }
 axum = { version = "0.7.5", optional = true, features = ["http2", "macros"] }

diff --git a/ipa-core/src/protocol/hybrid/step.rs b/ipa-core/src/protocol/hybrid/step.rs
@@ -1,4 +1,6 @@
 use ipa_step_derive::CompactStep;
 
 #[derive(CompactStep)]
-pub(crate) enum HybridStep {}
+pub(crate) enum HybridStep {
+    ReshardByTag,
+}
diff --git a/ipa-core/src/query/executor.rs b/ipa-core/src/query/executor.rs
@@ -39,7 +39,7 @@ use crate::{
         Gate,
     },
     query::{
-        runner::{HybridQuery, OprfIpaQuery, QueryResult},
+        runner::{OprfIpaQuery, QueryResult},
         state::RunningQuery,
     },
     sync::Arc,
@@ -164,20 +164,7 @@ pub fn execute<R: PrivateKeyRegistry>(
                 )
             },
         ),
-        (QueryType::SemiHonestHybrid(query_params), _) => do_query(
-            runtime,
-            config,
-            gateway,
-            input,
-            move |prss, gateway, config, input| {
-                let ctx = SemiHonestContext::new(prss, gateway);
-                Box::pin(
-                    HybridQuery::<_, BA32, R>::new(query_params, key_registry)
-                        .execute(ctx, config.size, input)
-                        .then(|res| ready(res.map(|out| Box::new(out) as Box<dyn Result>))),
-                )
-            },
-        ),
+        (QueryType::SemiHonestHybrid(_), _) => todo!(),
     }
 }