From 42f1af5d5a25ab5a95a16ef4a228d153d3f80e89 Mon Sep 17 00:00:00 2001 From: CoderKang Date: Thu, 18 Sep 2025 11:13:07 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E9=87=8D=E6=9E=84=20Magisk=20=E7=B1=BB?= =?UTF-8?q?=EF=BC=8C=E6=B7=BB=E5=8A=A0=E4=B8=8B=E8=BD=BD=E3=80=81=E6=8F=90?= =?UTF-8?q?=E5=8F=96=E5=92=8C=E5=A4=8D=E5=88=B6=E5=8A=9F=E8=83=BD=EF=BC=8C?= =?UTF-8?q?=E4=BC=98=E5=8C=96=E4=B8=BB=E6=9C=BA=E6=9E=B6=E6=9E=84=E6=A3=80?= =?UTF-8?q?=E6=B5=8B=EF=BC=8C=E7=A1=AE=E4=BF=9D=E4=B8=8E=E6=9C=80=E6=96=B0?= =?UTF-8?q?=E7=89=88=E6=9C=AC=E4=B8=80=E8=87=B4=EF=BC=8C=E5=B9=B6=E5=88=9B?= =?UTF-8?q?=E5=BB=BA=20bootanim.rc=20=E6=96=87=E4=BB=B6=E4=BB=A5=E6=94=AF?= =?UTF-8?q?=E6=8C=81=E5=AE=8C=E6=95=B4=E6=9E=84=E5=BB=BA=E6=B5=81=E7=A8=8B?= =?UTF-8?q?=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- magisk.py | 200 +++++++++++++++++++++++++++++++++++++++++++----------- 1 file changed, 161 insertions(+), 39 deletions(-) diff --git a/magisk.py b/magisk.py index 6f8c59e..4828039 100644 --- a/magisk.py +++ b/magisk.py @@ -1,21 +1,66 @@ +#!/usr/bin/env python3 import gzip +import hashlib import os -import shutil +import platform import re -from stuff.general import General -from tools.helper import bcolors, download_file, host, print_color, run, get_download_dir +import shutil +import subprocess +import urllib.request +import zipfile -class Magisk(General): - download_loc = get_download_dir() - dl_link = "https://github.com/topjohnwu/Magisk/releases/download/v30.2/Magisk-v30.2.apk" - dl_file_name = os.path.join(download_loc, "magisk.apk") - act_md5 = "2691c30ccf059af2536cb0af803c787c" - extract_to = "/tmp/magisk_unpack" - copy_dir = "./magisk" - magisk_dir = os.path.join(copy_dir, "system", "etc", "init", "magisk") - machine = host() - oringinal_bootanim = """ -service bootanim /system/bin/bootanimation + +def calculate_md5(file_path): + """计算文件的 MD5 值""" + hash_md5 = hashlib.md5() + with open(file_path, "rb") as f: + for chunk in iter(lambda: f.read(4096), b""): + hash_md5.update(chunk) + return hash_md5.hexdigest() + + +def download_file(url, file_path): + """下载文件""" + print(f"==> Downloading from {url}") + urllib.request.urlretrieve(url, file_path) + + +def get_host_arch(): + """获取主机架构""" + machine = platform.machine().lower() + if machine in ['amd64', 'x86_64']: + return 'arm64' # 默认构建 arm64 + elif machine in ['i386', 'i686', 'x86']: + return 'arm64' # 默认构建 arm64 + elif machine in ['aarch64', 'arm64']: + return 'arm64' + elif machine.startswith('arm'): + return 'arm' + else: + return 'arm64' # 默认返回 arm64 + + +def run_command(cmd): + """运行命令""" + try: + subprocess.run(cmd, check=True) + except subprocess.CalledProcessError as e: + print(f"Command failed: {' '.join(cmd)}") + raise e + + +class Magisk: + def __init__(self): + self.download_loc = os.path.join(os.path.dirname(os.path.abspath(__file__)), "downloads") + self.dl_link = "https://github.com/topjohnwu/Magisk/releases/download/v30.2/Magisk-v30.2.apk" + self.dl_file_name = os.path.join(self.download_loc, "magisk.apk") + self.act_md5 = "2691c30ccf059af2536cb0af803c787c" + self.extract_to = os.path.join(os.path.dirname(os.path.abspath(__file__)), "temp") + self.copy_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "magisk") + self.magisk_dir = os.path.join(self.copy_dir, "system", "etc", "init", "magisk") + self.machine_arch = get_host_arch() + + self.original_bootanim = """service bootanim /system/bin/bootanimation class core animation user graphics group graphics audio @@ -25,18 +70,21 @@ service bootanim /system/bin/bootanimation task_profiles MaxPerformance """ - bootanim_component = """ -on post-fs-data + + self.bootanim_component = """on post-fs-data start logd exec u:r:su:s0 root root -- {MAGISKSYSTEMDIR}/magiskpolicy --live --magisk exec u:r:magisk:s0 root root -- {MAGISKSYSTEMDIR}/magiskpolicy --live --magisk exec u:r:update_engine:s0 root root -- {MAGISKSYSTEMDIR}/magiskpolicy --live --magisk exec u:r:su:s0 root root -- {MAGISKSYSTEMDIR}/{magisk_name} --auto-selinux --setup-sbin {MAGISKSYSTEMDIR} {MAGISKTMP} exec u:r:su:s0 root root -- {MAGISKTMP}/magisk --auto-selinux --post-fs-data + on nonencrypted exec u:r:su:s0 root root -- {MAGISKTMP}/magisk --auto-selinux --service + on property:vold.decrypt=trigger_restart_framework exec u:r:su:s0 root root -- {MAGISKTMP}/magisk --auto-selinux --service + on property:sys.boot_completed=1 mkdir /data/adb/magisk 755 exec u:r:su:s0 root root -- {MAGISKTMP}/magisk --auto-selinux --boot-complete @@ -47,60 +95,134 @@ on property:init.svc.zygote=restarting on property:init.svc.zygote=stopped exec u:r:su:s0 root root -- {MAGISKTMP}/magisk --auto-selinux --zygote-restart - """.format(MAGISKSYSTEMDIR="/system/etc/init/magisk", MAGISKTMP="/sbin", magisk_name="magisk") +""".format(MAGISKSYSTEMDIR="/system/etc/init/magisk", MAGISKTMP="/sbin", magisk_name="magisk") def download(self): - print_color("Downloading latest Magisk-Delta now .....", bcolors.GREEN) - super().download() + """下载 Magisk APK""" + print("==> Downloading latest Magisk now .....") + + # 确保下载目录存在 + os.makedirs(self.download_loc, exist_ok=True) + + # 检查并下载 Magisk APK + need_download = True + if os.path.exists(self.dl_file_name): + print("==> Checking existing Magisk APK...") + actual_md5 = calculate_md5(self.dl_file_name) + if actual_md5 == self.act_md5: + print("==> Magisk APK already exists and verified!") + need_download = False + else: + print(f"==> MD5 mismatch. Expected: {self.act_md5}, Got: {actual_md5}") + + if need_download: + download_file(self.dl_link, self.dl_file_name) + actual_md5 = calculate_md5(self.dl_file_name) + if actual_md5 != self.act_md5: + raise Exception(f"Downloaded file MD5 verification failed! Expected: {self.act_md5}, Got: {actual_md5}") + print("==> Download completed and verified!") + + def extract(self): + """解压 APK 文件""" + print("==> Extracting Magisk APK...") + + # 清理并创建解压目录 + shutil.rmtree(self.extract_to, ignore_errors=True) + os.makedirs(self.extract_to, exist_ok=True) + + # 解压 APK + with zipfile.ZipFile(self.dl_file_name) as z: + z.extractall(self.extract_to) def copy(self): + """复制文件到目标目录""" + print("==> Copying magisk files now ...") + + # 清理并创建目标目录 if os.path.exists(self.copy_dir): shutil.rmtree(self.copy_dir) - if not os.path.exists(self.magisk_dir): - os.makedirs(self.magisk_dir, exist_ok=True) - - if not os.path.exists(os.path.join(self.copy_dir, "sbin")): - os.makedirs(os.path.join(self.copy_dir, "sbin"), exist_ok=True) - - print_color("Copying magisk libs now ...", bcolors.GREEN) + os.makedirs(self.magisk_dir, exist_ok=True) + os.makedirs(os.path.join(self.copy_dir, "sbin"), exist_ok=True) + # 架构映射 arch_map = { "x86": "x86", "x86_64": "x86_64", "arm": "armeabi-v7a", "arm64": "arm64-v8a" } - lib_dir = os.path.join(self.extract_to, "lib", arch_map[self.machine[0]]) + + # 复制主要架构的库文件 + lib_dir = os.path.join(self.extract_to, "lib", arch_map[self.machine_arch]) if os.path.exists(lib_dir): for parent, dirnames, filenames in os.walk(lib_dir): for filename in filenames: - o_path = os.path.join(lib_dir, filename) + o_path = os.path.join(lib_dir, filename) so_name = re.search(r'lib(.*)\.so', filename) if so_name: n_path = os.path.join(self.magisk_dir, so_name.group(1)) shutil.copyfile(o_path, n_path) - run(["chmod", "+x", n_path]) + run_command(["chmod", "+x", n_path]) - # 同时复制 arm32 的 magisk 二进制文件（如果存在） + # 复制 arm32 的 magisk 二进制文件（如果存在） lib32_path = os.path.join(self.extract_to, "lib", "armeabi-v7a") magisk32_src = os.path.join(lib32_path, "libmagisk32.so") magisk32_dst = os.path.join(self.magisk_dir, "magisk32") if os.path.exists(magisk32_src): shutil.copyfile(magisk32_src, magisk32_dst) - run(["chmod", "+x", magisk32_dst]) + run_command(["chmod", "+x", magisk32_dst]) - # Copy magisk.apk to target directory + # 复制 magisk.apk 到目标目录 apk_dst = os.path.join(self.magisk_dir, "magisk.apk") if os.path.exists(self.dl_file_name): shutil.copyfile(self.dl_file_name, apk_dst) + + # 生成 bootanim.rc 文件 + self._create_bootanim_rc() + + print("==> Magisk installation completed!") - # Updating Magisk from Magisk manager will modify bootanim.rc, - # So it is necessary to backup the original bootanim.rc. + def _create_bootanim_rc(self): + """创建 bootanim.rc 文件""" + # 创建备份的 gzip 文件 bootanim_path = os.path.join(self.copy_dir, "system", "etc", "init", "bootanim.rc") - gz_filename = os.path.join(bootanim_path)+".gz" - with gzip.open(gz_filename,'wb') as f_gz: - f_gz.write(self.oringinal_bootanim.encode('utf-8')) + gz_filename = bootanim_path + ".gz" + + # 确保目录存在 + os.makedirs(os.path.dirname(bootanim_path), exist_ok=True) + + # 创建 gzip 备份 + with gzip.open(gz_filename, 'wb') as f_gz: + f_gz.write(self.original_bootanim.encode('utf-8')) + + # 创建完整的 bootanim.rc with open(bootanim_path, "w") as initfile: - initfile.write(self.oringinal_bootanim+self.bootanim_component) + initfile.write(self.original_bootanim + self.bootanim_component) + + os.chmod(bootanim_path, 0o644) - os.chmod(bootanim_path, 0o644) \ No newline at end of file + def run(self): + """执行完整的构建流程""" + try: + self.download() + self.extract() + self.copy() + except Exception as e: + print(f"Error: {e}") + return False + return True + + +def main(): + """主函数""" + magisk = Magisk() + success = magisk.run() + if success: + print("==> Magisk vendor package created successfully!") + else: + print("==> Failed to create Magisk vendor package!") + exit(1) + + +if __name__ == '__main__': + main() \ No newline at end of file