staticmethods to get syscall info

trailofbits · Apr 30, 2021 · ba5597c · ba5597c
1 parent 101596e
commit ba5597c
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/manticore/platforms/linux.py b/manticore/platforms/linux.py
@@ -3329,6 +3329,21 @@ def _interp_total_size(interp):
         last = load_segs[-1]
         return last.header.p_vaddr + last.header.p_memsz
 
+    @staticmethod
+    def implemented_syscalls() -> Iterable[str]:
+        import inspect
+
+        return (
+            x[0].split("sys_", 1)[1]
+            for x in inspect.getmembers(Linux, predicate=inspect.isfunction)
+            if x[0].startswith("sys_")
+        )
+
+    @staticmethod
+    def print_implemented_syscalls() -> None:
+        for syscall in Linux.implemented_syscalls():
+            print(syscall)
+
 
 ############################################################################
 # Symbolic versions follows

diff --git a/manticore/platforms/linux_syscall_stubs.py b/manticore/platforms/linux_syscall_stubs.py
@@ -1,3 +1,5 @@
+from typing import Iterable
+
 from ..platforms.platform import SyscallNotImplemented, unimplemented
 from .linux_syscalls import amd64
 
@@ -1187,3 +1189,18 @@ def sys_wait4(self, upid, stat_addr, options, ru) -> int:
     def sys_waitid(self, which, upid, infop, options, ru) -> int:
         """ AUTOGENERATED UNIMPLEMENTED STUB """
         return self.simple_returns()
+
+    @staticmethod
+    def unimplemented_syscalls() -> Iterable[str]:
+        import inspect
+
+        return (
+            x[0].split("sys_", 1)[1]
+            for x in inspect.getmembers(SyscallStubs, predicate=inspect.isfunction)
+            if x[0].startswith("sys_")
+        )
+
+    @staticmethod
+    def print_unimplemented_syscalls() -> None:
+        for syscall in SyscallStubs.unimplemented_syscalls():
+            print(syscall)