You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
108 lines
3.1 KiB
108 lines
3.1 KiB
#!/usr/bin/env python
|
|
|
|
import os
|
|
import sys
|
|
import argparse
|
|
import shutil
|
|
|
|
|
|
parser = argparse.ArgumentParser(description='Android system files extractor')
|
|
parser.add_argument("-p", "--prefix", metavar="NAME", required=True,
|
|
help="Prefix for stored files, e.g. galaxy-s7-us")
|
|
|
|
|
|
SYSTEM_FILES = [
|
|
"/proc/cpuinfo",
|
|
"/sys/devices/system/cpu/kernel_max",
|
|
"/sys/devices/system/cpu/possible",
|
|
"/sys/devices/system/cpu/present",
|
|
]
|
|
|
|
CPU_FILES = [
|
|
"cpufreq/cpuinfo_max_freq",
|
|
"cpufreq/cpuinfo_min_freq",
|
|
"topology/physical_package_id",
|
|
"topology/core_siblings_list",
|
|
"topology/core_id",
|
|
"topology/thread_siblings_list",
|
|
]
|
|
|
|
CACHE_FILES = [
|
|
"allocation_policy",
|
|
"coherency_line_size",
|
|
"level",
|
|
"number_of_sets",
|
|
"shared_cpu_list",
|
|
"size",
|
|
"type",
|
|
"ways_of_associativity",
|
|
"write_policy",
|
|
]
|
|
|
|
def c_escape(string):
|
|
c_string = ""
|
|
for c in string:
|
|
if c == "\\":
|
|
c_string += "\\\\"
|
|
elif c == "\"":
|
|
c_string += "\\\""
|
|
elif c == "\t":
|
|
c_string += "\\t"
|
|
elif c == "\n":
|
|
c_string += "\\n"
|
|
elif c == "\r":
|
|
c_string += "\\r"
|
|
elif ord(c) == 0:
|
|
c_string += "\\0"
|
|
elif 32 <= ord(c) < 127:
|
|
c_string += c
|
|
else:
|
|
c_string += "x%02X" % ord(c)
|
|
return c_string
|
|
|
|
|
|
def dump_system_file(stream, path):
|
|
try:
|
|
with open(path, "rb") as device_file:
|
|
content = device_file.read()
|
|
stream.write("\t{\n")
|
|
stream.write("\t\t.path = \"%s\",\n" % path)
|
|
stream.write("\t\t.size = %d,\n" % len(content))
|
|
if len(content.splitlines()) > 1:
|
|
stream.write("\t\t.content =")
|
|
for line in content.splitlines(True):
|
|
stream.write("\n\t\t\t\"%s\"" % c_escape(line))
|
|
stream.write(",\n")
|
|
else:
|
|
stream.write("\t\t.content = \"%s\",\n" % c_escape(content))
|
|
stream.write("\t},\n")
|
|
return True
|
|
except IOError:
|
|
pass
|
|
|
|
|
|
def main(args):
|
|
options = parser.parse_args(args)
|
|
|
|
# with open(os.path.join("test", "dmesg", options.prefix + ".log"), "w") as dmesg_log:
|
|
# dmesg_log.write(device.Shell("dmesg"))
|
|
with open(os.path.join("test", options.prefix + ".h"), "w") as file_header:
|
|
file_header.write("struct cpuinfo_mock_file filesystem[] = {\n")
|
|
for path in SYSTEM_FILES:
|
|
dump_system_file(file_header, path)
|
|
for cpu in range(16):
|
|
for filename in CPU_FILES:
|
|
path = "/sys/devices/system/cpu/cpu%d/%s" % (cpu, filename)
|
|
dump_system_file(file_header, path)
|
|
for index in range(10):
|
|
for filename in CACHE_FILES:
|
|
path = "/sys/devices/system/cpu/cpu%d/cache/index%d/%s" % (cpu, index, filename)
|
|
dump_system_file(file_header, path)
|
|
file_header.write("\t{ NULL },\n")
|
|
file_header.write("};\n")
|
|
shutil.copy("/proc/cpuinfo",
|
|
os.path.join("test", "cpuinfo", options.prefix + ".log"))
|
|
|
|
if __name__ == "__main__":
|
|
main(sys.argv[1:])
|