Protect self-attach from execution as source file.

raphw · Dec 22, 2024 · da6026b · da6026b
1 parent 94f0e35
commit da6026b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/byte-buddy-agent/src/main/java/net/bytebuddy/agent/ByteBuddyAgent.java b/byte-buddy-agent/src/main/java/net/bytebuddy/agent/ByteBuddyAgent.java
@@ -1447,8 +1447,8 @@ private static File trySelfResolve(Class<?> installer) throws IOException {
                 }
                 URL location = codeSource.getLocation();
                 if (!location.getProtocol().equals("file")
-                        || location.getPath() == null
-                        || !location.getPath().endsWith(".class")) {
+                        || location.getPath() != null
+                        && location.getPath().endsWith(".java")) {
                     return null;
                 }
                 File agentJar;