266 lines
8.7 KiB
Java
266 lines
8.7 KiB
Java
/*
|
|
* 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;
|
|
}
|