+++ /dev/null
-/*
- * PROJECT: ReactOS api tests
- * LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory
- * PURPOSE: Test for TerminateProcess
- * PROGRAMMER: Thomas Faber <thomas.faber@reactos.org>
- */
-
-#include <apitest.h>
-
-#include <ndk/obfuncs.h>
-#include <strsafe.h>
-
-static
-HANDLE
-StartChild(
- _In_ PCWSTR Argument,
- _In_ DWORD Flags,
- _Out_opt_ PDWORD ProcessId)
-{
- BOOL Success;
- WCHAR FileName[MAX_PATH];
- WCHAR CommandLine[MAX_PATH];
- STARTUPINFOW StartupInfo;
- PROCESS_INFORMATION ProcessInfo;
-
- GetModuleFileNameW(NULL, FileName, _countof(FileName));
- StringCbPrintfW(CommandLine,
- sizeof(CommandLine),
- L"\"%ls\" TerminateProcess %ls",
- FileName,
- Argument);
-
- RtlZeroMemory(&StartupInfo, sizeof(StartupInfo));
- StartupInfo.cb = sizeof(StartupInfo);
- /* HACK: running the test under rosautotest seems to keep another reference
- * to the child process around until the test finishes (on both ROS and
- * Windows)... I'm too lazy to investigate very much so let's just redirect
- * the child std handles to nowhere. ok() is useless in half the child
- * processes anyway.
- */
- StartupInfo.dwFlags = STARTF_USESTDHANDLES;
-
- Success = CreateProcessW(FileName,
- CommandLine,
- NULL,
- NULL,
- FALSE,
- Flags,
- NULL,
- NULL,
- &StartupInfo,
- &ProcessInfo);
- if (!Success)
- {
- skip("CreateProcess failed with %lu\n", GetLastError());
- if (ProcessId)
- *ProcessId = 0;
- return NULL;
- }
- CloseHandle(ProcessInfo.hThread);
- if (ProcessId)
- *ProcessId = ProcessInfo.dwProcessId;
- return ProcessInfo.hProcess;
-}
-
-static
-VOID
-TraceHandleCount_(
- _In_ HANDLE hObject,
- _In_ PCSTR File,
- _In_ INT Line)
-{
- NTSTATUS Status;
- OBJECT_BASIC_INFORMATION BasicInfo;
-
- Status = NtQueryObject(hObject,
- ObjectBasicInformation,
- &BasicInfo,
- sizeof(BasicInfo),
- NULL);
- if (!NT_SUCCESS(Status))
- {
- ok_(File, Line)(0, "NtQueryObject failed with status 0x%lx\n", Status);
- return;
- }
- ok_(File, Line)(0, "Handle %p still has %lu open handles, %lu references\n", hObject, BasicInfo.HandleCount, BasicInfo.PointerCount);
-}
-
-#define WaitExpectSuccess(h, ms) WaitExpect_(h, ms, WAIT_OBJECT_0, __FILE__, __LINE__)
-#define WaitExpectTimeout(h, ms) WaitExpect_(h, ms, WAIT_TIMEOUT, __FILE__, __LINE__)
-static
-VOID
-WaitExpect_(
- _In_ HANDLE hWait,
- _In_ DWORD Milliseconds,
- _In_ DWORD ExpectedError,
- _In_ PCSTR File,
- _In_ INT Line)
-{
- DWORD Error;
-
- Error = WaitForSingleObject(hWait, Milliseconds);
- ok_(File, Line)(Error == ExpectedError, "Wait for %p return %lu\n", hWait, Error);
-}
-
-#define CloseProcessAndVerify(hp, pid, code) CloseProcessAndVerify_(hp, pid, code, __FILE__, __LINE__)
-static
-VOID
-CloseProcessAndVerify_(
- _In_ HANDLE hProcess,
- _In_ DWORD ProcessId,
- _In_ UINT ExpectedExitCode,
- _In_ PCSTR File,
- _In_ INT Line)
-{
- int i = 0;
- DWORD Error;
- DWORD ExitCode;
- BOOL Success;
-
- WaitExpect_(hProcess, 0, WAIT_OBJECT_0, File, Line);
- Success = GetExitCodeProcess(hProcess, &ExitCode);
- ok_(File, Line)(Success, "GetExitCodeProcess failed with %lu\n", GetLastError());
- CloseHandle(hProcess);
- while ((hProcess = OpenProcess(PROCESS_QUERY_INFORMATION, FALSE, ProcessId)) != NULL)
- {
- if (++i >= 100)
- {
- TraceHandleCount_(hProcess, File, Line);
- CloseHandle(hProcess);
- break;
- }
- CloseHandle(hProcess);
- Sleep(100);
- }
- Error = GetLastError();
- ok_(File, Line)(hProcess == NULL, "OpenProcess succeeded unexpectedly for pid 0x%lx\n", ProcessId);
- ok_(File, Line)(Error == ERROR_INVALID_PARAMETER, "Error = %lu\n", Error);
- ok_(File, Line)(ExitCode == ExpectedExitCode, "Exit code is %lu but expected %u\n", ExitCode, ExpectedExitCode);
-}
-
-static
-VOID
-TestTerminateProcess(
- _In_ HANDLE hEvent)
-{
- HANDLE hProcess;
- DWORD ProcessId;
-
- /* Regular child process that returns from the test function */
- /* HACK: These two tests don't work if stdout is a pipe. See StartChild */
- ResetEvent(hEvent);
- hProcess = StartChild(L"child", 0, &ProcessId);
- WaitExpectSuccess(hEvent, 5000);
- WaitExpectSuccess(hProcess, 5000);
- CloseProcessAndVerify(hProcess, ProcessId, 0);
-
- ResetEvent(hEvent);
- hProcess = StartChild(L"child", 0, &ProcessId);
- WaitExpectSuccess(hProcess, 5000);
- WaitExpectSuccess(hEvent, 0);
- CloseProcessAndVerify(hProcess, ProcessId, 0);
-
- /* Suspended process -- never gets a chance to initialize */
- ResetEvent(hEvent);
- hProcess = StartChild(L"child", CREATE_SUSPENDED, &ProcessId);
- WaitExpectTimeout(hEvent, 100);
- WaitExpectTimeout(hProcess, 100);
- TerminateProcess(hProcess, 123);
- WaitExpectSuccess(hProcess, 5000);
- CloseProcessAndVerify(hProcess, ProcessId, 123);
-
- /* Waiting process -- we have to terminate it */
- ResetEvent(hEvent);
- hProcess = StartChild(L"wait", 0, &ProcessId);
- WaitExpectTimeout(hProcess, 100);
- TerminateProcess(hProcess, 123);
- WaitExpectSuccess(hProcess, 5000);
- CloseProcessAndVerify(hProcess, ProcessId, 123);
-
- /* Process calls ExitProcess */
- ResetEvent(hEvent);
- hProcess = StartChild(L"child exit 456", 0, &ProcessId);
- WaitExpectSuccess(hEvent, 5000);
- WaitExpectSuccess(hProcess, 5000);
- CloseProcessAndVerify(hProcess, ProcessId, 456);
-
- /* Process calls TerminateProcess with GetCurrentProcess */
- ResetEvent(hEvent);
- hProcess = StartChild(L"child terminate 456", 0, &ProcessId);
- WaitExpectSuccess(hEvent, 5000);
- WaitExpectSuccess(hProcess, 5000);
- CloseProcessAndVerify(hProcess, ProcessId, 456);
-
- /* Process calls TerminateProcess with real handle to itself */
- ResetEvent(hEvent);
- hProcess = StartChild(L"child terminate2 456", 0, &ProcessId);
- WaitExpectSuccess(hEvent, 5000);
- WaitExpectSuccess(hProcess, 5000);
- CloseProcessAndVerify(hProcess, ProcessId, 456);
-}
-
-START_TEST(TerminateProcess)
-{
- HANDLE hEvent;
- BOOL Success;
- DWORD Error;
- int argc;
- char **argv;
-
- hEvent = CreateEventW(NULL, TRUE, FALSE, L"kernel32_apitest_TerminateProcess_event");
- Error = GetLastError();
- if (!hEvent)
- {
- skip("CreateEvent failed with error %lu\n", Error);
- return;
- }
- argc = winetest_get_mainargs(&argv);
- if (argc >= 3)
- {
- ok(Error == ERROR_ALREADY_EXISTS, "Error = %lu\n", Error);
- if (!strcmp(argv[2], "wait"))
- {
- WaitExpectSuccess(hEvent, 30000);
- }
- else
- {
- Success = SetEvent(hEvent);
- ok(Success, "SetEvent failed with return %d, error %lu\n", Success, GetLastError());
- }
- }
- else
- {
- ok(Error == NO_ERROR, "Error = %lu\n", Error);
- TestTerminateProcess(hEvent);
- }
- CloseHandle(hEvent);
- if (argc >= 5)
- {
- UINT ExitCode = strtol(argv[4], NULL, 10);
-
- fflush(stdout);
- if (!strcmp(argv[3], "exit"))
- ExitProcess(ExitCode);
- else if (!strcmp(argv[3], "terminate"))
- TerminateProcess(GetCurrentProcess(), ExitCode);
- else if (!strcmp(argv[3], "terminate2"))
- {
- HANDLE hProcess;
- hProcess = OpenProcess(PROCESS_TERMINATE, FALSE, GetCurrentProcessId());
- TerminateProcess(hProcess, ExitCode);
- }
- ok(0, "Should have terminated\n");
- }
-}
projects / reactos.git / blobdiff