OpenSoure_Repos/TheAlgorithms-Python
1
0
Fork 0
mirror of https://hub.njuu.cf/TheAlgorithms/Python.git synced 2023-10-11 13:06:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #230 from christianbender/master

Browse Source 
improvement XOR_cipher.py
This commit is contained in:
Harshil 2017-12-23 18:25:38 +05:30 committed by GitHub
commit eb9fe88e16
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
ciphers/XOR_cipher.py
View File

@ -38,12 +38,7 @@ class XORCipher(object):
# precondition
assert (isinstance(key,int) and isinstance(content,str))
# testing for default arguments
if (key == 0):
if (self.__key == 0):
key = 1
else:
key = self.__key
key = key or self.__key or 1
# make sure key can be any size
while (key > 255):
@ -51,7 +46,6 @@ class XORCipher(object):
# This will be returned
ans = []
resultNumber = 0
for ch in content:
ans.append(chr(ord(ch) ^ key))
@ -69,12 +63,7 @@ class XORCipher(object):
# precondition
assert (isinstance(key,int) and isinstance(content,list))
# testing for default arguments
if (key == 0):
if (self.__key == 0):
key = 1
else:
key = self.__key
key = key or self.__key or 1
# make sure key can be any size
while (key > 255):
@ -82,7 +71,6 @@ class XORCipher(object):
# This will be returned
ans = []
resultNumber = 0
for ch in content:
ans.append(chr(ord(ch) ^ key))
@ -101,12 +89,7 @@ class XORCipher(object):
# precondition
assert (isinstance(key,int) and isinstance(content,str))
# testing for default arguments
if (key == 0):
if (self.__key == 0):
key = 1
else:
key = self.__key
key = key or self.__key or 1
# make sure key can be any size
while (key > 255):
@ -114,7 +97,6 @@ class XORCipher(object):
# This will be returned
ans = ""
resultNumber = 0
for ch in content:
ans += chr(ord(ch) ^ key)
@ -132,12 +114,7 @@ class XORCipher(object):
# precondition
assert (isinstance(key,int) and isinstance(content,str))
# testing for default arguments
if (key == 0):
if (self.__key == 0):
key = 1
else:
key = self.__key
key = key or self.__key or 1
# make sure key can be any size
while (key > 255):
@ -145,7 +122,6 @@ class XORCipher(object):
# This will be returned
ans = ""
resultNumber = 0
for ch in content:
ans += chr(ord(ch) ^ key)
@ -166,16 +142,13 @@ class XORCipher(object):
assert (isinstance(file,str) and isinstance(key,int))
try:
fin = open(file,"r")
fout = open("encrypt.out","w+")
with open(file,"r") as fin:
with open("encrypt.out","w+") as fout:
# actual encrypt-process
for line in fin:
fout.write(self.encrypt_string(line,key))
fin.close()
fout.close()
except:
return False
@ -195,16 +168,13 @@ class XORCipher(object):
assert (isinstance(file,str) and isinstance(key,int))
try:
fin = open(file,"r")
fout = open("decrypt.out","w+")
with open(file,"r") as fin:
with open("decrypt.out","w+") as fout:
# actual encrypt-process
for line in fin:
fout.write(self.decrypt_string(line,key))
fin.close()
fout.close()
except:
return False
@ -233,7 +203,7 @@ class XORCipher(object):
# else:
# print "encrypt unsuccessful"
# if (crypt.decrypt_file("a.out",key)):
# if (crypt.decrypt_file("encrypt.out",key)):
# print "decrypt successful"
# else:
# print "decrypt unsuccessful"