Add test for invalid pointer sharing for primitive wrappers

Author: fangerer

graalpython/com.oracle.graal.python.cext/src/capi.c

@@ -876,6 +876,11 @@ int PyTruffle_Debug(void *arg) {
     return 0;
 }
 
+int PyTruffle_ToNative(void *arg) {
+    polyglot_invoke(PY_TRUFFLE_CEXT, "PyTruffle_ToNative", arg);
+    return 0;
+}
+
 int truffle_ptr_compare(void* x, void* y, int op) {
     switch (op) {
     case Py_LT:

graalpython/com.oracle.graal.python.test/src/tests/cpyext/test_misc.py

@@ -228,6 +228,46 @@ def compile_module(self, name):
         cmpfunc=unhandled_error_compare
     )
 
+    # Tests if wrapped Java primitive values do not share the same
+    # native pointer.
+    test_primitive_sharing = CPyExtFunction(
+        lambda args: True,
+        lambda: (
+            (123.0, ),
+        ),
+        code="""
+        // internal function defined in 'capi.c'
+        int PyTruffle_ToNative(void *);
+        
+        PyObject* primitive_sharing(PyObject* val) {
+            Py_ssize_t val_refcnt = Py_REFCNT(val);
+            // assume val's refcnt is X > 0
+            Py_INCREF(val);
+            // val's refcnt should now be X+1
+
+            double dval = PyFloat_AsDouble(val);
+
+            PyTruffle_ToNative(val);
+
+            // a fresh object with the same value
+            PyObject *val1 = PyFloat_FromDouble(dval);
+            PyTruffle_ToNative(val1);
+
+            // now, kill it
+            Py_DECREF(val1);
+
+            // reset val's refcnt to X
+            Py_DECREF(val);
+
+            return val_refcnt == Py_REFCNT(val) ? Py_True : Py_False;
+        }
+        """,
+        resultspec="O",
+        argspec="O",
+        arguments=["PyObject* val"],
+        cmpfunc=unhandled_error_compare
+    )
+
     test_PyOS_double_to_string = CPyExtFunction(
         _reference_format_float,
         lambda: (

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/cext/PythonCextBuiltins.java

@@ -212,6 +212,7 @@
 import com.oracle.truffle.api.CompilerDirectives;
 import com.oracle.truffle.api.CompilerDirectives.TruffleBoundary;
 import com.oracle.truffle.api.RootCallTarget;
+import com.oracle.truffle.api.TruffleLanguage.Env;
 import com.oracle.truffle.api.TruffleLogger;
 import com.oracle.truffle.api.dsl.Cached;
 import com.oracle.truffle.api.dsl.Cached.Exclusive;
@@ -232,6 +233,7 @@
 import com.oracle.truffle.api.interop.UnsupportedMessageException;
 import com.oracle.truffle.api.interop.UnsupportedTypeException;
 import com.oracle.truffle.api.library.CachedLibrary;
+import com.oracle.truffle.api.nodes.LanguageInfo;
 import com.oracle.truffle.api.nodes.Node;
 import com.oracle.truffle.api.nodes.RootNode;
 import com.oracle.truffle.api.object.DynamicObjectLibrary;
@@ -241,6 +243,7 @@
 import com.oracle.truffle.api.profiles.LoopConditionProfile;
 import com.oracle.truffle.api.profiles.ValueProfile;
 import com.oracle.truffle.api.utilities.CyclicAssumption;
+import com.oracle.truffle.llvm.api.Toolchain;
 
 @CoreFunctions(defineModule = PythonCextBuiltins.PYTHON_CEXT)
 @GenerateNodeFactory
@@ -269,6 +272,15 @@ public void initialize(Python3Core core) {
         builtinConstants.put("PyGILState_Release", new PyGILStateRelease());
     }
 
+    @Override
+    public void postInitialize(Python3Core core) {
+        super.postInitialize(core);
+        if (!core.getContext().getOption(PythonOptions.EnableDebuggingBuiltins)) {
+            PythonModule mod = core.lookupBuiltinModule(PYTHON_CEXT);
+            mod.setAttribute("PyTruffle_ToNative", PNone.NO_VALUE);
+        }
+    }
+
     @FunctionalInterface
     public interface TernaryFunction<T1, T2, T3, R> {
         R apply(T1 arg0, T2 arg1, T3 arg2);
@@ -2376,15 +2388,33 @@ Object tssDelete(Object key,
         }
     }
 
+    // directly called without landing function
     @Builtin(name = "PyTruffle_Debug", takesVarArgs = true)
     @GenerateNodeFactory
     public abstract static class PyTruffleDebugNode extends PythonBuiltinNode {
         @Specialization
         @TruffleBoundary
-        public Object doIt(Object[] args,
+        static Object doIt(Object[] args,
                         @Cached DebugNode debugNode) {
             debugNode.execute(args);
             return PNone.NONE;
         }
     }
+
+    // directly called without landing function
+    @Builtin(name = "PyTruffle_ToNative", minNumOfPositionalArgs = 1)
+    @GenerateNodeFactory
+    public abstract static class PyTruffleToNativeNode extends PythonUnaryBuiltinNode {
+        @Specialization
+        @TruffleBoundary
+        Object doIt(Object object) {
+            Env env = getContext().getEnv();
+            LanguageInfo llvmInfo = env.getInternalLanguages().get(PythonLanguage.LLVM_LANGUAGE);
+            Toolchain toolchain = env.lookup(llvmInfo, Toolchain.class);
+            if ("native".equals(toolchain.getIdentifier())) {
+                InteropLibrary.getUncached().toNative(object);
+            }
+            return PNone.NONE;
+        }
+    }
 }