add queue and threading

ThomasFarstrike · ThomasFarstrike · commit d6ee047239aa · 2025-05-19T12:07:42.000+02:00
diff --git a/internal_filesystem/lib/queue.py 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
@@ -96,6 +96,10 @@ def secp256k1_ecdsa_signature_serialize_der(self, ctx, output, outputlen, raw_si
         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 @@ def secp256k1_ecdsa_signature_serialize_compact(self, ctx, output, raw_sig):
         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 @@ def secp256k1_ecdsa_recoverable_signature_serialize_compact(self, ctx, output, r
         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 @@ def secp256k1_schnorrsig_sign_custom(self, ctx, sig64, msg, msg_len, keypair, au
         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 @@ def secp256k1_schnorrsig_verify(self, ctx, schnorr_sig, msg, msg_len, xonly_pubk
 
     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 @@ def secp256k1_ec_pubkey_serialize(self, ctx, output, outlen, pubkey, flags):
         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 @@ def secp256k1_ecdh(self, ctx, output, pubkey, seckey, hashfn=FFI.NULL, hasharg=F
         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
@@ -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()
