Imported Upstream version 1.0.0.9127upstream/1.0.0.9127

1 files changed, 168 insertions, 0 deletions

diff --git a/src/pal/tests/palsuite/threading/ExitThread/test2/test2.cpp b/src/pal/tests/palsuite/threading/ExitThread/test2/test2.cpp
new file mode 100644
index 0000000000..c31af8a079
--- /dev/null
+++ b/src/pal/tests/palsuite/threading/ExitThread/test2/test2.cpp
@@ -0,0 +1,168 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+// See the LICENSE file in the project root for more information.
+
+/*=============================================================================
+**
+** Source: test2.c
+**
+** Purpose: Test to ensure ExitThread() called from the last thread of
+**          a process shuts down that process and returns the proper
+**          exit code as specified in the ExitThread() call.
+** 
+** Dependencies: PAL_Initialize
+**               PAL_Terminate
+**               Fail
+**               ZeroMemory
+**               GetCurrentDirectoryW
+**               CreateProcessW
+**               WaitForSingleObject
+**               GetLastError
+**               strlen
+**               strncpy
+** 
+
+**
+**===========================================================================*/
+#include <palsuite.h>
+#include "myexitcode.h"
+
+
+static const char* rgchPathDelim = "\\";
+
+
+int 
+mkAbsoluteFilename( LPSTR dirName,  
+                    DWORD dwDirLength, 
+                    LPCSTR fileName, 
+                    DWORD dwFileLength,
+                    LPSTR absPathName )
+{
+    DWORD sizeDN, sizeFN, sizeAPN;
+
+    sizeDN = strlen( dirName );
+    sizeFN = strlen( fileName );
+    sizeAPN = (sizeDN + 1 + sizeFN + 1);
+
+    /* ensure ((dirName + DELIM + fileName + \0) =< _MAX_PATH ) */
+    if( sizeAPN > _MAX_PATH )
+    {
+        return ( 0 );
+    }
+    
+    strncpy( absPathName, dirName, dwDirLength +1 );
+    strncpy( absPathName, rgchPathDelim, 2 );
+    strncpy( absPathName, fileName, dwFileLength +1 );
+
+    return (sizeAPN);
+  
+} 
+
+
+int __cdecl main( int argc, char **argv ) 
+
+{
+    const char* rgchChildFile = "childprocess";
+    
+    STARTUPINFO si;
+    PROCESS_INFORMATION pi;
+
+    DWORD dwError;
+    DWORD dwExitCode;
+    DWORD dwFileLength;
+    DWORD dwDirLength;
+    DWORD dwSize;
+    DWORD dwExpected = TEST_EXIT_CODE;
+    
+    char  rgchDirName[_MAX_DIR];  
+    char  absPathBuf[_MAX_PATH];
+    char* rgchAbsPathName;
+
+    /* initialize the PAL */
+    if( PAL_Initialize(argc, argv) != 0 )
+    {
+	    return( FAIL );
+    }
+    
+    /* zero our process and startup info structures */
+    ZeroMemory( &si, sizeof(si) );
+    si.cb = sizeof( si );
+    ZeroMemory( &pi, sizeof(pi) );
+    
+    /* build the absolute path to the child process */
+    rgchAbsPathName = &absPathBuf[0];
+    dwFileLength = strlen( rgchChildFile );
+    
+    dwDirLength = GetCurrentDirectory( _MAX_PATH, rgchDirName );
+    if( dwDirLength == 0 ) 
+    {
+        dwError = GetLastError();
+        Fail( "GetCurrentDirectory call failed with error code %d\n", 
+              dwError ); 
+    }
+
+    dwSize = mkAbsoluteFilename(   rgchDirName,
+                                   dwDirLength,
+                                   rgchChildFile, 
+                                   dwFileLength,
+                                   rgchAbsPathName );
+    if( dwSize == 0 )
+    {
+        Fail( "Palsuite Code: mkAbsoluteFilename() call failed.  Could ", 
+              "not build absolute path name to file\n.  Exiting.\n" ); 
+    }
+
+    /* launch the child process */
+    if( !CreateProcess(     NULL,               /* module name to execute */
+                            rgchAbsPathName,    /* command line */
+                            NULL,               /* process handle not */
+                                                /* inheritable */
+                            NULL,               /* thread handle not */
+                                                /* inheritable */
+                            FALSE,              /* handle inheritance */
+                            CREATE_NEW_CONSOLE, /* dwCreationFlags */
+                            NULL,               /* use parent's environment */
+                            NULL,               /* use parent's starting */
+                                                /* directory */
+                            &si,                /* startup info struct */
+                            &pi )               /* process info struct */
+        )
+    {
+        dwError = GetLastError();
+        Fail( "CreateProcess call failed with error code %d\n", 
+              dwError ); 
+    }
+    
+    /* wait for the child process to complete */
+    WaitForSingleObject ( pi.hProcess, INFINITE );
+
+    /* check the exit code from the process */
+    if( ! GetExitCodeProcess( pi.hProcess, &dwExitCode ) )
+    {
+        dwError = GetLastError();
+        CloseHandle ( pi.hProcess );
+        CloseHandle ( pi.hThread );
+        Fail( "GetExitCodeProcess call failed with error code %d\n", 
+              dwError ); 
+    }
+    
+    /* close process and thread handle */
+    CloseHandle ( pi.hProcess );
+    CloseHandle ( pi.hThread );
+    
+    /* check for the expected exit code */
+    /* exit code for some systems is as small as a char, so that's all */
+    /* we'll compare for checking success                              */
+    if( LOBYTE(LOWORD(dwExitCode)) != LOBYTE(LOWORD(dwExpected)) )
+    {
+        Fail( "GetExitCodeProcess returned an incorrect exit code %d, "
+              "expected value is %d\n", 
+              LOWORD(dwExitCode), dwExpected ); 
+    }
+
+    /* terminate the PAL */
+    PAL_Terminate();
+    
+    /* return success */
+    return PASS; 
+}