Initial commit

venv/Lib/site-packages/passlib/crypto/scrypt/_gen_files.py

@@ -0,0 +1,154 @@
+"""passlib.utils.scrypt._gen_files - meta script that generates _salsa.py"""
+#==========================================================================
+# imports
+#==========================================================================
+# core
+import os
+# pkg
+# local
+#==========================================================================
+# constants
+#==========================================================================
+
+_SALSA_OPS = [
+        # row = (target idx, source idx 1, source idx 2, rotate)
+        # interpreted as salsa operation over uint32...
+        #   target = (source1+source2)<<rotate
+
+        ##/* Operate on columns. */
+        ##define R(a,b) (((a) << (b)) | ((a) >> (32 - (b))))
+        ##x[ 4] ^= R(x[ 0]+x[12], 7);  x[ 8] ^= R(x[ 4]+x[ 0], 9);
+        ##x[12] ^= R(x[ 8]+x[ 4],13);  x[ 0] ^= R(x[12]+x[ 8],18);
+        (  4,  0, 12,  7),
+        (  8,  4,  0,  9),
+        ( 12,  8,  4, 13),
+        (  0, 12,  8, 18),
+
+        ##x[ 9] ^= R(x[ 5]+x[ 1], 7);  x[13] ^= R(x[ 9]+x[ 5], 9);
+        ##x[ 1] ^= R(x[13]+x[ 9],13);  x[ 5] ^= R(x[ 1]+x[13],18);
+        (  9,  5,  1,  7),
+        ( 13,  9,  5,  9),
+        (  1, 13,  9, 13),
+        (  5,  1, 13, 18),
+
+        ##x[14] ^= R(x[10]+x[ 6], 7);  x[ 2] ^= R(x[14]+x[10], 9);
+        ##x[ 6] ^= R(x[ 2]+x[14],13);  x[10] ^= R(x[ 6]+x[ 2],18);
+        ( 14, 10,  6,  7),
+        (  2, 14, 10,  9),
+        (  6,  2, 14, 13),
+        ( 10,  6,  2, 18),
+
+        ##x[ 3] ^= R(x[15]+x[11], 7);  x[ 7] ^= R(x[ 3]+x[15], 9);
+        ##x[11] ^= R(x[ 7]+x[ 3],13);  x[15] ^= R(x[11]+x[ 7],18);
+        (  3, 15, 11,  7),
+        (  7,  3, 15,  9),
+        ( 11,  7,  3, 13),
+        ( 15, 11,  7, 18),
+
+        ##/* Operate on rows. */
+        ##x[ 1] ^= R(x[ 0]+x[ 3], 7);  x[ 2] ^= R(x[ 1]+x[ 0], 9);
+        ##x[ 3] ^= R(x[ 2]+x[ 1],13);  x[ 0] ^= R(x[ 3]+x[ 2],18);
+        (  1,  0,  3,  7),
+        (  2,  1,  0,  9),
+        (  3,  2,  1, 13),
+        (  0,  3,  2, 18),
+
+        ##x[ 6] ^= R(x[ 5]+x[ 4], 7);  x[ 7] ^= R(x[ 6]+x[ 5], 9);
+        ##x[ 4] ^= R(x[ 7]+x[ 6],13);  x[ 5] ^= R(x[ 4]+x[ 7],18);
+        (  6,  5,  4,  7),
+        (  7,  6,  5,  9),
+        (  4,  7,  6, 13),
+        (  5,  4,  7, 18),
+
+        ##x[11] ^= R(x[10]+x[ 9], 7);  x[ 8] ^= R(x[11]+x[10], 9);
+        ##x[ 9] ^= R(x[ 8]+x[11],13);  x[10] ^= R(x[ 9]+x[ 8],18);
+        ( 11, 10,  9,  7),
+        (  8, 11, 10,  9),
+        (  9,  8, 11, 13),
+        ( 10,  9,  8, 18),
+
+        ##x[12] ^= R(x[15]+x[14], 7);  x[13] ^= R(x[12]+x[15], 9);
+        ##x[14] ^= R(x[13]+x[12],13);  x[15] ^= R(x[14]+x[13],18);
+        ( 12, 15, 14,  7),
+        ( 13, 12, 15,  9),
+        ( 14, 13, 12, 13),
+        ( 15, 14, 13, 18),
+]
+
+def main():
+    target = os.path.join(os.path.dirname(__file__), "_salsa.py")
+    fh = file(target, "w")
+    write = fh.write
+
+    VNAMES = ["v%d" % i for i in range(16)]
+
+    PAD = " " * 4
+    PAD2 = " " * 8
+    PAD3 = " " * 12
+    TLIST = ", ".join("b%d" % i for i in range(16))
+    VLIST = ", ".join(VNAMES)
+    kwds = dict(
+        VLIST=VLIST,
+        TLIST=TLIST,
+    )
+
+    write('''\
+"""passlib.utils.scrypt._salsa - salsa 20/8 core, autogenerated by _gen_salsa.py"""
+#=================================================================
+# salsa function
+#=================================================================
+
+def salsa20(input):
+    \"""apply the salsa20/8 core to the provided input
+
+    :args input: input list containing 16 32-bit integers
+    :returns: result list containing 16 32-bit integers
+    \"""
+
+    %(TLIST)s = input
+    %(VLIST)s = \\
+        %(TLIST)s
+
+    i = 0
+    while i < 4:
+''' % kwds)
+
+    for idx, (target, source1, source2, rotate) in enumerate(_SALSA_OPS):
+        write('''\
+        # salsa op %(idx)d: [%(it)d] ^= ([%(is1)d]+[%(is2)d])<<<%(rot1)d
+        t = (%(src1)s + %(src2)s) & 0xffffffff
+        %(dst)s ^= ((t & 0x%(rmask)08x) << %(rot1)d) | (t >> %(rot2)d)
+
+''' % dict(
+        idx=idx, is1 = source1, is2=source2, it=target,
+        src1=VNAMES[source1],
+        src2=VNAMES[source2],
+        dst=VNAMES[target],
+        rmask=(1<<(32-rotate))-1,
+        rot1=rotate,
+        rot2=32-rotate,
+    ))
+
+    write('''\
+        i += 1
+
+''')
+
+    for idx in range(16):
+        write(PAD + "b%d = (b%d + v%d) & 0xffffffff\n" % (idx,idx,idx))
+
+    write('''\
+
+    return %(TLIST)s
+
+#=================================================================
+# eof
+#=================================================================
+''' % kwds)
+        
+if __name__ == "__main__":
+    main()
+
+#==========================================================================
+# eof
+#==========================================================================