Adding runtime typechecking to SharedMemory::_from_raw

bytecodealliance · Oct 25, 2024 · cd78fa1 · cd78fa1
1 parent 4440fe9
commit cd78fa1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/wasmtime/_sharedmemory.py b/wasmtime/_sharedmemory.py
@@ -21,6 +21,9 @@ def __init__(self, engine: Engine, ty: MemoryType):
 
     @classmethod
     def _from_raw(cls, sharedmemory: ffi.wasmtime_sharedmemory_t) -> "SharedMemory":
+        if not isinstance(sharedmemory, ffi.wasmtime_sharedmemory_t):
+            raise TypeError("wrong shared memory pointer type provided to _from_raw")
+
         ty: "SharedMemory" = cls.__new__(cls)
         ty._set_ptr(sharedmemory)
         return ty