Add weaklistoffset to builtin types

Author: qunaibit

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/PythonBuiltinClassType.java

@@ -99,6 +99,7 @@
 import java.util.HashSet;
 
 import com.oracle.graal.python.PythonLanguage;
+import com.oracle.graal.python.builtins.modules.WeakRefModuleBuiltins;
 import com.oracle.graal.python.builtins.objects.function.BuiltinMethodDescriptor;
 import com.oracle.graal.python.builtins.objects.type.SpecialMethodSlot;
 import com.oracle.graal.python.runtime.PythonContext;
@@ -499,6 +500,7 @@ private static class Flags {
     // initialized in static constructor
     @CompilationFinal private PythonBuiltinClassType type;
     @CompilationFinal private PythonBuiltinClassType base;
+    @CompilationFinal private int weaklistoffset;
 
     /**
      * @see #redefinesSlot(SpecialMethodSlot)
@@ -540,6 +542,7 @@ private static class Flags {
         this.isBuiltinWithDict = flags.isBuiltinWithDict;
         this.isException = flags == Flags.EXCEPTION;
         this.methodsFlags = methodsFlags;
+        this.weaklistoffset = -1;
     }
 
     PythonBuiltinClassType(String name, String module) {
@@ -610,6 +613,10 @@ public long getMethodsFlags() {
         return methodsFlags;
     }
 
+    public int getWeaklistoffset() {
+        return weaklistoffset;
+    }
+
     /**
      * Returns {@code true} if this method slot is redefined in Python code during initialization.
      * Values of such slots cannot be cached in {@link #specialMethodSlots}, because they are not
@@ -860,6 +867,8 @@ public final Shape getInstanceShape(PythonLanguage lang) {
             if (type.type == null && type.base != null) {
                 type.type = type.base.type;
             }
+
+            type.weaklistoffset = WeakRefModuleBuiltins.getBuiltinTypeWeaklistoffset(type);
         }
 
         // Finally, we set all remaining types to PythonClass.

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

@@ -105,7 +105,7 @@ protected List<? extends NodeFactory<? extends PythonBuiltinBaseNode>> getNodeFa
         return WeakRefModuleBuiltinsFactory.getFactories();
     }
 
-    private static long getBuiltinTypeWeaklistoffset(PythonBuiltinClassType cls) {
+    public static int getBuiltinTypeWeaklistoffset(PythonBuiltinClassType cls) {
         // @formatter:off
         return switch (cls) {
             case PythonObject, // object
@@ -393,7 +393,7 @@ public PReferenceType refType(Object cls, Object object, @SuppressWarnings("unus
             Object clazz = getClassNode.execute(this, obj);
             boolean allowed = true;
             if (clazz instanceof PythonBuiltinClassType type) {
-                allowed = getBuiltinTypeWeaklistoffset(type) != 0;
+                allowed = type.getWeaklistoffset() != 0;
             }
             if (!allowed) {
                 throw raise(TypeError, ErrorMessages.CANNOT_CREATE_WEAK_REFERENCE_TO, obj);