WinCommand.java revision 1952:1755493c5774 
1/*
2 * Copyright 2004 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
20 * CA 95054 USA or visit www.sun.com if you need additional information or
21 * have any questions.
22 */
23
24/* @test
25 * @bug 5006520
26 * @summary Check many different ways to run Windows programs
27 * @author Martin Buchholz
28 */
29
30import java.io.*;
31import java.util.*;
32import static java.lang.System.*;
33
34class 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
48class 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
70public 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}
184