/*
 * Copyright (c) 2010, 2015, 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 6893943 6937318
 * @summary exit code from javah with no args is 0
 * @modules jdk.compiler
 */

import java.io.*;
import java.util.*;

public class T6893943 {
    static final String[] NO_ARGS = { };
    static final String[] HELP = { "-help" };
    static final String NEWLINE = System.getProperty("line.separator");

    public static void main(String... args) throws Exception {
        new T6893943().run();
    }

    void run() throws Exception {
        testSimpleAPI(NO_ARGS, 1);
        testSimpleAPI(HELP, 0);
        testCommand(NO_ARGS, 1);
        testCommand(HELP, 0);
    }

    void testSimpleAPI(String[] args, int expect_rc) throws Exception {
        System.err.println("Test simple api: " + Arrays.asList(args));
        StringWriter sw = new StringWriter();
        PrintWriter pw = new PrintWriter(sw);
        int rc = com.sun.tools.javah.Main.run(args, pw);
        pw.close();
        expect("testSimpleAPI", sw.toString(), rc, expect_rc);
    }

    void testCommand(String[] args, int expect_rc) throws Exception {
        System.err.println("Test command: " + Arrays.asList(args));
        File javaHome = new File(System.getProperty("java.home"));
        List<String> command = new ArrayList<String>();
        command.add(new File(new File(javaHome, "bin"), "javah").getPath());
        command.addAll(Arrays.asList(args));
        //System.err.println("command: " + command);

        ProcessBuilder pb = new ProcessBuilder(command);
        pb.redirectErrorStream(true);
        Process p = pb.start();
        p.getOutputStream().close();
        StringWriter sw = new StringWriter();
        String line;
        BufferedReader in = new BufferedReader(new InputStreamReader(p.getInputStream()));
        while ((line = in.readLine()) != null)
            sw.write(line + NEWLINE);
        int rc = p.waitFor();
        expect("testCommand", sw.toString(), rc, expect_rc);
    }

    void expect(String name, String out, int actual_rc, int expect_rc) throws Exception {
        if (out.isEmpty())
            throw new Exception("No output from javah");

        if (!out.startsWith("Usage:")) {
            System.err.println(out);
            throw new Exception("Unexpected output from javah");
        }

        if (actual_rc != expect_rc)
            throw new Exception(name + ": unexpected exit: " + actual_rc + ", expected: " + expect_rc);
    }
}