arnaucube
arnaucube/gnark-plonky2-verifierPublic Self-hosted repository
gnark-plonky2-verifier/plonky2_verifier/deserialize.go
plonky2_verifier/deserialize.goRaw
  1package plonky2_verifier
  2
  3import (
  4	"encoding/json"
  5	. "gnark-plonky2-verifier/field"
  6	"gnark-plonky2-verifier/utils"
  7	"io/ioutil"
  8	"os"
  9)
 10
 11type ProofWithPublicInputsRaw struct {
 12	Proof struct {
 13		WiresCap []struct {
 14			Elements []uint64 `json:"elements"`
 15		} `json:"wires_cap"`
 16		PlonkZsPartialProductsCap []struct {
 17			Elements []uint64 `json:"elements"`
 18		} `json:"plonk_zs_partial_products_cap"`
 19		QuotientPolysCap []struct {
 20			Elements []uint64 `json:"elements"`
 21		} `json:"quotient_polys_cap"`
 22		Openings struct {
 23			Constants       [][]uint64 `json:"constants"`
 24			PlonkSigmas     [][]uint64 `json:"plonk_sigmas"`
 25			Wires           [][]uint64 `json:"wires"`
 26			PlonkZs         [][]uint64 `json:"plonk_zs"`
 27			PlonkZsNext     [][]uint64 `json:"plonk_zs_next"`
 28			PartialProducts [][]uint64 `json:"partial_products"`
 29			QuotientPolys   [][]uint64 `json:"quotient_polys"`
 30		} `json:"openings"`
 31		OpeningProof struct {
 32			CommitPhaseMerkleCaps []MerkleCapsRaw `json:"commit_phase_merkle_caps"`
 33			QueryRoundProofs      []struct {
 34				InitialTreesProof struct {
 35					EvalsProofs []EvalProofRaw `json:"evals_proofs"`
 36				} `json:"initial_trees_proof"`
 37				Steps []struct {
 38					Evals       [][]uint64     `json:"evals"`
 39					MerkleProof MerkleProofRaw `json:"merkle_proof"`
 40				} `json:"steps"`
 41			} `json:"query_round_proofs"`
 42			FinalPoly struct {
 43				Coeffs [][]uint64 `json:"coeffs"`
 44			} `json:"final_poly"`
 45			PowWitness uint64 `json:"pow_witness"`
 46		} `json:"opening_proof"`
 47	} `json:"proof"`
 48	PublicInputs []uint64 `json:"public_inputs"`
 49}
 50
 51type MerkleCapsRaw struct {
 52	hashes [][]uint64
 53}
 54
 55func (m *MerkleCapsRaw) UnmarshalJSON(data []byte) error {
 56	var merkleCaps []map[string][]uint64
 57	if err := json.Unmarshal(data, &merkleCaps); err != nil {
 58		panic(err)
 59	}
 60
 61	m.hashes = make([][]uint64, len(merkleCaps))
 62	for i := 0; i < len(merkleCaps); i++ {
 63		m.hashes[i] = merkleCaps[i]["elements"]
 64	}
 65	return nil
 66}
 67
 68type EvalProofRaw struct {
 69	leafElements []uint64
 70	merkleProof  MerkleProofRaw
 71}
 72
 73func (e *EvalProofRaw) UnmarshalJSON(data []byte) error {
 74	return json.Unmarshal(data, &[]interface{}{&e.leafElements, &e.merkleProof})
 75}
 76
 77type MerkleProofRaw struct {
 78	hash [][]uint64
 79}
 80
 81func (m *MerkleProofRaw) UnmarshalJSON(data []byte) error {
 82	type SiblingObject struct {
 83		Siblings []map[string][]uint64 // "siblings"
 84	}
 85
 86	var siblings SiblingObject
 87	if err := json.Unmarshal(data, &siblings); err != nil {
 88		panic(err)
 89	}
 90
 91	m.hash = make([][]uint64, len(siblings.Siblings))
 92	for siblingIdx, sibling := range siblings.Siblings {
 93		m.hash[siblingIdx] = sibling["elements"]
 94	}
 95
 96	return nil
 97}
 98
 99type CommonCircuitDataRaw struct {
100	Config struct {
101		NumWires                uint64 `json:"num_wires"`
102		NumRoutedWires          uint64 `json:"num_routed_wires"`
103		NumConstants            uint64 `json:"num_constants"`
104		UseBaseArithmeticGate   bool   `json:"use_base_arithmetic_gate"`
105		SecurityBits            uint64 `json:"security_bits"`
106		NumChallenges           uint64 `json:"num_challenges"`
107		ZeroKnowledge           bool   `json:"zero_knowledge"`
108		MaxQuotientDegreeFactor uint64 `json:"max_quotient_degree_factor"`
109		FriConfig               struct {
110			RateBits          uint64 `json:"rate_bits"`
111			CapHeight         uint64 `json:"cap_height"`
112			ProofOfWorkBits   uint64 `json:"proof_of_work_bits"`
113			ReductionStrategy struct {
114				ConstantArityBits []uint64 `json:"ConstantArityBits"`
115			} `json:"reduction_strategy"`
116			NumQueryRounds uint64 `json:"num_query_rounds"`
117		} `json:"fri_config"`
118	} `json:"config"`
119	FriParams struct {
120		Config struct {
121			RateBits          uint64 `json:"rate_bits"`
122			CapHeight         uint64 `json:"cap_height"`
123			ProofOfWorkBits   uint64 `json:"proof_of_work_bits"`
124			ReductionStrategy struct {
125				ConstantArityBits []uint64 `json:"ConstantArityBits"`
126			} `json:"reduction_strategy"`
127			NumQueryRounds uint64 `json:"num_query_rounds"`
128		} `json:"config"`
129		Hiding             bool     `json:"hiding"`
130		DegreeBits         uint64   `json:"degree_bits"`
131		ReductionArityBits []uint64 `json:"reduction_arity_bits"`
132	} `json:"fri_params"`
133	DegreeBits    uint64   `json:"degree_bits"`
134	Gates         []string `json:"gates"`
135	SelectorsInfo struct {
136		SelectorIndices []uint64 `json:"selector_indices"`
137		Groups          []struct {
138			Start uint64 `json:"start"`
139			End   uint64 `json:"end"`
140		} `json:"groups"`
141	} `json:"selectors_info"`
142	QuotientDegreeFactor uint64   `json:"quotient_degree_factor"`
143	NumGateConstraints   uint64   `json:"num_gate_constraints"`
144	NumConstants         uint64   `json:"num_constants"`
145	NumPublicInputs      uint64   `json:"num_public_inputs"`
146	KIs                  []uint64 `json:"k_is"`
147	NumPartialProducts   uint64   `json:"num_partial_products"`
148	CircuitDigest        struct {
149		Elements []uint64 `json:"elements"`
150	} `json:"circuit_digest"`
151}
152
153type VerifierOnlyCircuitDataRaw struct {
154	ConstantsSigmasCap []struct {
155		Elements []uint64 `json:"elements"`
156	} `json:"constants_sigmas_cap"`
157}
158
159func DeserializeMerkleCap(merkleCapRaw []struct{ Elements []uint64 }) MerkleCap {
160	n := len(merkleCapRaw)
161	merkleCap := make([]Hash, n)
162	for i := 0; i < n; i++ {
163		copy(merkleCap[i][:], utils.Uint64ArrayToFArray(merkleCapRaw[i].Elements))
164	}
165	return merkleCap
166}
167
168func DeserializeMerkleProof(merkleProofRaw struct{ Siblings []interface{} }) MerkleProof {
169	n := len(merkleProofRaw.Siblings)
170	var mp MerkleProof
171	mp.Siblings = make([]Hash, n)
172	for i := 0; i < n; i++ {
173		element := merkleProofRaw.Siblings[i].(struct{ Elements []uint64 })
174		copy(mp.Siblings[i][:], utils.Uint64ArrayToFArray(element.Elements))
175	}
176	return mp
177}
178
179func DeserializeOpeningSet(openingSetRaw struct {
180	Constants       [][]uint64
181	PlonkSigmas     [][]uint64
182	Wires           [][]uint64
183	PlonkZs         [][]uint64
184	PlonkZsNext     [][]uint64
185	PartialProducts [][]uint64
186	QuotientPolys   [][]uint64
187}) OpeningSet {
188	return OpeningSet{
189		Constants:       utils.Uint64ArrayToQuadraticExtensionArray(openingSetRaw.Constants),
190		PlonkSigmas:     utils.Uint64ArrayToQuadraticExtensionArray(openingSetRaw.PlonkSigmas),
191		Wires:           utils.Uint64ArrayToQuadraticExtensionArray(openingSetRaw.Wires),
192		PlonkZs:         utils.Uint64ArrayToQuadraticExtensionArray(openingSetRaw.PlonkZs),
193		PlonkZsNext:     utils.Uint64ArrayToQuadraticExtensionArray(openingSetRaw.PlonkZsNext),
194		PartialProducts: utils.Uint64ArrayToQuadraticExtensionArray(openingSetRaw.PartialProducts),
195		QuotientPolys:   utils.Uint64ArrayToQuadraticExtensionArray(openingSetRaw.QuotientPolys),
196	}
197}
198
199func DeserializeFriProof(openingProofRaw struct {
200	CommitPhaseMerkleCaps []MerkleCapsRaw
201	QueryRoundProofs      []struct {
202		InitialTreesProof struct {
203			EvalsProofs []EvalProofRaw
204		}
205		Steps []struct {
206			Evals       [][]uint64
207			MerkleProof MerkleProofRaw
208		}
209	}
210	FinalPoly struct {
211		Coeffs [][]uint64
212	}
213	PowWitness uint64
214}) FriProof {
215	var openingProof FriProof
216	openingProof.PowWitness = NewFieldElement(openingProofRaw.PowWitness)
217	openingProof.FinalPoly.Coeffs = utils.Uint64ArrayToQuadraticExtensionArray(openingProofRaw.FinalPoly.Coeffs)
218
219	openingProof.CommitPhaseMerkleCaps = make([]MerkleCap, len(openingProofRaw.CommitPhaseMerkleCaps))
220	for i := 0; i < len(openingProofRaw.CommitPhaseMerkleCaps); i++ {
221		openingProof.CommitPhaseMerkleCaps[i] = utils.Uint64ArrayToHashArray(openingProofRaw.CommitPhaseMerkleCaps[i].hashes)
222	}
223
224	numQueryRoundProofs := len(openingProofRaw.QueryRoundProofs)
225	openingProof.QueryRoundProofs = make([]FriQueryRound, numQueryRoundProofs)
226
227	for i := 0; i < numQueryRoundProofs; i++ {
228		numEvalProofs := len(openingProofRaw.QueryRoundProofs[i].InitialTreesProof.EvalsProofs)
229		openingProof.QueryRoundProofs[i].InitialTreesProof.EvalsProofs = make([]EvalProof, numEvalProofs)
230		for j := 0; j < numEvalProofs; j++ {
231			openingProof.QueryRoundProofs[i].InitialTreesProof.EvalsProofs[j].Elements = utils.Uint64ArrayToFArray(openingProofRaw.QueryRoundProofs[i].InitialTreesProof.EvalsProofs[j].leafElements)
232			openingProof.QueryRoundProofs[i].InitialTreesProof.EvalsProofs[j].MerkleProof.Siblings = utils.Uint64ArrayToHashArray(openingProofRaw.QueryRoundProofs[i].InitialTreesProof.EvalsProofs[j].merkleProof.hash)
233		}
234
235		numSteps := len(openingProofRaw.QueryRoundProofs[i].Steps)
236		openingProof.QueryRoundProofs[i].Steps = make([]FriQueryStep, numSteps)
237		for j := 0; j < numSteps; j++ {
238			openingProof.QueryRoundProofs[i].Steps[j].Evals = utils.Uint64ArrayToQuadraticExtensionArray(openingProofRaw.QueryRoundProofs[i].Steps[j].Evals)
239			openingProof.QueryRoundProofs[i].Steps[j].MerkleProof.Siblings = utils.Uint64ArrayToHashArray(openingProofRaw.QueryRoundProofs[i].Steps[j].MerkleProof.hash)
240		}
241	}
242
243	return openingProof
244}
245
246func DeserializeProofWithPublicInputs(path string) ProofWithPublicInputs {
247	jsonFile, err := os.Open(path)
248	if err != nil {
249		panic(err)
250	}
251
252	defer jsonFile.Close()
253	rawBytes, _ := ioutil.ReadAll(jsonFile)
254
255	var raw ProofWithPublicInputsRaw
256	err = json.Unmarshal(rawBytes, &raw)
257	if err != nil {
258		panic(err)
259	}
260
261	var proofWithPis ProofWithPublicInputs
262	proofWithPis.Proof.WiresCap = DeserializeMerkleCap([]struct{ Elements []uint64 }(raw.Proof.WiresCap))
263	proofWithPis.Proof.PlonkZsPartialProductsCap = DeserializeMerkleCap([]struct{ Elements []uint64 }(raw.Proof.PlonkZsPartialProductsCap))
264	proofWithPis.Proof.QuotientPolysCap = DeserializeMerkleCap([]struct{ Elements []uint64 }(raw.Proof.QuotientPolysCap))
265	proofWithPis.Proof.Openings = DeserializeOpeningSet(struct {
266		Constants       [][]uint64
267		PlonkSigmas     [][]uint64
268		Wires           [][]uint64
269		PlonkZs         [][]uint64
270		PlonkZsNext     [][]uint64
271		PartialProducts [][]uint64
272		QuotientPolys   [][]uint64
273	}(raw.Proof.Openings))
274	proofWithPis.Proof.OpeningProof = DeserializeFriProof(struct {
275		CommitPhaseMerkleCaps []MerkleCapsRaw
276		QueryRoundProofs      []struct {
277			InitialTreesProof struct {
278				EvalsProofs []EvalProofRaw
279			}
280			Steps []struct {
281				Evals       [][]uint64
282				MerkleProof MerkleProofRaw
283			}
284		}
285		FinalPoly  struct{ Coeffs [][]uint64 }
286		PowWitness uint64
287	}(raw.Proof.OpeningProof))
288	proofWithPis.PublicInputs = utils.Uint64ArrayToFArray(raw.PublicInputs)
289
290	return proofWithPis
291}
292
293func ReductionArityBits(
294	arityBits uint64,
295	finalPolyBits uint64,
296	degreeBits uint64,
297	rateBits uint64,
298	capHeight uint64,
299) []uint64 {
300	returnArr := make([]uint64, 0)
301
302	for degreeBits > finalPolyBits && degreeBits+rateBits-arityBits >= capHeight {
303		returnArr = append(returnArr, arityBits)
304		if degreeBits < arityBits {
305			panic("degreeBits < arityBits")
306		}
307		degreeBits -= arityBits
308	}
309
310	return returnArr
311}
312
313func DeserializeCommonCircuitData(path string) CommonCircuitData {
314	jsonFile, err := os.Open(path)
315	if err != nil {
316		panic(err)
317	}
318
319	defer jsonFile.Close()
320	rawBytes, _ := ioutil.ReadAll(jsonFile)
321
322	var raw CommonCircuitDataRaw
323	err = json.Unmarshal(rawBytes, &raw)
324	if err != nil {
325		panic(err)
326	}
327
328	var commonCircuitData CommonCircuitData
329	commonCircuitData.Config.NumWires = raw.Config.NumWires
330	commonCircuitData.Config.NumRoutedWires = raw.Config.NumRoutedWires
331	commonCircuitData.Config.NumConstants = raw.Config.NumConstants
332	commonCircuitData.Config.UseBaseArithmeticGate = raw.Config.UseBaseArithmeticGate
333	commonCircuitData.Config.SecurityBits = raw.Config.SecurityBits
334	commonCircuitData.Config.NumChallenges = raw.Config.NumChallenges
335	commonCircuitData.Config.ZeroKnowledge = raw.Config.ZeroKnowledge
336	commonCircuitData.Config.MaxQuotientDegreeFactor = raw.Config.MaxQuotientDegreeFactor
337
338	commonCircuitData.Config.FriConfig.RateBits = raw.Config.FriConfig.RateBits
339	commonCircuitData.Config.FriConfig.CapHeight = raw.Config.FriConfig.CapHeight
340	commonCircuitData.Config.FriConfig.ProofOfWorkBits = raw.Config.FriConfig.ProofOfWorkBits
341	commonCircuitData.Config.FriConfig.NumQueryRounds = raw.Config.FriConfig.NumQueryRounds
342
343	commonCircuitData.FriParams.DegreeBits = raw.FriParams.DegreeBits
344	commonCircuitData.FriParams.Config.RateBits = raw.FriParams.Config.RateBits
345	commonCircuitData.FriParams.Config.CapHeight = raw.FriParams.Config.CapHeight
346	commonCircuitData.FriParams.Config.ProofOfWorkBits = raw.FriParams.Config.ProofOfWorkBits
347	commonCircuitData.FriParams.Config.NumQueryRounds = raw.FriParams.Config.NumQueryRounds
348	commonCircuitData.FriParams.ReductionArityBits = raw.FriParams.ReductionArityBits
349	commonCircuitData.DegreeBits = raw.DegreeBits
350
351	commonCircuitData.Gates = []gate{}
352	for _, gate := range raw.Gates {
353		commonCircuitData.Gates = append(commonCircuitData.Gates, GateInstanceFromId(gate))
354	}
355
356	commonCircuitData.SelectorsInfo.selectorIndices = raw.SelectorsInfo.SelectorIndices
357	commonCircuitData.SelectorsInfo.groups = []Range{}
358	for _, group := range raw.SelectorsInfo.Groups {
359		commonCircuitData.SelectorsInfo.groups = append(commonCircuitData.SelectorsInfo.groups, Range{
360			start: group.Start,
361			end:   group.End,
362		})
363	}
364
365	commonCircuitData.QuotientDegreeFactor = raw.QuotientDegreeFactor
366	commonCircuitData.NumGateConstraints = raw.NumGateConstraints
367	commonCircuitData.NumConstants = raw.NumConstants
368	commonCircuitData.NumPublicInputs = raw.NumPublicInputs
369	commonCircuitData.KIs = utils.Uint64ArrayToFArray(raw.KIs)
370	commonCircuitData.NumPartialProducts = raw.NumPartialProducts
371	copy(commonCircuitData.CircuitDigest[:], utils.Uint64ArrayToFArray(raw.CircuitDigest.Elements))
372
373	return commonCircuitData
374}
375
376func DeserializeVerifierOnlyCircuitData(path string) VerifierOnlyCircuitData {
377	jsonFile, err := os.Open(path)
378	if err != nil {
379		panic(err)
380	}
381
382	defer jsonFile.Close()
383	rawBytes, _ := ioutil.ReadAll(jsonFile)
384
385	var raw VerifierOnlyCircuitDataRaw
386	err = json.Unmarshal(rawBytes, &raw)
387	if err != nil {
388		panic(err)
389	}
390
391	return VerifierOnlyCircuitData{
392		ConstantSigmasCap: DeserializeMerkleCap([]struct{ Elements []uint64 }(raw.ConstantsSigmasCap)),
393	}
394}