algorithms.py: Support SHA512/224 and SHA512/256.

[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()
latindances = '''
salsa20 salsa20/12 salsa20/8
salsa20-ietf salsa20/12-ietf salsa20/8-ietf
xsalsa20 xsalsa20/12 xsalsa20/8
chacha20 chacha12 chacha8
chacha20-ietf chacha12-ietf chacha8-ietf
xchacha20 xchacha12 xchacha8
'''.split()
streamciphers += map(lambda s: s.translate(None, '/'), latindances)
hashes = '''
md2 md4 md5 tiger has160
sha sha224 sha256 sha512/224 sha512/256 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.replace('/', '-')
  for j in pmodes:
    print '#include <catacomb/%s-%s.h>' % (i.replace('/', '-'), j)
for i in streamciphers:
  print '#include <catacomb/%s.h>' % i.replace('/', '-')
print
for i in hashes:
  print '#include <catacomb/%s.h>' % i.replace('/', '-')
  for j in hmodes:
    print '#include <catacomb/%s-%s.h>' % (i.replace('/', '-'), j)
print

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

print '#define RNGS(_) \\'
for i in (cross(prps, ['ofb', 'counter'])):
  print ('\t_("%(prim)s-%(mode)s", %(primid)s_keysz, ' +
         '%(primid)s_%(mode)srand, RNG_PLAIN, 0) \\') % \
         {'prim': i[0], 'mode': i[1],
          'primid': i[0].replace('-', '_').replace('/', '_')}
for i in (cross(hashes, 'mgf')):
  print ('\t_("%(prim)s-%(mode)s", %(primid)s_%(mode)skeysz, ' +
         '%(primid)s_%(mode)srand, RNG_PLAIN, 0) \\') % \
         {'prim': i[0], 'mode': i[1],
          'primid': i[0].replace('-', '_').replace('/', '_')}
print '\t_("rc4", rc4_keysz, rc4_rand, 0, 0) \\'
print '\t_("seal", seal_keysz, seal_rand, RNG_SEAL, 0) \\'
for i in latindances:
  for r in ['salsa20', 'xsalsa20', 'chacha', 'xchacha']:
    if i.startswith(r):
      root = r
      break
  else:
    raise ValueError, 'failed to find root name for %s' % i
  if i.endswith('-ietf'): root += '_ietf'
  print ('\t_("%(name)s", %(root)s_keysz, %(id)s_rand, ' +
         'RNG_LATIN, %(ROOT)s_NONCESZ) \\') % \
      {'name': i, 'id': i.translate(None, '/').replace('-', '_'),
       'root': root, 'ROOT': root.upper()}
print '\t/* end */'
print