[Merge-on-Red] - Implement Test Process Watcher

eng/Subsets.props

@@ -149,6 +149,7 @@
     <SubsetName Include="Host.Pkg" Description="The .NET host packages." />
     <SubsetName Include="Host.Tools" Description="The .NET hosting libraries." />
     <SubsetName Include="Host.Tests" Description="The .NET hosting tests." />
+    <SubsetName Include="Host.Watchdog" Description="The .NET tests watchdog (Experimental)." />
 
     <!-- Libs -->
     <SubsetName Include="Libs" Description="The libraries native part, refs and source assemblies, test infra and packages, but NOT the tests (use Libs.Tests to request those explicitly). Equivalent to: $(DefaultLibrariesSubsets)" />
@@ -387,6 +388,10 @@
     <ProjectToBuild Include="@(PkgprojProjectToBuild)" Pack="true" Category="host" />
   </ItemGroup>
 
+  <ItemGroup Condition="$(_subset.Contains('+host.watchdog+'))">
+    <ProjectToBuild Include="$(SharedNativeRoot)watchdog\build.proj" Category="host" />
+  </ItemGroup>
+
   <!-- Libraries sets -->
   <ItemGroup Condition="$(_subset.Contains('+libs.native+'))">
     <ProjectToBuild Include="$(SharedNativeRoot)libs\build-native.proj" Category="libs" />

src/native/watchdog/CMakeLists.txt

@@ -0,0 +1,26 @@
+cmake_minimum_required(VERSION 3.6.2)
+
+if (CLR_CMAKE_TARGET_IOS OR CLR_CMAKE_TARGET_TVOS)
+    # CMake 3.14.5 contains bug fixes for iOS
+    cmake_minimum_required(VERSION 3.14.5)
+elseif (CLR_CMAKE_TARGET_MACCATALYST)
+    # CMake 3.18.1 properly generates MacCatalyst C compiler
+    cmake_minimum_required(VERSION 3.18.1)
+endif ()
+
+if (WIN32)
+    cmake_policy(SET CMP0091 NEW)
+else ()
+    cmake_policy(SET CMP0042 NEW)
+endif ()
+
+project(watchdog)
+
+include(../../../eng/native/configurepaths.cmake)
+include(${CLR_ENG_NATIVE_DIR}/configurecompiler.cmake)
+
+set(CMAKE_INCLUDE_CURRENT_DIR ON)
+set(CMAKE_CXX_STANDARD 11)
+
+add_executable(watchdog watchdog.cpp)
+

src/native/watchdog/build-watchdog.sh

@@ -0,0 +1,4 @@
+#!/usr/bin/env bash
+
+echo "Called build script successfully!"
+

src/native/watchdog/build.proj

@@ -0,0 +1,41 @@
+<Project Sdk="Microsoft.Build.Traversal">
+
+  <PropertyGroup>
+    <AssemblyName>.NET Tests Watchdog</AssemblyName>
+    <_BuildGeneralArgs>-arch $(TargetArchitecture) -c $(Configuration) -os $(TargetOS)</_BuildGeneralArgs>
+  </PropertyGroup>
+
+
+  <!-- Instructions to build on Unix. -->
+
+  <Target Name="BuildForUnix"
+          Condition="!$([MSBuild]::IsOsPlatform(Windows))">
+
+    <PropertyGroup>
+      <_CMakeArgs Condition="'$(CMakeArgs)' != ''"> $(CMakeArgs)</_CMakeArgs>
+      <_BuildUnixArgs>$(_BuildGeneralArgs)$(_CMakeArgs)</_BuildUnixArgs>
+    </PropertyGroup>
+
+    <Message Text="$(MSBuildThisFileDirectory)build-watchdog.sh $(_BuildUnixArgs)" Importance="High" />
+    <Exec Command="&quot;$(MSBuildThisFileDirectory)build-watchdog.sh&quot; $(_BuildUnixArgs)" />
+
+  </Target>
+
+
+  <!-- Instructions to build on Windows. -->
+
+  <Target Name="BuildForWindows"
+          Condition="$([MSBuild]::IsOsPlatform(Windows))">
+
+    <PropertyGroup>
+      <_BuildWindowsArgs>$(_BuildGeneralArgs)</_BuildWindowsArgs>
+      <_BuildWindowsArgs Condition="'$(Ninja)' == 'false'">$(_BuildWindowsArgs) msbuild</_BuildWindowsArgs>
+    </PropertyGroup>
+
+    <Message Text="$(MSBuildThisFileDirectory)build-watchdog.cmd $(_BuildWindowsArgs)" Importance="High" />
+    <Exec Command="&quot;$(MSBuildThisFileDirectory)build-watchdog.cmd&quot; $(_BuildWindowsArgs)" />
+
+  </Target>
+
+</Project>
+

