For PR351:
Implement the new environment pointer for ExecuteAndWait
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@18928 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/System/Unix/Program.cpp b/lib/System/Unix/Program.cpp
index ae53720..35192ce 100644
--- a/lib/System/Unix/Program.cpp
+++ b/lib/System/Unix/Program.cpp
@@ -79,7 +79,8 @@
//
int
Program::ExecuteAndWait(const Path& path,
- const std::vector<std::string>& args) {
+ const std::vector<std::string>& args,
+ const char ** envp ) {
if (!path.executable())
throw path.toString() + " is not executable";
@@ -103,11 +104,15 @@
break;
// Child process: Execute the program.
- case 0:
- execve (path.c_str(), (char** const)argv, environ);
+ case 0: {
+ char** env = environ;
+ if (envp != 0)
+ env = (char**) envp;
+ execve (path.c_str(), (char** const)argv, env);
// If the execve() failed, we should exit and let the parent pick up
// our non-zero exit status.
exit (errno);
+ }
// Parent process: Break out of the switch to do our processing.
default:
diff --git a/lib/System/Unix/Program.inc b/lib/System/Unix/Program.inc
index ae53720..35192ce 100644
--- a/lib/System/Unix/Program.inc
+++ b/lib/System/Unix/Program.inc
@@ -79,7 +79,8 @@
//
int
Program::ExecuteAndWait(const Path& path,
- const std::vector<std::string>& args) {
+ const std::vector<std::string>& args,
+ const char ** envp ) {
if (!path.executable())
throw path.toString() + " is not executable";
@@ -103,11 +104,15 @@
break;
// Child process: Execute the program.
- case 0:
- execve (path.c_str(), (char** const)argv, environ);
+ case 0: {
+ char** env = environ;
+ if (envp != 0)
+ env = (char**) envp;
+ execve (path.c_str(), (char** const)argv, env);
// If the execve() failed, we should exit and let the parent pick up
// our non-zero exit status.
exit (errno);
+ }
// Parent process: Break out of the switch to do our processing.
default:
diff --git a/lib/System/Win32/Program.cpp b/lib/System/Win32/Program.cpp
index d6bd86a..4aff5e9 100644
--- a/lib/System/Win32/Program.cpp
+++ b/lib/System/Win32/Program.cpp
@@ -69,7 +69,8 @@
//
int
Program::ExecuteAndWait(const Path& path,
- const std::vector<std::string>& args) {
+ const std::vector<std::string>& args,
+ const char** env) {
if (!path.executable())
throw path.toString() + " is not executable";
@@ -124,8 +125,9 @@
PROCESS_INFORMATION pi;
memset(&pi, 0, sizeof(pi));
+ LPVOID lpEnvironment = envp;
if (!CreateProcess(path.c_str(), command, NULL, NULL, FALSE, 0,
- NULL, NULL, &si, &pi))
+ lpEnvironment, NULL, &si, &pi))
{
ThrowError(std::string("Couldn't execute program '") +
path.toString() + "'");
diff --git a/lib/System/Win32/Program.inc b/lib/System/Win32/Program.inc
index d6bd86a..4aff5e9 100644
--- a/lib/System/Win32/Program.inc
+++ b/lib/System/Win32/Program.inc
@@ -69,7 +69,8 @@
//
int
Program::ExecuteAndWait(const Path& path,
- const std::vector<std::string>& args) {
+ const std::vector<std::string>& args,
+ const char** env) {
if (!path.executable())
throw path.toString() + " is not executable";
@@ -124,8 +125,9 @@
PROCESS_INFORMATION pi;
memset(&pi, 0, sizeof(pi));
+ LPVOID lpEnvironment = envp;
if (!CreateProcess(path.c_str(), command, NULL, NULL, FALSE, 0,
- NULL, NULL, &si, &pi))
+ lpEnvironment, NULL, &si, &pi))
{
ThrowError(std::string("Couldn't execute program '") +
path.toString() + "'");