mirror of
https://github.com/cirosantilli/linux-kernel-module-cheat.git
synced 2025-08-09 04:13:50 +00:00
101 lines
4.7 KiB
Python
Executable File
101 lines
4.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import sys
|
|
|
|
import common
|
|
import lkmc.import_path
|
|
import path_properties
|
|
import signal
|
|
import thread_pool
|
|
|
|
class Main(common.TestCliFunction):
|
|
def __init__(self, *args, **kwargs):
|
|
if not 'description' in kwargs:
|
|
kwargs['description'] = '''\
|
|
Test userland executables in user mode, or baremetal executables in full system
|
|
depending on the value of the --mode option. See also:
|
|
|
|
* https://cirosantilli.com/linux-kernel-module-cheat#user-mode-tests
|
|
* https://cirosantilli.com/linux-kernel-module-cheat#baremetal-tests
|
|
* https://cirosantilli.com/linux-kernel-module-cheat#userland-setup-getting-started-natively
|
|
'''
|
|
if not 'defaults' in kwargs:
|
|
kwargs['defaults'] = {}
|
|
if not 'mode' in kwargs['defaults']:
|
|
kwargs['defaults']['mode'] = 'userland'
|
|
super().__init__(*args, **kwargs)
|
|
self.add_argument(
|
|
'tests',
|
|
nargs='*',
|
|
help='''\
|
|
If given, run only the given tests. Otherwise, run all tests.
|
|
'''
|
|
)
|
|
|
|
def setup_one(self):
|
|
self.env['tests'] = self.resolve_targets(
|
|
[
|
|
self.env['baremetal_source_dir'],
|
|
self.env['userland_source_dir']
|
|
],
|
|
self.env['tests']
|
|
)
|
|
|
|
def timed_main(self):
|
|
run_args = self.get_common_args()
|
|
rootdir_abs_len = len(self.env['root_dir'])
|
|
with thread_pool.ThreadPool(
|
|
self.run_test,
|
|
handle_output=self.handle_output_function,
|
|
nthreads=self.env['nproc'],
|
|
thread_id_arg='thread_id',
|
|
submit_raise_exit=self.env['quit_on_fail'],
|
|
) as my_thread_pool:
|
|
for test in self.env['tests']:
|
|
for path, in_dirnames, in_filenames in self.sh.walk(test):
|
|
path_abs = os.path.abspath(path)
|
|
dirpath_relative_root = path_abs[rootdir_abs_len + 1:]
|
|
for in_filename in in_filenames:
|
|
if os.path.splitext(in_filename)[1] in self.env['build_in_exts']:
|
|
path_relative_root = os.path.join(dirpath_relative_root, in_filename)
|
|
my_path_properties = path_properties.get(path_relative_root)
|
|
if my_path_properties.should_be_tested(self.env):
|
|
cur_run_args = run_args.copy()
|
|
cur_run_args.update({
|
|
self.env['mode']: [os.path.relpath(
|
|
os.path.join(path_abs, in_filename),
|
|
os.getcwd()
|
|
)],
|
|
})
|
|
cur_run_args.update(my_path_properties['test_run_args'])
|
|
if my_path_properties['test_stdin_data'] is not None:
|
|
cur_run_args['stdin_file'] = os.path.join(
|
|
path_abs,
|
|
'test_data',
|
|
my_path_properties['test_stdin_data'] + '.i'
|
|
)
|
|
run_test_args = {
|
|
'expected_exit_status': my_path_properties['exit_status'],
|
|
'run_args': cur_run_args,
|
|
'run_obj': lkmc.import_path.import_path_main('run'),
|
|
'test_id': '{} {}'.format(self.env['mode'], path_relative_root),
|
|
}
|
|
if (my_path_properties['qemu_unimplemented_instruction'] and
|
|
self.env['emulator'] == 'qemu' and
|
|
self.env['mode'] == 'userland'
|
|
):
|
|
run_test_args['expected_exit_status'] = -signal.Signals.SIGILL.value
|
|
my_signal = my_path_properties['signal_received']
|
|
if my_signal is not None:
|
|
if self.env['mode'] == 'baremetal':
|
|
run_test_args['expected_exit_status'] = 128 + my_signal.value
|
|
elif self.env['mode'] == 'userland':
|
|
# Python subprocess reports signals differently from Bash's 128 + signal rule.
|
|
run_test_args['expected_exit_status'] = -my_signal.value
|
|
my_thread_pool.submit(run_test_args)
|
|
return self._handle_thread_pool_errors(my_thread_pool)
|
|
|
|
if __name__ == '__main__':
|
|
Main().cli()
|