diff options
| -rw-r--r-- | kapi/gdb/boot_modules/boot_module.py | 35 |
1 files changed, 28 insertions, 7 deletions
diff --git a/kapi/gdb/boot_modules/boot_module.py b/kapi/gdb/boot_modules/boot_module.py index b26ecf1..97e6584 100644 --- a/kapi/gdb/boot_modules/boot_module.py +++ b/kapi/gdb/boot_modules/boot_module.py @@ -3,21 +3,42 @@ from teachos import format_size class KapiBootModulesBootModulePrinter(gdb.ValuePrinter): + class Iterator: + def __init__(self, begin: gdb.Value, end: gdb.Value): + self._item = begin + self._end = end + self._count = 0 + + def __iter__(self): + return self + + def __next__(self): + count = self._count + self._count = count + 1 + + if self._item == self._end: + raise StopIteration + + element = self._item.dereference() + self._item += 1 + return (f"[{count}]", element) + def __init__(self, val): + self.__pointer_type = gdb.lookup_type("std::byte").pointer() + self.__val = val self.__name = val["name"] self.__start = val["start_address"] - self.__size = val["size"] - self.__pointer_type = gdb.lookup_type("std::byte").pointer() + self.__size = int(val["size"]) + self.__begin = val["start_address"]["m_value"].cast(self.__pointer_type) + self.__end = self.__begin + self.__size 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(self.__size)}, at {self.__start.cast(self.__pointer_type)}" def children(self): - yield ("name", self.__name) - yield ("start_address", self.__start) - yield ("size", self.__size) + return self.Iterator(self.__begin, self.__end) def display_hint(self): - return None + return "array" |