Как правильно использовать Instrumentation.retransformClasses () из кода asm? - PullRequest
Новая
       11

Как правильно использовать Instrumentation.retransformClasses () из кода asm?

8 голосов
/ 02 марта 2012

Я использую библиотеку asm для выполнения некоторой модификации байт-кода Java - в частности, для изменения моих классов для реализации нового интерфейса и связанных методов. Мой текущий подход заключается в использовании основного API asm через javaagent. Я хотел бы сохранить этот динамический подход в отличие от статического изменения файлов .class.

На более высоком уровне моя проблема заключается в том, что если я решу изменить класс A, который выходит за пределы B, мне также необходимо изменить B. (Учитывая мое понимание того, как классы загружаются в JVM, я считаю, что класс B всегда будет передаваться трансформатору до класса А. (Пожалуйста, исправьте меня, если я ошибаюсь.) Учитывая это предположение, я думаю, что тогда мне нужно вернуться и ретрансформировать Б. Мой подход в этом бите кода: 

public byte[] transform(ClassLoader l, String name, Class<?> clazz, ProtectionDomain d, byte[] b) {
      throws IllegalClassFormatException {
    // **1**
    System.out.println("--->>> " + name);

    if (interestingClass(name)) {
        try {
            ClassReader cr = new ClassReader(b);
            ClassWriter cw = new ClassWriter(cr, ClassWriter.COMPUTE_MAXS | ClassWriter.COMPUTE_FRAMES);
            PyClassVisitorAdapter pv = new PyClassVisitorAdapter(cw, name);
            cr.accept(pv, 0);

            // **2** Retrieve the superclass and try to transform that
            if (! "Ljava/lang/Object;".equals(pv.getSuperName())) {
                String cName = classJvmToCanonical(pv.getSuperName());
                Class[] classes = inst.getAllLoadedClasses();
                for (Class c : classes) {
                    if (c.getName().equals(cName)) {
                        inst.retransformClasses(c);
                        break;
                    }
                }
            }

            // Dump the transformed class
            ClassReader cr2 = new ClassReader(cw.toByteArray());
            ClassWriter cw2 = new ClassWriter(cr2, 0);
            TraceClassVisitor tcv = new TraceClassVisitor(cw2, new PrintWriter(System.out));
            cr2.accept(tcv, 0);

            return cw2.toByteArray();
        } catch (Exception ex) {
            ex.printStackTrace();
            return null;
        }
    } else {
        return b;
    }
}

(inst - это дескриптор Instrumentation, который передается в конструктор)

Часть, с которой мне трудно работать, это блок, помеченный в комментариях **2**. Скажем еще раз, что A расширяет B, и я «заинтересован» в преобразовании A. Что я ожидаю, так это то, что я увижу имя суперкласса (B), печатаемого в **1** (но не преобразующегося, потому что я не не думаю, что это интересно на первом проходе), а затем, как только я доберусь до **2** и обнаружу, что суперклассом A является B, я должен попытаться повторно преобразовать B. В этот момент я ожидаю, что этот метод будет вызван снова ( через inst.retransformClasses()), и я бы увидел, что B печатается на **1**. Тем не менее, я не (Я добавил операторы печати и уверен, что достигну вызова ретрансформации. Я также проверил, что Instrumentation.isRetransformClassesSupported() и Instrumentation.isModifiableClass(c) оба возвращают значение true).

Мне кажется, я правильно настроил агента; установив в манифесте значения Can-Retransform-Classes и Can-Redefine-Classes равными true. Кроме того, когда я добавляю трансформатор к инструменту в методе агента premain, я делаю это: 

public static void premain(String agentArgs, Instrumentation inst) {
    inst.addTransformer(new PyClassFileTransformer(inst), true);
}

Есть какие-нибудь идеи относительно того, что я здесь делаю неправильно? Спасибо.

1 Ответ

1 голос
/ 02 марта 2012

Вы можете изменить свою стратегию использования байт-кода, поэтому, когда класс B загружен, вы найдете все его подклассы и в этот момент решите, нужно ли вам сейчас модифицировать класс B.Это можно оптимизировать, поддерживая хранилище метаданных класса или кэш в памяти (т. Е. Информацию об иерархии классов), поэтому вам не придется каждый раз загружать метаданные.

...