added files for vdf proofing

src/onionrproofs/vdf/__init__.py

@@ -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)
+

tests/test_block_vdf_proof.py

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