Geo-Encryption-Algorithm/app.py at master · KriAga/Geo-Encryption-Algorithm · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
import generateCMT as gCMT
import Encryption, Decryption
import generateCMT as gCMT
import json, os, filecmp, pickle, base64

gCMT.generateCodeMapTable(5, 100, 100, 1114112)
CMT = pickle.load(open("CMT.pickle","rb"))
# print(CMT)
# with open("Data.png", "rb") as image_file:
    # Data = str(base64.b64encode(image_file.read()))
    # print(type(base64.b64encode(image_file.read())))


# from array import array
# output_file = open('Encrypted', 'wb')
# float_array = array('d', encryptedData)
# float_array.tofile(output_file)
# output_file.close()
# with open("Encrypted.pickle","wb") as fe:
#     pickle.dump(encryptedData, fe)
# for i in range(100000):
def encryptDecryptTextFile(filename,CMT):
    Data = open(filename, "r").read()
    # print("Data: ", Data)
    encryptedData = Encryption.encryptString(Data, CMT)
    # print("encryptedData", encryptedData)

    EncryptedFile = open(filename+".encrypted","w+")
    EncryptedFile.write(encryptedData)
    EncryptedFile.close()

    EncryptedData = open(filename+".encrypted", "r").read()
    # print("EncryptedData", EncryptedData)
    decryptedData = Decryption.decryptString(EncryptedData, CMT)

    # print("decryptedData", decryptedData)


    # fh = open("Decrypted.png", "wb")
    # fh.write(decryptedData.decode('base64'))
    # fh.close()
    DecryptedFile = open(filename+".decrypted.txt","w+")
    # print(decryptedData)
    DecryptedFile.write(decryptedData)
    DecryptedFile.close()
    # print(bytearray(bytes(decryptedData, encoding='ascii'), encoding='ascii'))
    # decryptedData = decryptedData[2:-1]
    # decryptedDataBytes = bytes(decryptedData, encoding='ascii')
    # print(decryptedDataBytes)
    # with open("Decrypted.png", "wb") as fh:
        # fh.write(decryptedDataBytes)
    if filecmp.cmp(filename, filename+".decrypted.txt"):
        print("File Matched")
    else:
        print("File not Matched")
    if(Data != decryptedData):
        print("Jagteeeee rahooooooo")
    else:
        print("Yipeeeeeee Hurrraayyyyyy")
def encryptDecryptBinFile(filename,CMT):
    dataFile = open(filename,'rb')
    data = (dataFile.read())
    intArray = []
    for i in data:
        intArray.append(i)
    encryptedData = Encryption.encryptByteArray(intArray,CMT)
    EncryptedFile = open(filename+".encrypted","w+")
    EncryptedFile.write(encryptedData)
    EncryptedFile.close()
    # print("EncryptedData", EncryptedData)
    # print(encryptedData)
    decryptedData = Decryption.decryptToByteArray(encryptedData, CMT)
    print(len(decryptedData))
    decryptedFile  = open("decrypted."+filename,"wb")
    decryptedFile.write(bytes(decryptedData))
    # decryptedFile.close()
    # for i in range(len(decryptedData)):
    #     if(decryptedData[i] != intArray[i]):
    #         print(decryptedData[i] , intArray[i])
    #     decryptedFile.write(bytes([decryptedData[i]]))
    decryptedFile.close()

    # binDecryptedDataArray =[ bytes(i,encoding='utf-16') for i in decryptedData]
    # print(data[1],binDecryptedDataArray[1])
    # binDecryptedData = (b"".join(binDecryptedDataArray[:]))
    # print(binDecryptedData[:100])
    # for i in range(100):
    #     print(data[i],binDecryptedData[i])
    # decryptedFile  = open(filename+".decrypted.png","wb")
    # decryptedFile.write(binDecryptedData)
    # decryptedFile.close()


encryptDecryptBinFile("Data.mp3",CMT)