From f507d404c32cb75d76eaf3cd197a8bebd997702f Mon Sep 17 00:00:00 2001
From: Ondro Mihalyi <mihalyi@omnifish.ee>
Date: Mon, 2 Sep 2024 23:57:31 +0200
Subject: [PATCH] JPMS improvements: Experiments with GF Embedded

Discovered a flow in the solution for ProxyServiceImpl
---
 .../src/main/java/com/sun/ejb/codegen/ClassGenerator.java | 2 +-
 .../org/glassfish/weld/services/ProxyServicesImpl.java    | 8 ++++++--
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/appserver/ejb/ejb-container/src/main/java/com/sun/ejb/codegen/ClassGenerator.java b/appserver/ejb/ejb-container/src/main/java/com/sun/ejb/codegen/ClassGenerator.java
index 6b729bd898e..bff83afdc61 100644
--- a/appserver/ejb/ejb-container/src/main/java/com/sun/ejb/codegen/ClassGenerator.java
+++ b/appserver/ejb/ejb-container/src/main/java/com/sun/ejb/codegen/ClassGenerator.java
@@ -256,7 +256,7 @@ private static boolean useMethodHandles(final ClassLoader loader, final Class<?>
         }
         // The bootstrap CL used by embedded glassfish doesn't remember generated classes
         // Further ClassLoader.findClass calls will fail.
-        if (anchorClass == null || loader.getParent() == null || loader.getClass() == ASURLClassLoader.class) {
+        if (anchorClass == null || loader.getClass() == ASURLClassLoader.class) {
             return false;
         }
         // Use MethodHandles.Lookup only if the anchor run-time Package defined by CL.
diff --git a/appserver/web/weld-integration/src/main/java/org/glassfish/weld/services/ProxyServicesImpl.java b/appserver/web/weld-integration/src/main/java/org/glassfish/weld/services/ProxyServicesImpl.java
index 7797387ee6c..bf172a98295 100644
--- a/appserver/web/weld-integration/src/main/java/org/glassfish/weld/services/ProxyServicesImpl.java
+++ b/appserver/web/weld-integration/src/main/java/org/glassfish/weld/services/ProxyServicesImpl.java
@@ -50,7 +50,8 @@ public class ProxyServicesImpl implements ProxyServices {
 
     private final ClassLoaderHierarchy classLoaderHierarchy;
 
-    private Map<ClassLoader, ClassPool> classPoolMap = Collections.synchronizedMap(new WeakHashMap<>());
+    // probably shouldn't be static but moved to a singleton service
+    private static Map<ClassLoader, ClassPool> classPoolMap = Collections.synchronizedMap(new WeakHashMap<>());
 
     /**
      * @param services immediately used to find a {@link ClassLoaderHierarchy} service
@@ -59,13 +60,16 @@ public ProxyServicesImpl(final ServiceLocator services) {
         classLoaderHierarchy = services.getService(ClassLoaderHierarchy.class);
     }
 
+    /* This solution is not ideal - it creates the new class in the CL of the originalClass
+       (most often server CL or bootstrap CL), while the previous solution created it in the app classloader.
+    */
     @Override
     public Class<?> defineClass(final Class<?> originalClass, final String className, final byte[] classBytes,
         final int off, final int len) throws ClassFormatError {
         try {
             final String originalPackageName = originalClass.getPackageName();
             String modifiedClassName = originalClass.getName() + "_GlassFishWeldProxy";
-            final ClassLoader originalClassCL = getClassLoaderforBean(originalClass);
+            final ClassLoader originalClassCL = originalClass.getClassLoader();
             final ClassPool classPool = classPoolMap.computeIfAbsent(originalClassCL, cl -> new ClassPool());
             while (classPool.getOrNull(modifiedClassName) != null) {
                 modifiedClassName += "_";