| /* Copyright Joyent, Inc. and other Node contributors. All rights reserved. |
| * |
| * Permission is hereby granted, free of charge, to any person obtaining a copy |
| * of this software and associated documentation files (the "Software"), to |
| * deal in the Software without restriction, including without limitation the |
| * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or |
| * sell copies of the Software, and to permit persons to whom the Software is |
| * furnished to do so, subject to the following conditions: |
| * |
| * The above copyright notice and this permission notice shall be included in |
| * all copies or substantial portions of the Software. |
| * |
| * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
| * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
| * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
| * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
| * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING |
| * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS |
| * IN THE SOFTWARE. |
| */ |
| |
| #include <errno.h> |
| #include <stdio.h> |
| #include <string.h> |
| |
| #ifdef _WIN32 |
| # include <io.h> |
| #else |
| # include <unistd.h> |
| #endif |
| |
| #include "uv.h" |
| #include "runner.h" |
| #include "task.h" |
| |
| /* Actual tests and helpers are defined in test-list.h */ |
| #include "test-list.h" |
| |
| /* The time in milliseconds after which a single test times out. */ |
| #define TEST_TIMEOUT 5000 |
| |
| int ipc_helper(int listen_after_write); |
| int ipc_helper_tcp_connection(void); |
| int ipc_send_recv_helper(void); |
| int stdio_over_pipes_helper(void); |
| |
| static int maybe_run_test(int argc, char **argv); |
| |
| |
| int main(int argc, char **argv) { |
| platform_init(argc, argv); |
| |
| argv = uv_setup_args(argc, argv); |
| |
| switch (argc) { |
| case 1: return run_tests(TEST_TIMEOUT, 0); |
| case 2: return maybe_run_test(argc, argv); |
| case 3: return run_test_part(argv[1], argv[2]); |
| default: |
| LOGF("Too many arguments.\n"); |
| return 1; |
| } |
| } |
| |
| |
| static int maybe_run_test(int argc, char **argv) { |
| if (strcmp(argv[1], "--list") == 0) { |
| print_tests(stdout); |
| return 0; |
| } |
| |
| if (strcmp(argv[1], "ipc_helper_listen_before_write") == 0) { |
| return ipc_helper(0); |
| } |
| |
| if (strcmp(argv[1], "ipc_helper_listen_after_write") == 0) { |
| return ipc_helper(1); |
| } |
| |
| if (strcmp(argv[1], "ipc_send_recv_helper") == 0) { |
| return ipc_send_recv_helper(); |
| } |
| |
| if (strcmp(argv[1], "ipc_helper_tcp_connection") == 0) { |
| return ipc_helper_tcp_connection(); |
| } |
| |
| if (strcmp(argv[1], "stdio_over_pipes_helper") == 0) { |
| return stdio_over_pipes_helper(); |
| } |
| |
| if (strcmp(argv[1], "spawn_helper1") == 0) { |
| return 1; |
| } |
| |
| if (strcmp(argv[1], "spawn_helper2") == 0) { |
| printf("hello world\n"); |
| return 1; |
| } |
| |
| if (strcmp(argv[1], "spawn_helper3") == 0) { |
| char buffer[256]; |
| ASSERT(buffer == fgets(buffer, sizeof(buffer) - 1, stdin)); |
| buffer[sizeof(buffer) - 1] = '\0'; |
| fputs(buffer, stdout); |
| return 1; |
| } |
| |
| if (strcmp(argv[1], "spawn_helper4") == 0) { |
| /* Never surrender, never return! */ |
| while (1) uv_sleep(10000); |
| } |
| |
| if (strcmp(argv[1], "spawn_helper5") == 0) { |
| const char out[] = "fourth stdio!\n"; |
| #ifdef _WIN32 |
| DWORD bytes; |
| WriteFile((HANDLE) _get_osfhandle(3), out, sizeof(out) - 1, &bytes, NULL); |
| #else |
| { |
| ssize_t r; |
| |
| do |
| r = write(3, out, sizeof(out) - 1); |
| while (r == -1 && errno == EINTR); |
| |
| fsync(3); |
| } |
| #endif |
| return 1; |
| } |
| |
| if (strcmp(argv[1], "spawn_helper6") == 0) { |
| int r; |
| |
| r = fprintf(stdout, "hello world\n"); |
| ASSERT(r > 0); |
| |
| r = fprintf(stderr, "hello errworld\n"); |
| ASSERT(r > 0); |
| |
| return 1; |
| } |
| |
| if (strcmp(argv[1], "spawn_helper7") == 0) { |
| int r; |
| char *test; |
| /* Test if the test value from the parent is still set */ |
| test = getenv("ENV_TEST"); |
| ASSERT(test != NULL); |
| |
| r = fprintf(stdout, "%s", test); |
| ASSERT(r > 0); |
| |
| return 1; |
| } |
| |
| return run_test(argv[1], TEST_TIMEOUT, 0, 1); |
| } |