From d6ee047239aa96f13ef9a81a87601c5f653008dd Mon Sep 17 00:00:00 2001
From: Thomas Farstrike <thomasverstreken@protonmail.com>
Date: Mon, 19 May 2025 12:07:42 +0200
Subject: [PATCH] add queue and threading

---
 internal_filesystem/lib/queue.py            | 44 +++++++++++++++++++++
 internal_filesystem/lib/secp256k1_compat.py | 21 +++++++++-
 internal_filesystem/lib/threading.py        | 26 ++++++++++++
 3 files changed, 90 insertions(+), 1 deletion(-)
 create mode 100644 internal_filesystem/lib/queue.py
 create mode 100644 internal_filesystem/lib/threading.py

diff --git a/internal_filesystem/lib/queue.py b/internal_filesystem/lib/queue.py
new file mode 100644
index 00000000..88ba52dc
--- /dev/null
+++ b/internal_filesystem/lib/queue.py
@@ -0,0 +1,44 @@
+try:
+    import _thread
+except ImportError:
+    _thread = None
+
+class Queue:
+    def __init__(self, maxsize=0):
+        self._queue = []
+        self.maxsize = maxsize  # 0 means unlimited
+        self._lock = _thread.allocate_lock() if _thread else None
+
+    def put(self, item):
+        if self._lock:
+            with self._lock:
+                if self.maxsize > 0 and len(self._queue) >= self.maxsize:
+                    raise RuntimeError("Queue is full")
+                self._queue.append(item)
+        else:
+            if self.maxsize > 0 and len(self._queue) >= self.maxsize:
+                raise RuntimeError("Queue is full")
+            self._queue.append(item)
+
+    def get(self):
+        if self._lock:
+            with self._lock:
+                if not self._queue:
+                    raise RuntimeError("Queue is empty")
+                return self._queue.pop(0)
+        else:
+            if not self._queue:
+                raise RuntimeError("Queue is empty")
+            return self._queue.pop(0)
+
+    def qsize(self):
+        if self._lock:
+            with self._lock:
+                return len(self._queue)
+        return len(self._queue)
+
+    def empty(self):
+        return self.qsize() == 0
+
+    def full(self):
+        return self.maxsize > 0 and self.qsize() >= self.maxsize
diff --git a/internal_filesystem/lib/secp256k1_compat.py b/internal_filesystem/lib/secp256k1_compat.py
index d59a7ebe..e0cf1a11 100644
--- a/internal_filesystem/lib/secp256k1_compat.py
+++ b/internal_filesystem/lib/secp256k1_compat.py
@@ -96,6 +96,10 @@ class Lib:
         try:
             if isinstance(raw_sig, FFI.CData):
                 raw_sig = raw_sig._data
+            if isinstance(output, FFI.CData):
+                output = output._data
+            if isinstance(outputlen, FFI.CData):
+                outputlen = outputlen._data
             result = usecp256k1.ecdsa_signature_serialize_der(raw_sig)
             if result is None:
                 return 0
@@ -121,6 +125,8 @@ class Lib:
         try:
             if isinstance(raw_sig, FFI.CData):
                 raw_sig = raw_sig._data
+            if isinstance(output, FFI.CData):
+                output = output._data
             result = usecp256k1.ecdsa_signature_serialize_compact(raw_sig)
             if result is None:
                 return 0
@@ -181,6 +187,10 @@ class Lib:
         try:
             if isinstance(recover_sig, FFI.CData):
                 recover_sig = recover_sig._data
+            if isinstance(output, FFI.CData):
+                output = output._data
+            if isinstance(recid, FFI.CData):
+                recid = recid._data
             result, rec_id = usecp256k1.ecdsa_sign_recoverable(recover_sig)
             if result is None:
                 return 0
@@ -246,6 +256,8 @@ class Lib:
         try:
             if isinstance(keypair, FFI.CData):
                 keypair = keypair._data
+            if isinstance(sig64, FFI.CData):
+                sig64 = sig64._data
             result = usecp256k1.schnorrsig_sign(msg, keypair)
             if result is None:
                 return 0
@@ -264,6 +276,8 @@ class Lib:
 
     def secp256k1_tagged_sha256(self, ctx, hash32, tag, tag_len, msg, msg_len):
         try:
+            if isinstance(hash32, FFI.CData):
+                hash32 = hash32._data
             result = usecp256k1.tagged_sha256(tag, msg)
             if result is None:
                 return 0
@@ -276,12 +290,15 @@ class Lib:
         try:
             if isinstance(pubkey, FFI.CData):
                 pubkey = pubkey._data
+            if isinstance(output, FFI.CData):
+                output = output._data
             if isinstance(outlen, FFI.CData):
                 outlen = outlen._data
             result = usecp256k1.ec_pubkey_serialize(pubkey, flags)
             if result is None:
                 return 0
-            output[:outlen[0]] = result
+            output[:len(result)] = result
+            outlen[0] = len(result)
             return 1
         except (ValueError, AttributeError):
             return 0
@@ -405,6 +422,8 @@ class Lib:
         try:
             if isinstance(pubkey, FFI.CData):
                 pubkey = pubkey._data
+            if isinstance(output, FFI.CData):
+                output = output._data
             result = usecp256k1.ecdh(pubkey, seckey)
             if result is None:
                 return 0
diff --git a/internal_filesystem/lib/threading.py b/internal_filesystem/lib/threading.py
new file mode 100644
index 00000000..31c04ec7
--- /dev/null
+++ b/internal_filesystem/lib/threading.py
@@ -0,0 +1,26 @@
+try:
+    import _thread
+except ImportError:
+    _thread = None
+
+class Lock:
+    def __init__(self):
+        self._lock = _thread.allocate_lock() if _thread else None
+
+    def __enter__(self):
+        if self._lock:
+            self._lock.acquire()
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        if self._lock:
+            self._lock.release()
+
+    def acquire(self):
+        if self._lock:
+            return self._lock.acquire()
+        return True
+
+    def release(self):
+        if self._lock:
+            self._lock.release()