langtools/test/tools/javah/compareTest/CompareTest.java - edge/openjdk - Git at Google

 /*
  * Copyright (c) 2009, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  * or visit www.oracle.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;
 }
	/*
	* Copyright (c) 2009, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
	* or visit www.oracle.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;
	}