sander/exceptional
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

last nicer making

Browse source
This commit is contained in:
Shautvast 2024-06-23 20:45:00 +02:00
parent 862aad4462
commit 3414c3d510
1 changed files with 16 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
agent/src/main/java/com/github/shautvast/exceptional/Agent.java
View file

@ -12,7 +12,6 @@ import java.nio.file.Files;
import java.nio.file.Path; import java.nio.file.Path;
import java.security.ProtectionDomain; import java.security.ProtectionDomain;
@SuppressWarnings("ALL")
public class Agent { public class Agent {
private static final String MESSAGE = "--- Exceptional agent active"; private static final String MESSAGE = "--- Exceptional agent active";
@ -62,20 +61,21 @@ public class Agent {
}); });
} }
private static void instrumentMethodWithMyExceptionLogger(ClassBuilder classBuilder, MethodModel mm) { private static void instrumentMethodWithMyExceptionLogger(ClassBuilder classBuilder, MethodModel methodModel) {
classBuilder.withMethod(mm.methodName(), mm.methodType(), mm.flags().flagsMask(), methodModel.code().ifPresent(code ->
methodBuilder -> classBuilder.withMethod(methodModel.methodName(), methodModel.methodType(), methodModel.flags().flagsMask(),
methodBuilder.transformCode(mm.code().get(), ((builder, element) -> { methodBuilder ->
if (element instanceof ReturnInstruction) { methodBuilder.transformCode(code, ((builder, element) -> {
builder.aload(0); // load `this` on the stack, ie the Throwable instance if (element instanceof ReturnInstruction) {
builder.invokestatic( // call my code builder.aload(0); // load `this` on the stack, ie the Throwable instance
ClassDesc.of("com.github.shautvast.exceptional.ExceptionLogger"), "log", builder.invokestatic( // call my code
MethodTypeDesc.ofDescriptor("(Ljava/lang/Throwable;)V")); ClassDesc.of("com.github.shautvast.exceptional.ExceptionLogger"), "log",
builder.return_(); MethodTypeDesc.ofDescriptor("(Ljava/lang/Throwable;)V"));
} else { builder.return_();
builder.with(element); } else {
} builder.with(element); // leave any other bytecode in place
}) }
)); })
)));
} }
} }