通过反射+动态代理向 Java 程序中注入代码

小鱼儿 2022-12-07 12:23 156阅读 0赞

## 简介 ##

最近在学习 Java 中类加载器的时候，突发奇想。本篇文章主要记录，反射的使用方法，以及使用动态代理往 Java 程序中注入代码。

## 编写 Java 程序 ##

1、新建一个包`com.qianyu.controller`，在该包下新建一个类`Executor`，用于模拟程序的主要逻辑。

package com.qianyu.controller;
    
    import java.util.Random;
    
    public class Executor { 
        public String run() { 
            try { 
                Thread.sleep(1000);
            } catch (InterruptedException e) { 
                e.printStackTrace();
            }
            return new Random().nextLong() + "";
        }
    }

上面的程序很简单，暂停一秒之后，返回一个随机字符串

2、新建`com.qianyu.App`类，增加一个静态属性`executor`，编写主函数循环调用 run 方法

package com.qianyu;
    
    import com.qianyu.controller.Executor;
    
    public class App { 
        private static Executor executor = new Executor();
        public static void main(String[] args) { 
            while(true){ 
                String res = executor.run();
                System.out.println("res = " + res);
            }
        }
    }

3、先编译代码，然后打包为 app.jar

javac com/qianyu/App.java -d .
    jar -cvf app.jar -C . .

4、然后我们就可以运行 app.jar 程序

java -cp app.jar com.qianyu.App

运行结果：

![运行结果][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xpMzQ1NTI3NzkyNQ_size_16_color_FFFFFF_t_70_pic_center]

## 通过反射注入代码 ##

1、新建 maven 项目，引入 cglib 依赖

<dependency>
        <groupId>cglib</groupId>
        <artifactId>cglib</artifactId>
        <version>3.3.0</version>
    </dependency>

2、首先，使用类加载器，加载 app.jar 文件

String path = "file:E:\\User\\Desktop\\代码注入测试\\app.jar";
    URLClassLoader classLoader = new URLClassLoader(new URL[]{ new URL(path)});

3、获取 Executor 类，使用动态代理创建代理对象

Class<?> executorClazz = classLoader.loadClass("com.qianyu.controller.Executor");
    Enhancer enhancer = new Enhancer();
    enhancer.setSuperclass(executorClazz);
    enhancer.setCallback((MethodInterceptor) (object, method, args1, proxy) -> { 
        Object result = null;
        try { 
            result = proxy.invokeSuper(object, args1);
        } catch (Throwable throwable) { 
            throwable.printStackTrace();
        }
        System.out.println("获取函数的返回值 : " + result);
        System.out.println("自己的业务");
        return result;
    });
    Object newExecutor = enhancer.create();

4、然后将 App 类里面的静态属性换成代理对象

Class<?> appClazz = classLoader.loadClass("com.qianyu.App");
    Field executor = appClazz.getDeclaredField("executor");
    executor.setAccessible(true);
    executor.set(null, newExecutor);

5、通过反射执行 main 方法

Method method = appClazz.getDeclaredMethod("main", String[].class);
    method.invoke(null, (Object) new String[1]);

运行结果：

![运行结果][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xpMzQ1NTI3NzkyNQ_size_16_color_FFFFFF_t_70_pic_center 1]

## 全部代码 ##

package pers.qianyu;
    
    import net.sf.cglib.proxy.*;
    
    import java.lang.reflect.*;
    import java.net.*;
    
    public class App { 
        public static void main(String[] args) throws Exception { 
            String path = "file:E:\\User\\Desktop\\代码注入测试\\app.jar";
            URLClassLoader classLoader = new URLClassLoader(new URL[]{ new URL(path)});
    
            Class<?> executorClazz = classLoader.loadClass("com.qianyu.controller.Executor");
            Enhancer enhancer = new Enhancer();
            enhancer.setSuperclass(executorClazz);
            enhancer.setCallback((MethodInterceptor) (object, method, args1, proxy) -> { 
                Object result = null;
                try { 
                    result = proxy.invokeSuper(object, args1);
                } catch (Throwable throwable) { 
                    throwable.printStackTrace();
                }
                System.out.println("获取函数的返回值 : " + result);
                System.out.println("自己的业务");
                return result;
            });
            Object newExecutor = enhancer.create();
    
            Class<?> appClazz = classLoader.loadClass("com.qianyu.App");
            Field executor = appClazz.getDeclaredField("executor");
            executor.setAccessible(true);
            executor.set(null, newExecutor);
    
            Method method = appClazz.getDeclaredMethod("main", String[].class);
            method.invoke(null, (Object) new String[1]);
        }
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xpMzQ1NTI3NzkyNQ_size_16_color_FFFFFF_t_70_pic_center]: /images/20221123/cc734b02da744f6ea288d0d6a3f5c42f.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xpMzQ1NTI3NzkyNQ_size_16_color_FFFFFF_t_70_pic_center 1]: /images/20221123/b91b5e51ef654012a8c83d309e237b72.png