[Merge-on-Red] - Implement Test Process Watcher #78742
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,4 @@ | ||
| add_executable_clr(watchdog ${CMAKE_CURRENT_LIST_DIR}/watchdog.cpp) | ||
| install_clr(TARGETS watchdog DESTINATIONS . COMPONENT hosts) | ||
| install_clr(TARGETS watchdog DESTINATIONS . COMPONENT nativeaot) | ||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,139 @@ | ||
| // Licensed to the .NET Foundation under one or more agreements. | ||
| // The .NET Foundation licenses this file to you under the MIT license. | ||
|
|
||
| #include <cstdio> | ||
| #include <cstdlib> | ||
| #include <errno.h> | ||
| #include <signal.h> | ||
|
|
||
| #ifdef TARGET_WINDOWS | ||
|
|
||
| #include <windows.h> | ||
| #include <string> | ||
|
|
||
| #else // !TARGET_WINDOWS | ||
|
|
||
| #include <chrono> | ||
| #include <sys/wait.h> | ||
| #include <thread> | ||
| #include <unistd.h> | ||
| #include <vector> | ||
|
|
||
| #endif // TARGET_WINDOWS | ||
|
|
||
| int run_timed_process(const long, const int, const char *[]); | ||
|
|
||
| #ifdef TARGET_X86 | ||
| int __cdecl main(const int argc, const char *argv[]) | ||
| #else | ||
| int main(const int argc, const char *argv[]) | ||
| #endif | ||
| { | ||
| if (argc < 3) | ||
| { | ||
| printf("There are missing arguments. Got %d instead of 3+ :(\n", argc); | ||
| return EXIT_FAILURE; | ||
| } | ||
|
|
||
| const long timeout_sec = strtol(argv[1], nullptr, 10); | ||
| int exit_code = run_timed_process(timeout_sec * 1000L, argc-2, &argv[2]); | ||
|
|
||
| printf("App Exit Code: %d\n", exit_code); | ||
| return exit_code; | ||
| } | ||
|
|
||
| int run_timed_process(const long timeout_ms, const int proc_argc, const char *proc_argv[]) | ||
| { | ||
| #ifdef TARGET_WINDOWS | ||
| std::string cmdline(proc_argv[0]); | ||
|
|
||
| for (int i = 1; i < proc_argc; i++) | ||
| { | ||
| cmdline.append(" "); | ||
| cmdline.append(proc_argv[i]); | ||
| } | ||
|
|
||
| STARTUPINFOA startup_info; | ||
| PROCESS_INFORMATION proc_info; | ||
| unsigned long exit_code; | ||
|
|
||
| ZeroMemory(&startup_info, sizeof(startup_info)); | ||
| startup_info.cb = sizeof(startup_info); | ||
| ZeroMemory(&proc_info, sizeof(proc_info)); | ||
|
|
||
| if (!CreateProcessA(NULL, &cmdline[0], NULL, NULL, FALSE, 0, NULL, NULL, | ||
| &startup_info, &proc_info)) | ||
| { | ||
| int error_code = GetLastError(); | ||
| printf("Process creation failed... Code %d.\n", error_code); | ||
| return error_code; | ||
| } | ||
|
|
||
| WaitForSingleObject(proc_info.hProcess, timeout_ms); | ||
| GetExitCodeProcess(proc_info.hProcess, &exit_code); | ||
|
|
||
| CloseHandle(proc_info.hProcess); | ||
| CloseHandle(proc_info.hThread); | ||
| return exit_code; | ||
|
|
||
| #else // !TARGET_WINDOWS | ||
|
|
||
| // TODO: Describe what the 'ms_factor' is, and why it's being used here. | ||
|
There was a problem hiding this comment. Oops forgot that. Thanks for the catch. There was a problem hiding this comment. I actually removed it altogether. I needed it because originally, I was dealing with some C stuff that combined microseconds and milliseconds. After switching to C++'s |
||
| const int ms_factor = 40; | ||
| const int check_interval = 1000 / ms_factor; | ||
|
|
||
| int check_count = 0; | ||
| std::vector<const char*> args; | ||
|
|
||
| pid_t child_pid; | ||
| int child_status; | ||
| int wait_code; | ||
|
|
||
| for (int i = 0; i < proc_argc; i++) | ||
| { | ||
| args.push_back(proc_argv[i]); | ||
| } | ||
| args.push_back(NULL); | ||
|
|
||
| child_pid = fork(); | ||
|
|
||
| if (child_pid < 0) | ||
| { | ||
| // Fork failed. No memory remaining available :( | ||
| printf("Fork failed... Returning ENOMEM.\n"); | ||
| return ENOMEM; | ||
| } | ||
| else if (child_pid == 0) | ||
| { | ||
| // Instructions for child process! | ||
| execv(args[0], const_cast<char* const*>(args.data())); | ||
| } | ||
| else | ||
| { | ||
| do | ||
| { | ||
| // Instructions for the parent process! | ||
| wait_code = waitpid(child_pid, &child_status, WNOHANG); | ||
|
|
||
| if (wait_code == -1) | ||
| return EINVAL; | ||
|
|
||
| std::this_thread::sleep_for(std::chrono::milliseconds(check_interval)); | ||
|
|
||
| if (wait_code) | ||
| { | ||
| if (WIFEXITED(child_status)) | ||
| return WEXITSTATUS(child_status); | ||
| } | ||
| check_count += ms_factor; | ||
|
|
||
| } while (check_count < ((timeout_ms / check_interval) * ms_factor)); | ||
| } | ||
|
|
||
| printf("Child process took too long. Timed out... Exiting...\n"); | ||
| kill(child_pid, SIGKILL); | ||
|
|
||
| #endif // TARGET_WINDOWS | ||
| return ETIMEDOUT; | ||
| } | ||
|
|
||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Nit: Add license header