diff --git a/notes/rsa-testvectors/rt.py b/notes/rsa-testvectors/rt.py new file mode 100755 index 0000000..2617837 --- /dev/null +++ b/notes/rsa-testvectors/rt.py @@ -0,0 +1,179 @@ +#!/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_eq(f, s): + while True: + l = f.readline() + if l.strip() == 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, q, p, dP, dQ, qInv): + self.n = n + self.e = e + self.d = d + self.q = q + self.p = p + 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.q, self.p, self.dP, self.dQ, self.qInv) + +def read_key(f): + n = read_part(f, '# RSA modulus n') + e = read_part(f, '# RSA public exponent e') + d = read_part(f, '# RSA private exponent d') + q = read_part(f, '# Prime p') + p = read_part(f, '# Prime q') + dP = read_part(f, '# p\'s CRT exponent dP') + dQ = read_part(f, '# q\'s CRT exponent dQ') + qInv = read_part(f, '# CRT coefficient qInv') + k = RsaKey(n, e, d, q, p, dP, dQ, qInv) + return k + +class Signature(object): + def __init__(self, name, msg, salt, sig): + self.name = name + self.msg = msg + self.salt = salt + self.sig = sig + + def __str__(self): + return "{{\n \"{0}\",\n{1},\n{2},\n{3}\n}}\n,".format(self.name, self.msg, self.salt, self.sig) + +def read_sig(f): + name = read_until_start(f, '# RSASSA-PSS Signature Example').strip().lstrip('# ') + msg = read_part(f, '# Message to be signed') + salt = read_part(f, '# Salt') + sig = read_part(f, '# Signature') + s = Signature(name, msg, salt, sig) + return s + +class Example(object): + def __init__(self, name, key, s): + self.name = name + self.key = key + self.s = s + + def __str__(self): + res = "{{/* {0} */\n{1},\n{{".format(self.name, str(self.key)) + for i in self.s: + res += str(i) + '\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, '#') + s = [] + while l.strip().startswith('# --------------------------------'): + sig = read_sig(f) + s.append(sig) + l = read_until_start(f, '#') + + e = Example(name, key, s) + f.seek(-len(l), os.SEEK_CUR) + return e + +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 rsaSig { + const char* name; + int msg_l; + unsigned char msg[256]; + int salt_l; + unsigned char salt[256]; + int sig_l; + unsigned char sig[256]; +} rsaSig_t; + +typedef struct testcase { + rsaKey_t rsa; + rsaSig_t sig[6]; +} testcase_t; + +testcase_t testcases[] = + {''') + +with open(sys.argv[1], 'rb') as f: + ex = [] + while read_until_eq(f, '# ============================================='): + if f.tell() == os.path.getsize(sys.argv[1]): + break + e = read_example(f) +# print e + ex.append(e) + + for i in ex: + print(i) +f.close() +print('};\n')