test/jdk/tools/jpackage/createappimage/windows/JPackageCreateAppImageWinConsoleTest.java
--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/test/jdk/tools/jpackage/createappimage/windows/JPackageCreateAppImageWinConsoleTest.java Thu Jun 13 19:34:44 2019 -0400
@@ -0,0 +1,135 @@
+/*
+ * Copyright (c) 2018, 2019, 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.IOException;
+import java.io.File;
+import java.io.InputStream;
+import java.io.FileInputStream;
+
+/*
+ * @test
+ * @summary jpackage create image win console test
+ * @library ../../helpers
+ * @library ../
+ * @build JPackageHelper
+ * @build JPackagePath
+ * @build JPackageCreateAppImageBase
+ * @requires (os.family == "windows")
+ * @modules jdk.jpackage
+ * @run main/othervm -Xmx512m JPackageCreateAppImageWinConsoleTest
+ */
+
+public class JPackageCreateAppImageWinConsoleTest {
+ private static final String NAME = "test";
+ private static final String OUTPUT = "output";
+ private static final String OUTPUT_WIN_CONSOLE = "outputWinConsole";
+ private static final int BUFFER_SIZE = 512;
+ private static final int GUI_SUBSYSTEM = 2;
+ private static final int CONSOLE_SUBSYSTEM = 3;
+
+ private static final String [] CMD = {
+ "create-app-image",
+ "--input", "input",
+ "--output", OUTPUT,
+ "--name", NAME,
+ "--main-jar", "hello.jar",
+ "--main-class", "Hello",
+ };
+
+ private static final String [] CMD_WIN_CONSOLE = {
+ "create-app-image",
+ "--input", "input",
+ "--output", OUTPUT_WIN_CONSOLE,
+ "--name", NAME,
+ "--main-jar", "hello.jar",
+ "--main-class", "Hello",
+ "--win-console"
+ };
+
+ private static void checkSubsystem(boolean console) throws Exception {
+ String file = console ? OUTPUT_WIN_CONSOLE : OUTPUT;
+ file += File.separator;
+ file += NAME;
+ file += File.separator;
+ file += NAME + ".exe";
+
+ JPackageCreateAppImageBase.validate(file);
+
+ try (InputStream inputStream = new FileInputStream(file)) {
+ byte [] bytes = new byte[BUFFER_SIZE];
+ if (inputStream.read(bytes) != BUFFER_SIZE) {
+ throw new AssertionError("Wrong number of bytes read");
+ }
+
+ // Check PE header for console or Win GUI app.
+ // https://docs.microsoft.com/en-us/windows/desktop/api/winnt/ns-winnt-_image_nt_headers
+ for (int i = 0; i < (bytes.length - 4); i++) {
+ if (bytes[i] == 0x50 && bytes[i + 1] == 0x45 &&
+ bytes[i + 2] == 0x0 && bytes[i + 3] == 0x0) {
+ i = i + 4 + 20 + 68; // Signature, File Header and subsystem offset.
+ byte subsystem = bytes[i];
+ if (console) {
+ if (subsystem != CONSOLE_SUBSYSTEM) {
+ throw new AssertionError("Unexpected subsystem: " + subsystem);
+ } else {
+ return; // done
+ }
+ } else {
+ if (subsystem != GUI_SUBSYSTEM) {
+ throw new AssertionError("Unexpected subsystem: " + subsystem);
+ } else {
+ return; // done
+ }
+ }
+ }
+ }
+ }
+
+ throw new AssertionError("Unable to verify PE header");
+ }
+
+ private static void validate() throws Exception {
+ checkSubsystem(false);
+ checkSubsystem(true);
+ }
+
+ private static void testCreateAppImage(String [] cmd) throws Exception {
+ JPackageHelper.executeCLI(true, cmd);
+ }
+
+ private static void testCreateAppImageToolProvider(String [] cmd) throws Exception {
+ JPackageHelper.executeToolProvider(true, cmd);
+ }
+
+ public static void main(String[] args) throws Exception {
+ JPackageHelper.createHelloImageJar();
+ testCreateAppImage(CMD);
+ testCreateAppImage(CMD_WIN_CONSOLE);
+ validate();
+ JPackageHelper.deleteOutputFolder(OUTPUT);
+ JPackageHelper.deleteOutputFolder(OUTPUT_WIN_CONSOLE);
+ testCreateAppImageToolProvider(CMD);
+ testCreateAppImageToolProvider(CMD_WIN_CONSOLE);
+ validate();
+ }
+}