Tonny/jetbra
0
0
Fork 0
Code Pull Requests Activity

init

Browse Source
This commit is contained in:
novice.li
2023-11-25 14:55:08 +08:00
commit e060f1c114
29 changed files with 740 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
jetbra-agent/src/main/java/win/novice/li/AgentMain.java Normal file
View File

@@ -0,0 +1,50 @@
package win.novice.li;
import net.bytebuddy.agent.builder.AgentBuilder;
import net.bytebuddy.asm.Advice;
import net.bytebuddy.matcher.ElementMatchers;
import java.lang.instrument.Instrumentation;
import java.util.Set;
public class AgentMain {
public static void premain(String agentArgs, Instrumentation inst) throws Exception {
printLogo();
AgentBuilder agentBuilder = newAgentBuilder();
agentBuilder.type(ElementMatchers.named("java.security.cert.PKIXBuilderParameters"))
.transform((builder, typeDescription, classLoader, module, protectionDomain) -> builder
.visit(Advice.to(PKIXBuilderParameters.class)
.on(ElementMatchers.isConstructor().and(ElementMatchers.takesArgument(0, Set.class)))))
.asTerminalTransformation()
.type(ElementMatchers.named("sun.net.www.http.HttpClient"))
.transform((builder, typeDescription, classLoader, module, protectionDomain) -> builder
.visit(Advice.to(HttpClientAdvice.class)
.on(ElementMatchers.named("openServer"))))
.asTerminalTransformation()
.type(ElementMatchers.named("java.lang.System"))
.transform((builder, typeDescription, classLoader, module, protectionDomain) -> builder
.visit(Advice.to(SystemAdvice.class)
.on(ElementMatchers.named("getProperty"))))
.asTerminalTransformation()
.installOn(inst);
agentBuilder.installOn(inst);
}
static AgentBuilder newAgentBuilder() {
return new AgentBuilder.Default()
.with(AgentBuilder.RedefinitionStrategy.RETRANSFORMATION)
.with(AgentBuilder.InitializationStrategy.NoOp.INSTANCE)
.with(AgentBuilder.TypeStrategy.Default.REDEFINE)
.ignore(ElementMatchers.nameStartsWith("net.bytebuddy."));
}
static void printLogo() {
System.out.println(" _ _ _ \n" +
" | | ___| |_| |__ _ __ __ _ \n" +
" _ | |/ _ \\ __| '_ \\| '__/ _` |\n" +
"| |_| | __/ |_| |_) | | | (_| |\n" +
" \\___/ \\___|\\__|_.__/|_| \\__,_|");
}
}