Convert froml (AbstractMemory *) DATA_PTR(obj) to calls to rb_FFI_AbstractMemory_cast(obj, klass) to ensure an exception is raised for invalid objects

1 files changed, 6 insertions, 0 deletions

diff --git a/ext/ffi_c/AbstractMemory.h b/ext/ffi_c/AbstractMemory.h
index 2df3816..729f9cf 100644
--- a/ext/ffi_c/AbstractMemory.h
+++ b/ext/ffi_c/AbstractMemory.h
@@ -23,6 +23,12 @@ checkBounds(AbstractMemory* mem, long off, long len)
     }
 }
 
+#define MEMORY(obj) rb_FFI_AbstractMemory_cast((obj), rb_FFI_AbstractMemory_class)
+#define MEMORY_PTR(obj) MEMORY((obj))->address
+#define MEMORY_LEN(obj) MEMORY((obj))->size
+
+extern AbstractMemory* rb_FFI_AbstractMemory_cast(VALUE obj, VALUE klass);
+
 extern VALUE rb_FFI_AbstractMemory_class;
 #ifdef	__cplusplus
 }