-
Notifications
You must be signed in to change notification settings - Fork 4
/
patch_R.py
1189 lines (1033 loc) · 49.2 KB
/
patch_R.py
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
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
#!/usr/bin/env python
# encoding: utf-8
# Author: Liu DongMiao <[email protected]>
# Modified By Pzqqt <[email protected]>
# Modified By Saint-Theana@github
# https://github.com/Saint-Theana/Prevent-Q
import locale
import os
import sys
import shutil
import re
plus_flag = False
MESSAGE_ZH_TW = {
'%s not exists%s': '無法找到檔案 %s%s',
'%s patched %s%s': '成功為 %s 打了 %s 處補丁%s',
'%s patched %s, should patch %s%s': '為 %s 打了 %s 處補丁，應該打 %s 處%s',
'copying %s%s': '正在拷貝%s%s',
}
MESSAGE_ZH = {
'%s not exists%s': '无法找到文件 %s%s',
'%s patched %s%s': '成功为 %s 打了 %s 处补丁%s',
'%s patched %s, should patch %s%s': '为 %s 打了 %s 处补丁，应该打 %s 处%s',
'copying %s%s': '正在复制 %s%s',
}
def _(message):
# 对要在终端打印的字符串进行翻译和编码(仅在Python 2.x需要编码)
lang, encoding = locale.getdefaultlocale()
if not lang or not lang.startswith("zh"):
return message
if lang.startswith("zh_TW"):
translate = MESSAGE_ZH_TW.get(message, message)
else:
translate = MESSAGE_ZH.get(message, message)
if hasattr(translate, 'decode'):
return translate.decode('utf8').encode(encoding)
else:
return translate
class Patch(object):
def __init__(self, dir_services='services'):
# services.jar文件反编译后的路径 默认值为services
self.dir_services = dir_services
def build_path(self, paths, dir_path=None):
# 组合dir_path和paths路径(dir_path + paths)
# 由于paths字符串已定并且使用"/"作为路径分割符
# 所以这样做可以使其同时兼容win平台和Linux平台
if dir_path is None:
# 如果可选参数dir_path没有提供 则使用self.dir_services
# 再往上推 其实就是"services"目录
dir_path = self.dir_services
for path in os.listdir(dir_path):
if path.startswith("smali"):
if os.path.exists(dir_path+"/"+path+"/"+paths):
return dir_path+"/"+path+"/"+paths
sys.stderr.write(_("%s not exists%s") % (paths, os.linesep))
raise SystemExit(1)
# os.path.normpath(path)方法：规范path字符串形式
# 例如：
# path = r'D:\A\B\..\C'
# 则 os.path.normpath(path) == r'D:\A\C'
# ~ path = os.path.normpath(dir_path)
# ~ for x in paths.split("/"):
# ~ path = os.path.join(path, x)
# 上面两行代码可以像这样简写为一行吗？可以！
# ~ path = os.path.join(path, *paths.split("/"))
# ~ return path
# 一行代码 简单粗暴
def run(self):
patched = 0
# get_path方法定义了要打补丁的文件路径
path = self.build_path(self.get_path())
output = open(path + ".patched", "w")
for line in open(path, "r"):
# 打补丁动作在此
# 逐行读取smali文件 并调用patch方法
# 如果打补丁成功 则已打补丁处数(patched)加1
# 否则直接写入*.smali.patched文件
if self.patch(output, line):
patched += 1
else:
output.write(line)
output.close()
# remove .smali
# 移除*.smali文件(实际是重命名为*.smali.orig文件)
# 并将刚才打好补丁的smali文件(*.smali.patched)重命名为*.smali
# os.path.basename(path): 返回文件名
# 此处切片[:-6] 略过字符".smali" 也就是文件后缀
path_name = os.path.basename(path)[:-6]
if patched == self.get_patch_count():
os.rename(path, path + ".orig")
os.rename(path + ".patched", path)
sys.stdout.write(_("%s patched %s%s") % (
path_name, patched, os.linesep))
else:
sys.stderr.write(_("%s patched %s, should patch %s%s") % (
path_name, patched, self.get_patch_count(), os.linesep))
raise SystemExit(2)
return patched
def init_pr_methods(self, fixing, smali_path, dir_apk):
# 初始化Prevent的方法(methods)
# 用意？从apk目录中的指定smali文件中获取所有的method语句体
# 为什么要获取？这是补丁版黑域自己定义的method
# 使用它来替换原本的method以达到劫持的目的
# 此方法在下面的ActivityManagerService类中有使用
path = self.build_path(smali_path, dir_apk)
methods = {}
# method_signature 应该是起到了一个信号的作用 表示method语句的开始
# method_body method完整语句体 最终得到的字符串将包括".method" 和 ".end method"语句
# method_name method名
method_name = method_signature = method_body = ''
for line in open(path, "r"):
line_strip = line.strip()
if line.startswith(".method"):
method_signature = line_strip
method_body = line
method_name = self.find_method_name(method_signature)
elif line.startswith(".end method"):
method_body += line
# 如果method名在预设的fixing集合里
# 则添加元素到methods列表里
if method_name in fixing:
methods[method_signature] = (method_name, method_body)
method_name = method_signature = method_body = ''
elif line_strip.startswith(".line"):
# 跳过".line"行
continue
elif method_signature:
method_body += line
return methods
def get_path(self):
raise NotImplemented # 预先定义类的方法 直接调用则会...
def get_patch_count(self):
raise NotImplemented
def patch(self, output, line):
raise NotImplemented
@staticmethod # 装饰器 类的静态方法 不会对任何实例类型进行操作 可以理解为此方法与self无关
def find_method_name(line):
# 看起来这个函数是返回smali的方法名的
# 例如：
# line = "method public abstract setVersion(I)V"
# 则调用此函数后返回： "setVersion"
# 找出字符串line中第一次出现"("的位置索引
end = line.index("(")
# 找出字符串line中最后一次出现" "的位置索引 指定范围：开头(0) -> 第一次出现"("的位置索引(end)
start = line.rindex(" ", 0, end)
# 返回line的切片(start + 1 到 end)
return line[start + 1:end]
@staticmethod
def get_method_arguments(line):
if line.strip().startswith("iget-object"):
return [a[:-1] for a in line.strip().split()[1:3]]
# 获取smali的method参数
start = line.index("{")
end = line.index("}", start)
argument = line[start + 1:end]
# 如果argument中有".."字符串(省略了一些参数)
# 例如：
# line = "invoke-virtual/range {p0 .. p8}, Lcom/andro(后略)"
# 此时argument = "p0 .. p8"
if '..' in argument:
# 函数partition()
# 作用：按分割符将字符串分割为3部分 返回一个3元的元组
# 第一个为分隔符左边的子串 第二个为分隔符本身 第三个为分隔符右边的子串
# strip() 剔除字符串两侧空白
start, _, end = [x.strip() for x in argument.partition("..")]
sn = int(start[1:])
en = int(end[1:])
# 若argument的首尾参数不是以同一字母开头则抛出异常
# 字母？在smali语法中 p开头的寄存器为参数寄存器 v开头的寄存器为本地寄存器
assert start[0] == end[0]
prefix = start[0]
arguments = []
for x in range(sn, en + 1):
arguments.append("%s%d" % (prefix, x))
return arguments
else:
return [x.strip() for x in argument.split(",")]
# 最终返回argument包含的所有参数的列表
# 例如：
# argument = "p0, p1, p2, p3"
# 则返回 ["p0", "p1", "p2", "p3"]
# argument = "p0 .. p8"
# 则返回 ["p0", "p1", "p2", "p3", "p4", "p5", "p6", "p7", "p8"]
class IntentResolver(Patch): # 注意！此类继承于上面的"Patch"类！
def get_path(self):
# 定义要打补丁的文件相对路径
return "com/android/server/IntentResolver.smali"
def patch(self, output, line):
if 'Landroid/content/Intent;->isExcludingStopped(' in line:
# 字符串的replace方法
# 方法：replace(旧字符串, 新字符串[, 最大替换次数])
new_line = line.replace("invoke-virtual/range", "invoke-static/range") \
.replace("Landroid/content/Intent;->isExcludingStopped(",
"Lcom/android/server/am/PreventRunningUtils;->isExcludingStopped(Landroid/content/Intent;")
output.write(new_line)
# 返回新旧字符串是否相同 用以判断是打补丁是否成功
# 不同则返回True 相同则返回False
return new_line != line
elif 'Landroid/content/IntentFilter;->match(' in line:
new_line = line.replace("invoke-virtual/range", "invoke-static/range") \
.replace("Landroid/content/IntentFilter;->match(",
"Lcom/android/server/am/PreventRunningUtils;->match(Landroid/content/IntentFilter;")
output.write(new_line)
return new_line != line
# 此处应该有个"return None" 嘛。。。只是便于理解
def get_patch_count(self):
# 定义应打几处补丁
return 2
class ActivityManagerService(Patch):
methods = None
pkg_deps = ''
extra_count = 0
method_name_sp = ""
# 使用大括号创建了一个仅包含键的字典 称之为集合
# 特点: 检索迅速 不会有重复值
# 这个集合包含了所有要修正的method名
# 即：用补丁版黑域自己的的method替代原本的method
# 原本的method体不动 但method名重命名为"原method名"+"$Pr"
# 至于如何替换 替换为什么样的代码 参考父类的init_pr_methods方法
#fixing bindService
#fixing broadcastIntent
#fixing handleAppDiedLocked
#fixing moveActivityTaskToBack
#fixing startService
#bindService
#broadcastIntent
#handleAppDiedLocked
#moveActivityTaskToBack
#startActivity
#startService
fixing = {'startProcessLocked', 'startActivity', 'handleAppDiedLocked', 'cleanUpRemovedTaskLocked',
'moveActivityTaskToBack', 'startService', 'bindService', 'broadcastIntent'}
def __init__(self, dir_services=None, dir_apk='apk'):
if dir_services is None:
# super() 函数: 用于调用父类(超类)的一个方法
super(ActivityManagerService, self).__init__()
else:
super(ActivityManagerService, self).__init__(dir_services)
self.dir_apk = dir_apk
self.methods = self.init_pr_methods(self.fixing, self.get_path(), self.dir_apk)
def get_path(self):
return "com/android/server/am/ActivityManagerService.smali"
def patch(self, output, line):
line_strip = line.strip()
# 跳过空行
if not line_strip:
return False
# 跳过".line"行
if line_strip.startswith(".line"):
return False
if line.startswith(".method"):
method_signature = line_strip
self.method_name_sp = self.find_method_name(line_strip)
# 如果method体第一行在self.methods里也有
if method_signature == ".method public startActivity(Landroid/app/IApplicationThread;Ljava/lang/String;Landroid/content/Intent;Ljava/lang/String;Landroid/os/IBinder;Ljava/lang/String;IILandroid/app/ProfilerInfo;Landroid/os/Bundle;)I":
method_name, method_body = self.methods[".method public final startActivity(Landroid/app/IApplicationThread;Ljava/lang/String;Landroid/content/Intent;Ljava/lang/String;Landroid/os/IBinder;Ljava/lang/String;IILandroid/app/ProfilerInfo;Landroid/os/Bundle;)I"]
# 如果method名在method名修正集合里
if method_name in self.fixing:
#print("fixing "+method_name)
# 写入新的method语句体
# 写入一个空行
# 把原本的method语句块的(首行的)method名末尾加上"$Pr"
# 最后从self.fixing中移除此method名 返回True
output.write(method_body)
output.write(os.linesep)
output.write(line.replace(method_name, method_name + "$Pr"))
self.fixing.remove(method_name)
return True
if method_signature in self.methods:
# 从self.methods中取出method名和method完整语句体
method_name, method_body = self.methods[method_signature]
# 如果method名在method名修正集合里
if method_name in self.fixing:
#print("fixing "+method_name)
# 写入新的method语句体
# 写入一个空行
# 把原本的method语句块的(首行的)method名末尾加上"$Pr"
# 最后从self.fixing中移除此method名 返回True
output.write(method_body)
output.write(os.linesep)
output.write(line.replace(method_name, method_name + "$Pr"))
self.fixing.remove(method_name)
return True
# 此处应该有个"return None" 嘛。。。只是便于理解
# 下面类似的地方就不补充注释了
# 配合self.pkg_deps这个flag 判断此行是否需要重写
if self.pkg_deps and 'Landroid/util/ArraySet;->contains(Ljava/lang/Object;)Z' in line_strip:
output.write(" invoke-static {},"
" Lcom/android/server/am/PreventRunningUtils;->returnFalse()Z")
output.write(os.linesep)
# 此处patch比较特殊 故使用self.extra_count来特殊计数
# 为什么特殊？此处Patch在4.4下是打不上的
self.extra_count += 1
self.pkg_deps = ''
return True
# 此处Patch仅用于killPackageProcessesLocked方法
# 在8.x+中 有可能会给getPackageProcessState方法也打上补丁
# 这样做是多余的 所以这里添加一个额外的条件判断
if self.method_name_sp == "killPackageProcessesLocked":
if line_strip.startswith('iget-object'):
# self.pkg_deps用来设置一个flag 因为此行之后的一行需要重写
# 对此行以"iget-object"开头的语句进行判断
if 'Lcom/android/server/am/ProcessRecord;->pkgDeps:Landroid/util/ArraySet;' in line_strip:
self.pkg_deps = line_strip
else:
self.pkg_deps = ''
def get_patch_count(self):
# 4.4 必打 8+0 处补丁
# 5.x ~ 7.x 必打 8+1 处补丁
# 8.x+ 只打 7+1 处补丁 属于正常情况
global plus_flag
if plus_flag:
return 6 + self.extra_count
else:
return 7 + self.extra_count
def run(self):
super(ActivityManagerService, self).run()
src = self.build_path("com/android/server/am/", self.dir_apk)
dest = self.build_path("com/android/server/am/", self.dir_services)
# 复制apk目录下文件名以PreventRunning开头的文件 到services目录下(需要复制三个文件)
for path in os.listdir(src):
if path.startswith("PreventRunning"):
sys.stdout.write(_("copying %s%s") % (path[:-6], os.linesep))
shutil.copy(os.path.join(src, path), dest)
class ActivityStack(Patch):
# 此类和IntentResolver类类似 不再详解
patched = 0
method_name_sp = ""
startP1Tracing=False
p1Ocupied=False
p1Alternative="p1"
p1Identity="p1"
astActivityRecordRef=""
extrapatchcount=0
def get_path(self):
return "com/android/server/wm/ActivityStack.smali"
def patch(self, output, line):
# 写得非常不清真的代码
if line.startswith(".method"):
self.method_name_sp = self.find_method_name(line.strip())
if self.method_name_sp == "resumeTopActivityInnerLocked":
if "iget-object" in line:
if "Lcom/android/server/wm/ActivityRecord;->" in line:
self.lastActivityRecordRef=re.findall("iget-object [pv][0-9]+, ([pv][0-9]+)",line)[0]
if "Lcom/android/server/wm/ClientLifecycleManager;->scheduleTransaction(Landroid/app/servertransaction/ClientTransaction;)V" in line:
output.write(line)
output.write(os.linesep)
# 父类定义的get_method_arguments方法: 获取smali的method参数 返回参数列表
# 取此列表的第一个参数赋值给argument变量
argument = self.get_method_arguments(line)[0]
output.write("iget-object %s, %s, Lcom/android/server/wm/ActivityRecord;->appToken:Lcom/android/server/wm/ActivityRecord$Token;" %(argument, self.lastActivityRecordRef))
output.write(os.linesep)
output.write(" invoke-static/range {%s .. %s},"
" Lcom/android/server/am/PreventRunningUtils;"
"->onResumeActivity(Landroid/os/IBinder;)V" % (argument, argument))
output.write(os.linesep)
self.patched += 1
return True
if "Lcom/android/server/wm/ActivityTaskManagerServiceInjector;->destroyActivity(Landroid/content/pm/ActivityInfo;)V" in line:
output.write(os.linesep)
self.patched += 1
return True
if self.method_name_sp == "startPausingLocked":
if "iget-object" in line:
if "Lcom/android/server/wm/ActivityRecord;->" in line:
self.lastActivityRecordRef=re.findall("iget-object [pv][0-9]+, ([pv][0-9]+)",line)[0]
if "Landroid/app/servertransaction/PauseActivityItem;->obtain(ZZIZ)Landroid/app/servertransaction/PauseActivityItem;" in line:
self.p1Identity=self.get_method_arguments(line)[1]
self.p1Alternative=self.get_method_arguments(line)[0]
self.startP1Tracing=True
if "move-result-object" in line:
if self.startP1Tracing:
self.startP1Tracing=False
if "move-result-object "+self.p1Identity in line: #刚好占用的就是p1，中奖了
self.p1Ocupied=True
output.write("move-result-object "+self.p1Alternative)
output.write(os.linesep)
self.extrapatchcount+=1
self.patched += 1
return True
if "Lcom/android/server/wm/ClientLifecycleManager;->scheduleTransaction(Landroid/app/IApplicationThread;Landroid/os/IBinder;Landroid/app/servertransaction/ActivityLifecycleItem;)V" in line:
arguments = self.get_method_arguments(line)
if self.p1Ocupied:
output.write("invoke-virtual {%s, %s, %s, %s}, Lcom/android/server/wm/ClientLifecycleManager;->scheduleTransaction(Landroid/app/IApplicationThread;Landroid/os/IBinder;Landroid/app/servertransaction/ActivityLifecycleItem;)V"%(arguments[0],arguments[1],arguments[2],self.p1Alternative))
arguments[3]=self.p1Alternative
else:
output.write(line)
output.write(os.linesep)
output.write("iget-object %s, %s, Lcom/android/server/wm/ActivityRecord;->appToken:Lcom/android/server/wm/ActivityRecord$Token;" %(arguments[2],self.lastActivityRecordRef) )
output.write(os.linesep)
output.write("iget-boolean %s, %s, Lcom/android/server/wm/ActivityRecord;->finishing:Z" %(arguments[3],self.lastActivityRecordRef) )
output.write(os.linesep)
output.write("invoke-static {%s, %s, %s}, Lcom/android/server/am/PreventRunningUtils;->onUserLeavingActivity(Landroid/os/IBinder;ZZ)V"%(arguments[2],arguments[3],self.p1Identity) )
output.write(os.linesep)
self.patched += 1
return True
def get_patch_count(self):
# 打补丁的次数可能会多于3次？
shouldpatch=2
if self.extrapatchcount > 0:
shouldpatch+=self.extrapatchcount
if self.patched > self.extrapatchcount:
return self.patched
else:
return self.extrapatchcount
class ProcessList(Patch):
# 此类和IntentResolver类类似 不再详解
patched = 0
method_name_sp = ""
arg_sp = ""
arg2_sp = ""
startP1Tracing=False
p1Ocupied=False
p1Alternative="p1"
p1Identity="p1"
method_name_sp=""
condtracing=False
killPackageProcessesLockedPached=False
def get_path(self):
return "com/android/server/am/ProcessList.smali"
def patch(self, output, line):
# 写得非常不清真的代码
if line.startswith(".method"):
self.method_name_sp = self.find_method_name(line.strip())
if len(re.findall(".method final startProcessLocked.*Lcom/android/server/am/ProcessRecord;$",line))>0:
paramstr=re.findall("\((.*)\)",line)[0]
objparams=self.parseParam(paramstr)
output.write(os.linesep)
output.write(line) #方法体
output.write(os.linesep)
output.write(" .registers %s" %(len(objparams)+7))
output.write(os.linesep)
ref=self.locateParam(objparams,"Lcom/android/server/am/HostingRecord;")
#print(ref)
output.write("invoke-virtual {p%s}, Lcom/android/server/am/HostingRecord;->getType()Ljava/lang/String;" %(ref+1))
output.write(os.linesep)
output.write("move-result-object v4")
output.write(os.linesep)
output.write("invoke-virtual {p%s}, Lcom/android/server/am/HostingRecord;->getName()Ljava/lang/String;" %(ref+1))
output.write(os.linesep)
output.write("move-result-object v0")
output.write(os.linesep)
output.write("invoke-static {v0}, Landroid/content/ComponentName;->unflattenFromString(Ljava/lang/String;)Landroid/content/ComponentName;")
output.write(os.linesep)
output.write("move-result-object v5")
output.write(os.linesep)
ref=self.locateParam(objparams,"Ljava/lang/String;")
output.write("move-object v0, p%s" %(ref+1))
output.write(os.linesep)
ref=self.locateParam(objparams,"Landroid/content/pm/ApplicationInfo;")
output.write("move-object v1, p%s" %(ref+1))
output.write(os.linesep)
ref=self.locateParam(objparams,"Z")
output.write("move v2, p%s" %(ref+1))
ref=self.locateParam(objparams,"I")
output.write(os.linesep)
output.write("move v3, p%s" %(ref+1))
output.write(os.linesep)
output.write("invoke-static/range {v0 .. v5}, Lcom/android/server/am/PreventRunningUtils;->hookStartProcessLocked(Ljava/lang/String;Landroid/content/pm/ApplicationInfo;ZILjava/lang/String;Landroid/content/ComponentName;)Z")
output.write(os.linesep)
output.write("move-result v0")
output.write(os.linesep)
output.write("if-eqz v0, :cond_0")
output.write(os.linesep)
output.write("invoke-virtual/range {p0 .. p%s}, Lcom/android/server/am/ProcessList;->startProcessLocked$Pr(%s)Lcom/android/server/am/ProcessRecord;" %(len(objparams),paramstr))
output.write(os.linesep)
output.write("move-result-object v0")
output.write(os.linesep)
output.write("return-object v0")
output.write(os.linesep)
output.write(":cond_0")
output.write(os.linesep)
output.write("const/4 v0, 0x0")
output.write(os.linesep)
output.write("return-object v0")
output.write(os.linesep)
output.write(".end method")
output.write(os.linesep)
output.write(line.replace("startProcessLocked","startProcessLocked$Pr")) #方法体
output.write(os.linesep)
self.patched += 1
return True
else:
if self.method_name_sp == "killPackageProcessesLocked":
if self.killPackageProcessesLockedPached:
return False
if "iget" in line and "Lcom/android/server/am/ProcessRecord;->setAdj:I" in line:
self.condtracing=True
if self.condtracing:
if len(re.findall(" if-ltz [pv][0-9]+, :",line))>0:
ref=re.findall(" if-ltz ([pv][0-9]+), :",line)[0]
output.write(os.linesep)
output.write("invoke-static {%s}, Lcom/android/server/am/PreventRunningUtils;->returnFalse(Z)Z" %(ref))
output.write(os.linesep)
output.write("move-result %s" %(ref))
output.write(os.linesep)
output.write(line)
output.write(os.linesep)
self.killPackageProcessesLockedPached=True
self.patched += 1
return True
def locateParam(self,params,param,offset=0):
count=0
index=0
while index < len(params):
if param == params[index]:
if offset >=count:
return index
else:
count+=1
index+=1
return -1
def parseParam(self,input):
index=0
params=[]
while len(input) !=0:
if input[index] == "L" or input[index] == "[":
start=re.search("\[?L[/[a-zA-Z]+;",input).span()[0]
end=re.search("\[?L[/[a-zA-Z]+;",input).span()[1]
params.append(input[:end])
input=input[end:]
else:
params.append(input[0])
input=input[1:]
return params
def get_patch_count(self):
# 打补丁的次数可能会多于3次？
if self.patched > 2:
return self.patched
else:
return 2
class ActivityRecordToken(Patch):
# 此类和IntentResolver类类似 不再详解
patched = 0
def get_path(self):
return "com/android/server/wm/ActivityRecord$Token.smali"
def patch(self, output, line):
# 写得非常不清真的代码
if ".field private weakActivity:Ljava/lang/ref/WeakReference;" in line:
output.write(".field public weakActivity:Ljava/lang/ref/WeakReference;")
output.write(os.linesep)
self.patched += 1
return True
def get_patch_count(self):
# 打补丁的次数可能会多于3次？
if self.patched > 1:
return self.patched
else:
return 1
class ActivityTaskManagerService(Patch):
# 此类和IntentResolver类类似 不再详解
patched = 0
def get_path(self):
return "com/android/server/wm/ActivityTaskManagerService.smali"
def patch(self, output, line):
# 写得非常不清真的代码
if ".field mRootWindowContainer:Lcom/android/server/wm/RootWindowContainer;" in line:
output.write(".field public mRootWindowContainer:Lcom/android/server/wm/RootWindowContainer;")
output.write(os.linesep)
self.patched += 1
return True
def get_patch_count(self):
# 打补丁的次数可能会多于3次？
if self.patched > 1:
return self.patched
else:
return 1
class ActivityRecord(Patch):
# 此类和IntentResolver类类似 不再详解
patched = 0
method_name_sp = ""
astActivityRecordRef=""
extrapatchcount=0
def get_path(self):
return "com/android/server/wm/ActivityRecord.smali"
def patch(self, output, line):
# 写得非常不清真的代码
if line.startswith(".method"):
self.method_name_sp = self.find_method_name(line.strip())
if self.method_name_sp == "destroyImmediately":
if "iget-object" in line:
if "Lcom/android/server/wm/ActivityRecord;->" in line:
self.lastActivityRecordRef=re.findall("iget-object [pv][0-9]+, ([pv][0-9]+)",line)[0]
if "Lcom/android/server/wm/ClientLifecycleManager;->scheduleTransaction(Landroid/app/IApplicationThread;Landroid/os/IBinder;Landroid/app/servertransaction/ActivityLifecycleItem;)V" in line:
output.write(line)
output.write(os.linesep)
argument = self.get_method_arguments(line)[0]
output.write("iget-object %s, %s, Lcom/android/server/wm/ActivityRecord;->appToken:Lcom/android/server/wm/ActivityRecord$Token;" %(argument,self.lastActivityRecordRef))
output.write(os.linesep)
output.write("invoke-static/range {%s .. %s}, Lcom/android/server/am/PreventRunningUtils;->onDestroyActivity(Landroid/os/IBinder;)V" %(argument, argument))
output.write(os.linesep)
self.patched += 1
return True
if "Lcom/android/server/wm/ActivityTaskManagerServiceInjector;->destroyActivity(Landroid/content/pm/ActivityInfo;)V" in line:
output.write(os.linesep)
self.patched += 1
self.extrapatchcount+=1
return True
if ".class final Lcom/android/server/wm/ActivityRecord;" in line:
output.write(".class public final Lcom/android/server/wm/ActivityRecord;")
output.write(os.linesep)
self.patched += 1
return True
if ".class public final Lcom/android/server/wm/ActivityRecord;" in line:
output.write(line)
output.write(os.linesep)
self.patched += 1
return True
if ".method static forTokenLocked(Landroid/os/IBinder;)Lcom/android/server/wm/ActivityRecord;" in line:
output.write(".method public static forTokenLocked(Landroid/os/IBinder;)Lcom/android/server/wm/ActivityRecord;")
output.write(os.linesep)
self.patched += 1
return True
if ".method getStack()Lcom/android/server/wm/ActivityStack;" in line:
output.write(".method public getStack()Lcom/android/server/wm/ActivityStack;")
output.write(os.linesep)
self.patched += 1
return True
if ".field app:Lcom/android/server/wm/WindowProcessController;" in line:
output.write(".field public app:Lcom/android/server/wm/WindowProcessController;")
output.write(os.linesep)
self.patched += 1
return True
if ".field final info:Landroid/content/pm/ActivityInfo;" in line:
output.write(".field public final info:Landroid/content/pm/ActivityInfo;")
output.write(os.linesep)
self.patched += 1
return True
if ".field final packageName:Ljava/lang/String;" in line:
output.write(".field public final packageName:Ljava/lang/String;")
output.write(os.linesep)
self.patched += 1
return True
if ".field public final packageName:Ljava/lang/String;" in line:
output.write(line)
output.write(os.linesep)
self.patched += 1
return True
def get_patch_count(self):
# 打补丁的次数可能会多于3次？
sholdpatchcount=7
if self.extrapatchcount>0:
sholdpatchcount+=self.extrapatchcount
if self.patched > sholdpatchcount:
return self.patched
else:
return sholdpatchcount
class ActivityStackSupervisor(Patch):
# 同上 不再详解
patched = 0
extra_count = 0
lastActivityRecordRef=""
methods = None
fixing = {'cleanUpRemovedTaskLocked','cleanUpRemovedTaskLocked$Pr'}
arg_sp = ""
def __init__(self, dir_services=None, dir_apk='apk'):
if dir_services is None:
super(ActivityStackSupervisor, self).__init__()
else:
super(ActivityStackSupervisor, self).__init__(dir_services)
self.dir_apk = dir_apk
self.methods = self.init_pr_methods(self.fixing, self.get_path(), self.dir_apk)
def get_path(self):
return "com/android/server/wm/ActivityStackSupervisor.smali"
def patch(self, output, line):
line_strip = line.strip()
if not line_strip:
return False
if line_strip.startswith(".line"):
return False
if line.startswith(".method"):
method_signature = line_strip
if method_signature in self.methods:
method_name, method_body = self.methods[method_signature]
#print(self.methods)
if method_name in self.fixing:
output.write(method_body)
output.write(os.linesep)
# 注意！
# 这里不仅要把旧的method体改成任意名字使其废弃
output.write(line.replace(method_name, "private %s$Pr" %method_name))
self.fixing.remove(method_name)
self.extra_count += 1
self.patched += 1
return True
if "iget-object" in line:
if "Lcom/android/server/wm/ActivityRecord;->" in line:
self.lastActivityRecordRef=re.findall("iget-object [pv][0-9]+, ([pv][0-9]+)",line)[0]
if "Lcom/android/server/wm/ClientLifecycleManager;->scheduleTransaction(Landroid/app/servertransaction/ClientTransaction;)V" in line:
output.write(line)
output.write(os.linesep)
arguments = self.get_method_arguments(line)
argument = arguments[0]
output.write("iget-object %s, %s, Lcom/android/server/wm/ActivityRecord;->appToken:Lcom/android/server/wm/ActivityRecord$Token;" %(argument,self.lastActivityRecordRef))
output.write(os.linesep)
output.write(" invoke-static/range {%s .. %s},"
" Lcom/android/server/am/PreventRunningUtils;->onLaunchActivity(Landroid/os/IBinder;)V" % (argument, argument))
output.write(os.linesep)
self.patched += 1
return True
def get_patch_count(self):
if self.extra_count:
global plus_flag
plus_flag = True
# 生成一个flag文件 以帮助bat批处理判断是否是8.x+的services.jar
# 事后删除
# 若并非用于工具自带的批处理 可酌情将下方两行删掉或注释掉
with open("plus_flag.txt", "w") as f:
f.write("plus_flag")
# 7.x以及7.x之前: 1处或N处(加行xN, N>=1)
# 8.x: 1+1处或1+N处(方法Hook + 加行xN, N>=1)
# 9.0: 1处(方法Hook)
if self.patched > 1:
return self.patched
else:
return 1
class ConnectivityService(Patch):
patched = 0
method_name_sp = ""
def get_path(self):
return "com/android/server/ConnectivityService.smali"
def patch(self, output, line):
if line.startswith(".method"):
self.method_name_sp = self.find_method_name(line.strip())
if self.method_name_sp == "handleAsyncChannelDisconnected":
if ".locals" in line:
output.write(line)
output.write(os.linesep)
output.write(" invoke-static {},"
" Lcom/android/server/am/PreventRunningUtils;"
"->onVpnConnectionDisconnected()V")
output.write(os.linesep)
self.patched += 1
return True
def get_patch_count(self):
if self.patched > 1:
return self.patched
else:
return 1
class Vpn(Patch):
patched = 0
method_name_sp = ""
startTracing = False
def get_path(self):
return "com/android/server/connectivity/Vpn.smali"
def patch(self, output, line):
if line.startswith(".method"):
self.method_name_sp = self.find_method_name(line.strip())
if self.method_name_sp == "establish":
if "Established by " in line:
self.startTracing=True
return False
if self.startTracing:
if len(re.findall("iget-object [pv][1-9]+, [pv][1-9]+, Lcom/android/internal/net/VpnConfig;->user:Ljava/lang/String;",line))>0:
ref=re.findall("iget-object ([pv][1-9]+), [pv][1-9]+, Lcom/android/internal/net/VpnConfig;->user:Ljava/lang/String;",line)[0]
output.write(line)
output.write(os.linesep)
output.write(" invoke-static {%s},"
" Lcom/android/server/am/PreventRunningUtils;"
"->onActivityEstablishVpnConnection(Ljava/lang/String;)V" %(ref))
output.write(os.linesep)
self.patched += 1
self.startTracing=False
return True
def get_patch_count(self):
if self.patched > 1:
return self.patched
else:
return 1
class MediaFocusControl(Patch):
patched = 0
method_name_sp = ""
def get_path(self):
return "com/android/server/audio/MediaFocusControl.smali"
def patch(self, output, line):
if line.startswith(".method"):
self.method_name_sp = self.find_method_name(line.strip())
if self.method_name_sp == "requestAudioFocus":
if ".locals" in line:
output.write(line)
output.write(os.linesep)
output.write("invoke-static {}, Landroid/os/Binder;->getCallingUid()I")
output.write(os.linesep)
output.write("move-result v1")
output.write(os.linesep)
output.write("invoke-static {}, Landroid/os/Binder;->getCallingPid()I")
output.write(os.linesep)
output.write("move-result v2")
output.write(os.linesep)
output.write(" move-object/16 v3 , p5")
output.write(os.linesep)
output.write(" move-object/16 v4 , p6")
output.write(os.linesep)
output.write(" invoke-static/range {v1 .. v4},"
" Lcom/android/server/am/PreventRunningUtils;"
"->onActivityRequestAudioFocus(IILjava/lang/String;Ljava/lang/String;)V")
output.write(os.linesep)
self.patched += 1
return True
if self.method_name_sp == "abandonAudioFocus":
if ".locals" in line:
output.write(line)
output.write(os.linesep)
output.write("invoke-static {}, Landroid/os/Binder;->getCallingUid()I")
output.write(os.linesep)
output.write("move-result v1")
output.write(os.linesep)
output.write("invoke-static {}, Landroid/os/Binder;->getCallingPid()I")
output.write(os.linesep)
output.write("move-result v2")
output.write(os.linesep)
output.write(" move-object/16 v3 , p2")
output.write(os.linesep)
output.write(" invoke-static {v1 ,v2, v3},"
" Lcom/android/server/am/PreventRunningUtils;"
"->onActivityAbandonAudioFocus(IILjava/lang/String;)V")
output.write(os.linesep)
self.patched += 1
return True
if self.method_name_sp == "removeFocusStackEntryOnDeath":
if "Lcom/android/server/audio/FocusRequester;->release()V" in line:
output.write(line)
output.write(os.linesep)
arguments = self.get_method_arguments(line)
argument = arguments[0]
output.write(" invoke-virtual {%s},"
" Lcom/android/server/audio/FocusRequester;"
"->getClientId()Ljava/lang/String;" %(argument))
output.write(os.linesep)
output.write(" move-result-object %s" %(argument))
output.write(os.linesep)
output.write(" invoke-static {%s},"
" Lcom/android/server/am/PreventRunningUtils;"
"->onActivityLostAudioFocusOnDeath(Ljava/lang/String;)V" %(argument))
output.write(os.linesep)
self.patched += 1
return True
def get_patch_count(self):
if self.patched > 1:
return self.patched
else:
return 1
class ConfigurationContainer(Patch):
# 此类和IntentResolver类类似 不再详解
patched = 0
def get_path(self):
return "com/android/server/wm/ConfigurationContainer.smali"
def patch(self, output, line):
# 写得非常不清真的代码
if ".method protected abstract getChildAt(I)Lcom/android/server/wm/ConfigurationContainer;" in line:
output.write(".method public abstract getChildAt(I)Lcom/android/server/wm/ConfigurationContainer;")
output.write(os.linesep)
self.patched += 1
return True
if ".method protected abstract getChildCount()I" in line:
output.write(".method public abstract getChildCount()I")
output.write(os.linesep)
self.patched += 1
return True
def get_patch_count(self):
# 打补丁的次数可能会多于3次？
if self.patched > 2:
return self.patched
else:
return 2
class RootWindowContainer(Patch):
# 此类和IntentResolver类类似 不再详解
patched = 0
def get_path(self):
return "com/android/server/wm/RootWindowContainer.smali"
def patch(self, output, line):
# 写得非常不清真的代码
if ".method getDefaultTaskDisplayArea()Lcom/android/server/wm/TaskDisplayArea;" in line:
output.write(".method public getDefaultTaskDisplayArea()Lcom/android/server/wm/TaskDisplayArea;")
output.write(os.linesep)
self.patched += 1
return True
def get_patch_count(self):
# 打补丁的次数可能会多于3次？
if self.patched > 1:
return self.patched
else:
return 1
class WindowContainer(Patch):