228 lines
8.1 KiB
Python
228 lines
8.1 KiB
Python
#!/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() |