working now this is
This commit is contained in:
Shautvast
parent
02b15b3e72
commit
8f516d5612
6 changed files with 215 additions and 106 deletions
49
agent/pom.xml
Normal file
49
agent/pom.xml
Normal file
|
|
@ -0,0 +1,49 @@
|
|||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project xmlns="http://maven.apache.org/POM/4.0.0"
|
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
|
||||
<modelVersion>4.0.0</modelVersion>
|
||||
|
||||
<parent>
|
||||
<groupId>org.github.shautvast.exceptional</groupId>
|
||||
<artifactId>exceptional-parent</artifactId>
|
||||
<version>1.0-SNAPSHOT</version>
|
||||
</parent>
|
||||
|
||||
<artifactId>exceptional-agent</artifactId>
|
||||
<version>1.0-SNAPSHOT</version>
|
||||
|
||||
<properties>
|
||||
<maven.compiler.source>22</maven.compiler.source>
|
||||
<maven.compiler.target>22</maven.compiler.target>
|
||||
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
|
||||
</properties>
|
||||
<build>
|
||||
<plugins>
|
||||
<plugin>
|
||||
<groupId>org.apache.maven.plugins</groupId>
|
||||
<artifactId>maven-compiler-plugin</artifactId>
|
||||
<configuration>
|
||||
<source>22</source>
|
||||
<target>22</target>
|
||||
<compilerArgs>--enable-preview</compilerArgs>
|
||||
</configuration>
|
||||
</plugin>
|
||||
<plugin>
|
||||
<artifactId>maven-jar-plugin</artifactId>
|
||||
<version>3.1.0</version>
|
||||
<configuration>
|
||||
<archive>
|
||||
<manifestEntries>
|
||||
<Premain-Class>com.github.shautvast.exceptional.Agent</Premain-Class>
|
||||
<Can-Retransform-Classes>true</Can-Retransform-Classes>
|
||||
<Can-Redefine-Classes>true</Can-Redefine-Classes>
|
||||
<Boot-Class-Path>/Users/Shautvast/dev/exceptional/lib/target/exceptional-lib-1.0-SNAPSHOT.jar</Boot-Class-Path>
|
||||
</manifestEntries>
|
||||
</archive>
|
||||
</configuration>
|
||||
</plugin>
|
||||
</plugins>
|
||||
</build>
|
||||
|
||||
</project>
|
||||
111
agent/src/main/java/com/github/shautvast/exceptional/Agent.java
Normal file
111
agent/src/main/java/com/github/shautvast/exceptional/Agent.java
Normal file
|
|
@ -0,0 +1,111 @@
|
|||
package com.github.shautvast.exceptional;
|
||||
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.lang.classfile.*;
|
||||
import java.lang.constant.ClassDesc;
|
||||
import java.lang.constant.ConstantDescs;
|
||||
import java.lang.constant.MethodTypeDesc;
|
||||
import java.lang.instrument.ClassDefinition;
|
||||
import java.lang.instrument.ClassFileTransformer;
|
||||
import java.lang.instrument.IllegalClassFormatException;
|
||||
import java.lang.instrument.Instrumentation;
|
||||
import java.lang.reflect.Modifier;
|
||||
import java.security.ProtectionDomain;
|
||||
|
||||
@SuppressWarnings("ALL")
|
||||
public class Agent {
|
||||
|
||||
private static final String MESSAGE = "--- Exceptional agent active";
|
||||
|
||||
public static void premain(String agentArgs, Instrumentation instrumentation) throws Exception {
|
||||
instrumentation.addTransformer(new ClassFileTransformer() {
|
||||
@Override
|
||||
public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) {
|
||||
return instrumentTheThrowable(className, classfileBuffer);
|
||||
}
|
||||
|
||||
@Override
|
||||
public byte[] transform(Module module, ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {
|
||||
return instrumentTheThrowable(className, classfileBuffer);
|
||||
}
|
||||
}, true);
|
||||
|
||||
InputStream stream = Agent.class.getResourceAsStream("/java/lang/Throwable.class");
|
||||
byte[] bytecode = readFully(stream);
|
||||
|
||||
instrumentation.redefineClasses(new ClassDefinition(Throwable.class, bytecode));
|
||||
}
|
||||
|
||||
private static byte[] readFully(InputStream stream) throws IOException {
|
||||
ByteArrayOutputStream buffer = new ByteArrayOutputStream();
|
||||
int nRead;
|
||||
byte[] data = new byte[1024];
|
||||
while ((nRead = stream.read(data, 0, data.length)) != -1) {
|
||||
buffer.write(data, 0, nRead);
|
||||
}
|
||||
|
||||
buffer.flush();
|
||||
return buffer.toByteArray();
|
||||
}
|
||||
|
||||
private static byte[] instrumentTheThrowable(String className, byte[] classfileBuffer) {
|
||||
if (className.equals("java/lang/Throwable")) {
|
||||
ClassFile classFile = ClassFile.of();
|
||||
ClassModel classModel = classFile.parse(classfileBuffer);
|
||||
|
||||
return classFile.build(classModel.thisClass().asSymbol(), classBuilder -> {
|
||||
for (ClassElement ce : classModel) {
|
||||
if (ce instanceof MethodModel mm) {
|
||||
String signature = mm.methodName() + mm.methodType().stringValue();
|
||||
if (signature.equals("<init>()V")) {
|
||||
classBuilder.withMethod(mm.methodName(), mm.methodType(), Modifier.PUBLIC,
|
||||
methodBuilder -> methodBuilder.withCode(
|
||||
cb -> {
|
||||
// recreate existing code for this method, because... I don't know how to simply add the new code at the end🫣
|
||||
ClassDesc throwable = ClassDesc.of("java.lang.Throwable");
|
||||
cb.aload(0);
|
||||
cb.invokespecial(ConstantDescs.CD_Object, "<init>", MethodTypeDesc.ofDescriptor("()V"));
|
||||
cb.aload(0);
|
||||
cb.aload(0);
|
||||
cb.putfield(throwable, "cause", ClassDesc.ofDescriptor("Ljava/lang/Throwable;"));
|
||||
cb.aload(0);
|
||||
cb.getstatic(throwable, "UNASSIGNED_STACK", ClassDesc.ofDescriptor("[Ljava/lang/StackTraceElement;"));
|
||||
cb.putfield(throwable, "stackTrace", ClassDesc.ofDescriptor("[Ljava/lang/StackTraceElement;"));
|
||||
cb.aload(0);
|
||||
cb.invokevirtual(throwable, "fillInStackTrace", MethodTypeDesc.ofDescriptor("()Ljava/lang/Throwable;"));
|
||||
cb.pop();
|
||||
// cb.getstatic(throwable, "jfrTracing", ConstantDescs.CD_Boolean);
|
||||
// Label end = cb.newLabel();
|
||||
// cb.ifeq(end);
|
||||
// cb.aload(0);
|
||||
// cb.invokevirtual(ConstantDescs.CD_Object, "getClass", MethodTypeDesc.ofDescriptor("()Ljava/lang/Class;"));
|
||||
// cb.aconst_null();
|
||||
// cb.invokestatic(ClassDesc.of("jdk.internal.event.ThrowableTracer"), "traceThrowable", MethodTypeDesc.ofDescriptor("(Ljava/lang/Class;Ljava/lang/String;)V"));
|
||||
// cb.labelBinding(end);
|
||||
|
||||
cb.aload(0);
|
||||
cb.invokestatic(
|
||||
ClassDesc.of("com.github.shautvast.exceptional.ThrowableHandler"), "handle",
|
||||
MethodTypeDesc.ofDescriptor("(Ljava/lang/Throwable;)V"));
|
||||
cb.return_();
|
||||
}
|
||||
));
|
||||
continue;
|
||||
}
|
||||
}
|
||||
classBuilder.with(ce);
|
||||
}
|
||||
});
|
||||
} else {
|
||||
return classfileBuffer;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//getstatic #7 // Field java/lang/System.out:Ljava/io/PrintStream;
|
||||
//ldc #13 // class com/github/shautvast/exceptional/Foo
|
||||
//invokevirtual #15 // Method java/lang/Class.getModule:()Ljava/lang/Module;
|
||||
//invokevirtual #21 // Method java/lang/Module.getName:()Ljava/lang/String;
|
||||
//invokevirtual #27 // Method java/io/PrintStream.println:(Ljava/lang/String;)V
|
||||
37
lib/pom.xml
Normal file
37
lib/pom.xml
Normal file
|
|
@ -0,0 +1,37 @@
|
|||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project xmlns="http://maven.apache.org/POM/4.0.0"
|
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
|
||||
<modelVersion>4.0.0</modelVersion>
|
||||
|
||||
<parent>
|
||||
<groupId>org.github.shautvast.exceptional</groupId>
|
||||
<artifactId>exceptional-parent</artifactId>
|
||||
<version>1.0-SNAPSHOT</version>
|
||||
</parent>
|
||||
|
||||
<artifactId>exceptional-lib</artifactId>
|
||||
<version>1.0-SNAPSHOT</version>
|
||||
|
||||
<properties>
|
||||
<maven.compiler.source>22</maven.compiler.source>
|
||||
<maven.compiler.target>22</maven.compiler.target>
|
||||
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
|
||||
</properties>
|
||||
|
||||
<build>
|
||||
<plugins>
|
||||
<plugin>
|
||||
<groupId>org.apache.maven.plugins</groupId>
|
||||
<artifactId>maven-compiler-plugin</artifactId>
|
||||
<configuration>
|
||||
<source>22</source>
|
||||
<target>22</target>
|
||||
<compilerArgs>--enable-preview</compilerArgs>
|
||||
</configuration>
|
||||
</plugin>
|
||||
|
||||
</plugins>
|
||||
</build>
|
||||
|
||||
</project>
|
||||
|
|
@ -0,0 +1,11 @@
|
|||
package com.github.shautvast.exceptional;
|
||||
|
||||
import java.util.Arrays;
|
||||
|
||||
@SuppressWarnings("unused")
|
||||
public class ThrowableHandler {
|
||||
public static void handle(Throwable throwable) {
|
||||
System.out.print("Handling exception:");
|
||||
Arrays.stream(throwable.getStackTrace()).forEach(System.out::println);
|
||||
}
|
||||
}
|
||||
40
pom.xml
40
pom.xml
|
|
@ -4,40 +4,14 @@
|
|||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
|
||||
<modelVersion>4.0.0</modelVersion>
|
||||
|
||||
<groupId>org.example</groupId>
|
||||
<artifactId>Exceptional</artifactId>
|
||||
<groupId>org.github.shautvast.exceptional</groupId>
|
||||
<artifactId>exceptional-parent</artifactId>
|
||||
<version>1.0-SNAPSHOT</version>
|
||||
<packaging>pom</packaging>
|
||||
|
||||
<properties>
|
||||
<maven.compiler.source>21</maven.compiler.source>
|
||||
<maven.compiler.target>21</maven.compiler.target>
|
||||
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
|
||||
</properties>
|
||||
<build>
|
||||
<plugins>
|
||||
<plugin>
|
||||
<groupId>org.apache.maven.plugins</groupId>
|
||||
<artifactId>maven-compiler-plugin</artifactId>
|
||||
<configuration>
|
||||
<source>22</source>
|
||||
<target>22</target>
|
||||
<compilerArgs>--enable-preview</compilerArgs>
|
||||
</configuration>
|
||||
</plugin>
|
||||
<plugin>
|
||||
<artifactId>maven-jar-plugin</artifactId>
|
||||
<version>3.1.0</version>
|
||||
<configuration>
|
||||
<archive>
|
||||
<manifestEntries>
|
||||
<Premain-Class>com.github.shautvast.exceptional.Agent</Premain-Class>
|
||||
<Can-Retransform-Classes>true</Can-Retransform-Classes>
|
||||
<Can-Redefine-Classes>true</Can-Redefine-Classes>
|
||||
</manifestEntries>
|
||||
</archive>
|
||||
</configuration>
|
||||
</plugin>
|
||||
</plugins>
|
||||
</build>
|
||||
<modules>
|
||||
<module>agent</module>
|
||||
<module>lib</module>
|
||||
</modules>
|
||||
|
||||
</project>
|
||||
|
|
@ -1,73 +0,0 @@
|
|||
package com.github.shautvast.exceptional;
|
||||
|
||||
import java.io.InputStream;
|
||||
import java.lang.classfile.*;
|
||||
import java.lang.constant.ClassDesc;
|
||||
import java.lang.constant.MethodTypeDesc;
|
||||
import java.lang.instrument.ClassDefinition;
|
||||
import java.lang.instrument.ClassFileTransformer;
|
||||
import java.lang.instrument.IllegalClassFormatException;
|
||||
import java.lang.instrument.Instrumentation;
|
||||
import java.security.ProtectionDomain;
|
||||
|
||||
@SuppressWarnings("ALL")
|
||||
public class Agent {
|
||||
|
||||
private static final String MESSAGE = "--- Exceptional agent active";
|
||||
|
||||
public static void premain(String agentArgs, Instrumentation instrumentation) throws Exception {
|
||||
instrumentation.addTransformer(new ClassFileTransformer() {
|
||||
@Override
|
||||
public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) {
|
||||
return instrumentTheThrowable(className, classfileBuffer);
|
||||
}
|
||||
|
||||
@Override
|
||||
public byte[] transform(Module module, ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {
|
||||
return instrumentTheThrowable(className, classfileBuffer);
|
||||
}
|
||||
}, true);
|
||||
|
||||
InputStream stream = Agent.class.getResourceAsStream("/java/lang/Throwable.class");
|
||||
byte[] b = new byte[stream.available()];
|
||||
stream.read(b);
|
||||
|
||||
instrumentation.redefineClasses(new ClassDefinition(Throwable.class, b));
|
||||
}
|
||||
|
||||
private static byte[] instrumentTheThrowable(String className, byte[] classfileBuffer) {
|
||||
if (className.equals("java/lang/Throwable")) {
|
||||
|
||||
ClassFile classFile = ClassFile.of();
|
||||
ClassModel classModel = classFile.parse(classfileBuffer);
|
||||
|
||||
return classFile.build(classModel.thisClass().asSymbol(), classBuilder -> {
|
||||
for (ClassElement ce : classModel) {
|
||||
if (ce instanceof MethodModel mm) {
|
||||
String signature = mm.methodName() + "." + mm.methodType().stringValue();
|
||||
if (signature.equals("<init>.()V")) {
|
||||
classBuilder.withMethod(mm.methodName(), mm.methodType(), 1,
|
||||
methodBuilder -> methodBuilder.withCode(
|
||||
codebuilder ->
|
||||
// codebuilder.invokeInstruction(Opcode.LDC, );
|
||||
codebuilder.invokeInstruction(Opcode.INVOKESTATIC,
|
||||
ClassDesc.of("com.github.shautvast.exceptional.Agent"), "foo",
|
||||
MethodTypeDesc.ofDescriptor("()V"),
|
||||
false)
|
||||
));
|
||||
continue;
|
||||
}
|
||||
}
|
||||
classBuilder.with(ce);
|
||||
}
|
||||
});
|
||||
} else {
|
||||
return classfileBuffer;
|
||||
}
|
||||
}
|
||||
|
||||
public static void foo() {
|
||||
System.out.println("foo");
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Add table
Reference in a new issue