Mercurial

comparison third_party/libuv/test/runner-win.c @ 160:948de3f54cea

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
[ThirdParty] Added libuv
author June Park <parkjune1995@gmail.com>
date Wed, 14 Jan 2026 19:39:52 -0800
parents
children
comparison
equal deleted inserted replaced
159:05cf9467a1c3 160:948de3f54cea
1 /* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
2 *
3 * Permission is hereby granted, free of charge, to any person obtaining a copy
4 * of this software and associated documentation files (the "Software"), to
5 * deal in the Software without restriction, including without limitation the
6 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
7 * sell copies of the Software, and to permit persons to whom the Software is
8 * furnished to do so, subject to the following conditions:
9 *
10 * The above copyright notice and this permission notice shall be included in
11 * all copies or substantial portions of the Software.
12 *
13 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
17 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
18 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
19 * IN THE SOFTWARE.
20 */
21
22 #include <fcntl.h>
23 #include <io.h>
24 #include <malloc.h>
25 #include <stdio.h>
26 #include <process.h>
27 #if !defined(__MINGW32__)
28 # include <crtdbg.h>
29 #endif
30
31
32 #include "task.h"
33 #include "runner.h"
34
35
36 /*
37 * Define the stuff that MinGW doesn't have
38 */
39 #ifndef GetFileSizeEx
40 WINBASEAPI BOOL WINAPI GetFileSizeEx(HANDLE hFile,
41 PLARGE_INTEGER lpFileSize);
42 #endif
43
44
45 /* Do platform-specific initialization. */
46 void platform_init(int argc, char **argv) {
47 /* Disable the "application crashed" popup. */
48 SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOGPFAULTERRORBOX |
49 SEM_NOOPENFILEERRORBOX);
50 #if !defined(__MINGW32__)
51 _CrtSetReportMode(_CRT_ASSERT, _CRTDBG_MODE_DEBUG);
52 _CrtSetReportMode(_CRT_ERROR, _CRTDBG_MODE_DEBUG);
53 #endif
54
55 _setmode(0, _O_BINARY);
56 _setmode(1, _O_BINARY);
57 _setmode(2, _O_BINARY);
58
59 #ifdef _MSC_VER
60 _set_fmode(_O_BINARY);
61 #else
62 _fmode = _O_BINARY;
63 #endif
64
65 /* Disable stdio output buffering. */
66 setvbuf(stdout, NULL, _IONBF, 0);
67 setvbuf(stderr, NULL, _IONBF, 0);
68
69 strcpy(executable_path, argv[0]);
70 }
71
72
73 int process_start(char *name, char *part, process_info_t *p, int is_helper) {
74 HANDLE file = INVALID_HANDLE_VALUE;
75 HANDLE nul = INVALID_HANDLE_VALUE;
76 WCHAR path[MAX_PATH], filename[MAX_PATH];
77 WCHAR image[MAX_PATH + 1];
78 WCHAR args[MAX_PATH * 2];
79 STARTUPINFOW si;
80 PROCESS_INFORMATION pi;
81 DWORD result;
82
83 if (!is_helper) {
84 /* Give the helpers time to settle. Race-y, fix this. */
85 uv_sleep(250);
86 }
87
88 if (GetTempPathW(sizeof(path) / sizeof(WCHAR), (WCHAR*)&path) == 0)
89 goto error;
90 if (GetTempFileNameW((WCHAR*)&path, L"uv", 0, (WCHAR*)&filename) == 0)
91 goto error;
92
93 file = CreateFileW((WCHAR*)filename,
94 GENERIC_READ | GENERIC_WRITE,
95 0,
96 NULL,
97 CREATE_ALWAYS,
98 FILE_ATTRIBUTE_TEMPORARY | FILE_FLAG_DELETE_ON_CLOSE,
99 NULL);
100 if (file == INVALID_HANDLE_VALUE)
101 goto error;
102
103 if (!SetHandleInformation(file, HANDLE_FLAG_INHERIT, HANDLE_FLAG_INHERIT))
104 goto error;
105
106 nul = CreateFileA("nul",
107 GENERIC_READ,
108 FILE_SHARE_READ | FILE_SHARE_WRITE,
109 NULL,
110 OPEN_EXISTING,
111 FILE_ATTRIBUTE_NORMAL,
112 NULL);
113 if (nul == INVALID_HANDLE_VALUE)
114 goto error;
115
116 if (!SetHandleInformation(nul, HANDLE_FLAG_INHERIT, HANDLE_FLAG_INHERIT))
117 goto error;
118
119 result = GetModuleFileNameW(NULL,
120 (WCHAR*) &image,
121 sizeof(image) / sizeof(WCHAR));
122 if (result == 0 || result == sizeof(image))
123 goto error;
124
125 if (part) {
126 if (_snwprintf((WCHAR*)args,
127 sizeof(args) / sizeof(WCHAR),
128 L"\"%s\" %S %S",
129 image,
130 name,
131 part) < 0) {
132 goto error;
133 }
134 } else {
135 if (_snwprintf((WCHAR*)args,
136 sizeof(args) / sizeof(WCHAR),
137 L"\"%s\" %S",
138 image,
139 name) < 0) {
140 goto error;
141 }
142 }
143
144 memset((void*)&si, 0, sizeof(si));
145 si.cb = sizeof(si);
146 si.dwFlags = STARTF_USESTDHANDLES;
147 si.hStdInput = nul;
148 si.hStdOutput = file;
149 si.hStdError = file;
150
151 if (!CreateProcessW(image, args, NULL, NULL, TRUE,
152 0, NULL, NULL, &si, &pi))
153 goto error;
154
155 CloseHandle(pi.hThread);
156
157 SetHandleInformation(nul, HANDLE_FLAG_INHERIT, 0);
158 SetHandleInformation(file, HANDLE_FLAG_INHERIT, 0);
159
160 p->stdio_in = nul;
161 p->stdio_out = file;
162 p->process = pi.hProcess;
163 p->name = part;
164
165 return 0;
166
167 error:
168 if (file != INVALID_HANDLE_VALUE)
169 CloseHandle(file);
170 if (nul != INVALID_HANDLE_VALUE)
171 CloseHandle(nul);
172
173 return -1;
174 }
175
176
177 /* Timeout is in msecs. Set timeout < 0 to never time out. Returns 0 when all
178 * processes are terminated, -2 on timeout. */
179 int process_wait(process_info_t *vec, int n, int timeout) {
180 int i;
181 HANDLE handles[MAXIMUM_WAIT_OBJECTS];
182 DWORD timeout_api, result;
183
184 /* If there's nothing to wait for, return immediately. */
185 if (n == 0)
186 return 0;
187
188 ASSERT_LE(n, MAXIMUM_WAIT_OBJECTS);
189
190 for (i = 0; i < n; i++)
191 handles[i] = vec[i].process;
192
193 if (timeout >= 0) {
194 timeout_api = (DWORD)timeout;
195 } else {
196 timeout_api = INFINITE;
197 }
198
199 result = WaitForMultipleObjects(n, handles, TRUE, timeout_api);
200
201 if (result < WAIT_OBJECT_0 + n) {
202 /* All processes are terminated. */
203 return 0;
204 }
205 if (result == WAIT_TIMEOUT) {
206 return -2;
207 }
208 return -1;
209 }
210
211
212 long int process_output_size(process_info_t *p) {
213 LARGE_INTEGER size;
214 if (!GetFileSizeEx(p->stdio_out, &size))
215 return -1;
216 return (long int)size.QuadPart;
217 }
218
219
220 int process_copy_output(process_info_t* p, FILE* stream) {
221 char buf[1024];
222 int partial;
223 int fd, r;
224
225 fd = _open_osfhandle((intptr_t)p->stdio_out, _O_RDONLY | _O_TEXT);
226 if (fd == -1)
227 return -1;
228
229 r = _lseek(fd, 0, SEEK_SET);
230 if (r < 0)
231 return -1;
232
233 partial = 0;
234 while ((r = _read(fd, buf, sizeof(buf))) != 0)
235 partial = print_lines(buf, r, stream, partial);
236
237 _close(fd);
238 return 0;
239 }
240
241
242 int process_read_last_line(process_info_t *p,
243 char * buffer,
244 size_t buffer_len) {
245 DWORD size;
246 DWORD read;
247 DWORD start;
248 OVERLAPPED overlapped;
249
250 ASSERT_GT(buffer_len, 0);
251
252 size = GetFileSize(p->stdio_out, NULL);
253 if (size == INVALID_FILE_SIZE)
254 return -1;
255
256 if (size == 0) {
257 buffer[0] = '\0';
258 return 1;
259 }
260
261 memset(&overlapped, 0, sizeof overlapped);
262 if (size >= buffer_len)
263 overlapped.Offset = size - buffer_len - 1;
264
265 if (!ReadFile(p->stdio_out, buffer, buffer_len - 1, &read, &overlapped))
266 return -1;
267
268 start = read;
269 while (start-- > 0) {
270 if (buffer[start] == '\n' || buffer[start] == '\r')
271 break;
272 }
273
274 if (start > 0)
275 memmove(buffer, buffer + start, read - start);
276
277 buffer[read - start] = '\0';
278
279 return 0;
280 }
281
282
283 char* process_get_name(process_info_t *p) {
284 return p->name;
285 }
286
287
288 int process_terminate(process_info_t *p) {
289 if (!TerminateProcess(p->process, 1))
290 return -1;
291 return 0;
292 }
293
294
295 int process_reap(process_info_t *p) {
296 DWORD exitCode;
297 if (!GetExitCodeProcess(p->process, &exitCode))
298 return -1;
299 return (int)exitCode;
300 }
301
302
303 void process_cleanup(process_info_t *p) {
304 CloseHandle(p->process);
305 CloseHandle(p->stdio_in);
306 }
307
308
309 static int clear_line(void) {
310 HANDLE handle;
311 CONSOLE_SCREEN_BUFFER_INFO info;
312 COORD coord;
313 DWORD written;
314
315 handle = (HANDLE)_get_osfhandle(_fileno(stderr));
316 if (handle == INVALID_HANDLE_VALUE)
317 return -1;
318
319 if (!GetConsoleScreenBufferInfo(handle, &info))
320 return -1;
321
322 coord = info.dwCursorPosition;
323 if (coord.Y <= 0)
324 return -1;
325
326 coord.X = 0;
327
328 if (!SetConsoleCursorPosition(handle, coord))
329 return -1;
330
331 if (!FillConsoleOutputCharacterW(handle,
332 0x20,
333 info.dwSize.X,
334 coord,
335 &written)) {
336 return -1;
337 }
338
339 return 0;
340 }
341
342
343 void rewind_cursor() {
344 if (clear_line() == -1) {
345 /* If clear_line fails (stdout is not a console), print a newline. */
346 fprintf(stderr, "\n");
347 }
348 }