json mode

Author: noel2004

crates/prover-bin/src/dumper.rs

@@ -12,32 +12,39 @@ use scroll_zkvm_types::ProvingTask;
 #[derive(Default)]
 pub struct Dumper {
     #[allow(dead_code)]
-    target_path: String,
+    pub target_path: String,
+    pub json_mode: bool,
 }
 
 impl Dumper {
     fn dump(&self, input_string: &str) -> eyre::Result<()> {
         let task: ProvingTaskExt = serde_json::from_str(input_string)?;
         let task = ProvingTask::from(task);
 
-        // stream-encode serialized_witness to input_task.bin using bincode 2.0
-        let input_file = std::fs::File::create("input_task.bin")?;
-        let mut input_writer = std::io::BufWriter::new(input_file);
-        bincode::encode_into_std_write(
-            &task.serialized_witness,
-            &mut input_writer,
-            bincode::config::standard(),
-        )?;
-
-        // stream-encode aggregated_proofs to agg_proofs.bin using bincode 2.0
-        let agg_file = std::fs::File::create("agg_proofs.bin")?;
-        let mut agg_writer = std::io::BufWriter::new(agg_file);
-        for proof in &task.aggregated_proofs {
-            bincode::serde::encode_into_std_write(
-                &proof.proofs,
-                &mut agg_writer,
+        if self.json_mode {
+            let file = std::fs::File::create("input_task.json")?;
+            serde_json::to_writer(std::io::BufWriter::new(file), &task)?;
+        } else {
+            // stream-encode serialized_witness to input_task.bin using bincode 2.0
+            let input_file = std::fs::File::create("input_task.bin")?;
+            let mut input_writer = std::io::BufWriter::new(input_file);
+            bincode::encode_into_std_write(
+                &task.serialized_witness,
+                &mut input_writer,
                 bincode::config::standard(),
             )?;
+
+            // stream-encode aggregated_proofs to agg_proofs.bin using bincode 2.0
+            let agg_file = std::fs::File::create("agg_proofs.bin")?;
+            let mut agg_writer = std::io::BufWriter::new(agg_file);
+            for proof in &task.aggregated_proofs {
+                let sz = bincode::serde::encode_into_std_write(
+                    &proof.proofs,
+                    &mut agg_writer,
+                    bincode::config::standard(),
+                )?;
+                println!("written {sz} bytes for proof");
+            }
         }
 
         Ok(())

crates/prover-bin/src/main.rs

@@ -57,6 +57,8 @@ enum Commands {
         task_path: String,
     },
     Dump {
+        #[arg(long = "json", default_value = "false")]
+        json_mode: bool,
         task_type: TaskType,
         task_id: String,
     },
@@ -85,11 +87,21 @@ async fn main() -> eyre::Result<()> {
     let local_prover = LocalProver::new(cfg.clone());
 
     match args.command {
-        Some(Commands::Dump { task_type, task_id }) => {
-            let prover = ProverBuilder::new(sdk_config, dumper::Dumper::default())
-                .build()
-                .await
-                .map_err(|e| eyre::eyre!("build prover fail: {e}"))?;
+        Some(Commands::Dump {
+            json_mode,
+            task_type,
+            task_id,
+        }) => {
+            let prover = ProverBuilder::new(
+                sdk_config,
+                dumper::Dumper {
+                    json_mode,
+                    ..Default::default()
+                },
+            )
+            .build()
+            .await
+            .map_err(|e| eyre::eyre!("build prover fail: {e}"))?;
 
             std::sync::Arc::new(prover)
                 .one_shot(&[task_id], task_type.into())