You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
2.3 KiB
80 lines
2.3 KiB
package gocircomprover
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestSmallCircuitGenerateProf(t *testing.T) {
|
|
provingKeyJson, err := ioutil.ReadFile("testdata/small/proving_key.json")
|
|
require.Nil(t, err)
|
|
pk, err := ParseProvingKey(provingKeyJson)
|
|
require.Nil(t, err)
|
|
|
|
fmt.Println("polsA", pk.PolsA)
|
|
fmt.Println("polsB", pk.PolsB)
|
|
fmt.Println("polsC", pk.PolsC)
|
|
|
|
witnessJson, err := ioutil.ReadFile("testdata/small/witness.json")
|
|
require.Nil(t, err)
|
|
w, err := ParseWitness(witnessJson)
|
|
require.Nil(t, err)
|
|
|
|
assert.Equal(t, Witness{big.NewInt(1), big.NewInt(33), big.NewInt(3), big.NewInt(11)}, w)
|
|
|
|
proof, pubSignals, err := GenerateProof(pk, w)
|
|
assert.Nil(t, err)
|
|
fmt.Println("proof", proof)
|
|
fmt.Println("pubSignals", pubSignals)
|
|
|
|
proofStr, err := ProofToJson(proof)
|
|
assert.Nil(t, err)
|
|
fmt.Println("prover\n", string(proofStr))
|
|
|
|
err = ioutil.WriteFile("testdata/small/proof.json", proofStr, 0644)
|
|
assert.Nil(t, err)
|
|
publicStr, err := json.Marshal(ArrayBigIntToString(pubSignals))
|
|
assert.Nil(t, err)
|
|
err = ioutil.WriteFile("testdata/small/public.json", publicStr, 0644)
|
|
assert.Nil(t, err)
|
|
|
|
// to verify the proof:
|
|
// snarkjs verify --vk testdata/small/verification_key.json -p testdata/small/proof.json --pub testdata/small/public.json
|
|
}
|
|
|
|
func TestBigCircuitGenerateProf(t *testing.T) {
|
|
provingKeyJson, err := ioutil.ReadFile("testdata/big/proving_key.json")
|
|
require.Nil(t, err)
|
|
pk, err := ParseProvingKey(provingKeyJson)
|
|
require.Nil(t, err)
|
|
|
|
witnessJson, err := ioutil.ReadFile("testdata/big/witness.json")
|
|
require.Nil(t, err)
|
|
w, err := ParseWitness(witnessJson)
|
|
require.Nil(t, err)
|
|
|
|
proof, pubSignals, err := GenerateProof(pk, w)
|
|
assert.Nil(t, err)
|
|
fmt.Println("proof", proof)
|
|
fmt.Println("pubSignals", pubSignals)
|
|
|
|
proofStr, err := ProofToJson(proof)
|
|
assert.Nil(t, err)
|
|
fmt.Println("prover\n", string(proofStr))
|
|
|
|
err = ioutil.WriteFile("testdata/big/proof.json", proofStr, 0644)
|
|
assert.Nil(t, err)
|
|
publicStr, err := json.Marshal(ArrayBigIntToString(pubSignals))
|
|
assert.Nil(t, err)
|
|
err = ioutil.WriteFile("testdata/big/public.json", publicStr, 0644)
|
|
assert.Nil(t, err)
|
|
|
|
// to verify the proof:
|
|
// snarkjs verify --vk testdata/big/verification_key.json -p testdata/big/proof.json --pub testdata/big/public.json
|
|
}
|