[GR-62607] Fix JavaMainSupport.mainArgs.

PullRequest: graal/20186

Author: cstancu

substratevm/src/com.oracle.graal.pointsto/src/com/oracle/graal/pointsto/meta/AnalysisField.java

@@ -31,6 +31,7 @@
 import java.util.concurrent.ConcurrentMap;
 import java.util.concurrent.atomic.AtomicReferenceFieldUpdater;
 
+import com.oracle.graal.pointsto.BigBang;
 import com.oracle.graal.pointsto.api.PointstoOptions;
 import com.oracle.graal.pointsto.flow.ContextInsensitiveFieldTypeFlow;
 import com.oracle.graal.pointsto.flow.FieldTypeFlow;
@@ -268,6 +269,15 @@ public boolean registerAsWritten(Object reason) {
         });
     }
 
+    public void injectDeclaredType() {
+        BigBang bb = getUniverse().getBigbang();
+        if (getStorageKind().isObject()) {
+            bb.injectFieldTypes(this, List.of(this.getType()), true);
+        } else if (bb.trackPrimitiveValues() && getStorageKind().isPrimitive()) {
+            ((PointsToAnalysisField) this).saturatePrimitiveField();
+        }
+    }
+
     public boolean isGuaranteeFolded() {
         return getAnnotation(GuaranteeFolded.class) != null;
     }

substratevm/src/com.oracle.svm.core/src/com/oracle/svm/core/JavaMainWrapper.java

@@ -32,10 +32,10 @@
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collections;
+import java.util.EnumSet;
 import java.util.List;
 import java.util.function.BooleanSupplier;
 
-import jdk.graal.compiler.word.Word;
 import org.graalvm.nativeimage.CurrentIsolate;
 import org.graalvm.nativeimage.ImageSingletons;
 import org.graalvm.nativeimage.Isolate;
@@ -65,10 +65,14 @@
 import com.oracle.svm.core.c.function.CEntryPointOptions.NoEpilogue;
 import com.oracle.svm.core.c.function.CEntryPointOptions.NoPrologue;
 import com.oracle.svm.core.c.function.CEntryPointSetup;
+import com.oracle.svm.core.graal.snippets.CEntryPointSnippets;
 import com.oracle.svm.core.jdk.InternalVMMethod;
 import com.oracle.svm.core.jdk.RuntimeSupport;
 import com.oracle.svm.core.jni.JNIJavaVMList;
 import com.oracle.svm.core.jni.functions.JNIFunctionTables;
+import com.oracle.svm.core.layeredimagesingleton.ApplicationLayerOnlyImageSingleton;
+import com.oracle.svm.core.layeredimagesingleton.LayeredImageSingletonBuilderFlags;
+import com.oracle.svm.core.layeredimagesingleton.UnsavedSingleton;
 import com.oracle.svm.core.log.Log;
 import com.oracle.svm.core.thread.JavaThreads;
 import com.oracle.svm.core.thread.PlatformThreads;
@@ -80,6 +84,8 @@
 import com.oracle.svm.util.ClassUtil;
 import com.oracle.svm.util.ReflectionUtil;
 
+import jdk.graal.compiler.word.Word;
+
 @InternalVMMethod
 public class JavaMainWrapper {
     /*
@@ -90,8 +96,12 @@ public class JavaMainWrapper {
 
     private static UnsignedWord argvLength = Word.zero();
 
-    public static class JavaMainSupport {
-
+    /**
+     * In a layered build the {@link JavaMainSupport} is installed in the last layer. However, code
+     * that uses it may be compiled as part of the base layer, e.g., such as
+     * {@link CEntryPointSnippets}.
+     */
+    public static class JavaMainSupport implements ApplicationLayerOnlyImageSingleton, UnsavedSingleton {
         private final MethodHandle javaMainHandle;
         private final MethodHandle javaMainClassCtorHandle;
         final String javaMainClassName;
@@ -157,6 +167,10 @@ public List<String> getInputArguments() {
             return Collections.emptyList();
         }
 
+        @Override
+        public EnumSet<LayeredImageSingletonBuilderFlags> getImageBuilderFlags() {
+            return LayeredImageSingletonBuilderFlags.ALL_ACCESS;
+        }
     }
 
     public static void invokeMain(String[] args) throws Throwable {
@@ -189,6 +203,7 @@ public static void invokeMain(String[] args) throws Throwable {
      * Determines whether instance main methodes are enabled. See JDK-8306112: Implementation of JEP
      * 445: Unnamed Classes and Instance Main Methods (Preview).
      */
+    @Platforms(Platform.HOSTED_ONLY.class)
     public static boolean instanceMainMethodSupported() {
         var previewFeature = ReflectionUtil.lookupClass(true, "jdk.internal.misc.PreviewFeatures");
         try {

substratevm/src/com.oracle.svm.hosted/src/com/oracle/svm/hosted/ImageSingletonsSupportImpl.java

@@ -228,10 +228,6 @@ private void doAddInternal(Class<?> key, Object value) {
 
                 if (singleton instanceof MultiLayeredImageSingleton || ApplicationLayerOnlyImageSingleton.isSingletonInstanceOf(singleton)) {
 
-                    if (!key.equals(singleton.getClass())) {
-                        throw UserError.abort("The implementation class must be the same as the key class. key: %s, singleton: %s", key, singleton);
-                    }
-
                     if (singleton instanceof MultiLayeredImageSingleton && ApplicationLayerOnlyImageSingleton.isSingletonInstanceOf(singleton)) {
                         throw UserError.abort("Singleton cannot implement both %s and %s. singleton: %s", MultiLayeredImageSingleton.class, ApplicationLayerOnlyImageSingleton.class, singleton);
                     }

substratevm/src/com.oracle.svm.hosted/src/com/oracle/svm/hosted/imagelayer/LoadImageSingletonFeature.java

@@ -580,7 +580,7 @@ void assignSlots(HostedMetaAccess metaAccess) {
             int slotAssignment;
             LoadImageSingletonDataImpl info = entry.getValue();
             var hType = metaAccess.lookupJavaType(info.getLoadType());
-            if (hType.isInstantiated()) {
+            if (hType.getWrapped().isAnySubtypeInstantiated()) {
                 Class<?> keyClass = entry.getKey();
                 SlotInfo slotInfo = priorKeyToSlotInfoMap.get(entry.getKey());
 

substratevm/src/com.oracle.svm.hosted/src/com/oracle/svm/hosted/imagelayer/SVMImageLayerLoader.java

@@ -1157,9 +1157,15 @@ public void initializeBaseLayerField(AnalysisField analysisField) {
         if (!analysisField.isStatic() && (isAccessed || isRead)) {
             analysisField.getDeclaringClass().getInstanceFields(true);
         }
-        registerFlag(isAccessed, debug -> analysisField.registerAsAccessed(PERSISTED));
+        registerFlag(isAccessed, debug -> {
+            analysisField.injectDeclaredType();
+            analysisField.registerAsAccessed(PERSISTED);
+        });
         registerFlag(isRead, debug -> analysisField.registerAsRead(PERSISTED));
-        registerFlag(fieldData.getIsWritten(), debug -> analysisField.registerAsWritten(PERSISTED));
+        registerFlag(fieldData.getIsWritten(), debug -> {
+            analysisField.injectDeclaredType();
+            analysisField.registerAsWritten(PERSISTED);
+        });
         registerFlag(fieldData.getIsFolded(), debug -> analysisField.registerAsFolded(PERSISTED));
     }