/*
 * Copyright 2009 Sun Microsystems, Inc.  All Rights Reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
 * CA 95054 USA or visit www.sun.com if you need additional information or
 * have any questions.
 */

import java.io.DataInputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Enumeration;
import java.util.List;
import java.util.Set;
import java.util.TreeSet;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;

import com.sun.tools.classfile.AccessFlags;
import com.sun.tools.classfile.ClassFile;
import com.sun.tools.classfile.ConstantPoolException;
import com.sun.tools.classfile.Method;
import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.util.LinkedHashSet;

public class CompareTest {
    String[][] testCases = {
        { },
        { "-jni" },
//        { "-llni" },
    };

    public static void main(String... args) throws Exception {
        new CompareTest().run(args);
    }

    public void run(String... args) throws Exception {
        old_javah_cmd = new File(args[0]);
        rt_jar = new File(args[1]);

        Set<String> testClasses;
        if (args.length > 2) {
            testClasses = new LinkedHashSet<String>(Arrays.asList(Arrays.copyOfRange(args, 2, args.length)));
        } else
            testClasses = getNativeClasses(new JarFile(rt_jar));

        for (String[] options: testCases) {
            for (String name: testClasses) {
                test(Arrays.asList(options), rt_jar, name);
            }
        }

        if (errors == 0)
            System.out.println(count + " tests passed");
        else
            throw new Exception(errors + "/" + count + " tests failed");
    }

    public void test(List<String> options, File bootclasspath, String className)
            throws IOException, InterruptedException {
        System.err.println("test: " + options + " " + className);
        count++;

        testOptions = options;
        testClassName = className;

        File oldOutDir = initDir(file(new File("old"), className));
        int old_rc = old_javah(options, oldOutDir, bootclasspath, className);

        File newOutDir = initDir(file(new File("new"), className));
        int new_rc = new_javah(options, newOutDir, bootclasspath, className);

        if (old_rc != new_rc)
            error("return codes differ; old: " + old_rc + ", new: " + new_rc);

        compare(oldOutDir, newOutDir);
    }

    int old_javah(List<String> options, File outDir, File bootclasspath, String className)
            throws IOException, InterruptedException {
        List<String> cmd = new ArrayList<String>();
        cmd.add(old_javah_cmd.getPath());
        cmd.addAll(options);
        cmd.add("-d");
        cmd.add(outDir.getPath());
        cmd.add("-bootclasspath");
        cmd.add(bootclasspath.getPath());
        cmd.add(className);
        System.err.println("old_javah: " + cmd);
        ProcessBuilder pb = new ProcessBuilder(cmd);
        pb.redirectErrorStream(true);
        Process p = pb.start();
        BufferedReader in = new BufferedReader(new InputStreamReader(p.getInputStream()));
        String line;
        StringBuilder sb = new StringBuilder();
        while ((line = in.readLine()) != null) {
            sb.append(line);
            sb.append("\n");
        }
        System.err.println("old javah out: " + sb.toString());
        return p.waitFor();
    }

    int new_javah(List<String> options, File outDir, File bootclasspath, String className) {
        List<String> args = new ArrayList<String>();
        args.addAll(options);
        args.add("-d");
        args.add(outDir.getPath());
        args.add("-bootclasspath");
        args.add(bootclasspath.getPath());
        args.add(className);
        StringWriter sw = new StringWriter();
        PrintWriter pw = new PrintWriter(sw);
        int rc = com.sun.tools.javah.Main.run(args.toArray(new String[args.size()]), pw);
        pw.close();
        System.err.println("new javah out: " + sw.toString());
        return rc;
    }

    Set<String> getNativeClasses(JarFile jar) throws IOException, ConstantPoolException {
        System.err.println("getNativeClasses: " + jar.getName());
        Set<String> results = new TreeSet<String>();
        Enumeration<JarEntry> e = jar.entries();
        while (e.hasMoreElements()) {
            JarEntry je = e.nextElement();
            if (isNativeClass(jar, je)) {
                String name = je.getName();
                results.add(name.substring(0, name.length() - 6).replace("/", "."));
            }
        }
        return results;
    }

    boolean isNativeClass(JarFile jar, JarEntry entry) throws IOException, ConstantPoolException {
        String name = entry.getName();
        if (name.startsWith("META-INF") || !name.endsWith(".class"))
            return false;
        //String className = name.substring(0, name.length() - 6).replace("/", ".");
        //System.err.println("check " + className);
        InputStream in = jar.getInputStream(entry);
        ClassFile cf = ClassFile.read(in);
        for (int i = 0; i < cf.methods.length; i++) {
            Method m = cf.methods[i];
            if (m.access_flags.is(AccessFlags.ACC_NATIVE)) {
                // System.err.println(className);
                return true;
            }
        }
        return false;
    }

    void compare(File f1, File f2) throws IOException {
        if (f1.isFile() && f2.isFile())
            compareFiles(f1, f2);
        else if (f1.isDirectory() && f2.isDirectory())
            compareDirectories(f1, f2);
        else
            error("files differ: "
                + f1 + " (" + getFileType(f1) + "), "
                + f2 + " (" + getFileType(f2) + ")");
    }

    void compareDirectories(File d1, File d2) throws IOException {
        Set<String> list = new TreeSet<String>();
        list.addAll(Arrays.asList(d1.list()));
        list.addAll(Arrays.asList(d2.list()));
        for (String c: list)
            compare(new File(d1, c), new File(d2, c));
    }

    void compareFiles(File f1, File f2) throws IOException {
        byte[] c1 = readFile(f1);
        byte[] c2 = readFile(f2);
        if (!Arrays.equals(c1, c2))
            error("files differ: " + f1 + ", " + f2);
    }

    byte[] readFile(File file) throws IOException {
        int size = (int) file.length();
        byte[] data = new byte[size];
        DataInputStream in = new DataInputStream(new FileInputStream(file));
        try {
            in.readFully(data);
        } finally {
            in.close();
        }
        return data;
    }

    String getFileType(File f) {
        return f.isDirectory() ? "directory"
                : f.isFile() ? "file"
                : f.exists() ? "other"
                : "not found";
    }

    /**
     * Set up an empty directory.
     */
    public File initDir(File dir) {
        if (dir.exists())
            deleteAll(dir);
        dir.mkdirs();
        return dir;
    }

    /**
     * Delete a file or a directory (including all its contents).
     */
    boolean deleteAll(File file) {
        if (file.isDirectory()) {
            for (File f: file.listFiles())
                deleteAll(f);
        }
        return file.delete();
    }

    File file(File dir, String... path) {
        File f = dir;
        for (String p: path)
            f = new File(f, p);
        return f;
    }

    /**
     * Report an error.
     */
    void error(String msg, String... more) {
        System.err.println("test: " + testOptions + " " + testClassName);
        System.err.println("error: " + msg);
        for (String s: more)
            System.err.println(s);
        errors++;
        System.exit(1);
    }

    File old_javah_cmd;
    File rt_jar;
    List<String> testOptions;
    String testClassName;
    int count;
    int errors;
}