#!/usr/bin/env python3 import gzip import hashlib import os import platform import re import shutil import subprocess import urllib.request import zipfile 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 disabled oneshot ioprio rt 0 task_profiles MaxPerformance """ 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 exec -- /system/bin/sh -c "if [ ! -e /data/data/io.github.huskydg.magisk ] ; then pm install /system/etc/init/magisk/magisk.apk ; fi" on property:init.svc.zygote=restarting exec u:r:su:s0 root root -- {MAGISKTMP}/magisk --auto-selinux --zygote-restart 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") def download(self): """下载 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) 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_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) 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_command(["chmod", "+x", n_path]) # 复制 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_command(["chmod", "+x", magisk32_dst]) # 复制 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!") def _create_bootanim_rc(self): """创建 bootanim.rc 文件""" # 创建备份的 gzip 文件 bootanim_path = os.path.join(self.copy_dir, "system", "etc", "init", "bootanim.rc") 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.original_bootanim + self.bootanim_component) os.chmod(bootanim_path, 0o644) 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()