Addition of gates, and poseidon bn254

.gitignore

@@ -1,3 +1,5 @@
 .DS_Store
 main
-data
+data
+data_write/**/*.sol
+data_write/**/*.bin

README.md

@@ -2,18 +2,18 @@
 Wraps up plonky2 verifier as a groth16 circuit
 
 # TODOS
-- [ ] Implement constraints for rest of the plonky2 gates
-    - [ ] ArithmeticExtensionGate
-    - [ ] BaseSumGate
-    - [ ] CosetInterpolationGate
-    - [ ] ExponentiationGate
-    - [ ] LookupGate
-    - [ ] LookupTableGate
-    - [ ] MulExtensionGate
-    - [ ] NoopGate
-    - [ ] PoseidonMdsGate
-    - [ ] RandomAccessGate
-    - [ ] ReducingGate
-    - [ ] ReducingExtensionGate
+- [x] Implement constraints for rest of the plonky2 gates
+    - [x] ArithmeticExtensionGate
+    - [x] BaseSumGate
+    - [x] CosetInterpolationGate
+    - [x] ExponentiationGate
+    - [x] LookupGate
+    - [x] LookupTableGate
+    - [x] MulExtensionGate
+    - [x] NoopGate
+    - [x] PoseidonMdsGate
+    - [x] RandomAccessGate
+    - [x] ReducingGate
+    - [x] ReducingExtensionGate
 - [ ] Implement constraints for lookups in vanishing polynomial evaluation
-- [ ] Use poseidon over BN254 scalar field rather than goldilocks field; it will reduce constraints vastly. Also implement the corresponding config for plonky2 prover
+- [x] Use poseidon over BN254 scalar field rather than goldilocks field; it will reduce constraints vastly. Also implement the corresponding config for plonky2 prover

build_plonk.sh

@@ -0,0 +1,30 @@
+go build main.go
+go install
+# plonky2-groth16-verifier buildPlonk --common_data ./data/tendermint/common_data_struct.json
+# go run -tags=debug main.go buildPlonk --common_data ./data/tendermint/common_data_struct.json
+
+# plonky2-groth16-verifier provePlonk --plonky2_proof_path ./data/tendermint/proof_with_pis_struct.json --verifier_only_path ./data/tendermint/verifier_only_struct.json --plonky2_public_inputs_path ./data/tendermint/plonky2_pub_inputs_struct.json --gnark_public_inputs_path ./data/tendermint/gnark_pub_inputs_struct.json --proving_key_path ./data/pk.bin --r1cs_path ./data/r1cs.bin
+
+# plonky2-groth16-verifier verifyPlonk --plonkProofPath ./data/proofP --vkey_path ./data/vk.bin --gnark_public_inputs_path ./data/tendermint/gnark_pub_inputs_struct.json
+
+# plonky2-groth16-verifier exportSolPlonk --vkey_path ./data/vk.bin
+
+# osmosis [without recursion]
+# *** results [before adding sha]***
+# [plonk backend]
+# number of constraints: 20669098
+# proving time: 156349.328648
+# gas usage: 307932
+
+# osmosis [without recursion]
+# *** results [with sha]***
+# number of constraints: 26690559
+# proving time: 167878.632074
+# gas usage:
+
+
+# aggregated [osmosis]
+# [plonk backend]
+# number of constraints: 18294271
+# proving time: 145921.137651
+# gas usage: 307932

build_test.sh

@@ -6,3 +6,5 @@ plonky2-groth16-verifier build --common_data ./data/goldilocks/common_data.json
 plonky2-groth16-verifier prove --plonky2_proof_path ./data/goldilocks/proof_with_pis.json --verifier_only_path ./data/goldilocks/verifier_only.json --public_inputs_path ./data/goldilocks/pub_inputs.json --proving_key_path ./data/pk.bin --r1cs_path ./data/r1cs.bin --vk_path ./data/vk.bin
 
 plonky2-groth16-verifier verify --groth16_proof_path ./data/g16p --vkey_path ./data/vk.bin --pub_inputs_path ./data/goldilocks/pub_inputs.json
+
+plonky2-groth16-verifier exportSol --vkey_path ./data/vk.bin

cmd/build.go

