[java] Byte Buddy를 사용하여 인스턴스 생성 가로채는 방법은?

20 Nov 2023

java

인스턴스 생성을 가로채는 방법은 다음과 같습니다:

Byte Buddy 의존성 추가 먼저 프로젝트에 Byte Buddy를 추가해야 합니다. Maven을 사용하는 경우, pom.xml 파일에 다음과 같이 의존성을 추가합니다: ```xml

net.bytebuddy byte-buddy 1.10.19

2. InInterceptor 클래스 정의
인스턴스 생성을 가로채기 위해 Byte Buddy에서 제공하는 `AgentBuilder`를 사용합니다. 아래와 같이 `InInterceptor` 클래스를 만듭니다:

```java
import net.bytebuddy.agent.builder.AgentBuilder;
import net.bytebuddy.agent.builder.AgentBuilder.Transformer.ForMethodElement;
import net.bytebuddy.implementation.MethodDelegation;
import net.bytebuddy.implementation.bind.annotation.AllArguments;
import net.bytebuddy.implementation.bind.annotation.Origin;
import net.bytebuddy.implementation.bind.annotation.RuntimeType;
import net.bytebuddy.implementation.bind.annotation.SuperCall;
import net.bytebuddy.implementation.bind.annotation.This;
import net.bytebuddy.jar.asm.Opcodes;
import net.bytebuddy.matcher.ElementMatchers;

import java.lang.instrument.Instrumentation;
import java.lang.reflect.Constructor;
import java.util.concurrent.Callable;
import java.util.concurrent.atomic.AtomicReference;

public class InInterceptor {
    public static void premain(String agentArgs, Instrumentation instrumentation) {
        AgentBuilder agentBuilder = new AgentBuilder.Default()
                .with(AgentBuilder.RedefinitionStrategy.RETRANSFORMATION)
                .type(ElementMatchers.any())
                .transform(new ForMethodElement((builder, typeDescription, classLoader, module) ->
                        builder.visit(
                                MethodDelegation.to(Interceptor.class)
                                                .appendParameterBinder(RuntimeType.class, AllArguments.class)
                                                .appendParameterBinder(SuperCall.class, Origin.class, This.class)
                                                .appendParameterBinder(This.class, Origin.class, SuperCall.class)
                        )
                ));

        agentBuilder.installOn(instrumentation);
    }

    public static class Interceptor {
        public static AtomicReference<Constructor<?>> interceptedConstructor = new AtomicReference<>();

        @RuntimeType
        public static Object intercept(@SuperCall Callable<?> zuper, @This Object instance,
                                       @AllArguments Object[] arguments, @Origin Constructor<?> constructor) throws Exception {
            interceptedConstructor.set(constructor);

            // 여기에서 인스턴스 생성 가로채기가 가능합니다.
            // 원하는 동작을 수행할 수 있습니다.

            return zuper.call();
        }
    }
}

프로그램에 InInterceptor 추가 애플리케이션의 main 메서드에서 premain을 호출하여 InInterceptor를 프로그램에 추가합니다:

public class Main {
    public static void main(String[] args) {
        try {
            Instrumentation instrumentation = ByteBuddyAgent.install();
            InInterceptor.premain(null, instrumentation);

            // 프로그램 실행
            // 실제 인스턴스 생성에 InInterceptor 가 원하는 동작을 추가할 수 있습니다.

        } catch (Throwable throwable) {
            throwable.printStackTrace();
        }
    }
}

위와 같이 Byte Buddy를 사용하여 인스턴스 생성을 가로채고 수정할 수 있습니다. Interceptor 클래스의 intercept 메서드에서 원하는 동작을 구현하면 됩니다.

참고 자료: