8059632: Method reference compilation uses incorrect qualifying type

Reviewed-by: mcimadamore

Author: Srikanth Adayapalam

src/jdk.compiler/share/classes/com/sun/tools/javac/code/Types.java

@@ -119,6 +119,7 @@ protected Types(Context context) {
         messages = JavacMessages.instance(context);
         diags = JCDiagnostic.Factory.instance(context);
         noWarnings = new Warner(null);
+        qualifiedSymbolCache = new HashMap<>();
     }
     // </editor-fold>
 
@@ -3617,6 +3618,56 @@ public int rank(Type t) {
     }
     // </editor-fold>
 
+    /** Cache the symbol to reflect the qualifying type.
+     *  key: corresponding type
+     *  value: qualified symbol
+     */
+    private Map<Type, Symbol> qualifiedSymbolCache;
+
+    public void clearQualifiedSymbolCache() {
+        qualifiedSymbolCache.clear();
+    }
+
+    /** Construct a symbol to reflect the qualifying type that should
+     *  appear in the byte code as per JLS 13.1.
+     *
+     *  For {@literal target >= 1.2}: Clone a method with the qualifier as owner (except
+     *  for those cases where we need to work around VM bugs).
+     *
+     *  For {@literal target <= 1.1}: If qualified variable or method is defined in a
+     *  non-accessible class, clone it with the qualifier class as owner.
+     *
+     *  @param sym    The accessed symbol
+     *  @param site   The qualifier's type.
+     */
+    public Symbol binaryQualifier(Symbol sym, Type site) {
+
+        if (site.hasTag(ARRAY)) {
+            if (sym == syms.lengthVar ||
+                    sym.owner != syms.arrayClass)
+                return sym;
+            // array clone can be qualified by the array type in later targets
+            Symbol qualifier;
+            if ((qualifier = qualifiedSymbolCache.get(site)) == null) {
+                qualifier = new ClassSymbol(Flags.PUBLIC, site.tsym.name, site, syms.noSymbol);
+                qualifiedSymbolCache.put(site, qualifier);
+            }
+            return sym.clone(qualifier);
+        }
+
+        if (sym.owner == site.tsym ||
+                (sym.flags() & (STATIC | SYNTHETIC)) == (STATIC | SYNTHETIC)) {
+            return sym;
+        }
+
+        // leave alone methods inherited from Object
+        // JLS 13.1.
+        if (sym.owner == syms.objectType.tsym)
+            return sym;
+
+        return sym.clone(site.tsym);
+    }
+
     /**
      * Helper method for generating a string representation of a given type
      * accordingly to a given locale

src/jdk.compiler/share/classes/com/sun/tools/javac/comp/LambdaToMethod.java

@@ -518,6 +518,9 @@ public void visitReference(JCMemberReference tree) {
                 throw new InternalError("Should not have an invalid kind");
         }
 
+        if (init != null) {
+            refSym = (MethodSymbol) types.binaryQualifier(refSym, init.type);
+        }
         List<JCExpression> indy_args = init==null? List.nil() : translate(List.of(init), localContext.prev);
 
 

src/jdk.compiler/share/classes/com/sun/tools/javac/jvm/Gen.java

@@ -130,7 +130,6 @@ protected Gen(Context context) {
         // ignore cldc because we cannot have both stackmap formats
         this.stackMap = StackMapFormat.JSR202;
         annotate = Annotate.instance(context);
-        qualifiedSymbolCache = new HashMap<>();
     }
 
     /** Switches
@@ -172,12 +171,6 @@ protected Gen(Context context) {
     List<LocalItem> stackBeforeSwitchExpression;
     LocalItem switchResult;
 
-    /** Cache the symbol to reflect the qualifying type.
-     *  key: corresponding type
-     *  value: qualified symbol
-     */
-    Map<Type, Symbol> qualifiedSymbolCache;
-
     /** Generate code to load an integer constant.
      *  @param n     The integer to be loaded.
      */
@@ -221,46 +214,6 @@ void emitMinusOne(int tc) {
         }
     }
 
