[fixes #3406] Find jdt.core classloader if initialized in batch mode

Rawi01 · rspilker · commit 250bc0660b09 · 2023-04-16T18:10:19.000+02:00
diff --git a/src/eclipseAgent/lombok/launch/PatchFixesHider.java b/src/eclipseAgent/lombok/launch/PatchFixesHider.java
@@ -201,17 +201,49 @@ public static final class Transform {
 		private static synchronized void init(ClassLoader prepend) {
 			if (TRANSFORM != null) return;
 			
-			Main.prependClassLoader(prepend);
+			prependClassLoader(prepend);
+			if (!prepend.toString().contains("org.eclipse.jdt.core:")) {
+				ClassLoader jdtCoreClassLoader = findJdtCoreClassLoader(prepend);
+				prependClassLoader(jdtCoreClassLoader);
+			}
+			Class<?> shadowed = Util.shadowLoadClass("lombok.eclipse.TransformEclipseAST");
+			TRANSFORM = Util.findMethodAnyArgs(shadowed, "transform");
+			TRANSFORM_SWAPPED = Util.findMethodAnyArgs(shadowed, "transform_swapped");
+		}
+		
+		private static void prependClassLoader(ClassLoader classLoader) {
+			Main.prependClassLoader(classLoader);
 			try {
 				ClassLoader currentClassLoader = Transform.class.getClassLoader();
+				
 				Method prependParentMethod = Permit.getMethod(currentClassLoader.getClass(), "prependParent", ClassLoader.class);
-				Permit.invoke(prependParentMethod, currentClassLoader, prepend);
+				Permit.invoke(prependParentMethod, currentClassLoader, classLoader);
 			} catch (Throwable t) {
 				// Ignore
 			}
-			Class<?> shadowed = Util.shadowLoadClass("lombok.eclipse.TransformEclipseAST");
-			TRANSFORM = Util.findMethodAnyArgs(shadowed, "transform");
-			TRANSFORM_SWAPPED = Util.findMethodAnyArgs(shadowed, "transform_swapped");
+		}
+		
+		private static ClassLoader findJdtCoreClassLoader(ClassLoader classLoader) {
+			try {
+				Method getBundleMethod = Permit.getMethod(classLoader.getClass(), "getBundle");
+				Object bundle = Permit.invoke(getBundleMethod, classLoader);
+				
+				Method getBundleContextMethod = Permit.getMethod(bundle.getClass(), "getBundleContext");
+				Object bundleContext = Permit.invoke(getBundleContextMethod, bundle);
+				
+				Method getBundlesMethod = Permit.getMethod(bundleContext.getClass(), "getBundles");
+				Object[] bundles = (Object[]) Permit.invoke(getBundlesMethod, bundleContext);
+				
+				for (Object searchBundle : bundles) {
+					if (searchBundle.toString().startsWith("org.eclipse.jdt.core_")) {
+						Method getModuleClassLoaderMethod = Permit.getMethod(searchBundle.getClass(), "getModuleClassLoader", boolean.class);
+						return (ClassLoader) Permit.invoke(getModuleClassLoaderMethod, searchBundle, false);
+					}
+				}
+			} catch (Throwable t) {
+				// Ignore
+			}
+			return null;
 		}
 		
 		public static void transform(Object parser, Object ast) throws IOException {
