From 4d4eb6ba6268f10037ea1e78cce492d92c760e5e Mon Sep 17 00:00:00 2001
From: Rafael Winterhalter <rafael.wth@gmail.com>
Date: Wed, 25 Sep 2024 18:59:32 +0200
Subject: [PATCH] Prepend actual class file.

---
 .../main/java/net/bytebuddy/build/Plugin.java | 76 +++++++++++++++++--
 1 file changed, 70 insertions(+), 6 deletions(-)

diff --git a/byte-buddy-dep/src/main/java/net/bytebuddy/build/Plugin.java b/byte-buddy-dep/src/main/java/net/bytebuddy/build/Plugin.java
index 0f4ca42b68..a8e796329c 100644
--- a/byte-buddy-dep/src/main/java/net/bytebuddy/build/Plugin.java
+++ b/byte-buddy-dep/src/main/java/net/bytebuddy/build/Plugin.java
@@ -28,6 +28,7 @@
 import net.bytebuddy.pool.TypePool;
 import net.bytebuddy.utility.CompoundList;
 import net.bytebuddy.utility.FileSystem;
+import net.bytebuddy.utility.StreamDrainer;
 import net.bytebuddy.utility.nullability.AlwaysNull;
 import net.bytebuddy.utility.nullability.MaybeNull;
 
@@ -2959,7 +2960,7 @@ public static Source ofTypes(Collection<? extends Class<?>> types) {
                 /**
                  * Represents a collection of types as an in-memory source.
                  *
-                 * @param types The types to represent.
+                 * @param types          The types to represent.
                  * @param versionedTypes A versioned mapping of types to represent.
                  * @return A source representing the supplied types.
                  */
@@ -4884,16 +4885,20 @@ public Summary apply(Source source, Target target, List<? extends Plugin.Factory
                                     while (name.startsWith("/")) {
                                         name = name.substring(1);
                                     }
-                                    if (name.endsWith(ClassFileLocator.CLASS_FILE_EXTENSION) && !name.endsWith(PACKAGE_INFO) && !name.equals(MODULE_INFO)) {
+                                    if (name.endsWith(ClassFileLocator.CLASS_FILE_EXTENSION)
+                                            && (!name.startsWith("META-INF") && name.startsWith(ClassFileLocator.META_INF_VERSIONS))
+                                            && !name.endsWith(PACKAGE_INFO)
+                                            && !name.endsWith(MODULE_INFO)) {
                                         try {
+                                            String typeName = name.substring(name.startsWith(ClassFileLocator.META_INF_VERSIONS)
+                                                    ? name.indexOf('/', ClassFileLocator.META_INF_VERSIONS.length()) + 1
+                                                    : 0, name.length() - ClassFileLocator.CLASS_FILE_EXTENSION.length()).replace('/', '.');
                                             dispatcher.accept(new Preprocessor(element,
-                                                    name.substring(name.startsWith(ClassFileLocator.META_INF_VERSIONS)
-                                                            ? name.indexOf('/', ClassFileLocator.META_INF_VERSIONS.length()) + 1
-                                                            : 0, name.length() - ClassFileLocator.CLASS_FILE_EXTENSION.length()).replace('/', '.'),
+                                                    typeName,
                                                     name.startsWith(ClassFileLocator.META_INF_VERSIONS)
                                                             ? Integer.parseInt(name.substring(ClassFileLocator.META_INF_VERSIONS.length(), name.indexOf('/', ClassFileLocator.META_INF_VERSIONS.length())))
                                                             : 0,
-                                                    classFileLocator,
+                                                    new SourceEntryPrependingClassFileLocator(typeName, element, classFileLocator),
                                                     typePool,
                                                     listener,
                                                     plugins,
@@ -4946,6 +4951,65 @@ public Summary apply(Source source, Target target, List<? extends Plugin.Factory
                 }
             }
 
+            /**
+             * A class file locator that shadows a given {@link Source.Element}'s type with the explicit element.
+             * This avoids that caching yields the wrong class file in case of multi-release jars.
+             */
+            @HashCodeAndEqualsPlugin.Enhance
+            protected static class SourceEntryPrependingClassFileLocator implements ClassFileLocator {
+
+                /**
+                 * The name of the represented type.
+                 */
+                private final String name;
+
+                /**
+                 * The corresponding source element.
+                 */
+                private final Source.Element element;
+
+                /**
+                 * The actual class file locator to query for all other types.
+                 */
+                private final ClassFileLocator delegate;
+
+                /**
+                 * Creates a class file locator that prepends a {@link Source.Element}.
+                 *
+                 * @param name     The name of the represented type.
+                 * @param element  The corresponding source element.
+                 * @param delegate The actual class file locator to query for all other types.
+                 */
+                protected SourceEntryPrependingClassFileLocator(String name, Source.Element element, ClassFileLocator delegate) {
+                    this.name = name;
+                    this.element = element;
+                    this.delegate = delegate;
+                }
+
+                /**
+                 * {@inheritDoc}
+                 */
+                public Resolution locate(String name) throws IOException {
+                    if (name.endsWith(this.name)) {
+                        InputStream inputStream = element.getInputStream();
+                        try {
+                            return new Resolution.Explicit(StreamDrainer.DEFAULT.drain(inputStream));
+                        } finally {
+                            inputStream.close();
+                        }
+                    } else {
+                        return delegate.locate(name);
+                    }
+                }
+
+                /**
+                 * {@inheritDoc}
+                 */
+                public void close() throws IOException {
+                    delegate.close();
+                }
+            }
+
             /**
              * A preprocessor for a parallel plugin engine.
              */