def DSSE_add(username, add_token, enc_file):
rootdir = os.path.join(os.getcwd(), 'static/db/' + username)
with open(os.path.join(rootdir, 'TS.pickle'), 'r') as picklefile:
TS = pickle.load(picklefile)
with open(os.path.join(rootdir, 'TD.pickle'), 'r') as picklefile:
TD = pickle.load(picklefile)
with open(os.path.join(rootdir, 'AS.pickle'), 'r') as picklefile:
AS = pickle.load(picklefile)
with open(os.path.join(rootdir, 'AD.pickle'), 'r') as picklefile:
AD = pickle.load(picklefile)
filename = add_token[0]
del add_token[0]
filename_sha1 = myTool.CalcSha1(filename)
filename_md5 = int(myTool.CalcMD5(filename), 16)
TD[filename_sha1] = -1 ^ filename_md5
for word in add_token:
word_sha1 = myTool.CalcSha1(word)
word_md5 = int(myTool.CalcMD5(word), 16)
try:
head = TS[word_sha1]
except:
head = -1 ^ word_md5
eTS = TS[myTool.CalcSha1("#EMPTY#")] ^ int(myTool.CalcMD5("#EMPTY#"), 16)
eAsNext = AS[eTS][1]
AS[eTS] = [int(filename) ^ word_md5, head]
eTD = TD[myTool.CalcSha1("#EMPTY#")] ^ int(myTool.CalcMD5("#EMPTY#"), 16)
eAdNext = AD[eTD][1]
AD[eTD] = [myTool.sxor(word, myTool.CalcMD5(filename)), TD[filename_sha1]]
# AD[TD[myTool.CalcSha1("#EMPTY#")]] =
# [TS[myTool.CalcSha1("#EMPTY#")], TD[myTool.CalcSha1(filename)]]
TS[word_sha1] = eTS ^ word_md5
TS[myTool.CalcSha1("#EMPTY#")] = eAsNext
TD[filename_sha1] = eTD ^ filename_md5
TD[myTool.CalcSha1("#EMPTY#")] = eAdNext
with open(os.path.join(rootdir, 'TS.pickle'), 'w+') as outputfile:
pickle.dump(TS, outputfile)
with open(os.path.join(rootdir, 'TD.pickle'), 'w+') as outputfile:
pickle.dump(TD, outputfile)
with open(os.path.join(rootdir, 'AS.pickle'), 'w+') as outputfile:
pickle.dump(AS, outputfile)
with open(os.path.join(rootdir, 'AD.pickle'), 'w+') as outputfile:
pickle.dump(AD, outputfile)
with open(os.path.join(rootdir, enc_file['filename']), 'w+') as outputfile:
outputfile.write(enc_file['body'])
def DSSE_del(username, filename):
rootdir = os.path.join(os.getcwd(), 'static/db/' + username)
with open(os.path.join(rootdir, 'TS.pickle'), 'r') as picklefile:
TS = pickle.load(picklefile)
with open(os.path.join(rootdir, 'TD.pickle'), 'r') as picklefile:
TD = pickle.load(picklefile)
with open(os.path.join(rootdir, 'AS.pickle'), 'r') as picklefile:
AS = pickle.load(picklefile)
with open(os.path.join(rootdir, 'AD.pickle'), 'r') as picklefile:
AD = pickle.load(picklefile)
filename_sha1 = myTool.CalcSha1(filename)
filename_md5 = myTool.CalcMD5(filename)
filename_md5_hex = int(filename_md5, 16)
print 'TD[filename_sha1]: ', TD[filename_sha1] ^ filename_md5_hex
fHere = TD[filename_sha1] ^ filename_md5_hex
# del TD[myTool.CalcSha1(filename)]
print '-----------------DSSE_del------------------\n'
while fHere != -1:
fNext = AD[fHere][1] ^ filename_md5_hex
word = myTool.sxor(AD[fHere][0], filename_md5)
word_sha1 = myTool.CalcSha1(word)
word_md5 = int(myTool.CalcMD5(word), 16)
print 'AD[fHere][0]: ', AD[fHere][0]
print 'fNext: ', fNext
print 'word: ', word
print AS[TS[word_sha1] ^ word_md5][0] ^ word_md5
if myTool.CalcSha1(str(AS[TS[word_sha1] ^ word_md5][0] ^ word_md5)) == filename_sha1:
j = TS[word_sha1] ^ word_md5
TS[word_sha1] = AS[j][1]
else:
i = TS[word_sha1] ^ word_md5
j = AS[word_sha1][1] ^ word_md5
while AS[j][0] != filename ^ word_md5 and j != -1:
i = j
j = AS[j][1] ^ word_md5
if j != -1:
AS[i][1] = AS[j][1]
AS[j] = ["#EMPTY#", TS[myTool.CalcSha1("#EMPTY#")]]
TS[myTool.CalcSha1("#EMPTY#")] = j ^ int(myTool.CalcMD5("#EMPTY#"), 16)
AD[fHere] = ["#EMPTY#", TD[myTool.CalcSha1("#EMPTY#")]]
TD[myTool.CalcSha1("#EMPTY#")] = fHere ^ int(myTool.CalcMD5("#EMPTY#"), 16)
fHere = fNext
with open(os.path.join(rootdir, 'TS.pickle'), 'w+') as outputfile:
pickle.dump(TS, outputfile)
with open(os.path.join(rootdir, 'TD.pickle'), 'w+') as outputfile:
pickle.dump(TD, outputfile)
with open(os.path.join(rootdir, 'AS.pickle'), 'w+') as outputfile:
pickle.dump(AS, outputfile)
with open(os.path.join(rootdir, 'AD.pickle'), 'w+') as outputfile:
pickle.dump(AD, outputfile)
file_path = os.path.join(rootdir, filename + '.enc')
if os.path.isfile(file_path):
os.remove(file_path)