Java代理无法转换项目

时间:2016-04-22 09:54:02

标签: java bytecode javaagents

长话短说:

  • 我需要改造程序中的每个类(甚至是java) 在我的代理之前加载的库)。
  • 我找到了办法 但并不完全正常。我对新想法持开放态度。
  • 我的实际方法很奇怪:它应该在文件和控制台中打印相同的名称,但它们不是。我确信那些类到达了我的转换方法,因为如果我试图对它们进行检测 我收到了一个错误。

完整故事: 我创建了一个代理,以便在每个加载到我的项目中的类中注入一些自定义代码。我在运行时使用选项-javaagent添加此代理,这很好。

问题是当我的代理连接到JVM时,已经加载了很多类(例如java.io. *类),所以我的转换错过了一大堆类。 所以我的问题是:有一种方法可以用来记录我所缺少的那些课程吗?每个建议都更受欢迎。

现在我尝试过这样:

public class MyTransformer implements ClassFileTransformer {


 public static void premain(String agentArgs, Instrumentation inst) {
    final MyTransformer  t = MyTransformer .getInstance();
    inst.addTransformer(t, true);
    MyTransformer .log.info(t + " registered via JVM option -javaagent");

    // TEST
    // by the time we are attached, the classes to be
    // dumped may have been loaded already. So, check
    // for candidates in the loaded classes.
    Class[] classes = inst.getAllLoadedClasses();
    List<Class> candidates = new ArrayList<Class>();
    for (Class c : classes) {
        if (inst.isModifiableClass(c) && inst.isRetransformClassesSupported()){
            candidates.add(c);
        }
    }
    System.out.println("There are "+candidates.size()+" classes");
    try {
        // if we have matching candidates, then
        // retransform those classes so that we
        // will get callback to transform.
        if (! candidates.isEmpty()) {
            Iterator it = candidates.iterator();
            while(it.hasNext()){
                Class c = (Class)it.next();
                if(!c.getName().startsWith("javassist")){
                    System.out.println(" ========================> In Progress:"+c.getName());
                    inst.retransformClasses(c);
                }
            }
        }else{
            System.out.println("candidates.isEmpty()");
        }
    } catch (Exception e) {
        e.printStackTrace();
    }
  }

  public byte[] transform(ClassLoader loader, String className,
        Class<?> classBeingRedefined, ProtectionDomain protectionDomain,
        byte[] classfileBuffer) throws IllegalClassFormatException {

    byte[] byteCode = classfileBuffer;

    final String dot_classname    = className.replace('/', '.');
    // log classes that are going throug the instrumentor
    // just log if the java.io. is coming here
    try {
        PrintWriter out = new PrintWriter(new BufferedWriter(new FileWriter(path, true)));
        out.println(dot_classname);
        out.close();
     } catch (IOException ex) {
        Logger.getLogger(MyTransformer.class.getName()).log(Level.SEVERE, null, ex);
     }
     return byteCode;
  }
}

所以我的第一次尝试是接受&#34; premain&#34;每个类已经加载的方法,如果它可以通过方法&#34; retransfromClasses&#34;来转发它。

现在在我的转换方法中,我只有一个日志,可以将每个要转换的类写入文件。现在我的测试结果是:

