added files for vdf proofing
This commit is contained in:
parent
9369a22841
commit
acabeab7c4
45
src/onionrproofs/vdf/__init__.py
Normal file
45
src/onionrproofs/vdf/__init__.py
Normal file
@ -0,0 +1,45 @@
|
||||
"""Onionr - Private P2P Communication.
|
||||
|
||||
verifiable delay function proof
|
||||
"""
|
||||
from multiprocessing import Process
|
||||
from multiprocessing import Pipe
|
||||
|
||||
from mimcvdf import vdf_create
|
||||
from mimcvdf import vdf_verify
|
||||
"""
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.c
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
"""
|
||||
ROUND_MULTIPLIER_PER_BYTE = 100
|
||||
|
||||
|
||||
def create(data: bytes) -> str:
|
||||
rounds = len(data) * ROUND_MULTIPLIER_PER_BYTE
|
||||
return vdf_create(data, rounds)
|
||||
|
||||
|
||||
def multiproces_create(data: bytes) -> str:
|
||||
parent_conn, child_conn = Pipe()
|
||||
def __do_create(conn, data):
|
||||
conn.send(create(data))
|
||||
conn.close()
|
||||
p = Process(target=__do_create, args=(child_conn, data))
|
||||
p.start()
|
||||
return parent_conn.recv()
|
||||
|
||||
|
||||
def verify(data: bytes, block_id: str) -> bool:
|
||||
rounds = len(data) * ROUND_MULTIPLIER_PER_BYTE
|
||||
return vdf_verify(data, block_id, rounds)
|
||||
|
0
src/onionrproofs/vdf/hasher.py
Normal file
0
src/onionrproofs/vdf/hasher.py
Normal file
0
src/onionrproofs/vdf/verify.py
Normal file
0
src/onionrproofs/vdf/verify.py
Normal file
36
tests/test_block_vdf_proof.py
Normal file
36
tests/test_block_vdf_proof.py
Normal file
@ -0,0 +1,36 @@
|
||||
#!/usr/bin/env python3
|
||||
import sys, os
|
||||
sys.path.append(".")
|
||||
sys.path.append("src/")
|
||||
import uuid
|
||||
TEST_DIR = 'testdata/%s-%s' % (uuid.uuid4(), os.path.basename(__file__)) + '/'
|
||||
print("Test directory:", TEST_DIR)
|
||||
os.environ["ONIONR_HOME"] = TEST_DIR
|
||||
import unittest, json
|
||||
|
||||
from utils import identifyhome, createdirs
|
||||
from onionrsetup import setup_config
|
||||
from onionrproofs import vdf
|
||||
from time import time
|
||||
|
||||
createdirs.create_dirs()
|
||||
setup_config()
|
||||
|
||||
class TestVdf(unittest.TestCase):
|
||||
def test_vdf(self):
|
||||
res = vdf.create(b'test')
|
||||
int(res, 16)
|
||||
if len(res) == 0: raise ValueError
|
||||
self.assertEqual(vdf.multiproces_create(b'test'), res)
|
||||
def test_speed(self):
|
||||
t = time()
|
||||
vdf.create(b'test')
|
||||
self.assertTrue(time() - t <= 10)
|
||||
# test 2 kb
|
||||
t = time()
|
||||
vdf.create(b't'*10000)
|
||||
self.assertTrue(time() - t >= 10)
|
||||
#timeit(lambda: vdf.create(b'test'))
|
||||
|
||||
|
||||
unittest.main()
|
Loading…
Reference in New Issue
Block a user