X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=lib%2FSystem%2FWin32%2FProgram.inc;h=2d3580d3db35715bd9f6cd3fe8d4bb110882bed9;hb=a1b3d3d24cebeaa98fb7e2c2280b8fe4e1476761;hp=7a6b124d2d29fd3addb1386e29a28cfa9cf0121a;hpb=52a7efafba1045a8c7a3fe545487d11d53e6f93a;p=oota-llvm.git

diff --git a/lib/System/Win32/Program.inc b/lib/System/Win32/Program.inc
index 7a6b124d2d2..2d3580d3db3 100644
--- a/lib/System/Win32/Program.inc
+++ b/lib/System/Win32/Program.inc
@@ -1,8 +1,8 @@
-//===- Win32/Program.cpp - Win32 Program Implelementation ----- -*- C++ -*-===//
+//===- Win32/Program.cpp - Win32 Program Implementation ------- -*- C++ -*-===//
 // 
 //                     The LLVM Compiler Infrastructure
 //
-// This file was developed by Reid Spencer and is distributed under the 
+// This file was developed by Jeff Cohen and is distributed under the 
 // University of Illinois Open Source License. See LICENSE.TXT for details.
 // 
 //===----------------------------------------------------------------------===//
@@ -11,9 +11,159 @@
 //
 //===----------------------------------------------------------------------===//
 
+#include "Win32.h"
+#include <malloc.h>
+
 //===----------------------------------------------------------------------===//
-//=== WARNING: Implementation here must contain only generic Win32 code that
-//===          is guaranteed to work on *all* Win32 variants.
+//=== WARNING: Implementation here must contain only Win32 specific code 
+//===          and must not be UNIX code
 //===----------------------------------------------------------------------===//
 
+namespace llvm {
+using namespace sys;
+
+// This function just uses the PATH environment variable to find the program.
+Path
+Program::FindProgramByName(const std::string& progName) {
+
+  // Check some degenerate cases
+  if (progName.length() == 0) // no program
+    return Path();
+  Path temp;
+  if (!temp.setFile(progName)) // invalid name
+    return Path();
+  if (temp.executable()) // already executable as is
+    return temp;
+
+  // At this point, the file name is valid and its not executable.
+  // Let Windows search for it.
+  char buffer[MAX_PATH];
+  char *dummy = NULL;
+  DWORD len = SearchPath(NULL, progName.c_str(), ".exe", MAX_PATH,
+                         buffer, &dummy);
+
+  // See if it wasn't found.
+  if (len == 0)
+    return Path();
+
+  // See if we got the entire path.
+  if (len < MAX_PATH)
+    return Path(buffer);
+
+  // Buffer was too small; grow and retry.
+  while (true) {
+    char *b = reinterpret_cast<char *>(_alloca(len+1));
+    DWORD len2 = SearchPath(NULL, progName.c_str(), ".exe", len+1, b, &dummy);
+
+    // It is unlikely the search failed, but it's always possible some file
+    // was added or removed since the last search, so be paranoid...
+    if (len2 == 0)
+      return Path();
+    else if (len2 <= len)
+      return Path(b);
+
+    len = len2;
+  }
+}
+
+//
+int 
+Program::ExecuteAndWait(const Path& path, 
+                        const char** args,
+                        const char** envp,
+                        const Path** redirects,
+                        unsigned secondsToWait) {
+  if (!path.executable())
+    throw path.toString() + " is not executable"; 
+
+  // Windows wants a command line, not an array of args, to pass to the new
+  // process.  We have to concatenate them all, while quoting the args that
+  // have embedded spaces.
+
+  // First, determine the length of the command line.
+  std::string progname(path.getLast());
+  unsigned len = progname.length() + 1;
+  if (progname.find(' ') != std::string::npos)
+    len += 2;
+
+  for (unsigned i = 0; args[i]; i++) {
+    len += strlen(args[i]) + 1;
+    if (strchr(args[i], ' '))
+      len += 2;
+  }
+
+  // Now build the command line.
+  char *command = reinterpret_cast<char *>(_alloca(len));
+  char *p = command;
+
+  bool needsQuoting = progname.find(' ') != std::string::npos;
+  if (needsQuoting)
+    *p++ = '"';
+  memcpy(p, progname.c_str(), progname.length());
+  p += progname.length();
+  if (needsQuoting)
+    *p++ = '"';
+  *p++ = ' ';
+
+  for (unsigned i = 0; args[i]; i++) {
+    const char *arg = args[i];
+	size_t len = strlen(arg);
+    needsQuoting = strchr(arg, ' ') != 0;
+    if (needsQuoting)
+      *p++ = '"';
+    memcpy(p, arg, len);
+    p += len;
+    if (needsQuoting)
+      *p++ = '"';
+    *p++ = ' ';
+  }
+
+  *p = 0;
+
+  // Create a child process.
+  STARTUPINFO si;
+  memset(&si, 0, sizeof(si));
+  si.cb = sizeof(si);
+
+  // TODO: do replacement of standard input/output/error handles.
+
+  PROCESS_INFORMATION pi;
+  memset(&pi, 0, sizeof(pi));
+
+  if (!CreateProcess(path.c_str(), command, NULL, NULL, FALSE, 0,
+                     envp, NULL, &si, &pi))
+  {
+    ThrowError(std::string("Couldn't execute program '") + 
+               path.toString() + "'");
+  }
+
+  // Wait for it to terminate.
+  DWORD millisecondsToWait = INFINITE;
+  if (secondsToWait > 0)
+    millisecondsToWait = secondsToWait * 1000;
+
+  if (WaitForSingleObject(pi.hProcess, millisecondsToWait) == WAIT_TIMEOUT) {
+    if (!TerminateProcess(pi.hProcess, 1)) {
+      ThrowError(std::string("Failed to terminate timed-out program '") + 
+                 path.toString() + "'");
+    }
+    WaitForSingleObject(pi.hProcess, INFINITE);
+  }
+  
+  // Get its exit status.
+  DWORD status;
+  BOOL rc = GetExitCodeProcess(pi.hProcess, &status);
+
+  // Done with the handles; go close them.
+  CloseHandle(pi.hProcess);
+  CloseHandle(pi.hThread);
+
+  if (!rc)
+    ThrowError(std::string("Failed getting status for program '") + 
+               path.toString() + "'");
+
+  return status;
+}
+
+}
 // vim: sw=2 smartindent smarttab tw=80 autoindent expandtab