byte-buddy icon indicating copy to clipboard operation
byte-buddy copied to clipboard

Bytebuddy Advice not called

Open SaiprasadKrishnamurthy opened this issue 1 year ago • 1 comments

Newbie Alert!

This is my main application.

@SpringBootApplication(exclude = {DataSourceAutoConfiguration.class})
public class DataTransformationServiceApplication {
    public static void main(String[] args) throws Exception {
        DynamicType.Unloaded<BodyNode> unloaded = new ByteBuddy()
                .redefine(BodyNode.class)
                .visit(Advice.to(MyAdvices.class).on(ElementMatchers.isMethod()))
                .make();

        Class<? extends BodyNode> loaded = unloaded
                .load(ClassLoader.getSystemClassLoader(), ClassLoadingStrategy.Default.CHILD_FIRST)
                .getLoaded();



        SpringApplication.run(DataTransformationServiceApplication.class, args);
    }

My advice looks like this:

public class MyAdvices {
    @Advice.OnMethodEnter(suppress = Throwable.class)
    static long enter(@Advice.This Object thisObject,
                      @Advice.Origin String origin,
                      @Advice.Origin("#t #m") String detaildOrigin,
                      @Advice.AllArguments Object[] ary){

        System.out.println("Inside enter method . . .  ");

        if(ary != null) {
            for(int i =0 ; i < ary.length ; i++){
                System.out.println("Argument: " + i + " is " + ary[i]);
            }
        }

        System.out.println("Origin :" + origin);
        System.out.println("Detailed Origin :" + detaildOrigin);

        return System.nanoTime();
    }

    @Advice.OnMethodExit(suppress = Throwable.class, onThrowable = Throwable.class)
    static void exit(@Advice.Enter long time){
        System.out.println("Inside exit method . . .");
        System.out.println("Method Execution Time: " + (System.nanoTime() - time) + " nano seconds");
    }
}

My 3rd party library when it calls BodyNode.render(...) with a hard coded object within it's class (using new), my advice isn't getting called. What am I missing? I'm looking to log a few things and then pass the control back to the original method (similar to the around advice).

SaiprasadKrishnamurthy avatar Apr 28 '24 00:04 SaiprasadKrishnamurthy

By .load(ClassLoader.getSystemClassLoader(), ClassLoadingStrategy.Default.CHILD_FIRST) you create a new class in a child loader. Once your regular program loads the class, it won't pick it up from that loader but load the original class. You would need to use an injection strategy, or use a MethodHandle.Lookup to inject the class in the loader. To do so, you need to avoid loading the class:

    DynamicType.Unloaded<BodyNode> unloaded = new ByteBuddy()
            .redefine("your.packagelocation.BodyNode", ClassFileLocator.ForClassLoader.ofSystemLoader())
            .visit(Advice.to(MyAdvices.class).on(ElementMatchers.isMethod()))
            .make();

    Class<? extends BodyNode> loaded = unloaded
            .load(ClassLoader.getSystemClassLoader(), ClassLoadingStrategy.Default.INJECT) // or lookup when using Java 9+
            .getLoaded();

raphw avatar Apr 28 '24 07:04 raphw