aoqi@0: /*
aoqi@0:  * Copyright (c) 2010, Oracle and/or its affiliates. All rights reserved.
aoqi@0:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
aoqi@0:  *
aoqi@0:  * This code is free software; you can redistribute it and/or modify it
aoqi@0:  * under the terms of the GNU General Public License version 2 only, as
aoqi@0:  * published by the Free Software Foundation.
aoqi@0:  *
aoqi@0:  * This code is distributed in the hope that it will be useful, but WITHOUT
aoqi@0:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
aoqi@0:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
aoqi@0:  * version 2 for more details (a copy is included in the LICENSE file that
aoqi@0:  * accompanied this code).
aoqi@0:  *
aoqi@0:  * You should have received a copy of the GNU General Public License version
aoqi@0:  * 2 along with this work; if not, write to the Free Software Foundation,
aoqi@0:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
aoqi@0:  *
aoqi@0:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
aoqi@0:  * or visit www.oracle.com if you need additional information or have any
aoqi@0:  * questions.
aoqi@0:  */
aoqi@0: 
aoqi@0: /*
aoqi@0:  * @test
aoqi@0:  * @bug 4942232
aoqi@0:  * @summary missing param class processes without error
aoqi@0:  * @build ParamClassTest Test
aoqi@0:  * @run main Test
aoqi@0:  */
aoqi@0: 
aoqi@0: import java.io.*;
aoqi@0: import java.util.*;
aoqi@0: 
aoqi@0: public class Test {
aoqi@0:     public static void main(String... args) throws Exception {
aoqi@0:         new Test().run();
aoqi@0:     }
aoqi@0: 
aoqi@0:     void run() throws Exception {
aoqi@0:         File testSrc = new File(System.getProperty("test.src"));
aoqi@0:         File testClasses = new File(System.getProperty("test.classes"));
aoqi@0: 
aoqi@0:         // standard use of javah on valid class file
aoqi@0:         String[] test1Args = {
aoqi@0:             "-d", mkdir("test1/out").getPath(),
aoqi@0:             "-classpath", testClasses.getPath(),
aoqi@0:             "ParamClassTest"
aoqi@0:         };
aoqi@0:         test(test1Args, 0);
aoqi@0: 
aoqi@0:         // extended use of javah on valid source file
aoqi@0:         String[] test2Args = {
aoqi@0:             "-d", mkdir("test2/out").getPath(),
aoqi@0:             "-classpath", testSrc.getPath(),
aoqi@0:             "ParamClassTest"
aoqi@0:         };
aoqi@0:         test(test2Args, 0);
aoqi@0: 
aoqi@0:         // javah on class file with missing referents
aoqi@0:         File test3Classes = mkdir("test3/classes");
aoqi@0:         copy(new File(testClasses, "ParamClassTest.class"), test3Classes);
aoqi@0:         String[] test3Args = {
aoqi@0:             "-d", mkdir("test3/out").getPath(),
aoqi@0:             "-classpath", test3Classes.getPath(),
aoqi@0:             "ParamClassTest"
aoqi@0:         };
aoqi@0:         test(test3Args, 1);
aoqi@0: 
aoqi@0:         // javah on source file with missing referents
aoqi@0:         File test4Src = mkdir("test4/src");
aoqi@0:         String paramClassTestSrc = readFile(new File(testSrc, "ParamClassTest.java"));
aoqi@0:         writeFile(new File(test4Src, "ParamClassTest.java"),
aoqi@0:                 paramClassTestSrc.replaceAll("class Param \\{\\s+\\}", ""));
aoqi@0:         String[] test4Args = {
aoqi@0:             "-d", mkdir("test4/out").getPath(),
aoqi@0:             "-classpath", test4Src.getPath(),
aoqi@0:             "ParamClassTest"
aoqi@0:         };
aoqi@0:         test(test4Args, 15);
aoqi@0: 
aoqi@0:         if (errors > 0)
aoqi@0:             throw new Exception(errors + " errors occurred");
aoqi@0:     }
aoqi@0: 
aoqi@0:     void test(String[] args, int expect) {
aoqi@0:         System.err.println("test: " + Arrays.asList(args));
aoqi@0:         int rc = javah(args);
aoqi@0:         if (rc != expect)
aoqi@0:             error("Unexpected return code: " + rc + "; expected: " + expect);
aoqi@0:     }
aoqi@0: 
aoqi@0:     int javah(String... args) {
aoqi@0:         StringWriter sw = new StringWriter();
aoqi@0:         PrintWriter pw = new PrintWriter(sw);
aoqi@0:         int rc = com.sun.tools.javah.Main.run(args, pw);
aoqi@0:         pw.close();
aoqi@0:         String out = sw.toString();
aoqi@0:         if (!out.isEmpty())
aoqi@0:             System.err.println(out);
aoqi@0:         return rc;
aoqi@0:     }
aoqi@0: 
aoqi@0:     File mkdir(String path) {
aoqi@0:         File f = new File(path);
aoqi@0:         f.mkdirs();
aoqi@0:         return f;
aoqi@0:     }
aoqi@0: 
aoqi@0:     void copy(File from, File to) throws IOException {
aoqi@0:         if (to.isDirectory())
aoqi@0:             to = new File(to, from.getName());
aoqi@0:         try (DataInputStream in = new DataInputStream(new FileInputStream(from));
aoqi@0:                 FileOutputStream out = new FileOutputStream(to)) {
aoqi@0:             byte[] buf = new byte[(int) from.length()];
aoqi@0:             in.readFully(buf);
aoqi@0:             out.write(buf);
aoqi@0:         }
aoqi@0:     }
aoqi@0: 
aoqi@0:     String readFile(File f) throws IOException {
aoqi@0:         try (DataInputStream in = new DataInputStream(new FileInputStream(f))) {
aoqi@0:             byte[] buf = new byte[(int) f.length()];
aoqi@0:             in.readFully(buf);
aoqi@0:             return new String(buf);
aoqi@0:         }
aoqi@0:     }
aoqi@0: 
aoqi@0:     void writeFile(File f, String body) throws IOException {
aoqi@0:         try (FileWriter out = new FileWriter(f)) {
aoqi@0:             out.write(body);
aoqi@0:         }
aoqi@0:     }
aoqi@0: 
aoqi@0:     void error(String msg) {
aoqi@0:         System.err.println(msg);
aoqi@0:         errors++;
aoqi@0:     }
aoqi@0: 
aoqi@0:     int errors;
aoqi@0: }