#include "pex-common.h"
#include <windows.h>
#ifdef HAVE_STDLIB_H
#include <stdlib.h>
#endif
#ifdef HAVE_STRING_H
#include <string.h>
#endif
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#ifdef HAVE_SYS_WAIT_H
#include <sys/wait.h>
#endif
#include <assert.h>
#include <process.h>
#include <io.h>
#include <fcntl.h>
#include <signal.h>
#include <sys/stat.h>
#include <errno.h>
#include <ctype.h>
#ifndef _P_WAIT
# define _P_WAIT 0
# define _P_NOWAIT 1
# define _P_OVERLAY 2
# define _P_NOWAITO 3
# define _P_DETACH 4
# define WAIT_CHILD 0
# define WAIT_GRANDCHILD 1
#endif
#define MINGW_NAME "Minimalist GNU for Windows"
#define MINGW_NAME_LEN (sizeof(MINGW_NAME) - 1)
extern char *stpcpy (char *dst, const char *src);
static void
backslashify (char *s)
{
while ((s = strchr (s, '/')) != NULL)
*s = '\\';
return;
}
static int pex_win32_open_read (struct pex_obj *, const char *, int);
static int pex_win32_open_write (struct pex_obj *, const char *, int);
static long pex_win32_exec_child (struct pex_obj *, int, const char *,
char * const *, char * const *,
int, int, int, int,
const char **, int *);
static int pex_win32_close (struct pex_obj *, int);
static int pex_win32_wait (struct pex_obj *, long, int *,
struct pex_time *, int, const char **, int *);
static int pex_win32_pipe (struct pex_obj *, int *, int);
static FILE *pex_win32_fdopenr (struct pex_obj *, int, int);
static FILE *pex_win32_fdopenw (struct pex_obj *, int, int);
const struct pex_funcs funcs =
{
pex_win32_open_read,
pex_win32_open_write,
pex_win32_exec_child,
pex_win32_close,
pex_win32_wait,
pex_win32_pipe,
pex_win32_fdopenr,
pex_win32_fdopenw,
NULL
};
struct pex_obj *
pex_init (int flags, const char *pname, const char *tempbase)
{
return pex_init_common (flags, pname, tempbase, &funcs);
}
static int
pex_win32_open_read (struct pex_obj *obj ATTRIBUTE_UNUSED, const char *name,
int binary)
{
return _open (name, _O_RDONLY | (binary ? _O_BINARY : _O_TEXT));
}
static int
pex_win32_open_write (struct pex_obj *obj ATTRIBUTE_UNUSED, const char *name,
int binary)
{
return _open (name,
(_O_WRONLY | _O_CREAT | _O_TRUNC
| (binary ? _O_BINARY : _O_TEXT)),
_S_IREAD | _S_IWRITE);
}
static int
pex_win32_close (struct pex_obj *obj ATTRIBUTE_UNUSED, int fd)
{
return _close (fd);
}
#ifdef USE_MINGW_MSYS
static const char *mingw_keys[] = {"SOFTWARE", "Microsoft", "Windows", "CurrentVersion", "Uninstall", NULL};
static const char *
tack_on_executable (char *buf, const char *executable)
{
char *p = strchr (buf, '\0');
if (p > buf && (p[-1] == '\\' || p[-1] == '/'))
p[-1] = '\0';
backslashify (strcat (buf, executable));
return buf;
}
static HKEY
openkey (HKEY hStart, const char *keys[])
{
HKEY hKey, hTmp;
for (hKey = hStart; *keys; keys++)
{
LONG res;
hTmp = hKey;
res = RegOpenKey (hTmp, *keys, &hKey);
if (hTmp != HKEY_LOCAL_MACHINE)
RegCloseKey (hTmp);
if (res != ERROR_SUCCESS)
return NULL;
}
return hKey;
}
static const char *
mingw_rootify (const char *executable)
{
HKEY hKey, hTmp;
DWORD maxlen;
char *namebuf, *foundbuf;
DWORD i;
LONG res;
hKey = openkey (HKEY_LOCAL_MACHINE, mingw_keys);
if (!hKey)
return executable;
if (RegQueryInfoKey (hKey, NULL, NULL, NULL, NULL, &maxlen, NULL, NULL,
NULL, NULL, NULL, NULL) != ERROR_SUCCESS)
{
RegCloseKey (hKey);
return executable;
}
namebuf = XNEWVEC (char, ++maxlen);
foundbuf = XNEWVEC (char, maxlen);
foundbuf[0] = '\0';
if (!namebuf || !foundbuf)
{
RegCloseKey (hKey);
if (namebuf)
free (namebuf);
if (foundbuf)
free (foundbuf);
return executable;
}
for (i = 0; RegEnumKey (hKey, i, namebuf, maxlen) == ERROR_SUCCESS; i++)
{
int match = strcasecmp (namebuf, MINGW_NAME);
if (match < 0)
continue;
if (match > 0 && strncasecmp (namebuf, MINGW_NAME, MINGW_NAME_LEN) > 0)
continue;
if (strcasecmp (namebuf, foundbuf) > 0)
strcpy (foundbuf, namebuf);
}
free (namebuf);
if (!foundbuf[0])
{
free (foundbuf);
RegCloseKey (hKey);
return executable;
}
res = RegOpenKey (hKey, foundbuf, &hTmp);
RegCloseKey (hKey);
free (foundbuf);
if (res != ERROR_SUCCESS)
return executable;
maxlen = 0;
if (RegQueryValueEx (hTmp, "InstallLocation", 0, NULL, NULL,
&maxlen) != ERROR_SUCCESS || maxlen == 0)
{
RegCloseKey (hTmp);
return executable;
}
foundbuf = XNEWVEC (char, maxlen + strlen (executable));
if (!foundbuf)
{
free (foundbuf);
RegCloseKey (hTmp);
}
res = RegQueryValueEx (hTmp, "InstallLocation", 0, NULL, (LPBYTE) foundbuf,
&maxlen);
RegCloseKey (hTmp);
if (res != ERROR_SUCCESS)
{
free (foundbuf);
return executable;
}
return tack_on_executable (foundbuf, executable);
}
static const char *
msys_rootify (const char *executable)
{
size_t bufsize = 64;
size_t execlen = strlen (executable) + 1;
char *buf;
DWORD res = 0;
for (;;)
{
buf = XNEWVEC (char, bufsize + execlen);
if (!buf)
break;
res = GetPrivateProfileString ("InstallSettings", "InstallPath", NULL,
buf, bufsize, "msys.ini");
if (!res)
break;
if (strlen (buf) < bufsize)
break;
res = 0;
free (buf);
bufsize *= 2;
if (bufsize > 65536)
{
buf = NULL;
break;
}
}
if (res)
return tack_on_executable (buf, executable);
if (buf)
free (buf);
return executable;
}
#endif
static char *
argv_to_cmdline (char *const *argv)
{
char *cmdline;
char *p;
size_t cmdline_len;
int i, j, k;
cmdline_len = 0;
for (i = 0; argv[i]; i++)
{
for (j = 0; argv[i][j]; j++)
{
if (argv[i][j] == '"')
{
for (k = j - 1; k >= 0 && argv[i][k] == '\\'; k--)
cmdline_len++;
cmdline_len++;
}
}
for (k = j - 1; k >= 0 && argv[i][k] == '\\'; k--)
cmdline_len++;
cmdline_len += j;
cmdline_len += 3;
}
cmdline = xmalloc (cmdline_len);
p = cmdline;
for (i = 0; argv[i]; i++)
{
*p++ = '"';
for (j = 0; argv[i][j]; j++)
{
if (argv[i][j] == '"')
{
for (k = j - 1; k >= 0 && argv[i][k] == '\\'; k--)
*p++ = '\\';
*p++ = '\\';
}
*p++ = argv[i][j];
}
for (k = j - 1; k >= 0 && argv[i][k] == '\\'; k--)
*p++ = '\\';
*p++ = '"';
*p++ = ' ';
}
p[-1] = '\0';
return cmdline;
}
static const char *const
std_suffixes[] = {
".com",
".exe",
".bat",
".cmd",
0
};
static const char *const
no_suffixes[] = {
"",
0
};
static char *
find_executable (const char *program, BOOL search)
{
char *full_executable;
char *e;
size_t fe_len;
const char *path = 0;
const char *const *ext;
const char *p, *q;
size_t proglen = strlen (program);
int has_extension = !!strchr (program, '.');
int has_slash = (strchr (program, '/') || strchr (program, '\\'));
HANDLE h;
if (has_slash)
search = FALSE;
if (search)
path = getenv ("PATH");
if (!path)
path = "";
fe_len = 0;
for (p = path; *p; p = q)
{
q = p;
while (*q != ';' && *q != '\0')
q++;
if ((size_t)(q - p) > fe_len)
fe_len = q - p;
if (*q == ';')
q++;
}
fe_len = fe_len + 1 + proglen + (has_extension ? 1 : 5);
full_executable = xmalloc (fe_len);
p = path;
do
{
q = p;
while (*q != ';' && *q != '\0')
q++;
e = full_executable;
memcpy (e, p, q - p);
e += (q - p);
if (q - p)
*e++ = '\\';
strcpy (e, program);
if (*q == ';')
q++;
for (e = full_executable; *e; e++)
if (*e == '/')
*e = '\\';
for (ext = has_extension ? no_suffixes : std_suffixes; *ext; ext++)
{
*e = '\0';
strcat (full_executable, *ext);
h = CreateFile (full_executable, GENERIC_READ,
FILE_SHARE_READ | FILE_SHARE_WRITE,
0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
if (h != INVALID_HANDLE_VALUE)
goto found;
}
p = q;
}
while (*p);
free (full_executable);
return 0;
found:
CloseHandle (h);
return full_executable;
}
static int
env_compare (const void *a_ptr, const void *b_ptr)
{
const char *a;
const char *b;
unsigned char c1;
unsigned char c2;
a = *(const char **) a_ptr;
b = *(const char **) b_ptr;
do
{
c1 = (unsigned char) tolower (*a++);
c2 = (unsigned char) tolower (*b++);
if (c1 == '=')
c1 = '\0';
if (c2 == '=')
c2 = '\0';
}
while (c1 == c2 && c1 != '\0');
return c1 - c2;
}
static long
win32_spawn (const char *executable,
BOOL search,
char *const *argv,
char *const *env,
DWORD dwCreationFlags,
LPSTARTUPINFO si,
LPPROCESS_INFORMATION pi)
{
char *full_executable;
char *cmdline;
char **env_copy;
char *env_block = NULL;
full_executable = NULL;
cmdline = NULL;
if (env)
{
int env_size;
for (env_size = 0; env[env_size]; env_size++)
continue;
if (env_size > 0)
{
int var;
int total_size = 1;
char *bufptr;
env_copy = alloca (sizeof (char *) * env_size);
memcpy (env_copy, env, sizeof (char *) * env_size);
qsort (env_copy, env_size, sizeof (char *), env_compare);
for (var = 0; var < env_size; var++)
total_size += strlen (env[var]) + 1;
env_block = malloc (total_size);
bufptr = env_block;
for (var = 0; var < env_size; var++)
bufptr = stpcpy (bufptr, env_copy[var]) + 1;
*bufptr = '\0';
}
}
full_executable = find_executable (executable, search);
if (!full_executable)
goto error;
cmdline = argv_to_cmdline (argv);
if (!cmdline)
goto error;
if (!CreateProcess (full_executable, cmdline,
NULL,
NULL,
TRUE,
dwCreationFlags,
(LPVOID) env_block,
NULL,
si,
pi))
{
if (env_block)
free (env_block);
free (full_executable);
return -1;
}
CloseHandle (pi->hThread);
free (full_executable);
if (env_block)
free (env_block);
return (long) pi->hProcess;
error:
if (env_block)
free (env_block);
if (cmdline)
free (cmdline);
if (full_executable)
free (full_executable);
return -1;
}
static long
spawn_script (const char *executable, char *const *argv,
char* const *env,
DWORD dwCreationFlags,
LPSTARTUPINFO si,
LPPROCESS_INFORMATION pi)
{
int pid = -1;
int save_errno = errno;
int fd = _open (executable, _O_RDONLY);
if (fd >= 0)
{
char buf[MAX_PATH + 5];
int len = _read (fd, buf, sizeof (buf) - 1);
_close (fd);
if (len > 3)
{
char *eol;
buf[len] = '\0';
eol = strchr (buf, '\n');
if (eol && strncmp (buf, "#!", 2) == 0)
{
char *executable1;
const char ** avhere = (const char **) --argv;
do
*eol = '\0';
while (*--eol == '\r' || *eol == ' ' || *eol == '\t');
for (executable1 = buf + 2; *executable1 == ' ' || *executable1 == '\t'; executable1++)
continue;
backslashify (executable1);
*avhere = executable1;
#ifndef USE_MINGW_MSYS
executable = strrchr (executable1, '\\') + 1;
if (!executable)
executable = executable1;
pid = win32_spawn (executable, TRUE, argv, env,
dwCreationFlags, si, pi);
#else
if (strchr (executable1, '\\') == NULL)
pid = win32_spawn (executable1, TRUE, argv, env,
dwCreationFlags, si, pi);
else if (executable1[0] != '\\')
pid = win32_spawn (executable1, FALSE, argv, env,
dwCreationFlags, si, pi);
else
{
const char *newex = mingw_rootify (executable1);
*avhere = newex;
pid = win32_spawn (newex, FALSE, argv, env,
dwCreationFlags, si, pi);
if (executable1 != newex)
free ((char *) newex);
if (pid < 0)
{
newex = msys_rootify (executable1);
if (newex != executable1)
{
*avhere = newex;
pid = win32_spawn (newex, FALSE, argv, env,
dwCreationFlags, si, pi);
free ((char *) newex);
}
}
}
#endif
}
}
}
if (pid < 0)
errno = save_errno;
return pid;
}
static long
pex_win32_exec_child (struct pex_obj *obj ATTRIBUTE_UNUSED, int flags,
const char *executable, char * const * argv,
char* const* env,
int in, int out, int errdes,
int toclose ATTRIBUTE_UNUSED,
const char **errmsg,
int *err)
{
long pid;
HANDLE stdin_handle;
HANDLE stdout_handle;
HANDLE stderr_handle;
DWORD dwCreationFlags;
OSVERSIONINFO version_info;
STARTUPINFO si;
PROCESS_INFORMATION pi;
stdin_handle = INVALID_HANDLE_VALUE;
stdout_handle = INVALID_HANDLE_VALUE;
stderr_handle = INVALID_HANDLE_VALUE;
stdin_handle = (HANDLE) _get_osfhandle (in);
stdout_handle = (HANDLE) _get_osfhandle (out);
if (!(flags & PEX_STDERR_TO_STDOUT))
stderr_handle = (HANDLE) _get_osfhandle (errdes);
else
stderr_handle = stdout_handle;
version_info.dwOSVersionInfoSize = sizeof (version_info);
GetVersionEx (&version_info);
if (version_info.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS)
dwCreationFlags = 0;
else
{
HANDLE conout_handle;
conout_handle = CreateFile("CONOUT$",
GENERIC_WRITE,
FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL,
NULL);
if (conout_handle == INVALID_HANDLE_VALUE)
dwCreationFlags = CREATE_NO_WINDOW;
else
{
CloseHandle(conout_handle);
dwCreationFlags = 0;
}
}
memset (&si, 0, sizeof (si));
si.cb = sizeof (si);
si.dwFlags = STARTF_USESTDHANDLES;
si.hStdInput = stdin_handle;
si.hStdOutput = stdout_handle;
si.hStdError = stderr_handle;
pid = win32_spawn (executable, (flags & PEX_SEARCH) != 0,
argv, env, dwCreationFlags, &si, &pi);
if (pid == -1)
pid = spawn_script (executable, argv, env, dwCreationFlags,
&si, &pi);
if (pid == -1)
{
*err = ENOENT;
*errmsg = "CreateProcess";
}
if (out != STDOUT_FILENO)
obj->funcs->close (obj, out);
if (errdes != STDERR_FILENO)
obj->funcs->close (obj, errdes);
return pid;
}
static int
pex_win32_wait (struct pex_obj *obj ATTRIBUTE_UNUSED, long pid,
int *status, struct pex_time *time, int done ATTRIBUTE_UNUSED,
const char **errmsg, int *err)
{
DWORD termstat;
HANDLE h;
if (time != NULL)
memset (time, 0, sizeof *time);
h = (HANDLE) pid;
if (WaitForSingleObject (h, INFINITE) != WAIT_OBJECT_0)
{
CloseHandle (h);
*err = ECHILD;
*errmsg = "WaitForSingleObject";
return -1;
}
GetExitCodeProcess (h, &termstat);
CloseHandle (h);
if (termstat == 3)
*status = SIGABRT;
else
*status = (termstat & 0xff) << 8;
return 0;
}
static int
pex_win32_pipe (struct pex_obj *obj ATTRIBUTE_UNUSED, int *p,
int binary)
{
return _pipe (p, 256, binary ? _O_BINARY : _O_TEXT);
}
static FILE *
pex_win32_fdopenr (struct pex_obj *obj ATTRIBUTE_UNUSED, int fd,
int binary)
{
return fdopen (fd, binary ? "rb" : "r");
}
static FILE *
pex_win32_fdopenw (struct pex_obj *obj ATTRIBUTE_UNUSED, int fd,
int binary)
{
HANDLE h = (HANDLE) _get_osfhandle (fd);
if (h == INVALID_HANDLE_VALUE)
return NULL;
if (! SetHandleInformation (h, HANDLE_FLAG_INHERIT, 0))
return NULL;
return fdopen (fd, binary ? "wb" : "w");
}
#ifdef MAIN
#include <stdio.h>
int
main (int argc ATTRIBUTE_UNUSED, char **argv)
{
char const *errmsg;
int err;
argv++;
printf ("%ld\n", pex_win32_exec_child (NULL, PEX_SEARCH, argv[0], argv, NULL, 0, 0, 1, 2, &errmsg, &err));
exit (0);
}
#endif