Fix MemberName fields after Fast HCR and FSD

runtime/jcl/common/java_lang_invoke_MethodHandleNatives.cpp

@@ -44,11 +44,6 @@ extern "C" {
  * These constants are validated by the MethodHandleNatives$Constants.verifyConstants()
  * method when Java assertions are enabled
  */
-#define MN_IS_METHOD		0x00010000
-#define MN_IS_CONSTRUCTOR	0x00020000
-#define MN_IS_FIELD			0x00040000
-#define MN_IS_TYPE			0x00080000
-#define MN_CALLER_SENSITIVE	0x00100000
 
 #define MN_REFERENCE_KIND_SHIFT	24
 #define MN_REFERENCE_KIND_MASK	0xF		/* (flag >> MN_REFERENCE_KIND_SHIFT) & MN_REFERENCE_KIND_MASK */

runtime/jvmti/jvmtiClass.c

@@ -1195,6 +1195,11 @@ redefineClassesCommon(jvmtiEnv* env,
 			/* Fix JNI */
 			fixJNIRefs(currentThread, classPairs, TRUE, extensionsUsed);
 
+#if defined(J9VM_OPT_OPENJDK_METHODHANDLE)
+			/* Fix MemberNames (vmtarget) */
+			fixMemberNames(currentThread, classPairs);
+#endif /* defined(J9VM_OPT_OPENJDK_METHODHANDLE) */
+
 			/* Fix resolved constant pool references to point to new methods. */
 			fixConstantPoolsForFastHCR(currentThread, classPairs, methodPairs);
 
@@ -1254,6 +1259,11 @@ redefineClassesCommon(jvmtiEnv* env,
 				fixVTables_forNormalRedefine(currentThread, classPairs, methodPairs, FALSE, &methodEquivalences);
 			}
 
+#if defined(J9VM_OPT_OPENJDK_METHODHANDLE)
+			/* Fix MemberNames (vmtarget) */
+			fixMemberNames(currentThread, classPairs);
+#endif /* defined(J9VM_OPT_OPENJDK_METHODHANDLE) */
+
 			/* Restore breakpoints in the implicitly replaced classes */
 			restoreBreakpointsInClasses(currentThread, classPairs);
 

runtime/oti/j9nonbuilder.h

@@ -2247,6 +2247,16 @@ typedef struct J9ROMMethodHandleRef {
 #define MH_REF_INVOKEVIRTUAL  5
 #define MH_REF_PUTFIELD  3
 
+/* Constants mapped from java.lang.invoke.MethodHandleNatives$Constants
+ * These constants are validated by the MethodHandleNatives$Constants.verifyConstants()
+ * method when Java assertions are enabled
+ */
+#define MN_IS_METHOD		0x00010000
+#define MN_IS_CONSTRUCTOR	0x00020000
+#define MN_IS_FIELD			0x00040000
+#define MN_IS_TYPE			0x00080000
+#define MN_CALLER_SENSITIVE	0x00100000
+
 typedef struct J9ROMMethodRef {
 	U_32 classRefCPIndex;
 	J9SRP nameAndSignature;

runtime/oti/util_api.h

@@ -2174,6 +2174,11 @@ fixArrayClasses(J9VMThread * currentThread, J9HashTable* classHashTable);
 void
 fixJNIRefs (J9VMThread * currentThread, J9HashTable* classHashTable, BOOLEAN fastHCR, UDATA extensionsUsed);
 
+#if defined(J9VM_OPT_OPENJDK_METHODHANDLE)
+void
+fixMemberNames(J9VMThread * currentThread, J9HashTable * classHashTable);
+#endif /* defined(J9VM_OPT_OPENJDK_METHODHANDLE) */
+
 void
 fixConstantPoolsForFastHCR(J9VMThread *currentThread, J9HashTable *classPairs, J9HashTable *methodPair);
 

runtime/util/hshelp.c

@@ -90,6 +90,9 @@ static int compareClassDepth (const void *leftPair, const void *rightPair);
 static UDATA utfsAreIdentical(J9UTF8 * utf1, J9UTF8 * utf2);
 static UDATA areUTFPairsIdentical(J9UTF8 * leftUtf1, J9UTF8 * leftUtf2, J9UTF8 * rightUtf1, J9UTF8 * rightUtf2);
 static jvmtiError fixJNIMethodID(J9VMThread *currentThread, J9Method *oldMethod, J9Method *newMethod, BOOLEAN equivalent, UDATA extensionsUsed);
+#if defined(J9VM_OPT_OPENJDK_METHODHANDLE)
+static jvmtiIterationControl fixMemberNamesObjectIteratorCallback(J9JavaVM *vm, J9MM_IterateObjectDescriptor *objectDesc, void *userData);
+#endif /* defined(J9VM_OPT_OPENJDK_METHODHANDLE) */
 
 static jvmtiIterationControl fixHeapRefsHeapIteratorCallback(J9JavaVM *vm, J9MM_IterateHeapDescriptor *heapDesc, void *userData);
 static jvmtiIterationControl fixHeapRefsSpaceIteratorCallback(J9JavaVM *vm, J9MM_IterateSpaceDescriptor *spaceDesc, void *userData);
@@ -1701,6 +1704,75 @@ fixJNIRefs(J9VMThread * currentThread, J9HashTable * classPairs, BOOLEAN fastHCR
 }
 
 
+typedef struct J9ThreadHashPair {
+	J9VMThread *currentThread;
+	void *userData;
+} J9ThreadHashPair;
+
+#if defined(J9VM_OPT_OPENJDK_METHODHANDLE)
+void
+fixMemberNames(J9VMThread * currentThread, J9HashTable * classHashTable)
+{
+	PORT_ACCESS_FROM_JAVAVM(currentThread->javaVM);
+
+	J9ThreadHashPair data;
+	data.currentThread = currentThread;
+	data.userData = classHashTable;
+
+	/* iterate over all objects fixing up vmtarget refs if object is a MemberName */
+	currentThread->javaVM->memoryManagerFunctions->j9mm_iterate_all_objects(currentThread->javaVM, PORTLIB, 0, fixMemberNamesObjectIteratorCallback, &data);
+}
+
+static jvmtiIterationControl
+fixMemberNamesObjectIteratorCallback(J9JavaVM *vm, J9MM_IterateObjectDescriptor *objectDesc, void *userData)
+{
+	J9ThreadHashPair *data = userData;
+	J9VMThread *currentThread = data->currentThread;
+	J9HashTable *classHashTable = data->userData;
+	j9object_t object = objectDesc->object;
+	J9Class *clazz = J9OBJECT_CLAZZ_VM(vm, object);
+
+	if (NULL != classHashTable) {
+		if (clazz == J9VMJAVALANGINVOKEMEMBERNAME(vm) && 0 != J9OBJECT_U64_LOAD(currentThread, object, vm->vmindexOffset)) {
+			U_64 vmindex = J9OBJECT_U64_LOAD(currentThread, object, vm->vmindexOffset);
+
+			J9JVMTIClassPair exemplar;
+			J9JVMTIClassPair *result = NULL;
+
+			j9object_t membernameClazz = J9VMJAVALANGINVOKEMEMBERNAME_CLAZZ(currentThread, object);
+			exemplar.replacementClass.ramClass = J9VM_J9CLASS_FROM_HEAPCLASS(currentThread, membernameClazz);
+			result = hashTableFind(classHashTable, &exemplar);
+			if (NULL != result) {
+				jint flags = J9VMJAVALANGINVOKEMEMBERNAME_FLAGS(currentThread, object);
+				if (J9_ARE_ANY_BITS_SET(flags, MN_IS_FIELD)) {
+					/* Update vmtarget to vmindex->offset */
+					J9JNIFieldID *fieldID = (J9JNIFieldID *)vmindex;
+					J9ROMFieldShape *romField = fieldID->field;
+					UDATA offset = fieldID->offset;
+
+					if (J9_ARE_ANY_BITS_SET(romField->modifiers, J9AccStatic)) {
+						offset |= J9_SUN_STATIC_FIELD_OFFSET_TAG;
+						if (J9_ARE_ANY_BITS_SET(romField->modifiers, J9AccFinal)) {
+							offset |= J9_SUN_FINAL_FIELD_OFFSET_TAG;
+						}
+					}
+					J9OBJECT_U64_STORE(currentThread, object, vm->vmtargetOffset, (U_64)offset);
+
+				} else if (J9_ARE_ANY_BITS_SET(flags, MN_IS_METHOD | MN_IS_CONSTRUCTOR)) {
+					/* Update vmtarget to vmindex->method */
+					J9JNIMethodID *methodID = (J9JNIMethodID *)vmindex;
+					J9OBJECT_U64_STORE(currentThread, object, vm->vmtargetOffset, (U_64)methodID->method);
+
+				}
+			}
+		}
+	}
+
+	return JVMTI_ITERATION_CONTINUE;
+}
+#endif /* defined(J9VM_OPT_OPENJDK_METHODHANDLE) */
+
+
 /**
  * \brief  Reresolve the passed in constant pool
  * \ingroup jvmtiClass