1package plonky2_verifier
  2
  3import (
  4	"gnark-ed25519/field"
  5	. "gnark-ed25519/field"
  6	. "gnark-ed25519/poseidon"
  7	"gnark-ed25519/utils"
  8	"testing"
  9
 10	"github.com/consensys/gnark/frontend"
 11	"github.com/consensys/gnark/test"
 12)
 13
 14type TestChallengerCircuit struct {
 15	PublicInputs  [3]frontend.Variable
 16	CircuitDigest [4]frontend.Variable
 17	WiresCap      [16][4]frontend.Variable
 18}
 19
 20func (circuit *TestChallengerCircuit) Define(api frontend.API) error {
 21	field := field.NewFieldAPI(api)
 22	poseidonChip := NewPoseidonChip(api, field)
 23	challengerChip := NewChallengerChip(api, field, *poseidonChip)
 24
 25	var circuitDigest [4]F
 26	for i := 0; i < len(circuitDigest); i++ {
 27		circuitDigest[i] = field.FromBinary(api.ToBinary(circuit.CircuitDigest[i], 64)).(F)
 28	}
 29
 30	var publicInputs [3]F
 31	for i := 0; i < len(publicInputs); i++ {
 32		publicInputs[i] = field.FromBinary(api.ToBinary(circuit.PublicInputs[i], 64)).(F)
 33	}
 34
 35	var wiresCap [16][4]F
 36	for i := 0; i < len(wiresCap); i++ {
 37		for j := 0; j < len(wiresCap[0]); j++ {
 38			wiresCap[i][j] = field.FromBinary(api.ToBinary(circuit.WiresCap[i][j], 64)).(F)
 39		}
 40	}
 41
 42	publicInputHash := poseidonChip.HashNoPad(publicInputs[:])
 43	challengerChip.ObserveHash(circuitDigest)
 44	challengerChip.ObserveHash(publicInputHash)
 45	challengerChip.ObserveCap(wiresCap[:])
 46
 47	numChallenges := uint64(2)
 48	plonkBetas := challengerChip.GetNChallenges(numChallenges)
 49	plonkGammas := challengerChip.GetNChallenges(numChallenges)
 50
 51	expectedPublicInputHash := [4]F{
 52		NewFieldElementFromString("8416658900775745054"),
 53		NewFieldElementFromString("12574228347150446423"),
 54		NewFieldElementFromString("9629056739760131473"),
 55		NewFieldElementFromString("3119289788404190010"),
 56	}
 57
 58	for i := 0; i < 4; i++ {
 59		field.AssertIsEqual(publicInputHash[i], expectedPublicInputHash[i])
 60	}
 61
 62	expectedPlonkBetas := [2]F{
 63		NewFieldElementFromString("4678728155650926271"),
 64		NewFieldElementFromString("13611962404289024887"),
 65	}
 66
 67	expectedPlonkGammas := [2]frontend.Variable{
 68		NewFieldElementFromString("13237663823305715949"),
 69		NewFieldElementFromString("15389314098328235145"),
 70	}
 71
 72	for i := 0; i < 2; i++ {
 73		field.AssertIsEqual(plonkBetas[i], expectedPlonkBetas[i])
 74		field.AssertIsEqual(plonkGammas[i], expectedPlonkGammas[i])
 75	}
 76
 77	return nil
 78}
 79
 80func TestChallengerWitness(t *testing.T) {
 81	assert := test.NewAssert(t)
 82
 83	testCase := func(publicInputs [3]frontend.Variable, circuitDigest [4]frontend.Variable, wiresCap [16][4]frontend.Variable) {
 84		circuit := TestChallengerCircuit{PublicInputs: publicInputs, CircuitDigest: circuitDigest, WiresCap: wiresCap}
 85		witness := TestChallengerCircuit{PublicInputs: publicInputs, CircuitDigest: circuitDigest, WiresCap: wiresCap}
 86		err := test.IsSolved(&circuit, &witness, TEST_CURVE.ScalarField())
 87		assert.NoError(err)
 88	}
 89
 90	publicInputsStr := []string{"0", "1", "3736710860384812976"}
 91	circuitDigestStr := []string{"7754113318730736048", "18436136620016916513", "18054530212389526288", "5893739326632906028"}
 92	wiresCapStr := [][]string{
 93		{"13884351014873073118", "5174249846243191862", "2208632528791973868", "1071582828677910652"},
 94		{"11475361245556894879", "14867351574926692044", "17013374066934071379", "1027671036932569748"},
 95		{"5604634992452399010", "3684464596850094189", "5565599237356852406", "4136295609943151014"},
 96		{"8463721840990025805", "5922588965472526198", "8096699027533803435", "2210089353004111478"},
 97		{"17531628199677307555", "11513452064460680964", "1482441508929181375", "5139566233781982440"},
 98		{"13271417993289093233", "17257193898955790413", "16883807866578566670", "7423179920948669117"},
 99		{"13462567520785358202", "15555103598281658890", "5859961276885232601", "4464568704709749394"},
100		{"153012620162729043", "14072764618167122665", "3025694603779494447", "15948104906680148838"},
101		{"18050235253694287284", "11467396424826912141", "11302553396166323353", "10976271719722841224"},
102		{"15208241660644051470", "8520722208187871063", "10775022596056682771", "16048513824198271730"},
103		{"6929477084755896240", "11382029470138215117", "13205948643259905511", "9421863267852221772"},
104		{"15449187573546292268", "10216729601353604194", "9493934392442974211", "9848643714440191835"},
105		{"2172475758127444753", "16681095938683502188", "9983383760611275566", "2603547977557388755"},
106		{"17440301588003279095", "11799356585691460705", "1386003375936412946", "11059100806278290279"},
107		{"10758265002546797581", "1374136260999724547", "7200401521491969338", "219493657547391496"},
108		{"5995963332181008902", "4442996285152250372", "2005936434281221193", "6869325719052666642"},
109	}
110
111	var publicInputs [3]frontend.Variable
112	var circuitDigest [4]frontend.Variable
113	var wiresCap [16][4]frontend.Variable
114
115	copy(publicInputs[:], utils.StrArrayToFrontendVariableArray(publicInputsStr))
116	copy(circuitDigest[:], utils.StrArrayToFrontendVariableArray(circuitDigestStr))
117	for i := 0; i < len(wiresCapStr); i++ {
118		copy(wiresCap[i][:], utils.StrArrayToFrontendVariableArray(wiresCapStr[i]))
119	}
120
121	testCase(publicInputs, circuitDigest, wiresCap)
122}