1 files changed, 17 insertions, 25 deletions
diff --git a/lib/System/Win32/Program.inc b/lib/System/Win32/Program.inc
index af6cce6..f1faff0 100644
--- a/lib/System/Win32/Program.inc
+++ b/lib/System/Win32/Program.inc
@@ -25,21 +25,6 @@
 namespace llvm {
 using namespace sys;
 
-Program::Program() : Data_(0) {}
-
-Program::~Program() {
-  if (Data_) {
-    HANDLE hProcess = reinterpret_cast<HANDLE>(Data_);
-    CloseHandle(hProcess);
-    Data_ = 0;
-  }
-}
-
-unsigned Program::GetPid() const {
-  HANDLE hProcess = reinterpret_cast<HANDLE>(Data_);
-  return GetProcessId(hProcess);
-}
-
 // This function just uses the PATH environment variable to find the program.
 Path
 Program::FindProgramByName(const std::string& progName) {
@@ -137,11 +122,6 @@ Program::Execute(const Path& path,
                  const Path** redirects,
                  unsigned memoryLimit,
                  std::string* ErrMsg) {
-  if (Data_) {
-    HANDLE hProcess = reinterpret_cast<HANDLE>(Data_);
-    CloseHandle(Data_);
-    Data_ = 0;
-  }
 
   if (!path.canExecute()) {
     if (ErrMsg)
@@ -269,9 +249,10 @@ Program::Execute(const Path& path,
                path.str() + "'");
     return false;
   }
-  Data_ = reinterpret_cast<void*>(pi.hProcess);
+  Pid_ = pi.dwProcessId;
 
   // Make sure these get closed no matter what.
+  AutoHandle hProcess(pi.hProcess);
   AutoHandle hThread(pi.hThread);
 
   // Assign the process to a job if a memory limit is defined.
@@ -305,12 +286,17 @@ Program::Execute(const Path& path,
 int
 Program::Wait(unsigned secondsToWait,
               std::string* ErrMsg) {
-  if (Data_ == 0) {
+  if (Pid_ == 0) {
     MakeErrMsg(ErrMsg, "Process not started!");
     return -1;
   }
 
-  HANDLE hProcess = reinterpret_cast<HANDLE>(Data_);
+  HANDLE hOpen = OpenProcess(SYNCHRONIZE, FALSE, Pid_);
+  if (hOpen == NULL) {
+    MakeErrMsg(ErrMsg, "OpenProcess failed!");
+    return -1;
+  }
+  AutoHandle hProcess(hOpen);
 
   // Wait for the process to terminate.
   DWORD millisecondsToWait = INFINITE;
@@ -341,12 +327,18 @@ Program::Wait(unsigned secondsToWait,
 
 bool
 Program::Kill(std::string* ErrMsg) {
-  if (Data_ == 0) {
+  if (Pid_ == 0) {
     MakeErrMsg(ErrMsg, "Process not started!");
     return true;
   }
 
-  HANDLE hProcess = reinterpret_cast<HANDLE>(Data_);
+  HANDLE hOpen = OpenProcess(PROCESS_TERMINATE, FALSE, Pid_);
+  if (hOpen == NULL) {
+    MakeErrMsg(ErrMsg, "OpenProcess failed!");
+    return true;
+  }
+  AutoHandle hProcess(hOpen);
+
   if (TerminateProcess(hProcess, 1) == 0) {
     MakeErrMsg(ErrMsg, "The process couldn't be killed!");
     return true;