mirror of
https://github.com/logos-storage/logos-storage-nim.git
synced 2026-05-16 08:19:46 +00:00
617d2982a3
Part of https://github.com/logos-storage/logos-storage-nim/issues/896 Signed-off-by: Chrysostomos Nanakos <chris@include.gr>
127 lines
3.2 KiB
Nim
127 lines
3.2 KiB
Nim
import std/options
|
|
|
|
import ../../../asynctest
|
|
|
|
import pkg/chronos
|
|
import pkg/taskpools
|
|
import pkg/poseidon2
|
|
import pkg/serde/json
|
|
|
|
import pkg/codex/slots {.all.}
|
|
import pkg/codex/slots/types {.all.}
|
|
import pkg/codex/merkletree
|
|
import pkg/codex/codextypes
|
|
import pkg/codex/manifest
|
|
import pkg/codex/stores
|
|
import pkg/codex/erasure
|
|
|
|
import ./helpers
|
|
import ../helpers
|
|
import ../../helpers
|
|
|
|
suite "Test Circom Compat Backend - control inputs":
|
|
let
|
|
r1cs = "tests/circuits/fixtures/proof_main.r1cs"
|
|
wasm = "tests/circuits/fixtures/proof_main.wasm"
|
|
zkey = "tests/circuits/fixtures/proof_main.zkey"
|
|
|
|
var
|
|
circom: CircomCompat
|
|
proofInputs: ProofInputs[Poseidon2Hash]
|
|
|
|
setup:
|
|
let
|
|
inputData = readFile("tests/circuits/fixtures/input.json")
|
|
inputJson = !JsonNode.parse(inputData)
|
|
|
|
proofInputs = Poseidon2Hash.jsonToProofInput(inputJson)
|
|
circom = CircomCompat.init(r1cs, wasm, zkey)
|
|
|
|
teardown:
|
|
circom.release() # this comes from the rust FFI
|
|
|
|
test "Should verify with correct inputs":
|
|
let proof = circom.prove(proofInputs).tryGet
|
|
|
|
check circom.verify(proof, proofInputs).tryGet
|
|
|
|
test "Should not verify with incorrect inputs":
|
|
proofInputs.slotIndex = 1 # change slot index
|
|
|
|
let proof = circom.prove(proofInputs).tryGet
|
|
|
|
check circom.verify(proof, proofInputs).tryGet == false
|
|
|
|
suite "Test Circom Compat Backend":
|
|
let
|
|
ecK = 2
|
|
ecM = 2
|
|
slotId = 3
|
|
samples = 5
|
|
numDatasetBlocks = 8
|
|
blockSize = DefaultBlockSize
|
|
cellSize = DefaultCellSize
|
|
|
|
r1cs = "tests/circuits/fixtures/proof_main.r1cs"
|
|
wasm = "tests/circuits/fixtures/proof_main.wasm"
|
|
zkey = "tests/circuits/fixtures/proof_main.zkey"
|
|
|
|
repoTmp = TempLevelDb.new()
|
|
metaTmp = TempLevelDb.new()
|
|
|
|
var
|
|
store: BlockStore
|
|
manifest: Manifest
|
|
protected: Manifest
|
|
verifiable: Manifest
|
|
circom: CircomCompat
|
|
proofInputs: ProofInputs[Poseidon2Hash]
|
|
challenge: array[32, byte]
|
|
builder: Poseidon2Builder
|
|
sampler: Poseidon2Sampler
|
|
taskpool: Taskpool
|
|
erasure: Erasure
|
|
|
|
setup:
|
|
let
|
|
repoDs = repoTmp.newDb()
|
|
metaDs = metaTmp.newDb()
|
|
|
|
store = RepoStore.new(repoDs, metaDs)
|
|
taskpool = Taskpool.new()
|
|
erasure = Erasure.new(store, leoEncoderProvider, leoDecoderProvider, taskpool)
|
|
|
|
(manifest, protected, verifiable) = await createVerifiableManifest(
|
|
store, numDatasetBlocks, ecK, ecM, blockSize, cellSize, erasure
|
|
)
|
|
|
|
builder = Poseidon2Builder.new(store, verifiable, erasure).tryGet
|
|
sampler = Poseidon2Sampler.new(slotId, store, builder).tryGet
|
|
|
|
circom = CircomCompat.init(r1cs, wasm, zkey)
|
|
challenge = 1234567.toF.toBytes.toArray32
|
|
|
|
proofInputs = (await sampler.getProofInput(challenge, samples)).tryGet
|
|
|
|
teardown:
|
|
circom.release() # this comes from the rust FFI
|
|
await repoTmp.destroyDb()
|
|
await metaTmp.destroyDb()
|
|
|
|
if not taskpool.isNil:
|
|
taskpool.shutdown()
|
|
|
|
reset(taskpool)
|
|
|
|
test "Should verify with correct input":
|
|
var proof = circom.prove(proofInputs).tryGet
|
|
|
|
check circom.verify(proof, proofInputs).tryGet
|
|
|
|
test "Should not verify with incorrect input":
|
|
proofInputs.slotIndex = 1 # change slot index
|
|
|
|
let proof = circom.prove(proofInputs).tryGet
|
|
|
|
check circom.verify(proof, proofInputs).tryGet == false
|