From 817f43466625a1dafa8c41d49ff1a4359b406567 Mon Sep 17 00:00:00 2001
From: malwarefrank <42877127+malwarefrank@users.noreply.github.com>
Date: Sun, 17 Mar 2024 01:52:20 +0000
Subject: [PATCH] add function to convert int to compressed int

---
 src/dnfile/utils.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/src/dnfile/utils.py b/src/dnfile/utils.py
index 2c9ed0d..90232bf 100644
--- a/src/dnfile/utils.py
+++ b/src/dnfile/utils.py
@@ -57,6 +57,30 @@ def read_compressed_int(data) -> Optional[Tuple[int, int]]:
         return None
 
 
+def compress_int(i: int) -> Optional[bytes]:
+    """
+    Given integer, return bytes representing a compressed integer per
+    spec ECMA-335 II.23.2 Blobs and signatures.
+    """
+
+    if not isinstance(i, int):
+        raise TypeError(f"expected int, given {type(i)}")
+    if i < 0:
+        raise ValueError(f"expected positive int, given {i}")
+
+    if i <= 0x7f:
+        return int.to_bytes(i, length=1, byteorder="big")
+    elif i <= 0x3fff:
+        b = int.to_bytes(i, length=2, byteorder="big")
+        return bytes((0x80 | b[0], b[1]))
+    elif i <= 0x1fffffff:
+        b = int.to_bytes(i, length=4, byteorder="big")
+        return bytes((0x80 | 0x40 | b[0], b[1], b[2], b[3]))
+    else:
+        logger.warning(f"invalid int {i}, max value 0x1fffffff")
+        return None
+
+
 def two_way_dict(pairs):
     return dict([(e[1], e[0]) for e in pairs] + pairs)