TGC

project2/TableGenerator copy.py

@@ -0,0 +1,69 @@
+#
+# Project 2
+#
+# 01435 Practiacal Cryptanalysis
+# Technical University of Denmark
+
+# Markus Faerevaag              (s123692@student.dtu.dk)
+# Christian Mathias Rohde Kiaer (s123812@student.dtu.dk)
+# Jonathan Becktor              (s123094@student.dtu.dk)
+#
+
+import md5, random, csv, time
+
+BIT_SIZE   = 28
+NUM_CHAINS = 2**18
+CHAIN_LEN  = 2**10
+TABLE_NAME = "table.csv"
+LOG_FREQ   = 5000
+
+u = "daffeda"
+
+
+def md5_redux(s, i=0):
+    """Lowest 28 bits of (MD5(s||u) % i)"""    
+    digest = '0x' + md5.new(str(s) + str(u)).hexdigest()
+    result = hex((int(digest, 16) + i) % 2**BIT_SIZE)[:BIT_SIZE/4+2]
+    return result
+
+
+def generate_table():
+    """Generates a rainbow table"""
+    dict = {}
+
+    for i in xrange(0, NUM_CHAINS):
+        red = hex(random.getrandbits(28))[:-1]
+
+        red_start_point = red
+
+        for x in xrange(0, CHAIN_LEN):
+            red = md5_redux(red, x)
+
+        red_end_point = red
+
+        if i % LOG_FREQ == 0:
+            print "Took i calls: %d" % (i)
+        dict[red_start_point] = red_end_point
+
+    write_to_csv(dict)
+
+
+def write_to_csv(dict):
+    """Writes the rainbow table to a .csv file"""
+    w = csv.writer(open(TABLE_NAME, 'w'))
+    for key, value in dict.items():
+        w.writerow([key, value])
+
+
+def main():
+    start = time.clock()
+
+    generate_table()
+
+    end = time.clock()
+    print "Took: ", (end - start), " s"
+
+
+if __name__ == '__main__':
+    main()
+