yapper optionally read enc_key from contract

Author: alxiong

Cargo.lock

@@ -218,8 +218,16 @@ test-dyn-comm: build_release_until build-test-utils
                  -u http://localhost:8545 \
                  -k 0x2bbf15bc655c4cc157b769cfcb1ea9924b9e1a35 \
                  -c test-configs/c1/committee.toml" \
-    --run   "8:sleep 5" \
-    --spawn "9:target/release/yapper --keyset-file test-configs/c1/committee.toml" \
+    --run   "8:sleep 8" \
+    --run   "9:target/release/register \
+                 -u http://localhost:8545 \
+                 -k 0x2bbf15bc655c4cc157b769cfcb1ea9924b9e1a35 \
+                 -c test-configs/c0/committee.toml \
+                 -a threshold-enc-key" \
+    --spawn "10:target/release/yapper \
+                  --keyset-file test-configs/c1/committee.toml \
+                  --parent-url http://localhost:8545 \
+                  --key-manager-contract 0x2bbf15bc655c4cc157b769cfcb1ea9924b9e1a35" \
     target/release/run-committee -- --configs test-configs/c1/ \
       --committee 1 \
       --timeboost target/release/timeboost \

justfile

@@ -14,6 +14,7 @@ cliquenet = { path = "../cliquenet" }
 futures = { workspace = true }
 reqwest = { workspace = true }
 timeboost = { path = "../timeboost" }
+timeboost-contract = { path = "../timeboost-contract" }
 timeboost-utils = { path = "../timeboost-utils" }
 tokio = { workspace = true }
 tracing = { workspace = true }

yapper/Cargo.toml

@@ -1,6 +1,7 @@
 use bon::Builder;
 use cliquenet::Address;
 use reqwest::Url;
+use timeboost::crypto::prelude::ThresholdEncKey;
 
 #[derive(Debug, Clone, Builder)]
 pub(crate) struct YapperConfig {
@@ -12,4 +13,6 @@ pub(crate) struct YapperConfig {
     pub(crate) nitro_url: Option<Url>,
     /// Chain id for l2 chain
     pub(crate) chain_id: u64,
+    /// Use a given threshold encryption key, instead of fetching from nodes
+    pub(crate) threshold_enc_key: Option<ThresholdEncKey>,
 }

yapper/src/config.rs

@@ -6,11 +6,13 @@
 
 use std::path::PathBuf;
 
+use alloy::{primitives::Address, providers::ProviderBuilder};
 use anyhow::{Context, Result};
 
 use clap::Parser;
 use reqwest::Url;
-use timeboost::config::CommitteeConfig;
+use timeboost::{config::CommitteeConfig, crypto::prelude::ThresholdEncKey};
+use timeboost_contract::KeyManager;
 use timeboost_utils::types::logging::init_logging;
 use timeboost_utils::wait_for_live_peer;
 use tokio::signal::{
@@ -45,6 +47,14 @@ struct Cli {
     /// Nitro node url.
     #[clap(long)]
     nitro_url: Option<Url>,
+
+    /// Parent chain where KeyManager is deployed
+    #[clap(long)]
+    parent_url: Option<Url>,
+
+    /// KeyManager contract address on the parent chain
+    #[clap(long)]
+    key_manager_contract: Option<Address>,
 }
 
 #[tokio::main]
@@ -66,10 +76,25 @@ async fn main() -> Result<()> {
         addresses.push(node.http_api);
     }
 
+    let threshold_enc_key = if let Some(rpc) = cli.parent_url {
+        let km_addr = cli
+            .key_manager_contract
+            .expect("provide both parent chain and key manager contract");
+        let provider = ProviderBuilder::new().connect_http(rpc);
+        let contract = KeyManager::new(km_addr, provider);
+
+        Some(ThresholdEncKey::from_bytes(
+            &contract.thresholdEncryptionKey().call().await?.0,
+        )?)
+    } else {
+        None
+    };
+
     let config = YapperConfig::builder()
         .addresses(addresses)
         .tps(cli.tps)
         .maybe_nitro_url(cli.nitro_url)
+        .maybe_threshold_enc_key(threshold_enc_key)
         .chain_id(cli.chain_id)
         .build();
     let yapper = Yapper::new(config).await?;

yapper/src/main.rs

@@ -10,7 +10,7 @@ use alloy::{
 use anyhow::{Context, Result};
 use futures::future::join_all;
 use reqwest::{Client, Url};
-use timeboost::types::BundleVariant;
+use timeboost::{crypto::prelude::ThresholdEncKey, types::BundleVariant};
 use timeboost_utils::enc_key::ThresholdEncKeyCellAccumulator;
 use timeboost_utils::load_generation::{TxInfo, make_bundle, make_dev_acct_bundle, tps_to_millis};
 use tokio::time::interval;
@@ -42,6 +42,7 @@ pub(crate) struct Yapper {
     interval: Duration,
     chain_id: u64,
     provider: Option<RootProvider>,
+    enc_key: Option<ThresholdEncKey>,
 }
 
 impl Yapper {
@@ -68,12 +69,14 @@ impl Yapper {
         } else {
             None
         };
+
         Ok(Self {
             urls,
             interval: Duration::from_millis(tps_to_millis(cfg.tps)),
             client,
             provider,
             chain_id: cfg.chain_id,
+            enc_key: cfg.threshold_enc_key,
         })
     }
 
@@ -93,25 +96,31 @@ impl Yapper {
                     warn!("failed to prepare txn");
                     continue;
                 };
-                let enc_key = match acc.enc_key().await {
+                let enc_key = match self.enc_key.as_ref() {
                     Some(key) => key,
-                    None => {
-                        warn!("encryption key not available yet");
-                        continue;
-                    }
+                    None => match acc.enc_key().await {
+                        Some(key) => key,
+                        None => {
+                            warn!("encryption key not available yet");
+                            continue;
+                        }
+                    },
                 };
                 let Ok(b) = make_dev_acct_bundle(enc_key, txn) else {
                     warn!("failed to generate dev account bundle");
                     continue;
                 };
                 b
             } else {
-                let enc_key = match acc.enc_key().await {
+                let enc_key = match self.enc_key.as_ref() {
                     Some(key) => key,
-                    None => {
-                        warn!("encryption key not available yet");
-                        continue;
-                    }
+                    None => match acc.enc_key().await {
+                        Some(key) => key,
+                        None => {
+                            warn!("encryption key not available yet");
+                            continue;
+                        }
+                    },
                 };
                 let Ok(b) = make_bundle(enc_key) else {
                     warn!("failed to generate bundle");