from pymd5 import md5, padding
import os

with open("prefix") as f:
    prefix = f.read().strip()

os.system("./fastcoll -p prefix -o col0 col1")
with open("col0") as f:
    col0 = f.read().strip()
with open("col1") as f:
    col1 = f.read().strip()

blobs = []
index = 0
chunk_size = 2
blob2old = ""
blob3old = ""

# while True:
#     if ('(' or ')' in col0) or ('(' or ')' in col1):
#         os.system("./fastcoll -p prefix -o col0 col1")

blobs.append(col0)
blobs.append(col1)


while True:
# for i in range(0, 5):
    os.system("./fastcoll -i " + md5(blobs[index]).hexdigest() + " -o blob2 blob3")

    with open("blob2") as f:
        blob2 = f.read().strip()
    with open("blob3") as f:
        blob3 = f.read().strip()

    if ('(' in blob2) or (')' in blob2) or ('(' in blob3) or (')' in blob3):
        if blob2 != blob2old:
            print "blob2 diff"
        blob2old = blob2
        continue
    # while True:
    #     os.system("./fastcoll -i " + md5(blobs[index]).hexdigest() + " -o blob2 blob3")
    #     with open("blob2") as f:
    #         blob2 = f.read().strip()
    #     with open("blob3") as f:
    #         blob3 = f.read().strip()
    #
    #     if '(' or ')' in blob2 or '(' or ')' in blob3:
    #         if blob2 != blob2old:
    #             print "blob2 diff"
    #         blob2old = blob2
    #         continue
    #     else:
    #         break

    for j in range(0, chunk_size):
        blobs.append(blobs[index+j] + padding(len(blobs[index+j])*8) + blob2)
        print md5(blobs[index+j] + padding(len(blobs[index+j])*8) + blob2).hexdigest()
        blobs.append(blobs[index+j] + padding(len(blobs[index+j])*8) + blob3)
        print md5(blobs[index+j] + padding(len(blobs[index+j])*8) + blob3).hexdigest()

    index += chunk_size
    chunk_size *=2
    if chunk_size == 64:
        break

index = len(blobs) - 64
for k in range(0,64):
    filename = "blobs/blob" + str(k)
    with open(filename, "w") as f:
        f.write(blobs[index+k])