mirror of
https://github.com/logos-storage/logos-storage-nim.git
synced 2026-01-14 03:18:17 +00:00
110 lines
2.6 KiB
Nim
110 lines
2.6 KiB
Nim
## Nim-Codex
|
|
## Copyright (c) 2024 Status Research & Development GmbH
|
|
## Licensed under either of
|
|
## * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
|
|
## * MIT license ([LICENSE-MIT](LICENSE-MIT))
|
|
## at your option.
|
|
## This file may not be copied, modified, or distributed except according to
|
|
## those terms.
|
|
##
|
|
|
|
import pkg/chronos
|
|
import pkg/chronicles
|
|
import pkg/circomcompat
|
|
import pkg/poseidon2
|
|
import pkg/taskpools
|
|
import pkg/questionable/results
|
|
|
|
import pkg/libp2p/cid
|
|
|
|
import ../../manifest
|
|
import ../../merkletree
|
|
import ../../stores
|
|
import ../../market
|
|
import ../../utils/poseidon2digest
|
|
import ../../conf
|
|
|
|
import ../builder
|
|
import ../sampler
|
|
|
|
import ./backends
|
|
import ../types
|
|
|
|
export backends
|
|
|
|
logScope:
|
|
topics = "codex prover"
|
|
|
|
type
|
|
Prover* = ref object
|
|
case backendKind: ProverBackendCmd
|
|
of ProverBackendCmd.nimGroth16:
|
|
groth16Backend*: NimGroth16BackendRef
|
|
of ProverBackendCmd.circomCompat:
|
|
circomCompatBackend*: CircomCompatBackendRef
|
|
nSamples: int
|
|
tp: Taskpool
|
|
|
|
AnyBackend* = NimGroth16BackendRef or CircomCompatBackendRef
|
|
|
|
proc prove*[SomeSampler](
|
|
self: Prover,
|
|
sampler: SomeSampler,
|
|
manifest: Manifest,
|
|
challenge: ProofChallenge,
|
|
verify = false,
|
|
): Future[?!(Groth16Proof, ?bool)] {.async: (raises: [CancelledError]).} =
|
|
## Prove a statement using backend.
|
|
## Returns a future that resolves to a proof.
|
|
|
|
logScope:
|
|
cid = manifest.treeCid
|
|
slot = slotIdx
|
|
challenge = challenge
|
|
|
|
trace "Received proof challenge"
|
|
|
|
let
|
|
proofInput = ?await sampler.getProofInput(challenge, self.nSamples)
|
|
# prove slot
|
|
|
|
case self.backendKind
|
|
of ProverBackendCmd.nimGroth16:
|
|
let
|
|
proof = ?await self.groth16Backend.prove(proofInput)
|
|
verified =
|
|
if verify:
|
|
(?await self.groth16Backend.verify(proof)).some
|
|
else:
|
|
bool.none
|
|
return success (proof.toGroth16Proof, verified)
|
|
of ProverBackendCmd.circomCompat:
|
|
let
|
|
proof = ?await self.circomCompatBackend.prove(proofInput)
|
|
verified =
|
|
if verify:
|
|
(?await self.circomCompatBackend.verify(proof, proofInput)).some
|
|
else:
|
|
bool.none
|
|
return success (proof.toGroth16Proof, verified)
|
|
|
|
proc new*(
|
|
_: type Prover, backend: CircomCompatBackendRef, nSamples: int, tp: Taskpool
|
|
): Prover =
|
|
Prover(
|
|
circomCompatBackend: backend,
|
|
backendKind: ProverBackendCmd.circomCompat,
|
|
nSamples: nSamples,
|
|
tp: tp,
|
|
)
|
|
|
|
proc new*(
|
|
_: type Prover, backend: NimGroth16BackendRef, nSamples: int, tp: Taskpool
|
|
): Prover =
|
|
Prover(
|
|
groth16Backend: backend,
|
|
backendKind: ProverBackendCmd.nimGroth16,
|
|
nSamples: nSamples,
|
|
tp: tp,
|
|
)
|