5 files changed, 134 insertions, 0 deletions
diff --git a/libs/kstd/gdb/__init__.py b/libs/kstd/gdb/__init__.py
new file mode 100644
index 0000000..fc5e8fb
--- /dev/null
+++ b/libs/kstd/gdb/__init__.py
@@ -0,0 +1,20 @@
+import gdb.printing
+
+from .vector import KstdVectorPrinter
+from .string import KstdStringPrinter
+from .std_types import StdBytePrinter
+from .smart_pointers import KstdUniquePtrPrinter, KstdSharedPtrPrinter
+
+
+def build_pretty_printers():
+    pp = gdb.printing.RegexpCollectionPrettyPrinter("kstd")
+    pp.add_printer("vector", "^kstd::vector<.*>$", KstdVectorPrinter)
+    pp.add_printer("string", "^kstd::string$", KstdStringPrinter)
+    pp.add_printer("std_byte", "^std::byte$", StdBytePrinter)
+    pp.add_printer("unique_ptr", "^kstd::unique_ptr<.*>$", KstdUniquePtrPrinter)
+    pp.add_printer("shared_ptr", "^kstd::shared_ptr<.*>$", KstdSharedPtrPrinter)
+    return pp
+
+
+def register_printers(objfile):
+    gdb.printing.register_pretty_printer(objfile, build_pretty_printers(), replace=True)
diff --git a/libs/kstd/gdb/smart_pointers.py b/libs/kstd/gdb/smart_pointers.py
new file mode 100644
index 0000000..6e4a4f9
--- /dev/null
+++ b/libs/kstd/gdb/smart_pointers.py
@@ -0,0 +1,50 @@
+import gdb
+
+
+class KstdUniquePtrPrinter:
+
+    def __init__(self, val):
+        self.val = val
+        self.type = val.type.template_argument(0)
+
+    def to_string(self):
+        pointer = self.val["pointer"]
+        if int(pointer) == 0:
+            return f"kstd::unique_ptr<{self.type}> (empty)"
+        return f"kstd::unique_ptr<{self.type}>"
+
+    def children(self):
+        pointer = self.val["pointer"]
+        if int(pointer) != 0:
+            yield ("get()", pointer)
+            yield ("*get()", pointer.dereference())
+
+
+class KstdSharedPtrPrinter:
+
+    def __init__(self, val):
+        self.val = val
+        self.type = val.type.template_argument(0)
+
+    def to_string(self):
+        pointer = self.val["pointer"]
+        control_block = self.val["control"]
+
+        if int(pointer) == 0 or int(control_block) == 0:
+            return f"kstd::shared_ptr<{self.type}> (empty)"
+
+        strong_refs = int(control_block["shared_count"]["_M_i"])
+        weak_refs = int(control_block["weak_count"]["_M_i"])
+
+        return f"kstd::shared_ptr<{self.type}> (use_count={strong_refs}, weak_count={weak_refs})"
+
+    def children(self):
+        pointer = self.val["pointer"]
+        control_block = self.val["control"]
+
+        if int(pointer) != 0:
+            yield ("get()", pointer)
+            yield ("*get()", pointer.dereference())
+
+        if int(control_block) != 0:
+            yield ("[control_block]", control_block.dereference())
diff --git a/libs/kstd/gdb/std_types.py b/libs/kstd/gdb/std_types.py
new file mode 100644
index 0000000..78d094c
--- /dev/null
+++ b/libs/kstd/gdb/std_types.py
@@ -0,0 +1,15 @@
+import gdb
+
+
+class StdBytePrinter:
+
+    def __init__(self, val):
+        self.val = val
+
+    def to_string(self):
+        try:
+            uint8_type = gdb.lookup_type("unsigned char")
+            numeric_value = int(self.val.cast(uint8_type))
+            return f"{numeric_value:#04x}"
+        except gdb.error:
+            return f"<optimized out or unreadable>"
diff --git a/libs/kstd/gdb/string.py b/libs/kstd/gdb/string.py
new file mode 100644
index 0000000..8230b21
--- /dev/null
+++ b/libs/kstd/gdb/string.py
@@ -0,0 +1,27 @@
+import gdb
+
+
+class KstdStringPrinter:
+
+    def __init__(self, val):
+        self.val = val
+
+    def to_string(self):
+        storage = self.val["m_storage"]
+        storage_size = int(storage["m_size"])
+
+        if storage_size <= 0:
+            return '""'
+
+        data_pointer = storage["m_data"]
+        string_length = storage_size - 1
+
+        try:
+            if hasattr(data_pointer, "lazy_string"):
+                return data_pointer.lazy_string(encoding="utf-8", length=string_length)
+            return data_pointer.string(encoding="utf-8", length=string_length)
+        except gdb.error:
+            return "<unreadable memory>"
+
+    def display_hint(self):
+        return "string"
diff --git a/libs/kstd/gdb/vector.py b/libs/kstd/gdb/vector.py
new file mode 100644
index 0000000..597ffdc
--- /dev/null
+++ b/libs/kstd/gdb/vector.py
@@ -0,0 +1,22 @@
+import gdb
+
+
+class KstdVectorPrinter:
+
+    def __init__(self, val):
+        self.val = val
+        self.type = val.type.template_argument(0)
+
+    def to_string(self):
+        size = int(self.val["m_size"])
+        capacity = int(self.val["m_capacity"])
+        return f"kstd::vector of length {size}, capacity {capacity}"
+
+    def children(self):
+        size = int(self.val["m_size"])
+        data_pointer = self.val["m_data"]
+        for i in range(size):
+            yield (f"[{i}]", (data_pointer + i).dereference())
+
+    def display_hint(self):
+        return "array"