test/langtools/tools/javah/6572945/T6572945.java - platform/libcore - Gitiles

 /*
  * Copyright (c) 2007, 2016, 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.
  */

 /*
  * @test
  * @bug 6572945
  * @summary rewrite javah as an annotation processor, instead of as a doclet
  * @modules jdk.compiler/com.sun.tools.javah
  * @build TestClass1 TestClass2 TestClass3
  * @run main T6572945
  */

 import java.io.*;
 import java.util.*;
 import com.sun.tools.javah.Main;

 public class T6572945
 {
     static File testSrc = new File(System.getProperty("test.src", "."));
     static File testClasses = new File(System.getProperty("test.classes", "."));
     static boolean isWindows = System.getProperty("os.name").startsWith("Windows");

     public static void main(String... args)
         throws IOException, InterruptedException
     {
         boolean ok = new T6572945().run(args);
         if (!ok)
             throw new Error("Test Failed");
     }

     public boolean run(String[] args)
         throws IOException, InterruptedException
     {
         if (args.length == 1)
             jdk = new File(args[0]);

         test("-o", "jni.file.1",  "-jni", "TestClass1");
         test("-o", "jni.file.2",  "-jni", "TestClass1", "TestClass2");
         test("-d", "jni.dir.1",   "-jni", "TestClass1", "TestClass2");
         test("-o", "jni.file.3",  "-jni", "TestClass3");

         // The following tests are disabled because llni support has been
         // discontinued, and because bugs in old javah means that character
         // for character testing against output from old javah does not work.
         // In fact, the LLNI impl in new javah is actually better than the
         // impl in old javah because of a couple of significant bug fixes.

 //        test("-o", "llni.file.1", "-llni", "TestClass1");
 //        test("-o", "llni.file.2", "-llni", "TestClass1", "TestClass2");
 //        test("-d", "llni.dir.1",  "-llni", "TestClass1", "TestClass2");
 //        test("-o", "llni.file.3", "-llni", "TestClass3");

         return (errors == 0);
     }

     void test(String... args)
         throws IOException, InterruptedException
     {
         String[] cp_args = new String[args.length + 2];
         cp_args[0] = "-classpath";
         cp_args[1] = testClasses.getPath();
         System.arraycopy(args, 0, cp_args, 2, args.length);

         if (jdk != null)
             init(cp_args);

         File out = null;
         for (int i = 0; i < args.length; i++) {
             if (args[i].equals("-o")) {
                 out = new File(args[++i]);
                 break;
             } else if (args[i].equals("-d")) {
                 out = new File(args[++i]);
                 out.mkdirs();
                 break;
             }
         }

         try {
             System.out.println("test: " + Arrays.asList(cp_args));

 //            // Uncomment and use the following lines to execute javah via the
 //            // command line -- for example, to run old javah and set up the golden files
 //            List<String> cmd = new ArrayList<String>();
 //            File javaHome = new File(System.getProperty("java.home"));
 //            if (javaHome.getName().equals("jre"))
 //                javaHome = javaHome.getParentFile();
 //            File javah = new File(new File(javaHome, "bin"), "javah");
 //            cmd.add(javah.getPath());
 //            cmd.addAll(Arrays.asList(cp_args));
 //            ProcessBuilder pb = new ProcessBuilder(cmd);
 //            pb.redirectErrorStream(true);
 //            pb.start();
 //            Process p = pb.start();
 //            String line;
 //            BufferedReader in = new BufferedReader(new InputStreamReader(p.getInputStream()));
 //            while ((line = in.readLine()) != null)
 //                System.err.println(line);
 //            in.close();
 //            int rc = p.waitFor();

             // Use new javah
             PrintWriter err = new PrintWriter(System.err, true);
             int rc = Main.run(cp_args, err);

             if (rc != 0) {
                 error("javah failed: rc=" + rc);
                 return;
             }

             // The golden files use the LL suffix for long constants, which
             // is used on Linux and Solaris.   On Windows, the suffix is i64,
             // so compare will update the golden files on the fly before the
             // final comparison.
             compare(new File(new File(testSrc, "gold"), out.getName()), out);
         } catch (Throwable t) {
             t.printStackTrace();
             error("javah threw exception");
         }
     }

     void init(String[] args) throws IOException, InterruptedException {
         String[] cmdArgs = new String[args.length + 1];
         cmdArgs[0] = new File(new File(jdk, "bin"), "javah").getPath();
         System.arraycopy(args, 0, cmdArgs, 1, args.length);

         System.out.println("init: " + Arrays.asList(cmdArgs));

         ProcessBuilder pb = new ProcessBuilder(cmdArgs);
         pb.directory(new File(testSrc, "gold"));
         pb.redirectErrorStream(true);
         Process p = pb.start();
         BufferedReader in = new BufferedReader(new InputStreamReader(p.getInputStream()));
         String line;
         while ((line = in.readLine()) != null)
             System.out.println("javah: " + line);
         int rc = p.waitFor();
         if (rc != 0)
             error("javah: exit code " + rc);
     }

     /** Compare two directories.
      *  @param f1 The golden directory
      *  @param f2 The directory to be compared
      */
     void compare(File f1, File f2) {
         compare(f1, f2, null);
     }

     /** Compare two files or directories
      *  @param f1 The golden directory
      *  @param f2 The directory to be compared
      *  @param p An optional path identifying a file within the two directories
      */
     void compare(File f1, File f2, String p) {
         File f1p = (p == null ? f1 : new File(f1, p));
         File f2p = (p == null ? f2 : new File(f2, p));
         System.out.println("compare " + f1p + " " + f2p);
         if (f1p.isDirectory() && f2p.isDirectory()) {
             Set<String> children = new HashSet<String>();
             children.addAll(Arrays.asList(f1p.list()));
             children.addAll(Arrays.asList(f2p.list()));
             for (String c: children) {
                 compare(f1, f2, new File(p, c).getPath()); // null-safe for p
             }
         }
         else if (f1p.isFile() && f2p.isFile()) {
             String s1 = read(f1p);
             if (isWindows) {
                 // f1/s1 is the golden file
                 // on Windows, long constants use the i64 suffix, not LL
                 s1 = s1.replaceAll("( [0-9]+)LL\n", "$1i64\n");
             }
             String s2 = read(f2p);
             if (!s1.equals(s2)) {
                 System.out.println("File: " + f1p + "\n" + s1);
                 System.out.println("File: " + f2p + "\n" + s2);
                 error("Files differ: " + f1p + " " + f2p);
             }
         }
         else if (f1p.exists() && !f2p.exists())
             error("Only in " + f1 + ": " + p);
         else if (f2p.exists() && !f1p.exists())
             error("Only in " + f2 + ": " + p);
         else
             error("Files differ: " + f1p + " " + f2p);
     }

     private String read(File f) {
         try {
             BufferedReader in = new BufferedReader(new FileReader(f));
             try {
                 StringBuilder sb = new StringBuilder((int) f.length());
                 String line;
                 while ((line = in.readLine()) != null) {
                     sb.append(line);
                     sb.append("\n");
                 }
                 return sb.toString();
             } finally {
                 try {
                     in.close();
                 } catch (IOException e) {
                 }
             }
         } catch (IOException e) {
             error("error reading " + f + ": " + e);
             return "";
         }
     }


     private void error(String msg) {
         System.out.println(msg);
         errors++;
     }

     private int errors;
     private File jdk;
 }
	/*
	* Copyright (c) 2007, 2016, 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.
	*/

	/*
	* @test
	* @bug 6572945
	* @summary rewrite javah as an annotation processor, instead of as a doclet
	* @modules jdk.compiler/com.sun.tools.javah
	* @build TestClass1 TestClass2 TestClass3
	* @run main T6572945
	*/

	import java.io.*;
	import java.util.*;
	import com.sun.tools.javah.Main;

	public class T6572945
	{
	static File testSrc = new File(System.getProperty("test.src", "."));
	static File testClasses = new File(System.getProperty("test.classes", "."));
	static boolean isWindows = System.getProperty("os.name").startsWith("Windows");

	public static void main(String... args)
	throws IOException, InterruptedException
	{
	boolean ok = new T6572945().run(args);
	if (!ok)
	throw new Error("Test Failed");
	}

	public boolean run(String[] args)
	throws IOException, InterruptedException
	{
	if (args.length == 1)
	jdk = new File(args[0]);

	test("-o", "jni.file.1", "-jni", "TestClass1");
	test("-o", "jni.file.2", "-jni", "TestClass1", "TestClass2");
	test("-d", "jni.dir.1", "-jni", "TestClass1", "TestClass2");
	test("-o", "jni.file.3", "-jni", "TestClass3");

	// The following tests are disabled because llni support has been
	// discontinued, and because bugs in old javah means that character
	// for character testing against output from old javah does not work.
	// In fact, the LLNI impl in new javah is actually better than the
	// impl in old javah because of a couple of significant bug fixes.

	// test("-o", "llni.file.1", "-llni", "TestClass1");
	// test("-o", "llni.file.2", "-llni", "TestClass1", "TestClass2");
	// test("-d", "llni.dir.1", "-llni", "TestClass1", "TestClass2");
	// test("-o", "llni.file.3", "-llni", "TestClass3");

	return (errors == 0);
	}

	void test(String... args)
	throws IOException, InterruptedException
	{
	String[] cp_args = new String[args.length + 2];
	cp_args[0] = "-classpath";
	cp_args[1] = testClasses.getPath();
	System.arraycopy(args, 0, cp_args, 2, args.length);

	if (jdk != null)
	init(cp_args);

	File out = null;
	for (int i = 0; i < args.length; i++) {
	if (args[i].equals("-o")) {
	out = new File(args[++i]);
	break;
	} else if (args[i].equals("-d")) {
	out = new File(args[++i]);
	out.mkdirs();
	break;
	}
	}

	try {
	System.out.println("test: " + Arrays.asList(cp_args));

	// // Uncomment and use the following lines to execute javah via the
	// // command line -- for example, to run old javah and set up the golden files
	// List<String> cmd = new ArrayList<String>();
	// File javaHome = new File(System.getProperty("java.home"));
	// if (javaHome.getName().equals("jre"))
	// javaHome = javaHome.getParentFile();
	// File javah = new File(new File(javaHome, "bin"), "javah");
	// cmd.add(javah.getPath());
	// cmd.addAll(Arrays.asList(cp_args));
	// ProcessBuilder pb = new ProcessBuilder(cmd);
	// pb.redirectErrorStream(true);
	// pb.start();
	// Process p = pb.start();
	// String line;
	// BufferedReader in = new BufferedReader(new InputStreamReader(p.getInputStream()));
	// while ((line = in.readLine()) != null)
	// System.err.println(line);
	// in.close();
	// int rc = p.waitFor();

	// Use new javah
	PrintWriter err = new PrintWriter(System.err, true);
	int rc = Main.run(cp_args, err);

	if (rc != 0) {
	error("javah failed: rc=" + rc);
	return;
	}

	// The golden files use the LL suffix for long constants, which
	// is used on Linux and Solaris. On Windows, the suffix is i64,
	// so compare will update the golden files on the fly before the
	// final comparison.
	compare(new File(new File(testSrc, "gold"), out.getName()), out);
	} catch (Throwable t) {
	t.printStackTrace();
	error("javah threw exception");
	}
	}

	void init(String[] args) throws IOException, InterruptedException {
	String[] cmdArgs = new String[args.length + 1];
	cmdArgs[0] = new File(new File(jdk, "bin"), "javah").getPath();
	System.arraycopy(args, 0, cmdArgs, 1, args.length);

	System.out.println("init: " + Arrays.asList(cmdArgs));

	ProcessBuilder pb = new ProcessBuilder(cmdArgs);
	pb.directory(new File(testSrc, "gold"));
	pb.redirectErrorStream(true);
	Process p = pb.start();
	BufferedReader in = new BufferedReader(new InputStreamReader(p.getInputStream()));
	String line;
	while ((line = in.readLine()) != null)
	System.out.println("javah: " + line);
	int rc = p.waitFor();
	if (rc != 0)
	error("javah: exit code " + rc);
	}

	/** Compare two directories.
	* @param f1 The golden directory
	* @param f2 The directory to be compared
	*/
	void compare(File f1, File f2) {
	compare(f1, f2, null);
	}

	/** Compare two files or directories
	* @param f1 The golden directory
	* @param f2 The directory to be compared
	* @param p An optional path identifying a file within the two directories
	*/
	void compare(File f1, File f2, String p) {
	File f1p = (p == null ? f1 : new File(f1, p));
	File f2p = (p == null ? f2 : new File(f2, p));
	System.out.println("compare " + f1p + " " + f2p);
	if (f1p.isDirectory() && f2p.isDirectory()) {
	Set<String> children = new HashSet<String>();
	children.addAll(Arrays.asList(f1p.list()));
	children.addAll(Arrays.asList(f2p.list()));
	for (String c: children) {
	compare(f1, f2, new File(p, c).getPath()); // null-safe for p
	}
	}
	else if (f1p.isFile() && f2p.isFile()) {
	String s1 = read(f1p);
	if (isWindows) {
	// f1/s1 is the golden file
	// on Windows, long constants use the i64 suffix, not LL
	s1 = s1.replaceAll("( [0-9]+)LL\n", "$1i64\n");
	}
	String s2 = read(f2p);
	if (!s1.equals(s2)) {
	System.out.println("File: " + f1p + "\n" + s1);
	System.out.println("File: " + f2p + "\n" + s2);
	error("Files differ: " + f1p + " " + f2p);
	}
	}
	else if (f1p.exists() && !f2p.exists())
	error("Only in " + f1 + ": " + p);
	else if (f2p.exists() && !f1p.exists())
	error("Only in " + f2 + ": " + p);
	else
	error("Files differ: " + f1p + " " + f2p);
	}

	private String read(File f) {
	try {
	BufferedReader in = new BufferedReader(new FileReader(f));
	try {
	StringBuilder sb = new StringBuilder((int) f.length());
	String line;
	while ((line = in.readLine()) != null) {
	sb.append(line);
	sb.append("\n");
	}
	return sb.toString();
	} finally {
	try {
	in.close();
	} catch (IOException e) {
	}
	}
	} catch (IOException e) {
	error("error reading " + f + ": " + e);
	return "";
	}
	}


	private void error(String msg) {
	System.out.println(msg);
	errors++;
	}

	private int errors;
	private File jdk;
	}