1 /*
2 * Copyright (c) 2004, 2013 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 /* @test
25 * @bug 5006520
26 * @summary Check many different ways to run Windows programs
27 * @author Martin Buchholz
28 */
29
30 import java.io.*;
31 import java.util.*;
32 import static java.lang.System.*;
33
34 class StreamDrainer extends Thread {
35 private final InputStream is;
36 private final ByteArrayOutputStream os = new ByteArrayOutputStream();
37 public StreamDrainer(InputStream is) { this.is = is; }
38 public void run() {
39 try {
40 int i;
41 while ((i = is.read()) >= 0)
42 os.write(i);
43 } catch (Exception e) {}
44 }
45 public String toString() { return os.toString(); }
46 }
47
48 class CommandRunner {
49 private static Random generator = new Random();
50 public final int exitValue;
51 public final String out;
52 public final String err;
53 CommandRunner(String... args) throws Exception {
54 Process p = (generator.nextInt(2) == 0)
55 ? new ProcessBuilder(args).start()
56 : Runtime.getRuntime().exec(args);
57 StreamDrainer d1 = new StreamDrainer(p.getInputStream());
58 StreamDrainer d2 = new StreamDrainer(p.getErrorStream());
59 d1.start();
60 d2.start();
61 p.waitFor();
62 d1.join();
63 d2.join();
64 this.exitValue = p.exitValue();
65 this.out = d1.toString();
66 this.err = d2.toString();
67 }
68 }
69
70 public class WinCommand {
71 private static int failed = 0;
72
73 private static void fail(String msg) {
74 err.printf("FAIL: %s%n", msg);
75 failed++;
76 }
77
78 private static String outputOf(String... args) {
79 try {
80 CommandRunner cr = new CommandRunner(args);
81 if (cr.exitValue != 0)
82 fail("exitValue != 0");
83 if (! cr.err.equals(""))
84 fail("stderr: " + cr.err);
85 return cr.out.replaceFirst("[\r\n]+$", "");
86 } catch (Exception e) {
87 fail(e.toString());
88 return "";
89 }
90 }
91
92 private static void checkCD(String... filespecs) {
93 String firstCD = null;
94 for (String filespec : filespecs) {
95 String CD = outputOf(filespec, "/C", "CD");
96 out.printf("%s CD ==> %s%n", filespec, CD);
97 if (firstCD == null) {
98 firstCD = CD;
99 checkDir(CD);
100 }
101 if (! CD.equals(firstCD)) {
102 fail("Inconsistent result from CD subcommand");
103 checkDir(CD);
104 }
105 }
106 }
107
108 private static void checkDir(String dirname) {
109 if (! new File(dirname).isDirectory())
110 fail(String.format("Not a directory: %s%n", dirname));
111 }
112
113 private static void writeFile(String filename, String contents) {
114 try {
115 FileOutputStream fos = new FileOutputStream(filename);
116 fos.write(contents.getBytes());
117 fos.close();
118 } catch (Exception e) {
119 fail("Unexpected exception" + e.toString());
120 }
121 }
122
123 public static void main(String[] args) throws Exception {
124 File systemRoot =
125 getenv("SystemRoot") != null ? new File(getenv("SystemRoot")) :
126 getenv("WINDIR") != null ? new File(getenv ("WINDIR")) :
127 null;
128 if (systemRoot == null || ! systemRoot.isDirectory())
129 return; // Not Windows as we know it
130
131 String systemDirW = new File(systemRoot, "System32").getPath();
132 String systemDirM = systemDirW.replace('\\', '/');
133 out.printf("systemDirW=%s%n", systemDirW);
134 out.printf("systemDirM=%s%n", systemDirM);
135
136 // Win9x systems don't have a cmd.exe
137 if (new File(systemDirW, "cmd.exe").exists()) {
138 try {
139 out.println("Running cmd.exe tests...");
140 writeFile("cdcmd.cmd", "@echo off\r\nCD\r\n");
141 writeFile("cdbat.bat", "@echo off\r\nCD\r\n");
142 checkCD("cmd",
143 "cmd.exe",
144 systemDirW + "\\cmd.exe",
145 // Only the ".exe" extension can be omitted
146 systemDirW + "\\cmd",
147 systemDirM + "/cmd.exe",
148 systemDirM + "/cmd",
149 "/" + systemDirM + "/cmd",
150 "cdcmd.cmd", "./cdcmd.cmd", ".\\cdcmd.cmd",
151 "cdbat.bat", "./cdbat.bat", ".\\cdbat.bat");
152 } finally {
153 new File("cdcmd.cmd").delete();
154 new File("cdbat.bat").delete();
155 }
156 }
157
158 // 16-bit apps like command.com must have a console;
159 // fix this someday...
160
161 // // Win64 systems don't have a command.com
162 // if (new File(systemDirW, "command.com").exists()
163 // // no output if running without a console;
164 // // fix this in Mustang
165 // && ! outputOf("command.com", "/C", "CD").equals("")) {
166 // out.println("Running command.com tests...");
167 // checkCD("command.com",
168 // systemDirM + "/command.com",
169 // systemDirW + "\\command.com");
170 // }
171
172 // Win9x systems have a %SYSTEMDRIVE%\command.com
173 // if (new File("C:\\COMMAND.COM").exists()
174 // && ! outputOf("COMMAND.COM", "/C", "CD").equals("")) {
175 // out.println("Running COMMAND.COM tests...");
176 // checkCD("C:/command.com",
177 // "C:\\command.com");
178 // }
179
180 if (failed > 0)
181 throw new Exception(failed + " tests failed");
182 }
183 }