pock: Use floor division on integers.

[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
cmac pmac1
ccm eax gcm ocb1 ocb3
'''.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.replace('/', ''), latindances)
hashes = '''
md2 md4 md5 tiger has160
sha sha224 sha256 sha512/224 sha512/256 sha384 sha512
rmd128 rmd160 rmd256 rmd320
whirlpool whirlpool256
sha3-224 sha3-256 sha3-384 sha3-512
'''.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.replace('/', '').replace('-', '_'),
         'root': root, 'ROOT': root.upper()})
for i in [128, 256]:
  print(('\t_("shake%(w)d", shake%(w)d_keysz, cshake%(w)d_rand, ' +
         'RNG_SHAKE, 0) \\') % \
        {'w': i})
  print(('\t_("kmac%(w)d", kmac%(w)d_keysz, kmac%(w)d_rand, ' +
         'RNG_KMAC, 0) \\') % \
        {'w': i})
print('\t/* end */')
print('')