aboutsummaryrefslogtreecommitdiff
path: root/butl/process.cxx
diff options
context:
space:
mode:
authorBoris Kolpackov <boris@codesynthesis.com>2017-05-01 14:02:49 +0200
committerBoris Kolpackov <boris@codesynthesis.com>2017-05-01 14:09:24 +0200
commit442c1a6790e52baa0c081f310d4d9e9b6f1ff638 (patch)
treeeb5b95db50367e8326407e1ae5fff52da21a59f3 /butl/process.cxx
parent0eff5aede5ad7622ef184fcd0c24f6e088e9d031 (diff)
Try to detect MSYS2 processes
Diffstat (limited to 'butl/process.cxx')
-rw-r--r--butl/process.cxx55
1 files changed, 51 insertions, 4 deletions
diff --git a/butl/process.cxx b/butl/process.cxx
index db529a6..940560c 100644
--- a/butl/process.cxx
+++ b/butl/process.cxx
@@ -12,6 +12,8 @@
#else
# include <butl/win32-utility>
+# include <psapi.h> // EnumProcessModules(), etc
+
# include <io.h> // _get_osfhandle(), _close()
# include <stdlib.h> // _MAX_PATH
# include <sys/types.h> // stat
@@ -43,6 +45,8 @@
#include <butl/fdstream> // fdnull()
#include <butl/process-details>
+#include <iostream>
+
using namespace std;
#ifdef _WIN32
@@ -1166,7 +1170,9 @@ namespace butl
// be in the returned list if it has failed to initialize). With this
// improvement we could then wait longer and try harder.
//
- for (size_t ret (0); ret != 20; ++ret)
+ optional<bool> msys; // Absent if we don't know.
+
+ for (size_t ret (0); ret != 5; ++ret)
{
if (!CreateProcess (
batch != nullptr ? batch : pp.effect_string (),
@@ -1183,13 +1189,54 @@ namespace butl
auto_handle (pi.hThread).reset (); // Close.
- // Wait a bit (50ms) and check if the process has terminated. If it is
+ // Detect if this is an MSYS2 process by checking if the process has
+ // loaded msys-2.0.dll.
+ //
+ size_t wait (200);
+
+ if (!msys)
+ {
+ // Wait a bit for the process to load its DLLs.
+ //
+ if (WaitForSingleObject (pi.hProcess, 50) == WAIT_TIMEOUT)
+ {
+ wait -= 50;
+
+ DWORD mn;
+ HMODULE ms[32]; // Normally it is one of the first.
+
+ if (EnumProcessModules (pi.hProcess, ms, sizeof (ms), &mn))
+ {
+ for (DWORD i (0); !msys && i != mn / sizeof (HMODULE); ++i)
+ {
+ char p[_MAX_PATH + 1];
+ if (GetModuleFileNameExA (pi.hProcess, ms[i], p, sizeof (p)))
+ {
+ size_t n (strlen (p));
+ if (n >= 12 && casecmp (p + n - 12, "msys-2.0.dll") == 0)
+ msys = true;
+ }
+ }
+
+ if (!msys)
+ msys = false;
+ }
+ // EnumProcessModules() failed (presumably because the process has
+ // already exited), fall through.
+ }
+ // Process exited, fall through.
+ }
+
+ if (msys && !*msys)
+ break;
+
+ // Wait a bit longer and check if the process has terminated. If it is
// still running then we assume all is good. Otherwise, retry if this
// is the DLL initialization error.
//
DWORD s;
- if (WaitForSingleObject (pi.hProcess, 50) != WAIT_OBJECT_0 ||
- !GetExitCodeProcess (pi.hProcess, &s) ||
+ if (WaitForSingleObject (pi.hProcess, wait) != WAIT_OBJECT_0 ||
+ !GetExitCodeProcess (pi.hProcess, &s) ||
s != STATUS_DLL_INIT_FAILED)
break;
}