Fill up verifier implementation
This commit is contained in:
parent
c416b173d4
commit
102d9f1b4b
|
@ -1,20 +1,24 @@
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
from typing import List, Optional
|
from hashlib import sha3_256
|
||||||
|
from typing import List, Optional, Sequence
|
||||||
|
|
||||||
|
from eth2spec.deneb.mainnet import BLSFieldElement
|
||||||
from eth2spec.eip7594.mainnet import (
|
from eth2spec.eip7594.mainnet import (
|
||||||
KZGCommitment as Commitment,
|
KZGCommitment as Commitment,
|
||||||
KZGProof as Proof,
|
KZGProof as Proof,
|
||||||
BYTES_PER_CELL as BYTES_PER_CHUNK
|
|
||||||
)
|
)
|
||||||
from itertools import batched
|
from itertools import batched
|
||||||
|
|
||||||
from da.common import Column, Chunk, Attestation
|
from da.common import Column, Chunk, Attestation
|
||||||
|
from da.encoder import DAEncoder
|
||||||
|
from da.kzg_rs import kzg
|
||||||
|
from da.kzg_rs.common import ROOTS_OF_UNITY, GLOBAL_PARAMETERS, BLS_MODULUS
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class DABlob:
|
class DABlob:
|
||||||
# this should be removed, but for now it shows the purpose
|
|
||||||
index: int
|
index: int
|
||||||
column: bytearray
|
column: Column
|
||||||
column_commitment: Commitment
|
column_commitment: Commitment
|
||||||
aggregated_column_commitment: Commitment
|
aggregated_column_commitment: Commitment
|
||||||
aggregated_column_proof: Proof
|
aggregated_column_proof: Proof
|
||||||
|
@ -32,29 +36,37 @@ class DAVerifier:
|
||||||
column_commitment: Commitment,
|
column_commitment: Commitment,
|
||||||
aggregated_column_commitment: Commitment,
|
aggregated_column_commitment: Commitment,
|
||||||
aggregated_column_proof: Proof,
|
aggregated_column_proof: Proof,
|
||||||
# this is temporary and should be removed
|
|
||||||
index: int
|
index: int
|
||||||
) -> bool:
|
) -> bool:
|
||||||
|
|
||||||
# 1. compute commitment for column
|
# 1. compute commitment for column
|
||||||
|
_, computed_column_commitment = kzg.bytes_to_commitment(column.as_bytes(), GLOBAL_PARAMETERS)
|
||||||
# 2. If computed column commitment != column commitment, fail
|
# 2. If computed column commitment != column commitment, fail
|
||||||
|
if column_commitment != computed_column_commitment:
|
||||||
|
return False
|
||||||
# 3. compute column hash
|
# 3. compute column hash
|
||||||
column_hash: bytearray = bytearray(hash(column))
|
column_hash = DAEncoder.hash_column_and_commitment(column, column_commitment)
|
||||||
# 4. Check proof with commitment and proof over the aggregated column commitment
|
# 4. Check proof with commitment and proof over the aggregated column commitment
|
||||||
return False
|
chunk = BLSFieldElement.from_bytes(column_hash)
|
||||||
|
return kzg.verify_element_proof(
|
||||||
|
chunk, aggregated_column_commitment, aggregated_column_proof, index, ROOTS_OF_UNITY
|
||||||
|
)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _verify_chunk(chunk: Chunk, commitment: Commitment, proof: Proof) -> bool:
|
def _verify_chunk(chunk: Chunk, commitment: Commitment, proof: Proof, index: int) -> bool:
|
||||||
pass
|
chunk = BLSFieldElement.from_bytes((int.from_bytes(chunk) % BLS_MODULUS).to_bytes())
|
||||||
|
return kzg.verify_element_proof(chunk, commitment, proof, index, ROOTS_OF_UNITY)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _verify_chunks(
|
def _verify_chunks(
|
||||||
chunks: List[Chunk],
|
chunks: Sequence[Chunk],
|
||||||
commitments: List[Commitment],
|
commitments: Sequence[Commitment],
|
||||||
proofs: List[Proof]
|
proofs: Sequence[Proof],
|
||||||
|
index: int
|
||||||
) -> bool:
|
) -> bool:
|
||||||
|
if not (len(chunks) == len(commitments) == len(proofs)):
|
||||||
|
return False
|
||||||
for chunk, commitment, proof in zip(chunks, commitments, proofs):
|
for chunk, commitment, proof in zip(chunks, commitments, proofs):
|
||||||
if not DAVerifier._verify_chunk(chunk, commitment, proof):
|
if not DAVerifier._verify_chunk(chunk, commitment, proof, index):
|
||||||
return False
|
return False
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
@ -62,16 +74,15 @@ class DAVerifier:
|
||||||
return Attestation()
|
return Attestation()
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def verify(self, blob: DABlob) -> Optional[Attestation]:
|
def verify(blob: DABlob) -> Optional[Attestation]:
|
||||||
is_column_verified = DAVerifier._verify_column(
|
is_column_verified = DAVerifier._verify_column(
|
||||||
blob.column, blob.aggregated_column_commitment, blob.aggregated_column_proof, blob.index
|
blob.column, blob.aggregated_column_commitment, blob.aggregated_column_proof, blob.index
|
||||||
)
|
)
|
||||||
if not is_column_verified:
|
if not is_column_verified:
|
||||||
return
|
return
|
||||||
chunks = batched(blob.column, BYTES_PER_CHUNK)
|
|
||||||
are_chunks_verified = DAVerifier._verify_chunks(
|
are_chunks_verified = DAVerifier._verify_chunks(
|
||||||
chunks, blob.rows_commitments, blob.rows_proofs
|
blob.column, blob.rows_commitments, blob.rows_proofs, blob.index
|
||||||
)
|
)
|
||||||
if not are_chunks_verified:
|
if not are_chunks_verified:
|
||||||
return
|
return
|
||||||
return self._build_attestation(blob)
|
return DAVerifier._build_attestation(blob)
|
||||||
|
|
Loading…
Reference in New Issue