Pass 'null' frame to call node in 'atexit' handler.

Author: fangerer

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/AtexitModuleBuiltins.java

@@ -42,17 +42,23 @@
 
 import java.util.List;
 
+import com.oracle.graal.python.PythonLanguage;
 import com.oracle.graal.python.builtins.Builtin;
 import com.oracle.graal.python.builtins.CoreFunctions;
 import com.oracle.graal.python.builtins.PythonBuiltins;
 import com.oracle.graal.python.builtins.objects.PNone;
 import com.oracle.graal.python.builtins.objects.function.PKeyword;
+import com.oracle.graal.python.nodes.PNodeWithGlobalState;
+import com.oracle.graal.python.nodes.PNodeWithGlobalState.DefaultContextManager;
 import com.oracle.graal.python.nodes.call.CallNode;
 import com.oracle.graal.python.nodes.function.PythonBuiltinBaseNode;
 import com.oracle.graal.python.nodes.function.builtins.PythonUnaryBuiltinNode;
 import com.oracle.graal.python.nodes.function.builtins.PythonVarargsBuiltinNode;
+import com.oracle.graal.python.runtime.PythonContext;
+import com.oracle.graal.python.runtime.exception.PException;
 import com.oracle.truffle.api.Truffle;
 import com.oracle.truffle.api.TruffleLanguage;
+import com.oracle.truffle.api.TruffleLanguage.ContextReference;
 import com.oracle.truffle.api.dsl.GenerateNodeFactory;
 import com.oracle.truffle.api.dsl.NodeFactory;
 import com.oracle.truffle.api.dsl.Specialization;
@@ -70,7 +76,10 @@ protected List<? extends NodeFactory<? extends PythonBuiltinBaseNode>> getNodeFa
     @GenerateNodeFactory
     abstract static class RegisterNode extends PythonVarargsBuiltinNode {
         private static class AtExitCallTarget extends RootNode {
-            @Child CallNode callNode = CallNode.create();
+            @Child private CallNode callNode = CallNode.create();
+
+            private final ContextReference<PythonContext> contextRef = lookupContextReference(PythonLanguage.class);
+
             private Object callable;
             private Object[] arguments;
             private PKeyword[] keywords;
@@ -83,8 +92,12 @@ protected AtExitCallTarget(TruffleLanguage<?> language, Object callable, Object[
             }
 
             @Override
+            @SuppressWarnings("try")
             public Object execute(VirtualFrame frame) {
-                return callNode.execute(frame, callable, arguments, keywords);
+                // we deliberately pass 'null' frame here
+                try (DefaultContextManager cm = PNodeWithGlobalState.transferToContext(contextRef, PException.NO_EXCEPTION)) {
+                    return callNode.execute(null, callable, arguments, keywords);
+                }
             }
         }