50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
import time
|
|
|
|
from kasten import Kasten, generator
|
|
from kasten.types import KastenPacked, KastenChecksum
|
|
from kasten.exceptions import InvalidID
|
|
|
|
from mimcvdf import vdf_create, vdf_verify
|
|
|
|
from onionrblocks.universalrules import check_block_sanity
|
|
|
|
class AnonVDFGenerator(generator.KastenBaseGenerator):
|
|
@classmethod
|
|
def get_ttl_seconds_per_rounds(cls, rounds: int):
|
|
# 8000 rounds = 1 second (2.8ghz python) = 1 hour storage
|
|
if rounds < 8000:
|
|
raise ValueError(
|
|
"Rounds must be at least 8000")
|
|
return (rounds / 8000) * 60
|
|
|
|
@classmethod
|
|
def generate(
|
|
cls, packed_bytes: KastenPacked, rounds: int = 5000) -> Kasten:
|
|
return Kasten(
|
|
vdf_create(
|
|
packed_bytes,
|
|
rounds, dec=True
|
|
).to_bytes(
|
|
64, "big"), packed_bytes, cls, auto_check_generator=False)
|
|
|
|
@staticmethod
|
|
def validate_id(
|
|
hash: KastenChecksum,
|
|
packed_bytes: KastenPacked, rounds=5000) -> None:
|
|
|
|
try:
|
|
hash = int.from_bytes(hash, byteorder="big")
|
|
except TypeError:
|
|
pass
|
|
if not vdf_verify(packed_bytes, hash, rounds):
|
|
raise InvalidID
|
|
|
|
test_obj = Kasten(
|
|
None, packed_bytes, None, auto_check_generator=False)
|
|
allowed_age_seconds = AnonVDFGenerator.get_ttl_seconds_per_rounds(
|
|
rounds)
|
|
if time.time() > test_obj.get_timestamp() + allowed_age_seconds:
|
|
raise ValueError(
|
|
f"Block rounds only vaild through {allowed_age_seconds}")
|
|
|
|
return None |