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