def get_encrypt_filepath(destfile):
basename = os.path.basename(destfile)
name, ext = os.path.splitext(basename)
if ext == ".jar":
filename = '%s%s' % (basename, '.en')
else:
filename = '%s%s' % (basename, '.de')
dirname = os.path.dirname(destfile) or os.path.dirname(__file__)
filepath = os.path.join(dirname, filename)
return filepath
def xor_encrypt_file(destfile, keyfile):
dest_handle = open(destfile, 'rb')
dest_end_pos = os.path.getsize(destfile)
target_file = get_encrypt_filepath(destfile)
target_file_handle = open(target_file, 'wb')
key_handle = open(keyfile, 'rb')
key_end_pos = os.path.getsize(keyfile)
while True:
dest_byte = dest_handle.read(1)
key_byte = key_handle.read(1)
current_dest_pos = dest_handle.tell()
# print "current_dest_pos",current_dest_pos
current_key_pos = key_handle.tell()
# print "current_key_pos",current_key_pos
if current_key_pos == key_end_pos:
key_handle.seek(0)
key_byte = key_handle.read(1)
xor_value = ord(dest_byte) ^ ord(key_byte)
xor_byte = struct.pack(
'B',
xor_value
)
target_file_handle.write(xor_byte)
if current_dest_pos == dest_end_pos:
break
target_file_handle.flush()
target_file_handle.close()
dest_handle.close()
key_handle.close()
print "target_file_handle",target_file_handle
xor_encrypt_file(dest_file, key_file)