1 //===- Win32/Program.cpp - Win32 Program Implementation ------- -*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file was developed by Jeff Cohen and is distributed under the
6 // University of Illinois Open Source License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file provides the Win32 specific implementation of the Program class.
12 //===----------------------------------------------------------------------===//
17 //===----------------------------------------------------------------------===//
18 //=== WARNING: Implementation here must contain only Win32 specific code
19 //=== and must not be UNIX code
20 //===----------------------------------------------------------------------===//
25 // This function just uses the PATH environment variable to find the program.
27 Program::FindProgramByName(const std::string& progName) {
29 // Check some degenerate cases
30 if (progName.length() == 0) // no program
33 if (!temp.setFile(progName)) // invalid name
35 if (temp.executable()) // already executable as is
38 // At this point, the file name is valid and its not executable.
39 // Let Windows search for it.
40 char buffer[MAX_PATH];
42 DWORD len = SearchPath(NULL, progName.c_str(), ".exe", MAX_PATH,
45 // See if it wasn't found.
49 // See if we got the entire path.
53 // Buffer was too small; grow and retry.
55 char *b = reinterpret_cast<char *>(_alloca(len+1));
56 DWORD len2 = SearchPath(NULL, progName.c_str(), ".exe", len+1, b, &dummy);
58 // It is unlikely the search failed, but it's always possible some file
59 // was added or removed since the last search, so be paranoid...
71 Program::ExecuteAndWait(const Path& path,
72 const std::vector<std::string>& args,
74 if (!path.executable())
75 throw path.toString() + " is not executable";
77 // Windows wants a command line, not an array of args, to pass to the new
78 // process. We have to concatenate them all, while quoting the args that
79 // have embedded spaces.
81 // First, determine the length of the command line.
82 std::string progname(path.getLast());
83 unsigned len = progname.length() + 1;
84 if (progname.find(' ') != std::string::npos)
87 for (unsigned i = 0; i < args.size(); i++) {
88 len += args[i].length() + 1;
89 if (args[i].find(' ') != std::string::npos)
93 // Now build the command line.
94 char *command = reinterpret_cast<char *>(_alloca(len));
97 bool needsQuoting = progname.find(' ') != std::string::npos;
100 memcpy(p, progname.c_str(), progname.length());
101 p += progname.length();
106 for (unsigned i = 0; i < args.size(); i++) {
107 const std::string& arg = args[i];
108 needsQuoting = arg.find(' ') != std::string::npos;
111 memcpy(p, arg.c_str(), arg.length());
120 // Create a child process.
122 memset(&si, 0, sizeof(si));
125 PROCESS_INFORMATION pi;
126 memset(&pi, 0, sizeof(pi));
128 if (!CreateProcess(path.c_str(), command, NULL, NULL, FALSE, 0,
129 envp, NULL, &si, &pi))
131 ThrowError(std::string("Couldn't execute program '") +
132 path.toString() + "'");
135 // Wait for it to terminate.
136 WaitForSingleObject(pi.hProcess, INFINITE);
138 // Get its exit status.
140 BOOL rc = GetExitCodeProcess(pi.hProcess, &status);
142 // Done with the handles; go close them.
143 CloseHandle(pi.hProcess);
144 CloseHandle(pi.hThread);
147 ThrowError(std::string("Failed getting status for program '") +
148 path.toString() + "'");
154 // vim: sw=2 smartindent smarttab tw=80 autoindent expandtab