Initial check-in of catacomb-python.

[catacomb-python] / algorithms.py

## -*-python-*-

def cross(*seq):
  if not len(seq):
    return [(),]
  x = seq[0]
  if type(x) is not tuple and type(x) is not list:
    x = x,
  r = []
  for i in x:
    for j in cross(*seq[1:]):
      r.append((i,) + j)
  return r

prps = '''
des desx des3 mars
idea safer safersk
blowfish twofish
tea xtea
rc2 rc5
skipjack
cast128 cast256
square rijndael rijndael192 rijndael256
serpent noekeon
'''.split()
pmodes = '''
ecb cbc cfb ofb counter
'''.split()
streamciphers = '''
rc4 seal
'''.split()
hashes = '''
md2 md4 md5 tiger has160
sha sha224 sha256 sha384 sha512
rmd128 rmd160 rmd256 rmd320
whirlpool whirlpool256
'''.split()
hmodes = '''
mgf hmac
'''.split()

print '/* algorithms.h [generated] */'
print

for i in prps:
  print '#include <catacomb/%s.h>' % i
  for j in pmodes:
    print '#include <catacomb/%s-%s.h>' % (i, j)
for i in streamciphers:
  print '#include <catacomb/%s.h>' % i  
print
for i in hashes:
  print '#include <catacomb/%s.h>' % i
  for j in hmodes:
    print '#include <catacomb/%s-%s.h>' % (i, j)
print

print '#define PRPS(DO) \\'
for i in prps:
  print '  DO(%s, %s) \\' % (i.upper(), i)
print '  /* end */'
print

print '#define RNGS(DO) \\'
for i in (cross(prps, ['ofb', 'counter']) +
          cross(hashes, 'mgf')):
  print '  DO("%(prim)s-%(mode)s", %(prim)s_%(mode)srand) \\' % \
        {'prim': i[0], 'mode': i[1]}
print '  DO("rc4", rc4_rand) \\'
print '  DO("seal", seal_randkludge) \\'
print '  /* end */'
print