java--简单实现代码提交校验

简单实现代码提交校验

现在很多面试页面中或者ACM中会有代码提交校验你的结果正确性，因此我也打算自己简单实现一下这样的效果。

我简单的讲下我的思路，执行外部传过来的代表一个Java代码，然后通过  进行编译成class文件字节数组，通过JavaClassExecuter处理字节数组，并且把System.out替换成自定义的输出类。

• 简单的类介绍
• 替换后比较

---

简单的类介绍

先罗列出我的一些类 1、  自定义类加载器2、  自定义jdk编译器3、  自定义System替换java代码中的system.out4、  ByteUtils int<->byte string<->byte之间的转换5、  一个modifier（修改常量池中CONSTANT_Utf8_info常量的内容），从字节上做替换常量池指向的类6、  最后一个类做协调处理

首先看下Test类

public static void main(String[] args) throws Throwable {        StringBuilder sb = new StringBuilder();        sb.append("package Dongshuai;").append("\n");        sb.append("public class Test1{").append("\n");        sb.append("    public static void main(String[]args){").append("\n");        sb.append("System.out.println(\"Hello Wolrd\");").append("\n");        sb.append("    }").append("\n");        sb.append("}").append("\n");        //初始化编译器        UpicJdkCompiler jdkCompiler = new UpicJdkCompiler();        //自定义包名类名去加载，并返回加载好后的class字节数组        byte[] clazzByte = jdkCompiler.doCompile("Dongshuai.Test1", sb.toString());        //把字节数组交给处理类替换system.out        String execute = JavaClassExecuter.execute(clazzByte);        System.out.println(execute.equals("Hello Wolrd")?Boolean.TRUE:Boolean.FALSE);   }

这个类主要是拼装一个java类的伪代码，然后交给写好的编译器编译。编译好后拿到字节数组，交给JavaClassExcuter作处理。最后拿到运行main方法所输出的值做对比，这样就类似与ACM中校验答案正确性（我这个只是非常简单的思路，也可以做成web应用，我这边就简便了下）。

2、看下JavaClassExecuter类

/**     * 执行外部传过来的代表一个Java类的byte数组     * 将输入类的byte数组中代表java.lang.System的CONSTANT_Utf8_info常量修改为劫持后的HackSystem类     * 执行方法为该类的static main(String[]args)方法,输出结果为该类向System.out/err输出的信息     *      * @param classByte代表一个Java类的byte数组     * @return执行结果     */    public static String execute(byte[]classByte) {//      byte[]classByte=objectToBytes(clazzLoad);        HackSystem.clearBuffer();        ClassModifier cm = new ClassModifier(classByte);        byte[] modiBytes = cm.modifyUTF8Constant("java/lang/System", "com/upic/System/HackSystem");        UpicClassLoader loader = new UpicClassLoader();        Class<?> clazz = loader.loadByte(modiBytes);        try {            //为了查看替换后的class文件//          FileOutputStream f=new FileOutputStream("F:\\java\\Java tools\\java_基础\\newTest.class");//          f.write(modiBytes);//          f.close();            //执行main方法            Method method = clazz.getMethod("main", new Class[] { String[].class });            method.invoke(null, new String[] { null });        } catch (Throwable e) {            e.printStackTrace(HackSystem.out);        }        return HackSystem.getBufferString();    }

这个类就不详细解释了，上面注释应该写的比较清楚了。

3、我们看下HackSystem

public final static InputStream in = System.in;    private static ByteArrayOutputStream buffer = new ByteArrayOutputStream();    public final static PrintStream out = new PrintStream(buffer);    public final static PrintStream err = out;    public static String getBufferString() {        return buffer.toString();    }    public static void clearBuffer() {        buffer.reset();    }//.............和system类似

这里展示了部分代码，主要是把原来的System替换掉，把结果保存在ByteArrayOutputStream里边方便拿取。

4、UpicJdkCompiler
构造器 初始化

public UpicJdkCompiler() {    compiler =ToolProvider.getSystemJavaCompiler();    options = new ArrayList<String>()；    final ClassLoader loader =     Thread.currentThread().getContextClassLoader();    classLoader = new InnerUpicClassLoader(loader);    StandardJavaFileManager=manager=compiler.getStandardFileManager(diagnosticCollector, null, null);javaFileManager=newJavaFileManagerImpl(manager,classLoader);    }

主要做初始化所需要用到的工具类

方法doCompile(….)

public byte[] doCompile(String name, String sourceCode) throws Throwable {        int i = name.lastIndexOf('.');        // 获得包名        String packageName = i < 0 ? "" : name.substring(0, i);        // 获得类名        String className = i < 0 ? name : name.substring(i + 1);        JavaFileObjectImpl javaFileObject = new JavaFileObjectImpl(className, sourceCode);        javaFileManager.putFileForInput(StandardLocation.SOURCE_PATH, packageName, className + EXT_JAVA,                javaFileObject);                Boolean result = compiler                .getTask(null, javaFileManager, diagnosticCollector, options, null, Arrays.asList(javaFileObject))                .call();        if (result == null || !result) {            throw new IllegalStateException(                    "Compilation failed. class: " + name + ", diagnostics: " + diagnosticCollector);        }        return classes.get(name)!=null?((JavaFileObjectImpl)classes.get(name)).getByteCode():null;    }/**……**/

这个类主要把java伪代码编译成class文件并放入流中，并且返回字节数组,关于编译可以去参考javaAPI文档。

观察输出结果：true。

替换后比较

我们现在就来做个测试观察下他的字节码的替换

这是原来的类

public class Test1 {    public static void main(String[] args) {        System.out.println("Hello World");    }}

我们查看他的常量池结构

 #1 = Methodref          #6.#15         // java/lang/Object."<init>":()V   #2 = Fieldref           #16.#17        // java/lang/System.out:Ljava/io/PrintStream;

这时候的常量池#2是指向java/lang/System.out:Ljava/io/PrintStream

然后我们再把下面代码去掉注释

//          FileOutputStream f=new FileOutputStream("F:\\java\\Java tools\\java_基础\\newTest.class");//          f.write(modiBytes);//          f.close();

再次运行程序
找到目录下的class文件，并且查看他的结构

 #1 = Methodref          #6.#15         // java/lang/Object."<init>":()V   #2 = Fieldref           #16.#17        // com/upic/System/HackSystem.out:Ljava/io/PrintStream;

就会发现，原来的System已经被替换，所以能获取用户输出的结果。

个人对使用java原生态的编译器的意见：
tool.jar包中的大多数类和接口都定义了编译器（和一般工具）的 API，但最好不要在应用程序中使用接口 DiagnosticListener、JavaFileManager、FileObject 和 JavaFileObject。应该实现这些接口，用于为编译器提供自定义服务，从而定义编译器的 SPI。

这是我个人的看法和一个简单的实现，如果大家有什么好的想法多多留言交流哈