-    /** Construct a symbol to reflect the qualifying type that should
-     *  appear in the byte code as per JLS 13.1.
-     *
-     *  For {@literal target >= 1.2}: Clone a method with the qualifier as owner (except
-     *  for those cases where we need to work around VM bugs).
-     *
-     *  For {@literal target <= 1.1}: If qualified variable or method is defined in a
-     *  non-accessible class, clone it with the qualifier class as owner.
-     *
-     *  @param sym    The accessed symbol
-     *  @param site   The qualifier's type.
-     */
-    Symbol binaryQualifier(Symbol sym, Type site) {
-
-        if (site.hasTag(ARRAY)) {
-            if (sym == syms.lengthVar ||
-                sym.owner != syms.arrayClass)
-                return sym;
-            // array clone can be qualified by the array type in later targets
-            Symbol qualifier;
-            if ((qualifier = qualifiedSymbolCache.get(site)) == null) {
-                qualifier = new ClassSymbol(Flags.PUBLIC, site.tsym.name, site, syms.noSymbol);
-                qualifiedSymbolCache.put(site, qualifier);
-            }
-            return sym.clone(qualifier);
-        }
-
-        if (sym.owner == site.tsym ||
-            (sym.flags() & (STATIC | SYNTHETIC)) == (STATIC | SYNTHETIC)) {
-            return sym;
-        }
-
-        // leave alone methods inherited from Object
-        // JLS 13.1.
-        if (sym.owner == syms.objectType.tsym)
-            return sym;
-
-        return sym.clone(site.tsym);
-    }
-
     /** Insert a reference to given type in the constant pool,
      *  checking for an array with too many dimensions;
      *  return the reference's index.
@@ -2280,11 +2233,11 @@ public void visitIdent(JCIdent tree) {
             result = items.makeLocalItem((VarSymbol)sym);
         } else if ((sym.flags() & STATIC) != 0) {
             if (!isAccessSuper(env.enclMethod))
-                sym = binaryQualifier(sym, env.enclClass.type);
+                sym = types.binaryQualifier(sym, env.enclClass.type);
             result = items.makeStaticItem(sym);
         } else {
             items.makeThisItem().load();
-            sym = binaryQualifier(sym, env.enclClass.type);
+            sym = types.binaryQualifier(sym, env.enclClass.type);
             result = items.makeMemberItem(sym, nonVirtualForPrivateAccess(sym));
         }
     }
@@ -2337,7 +2290,7 @@ public void visitSelect(JCFieldAccess tree) {
                 result = items.makeDynamicItem(sym);
                 return;
             } else {
-                sym = binaryQualifier(sym, tree.selected.type);
+                sym = types.binaryQualifier(sym, tree.selected.type);
             }
             if ((sym.flags() & STATIC) != 0) {
                 if (!selectSuper && (ssym == null || ssym.kind != TYP))
@@ -2443,7 +2396,7 @@ public boolean genClass(Env<AttrContext> env, JCClassDecl cdef) {
             toplevel = null;
             endPosTable = null;
             nerrs = 0;
-            qualifiedSymbolCache.clear();
+            types.clearQualifiedSymbolCache();
         }
     }
 

test/langtools/tools/javac/lambda/methodReference/8059632/MethodRefQualifyingTypeTest.java

@@ -0,0 +1,77 @@
+/*
+ * Copyright (c) 2022, Oracle and/or its affiliates. All rights reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation.
+ *
+ * This code is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
+ * or visit www.oracle.com if you need additional information or have any
+ * questions.
+ */
+
+/**
+ * @test
+ * @bug 8059632
+ * @summary Method reference compilation uses incorrect qualifying type
+ * @compile MethodRefQualifyingTypeTest.java
+ * @compile MethodSupplierImpl.java
+ * @run main MethodRefQualifyingTypeTest
+ */
+
+import java.lang.reflect.Method;
+
+public class MethodRefQualifyingTypeTest {
+
+    static abstract class MethodSupplierImpl implements MethodSupplier {
+        class Inner {
+            MethodRefQualifyingTypeTest.MyFunctionalInterface createMethodReference() {
+                return MethodSupplierImpl.this::<Integer>m;
+            }
+        }
+    }
+
+    interface MethodSupplier {
+        int m(int a);
+    }
+
+    interface MyFunctionalInterface {
+        int invokeMethodReference(int a);
+    }
+
+    static class MethodInvoker {
+        public static void invoke() {
+            MyFunctionalInterface instance = null;
+            MethodSupplierImpl ms = new MethodSupplierImpl() {
+                    public int m(int a) {
+                        return a;
+                    }
+                };
+            instance = ms.new Inner().createMethodReference();
+            instance.invokeMethodReference(1);
+        }
+    }
+
+    public static void main(String argv[]) throws Exception {
+
+        // Without the fix for JDK-8059632, the invocation below will fail with
+        // java.lang.invoke.LambdaConversionException: Invalid receiver type class MethodRefQualifyingTypeTest$MethodSupplierImpl; not a subtype of implementation type interface MethodRefQualifyingTypeTest$MethodSupplier
+
+        // With the fix for JDK-8059632, the invocation would succeed since the bootstrap section
+        // would refer to the type of the receiver and not the type of the declaring interface,
+        // per JLS 13.1 (see "the qualifying type of the method invocation").
+
+        Class.forName("MethodRefQualifyingTypeTest$MethodInvoker").getMethod("invoke").invoke(null);
+    }
+}

test/langtools/tools/javac/lambda/methodReference/8059632/MethodSupplierImpl.java

@@ -0,0 +1,26 @@
+/*
+ * Copyright (c) 2022, Oracle and/or its affiliates. All rights reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation.
+ *
+ * This code is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
+ * or visit www.oracle.com if you need additional information or have any
+ * questions.
+ */
+
+class MethodRefQualifyingTypeTest$MethodSupplierImpl {
+    int m(int a) { return a; }
+}