From e21922b1f18068dcfbb550900c847ba47b576483 Mon Sep 17 00:00:00 2001 From: MasOnShi Date: Sun, 3 Apr 2022 20:22:25 +0800 Subject: [PATCH] dev commits. --- auto-analysis/utils.py | 14 ++++++++------ LICENSE => exports/LICENSE | 0 README.md => exports/README.md | 0 README_CN.md => exports/README_CN.md | 0 masar.py => exports/masar.py | 0 requirements.txt => exports/requirements.txt | 0 typora.py => exports/typora.py | 0 7 files changed, 8 insertions(+), 6 deletions(-) rename LICENSE => exports/LICENSE (100%) rename README.md => exports/README.md (100%) rename README_CN.md => exports/README_CN.md (100%) rename masar.py => exports/masar.py (100%) rename requirements.txt => exports/requirements.txt (100%) rename typora.py => exports/typora.py (100%) diff --git a/auto-analysis/utils.py b/auto-analysis/utils.py index 56b4f33..25924d5 100644 --- a/auto-analysis/utils.py +++ b/auto-analysis/utils.py @@ -44,13 +44,15 @@ def extract_file(from_path, to_path): subprocess.check_call(["innoextract", from_path, "-d", to_path]) -def patch_file(_key, _iv): - patch_file_path = os.path.join(BASE_DIR, "../typora.py") - fd = open(patch_file_path, "a+") - content = fd.read() +def patch_file(_key, _iv, to_dir): + patch_file_path = os.path.join(BASE_DIR, "../exports/typora.py") + content = open(patch_file_path, "r").read() content = content.replace("{AES_KEY}", f"b''.fromhex('{_key}')") content = content.replace("{AES_IV}", f"b''.fromhex('{_iv}')") - fd.write(content) + save_dir = os.path.join(to_dir, "build") + os.makedirs() + open(os.path.join(to_dir, "typora.py"), "w").write(content) + def win_x64_run(): @@ -68,7 +70,7 @@ def win_x64_run(): key, iv = analysis.get_aes_key_and_iv(main_node_path) log.success("analysis done") - patch_file(key.hex(), iv.hex()) + patch_file(key.hex(), iv.hex(), basedir) log.success("patch done") diff --git a/LICENSE b/exports/LICENSE similarity index 100% rename from LICENSE rename to exports/LICENSE diff --git a/README.md b/exports/README.md similarity index 100% rename from README.md rename to exports/README.md diff --git a/README_CN.md b/exports/README_CN.md similarity index 100% rename from README_CN.md rename to exports/README_CN.md diff --git a/masar.py b/exports/masar.py similarity index 100% rename from masar.py rename to exports/masar.py diff --git a/requirements.txt b/exports/requirements.txt similarity index 100% rename from requirements.txt rename to exports/requirements.txt diff --git a/typora.py b/exports/typora.py similarity index 100% rename from typora.py rename to exports/typora.py