🗓 May 2, 2025 8:44:49 PM

✨ int_to_bytes

Author: securisec

TODO

@@ -27,6 +27,7 @@ New ideas:
   ☐ ✨ whitespace encoding https://www.dcode.fr/whitespace-language
   ☐ beaufort
   ☐ 🔥 update python_requires in setup.py on python version change
+  ☐ update base85 https://github.com/gorakhargosh/mom/blob/master/mom/codec/base85.py#L343
 
 Bug:
 

chepy/modules/dataformat.py

@@ -750,6 +750,24 @@ def str_to_hex(self, delimiter: Union[str, bytes] = b"") -> DataFormatT:
         self.state = out
         return self
 
+    @ChepyDecorators.call_stack
+    def int_to_bytes(self, length: Union[int, None]=None, byteorder: str = 'big'):
+        """Int to bytes conversion
+
+        Args:
+            length (int, optional): Length of bytes. If not specified, byteorder is ignored and long_to_bytes is used. 
+            byteorder (str, optional): Endianness. Defaults to 'big'.
+
+        Returns:
+            Chepy: The Chepy object.
+        """
+        data = self._convert_to_int()
+        if not length:
+            self.state = crypto_number.long_to_bytes(self.state)
+        else:
+            self.state = data.to_bytes(int(length), byteorder=byteorder)
+        return self
+
     @ChepyDecorators.call_stack
     def int_to_hex(self, unsigned_int: bool = True, bit_size: int = 64) -> DataFormatT:
         """Converts an integer into its hex equivalent

chepy/modules/dataformat.pyi

@@ -37,6 +37,7 @@ class DataFormat(ChepyCore):
     def hex_to_bytes(self: DataFormatT) -> DataFormatT: ...
     def hex_to_str(self: DataFormatT, ignore: bool=...) -> DataFormatT: ...
     def str_to_hex(self: DataFormatT, delimiter: Union[str, bytes]=b'') -> DataFormatT: ...
+    def int_to_bytes(self: DataFormatT, length: Union[int, None], byteorder: Literal['big,little']='big') -> DataFormatT: ...
     def int_to_hex(self: DataFormatT, unsigned_int: bool=True, bit_size: int=64) -> DataFormatT: ...
     def int_to_str(self: DataFormatT) -> DataFormatT: ...
     def binary_to_hex(self: DataFormatT) -> DataFormatT: ...

tests/test_dataformat.py

@@ -202,6 +202,14 @@ def test_hex_to_bytes():
     assert Chepy("ab00").hex_to_bytes().o == b"\xab\x00"
 
 
+def test_int_to_bytes():
+    data = [17235, 17223]
+    assert Chepy(data[0]).int_to_bytes().o == b"CS"
+    assert Chepy(data[0]).int_to_bytes(2).o == b"CS"
+    assert Chepy(data[0]).int_to_bytes(2, "little").o == b"SC"
+    assert Chepy(data[0]).int_to_bytes(4, "big").o == b"\x00\x00CS"
+
+
 def test_int_to_hex():
     assert Chepy(101).int_to_hex().o == b"65"
     assert Chepy(-0x2152411021524111).int_to_hex().o == b"deadbeefdeadbeef"