added files for vdf proofing

master
Kevin Froman 2020-03-22 19:16:26 -05:00
parent 9369a22841
commit acabeab7c4
4 changed files with 81 additions and 0 deletions

View 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)

View File

View File

View 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()