blackteay/kobackupdec
1
0
Fork 0
mirror of https://github.com/RealityNet/kobackupdec.git synced 2025-10-09 17:59:57 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #10 from lp4n6/patch-3

Browse Source 
Update kobackupdec.py
This commit is contained in:
Francesco Picasso 2020-04-06 15:55:59 +02:00 committed by GitHub
commit 0e7fca2738
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
kobackupdec.py
View File

@ -657,7 +657,7 @@ def decrypt_files_in_root(decrypt_info, path_in, path_out):
data_apk_dir = path_out.absolute().joinpath('data/app') data_apk_dir = path_out.absolute().joinpath('data/app')
data_app_dir = path_out.absolute().joinpath('data/data') data_app_dir = path_out.absolute().joinpath('data/data')
#data_app_dir.mkdir(parents=True, exist_ok=True) #data_app_dir.mkdir(0o755, parents=True, exist_ok=True)
data_unk_dir = path_out.absolute().joinpath('unknown') data_unk_dir = path_out.absolute().joinpath('unknown')
for entry in path_in.glob('*'): for entry in path_in.glob('*'):
@ -673,7 +673,7 @@ def decrypt_files_in_root(decrypt_info, path_in, path_out):
if extension == '.apk': if extension == '.apk':
dest_file = data_apk_dir.joinpath(entry.name + '-1') dest_file = data_apk_dir.joinpath(entry.name + '-1')
dest_file.mkdir(parents=True, exist_ok=True) dest_file.mkdir(0o755, parents=True, exist_ok=True)
dest_file = dest_file.joinpath('base.apk') dest_file = dest_file.joinpath('base.apk')
dest_file.write_bytes(entry.read_bytes()) dest_file.write_bytes(entry.read_bytes())
@ -683,7 +683,7 @@ def decrypt_files_in_root(decrypt_info, path_in, path_out):
search=True) search=True)
if cleartext: if cleartext:
dest_file = data_app_dir.joinpath(entry.name) dest_file = data_app_dir.joinpath(entry.name)
dest_file.parent.mkdir(parents=True, exist_ok=True) dest_file.parent.mkdir(0o755, parents=True, exist_ok=True)
dest_file.write_bytes(cleartext) dest_file.write_bytes(cleartext)
else: else:
logging.warning('unable to decrypt entry %s', entry.name) logging.warning('unable to decrypt entry %s', entry.name)
@ -703,7 +703,7 @@ def decrypt_files_in_root(decrypt_info, path_in, path_out):
else: else:
logging.warning('entry %s unmanged, copying it', entry.name) logging.warning('entry %s unmanged, copying it', entry.name)
dest_file = data_unk_dir.joinpath(entry.name) dest_file = data_unk_dir.joinpath(entry.name)
dest_file.parent.mkdir(parents=True, exist_ok=True) dest_file.parent.mkdir(0o755, parents=True, exist_ok=True)
dest_file.write_bytes(entry.read_bytes()) dest_file.write_bytes(entry.read_bytes())
# --- decrypt_files_in_folder ------------------------------------------------- # --- decrypt_files_in_folder -------------------------------------------------
@ -735,7 +735,7 @@ def decrypt_files_in_folder(decrypt_info, folder, path_out):
if cleartext and decrypt_material: if cleartext and decrypt_material:
tmp_path = decrypt_material.path.lstrip('/').lstrip('\\') tmp_path = decrypt_material.path.lstrip('/').lstrip('\\')
dest_file = path_out.joinpath(tmp_path) dest_file = path_out.joinpath(tmp_path)
dest_file.parent.mkdir(parents=True, exist_ok=True) dest_file.parent.mkdir(0o755, parents=True, exist_ok=True)
dest_file.write_bytes(cleartext) dest_file.write_bytes(cleartext)
continue continue
@ -752,7 +752,7 @@ def decrypt_files_in_folder(decrypt_info, folder, path_out):
decrypt_material, entry.read_bytes()) decrypt_material, entry.read_bytes())
if cleartext: if cleartext:
dest_file = media_out_dir.joinpath(entry.relative_to(folder)) dest_file = media_out_dir.joinpath(entry.relative_to(folder))
dest_file.parent.mkdir(parents=True, exist_ok=True) dest_file.parent.mkdir(0o755, parents=True, exist_ok=True)
dest_file.write_bytes(cleartext) dest_file.write_bytes(cleartext)
continue continue
@ -764,7 +764,7 @@ def decrypt_files_in_folder(decrypt_info, folder, path_out):
decrypt_material, entry.read_bytes()) decrypt_material, entry.read_bytes())
if cleartext: if cleartext:
dest_file = media_out_dir.joinpath(entry.relative_to(folder)) dest_file = media_out_dir.joinpath(entry.relative_to(folder))
dest_file.parent.mkdir(parents=True, exist_ok=True) dest_file.parent.mkdir(0o755, parents=True, exist_ok=True)
if entry.suffix.lower() == '.tar': if entry.suffix.lower() == '.tar':
with tarfile.open(fileobj=io.BytesIO(cleartext)) as tdata: with tarfile.open(fileobj=io.BytesIO(cleartext)) as tdata:
if os.name == 'nt': if os.name == 'nt':
@ -775,14 +775,14 @@ def decrypt_files_in_folder(decrypt_info, folder, path_out):
if dest_file.exists(): if dest_file.exists():
new_name = str(folder.name) + '_' + str(dest_file.name) new_name = str(folder.name) + '_' + str(dest_file.name)
dest_file = dest_file.parent.joinpath(new_name) dest_file = dest_file.parent.joinpath(new_name)
dest_file.parent.mkdir(parents=True, exist_ok=True) dest_file.parent.mkdir(0o755, parents=True, exist_ok=True)
dest_file.write_bytes(cleartext) dest_file.write_bytes(cleartext)
continue continue
if cleartext is None: if cleartext is None:
logging.warning('decrypting [%s] failed, copying it', entry.name) logging.warning('decrypting [%s] failed, copying it', entry.name)
dest_file = media_unk_dir.joinpath(entry.name) dest_file = media_unk_dir.joinpath(entry.name)
dest_file.parent.mkdir(parents=True, exist_ok=True) dest_file.parent.mkdir(0o755, parents=True, exist_ok=True)
dest_file.write_bytes(entry.read_bytes()) dest_file.write_bytes(entry.read_bytes())
@ -876,7 +876,14 @@ def main(password, backup_path_in, dest_path_out):
logging.info('setting all decrypted files to read-only') logging.info('setting all decrypted files to read-only')
for entry in dest_path_out.glob('**/*'): for entry in dest_path_out.glob('**/*'):
os.chmod(entry, 0o444) # Set read-only permission if entry is a file.
if os.path.isfile(entry):
os.chmod(entry, 0o444)
# *nix directories require execute permission to read/traverse
elif os.path.isdir(entry):
os.chmod(entry, 0o555)
# --- entry point and parameters checks --------------------------------------- # --- entry point and parameters checks ---------------------------------------
@ -917,6 +924,8 @@ if __name__ == '__main__':
dest_path = pathlib.Path(args.dest_path) dest_path = pathlib.Path(args.dest_path)
if dest_path.is_dir(): if dest_path.is_dir():
sys.exit('Destination folder already exists!') sys.exit('Destination folder already exists!')
dest_path.mkdir(parents=True)
# Make directory with read and execute permission (=read and traverse)
dest_path.mkdir(0o755,parents=True)
main(user_password, backup_path, dest_path) main(user_password, backup_path, dest_path)