[ASM 5.0] 3. BCI를 이용한 클래스 변형하기

이번에는 이전 시간에 만든 writeClass()를 이용해서 생성된 클래스를 변형을 해보도록 하겠습니다.

혹시나 "클래스 생성하기" 챕터를 건너뛰신 분들을 위해서 저~~~~ 아래쪽에 writeClass() 메소드 소스를 올려놓도록 하겠습니다.

자 그럼 시작해볼까요...

변형하는거는 그리 어렵지 않습니다. ClassVisitor를 이용하면 됩니다. 제일 첫번째 시간에 MyClassVisitor를 만들었으니 이걸 이용해보도록 하겠습니다.

큰 그림부터 한번 볼까요?? 순서는 아래와 같습니다.

1. ClassWriter 생성

2. ClassWriter를 이용하여 ClassVisitor 생성 ( ClassWriter를 인자로 넘겨줌 )

3. ClassReader를 이용하여 변형하고자하는 클래스 정보를 읽어들임

4. cr.accept( ClassVisitor, int ) 메소드를 이용해서 클래스 변형

실질적으로 변형을 하는 녀석은 바로 이 ClassVisitor라는 것을 알고 넘어가시면 됩니다.

자, 그럼 우선 클래스를 변형하기위한 메소드를 하나 만들어보죠.

private byte[] transformClass(){
        ClassWriter cw = new ClassWriter(0);
        ClassVisitor cv = new MyClassVisitor(Opcodes.ASM5, cw){};
        ClassReader cr = new ClassReader(writeClass());
        cr.accept(cv, 0);
        return cw.toByteArray();
}

여기서 writeClass()의 소스는 이 포스트 하단에 있습니다.

간단하게 소스설명을 드리자면 우선 cw를 만들고 cv를 생성할 때 인자로 넣어줍니다. 여기서 기존에 만들었던 MyClassVisitor에는 위 소스에서 사용한 생성자가 없으므로 하나를 추가로 생성해주셔야 합니다.

MyClassVisitor클래스에 아래 생성자를 하나 추가해주세요.

public MyClassVisitor(int version, ClassWriter cw) {
        super(version, cw);
}

다시 소스설명으로 넘어와서....

이렇게 cv를 생성을 했으면 이제 변형할 클래스를 읽어옵니다.

그리고 cr의 accept()를 이용해서 방문(?)하면서 변환을 합니다. 이렇게 변환이 된 클래스 정보는 cw에 들어가게 됩니다.

뭔가 간단한것 같으면서도 어지러운것 같죠??

자, 그럼 테스트를 해봐야 하겠죠..

우리가 만들었던 MyClassVisitor의 visit() 메소드를 한번 볼까요?

@Override
    public void visit(int version, int access, String name,
            String signature, String superName, String[] interfaces) {
        name = "BCItest/Modified";
        super.visit(version, access, name, signature, superName, interfaces);
        System.out.println(name + " extends " + superName + " {");
}

자, 위에서 오렌지색으로 표시된 부분이 추가된 부분입니다. 이게 뭘 할건지 짐작이 가시나요?

자 그리고 테스트 클래스에서 지난 시간에 했던 생성된 클래스 테스트해보는 소스를 조금만 수정해보겠습니다.

// 기존 테스트 소스

Class c = new MyClassLoader().defineClass("BCItest.Comparable", writeClass());
System.out.println(c.getName());

// 수정한 테스트 소스

Class c = new MyClassLoader().defineClass("BCItest.Comparable", transformClass());
System.out.println(c.getName());

변경된 부분은 오렌지색으로 표시했습니다.

이렇게 해서 실행을 하면 뭐가 나올까요?

ClassLoader로 읽어오는 과정에서 아래와 같은 오류가 납니다.

java.lang.NoClassDefFoundError: BCItest/Comparable (wrong name: BCItest/Modified)

자....우리가 클래스로더를 이용해서 로드하려고한 클래스명은 Comparable이었는데 transformClass()에서 만들어진 클래스는 Modified이기 때문입니다.

그럼 이번에는 BCItest.Modified를 읽어오도록 해봅시다.

Class c = new MyClassLoader().defineClass("BCItest.Modified", transformClass());
 

위 처럼 defineClass() 메소드에 들어가는 인자를 바꿔주세요.

이제 다시 실행해보면 BCItest.Modified 라고 출력이 되는 것을 확인하실 수 있을겁니다.

참고로 지난번에 했던 MyClassVisitor를 그대로 사용하신다면 클래스 구조가 출력되는 것도 보실 수가 있습니다.

어렵지 않죠?

자 이제 여러분도 BCI의 기본기를 모두 익힌 셈입니다.

클래스 읽기, 쓰기, 수정 까지 했으니까요.

ASM을 이용해서 클래스를 읽어오고, 생성 및 변환하는 부분까지 공부를 해보았는데요

제가 설명하는 능력이 약해서 첫시간부터 지금까지 한 내용을 소스를 첨부했으니 소스를 보면서 천천히 한번 공부해보세요 ^_^

다음시간에는 수정할때 최적화 관련된 부분에 대해서 알아보도록 하겠습니다.

그럼 여러분 안녀~엉~

출처 : ASM 4 가이드 문서

-----------------writeClass()-----------------------

BCItest 패키지를 기준으로 소스가 작성되었습니다.

private byte[] writeClass(){
        ClassWriter cw = new ClassWriter(0);
        cw.visit(Opcodes.V1_5, Opcodes.ACC_PUBLIC + Opcodes.ACC_ABSTRACT + Opcodes.ACC_INTERFACE,
                "BCItest/Comparable", null, "java/lang/Object",
                null);
        cw.visitField(Opcodes.ACC_PUBLIC + Opcodes.ACC_FINAL + Opcodes.ACC_STATIC, "LESS", "I",
        null, new Integer(-1)).visitEnd();
        cw.visitField(Opcodes.ACC_PUBLIC + Opcodes.ACC_FINAL + Opcodes.ACC_STATIC, "EQUAL", "I",
        null, new Integer(0)).visitEnd();
        cw.visitField(Opcodes.ACC_PUBLIC + Opcodes.ACC_FINAL + Opcodes.ACC_STATIC, "GREATER", "I",
        null, new Integer(1)).visitEnd();
        cw.visitMethod(Opcodes.ACC_PUBLIC + Opcodes.ACC_ABSTRACT, "compareTo",
        "(Ljava/lang/Object;)I", null, null).visitEnd();
        cw.visitEnd();

        return cw.toByteArray();
}