Raise a ruby exception if a non-native pointer backed struct is put into

a pointer field
jruby · May 21, 2009 · 49968e2 · 49968e2
1 parent e7d6ace
commit 49968e2
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/org/jruby/ext/ffi/StructLayoutBuilder.java b/src/org/jruby/ext/ffi/StructLayoutBuilder.java
@@ -319,7 +319,13 @@ public void put(Ruby runtime, StructLayout.Storage cache, IRubyObject ptr, IRuby
             if (value instanceof Pointer) {
                 getMemoryIO(ptr).putMemoryIO(getOffset(ptr), ((Pointer) value).getMemoryIO());
             } else if (value instanceof Struct) {
-                getMemoryIO(ptr).putMemoryIO(getOffset(ptr), ((Struct) value).getMemoryIO());
+                MemoryIO mem = ((Struct) value).getMemoryIO();
+
+                if (!(mem instanceof DirectMemoryIO)) {
+                    throw runtime.newArgumentError("Struct memory not backed by a native pointer");
+                }
+                getMemoryIO(ptr).putMemoryIO(getOffset(ptr), mem);
+
             } else if (value instanceof RubyInteger) {
                 getMemoryIO(ptr).putAddress(offset, Util.int64Value(ptr));
             } else if (value.respondsTo("to_ptr")) {