1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
| import os
import subprocess as sp
from argparse import ArgumentParser
from pathlib import Path
from time import sleep
from typing import Optional, Union
from xml.dom import minidom
DEFAULT_ANDROID_SDK_DIR = Path(os.getenv("LOCALAPPDATA"), "Android", "Sdk")
def find_bin_path(filename: str, *extra_search_dirs: list[os.PathLike]) -> Optional[Path]:
"""Try to find bin file in system path and return full file path."""
search_dirs = os.get_exec_path()
search_dirs.extend(extra_search_dirs)
for p in map(Path, search_dirs):
exec_path = p.joinpath(filename)
if exec_path.is_file():
return exec_path.resolve()
return None
class AndroidSdkBuildTool:
def __init__(self, tool_dir) -> None:
self._dir = Path(tool_dir).resolve()
@property
def version(self):
return self._dir.name
@property
def apksigner(self):
return self._dir.joinpath("lib", "apksigner.jar")
@property
def zipalign(self):
return self._dir.joinpath("zipalign.exe")
class AndroidSdk:
def __init__(self, sdk_dir: os.PathLike) -> None:
self._dir = Path(sdk_dir).resolve()
def get_build_tools(self):
return [AndroidSdkBuildTool(p) for p in self._dir.joinpath("build-tools").iterdir()]
class AndroidManifest:
def __init__(self, path: os.PathLike) -> None:
self._tree = minidom.parse(str(path))
def enable_debuggable(self, flag: bool = True):
application_node = self._tree.getElementsByTagName("application")[0]
application_node.setAttribute("android:debuggable", ("true" if flag else "false"))
def save(self, path: os.PathLike):
path = Path(path)
path.write_bytes(self._tree.toxml(encoding="utf-8", standalone=False))
return path
class APKRepacker:
def __init__(
self,
*extra_bin_dirs,
android_sdk_dir: Optional[Path] = None,
ks_file: Optional[os.PathLike] = None,
ks_pwd: Optional[str] = None
) -> None:
if android_sdk_dir is None:
android_sdk_dir = DEFAULT_ANDROID_SDK_DIR
self._java_path = find_bin_path("java.exe", *extra_bin_dirs)
self._apktool_path = find_bin_path("apktool.jar", *extra_bin_dirs)
self._android_sdk = AndroidSdk(android_sdk_dir)
self._android_build_tool = self._android_sdk.get_build_tools()[-1]
self._ks_file = None if ks_file is None else Path(ks_file)
self._ks_pwd = ks_pwd
def _exec(self, bin_path: os.PathLike, *args):
args = [str(Path(bin_path).resolve()), *map(str, args)]
print(f"Exec: {args}")
return sp.run(args)
def _exec_jar(self, jar_path: os.PathLike, *args):
return self._exec(self._java_path, "-jar", jar_path, args)
def print_executables(self):
print(f"java: {self._java_path}")
print(f"apktool: {self._apktool_path}")
print(f"zipalign: {self._android_build_tool.zipalign}")
print(f"apksigner: {self._android_build_tool.apksigner}")
def unpack(self, apk_path: os.PathLike, output_dir: os.PathLike):
self._exec_jar(self._apktool_path, "-f", "d", str(apk_path), "-o", str(output_dir))
return Path(output_dir)
def pack(self, apk_dir: os.PathLike, output_path: os.PathLike):
self._exec_jar(self._apktool_path, "-f", "b", str(apk_dir), "-o", str(output_path))
return Path(output_path)
def align(self, apk_path: os.PathLike, output_path: os.PathLike):
apk_path = Path(apk_path).resolve()
output_path = Path(output_path).resolve()
self._exec(self._android_build_tool.zipalign, "-f", "-p", "4", str(apk_path), str(output_path))
return Path(output_path)
def sign(self, apk_path: os.PathLike, output_path: os.PathLike):
if self._ks_file is None or self._ks_pwd is None:
raise ValueError("apksigner needs ks file and ks password")
p = sp.Popen(
[str(self._java_path), "-jar",
str(self._android_build_tool.apksigner), "sign",
"--ks", str(self._ks_file), "--out", str(output_path), str(apk_path)],
stdin=sp.PIPE
)
sleep(1)
print(self._ks_pwd, file=p.stdin, end="\n")
return Path(output_path)
def add_debuggable_flag(self, apk_dir: os.PathLike):
manifest_path = Path(apk_dir).joinpath("AndroidManifest.xml")
manifest = AndroidManifest(manifest_path)
manifest.enable_debuggable(True)
manifest.save(manifest_path)
def main():
parser = ArgumentParser()
parser.add_argument("-p", "--apk", type=str, help="要修改的 APK 文件路径", required=True)
parser.add_argument("-o", "--output", type=str, help="重新打包后 APK 的输出路径", required=True)
parser.add_argument("--sdk-dir", type=str, help="Android Sdk 目录")
parser.add_argument("--extra-search-dir", nargs="+", type=str, help="可执行文件或者 jar 包的额外搜索路径", default=())
parser.add_argument("--ks-file", type=str, help="keytool 生成的自签名证书文件")
parser.add_argument("--ks-pwd", type=str, help="ks 文件的口令")
parser.add_argument("--enable-debuggable", action="store_true", help="在 AndroidManifest.xml 增加可调试标记")
args = parser.parse_args()
tmp = Path("./tmp").resolve()
tmp.mkdir(parents=True, exist_ok=True)
repacker = APKRepacker(
*args.extra_search_dir,
android_sdk_dir=args.sdk_dir,
ks_file=args.ks_file,
ks_pwd=args.ks_pwd
)
repacker.print_executables()
apk_path = Path(args.apk)
apk_name = apk_path.stem
apk_dir = tmp.joinpath(apk_name)
apk_repack_path = tmp.joinpath(f"{apk_name}_repack.apk")
apk_align_path = tmp.joinpath(f"{apk_name}_align.apk")
apk_output_path = Path(args.output)
repacker.unpack(apk_path, apk_dir)
if args.enable_debuggable:
repacker.add_debuggable_flag(apk_dir)
repacker.pack(apk_dir, apk_repack_path)
repacker.align(apk_repack_path, apk_align_path)
repacker.sign(apk_align_path, apk_output_path)
if __name__ == "__main__":
main()
|