There are 1486 classes  
==> In Progress:com.sun.org.apache.xerces.internal.parsers.AbstractXMLDocumentParser
==> In Progress:java.lang.Enum  
==> In Progress:java.lang.ThreadGroup  
==> In Progress:java.nio.file.FileSystem  
==> In Progress:java.util.regex.Pattern$Prolog  
==> In Progress:com.sun.org.apache.xerces.internal.dom.AttributeMap 
==> In Progress:java.util.regex.Matcher  
==> In Progress:org.apache.commons.beanutils.converters.ShortConverter 
==> In Progress:com.google.gson.JsonNull 
==> In Progress:java.util.concurrent.CopyOnWriteArrayList$COWIterator 
==> In Progress:java.util.concurrent.locks.ReentrantLock 
==> In Progress:java.lang.NoSuchMethodError 
==> In Progress:org.apache.commons.lang.BooleanUtils 
==> In Progress:java.lang.reflect.WeakCache$CacheValue 
==> In Progress:com.google.gson.internal.bind.TypeAdapters$33 
==> In Progress:java.lang.reflect.Type 
==> In Progress:sun.reflect.generics.scope.AbstractScope 
==> In Progress:org.apache.log4j.helpers.DateTimeDateFormat 
==> In Progress:sun.nio.cs.MS1252 
==> In Progress:java.lang.Integer$IntegerCache 
==> In Progress:com.sun.org.apache.xerces.internal.utils.SecuritySupport$3 
==> In Progress:org.apache.commons.configuration.MapConfiguration 
==> In Progress:org.apache.commons.beanutils.IntrospectionContext 
==> In Progress:java.io.Reader 
==> In Progress:java.util.WeakHashMap$Holder 
==> In Progress:java.util.ServiceLoader$LazyIterator 
==> In Progress:java.util.regex.Pattern$Branch 
==> In Progress:java.lang.IllegalMonitorStateException 
==> In Progress:java.util.regex.Pattern$Curly 
==> In Progress:org.apache.commons.configuration.resolver.EntityRegistry 
==> In Progress:java.io.IOException 
==> In Progress:java.io.FilterOutputStream 
==> In Progress:org.apache.log4j.LogManager 
==> In Progress:sun.util.logging.PlatformLogger$Level 
==> In Progress:java.nio.charset.CoderResult$1 
==> In Progress:com.google.gson.FieldNamingPolicy$5 
==> In Progress:com.google.gson.internal.ObjectConstructor 
==> In Progress:sun.util.calendar.BaseCalendar$Date

所以你可以注意到我可以找到我感兴趣的每个类(甚至是java.io)。现在,如果我们看看应该包含MyTransformer试图转换的每个类的文件,我们注意到没有常见的条目,这真的很奇怪。

org.apache.commons.beanutils.converters.ShortConverter
com.google.gson.JsonNull
org.apache.commons.lang.BooleanUtils
com.google.gson.internal.bind.TypeAdapters$33
org.apache.log4j.helpers.DateTimeDateFormat
org.apache.commons.configuration.MapConfiguration
org.apache.commons.beanutils.IntrospectionContext
org.apache.commons.configuration.resolver.EntityRegistry
org.apache.log4j.LogManager
com.google.gson.FieldNamingPolicy$5
com.google.gson.internal.ObjectConstructor
org.apache.log4j.Layout
com.google.gson.internal.bind.TypeAdapters
org.apache.log4j.PropertyConfigurator
com.sap.psr.vulas.java.JavaEnumId
org.apache.commons.collections.collection.AbstractSerializableCollectionDecorator
org.apache.commons.logging.impl.LogFactoryImpl
org.apache.commons.lang.text.StrTokenizer

我发现的另一个问题是,如果我尝试在Mytransformer.transform(..)方法中插入一些字节码中的篡改,它将破坏我的执行。我的意思是我正在使用的转换操作非常好,因为我在附加代理程序后加载的每个类上使用它。但不知何故,如果我尝试使用它,那么&#34;重新转换&#34;我会提出这个错误:

 ==> In Progress:org.apache.commons.lang.text.StrMatcher$TrimMatcher java.lang.UnsupportedOperationException: class redefinition failed: attempted to change the schema (add/remove fields)
        at sun.instrument.InstrumentationImpl.retransformClasses0(Native Method)
        at sun.instrument.InstrumentationImpl.retransformClasses(InstrumentationImpl.java:144)

我知道本机类的转换存在限制,但我的转换操作试图改变架构并不重要。我会调查这一点并发布更新。

2 个答案:

答案 0 :(得分:0)

This link与您提出的问题类似。看起来您可以修改本机方法的主体,但不能添加方法或字段

我可以在您的代码中看到您正在使用Javassist来执行此操作。我花了几个月试图让它工作,但它对我不起作用,所以我结果使用了ASM(顺便说一下工作)。

所以你是怎么做到的?首先,我会使用Attach API挂钩到rt.jar(因为它已经加载了你需要将你的代理连接到java进程。然后你可以从agentmain方法将变换器添加到java代理。然后为要编辑的类创建类阅读器,并接受扩展的ClassVisitor。然后,覆盖方法,如果是方法你想要的,用其他代码替换代码。你可以看到一个例子on my Github。你需要每次都转换它,因为它加载到内存中 - 你不必担心删除代码。

我希望这有帮助!如果您有任何疑问,请发表评论:)

答案 1 :(得分:0)

您是否已将所需的清单条目添加到您的代理商jar manifest.mf

你的表现如何.mf&#39;好像 ?它应该有类似

的东西

Can-Redefine-Classes: true

Can-Retransform-Classes: true

问候,

Grzesiek