src/native/watchdog/watchdog.cpp

@@ -0,0 +1,128 @@
+#include <cstdio>
+#include <cstdlib>
+#include <cstdarg>
+#include <errno.h>
+#include <signal.h>
+
+#ifdef _WIN32
+#include <windows.h>
+#include <string>
+#else
+#include <unistd.h>
+#include <sys/wait.h>
+#endif
+
+int run_timed_process(const long, const int, const char *[]);
+
+int main(const int argc, const char *argv[])
+{
+    if (argc < 3)
+    {
+        printf("There are missing arguments. Got %d instead of 3+ :(\n", argc);
+        return EXIT_FAILURE;
+    }
+
+    const long timeout_ms = strtol(argv[1], nullptr, 10);
+    int exit_code = run_timed_process(timeout_ms, argc-2, &argv[2]);
+
+    printf("App Exit Code: %d\n", exit_code);
+    return EXIT_SUCCESS;
+}
+
+int run_timed_process(const long timeout, const int exe_argc, const char *exe_path_and_argv[])
+{
+#ifdef _WIN32
+    std::string cmdline(exe_path_and_argv[0]);
+
+    for (int i = 1; i < exe_argc; i++)
+    {
+        cmdline.append(" ");
+        cmdline.append(exe_path_and_argv[i]);
+    }
+
+    STARTUPINFO startup_info;
+    PROCESS_INFORMATION proc_info;
+    int exit_code;
+
+    ZeroMemory(&startup_info, sizeof(startup_info));
+    startup_info.cb = sizeof(startup_info);
+    ZeroMemory(&proc_info, sizeof(proc_info));
+
+    if (!CreateProcess(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);
+    GetExitCodeProcess(proc_info.hProcess, &exit_code);
+
+    CloseHandle(proc_info.hProcess);
+    CloseHandle(proc_info.hThread);
+    return exit_code;
+
+#else
+    const int check_interval = 1000;
+    int check_count = 0;
+    char **args = new char *[exe_argc];
+
+    pid_t child_pid;
+    int child_status;
+    int wait_code;
+
+    for (int i = 0; i < exe_argc; i++)
+    {
+        args[i] = (char *) exe_path_and_argv[i];
+    }
+
+    // This is just for development. Will remove it when it's ready to be submitted :)
+    for (int j = 0; j < exe_argc; j++)
+    {
+        printf("[%d]: %s\n", j, args[j]);
+    }
+
+    child_pid = fork();
+
+    if (child_pid < 0)
+    {
+        printf("Fork failed... No memory available.\n");
+        return ENOMEM;
+    }
+    else if (child_pid == 0)
+    {
+        printf("Running child process...\n");
+        execv(args[0], &args[0]);
+    }
+    else
+    {
+        do
+        {
+            wait_code = waitpid(child_pid, &child_status, WNOHANG);
+
+            // Something went terribly wrong.
+            if (wait_code == -1)
+                return EINVAL;
+
+            // Passing ms * 1000 because usleep() receives its parameter in microseconds.
+            usleep(check_interval * 1000);
+
+            if (wait_code)
+            {
+                if (WIFEXITED(child_status))
+                {
+                    printf("Child process exited successfully with status %d.\n",
+                            WEXITSTATUS(child_status));
+                    return WEXITSTATUS(child_status);
+                }
+            }
+        } while (check_count++ < (timeout / check_interval));
+    }
+
+    printf("Child process took too long and timed out... Exiting it...\n");
+    kill(child_pid, SIGKILL);
+#endif
+    return ETIMEDOUT;
+}
+