aboutsummaryrefslogtreecommitdiff
path: root/src/test/java/org/apache/bcel/util/BCELifierTestCase.java
blob: b42b94e9f59fd619ff14d16d2d7fd4e74cafd01c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package org.apache.commons.bcel6.util;

import static org.junit.Assert.*;

import java.io.BufferedInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import org.apache.commons.bcel6.classfile.JavaClass;
import org.junit.Test;


public class BCELifierTestCase {

    @Test
    public void test() throws Exception {
        OutputStream os = new ByteArrayOutputStream();
        JavaClass java_class = BCELifier.getJavaClass("Java8Example");
        assertNotNull(java_class);
        BCELifier bcelifier = new BCELifier(java_class, os);
        bcelifier.start();
    }

    /*
     * Dump a class using "javap" and compare with the same class recreated
     * using BCELifier, "javac", "java" and dumped with "javap"
     * TODO: detect if JDK present and skip test if not 
     */
    @Test
    @org.junit.Ignore // does not work properly on some systems. Also the output is rather different 
    public void testJavapCompare() throws Exception {
        testClassOnPath("target/test-classes/Java8Example.class");
    }

    private void testClassOnPath(final String javaClass) throws Exception {
        // Get javap of the input class
        final String initial = exec(null, "javap", "-p", "-c", javaClass);

        final File workDir = new File("target");
        File infile = new File(javaClass);
        JavaClass java_class = BCELifier.getJavaClass(infile.getName().replace(".class", ""));
        assertNotNull(java_class);
        File outfile = new File(workDir,infile.getName().replace(".class", "Creator.java"));
        FileOutputStream fos = new FileOutputStream(outfile);
        BCELifier bcelifier = new BCELifier(java_class, fos);
        bcelifier.start();
        fos.close();
        exec(workDir, "javac", "-cp", "classes", outfile.getName());
        exec(workDir, "java", "-cp", ".:classes", outfile.getName().replace(".java", ""));
        final String output = exec(workDir, "javap", "-p", "-c", infile.getName());
        assertEquals(initial, output);
    }

    private String exec(final File workDir, final String ... args) throws Exception {
//        System.err.println(java.util.Arrays.toString(args));
        ProcessBuilder pb = new ProcessBuilder( args );
        pb.directory(workDir);
        Process proc = pb.start();
        BufferedInputStream is = new BufferedInputStream(proc.getInputStream());
        InputStream es = proc.getErrorStream();
        proc.waitFor();
        byte []buff=new byte[2048];
        int len;
        while((len=es.read(buff)) != -1) {
            System.err.print(new String(buff,0,len));
        }
        
        StringBuilder sb = new StringBuilder();
        while((len=is.read(buff)) != -1) {
            sb.append(new String(buff,0,len));
        }
        is.close();
        return sb.toString();
    }

}