191 lines
5.8 KiB
C
191 lines
5.8 KiB
C
/*-
|
|
* Copyright (c) 2009-2012 Michihiro NAKAJIMA
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
|
|
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
|
|
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
|
|
* IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
|
|
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
|
|
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
|
|
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "archive_platform.h"
|
|
|
|
#if defined(_WIN32) && !defined(__CYGWIN__)
|
|
#include "archive_cmdline_private.h"
|
|
#include "archive_string.h"
|
|
|
|
#include "filter_fork.h"
|
|
|
|
pid_t
|
|
__archive_create_child(const char *cmd, int *child_stdin, int *child_stdout)
|
|
{
|
|
HANDLE childStdout[2], childStdin[2],childStderr;
|
|
SECURITY_ATTRIBUTES secAtts;
|
|
STARTUPINFOA staInfo;
|
|
PROCESS_INFORMATION childInfo;
|
|
struct archive_string cmdline;
|
|
struct archive_string fullpath;
|
|
struct archive_cmdline *acmd;
|
|
char *arg0, *ext;
|
|
int i, l;
|
|
DWORD fl, fl_old;
|
|
|
|
childStdout[0] = childStdout[1] = INVALID_HANDLE_VALUE;
|
|
childStdin[0] = childStdin[1] = INVALID_HANDLE_VALUE;
|
|
childStderr = INVALID_HANDLE_VALUE;
|
|
archive_string_init(&cmdline);
|
|
archive_string_init(&fullpath);
|
|
|
|
acmd = __archive_cmdline_allocate();
|
|
if (acmd == NULL)
|
|
goto fail;
|
|
if (__archive_cmdline_parse(acmd, cmd) != ARCHIVE_OK)
|
|
goto fail;
|
|
|
|
/*
|
|
* Search the full path of 'path'.
|
|
* NOTE: This does not need if we give CreateProcessA 'path' as
|
|
* a part of the cmdline and give CreateProcessA NULL as first
|
|
* parameter, but I do not like that way.
|
|
*/
|
|
ext = strrchr(acmd->path, '.');
|
|
if (ext == NULL || strlen(ext) > 4)
|
|
/* 'path' does not have a proper extension, so we have to
|
|
* give SearchPath() ".exe" as the extension. */
|
|
ext = ".exe";
|
|
else
|
|
ext = NULL;/* 'path' has an extension. */
|
|
|
|
fl = MAX_PATH;
|
|
do {
|
|
if (archive_string_ensure(&fullpath, fl) == NULL)
|
|
goto fail;
|
|
fl_old = fl;
|
|
fl = SearchPathA(NULL, acmd->path, ext, fl, fullpath.s,
|
|
&arg0);
|
|
} while (fl != 0 && fl > fl_old);
|
|
if (fl == 0)
|
|
goto fail;
|
|
|
|
/*
|
|
* Make a command line.
|
|
*/
|
|
for (l = 0, i = 0; acmd->argv[i] != NULL; i++) {
|
|
if (i == 0)
|
|
continue;
|
|
l += (int)strlen(acmd->argv[i]) + 1;
|
|
}
|
|
if (archive_string_ensure(&cmdline, l + 1) == NULL)
|
|
goto fail;
|
|
for (i = 0; acmd->argv[i] != NULL; i++) {
|
|
if (i == 0) {
|
|
const char *p, *sp;
|
|
|
|
if ((p = strchr(acmd->argv[i], '/')) != NULL ||
|
|
(p = strchr(acmd->argv[i], '\\')) != NULL)
|
|
p++;
|
|
else
|
|
p = acmd->argv[i];
|
|
if ((sp = strchr(p, ' ')) != NULL)
|
|
archive_strappend_char(&cmdline, '"');
|
|
archive_strcat(&cmdline, p);
|
|
if (sp != NULL)
|
|
archive_strappend_char(&cmdline, '"');
|
|
} else {
|
|
archive_strappend_char(&cmdline, ' ');
|
|
archive_strcat(&cmdline, acmd->argv[i]);
|
|
}
|
|
}
|
|
if (i <= 1) {
|
|
const char *sp;
|
|
|
|
if ((sp = strchr(arg0, ' ')) != NULL)
|
|
archive_strappend_char(&cmdline, '"');
|
|
archive_strcat(&cmdline, arg0);
|
|
if (sp != NULL)
|
|
archive_strappend_char(&cmdline, '"');
|
|
}
|
|
|
|
secAtts.nLength = sizeof(SECURITY_ATTRIBUTES);
|
|
secAtts.bInheritHandle = TRUE;
|
|
secAtts.lpSecurityDescriptor = NULL;
|
|
if (CreatePipe(&childStdout[0], &childStdout[1], &secAtts, 0) == 0)
|
|
goto fail;
|
|
if (!SetHandleInformation(childStdout[0], HANDLE_FLAG_INHERIT, 0))
|
|
goto fail;
|
|
if (CreatePipe(&childStdin[0], &childStdin[1], &secAtts, 0) == 0)
|
|
goto fail;
|
|
if (!SetHandleInformation(childStdin[1], HANDLE_FLAG_INHERIT, 0))
|
|
goto fail;
|
|
if (DuplicateHandle(GetCurrentProcess(), GetStdHandle(STD_ERROR_HANDLE),
|
|
GetCurrentProcess(), &childStderr, 0, TRUE,
|
|
DUPLICATE_SAME_ACCESS) == 0)
|
|
goto fail;
|
|
|
|
memset(&staInfo, 0, sizeof(staInfo));
|
|
staInfo.cb = sizeof(staInfo);
|
|
staInfo.hStdError = childStderr;
|
|
staInfo.hStdOutput = childStdout[1];
|
|
staInfo.hStdInput = childStdin[0];
|
|
staInfo.wShowWindow = SW_HIDE;
|
|
staInfo.dwFlags = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
|
|
if (CreateProcessA(fullpath.s, cmdline.s, NULL, NULL, TRUE, 0,
|
|
NULL, NULL, &staInfo, &childInfo) == 0)
|
|
goto fail;
|
|
WaitForInputIdle(childInfo.hProcess, INFINITE);
|
|
CloseHandle(childInfo.hProcess);
|
|
CloseHandle(childInfo.hThread);
|
|
|
|
*child_stdout = _open_osfhandle((intptr_t)childStdout[0], _O_RDONLY);
|
|
*child_stdin = _open_osfhandle((intptr_t)childStdin[1], _O_WRONLY);
|
|
|
|
CloseHandle(childStdout[1]);
|
|
CloseHandle(childStdin[0]);
|
|
|
|
archive_string_free(&cmdline);
|
|
archive_string_free(&fullpath);
|
|
__archive_cmdline_free(acmd);
|
|
return (childInfo.dwProcessId);
|
|
|
|
fail:
|
|
if (childStdout[0] != INVALID_HANDLE_VALUE)
|
|
CloseHandle(childStdout[0]);
|
|
if (childStdout[1] != INVALID_HANDLE_VALUE)
|
|
CloseHandle(childStdout[1]);
|
|
if (childStdin[0] != INVALID_HANDLE_VALUE)
|
|
CloseHandle(childStdin[0]);
|
|
if (childStdin[1] != INVALID_HANDLE_VALUE)
|
|
CloseHandle(childStdin[1]);
|
|
if (childStderr != INVALID_HANDLE_VALUE)
|
|
CloseHandle(childStderr);
|
|
archive_string_free(&cmdline);
|
|
archive_string_free(&fullpath);
|
|
__archive_cmdline_free(acmd);
|
|
return (-1);
|
|
}
|
|
|
|
void
|
|
__archive_check_child(int in, int out)
|
|
{
|
|
(void)in; /* UNUSED */
|
|
(void)out; /* UNUSED */
|
|
Sleep(100);
|
|
}
|
|
|
|
#endif /* _WIN32 && !__CYGWIN__ */
|