ptrace-syscalls 0.0.0-experimental.3

Analyze syscalls via ptrace
Documentation 
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

#!/usr/bin/env -S uv run

import system_calls
import os


def generate_syscall_number_test(rust_arch, arch):
    test_header = f"""
  #[cfg(target_arch = "{rust_arch}")]
  #[test]
  fn test_{rust_arch}_syscall_numbers() \u007b
"""
    test_footer = """
  }
    """
    test_body = ""
    for name, value in system_calls.syscalls().load_arch_table(arch).items():
        if name not in {"epoll_ctl_old", "epoll_wait_old"}:
            test_body += f"    assert_eq!(ptrace_syscalls::SYS_{name}, {value});\n"
    return test_header + test_body + test_footer

def write_syscall_number_test():
    me = __file__
    test_file = os.path.realpath(f"{me}/../../tests/syscall_number_tests.rs")
    header = """
/// Automatically generated syscall number tests by generate_syscall_number_test.py
#[cfg(test)]
mod syscall_number_tests \u007b
    """
    footer = """
}
"""
    with open(test_file, "w+") as f:
        f.write(header)
        for (rust_arch, arch) in [("riscv64", "riscv64"), ("x86_64", "x86_64"), ("aarch64", "arm64")]:
            print(f"Generating syscall number test for {arch} (rust: {rust_arch})")
            f.write(generate_syscall_number_test(rust_arch, arch))
        f.write(footer)

if __name__ == '__main__':
    write_syscall_number_test()