mirror of
https://github.com/markqvist/Reticulum.git
synced 2024-12-26 07:59:36 -05:00
Updated tests
This commit is contained in:
Mark Qvist
parent
534a8825eb
commit
a8ea7bcca6
@ -27,7 +27,125 @@ class TestIdentity(unittest.TestCase):
|
|||||||
self.assertEqual(i.hash, bytes.fromhex(id_hash))
|
self.assertEqual(i.hash, bytes.fromhex(id_hash))
|
||||||
self.assertEqual(i.get_private_key(), bytes.fromhex(key))
|
self.assertEqual(i.get_private_key(), bytes.fromhex(key))
|
||||||
|
|
||||||
def test_1_encrypt(self):
|
def test_1_sign(self):
|
||||||
|
print("")
|
||||||
|
|
||||||
|
# Test known signature
|
||||||
|
fid = RNS.Identity.from_bytes(bytes.fromhex(fixed_keys[0][0]))
|
||||||
|
sig = fid.sign(signed_message.encode("utf-8"))
|
||||||
|
|
||||||
|
self.assertEqual(sig, bytes.fromhex(sig_from_key_0))
|
||||||
|
|
||||||
|
# Test signature time jitter
|
||||||
|
id1 = RNS.Identity()
|
||||||
|
id2 = RNS.Identity(create_keys=False)
|
||||||
|
id2.load_public_key(id1.get_public_key())
|
||||||
|
|
||||||
|
if RNS.Cryptography.backend() == "internal":
|
||||||
|
rounds = 2000
|
||||||
|
else:
|
||||||
|
rounds = 20000
|
||||||
|
|
||||||
|
times = []
|
||||||
|
for i in range(1, rounds):
|
||||||
|
msg = os.urandom(512)
|
||||||
|
start = time.time()
|
||||||
|
signature = id1.sign(msg)
|
||||||
|
t = time.time() - start
|
||||||
|
times.append(t)
|
||||||
|
|
||||||
|
import statistics
|
||||||
|
tmin = min(times)*1000
|
||||||
|
tmax = max(times)*1000
|
||||||
|
tmean = (sum(times)/len(times))*1000
|
||||||
|
tmed = statistics.median(times)*1000
|
||||||
|
tmdev = tmax - tmin
|
||||||
|
mpct = (tmax/tmed)*100
|
||||||
|
print("Random messages:")
|
||||||
|
print(" Signature timing min/avg/med/max/mdev: "+str(round(tmin, 3))+"/"+str(round(tmean, 3))+"/"+str(round(tmed, 3))+"/"+str(round(tmax, 3))+"/"+str(round(tmdev, 3)))
|
||||||
|
print(" Max deviation from median: "+str(round(mpct, 1))+"%")
|
||||||
|
print()
|
||||||
|
|
||||||
|
id1 = RNS.Identity()
|
||||||
|
id2 = RNS.Identity(create_keys=False)
|
||||||
|
id2.load_public_key(id1.get_public_key())
|
||||||
|
|
||||||
|
times = []
|
||||||
|
for i in range(1, rounds):
|
||||||
|
msg = bytes([0x00])*512
|
||||||
|
start = time.time()
|
||||||
|
signature = id1.sign(msg)
|
||||||
|
t = time.time() - start
|
||||||
|
times.append(t)
|
||||||
|
|
||||||
|
tmin = min(times)*1000
|
||||||
|
tmax = max(times)*1000
|
||||||
|
tmean = (sum(times)/len(times))*1000
|
||||||
|
tmed = statistics.median(times)*1000
|
||||||
|
tmdev = tmax - tmin
|
||||||
|
mpct = (tmax/tmed)*100
|
||||||
|
print("All 0xff messages:")
|
||||||
|
print(" Signature timing min/avg/med/max/mdev: "+str(round(tmin, 3))+"/"+str(round(tmean, 3))+"/"+str(round(tmed, 3))+"/"+str(round(tmax, 3))+"/"+str(round(tmdev, 3)))
|
||||||
|
print(" Max deviation from median: "+str(round(mpct, 1))+"%")
|
||||||
|
print()
|
||||||
|
|
||||||
|
id1 = RNS.Identity()
|
||||||
|
id2 = RNS.Identity(create_keys=False)
|
||||||
|
id2.load_public_key(id1.get_public_key())
|
||||||
|
|
||||||
|
times = []
|
||||||
|
for i in range(1, rounds):
|
||||||
|
msg = bytes([0xff])*512
|
||||||
|
start = time.time()
|
||||||
|
signature = id1.sign(msg)
|
||||||
|
t = time.time() - start
|
||||||
|
times.append(t)
|
||||||
|
|
||||||
|
tmin = min(times)*1000
|
||||||
|
tmax = max(times)*1000
|
||||||
|
tmean = (sum(times)/len(times))*1000
|
||||||
|
tmed = statistics.median(times)*1000
|
||||||
|
tmdev = tmax - tmin
|
||||||
|
mpct = (tmax/tmed)*100
|
||||||
|
print("All 0x00 messages:")
|
||||||
|
print(" Signature timing min/avg/med/max/mdev: "+str(round(tmin, 3))+"/"+str(round(tmean, 3))+"/"+str(round(tmed, 3))+"/"+str(round(tmax, 3))+"/"+str(round(tmdev, 3)))
|
||||||
|
print(" Max deviation from median: "+str(round(mpct, 1))+"%")
|
||||||
|
print()
|
||||||
|
|
||||||
|
b = 0
|
||||||
|
t = 0
|
||||||
|
for i in range(1, 500):
|
||||||
|
mlen = i % (RNS.Reticulum.MTU//2) + (RNS.Reticulum.MTU//2)
|
||||||
|
msg = os.urandom(mlen)
|
||||||
|
b += mlen
|
||||||
|
id1 = RNS.Identity()
|
||||||
|
id2 = RNS.Identity(create_keys=False)
|
||||||
|
id2.load_public_key(id1.get_public_key())
|
||||||
|
|
||||||
|
start = time.time()
|
||||||
|
signature = id1.sign(msg)
|
||||||
|
self.assertEqual(True, id2.validate(signature, msg))
|
||||||
|
t += time.time() - start
|
||||||
|
|
||||||
|
print("Sign/validate chunks < MTU: "+self.size_str(b/t, "b")+"ps")
|
||||||
|
|
||||||
|
for i in range(1, 500):
|
||||||
|
mlen = 16*1024
|
||||||
|
msg = os.urandom(mlen)
|
||||||
|
b += mlen
|
||||||
|
id1 = RNS.Identity()
|
||||||
|
id2 = RNS.Identity(create_keys=False)
|
||||||
|
id2.load_public_key(id1.get_public_key())
|
||||||
|
|
||||||
|
start = time.time()
|
||||||
|
signature = id1.sign(msg)
|
||||||
|
self.assertEqual(True, id2.validate(signature, msg))
|
||||||
|
t += time.time() - start
|
||||||
|
|
||||||
|
print("Sign/validate 16KB chunks: "+self.size_str(b/t, "b")+"ps")
|
||||||
|
|
||||||
|
|
||||||
|
def test_2_encrypt(self):
|
||||||
print("")
|
print("")
|
||||||
|
|
||||||
# Test decryption of known token
|
# Test decryption of known token
|
||||||
@ -83,48 +201,6 @@ class TestIdentity(unittest.TestCase):
|
|||||||
print("Encrypt "+self.size_str(mlen)+" chunks: "+self.size_str(b/e_t, "b")+"ps")
|
print("Encrypt "+self.size_str(mlen)+" chunks: "+self.size_str(b/e_t, "b")+"ps")
|
||||||
print("Decrypt "+self.size_str(mlen)+" chunks: "+self.size_str(b/d_t, "b")+"ps")
|
print("Decrypt "+self.size_str(mlen)+" chunks: "+self.size_str(b/d_t, "b")+"ps")
|
||||||
|
|
||||||
def test_2_sign(self):
|
|
||||||
print("")
|
|
||||||
|
|
||||||
# Test known signature
|
|
||||||
fid = RNS.Identity.from_bytes(bytes.fromhex(fixed_keys[0][0]))
|
|
||||||
sig = fid.sign(signed_message.encode("utf-8"))
|
|
||||||
|
|
||||||
self.assertEqual(sig, bytes.fromhex(sig_from_key_0))
|
|
||||||
|
|
||||||
b = 0
|
|
||||||
t = 0
|
|
||||||
for i in range(1, 500):
|
|
||||||
mlen = i % (RNS.Reticulum.MTU//2) + (RNS.Reticulum.MTU//2)
|
|
||||||
msg = os.urandom(mlen)
|
|
||||||
b += mlen
|
|
||||||
id1 = RNS.Identity()
|
|
||||||
id2 = RNS.Identity(create_keys=False)
|
|
||||||
id2.load_public_key(id1.get_public_key())
|
|
||||||
|
|
||||||
start = time.time()
|
|
||||||
signature = id1.sign(msg)
|
|
||||||
self.assertEqual(True, id2.validate(signature, msg))
|
|
||||||
t += time.time() - start
|
|
||||||
|
|
||||||
print("Sign/validate chunks < MTU: "+self.size_str(b/t, "b")+"ps")
|
|
||||||
|
|
||||||
for i in range(1, 500):
|
|
||||||
mlen = 16*1024
|
|
||||||
msg = os.urandom(mlen)
|
|
||||||
b += mlen
|
|
||||||
id1 = RNS.Identity()
|
|
||||||
id2 = RNS.Identity(create_keys=False)
|
|
||||||
id2.load_public_key(id1.get_public_key())
|
|
||||||
|
|
||||||
start = time.time()
|
|
||||||
signature = id1.sign(msg)
|
|
||||||
self.assertEqual(True, id2.validate(signature, msg))
|
|
||||||
t += time.time() - start
|
|
||||||
|
|
||||||
print("Sign/validate 16KB chunks: "+self.size_str(b/t, "b")+"ps")
|
|
||||||
|
|
||||||
|
|
||||||
def size_str(self, num, suffix='B'):
|
def size_str(self, num, suffix='B'):
|
||||||
units = ['','K','M','G','T','P','E','Z']
|
units = ['','K','M','G','T','P','E','Z']
|
||||||
last_unit = 'Y'
|
last_unit = 'Y'
|
||||||
|
|||||||
@ -19,11 +19,11 @@ fixed_keys = [
|
|||||||
|
|
||||||
def targets_job(caller):
|
def targets_job(caller):
|
||||||
cmd = "python -c \"from tests.link import targets; targets()\""
|
cmd = "python -c \"from tests.link import targets; targets()\""
|
||||||
print("Opening subprocess for "+str(caller)+"...", RNS.LOG_VERBOSE)
|
print("Opening subprocess for "+str(cmd)+"...", RNS.LOG_VERBOSE)
|
||||||
ppath = os.getcwd()
|
ppath = os.getcwd()
|
||||||
|
|
||||||
try:
|
try:
|
||||||
caller.process = subprocess.Popen(shlex.split(cmd), cwd=ppath)
|
caller.process = subprocess.Popen(shlex.split(cmd), cwd=ppath, stdout=subprocess.PIPE)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise e
|
raise e
|
||||||
caller.pipe_is_open = False
|
caller.pipe_is_open = False
|
||||||
@ -39,10 +39,19 @@ def init_rns(caller=None):
|
|||||||
c_rns.m_proc = caller.process
|
c_rns.m_proc = caller.process
|
||||||
print("Done starting local RNS instance...")
|
print("Done starting local RNS instance...")
|
||||||
|
|
||||||
|
def close_rns():
|
||||||
|
global c_rns
|
||||||
|
if c_rns != None:
|
||||||
|
c_rns.m_proc.kill()
|
||||||
|
|
||||||
class TestLink(unittest.TestCase):
|
class TestLink(unittest.TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def tearDownClass(cls):
|
||||||
|
close_rns()
|
||||||
|
|
||||||
def test_0_establish(self):
|
def test_0_establish(self):
|
||||||
init_rns(self)
|
init_rns(self)
|
||||||
print("")
|
print("")
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user