script to send proofs with random addresses (#1865)

Co-authored-by: Mauro Toscano <[email protected]>

Author: Mechanix97

Makefile

@@ -572,6 +572,11 @@ batcher_send_burst_groth16: batcher/target/release/aligned
 	@mkdir -p scripts/test_files/gnark_groth16_bn254_infinite_script/infinite_proofs
 	@./batcher/aligned/send_burst_tasks.sh $(BURST_SIZE) $(START_COUNTER)
 
+batcher_send_proof_with_random_address:
+	@cd batcher/aligned/ && ./send_proof_with_random_address.sh
+
+batcher_send_burst_with_random_address:
+	@cd batcher/aligned/ && ./send_burst_with_random_address.sh
 
 __TASK_SENDER__:
 BURST_TIME_SECS ?= 3

batcher/aligned/send_burst_with_random_address.sh

@@ -0,0 +1,21 @@
+#!/bin/bash
+
+# This script send periodically a burst of proof with a specific amount of repetitions
+# the proofs always have a random address
+# Params:
+# PROOF_TYPE = sp1|groth16|plonk|risc0  (default sp1)
+# RPC_URL (default localhost:8545)
+# NETWORK   devnet|holesky-stage|holesky
+# REPETITIONS (default 1)
+# BURST_DELAY in secs (default 30)
+
+if [ -z $BURST_DELAY ]; then
+    echo "Using default burst delay 30"
+    BURST_DELAY=30
+fi
+
+while true 
+do
+    . ./send_proof_with_random_address.sh &
+    sleep $BURST_DELAY
+done

batcher/aligned/send_proof_with_random_address.sh

@@ -0,0 +1,78 @@
+#!/bin/bash
+
+# Params:
+# PROOF_TYPE = sp1|groth16|plonk|risc0  (default sp1)
+# RPC_URL (default localhost:8545)
+# NETWORK   devnet|holesky-stage|holesky
+# REPETITIONS (default 1)
+
+if [ -z "$NETWORK" ]; then
+    echo "NETWORK is not set. Setting it to devnet"
+    NETWORK="devnet"
+fi
+
+if [ -z "$RPC_URL" ]; then
+    echo "RPC_URL is not set. Setting it to localhost:8545"
+    RPC_URL="http://localhost:8545"
+fi
+
+if [ -z $PROOF_TYPE ]; then
+    echo "Proof type not provided, using SP1 default"
+    PROOF_TYPE="sp1" #sp1|groth16|plonk|risc0
+fi
+
+if [ -z $REPETITIONS ]; then
+    echo "REPETITIONS not provided, using 1 as default"
+    REPETITIONS=1
+fi
+
+echo "Sending $REPETITIONS $PROOF_TYPE proof/s to the batcher"
+echo "Batcher in $NETWORK and endpoint at $RPC_URL"
+
+if [[ $PROOF_TYPE == "sp1" ]]; then
+    aligned submit \
+		--proving_system SP1 \
+		--proof ../../scripts/test_files/sp1/sp1_fibonacci_4_1_3.proof \
+		--vm_program ../../scripts/test_files/sp1/sp1_fibonacci_4_1_3.elf \
+        --random_address \
+        --repetitions $REPETITIONS \
+		--rpc_url $RPC_URL \
+		--network $NETWORK
+
+elif [[ $PROOF_TYPE == "groth16" ]]; then
+    aligned submit \
+		--proving_system Groth16Bn254 \
+		--proof ../../scripts/test_files/gnark_groth16_bn254_script/groth16.proof \
+		--public_input ../../scripts/test_files/gnark_groth16_bn254_script/groth16.pub \
+		--vk ../../scripts/test_files/gnark_groth16_bn254_script/groth16.vk \
+        --random_address \
+        --repetitions $REPETITIONS \
+		--rpc_url $RPC_URL \
+		--network $NETWORK
+
+elif [[ $PROOF_TYPE == "plonk" ]]; then
+    aligned submit \
+		--proving_system GnarkPlonkBn254 \
+		--proof ../../scripts/test_files/gnark_plonk_bn254_script/plonk.proof \
+		--public_input ../../scripts/test_files/gnark_plonk_bn254_script/plonk_pub_input.pub \
+		--vk ../../scripts/test_files/gnark_plonk_bn254_script/plonk.vk \
+		--random_address \
+        --repetitions $REPETITIONS \
+		--rpc_url $RPC_URL \
+		--network $NETWORK
+
+elif [[ $PROOF_TYPE == "risc0" ]]; then
+	aligned submit \
+		--proving_system Risc0 \
+		--proof ../../scripts/test_files/risc_zero/fibonacci_proof_generator/risc_zero_fibonacci_2_0.proof \
+        --vm_program ../../scripts/test_files/risc_zero/fibonacci_proof_generator/fibonacci_id_2_0.bin \
+        --public_input ../../scripts/test_files/risc_zero/fibonacci_proof_generator/risc_zero_fibonacci_2_0.pub \
+		--random_address \
+        --repetitions $REPETITIONS \
+		--rpc_url $RPC_URL \
+		--network $NETWORK
+
+else
+    echo "Incorrect proof type provided $1"
+    exit 1
+fi