@@ -5,37 +5,42 @@ package cmd
 
 import (
 	"fmt"
+	"math"
+	"math/big"
 	"os"
 
 	"github.com/Electron-Labs/plonky2-groth16-verifier/verifier"
 	"github.com/consensys/gnark-crypto/ecc"
+	kzg_bn254 "github.com/consensys/gnark-crypto/ecc/bn254/kzg"
 	"github.com/consensys/gnark/backend/groth16"
+	"github.com/consensys/gnark/backend/plonk"
 	"github.com/consensys/gnark/frontend"
 	"github.com/consensys/gnark/frontend/cs/r1cs"
+	"github.com/consensys/gnark/frontend/cs/scs"
 	"github.com/spf13/cobra"
 )
 
 var common_data_path string
 
-// buildCmd represents the build command
-var buildCmd = &cobra.Command{
-	Use:   "build",
+// buildGroth16Cmd represents the buildGroth16
+var buildGroth16Cmd = &cobra.Command{
+	Use:   "buildGroth16",
 	Short: "Build gnark groth16 circuit",
 	Long:  `Builds gnark groth16 circuit corresponding to provided common_data and plonky2 config.`,
 	Run: func(cmd *cobra.Command, args []string) {
 		fmt.Printf("build called:\n common data: %s\n ", common_data_path)
 
-		common_data, err := read_common_data_from_file(common_data_path)
+		common_data, err := ReadCommonDataFromFile(common_data_path)
 		if err != nil {
 			fmt.Println("Failed to read common data file:", err)
 			os.Exit(1)
 		}
-		circuitConstraints := getCircuitConstants(common_data)
+		circuitConstants := GetCircuitConstants(common_data)
 
 		var myCircuit verifier.Runner
 
 		// Arrays are resized according to circuitConstants before compiling
-		myCircuit.Make(circuitConstraints, common_data)
+		myCircuit.Make(circuitConstants, common_data)
 
 		r1cs, _ := frontend.Compile(ecc.BN254.ScalarField(), r1cs.NewBuilder, &myCircuit)
 		pk, vk, _ := groth16.Setup(r1cs)
@@ -63,8 +68,67 @@ var buildCmd = &cobra.Command{
 	},
 }
 
+// buildPlonkCmd represents the buildPlonk
+var buildPlonkCmd = &cobra.Command{
+	Use:   "buildPlonk",
+	Short: "Build gnark plonk circuit",
+	Long:  `Builds gnark plonk circuit corresponding to provided common_data and plonky2 config.`,
+	Run: func(cmd *cobra.Command, args []string) {
+		fmt.Printf("build called:\n common data: %s\n ", common_data_path)
+
+		common_data, err := ReadCommonDataFromFile(common_data_path)
+		if err != nil {
+			fmt.Println("Failed to read common data file:", err)
+			os.Exit(1)
+		}
+		circuitConstants := GetCircuitConstants(common_data)
+
+		var myCircuit verifier.Runner
+
+		// Arrays are resized according to circuitConstants before compiling
+		myCircuit.Make(circuitConstants, common_data)
+
+		ccs, _ := frontend.Compile(ecc.BN254.ScalarField(), scs.NewBuilder, &myCircuit)
+		// r1cs := ccs.(*cs.SparseR1CS)
+		// scs := r1cs.(*cs.SparseR1CS)
+		// srs, err := test.NewKZGSRS(scs)
+		srs, err := kzg_bn254.NewSRS(uint64(math.Pow(2, 28)), big.NewInt(-1))
+		if err != nil {
+			panic(err)
+		}
+		// pk, vk, _ := plonk.Setup(r1cs, srs)
+		pk, vk, _ := plonk.Setup(ccs, srs)
+
+		f_r1cs, err := os.Create("data/r1cs.bin")
+		if err != nil {
+			fmt.Println("Failed to create r1cs file:", err)
+			os.Exit(1)
+		}
+		// r1cs.WriteTo(f_r1cs)
+		ccs.WriteTo(f_r1cs)
+
+		f_pd, _ := os.Create("data/pk.bin")
+		if err != nil {
+			fmt.Println("Failed to create pk file:", err)
+			os.Exit(1)
+		}
+		pk.WriteTo(f_pd)
+
+		f_vk, err := os.Create("data/vk.bin")
+		if err != nil {
+			fmt.Println("Failed to create vk file:", err)
+			os.Exit(1)
+		}
+		vk.WriteTo(f_vk)
+	},
+}
+
 func init() {
-	buildCmd.Flags().StringVarP(&common_data_path, "common_data", "d", "", "JSON File path to common data of plonky2 circuit")
-	_ = buildCmd.MarkFlagRequired("common_data")
-	rootCmd.AddCommand(buildCmd)
+	buildGroth16Cmd.Flags().StringVarP(&common_data_path, "common_data", "d", "", "JSON File path to common data of plonky2 circuit")
+	_ = buildGroth16Cmd.MarkFlagRequired("common_data")
+	rootCmd.AddCommand(buildGroth16Cmd)
+
+	buildPlonkCmd.Flags().StringVarP(&common_data_path, "common_data", "d", "", "JSON File path to common data of plonky2 circuit")
+	_ = buildPlonkCmd.MarkFlagRequired("common_data")
+	rootCmd.AddCommand(buildPlonkCmd)
 }

cmd/export_solidity.go

@@ -0,0 +1,85 @@
+/*
+Copyright © 2023 NAME HERE <EMAIL ADDRESS>
+*/
+package cmd
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/consensys/gnark-crypto/ecc"
+	"github.com/consensys/gnark/backend/groth16"
+	"github.com/consensys/gnark/backend/plonk"
+	"github.com/spf13/cobra"
+)
+
+// exportSolGroth16Cmd represents the exportSolGroth16 command
+var exportSolGroth16Cmd = &cobra.Command{
+	Use:   "exportSolGroth16",
+	Short: "Exports Solidity contract for groth16 backend",
+	Long:  `Exports VerifyingKey as a Solidity contract for groth16 backend`,
+	Run: func(cmd *cobra.Command, args []string) {
+		fmt.Printf("export_solidity called:\n vkey: %s\n ", vkey_path)
+
+		vk := groth16.NewVerifyingKey(ecc.BN254)
+		vkFile, err := os.Open(vkey_path)
+		if err != nil {
+			fmt.Println("vkFile open wrong: ", err)
+			os.Exit(1)
+		}
+		vk.ReadFrom(vkFile)
+
+		f_sol, err := os.Create("data/Verifier.sol")
+		if err != nil {
+			fmt.Println("Failed to create Verifier.sol file:", err)
+			os.Exit(1)
+		}
+		err = vk.ExportSolidity(f_sol)
+		if err != nil {
+			fmt.Println("Failed to export Solidity Verifier")
+			os.Exit(1)
+		} else {
+			fmt.Println("Exported succesfully")
+		}
+	},
+}
+
+var exportSolPlonkCmd = &cobra.Command{
+	Use:   "exportSolPlonk",
+	Short: "Exports Solidity contract for groth16 backend",
+	Long:  `Exports VerifyingKey as a Solidity contract for plonk backend`,
+	Run: func(cmd *cobra.Command, args []string) {
+		fmt.Printf("export_solidity called:\n vkey: %s\n ", vkey_path)
+
+		vk := plonk.NewVerifyingKey(ecc.BN254)
+		vkFile, err := os.Open(vkey_path)
+		if err != nil {
+			fmt.Println("vkFile open wrong: ", err)
+			os.Exit(1)
+		}
+		vk.ReadFrom(vkFile)
+
+		f_sol, err := os.Create("data/Verifier.sol")
+		if err != nil {
+			fmt.Println("Failed to create Verifier.sol file:", err)
+			os.Exit(1)
+		}
+		err = vk.ExportSolidity(f_sol)
+		if err != nil {
+			fmt.Println("Failed to export Solidity Verifier")
+			os.Exit(1)
+		} else {
+			fmt.Println("Exported succesfully")
+		}
+	},
+}
+
+func init() {
+	exportSolGroth16Cmd.Flags().StringVarP(&vkey_path, "vkey_path", "v", "", "path to the vk.bin file")
+	_ = exportSolGroth16Cmd.MarkFlagRequired("vkey")
+	rootCmd.AddCommand(exportSolGroth16Cmd)
+
+	exportSolPlonkCmd.Flags().StringVarP(&vkey_path, "vkey_path", "v", "", "path to the vk.bin file")
+	_ = exportSolPlonkCmd.MarkFlagRequired("vkey")
+	rootCmd.AddCommand(exportSolPlonkCmd)
+}