summaryrefslogtreecommitdiffhomepage
path: root/libtomcrypt/notes/rsa-testvectors/rt.py
diff options
context:
space:
mode:
authorMatt Johnston <matt@ucc.asn.au>2018-02-09 21:44:05 +0800
committerMatt Johnston <matt@ucc.asn.au>2018-02-09 21:44:05 +0800
commit4f2eb1914bdac3ed3ee504ad86061281dbe0d074 (patch)
tree078293375c3f3ee2d485cf9559a08d65d460786a /libtomcrypt/notes/rsa-testvectors/rt.py
parentd72f50ff3284e15124a0f233c26339229fe305ac (diff)
Update to libtomcrypt 1.18.1, merged with Dropbear changes
Diffstat (limited to 'libtomcrypt/notes/rsa-testvectors/rt.py')
-rwxr-xr-xlibtomcrypt/notes/rsa-testvectors/rt.py253
1 files changed, 253 insertions, 0 deletions
diff --git a/libtomcrypt/notes/rsa-testvectors/rt.py b/libtomcrypt/notes/rsa-testvectors/rt.py
new file mode 100755
index 0000000..3f6ab7f
--- /dev/null
+++ b/libtomcrypt/notes/rsa-testvectors/rt.py
@@ -0,0 +1,253 @@
+#!/usr/bin/env python
+
+import sys
+import os
+import hashlib
+
+def md5_for_file(path, block_size=256*128):
+ '''
+ Block size directly depends on the block size of your filesystem
+ to avoid performances issues
+ Here I have blocks of 4096 octets (Default NTFS)
+ '''
+ md5 = hashlib.md5()
+ with open(path,'rb') as f:
+ for chunk in iter(lambda: f.read(block_size), b''):
+ md5.update(chunk)
+ f.close()
+ return md5.hexdigest()
+
+def read_until_ends(f, s):
+ while True:
+ l = f.readline()
+ if l.strip().endswith(s):
+ break
+ return l
+
+def read_until_start(f, s):
+ while True:
+ l = f.readline()
+ if l.startswith(s):
+ break
+ return l
+
+def read_hex(f):
+ t = []
+ while True:
+ l = f.readline()
+ if l.strip() == '':
+ break
+ t.extend(l.strip().split(' '))
+ return t
+
+class NamedData(object):
+ def __init__(self, name, data):
+ self.name = name
+ self.data = data
+
+ def __str__(self):
+ return " /* {0} */\n {1},\n {{ {2} }}\n".format(self.name, len(self.data), ', '.join('0x' + x for x in self.data))
+
+def read_part(f, s):
+ name = read_until_start(f, s).strip().lstrip('# ').rstrip(':')
+ data = read_hex(f)
+ e = NamedData(name, data)
+ return e
+
+class RsaKey(object):
+ def __init__(self, n, e, d, p, q, dP, dQ, qInv):
+ self.n = n
+ self.e = e
+ self.d = d
+ self.p = p
+ self.q = q
+ self.dP = dP
+ self.dQ = dQ
+ self.qInv = qInv
+
+ def __str__(self):
+ return "{{\n{0},\n{1},\n{2},\n{3},\n{4},\n{5},\n{6},\n{7}\n}}\n".format(self.n, self.e, self.d, self.p, self.q, self.dP, self.dQ, self.qInv)
+
+def read_key(f):
+ if ftype.version == 1:
+ read_until_start(f, '# Private key')
+ n = read_part(f, ftype.n)
+ e = read_part(f, ftype.e)
+ d = read_part(f, ftype.d)
+ p = read_part(f, ftype.p)
+ q = read_part(f, ftype.q)
+ dP = read_part(f, ftype.dP)
+ dQ = read_part(f, ftype.dQ)
+ qInv = read_part(f, ftype.qInv)
+ k = RsaKey(n, e, d, p, q, dP, dQ, qInv)
+ return k
+
+class Data(object):
+ def __init__(self, name, obj1, obj2, obj3):
+ self.name = name
+ self.obj1 = obj1
+ self.obj2 = obj2
+ self.obj3 = obj3
+
+ def __str__(self):
+ if self.obj3 == None:
+ return "{{\n \"{0}\",\n{1},\n{2}\n}}\n,".format(self.name, self.obj1, self.obj2)
+ else:
+ return "{{\n \"{0}\",\n{1},\n{2},\n{3}\n}}\n,".format(self.name, self.obj1, self.obj2, self.obj3)
+
+def read_data(f):
+ name = read_until_start(f, ftype.o).strip().lstrip('# ')
+ obj1 = read_part(f, ftype.o1)
+ obj2 = read_part(f, ftype.o2)
+ if ftype.name == 'emsa':
+ obj3 = None
+ else:
+ obj3 = read_part(f, ftype.o3)
+ s = Data(name, obj1, obj2, obj3)
+ return s
+
+class Example(object):
+ def __init__(self, name, key, data):
+ self.name = name
+ self.key = key
+ self.data = data
+
+ def __str__(self):
+ res = "{{\n \"{0}\",\n{1},\n{{".format(self.name, str(self.key))
+ for idx, d in enumerate(self.data, 1):
+ if idx == 2:
+ res += '#ifdef LTC_TEST_EXT\n'
+ res += str(d) + '\n'
+ if idx == ftype.numcases:
+ res += '#endif /* LTC_TEST_EXT */\n'
+ res += '}\n},'
+ return res
+
+def read_example(f):
+ name = read_until_start(f, '# Example').strip().lstrip('# ')
+ key = read_key(f)
+ l = read_until_start(f, ftype.sod)
+ d = []
+ while l.strip().startswith(ftype.sod):
+ if ftype.version == 1:
+ f.seek(-len(l), os.SEEK_CUR)
+ data = read_data(f)
+ d.append(data)
+ l = read_until_start(f, '#')
+
+ e = Example(name, key, d)
+ f.seek(-len(l), os.SEEK_CUR)
+ return e
+
+
+class PkcsType(object):
+ def __init__(self, name):
+ if name == 'pss':
+ self.o = '# RSASSA-PSS Signature Example'
+ self.o1 = '# Message to be signed'
+ self.o2 = '# Salt'
+ self.o3 = '# Signature'
+ elif name == 'oaep':
+ self.o = '# RSAES-OAEP Encryption Example'
+ self.o1 = '# Message to be encrypted'
+ self.o2 = '# Seed'
+ self.o3 = '# Encryption'
+ elif name == 'emsa':
+ self.o = '# PKCS#1 v1.5 Signature Example'
+ self.o1 = '# Message to be signed'
+ self.o2 = '# Signature'
+ elif name == 'eme':
+ self.o = '# PKCS#1 v1.5 Encryption Example'
+ self.o1 = '# Message'
+ self.o2 = '# Seed'
+ self.o3 = '# Encryption'
+ else:
+ raise ValueError('Type unknown: ' + name)
+
+ if name == 'pss' or name == 'oaep':
+ self.version = 2
+ self.numcases = 6
+ self.n = '# RSA modulus n'
+ self.e = '# RSA public exponent e'
+ self.d = '# RSA private exponent d'
+ self.p = '# Prime p'
+ self.q = '# Prime q'
+ self.dP = '# p\'s CRT exponent dP'
+ self.dQ = '# q\'s CRT exponent dQ'
+ self.qInv = '# CRT coefficient qInv'
+ self.sod = '# --------------------------------'
+ elif name == 'emsa' or name == 'eme':
+ self.version = 1
+ self.numcases = 20
+ self.n = '# Modulus'
+ self.e = '# Public exponent'
+ self.d = '# Exponent'
+ self.p = '# Prime 1'
+ self.q = '# Prime 2'
+ self.dP = '# Prime exponent 1'
+ self.dQ = '# Prime exponent 2'
+ self.qInv = '# Coefficient'
+ self.sod = self.o
+ self.name = name
+
+ftype = PkcsType(sys.argv[2])
+
+print('/* Generated from file: %s\n * with md5 hash: %s\n */\n' % (sys.argv[1], md5_for_file(sys.argv[1])))
+print('''
+typedef struct rsaKey {
+ int n_l;
+ unsigned char n[256];
+ int e_l;
+ unsigned char e[256];
+ int d_l;
+ unsigned char d[256];
+ int p_l;
+ unsigned char p[256];
+ int q_l;
+ unsigned char q[256];
+ int dP_l;
+ unsigned char dP[256];
+ int dQ_l;
+ unsigned char dQ[256];
+ int qInv_l;
+ unsigned char qInv[256];
+} rsaKey_t;
+
+typedef struct rsaData {
+ const char* name;
+ int o1_l;
+ unsigned char o1[256];
+ int o2_l;
+ unsigned char o2[256];''')
+
+if ftype.name != 'emsa':
+ print(''' int o3_l;
+ unsigned char o3[256];''')
+
+print('''} rsaData_t;
+
+typedef struct testcase {
+ const char* name;
+ rsaKey_t rsa;
+#ifdef LTC_TEST_EXT
+ rsaData_t data[%d];
+#else
+ rsaData_t data[1];
+#endif /* LTC_TEST_EXT */
+} testcase_t;
+
+testcase_t testcases_%s[] =
+ {''' % (ftype.numcases, sys.argv[2]))
+
+with open(sys.argv[1], 'rb') as f:
+ ex = []
+ while read_until_ends(f, '============================================='):
+ if f.tell() == os.path.getsize(sys.argv[1]):
+ break
+ e = read_example(f)
+ ex.append(e)
+
+ for i in ex:
+ print(i)
+f.close()
+print('};\n')