debug: add pretty printer for boot modules registry

4 files changed, 28 insertions, 2 deletions

diff --git a/kapi/gdb/__init__.py b/kapi/gdb/__init__.py
index 88b1d05..1b36753 100644
--- a/kapi/gdb/__init__.py
+++ b/kapi/gdb/__init__.py
@@ -23,6 +23,11 @@ def build_pretty_printers():
         "^kapi::boot_modules::boot_module$",
         KapiBootModulesBootModulePrinter,
     )
+    pp.add_printer(
+        "kapi_boot_modules_boot_module_registry",
+        "^kapi::boot_modules::boot_module_registry$",
+        KapiBootModulesBootModuleRegistryPrinter,
+    )
     return pp
 
 
diff --git a/kapi/gdb/boot_modules/__init__.py b/kapi/gdb/boot_modules/__init__.py
index fedab65..2a54136 100644
--- a/kapi/gdb/boot_modules/__init__.py
+++ b/kapi/gdb/boot_modules/__init__.py
@@ -1 +1,2 @@
 from .boot_module import KapiBootModulesBootModulePrinter
+from .boot_module_registry import KapiBootModulesBootModuleRegistryPrinter
diff --git a/kapi/gdb/boot_modules/boot_module.py b/kapi/gdb/boot_modules/boot_module.py
index f0d558b..b26ecf1 100644
--- a/kapi/gdb/boot_modules/boot_module.py
+++ b/kapi/gdb/boot_modules/boot_module.py
@@ -9,10 +9,10 @@ class KapiBootModulesBootModulePrinter(gdb.ValuePrinter):
         self.__start = val["start_address"]
         self.__size = val["size"]
         self.__pointer_type = gdb.lookup_type("std::byte").pointer()
-        self.__pretty_name = self.__name if str(self.__name) != '""' else "<no name>"
+        self.__pretty_name = " " + str(self.__name) if str(self.__name) != '""' else ""
 
     def to_string(self):
-        return f"boot module {self.__pretty_name} of size {format_size(int(self.__size))}, at {self.__start.cast(self.__pointer_type)}"
+        return f"boot module{self.__pretty_name} of size {format_size(int(self.__size))}, at {self.__start.cast(self.__pointer_type)}"
 
     def children(self):
         yield ("name", self.__name)
diff --git a/kapi/gdb/boot_modules/boot_module_registry.py b/kapi/gdb/boot_modules/boot_module_registry.py
new file mode 100644
index 0000000..599a823
--- /dev/null
+++ b/kapi/gdb/boot_modules/boot_module_registry.py
@@ -0,0 +1,20 @@
+import gdb
+from teachos import format_size
+
+
+class KapiBootModulesBootModuleRegistryPrinter(gdb.ValuePrinter):
+    def __init__(self, val: gdb.Value):
+        self.__val = val
+        self.__modules = val["m_modules"]
+        self.__size = int(self.__modules["m_size"])
+        self.__element_type = gdb.lookup_type("kapi::boot_modules::boot_module")
+
+    def to_string(self):
+        return f"boot module registry of size {self.__size}"
+
+    def children(self):
+        yield ("[size]", self.__size)
+        yield ("m_modules", self.__modules)
+
+    def display_hint(self):
+        return None