|
Packit |
b5b901 |
/* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
|
|
Packit |
b5b901 |
*
|
|
Packit |
b5b901 |
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
Packit |
b5b901 |
* of this software and associated documentation files (the "Software"), to
|
|
Packit |
b5b901 |
* deal in the Software without restriction, including without limitation the
|
|
Packit |
b5b901 |
* rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
|
|
Packit |
b5b901 |
* sell copies of the Software, and to permit persons to whom the Software is
|
|
Packit |
b5b901 |
* furnished to do so, subject to the following conditions:
|
|
Packit |
b5b901 |
*
|
|
Packit |
b5b901 |
* The above copyright notice and this permission notice shall be included in
|
|
Packit |
b5b901 |
* all copies or substantial portions of the Software.
|
|
Packit |
b5b901 |
*
|
|
Packit |
b5b901 |
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
Packit |
b5b901 |
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
Packit |
b5b901 |
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
Packit |
b5b901 |
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
Packit |
b5b901 |
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
Packit |
b5b901 |
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
|
|
Packit |
b5b901 |
* IN THE SOFTWARE.
|
|
Packit |
b5b901 |
*/
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
#include <fcntl.h>
|
|
Packit |
b5b901 |
#include <io.h>
|
|
Packit |
b5b901 |
#include <malloc.h>
|
|
Packit |
b5b901 |
#include <stdio.h>
|
|
Packit |
b5b901 |
#include <process.h>
|
|
Packit |
b5b901 |
#if !defined(__MINGW32__)
|
|
Packit |
b5b901 |
# include <crtdbg.h>
|
|
Packit |
b5b901 |
#endif
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
#include "task.h"
|
|
Packit |
b5b901 |
#include "runner.h"
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
/*
|
|
Packit |
b5b901 |
* Define the stuff that MinGW doesn't have
|
|
Packit |
b5b901 |
*/
|
|
Packit |
b5b901 |
#ifndef GetFileSizeEx
|
|
Packit |
b5b901 |
WINBASEAPI BOOL WINAPI GetFileSizeEx(HANDLE hFile,
|
|
Packit |
b5b901 |
PLARGE_INTEGER lpFileSize);
|
|
Packit |
b5b901 |
#endif
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
/* Do platform-specific initialization. */
|
|
Packit Service |
e08953 |
void platform_init(int argc, char **argv) {
|
|
Packit |
b5b901 |
/* Disable the "application crashed" popup. */
|
|
Packit |
b5b901 |
SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOGPFAULTERRORBOX |
|
|
Packit |
b5b901 |
SEM_NOOPENFILEERRORBOX);
|
|
Packit |
b5b901 |
#if !defined(__MINGW32__)
|
|
Packit |
b5b901 |
_CrtSetReportMode(_CRT_ASSERT, _CRTDBG_MODE_DEBUG);
|
|
Packit |
b5b901 |
_CrtSetReportMode(_CRT_ERROR, _CRTDBG_MODE_DEBUG);
|
|
Packit |
b5b901 |
#endif
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
_setmode(0, _O_BINARY);
|
|
Packit |
b5b901 |
_setmode(1, _O_BINARY);
|
|
Packit |
b5b901 |
_setmode(2, _O_BINARY);
|
|
Packit |
b5b901 |
|
|
Packit Service |
e08953 |
#ifdef _MSC_VER
|
|
Packit Service |
e08953 |
_set_fmode(_O_BINARY);
|
|
Packit Service |
e08953 |
#else
|
|
Packit Service |
e08953 |
_fmode = _O_BINARY;
|
|
Packit Service |
e08953 |
#endif
|
|
Packit Service |
e08953 |
|
|
Packit |
b5b901 |
/* Disable stdio output buffering. */
|
|
Packit |
b5b901 |
setvbuf(stdout, NULL, _IONBF, 0);
|
|
Packit |
b5b901 |
setvbuf(stderr, NULL, _IONBF, 0);
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
strcpy(executable_path, argv[0]);
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
int process_start(char *name, char *part, process_info_t *p, int is_helper) {
|
|
Packit |
b5b901 |
HANDLE file = INVALID_HANDLE_VALUE;
|
|
Packit |
b5b901 |
HANDLE nul = INVALID_HANDLE_VALUE;
|
|
Packit |
b5b901 |
WCHAR path[MAX_PATH], filename[MAX_PATH];
|
|
Packit |
b5b901 |
WCHAR image[MAX_PATH + 1];
|
|
Packit |
b5b901 |
WCHAR args[MAX_PATH * 2];
|
|
Packit |
b5b901 |
STARTUPINFOW si;
|
|
Packit |
b5b901 |
PROCESS_INFORMATION pi;
|
|
Packit |
b5b901 |
DWORD result;
|
|
Packit |
b5b901 |
|
|
Packit Service |
e08953 |
if (!is_helper) {
|
|
Packit Service |
e08953 |
/* Give the helpers time to settle. Race-y, fix this. */
|
|
Packit Service |
e08953 |
uv_sleep(250);
|
|
Packit Service |
e08953 |
}
|
|
Packit Service |
e08953 |
|
|
Packit |
b5b901 |
if (GetTempPathW(sizeof(path) / sizeof(WCHAR), (WCHAR*)&path) == 0)
|
|
Packit |
b5b901 |
goto error;
|
|
Packit |
b5b901 |
if (GetTempFileNameW((WCHAR*)&path, L"uv", 0, (WCHAR*)&filename) == 0)
|
|
Packit |
b5b901 |
goto error;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
file = CreateFileW((WCHAR*)filename,
|
|
Packit |
b5b901 |
GENERIC_READ | GENERIC_WRITE,
|
|
Packit |
b5b901 |
0,
|
|
Packit |
b5b901 |
NULL,
|
|
Packit |
b5b901 |
CREATE_ALWAYS,
|
|
Packit |
b5b901 |
FILE_ATTRIBUTE_TEMPORARY | FILE_FLAG_DELETE_ON_CLOSE,
|
|
Packit |
b5b901 |
NULL);
|
|
Packit |
b5b901 |
if (file == INVALID_HANDLE_VALUE)
|
|
Packit |
b5b901 |
goto error;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
if (!SetHandleInformation(file, HANDLE_FLAG_INHERIT, HANDLE_FLAG_INHERIT))
|
|
Packit |
b5b901 |
goto error;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
nul = CreateFileA("nul",
|
|
Packit |
b5b901 |
GENERIC_READ,
|
|
Packit |
b5b901 |
FILE_SHARE_READ | FILE_SHARE_WRITE,
|
|
Packit |
b5b901 |
NULL,
|
|
Packit |
b5b901 |
OPEN_EXISTING,
|
|
Packit |
b5b901 |
FILE_ATTRIBUTE_NORMAL,
|
|
Packit |
b5b901 |
NULL);
|
|
Packit |
b5b901 |
if (nul == INVALID_HANDLE_VALUE)
|
|
Packit |
b5b901 |
goto error;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
if (!SetHandleInformation(nul, HANDLE_FLAG_INHERIT, HANDLE_FLAG_INHERIT))
|
|
Packit |
b5b901 |
goto error;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
result = GetModuleFileNameW(NULL,
|
|
Packit |
b5b901 |
(WCHAR*) &image,
|
|
Packit |
b5b901 |
sizeof(image) / sizeof(WCHAR));
|
|
Packit |
b5b901 |
if (result == 0 || result == sizeof(image))
|
|
Packit |
b5b901 |
goto error;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
if (part) {
|
|
Packit |
b5b901 |
if (_snwprintf((WCHAR*)args,
|
|
Packit |
b5b901 |
sizeof(args) / sizeof(WCHAR),
|
|
Packit |
b5b901 |
L"\"%s\" %S %S",
|
|
Packit |
b5b901 |
image,
|
|
Packit |
b5b901 |
name,
|
|
Packit |
b5b901 |
part) < 0) {
|
|
Packit |
b5b901 |
goto error;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
} else {
|
|
Packit |
b5b901 |
if (_snwprintf((WCHAR*)args,
|
|
Packit |
b5b901 |
sizeof(args) / sizeof(WCHAR),
|
|
Packit |
b5b901 |
L"\"%s\" %S",
|
|
Packit |
b5b901 |
image,
|
|
Packit |
b5b901 |
name) < 0) {
|
|
Packit |
b5b901 |
goto error;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
memset((void*)&si, 0, sizeof(si));
|
|
Packit |
b5b901 |
si.cb = sizeof(si);
|
|
Packit |
b5b901 |
si.dwFlags = STARTF_USESTDHANDLES;
|
|
Packit |
b5b901 |
si.hStdInput = nul;
|
|
Packit |
b5b901 |
si.hStdOutput = file;
|
|
Packit |
b5b901 |
si.hStdError = file;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
if (!CreateProcessW(image, args, NULL, NULL, TRUE,
|
|
Packit |
b5b901 |
0, NULL, NULL, &si, &pi))
|
|
Packit |
b5b901 |
goto error;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
CloseHandle(pi.hThread);
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
SetHandleInformation(nul, HANDLE_FLAG_INHERIT, 0);
|
|
Packit |
b5b901 |
SetHandleInformation(file, HANDLE_FLAG_INHERIT, 0);
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
p->stdio_in = nul;
|
|
Packit |
b5b901 |
p->stdio_out = file;
|
|
Packit |
b5b901 |
p->process = pi.hProcess;
|
|
Packit |
b5b901 |
p->name = part;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
return 0;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
error:
|
|
Packit |
b5b901 |
if (file != INVALID_HANDLE_VALUE)
|
|
Packit |
b5b901 |
CloseHandle(file);
|
|
Packit |
b5b901 |
if (nul != INVALID_HANDLE_VALUE)
|
|
Packit |
b5b901 |
CloseHandle(nul);
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
return -1;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
/* Timeout is in msecs. Set timeout < 0 to never time out. Returns 0 when all
|
|
Packit |
b5b901 |
* processes are terminated, -2 on timeout. */
|
|
Packit |
b5b901 |
int process_wait(process_info_t *vec, int n, int timeout) {
|
|
Packit |
b5b901 |
int i;
|
|
Packit |
b5b901 |
HANDLE handles[MAXIMUM_WAIT_OBJECTS];
|
|
Packit |
b5b901 |
DWORD timeout_api, result;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
/* If there's nothing to wait for, return immediately. */
|
|
Packit |
b5b901 |
if (n == 0)
|
|
Packit |
b5b901 |
return 0;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
ASSERT(n <= MAXIMUM_WAIT_OBJECTS);
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
for (i = 0; i < n; i++)
|
|
Packit |
b5b901 |
handles[i] = vec[i].process;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
if (timeout >= 0) {
|
|
Packit |
b5b901 |
timeout_api = (DWORD)timeout;
|
|
Packit |
b5b901 |
} else {
|
|
Packit |
b5b901 |
timeout_api = INFINITE;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
result = WaitForMultipleObjects(n, handles, TRUE, timeout_api);
|
|
Packit |
b5b901 |
|
|
Packit Service |
e08953 |
if (result < WAIT_OBJECT_0 + n) {
|
|
Packit |
b5b901 |
/* All processes are terminated. */
|
|
Packit |
b5b901 |
return 0;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
if (result == WAIT_TIMEOUT) {
|
|
Packit |
b5b901 |
return -2;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
return -1;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
long int process_output_size(process_info_t *p) {
|
|
Packit |
b5b901 |
LARGE_INTEGER size;
|
|
Packit |
b5b901 |
if (!GetFileSizeEx(p->stdio_out, &size))
|
|
Packit |
b5b901 |
return -1;
|
|
Packit |
b5b901 |
return (long int)size.QuadPart;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
int process_copy_output(process_info_t* p, FILE* stream) {
|
|
Packit |
b5b901 |
char buf[1024];
|
|
Packit |
b5b901 |
int fd, r;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
fd = _open_osfhandle((intptr_t)p->stdio_out, _O_RDONLY | _O_TEXT);
|
|
Packit |
b5b901 |
if (fd == -1)
|
|
Packit |
b5b901 |
return -1;
|
|
Packit |
b5b901 |
|
|
Packit Service |
e08953 |
r = _lseek(fd, 0, SEEK_SET);
|
|
Packit |
b5b901 |
if (r < 0)
|
|
Packit |
b5b901 |
return -1;
|
|
Packit |
b5b901 |
|
|
Packit Service |
e08953 |
while ((r = _read(fd, buf, sizeof(buf))) != 0)
|
|
Packit Service |
e08953 |
print_lines(buf, r, stream);
|
|
Packit |
b5b901 |
|
|
Packit Service |
e08953 |
_close(fd);
|
|
Packit |
b5b901 |
return 0;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
int process_read_last_line(process_info_t *p,
|
|
Packit |
b5b901 |
char * buffer,
|
|
Packit |
b5b901 |
size_t buffer_len) {
|
|
Packit |
b5b901 |
DWORD size;
|
|
Packit |
b5b901 |
DWORD read;
|
|
Packit |
b5b901 |
DWORD start;
|
|
Packit |
b5b901 |
OVERLAPPED overlapped;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
ASSERT(buffer_len > 0);
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
size = GetFileSize(p->stdio_out, NULL);
|
|
Packit |
b5b901 |
if (size == INVALID_FILE_SIZE)
|
|
Packit |
b5b901 |
return -1;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
if (size == 0) {
|
|
Packit |
b5b901 |
buffer[0] = '\0';
|
|
Packit |
b5b901 |
return 1;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
memset(&overlapped, 0, sizeof overlapped);
|
|
Packit |
b5b901 |
if (size >= buffer_len)
|
|
Packit |
b5b901 |
overlapped.Offset = size - buffer_len - 1;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
if (!ReadFile(p->stdio_out, buffer, buffer_len - 1, &read, &overlapped))
|
|
Packit |
b5b901 |
return -1;
|
|
Packit |
b5b901 |
|
|
Packit Service |
e08953 |
start = read;
|
|
Packit Service |
e08953 |
while (start-- > 0) {
|
|
Packit |
b5b901 |
if (buffer[start] == '\n' || buffer[start] == '\r')
|
|
Packit |
b5b901 |
break;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
if (start > 0)
|
|
Packit |
b5b901 |
memmove(buffer, buffer + start, read - start);
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
buffer[read - start] = '\0';
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
return 0;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
char* process_get_name(process_info_t *p) {
|
|
Packit |
b5b901 |
return p->name;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
int process_terminate(process_info_t *p) {
|
|
Packit |
b5b901 |
if (!TerminateProcess(p->process, 1))
|
|
Packit |
b5b901 |
return -1;
|
|
Packit |
b5b901 |
return 0;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
int process_reap(process_info_t *p) {
|
|
Packit |
b5b901 |
DWORD exitCode;
|
|
Packit |
b5b901 |
if (!GetExitCodeProcess(p->process, &exitCode))
|
|
Packit |
b5b901 |
return -1;
|
|
Packit |
b5b901 |
return (int)exitCode;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
void process_cleanup(process_info_t *p) {
|
|
Packit |
b5b901 |
CloseHandle(p->process);
|
|
Packit |
b5b901 |
CloseHandle(p->stdio_in);
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
|
|
Packit Service |
e08953 |
static int clear_line(void) {
|
|
Packit |
b5b901 |
HANDLE handle;
|
|
Packit |
b5b901 |
CONSOLE_SCREEN_BUFFER_INFO info;
|
|
Packit |
b5b901 |
COORD coord;
|
|
Packit |
b5b901 |
DWORD written;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
handle = (HANDLE)_get_osfhandle(fileno(stderr));
|
|
Packit |
b5b901 |
if (handle == INVALID_HANDLE_VALUE)
|
|
Packit |
b5b901 |
return -1;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
if (!GetConsoleScreenBufferInfo(handle, &info))
|
|
Packit |
b5b901 |
return -1;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
coord = info.dwCursorPosition;
|
|
Packit |
b5b901 |
if (coord.Y <= 0)
|
|
Packit |
b5b901 |
return -1;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
coord.X = 0;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
if (!SetConsoleCursorPosition(handle, coord))
|
|
Packit |
b5b901 |
return -1;
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
if (!FillConsoleOutputCharacterW(handle,
|
|
Packit |
b5b901 |
0x20,
|
|
Packit |
b5b901 |
info.dwSize.X,
|
|
Packit |
b5b901 |
coord,
|
|
Packit |
b5b901 |
&written)) {
|
|
Packit |
b5b901 |
return -1;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
return 0;
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
|
|
Packit |
b5b901 |
void rewind_cursor() {
|
|
Packit |
b5b901 |
if (clear_line() == -1) {
|
|
Packit |
b5b901 |
/* If clear_line fails (stdout is not a console), print a newline. */
|
|
Packit |
b5b901 |
fprintf(stderr, "\n");
|
|
Packit |
b5b901 |
}
|
|
Packit |
b5b901 |
}
|