forked from ColdWindScholar/MIO-KITCHEN-SOURCE
-
Notifications
You must be signed in to change notification settings - Fork 0
/
contextpatch.py
81 lines (73 loc) · 3.09 KB
/
contextpatch.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
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
from re import sub
from difflib import SequenceMatcher
fix_permission = {"/vendor/bin/hw/android.hardware.wifi@1.0": "u:object_r:hal_wifi_default_exec:s0"}
def scan_context(file) -> dict: # 读取context文件返回一个字典
context = {}
with open(file, "r", encoding='utf-8') as file_:
for i in file_.readlines():
filepath, *other = i.strip().replace('\\', '').split()
context[filepath] = other
if len(other) > 1:
print(f"[Warn] {i[0]} has too much data.")
return context
def scan_dir(folder) -> list: # 读取解包的目录,返回一个字典
part_name = os.path.basename(folder)
allfiles = ['/', '/lost+found', f'/{part_name}/lost+found', f'/{part_name}', f'/{part_name}/']
for root, dirs, files in os.walk(folder, topdown=True):
for dir_ in dirs:
yield os.path.join(root, dir_).replace(folder, '/' + part_name).replace('\\', '/')
for file in files:
yield os.path.join(root, file).replace(folder, '/' + part_name).replace('\\', '/')
for rv in allfiles:
yield rv
def context_patch(fs_file, dir_path) -> tuple: # 接收两个字典对比
new_fs = {}
r_new_fs = {}
add_new = 0
permission_d = None
print("ContextPatcher: Load origin %d" % (len(fs_file.keys())) + " entries")
try:
permission_d = fs_file.get(list(fs_file)[5])
except IndexError:
...
if not permission_d:
permission_d = [f'u:object_r:{os.path.basename(dir_path)}_file:s0']
for i in scan_dir(os.path.abspath(dir_path)):
if not i.isprintable():
tmp = ''
for c in i:
tmp += c if c.isprintable() else '*'
i = tmp
if ' ' in i:
i = i.replace(' ', '*')
if fs_file.get(i):
new_fs[sub(r'([^-_/a-zA-Z0-9])', r'\\\1', i)] = fs_file[i]
else:
permission = permission_d
if r_new_fs.get(i):
continue
if i:
if i in fix_permission.keys():
permission = fix_permission[i]
else:
for e in fs_file.keys():
if SequenceMatcher(None, (path := os.path.dirname(i)), e).quick_ratio() >= 0.85:
if e == path:
continue
permission = fs_file[e]
break
else:
permission = permission_d
print(f"ADD [{i} {permission}]")
add_new += 1
r_new_fs[i] = permission
new_fs[sub(r'([^-_/a-zA-Z0-9])', r'\\\1', i)] = permission
return new_fs, add_new
def main(dir_path, fs_config) -> None:
new_fs, add_new = context_patch(scan_context(os.path.abspath(fs_config)), dir_path)
with open(fs_config, "w+", encoding='utf-8', newline='\n') as f:
f.writelines([i + " " + " ".join(new_fs[i]) + "\n" for i in sorted(new_fs.keys())])
print('ContextPatcher: Add %d' % add_new + " entries")