feat: implement to_bytes and from_bytes

Still have a circular import to solve though

Author: MrIndeciso

libdestruct/c/c_int.py

@@ -19,6 +19,13 @@ def get(self: c_int) -> int:
         """Return the value of the integer."""
         return int.from_bytes(self.memory[self.address : self.address + self.size], self.endianness, signed=True)
 
+    def to_bytes(self: obj) -> bytes:
+        """Return the serialized representation of the object."""
+        if self._frozen:
+            return self._frozen_value.to_bytes(self.size, self.endianness, signed=True)
+
+        return self.memory[self.address : self.address + self.size]
+
     def _set(self: c_int, value: int) -> None:
         """Set the value of the integer to the given value."""
         self.memory[self.address : self.address + self.size] = value.to_bytes(self.size, self.endianness, signed=True)

libdestruct/c/c_long.py

@@ -19,6 +19,13 @@ def get(self: c_long) -> int:
         """Return the value of the integer."""
         return int.from_bytes(self.memory[self.address : self.address + self.size], self.endianness, signed=True)
 
+    def to_bytes(self: obj) -> bytes:
+        """Return the serialized representation of the object."""
+        if self._frozen:
+            return self._frozen_value.to_bytes(self.size, self.endianness, signed=True)
+
+        return self.memory[self.address : self.address + self.size]
+
     def _set(self: c_long, value: int) -> None:
         """Set the value of the integer to the given value."""
         self.memory[self.address : self.address + self.size] = value.to_bytes(self.size, self.endianness, signed=True)

libdestruct/c/c_str.py

@@ -35,6 +35,10 @@ def get(self: c_str, index: int = -1) -> bytes:
 
         return bytes([self.memory[self.address + index]])
 
+    def to_bytes(self: obj) -> bytes:
+        """Return the serialized representation of the object."""
+        return self.memory[self.address : self.address + self.size()]
+
     def _set(self: c_str, value: bytes, index: int = -1) -> None:
         """Set the character at the given index to the given value."""
         if index != -1 and index < 0 or index >= self.size():

libdestruct/c/c_uint.py

@@ -19,6 +19,13 @@ def get(self: c_uint) -> int:
         """Return the value of the integer."""
         return int.from_bytes(self.memory[self.address : self.address + self.size], self.endianness, signed=False)
 
+    def to_bytes(self: obj) -> bytes:
+        """Return the serialized representation of the object."""
+        if self._frozen:
+            return self._frozen_value.to_bytes(self.size, self.endianness, signed=False)
+
+        return self.memory[self.address : self.address + self.size]
+
     def _set(self: c_uint, value: int) -> None:
         """Set the value of the integer to the given value."""
         self.memory[self.address : self.address + self.size] = value.to_bytes(self.size, self.endianness, signed=False)

libdestruct/c/c_ulong.py

@@ -19,6 +19,13 @@ def get(self: c_ulong) -> int:
         """Return the value of the integer."""
         return int.from_bytes(self.memory[self.address : self.address + self.size], self.endianness, signed=False)
 
+    def to_bytes(self: obj) -> bytes:
+        """Return the serialized representation of the object."""
+        if self._frozen:
+            return self._frozen_value.to_bytes(self.size, self.endianness, signed=False)
+
+        return self.memory[self.address : self.address + self.size]
+
     def _set(self: c_ulong, value: int) -> None:
         """Set the value of the integer to the given value."""
         self.memory[self.address : self.address + self.size] = value.to_bytes(self.size, self.endianness, signed=False)

libdestruct/common/obj.py

@@ -63,6 +63,17 @@ def get(self: obj) -> object:
     def _set(self: obj, value: object) -> None:
         """Set the value of the object to the given value."""
 
+    @abstractmethod
+    def to_bytes(self: obj) -> bytes:
+        """Serialize the object to bytes."""
+
+    @classmethod
+    def from_bytes(cls, data: bytes) -> obj:
+        """Deserialize the object from bytes."""
+        item = cls(data, 0)
+        item.freeze()
+        return item
+
     def set(self: obj, value: object) -> None:
         """Set the value of the object to the given value."""
         if self._frozen:
@@ -108,3 +119,7 @@ def __eq__(self, value: object) -> bool:
             return False
 
         return self.get() == value.get()
+
+    def __bytes__(self: obj) -> bytes:
+        """Return the serialized object."""
+        return self.to_bytes()

libdestruct/common/ptr.py

@@ -30,6 +30,13 @@ def get(self: ptr) -> int:
         """Return the value of the pointer."""
         return int.from_bytes(self.memory[self.address : self.address + self.size], self.endianness)
 
+    def to_bytes(self: obj) -> bytes:
+        """Return the serialized representation of the object."""
+        if self._frozen:
+            return self._frozen_value.to_bytes(self.size, self.endianness)
+
+        return self.memory[self.address : self.address + self.size]
+
     def _set(self: ptr, value: int) -> None:
         """Set the value of the pointer to the given value."""
         self.memory[self.address : self.address + self.size] = value.to_bytes(self.size, self.endianness)

libdestruct/common/struct/struct.py

@@ -6,12 +6,25 @@
 
 from __future__ import annotations
 
+from typing import TYPE_CHECKING
+
 from libdestruct.common.obj import obj
 
+if TYPE_CHECKING:
+    from libdestruct.common.struct.struct_impl import struct_impl
 
 class struct(obj):
     """A C struct."""
 
     def __init__(self: obj) -> None:
         """Initialize the struct."""
         raise RuntimeError("This type should not be directly instantiated.")
+
+    @classmethod
+    def from_bytes(cls: type[struct], data: bytes) -> struct_impl:
+        """Create a struct from a serialized representation."""
+        from libdestruct import inflater
+
+        type_inflater = inflater(data)
+
+        return type_inflater.inflate_struct(cls, 0)

libdestruct/common/struct/struct_impl.py

@@ -56,6 +56,10 @@ def get(self: struct_impl) -> str:
         """Return the value of the struct."""
         return f"{self.name}(address={self.address}, size={self.size})"
 
+    def to_bytes(self: struct_impl) -> bytes:
+        """Return the serialized representation of the struct."""
+        return b"".join(member.to_bytes() for member in self._members.values())
+
     def _set(self: struct_impl, _: str) -> None:
         """Set the value of the struct to the given value."""
         raise RuntimeError("Cannot set the value of a struct.")