add check-hash-based-pycs cli option

Author: cosminbasca

graalpython/com.oracle.graal.python.shell/src/com/oracle/graal/python/shell/GraalPythonMain.java

@@ -89,6 +89,7 @@ public static void main(String[] args) {
     private Map<String, String> enginePolyglotOptions;
     private boolean dontWriteBytecode = false;
     private String warnOptions = null;
+    private String checkHashPycsMode = "default";
 
     @Override
     protected List<String> preprocessArguments(List<String> givenArgs, Map<String, String> polyglotOptions) {
@@ -240,6 +241,10 @@ protected List<String> preprocessArguments(List<String> givenArgs, Map<String, S
                     addRelaunchArg(arg);
                     unrecognized.add(arg);
                     break;
+                case "--check-hash-based-pycs":
+                    i += 1;
+                    checkHashPycsMode = arguments.get(i);
+                    break;
                 default:
                     if (!arg.startsWith("-")) {
                         inputFile = arg;
@@ -453,6 +458,8 @@ protected void launch(Builder contextBuilder) {
         contextBuilder.option("python.TerminalWidth", Integer.toString(consoleHandler.getTerminalWidth()));
         contextBuilder.option("python.TerminalHeight", Integer.toString(consoleHandler.getTerminalHeight()));
 
+        contextBuilder.option("python.CheckHashPycsMode", checkHashPycsMode);
+
         if (multiContext) {
             contextBuilder.engine(Engine.newBuilder().allowExperimentalOptions(true).options(enginePolyglotOptions).build());
         }

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

@@ -86,6 +86,7 @@
 import com.oracle.graal.python.parser.sst.SerializationUtils;
 import com.oracle.graal.python.runtime.ExecutionContext.ForeignCallContext;
 import com.oracle.graal.python.runtime.PythonContext;
+import com.oracle.graal.python.runtime.PythonCore;
 import com.oracle.graal.python.runtime.PythonOptions;
 import com.oracle.graal.python.runtime.exception.PException;
 import com.oracle.graal.python.util.PythonUtils;
@@ -124,6 +125,14 @@ protected List<? extends NodeFactory<? extends PythonBuiltinBaseNode>> getNodeFa
         return ImpModuleBuiltinsFactory.getFactories();
     }
 
+    @Override
+    public void postInitialize(PythonCore core) {
+        super.postInitialize(core);
+        PythonContext context = core.getContext();
+        PythonModule mod = core.lookupBuiltinModule("_imp");
+        mod.setAttribute("check_hash_based_pycs", context.getOption(PythonOptions.CheckHashPycsMode));
+    }
+
     @Builtin(name = "acquire_lock")
     @GenerateNodeFactory
     public abstract static class AcquireLock extends PythonBuiltinNode {
@@ -498,7 +507,7 @@ public abstract static class CreateBuiltin extends PythonBuiltinNode {
         @Specialization
         @TruffleBoundary
         public Object run(PythonObject moduleSpec,
-                          @Cached CastToJavaStringNode toJavaStringNode) {
+                        @Cached CastToJavaStringNode toJavaStringNode) {
             Object name = moduleSpec.getAttribute("name");
             PythonModule builtinModule = getCore().lookupBuiltinModule(toJavaStringNode.execute(name));
             if (builtinModule != null) {

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/runtime/PythonOptions.java

@@ -120,6 +120,17 @@ private PythonOptions() {
     @Option(category = OptionCategory.USER, help = "Equivalent to setting the PYTHONWARNINGS environment variable for the standard launcher.", stability = OptionStability.STABLE) //
     public static final OptionKey<String> WarnOptions = new OptionKey<>("");
 
+    @Option(category = OptionCategory.USER, help = "Value of the --check-hash-based-pycs command line option" +
+                    "- 'default' means the 'check_source' flag in hash-based pycs" +
+                    "  determines invalidation" +
+                    "- 'always' causes the interpreter to hash the source file for" +
+                    "  invalidation regardless of value of 'check_source' bit" +
+                    "- 'never' causes the interpreter to always assume hash-based pycs are" +
+                    "  valid" +
+                    "The default value is 'default'." +
+                    "See PEP 552 'Deterministic pycs' for more details.", stability = OptionStability.STABLE) //
+    public static final OptionKey<String> CheckHashPycsMode = new OptionKey<>("default");
+
     @Option(category = OptionCategory.INTERNAL, help = "Set the location of C API home. Overrides any environment variables or Java options.", stability = OptionStability.STABLE) //
     public static final OptionKey<String> CAPI = new OptionKey<>("");