1 /*

     2  * Copyright (c) 2007, 2016, Oracle and/or its affiliates. All rights reserved.

     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

     4  *

     5  * This code is free software; you can redistribute it and/or modify it

     6  * under the terms of the GNU General Public License version 2 only, as

     7  * published by the Free Software Foundation.

     8  *

     9  * This code is distributed in the hope that it will be useful, but WITHOUT

    10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    12  * version 2 for more details (a copy is included in the LICENSE file that

    13  * accompanied this code).

    14  *

    15  * You should have received a copy of the GNU General Public License version

    16  * 2 along with this work; if not, write to the Free Software Foundation,

    17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    18  *

    19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

    20  * or visit www.oracle.com if you need additional information or have any

    21  * questions.

    22  */

    23 

    24 /*

    25  * @test

    26  * @bug 7150368 8003412 8000407 8031545

    27  * @summary javac should include basic ability to generate native headers

    28  * @modules jdk.compiler/com.sun.tools.javah

    29  */

    30 

    31 import java.io.File;

    32 import java.io.IOException;

    33 import java.io.PrintWriter;

    34 import java.nio.file.Files;

    35 import java.util.ArrayList;

    36 import java.util.Arrays;

    37 import java.util.HashSet;

    38 import java.util.List;

    39 import java.util.Set;

    40 

    41 public class CompareTest {

    42     public static void main(String... args) throws Exception {

    43         new CompareTest().run();

    44     }

    45 

    46     void run() throws Exception {

    47         File srcDir = new File(System.getProperty("test.src"));

    48         File classesDir = new File("classes");

    49         classesDir.mkdirs();

    50         File javacHeaders = new File("headers.javac");

    51         javacHeaders.mkdirs();

    52         File javahHeaders = new File("headers.javah");

    53         javahHeaders.mkdirs();

    54 

    55         List<String> javacArgs = new ArrayList<String>();

    56         javacArgs.add("-d");

    57         javacArgs.add(classesDir.getPath());

    58         javacArgs.add("-h");

    59         javacArgs.add(javacHeaders.getPath());

    60         javacArgs.add("-XDjavah:full");

    61 

    62         for (File f: srcDir.listFiles()) {

    63             if (f.getName().matches("TestClass[0-9]+\\.java")) {

    64                 sourceFileCount++;

    65                 javacArgs.add(f.getPath());

    66             }

    67         }

    68 

    69         int rc = com.sun.tools.javac.Main.compile(javacArgs.toArray(new String[javacArgs.size()]));

    70         if (rc != 0)

    71             throw new Exception("javac failed; rc=" + rc);

    72 

    73         List<String> javahArgs = new ArrayList<String>();

    74         javahArgs.add("-d");

    75         javahArgs.add(javahHeaders.getPath());

    76 

    77         for (File f: classesDir.listFiles()) {

    78             if (f.getName().endsWith(".class")) {

    79                 javahArgs.add(inferBinaryName(f));

    80             }

    81         }

    82 

    83         PrintWriter pw = new PrintWriter(System.out, true);

    84         rc = com.sun.tools.javah.Main.run(javahArgs.toArray(new String[javahArgs.size()]), pw);

    85         if (rc != 0)

    86             throw new Exception("javah failed; rc=" + rc);

    87 

    88         compare(javahHeaders, javacHeaders);

    89 

    90         int javahHeaderCount = javahHeaders.list().length;

    91         int javacHeaderCount = javacHeaders.list().length;

    92 

    93         System.out.println(sourceFileCount + " .java files found");

    94         System.out.println(javacHeaderCount + " .h files generated by javac");

    95         System.out.println(javahHeaderCount + " .h files generated by javah");

    96         System.out.println(compareCount + " header files compared");

    97 

    98         if (javacHeaderCount != javahHeaderCount || javacHeaderCount != compareCount)

    99             error("inconsistent counts");

   100 

   101         if (errors > 0)

   102             throw new Exception(errors + " errors occurred");

   103     }

   104 

   105     String inferBinaryName(File file) {

   106         String name = file.getName();

   107         return name.substring(0, name.length() - ".class".length()).replace("$", ".");

   108     }

   109 

   110     /** Compare two directories.

   111      *  @param f1 The golden directory

   112      *  @param f2 The directory to be compared

   113      */

   114     void compare(File f1, File f2) {

   115         compare(f1, f2, null);

   116     }

   117 

   118     /** Compare two files or directories

   119      *  @param f1 The golden directory

   120      *  @param f2 The directory to be compared

   121      *  @param p An optional path identifying a file within the two directories

   122      */

   123     void compare(File f1, File f2, String p) {

   124         File f1p = (p == null ? f1 : new File(f1, p));

   125         File f2p = (p == null ? f2 : new File(f2, p));

   126         if (f1p.isDirectory() && f2p.isDirectory()) {

   127             Set<String> children = new HashSet<String>();

   128             children.addAll(Arrays.asList(f1p.list()));

   129             children.addAll(Arrays.asList(f2p.list()));

   130             for (String c: children) {

   131                 compare(f1, f2, new File(p, c).getPath()); // null-safe for p

   132             }

   133         }

   134         else if (f1p.isFile() && f2p.isFile()) {

   135             System.out.println("checking " + p);

   136             compareCount++;

   137             String s1 = read(f1p);

   138             String s2 = read(f2p);

   139             if (!s1.equals(s2)) {

   140                 System.out.println("File: " + f1p + "\n" + s1);

   141                 System.out.println("File: " + f2p + "\n" + s2);

   142                 error("Files differ: " + f1p + " " + f2p);

   143             }

   144         }

   145         else if (f1p.exists() && !f2p.exists())

   146             error("Only in " + f1 + ": " + p);

   147         else if (f2p.exists() && !f1p.exists())

   148             error("Only in " + f2 + ": " + p);

   149         else

   150             error("Files differ: " + f1p + " " + f2p);

   151     }

   152 

   153     private String read(File f) {

   154         try {

   155             return new String(Files.readAllBytes(f.toPath()));

   156         } catch (IOException e) {

   157             error("error reading " + f + ": " + e);

   158             return "";

   159         }

   160     }

   161 

   162     private void error(String msg) {

   163         System.out.println(msg);

   164         errors++;

   165     }

   166 

   167     private int errors;

   168     private int compareCount;

   169     private int sourceFileCount;

   170 }