From 90d83310eea10f4d867caa15dfd70c063e1b3b48 Mon Sep 17 00:00:00 2001 From: pengzhile Date: Tue, 15 Feb 2022 17:34:24 +0800 Subject: [PATCH] new feature: management transformer Signed-off-by: pengzhile --- README.md | 2 +- pom.xml | 2 +- .../java/com/janetfilter/core/Dispatcher.java | 78 +++++++++++-------- .../com/janetfilter/core/Environment.java | 14 +++- .../com/janetfilter/core/Initializer.java | 5 +- .../java/com/janetfilter/core/Launcher.java | 4 +- .../core/plugin/MyTransformer.java | 10 +++ .../core/plugin/PluginManager.java | 4 +- 8 files changed, 74 insertions(+), 45 deletions(-) diff --git a/README.md b/README.md index 693e2f3..818eb32 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,4 @@ -# ja-netfilter v2.2.3 +# ja-netfilter v2.3.0 ### A javaagent framework diff --git a/pom.xml b/pom.xml index 73f3697..7dde3b4 100644 --- a/pom.xml +++ b/pom.xml @@ -6,7 +6,7 @@ com.ja-netfilter ja-netfilter - 2.2.3 + 2.3.0 ja-netfilter A javaagent framework diff --git a/src/main/java/com/janetfilter/core/Dispatcher.java b/src/main/java/com/janetfilter/core/Dispatcher.java index 0c4cbbc..30eaa03 100644 --- a/src/main/java/com/janetfilter/core/Dispatcher.java +++ b/src/main/java/com/janetfilter/core/Dispatcher.java @@ -13,12 +13,17 @@ public final class Dispatcher implements ClassFileTransformer { private final Set classSet = new TreeSet<>(); private final Map> transformerMap = new HashMap<>(); private final List globalTransformers = new ArrayList<>(); + private final List manageTransformers = new ArrayList<>(); public Dispatcher(Environment environment) { this.environment = environment; } - public synchronized void addTransformer(MyTransformer transformer) { + public void addTransformer(MyTransformer transformer) { + if (null == transformer) { + return; + } + if (environment.isAttachMode() && !transformer.attachMode()) { DebugInfo.debug("Transformer: " + transformer.getClass().getName() + " is set to not load in attach mode, ignored."); return; @@ -29,16 +34,23 @@ public final class Dispatcher implements ClassFileTransformer { return; } - String className = transformer.getHookClassName(); - if (null == className) { - globalTransformers.add(transformer); - return; - } + synchronized (this) { + String className = transformer.getHookClassName(); + if (null == className) { + globalTransformers.add(transformer); - classSet.add(className.replace('/', '.')); - List transformers = transformerMap.computeIfAbsent(className, k -> new ArrayList<>()); + if (transformer.isManager()) { + manageTransformers.add(transformer); + } + + return; + } - transformers.add(transformer); + classSet.add(className.replace('/', '.')); + List transformers = transformerMap.computeIfAbsent(className, k -> new ArrayList<>()); + + transformers.add(transformer); + } } public void addTransformers(List transformers) { @@ -64,42 +76,40 @@ public final class Dispatcher implements ClassFileTransformer { } public byte[] transform(ClassLoader loader, String className, Class classBeingRedefined, ProtectionDomain protectionDomain, byte[] classFileBuffer) throws IllegalClassFormatException { - do { - if (null == className) { - break; - } + if (null == className) { + return classFileBuffer; + } - List transformers = transformerMap.get(className); - if (null == transformers) { - break; - } + List transformers = transformerMap.get(className); + List globalTransformers = null == transformers ? this.manageTransformers : this.globalTransformers; - int order = 0; + int order = 0; - try { - for (MyTransformer transformer : globalTransformers) { - transformer.before(className, classFileBuffer); - } + try { + for (MyTransformer transformer : globalTransformers) { + transformer.before(className, classFileBuffer); + } - for (MyTransformer transformer : globalTransformers) { - classFileBuffer = transformer.preTransform(className, classFileBuffer, order++); - } + for (MyTransformer transformer : globalTransformers) { + classFileBuffer = transformer.preTransform(className, classFileBuffer, order++); + } + if (null != transformers) { for (MyTransformer transformer : transformers) { classFileBuffer = transformer.transform(className, classFileBuffer, order++); } + } - for (MyTransformer transformer : globalTransformers) { - classFileBuffer = transformer.postTransform(className, classFileBuffer, order++); - } + for (MyTransformer transformer : globalTransformers) { + classFileBuffer = transformer.postTransform(className, classFileBuffer, order++); + } - for (MyTransformer transformer : globalTransformers) { - transformer.after(className, classFileBuffer); - } - } catch (Throwable e) { - DebugInfo.error("Transform class failed: " + className, e); + for (MyTransformer transformer : globalTransformers) { + transformer.after(className, classFileBuffer); } - } while (false); + } catch (Throwable e) { + DebugInfo.error("Transform class failed: " + className, e); + } return classFileBuffer; } diff --git a/src/main/java/com/janetfilter/core/Environment.java b/src/main/java/com/janetfilter/core/Environment.java index a1a4ad6..2deb81f 100644 --- a/src/main/java/com/janetfilter/core/Environment.java +++ b/src/main/java/com/janetfilter/core/Environment.java @@ -4,6 +4,7 @@ import com.janetfilter.core.utils.ProcessUtils; import com.janetfilter.core.utils.StringUtils; import java.io.File; +import java.lang.instrument.Instrumentation; public final class Environment { private final String pid; @@ -18,11 +19,14 @@ public final class Environment { private final String disabledPluginSuffix; private final boolean attachMode; - public Environment(File agentFile, boolean attachMode) { - this(agentFile, null, attachMode); + private final Instrumentation instrumentation; + + public Environment(Instrumentation instrumentation, File agentFile, boolean attachMode) { + this(instrumentation, agentFile, null, attachMode); } - public Environment(File agentFile, String app, boolean attachMode) { + public Environment(Instrumentation instrumentation, File agentFile, String app, boolean attachMode) { + this.instrumentation = instrumentation; this.agentFile = agentFile; baseDir = agentFile.getParentFile(); @@ -93,6 +97,10 @@ public final class Environment { return !attachMode; } + public Instrumentation getInstrumentation() { + return instrumentation; + } + @Override public String toString() { return "Environment: {" + diff --git a/src/main/java/com/janetfilter/core/Initializer.java b/src/main/java/com/janetfilter/core/Initializer.java index 963703f..a473933 100644 --- a/src/main/java/com/janetfilter/core/Initializer.java +++ b/src/main/java/com/janetfilter/core/Initializer.java @@ -7,13 +7,14 @@ import java.lang.instrument.Instrumentation; import java.util.Set; public class Initializer { - public static void init(Instrumentation inst, Environment environment) { + public static void init(Environment environment) { DebugInfo.useFile(environment.getLogsDir()); DebugInfo.info(environment.toString()); Dispatcher dispatcher = new Dispatcher(environment); - new PluginManager(inst, dispatcher, environment).loadPlugins(); + new PluginManager(dispatcher, environment).loadPlugins(); + Instrumentation inst = environment.getInstrumentation(); inst.addTransformer(dispatcher, true); inst.setNativeMethodPrefix(dispatcher, environment.getNativePrefix()); diff --git a/src/main/java/com/janetfilter/core/Launcher.java b/src/main/java/com/janetfilter/core/Launcher.java index 966523d..fae7ad6 100644 --- a/src/main/java/com/janetfilter/core/Launcher.java +++ b/src/main/java/com/janetfilter/core/Launcher.java @@ -12,7 +12,7 @@ import java.util.jar.JarFile; public class Launcher { public static final String ATTACH_ARG = "--attach"; - public static final String VERSION = "v2.2.3"; + public static final String VERSION = "v2.3.0"; private static boolean loaded = false; @@ -78,7 +78,7 @@ public class Launcher { return; } - Initializer.init(inst, new Environment(agentFile, args, attachMode)); // for some custom UrlLoaders + Initializer.init(new Environment(inst, agentFile, args, attachMode)); // for some custom UrlLoaders } private static void printUsage() { diff --git a/src/main/java/com/janetfilter/core/plugin/MyTransformer.java b/src/main/java/com/janetfilter/core/plugin/MyTransformer.java index 13964d8..66b2aad 100644 --- a/src/main/java/com/janetfilter/core/plugin/MyTransformer.java +++ b/src/main/java/com/janetfilter/core/plugin/MyTransformer.java @@ -20,6 +20,16 @@ public interface MyTransformer { return true; } + /** + * for global transformers only + * whether it is a management transformer + * + * @return return true to handle the transform of all classes + */ + default boolean isManager() { + return false; + } + /** * for global transformers only */ diff --git a/src/main/java/com/janetfilter/core/plugin/PluginManager.java b/src/main/java/com/janetfilter/core/plugin/PluginManager.java index c29d86f..127b15f 100644 --- a/src/main/java/com/janetfilter/core/plugin/PluginManager.java +++ b/src/main/java/com/janetfilter/core/plugin/PluginManager.java @@ -22,8 +22,8 @@ public final class PluginManager { private final Dispatcher dispatcher; private final Environment environment; - public PluginManager(Instrumentation inst, Dispatcher dispatcher, Environment environment) { - this.inst = inst; + public PluginManager(Dispatcher dispatcher, Environment environment) { + this.inst = environment.getInstrumentation(); this.dispatcher = dispatcher; this.environment = environment; }