mirror of
https://github.com/cirosantilli/linux-kernel-module-cheat.git
synced 2026-01-25 19:21:35 +01:00
3cc1b793cb
Please python stop torturing me with refactors. Make ./run -u blow up if executable not found, otherwise I go crazy. Get ./test-gdb back to life after the ./run relative path refactor, forgot to test this.
58 lines
2.0 KiB
Python
Executable File
58 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import common
|
|
from shell_helpers import LF
|
|
|
|
class Main(common.LkmcCliFunction):
|
|
def __init__(self):
|
|
super().__init__(
|
|
description='''Trace the PIC addresses executed on a Linux kernel boot.
|
|
|
|
More information at: https://github.com/cirosantilli/linux-kernel-module-cheat#tracing
|
|
'''
|
|
)
|
|
|
|
def timed_main(self):
|
|
args = self.get_common_args()
|
|
run = common.import_path_main('run')
|
|
if self.env['emulator'] == 'gem5':
|
|
args['trace'] = 'Exec,-ExecSymbol,-ExecMicro'
|
|
run.main(**args)
|
|
elif self.env['emulator'] == 'qemu':
|
|
run_args = args.copy()
|
|
run_args['trace'] = 'exec_tb'
|
|
run_args['quit_after_boot'] = True
|
|
run.main(**run_args)
|
|
qemu_trace2txt = common.import_path_main('qemu-trace2txt')
|
|
qemu_trace2txt.main(**args)
|
|
# Instruction count.
|
|
# We could put this on a separate script, but it just adds more arch boilerplate to a new script.
|
|
# So let's just leave it here for now since it did not add a significant processing time.
|
|
kernel_entry_addr = hex(self.get_elf_entry(self.env['vmlinux']))
|
|
nlines = 0
|
|
nlines_firmware = 0
|
|
with open(self.env['qemu_trace_txt_file'], 'r') as trace_file:
|
|
in_firmware = True
|
|
for line in trace_file:
|
|
line = line.rstrip()
|
|
nlines += 1
|
|
pc = line.split('=')[-1]
|
|
if pc == kernel_entry_addr:
|
|
in_firmware = False
|
|
if in_firmware:
|
|
nlines_firmware += 1
|
|
print('''\
|
|
instructions {}
|
|
entry_address {}
|
|
instructions_firmware {}
|
|
'''.format(
|
|
nlines,
|
|
kernel_entry_addr,
|
|
nlines_firmware
|
|
),
|
|
end=''
|
|
)
|
|
|
|
if __name__ == '__main__':
|
|
Main().cli()
|