mirror of
https://github.com/eclipse-cdt/cdt
synced 2025-04-23 22:52:11 +02:00
Removed killer.cpp.
This commit is contained in:
Doug Schaefer
parent
07fa3d405e
commit
2ba5d92ebb
4 changed files with 2 additions and 399 deletions
|
|
@ -1,269 +0,0 @@
|
||||||
/*******************************************************************************
|
|
||||||
* Copyright (c) 2002 - 2005 QNX Software Systems and others.
|
|
||||||
* All rights reserved. This program and the accompanying materials
|
|
||||||
* are made available under the terms of the Eclipse Public License v1.0
|
|
||||||
* which accompanies this distribution, and is available at
|
|
||||||
* http://www.eclipse.org/legal/epl-v10.html
|
|
||||||
*
|
|
||||||
* Contributors:
|
|
||||||
* QNX Software Systems - initial API and implementation
|
|
||||||
*
|
|
||||||
* starter.cpp
|
|
||||||
*
|
|
||||||
* This is a helper function for the process killing
|
|
||||||
* Implementation based on the article "Terminating Windows Processes"
|
|
||||||
* see http://www.alexfedotov.com/articles/killproc.asp
|
|
||||||
***********************************************************************/
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#define STRICT
|
|
||||||
#include <Windows.h>
|
|
||||||
#include <Tlhelp32.h>
|
|
||||||
#include <process.h>
|
|
||||||
#include <tchar.h>
|
|
||||||
#include <stdio.h>
|
|
||||||
|
|
||||||
#include "killer.h"
|
|
||||||
|
|
||||||
#define SystemProcessesAndThreadsInformation 5
|
|
||||||
|
|
||||||
#define MAX_CMD_LINE_LENGTH 512
|
|
||||||
|
|
||||||
//#define DEBUG_MONITOR
|
|
||||||
|
|
||||||
void DisplayErrorMessage();
|
|
||||||
|
|
||||||
BOOL KillProcessEx(
|
|
||||||
IN DWORD dwProcessId // Handle of the process
|
|
||||||
)
|
|
||||||
{
|
|
||||||
|
|
||||||
OSVERSIONINFO osvi;
|
|
||||||
DWORD dwError;
|
|
||||||
#ifdef DEBUG_MONITOR
|
|
||||||
_TCHAR buffer[MAX_CMD_LINE_LENGTH];
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// determine operating system version
|
|
||||||
osvi.dwOSVersionInfoSize = sizeof(osvi);
|
|
||||||
GetVersionEx(&osvi);
|
|
||||||
|
|
||||||
if (osvi.dwPlatformId == VER_PLATFORM_WIN32_NT)
|
|
||||||
{
|
|
||||||
HINSTANCE hNtDll;
|
|
||||||
NTSTATUS (WINAPI * pZwQuerySystemInformation)(UINT, PVOID,
|
|
||||||
ULONG, PULONG);
|
|
||||||
|
|
||||||
// get NTDLL.DLL handle
|
|
||||||
hNtDll = GetModuleHandleW(_T("ntdll.dll"));
|
|
||||||
if(hNtDll == NULL) {
|
|
||||||
#ifdef DEBUG_MONITOR
|
|
||||||
_stprintf(buffer, _T("Failed to get ntdll.dll handle"));
|
|
||||||
OutputDebugStringW(buffer);
|
|
||||||
#endif
|
|
||||||
return FALSE;
|
|
||||||
}
|
|
||||||
|
|
||||||
// find address of ZwQuerySystemInformation
|
|
||||||
*(FARPROC *)&pZwQuerySystemInformation =
|
|
||||||
GetProcAddress(hNtDll, "ZwQuerySystemInformation");
|
|
||||||
if (pZwQuerySystemInformation == NULL)
|
|
||||||
return SetLastError(ERROR_PROC_NOT_FOUND), NULL;
|
|
||||||
|
|
||||||
// get default process heap handle
|
|
||||||
HANDLE hHeap = GetProcessHeap();
|
|
||||||
|
|
||||||
NTSTATUS Status;
|
|
||||||
ULONG cbBuffer = 0x8000;
|
|
||||||
PVOID pBuffer = NULL;
|
|
||||||
|
|
||||||
// it is difficult to predict what buffer size will be
|
|
||||||
// enough, so we start with 32K buffer and increase its
|
|
||||||
// size as needed
|
|
||||||
do
|
|
||||||
{
|
|
||||||
pBuffer = HeapAlloc(hHeap, 0, cbBuffer);
|
|
||||||
if (pBuffer == NULL)
|
|
||||||
return SetLastError(ERROR_NOT_ENOUGH_MEMORY), FALSE;
|
|
||||||
|
|
||||||
Status = pZwQuerySystemInformation(
|
|
||||||
SystemProcessesAndThreadsInformation,
|
|
||||||
pBuffer, cbBuffer, NULL);
|
|
||||||
|
|
||||||
if (Status == STATUS_INFO_LENGTH_MISMATCH)
|
|
||||||
{
|
|
||||||
HeapFree(hHeap, 0, pBuffer);
|
|
||||||
cbBuffer *= 2;
|
|
||||||
}
|
|
||||||
else if (!NT_SUCCESS(Status))
|
|
||||||
{
|
|
||||||
HeapFree(hHeap, 0, pBuffer);
|
|
||||||
return SetLastError(Status), NULL;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
while (Status == STATUS_INFO_LENGTH_MISMATCH);
|
|
||||||
|
|
||||||
// call the helper
|
|
||||||
dwError = KillProcessTreeNtHelper(
|
|
||||||
(PSYSTEM_PROCESS_INFORMATION)pBuffer,
|
|
||||||
dwProcessId);
|
|
||||||
|
|
||||||
HeapFree(hHeap, 0, pBuffer);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
// call the helper
|
|
||||||
dwError = KillProcessTreeWinHelper(dwProcessId);
|
|
||||||
}
|
|
||||||
|
|
||||||
SetLastError(dwError);
|
|
||||||
return dwError == ERROR_SUCCESS;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Heloer function for process killing
|
|
||||||
|
|
||||||
static BOOL KillProcess(
|
|
||||||
IN DWORD dwProcessId
|
|
||||||
)
|
|
||||||
{
|
|
||||||
// get process handle
|
|
||||||
HANDLE hProcess = OpenProcess(PROCESS_TERMINATE, FALSE, dwProcessId);
|
|
||||||
if (hProcess == NULL)
|
|
||||||
return FALSE;
|
|
||||||
|
|
||||||
DWORD dwError = ERROR_SUCCESS;
|
|
||||||
|
|
||||||
// try to terminate the process
|
|
||||||
if (!TerminateProcess(hProcess, (DWORD)-1))
|
|
||||||
dwError = GetLastError();
|
|
||||||
|
|
||||||
// close process handle
|
|
||||||
CloseHandle(hProcess);
|
|
||||||
|
|
||||||
SetLastError(dwError);
|
|
||||||
#ifdef DEBUG_MONITOR
|
|
||||||
if(dwError != ERROR_SUCCESS) {
|
|
||||||
_stprintf(buffer, _T("Process %i killed"), dwProcessId);
|
|
||||||
OutputDebugStringW(buffer);
|
|
||||||
} else {
|
|
||||||
_stprintf(buffer, _T("Failed to kill process %i"), dwProcessId);
|
|
||||||
OutputDebugStringW(buffer);
|
|
||||||
DisplayMessage();
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
return dwError == ERROR_SUCCESS;
|
|
||||||
}
|
|
||||||
|
|
||||||
// a helper function that walks a process tree recursively
|
|
||||||
// on Windows NT and terminates all processes in the tree
|
|
||||||
static BOOL KillProcessTreeNtHelper(
|
|
||||||
IN PSYSTEM_PROCESS_INFORMATION pInfo,
|
|
||||||
IN DWORD dwProcessId
|
|
||||||
)
|
|
||||||
{
|
|
||||||
#ifdef DEBUG_MONITOR
|
|
||||||
_TCHAR buffer[MAX_CMD_LINE_LENGTH];
|
|
||||||
#endif
|
|
||||||
if(pInfo == NULL) {
|
|
||||||
#ifdef DEBUG_MONITOR
|
|
||||||
_stprintf(buffer, _T("KillProcessTreeNtHelper: wrong parameter"));
|
|
||||||
OutputDebugStringW(buffer);
|
|
||||||
#endif
|
|
||||||
return FALSE;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// terminate all children first
|
|
||||||
for (;;)
|
|
||||||
{
|
|
||||||
if (pInfo->InheritedFromProcessId == dwProcessId)
|
|
||||||
KillProcessTreeNtHelper(pInfo, pInfo->ProcessId);
|
|
||||||
|
|
||||||
if (pInfo->NextEntryDelta == 0)
|
|
||||||
break;
|
|
||||||
|
|
||||||
// find address of the next structure
|
|
||||||
pInfo = (PSYSTEM_PROCESS_INFORMATION)(((PUCHAR)pInfo)
|
|
||||||
+ pInfo->NextEntryDelta);
|
|
||||||
}
|
|
||||||
|
|
||||||
// terminate the specified process
|
|
||||||
if (!KillProcess(dwProcessId))
|
|
||||||
return GetLastError();
|
|
||||||
|
|
||||||
return ERROR_SUCCESS;
|
|
||||||
}
|
|
||||||
|
|
||||||
// a helper function that walks a process tree recursively
|
|
||||||
// on Windows 9x and terminates all processes in the tree
|
|
||||||
static BOOL KillProcessTreeWinHelper(
|
|
||||||
IN DWORD dwProcessId
|
|
||||||
)
|
|
||||||
{
|
|
||||||
#ifdef DEBUG_MONITOR
|
|
||||||
_TCHAR buffer[MAX_CMD_LINE_LENGTH];
|
|
||||||
#endif
|
|
||||||
HINSTANCE hKernel;
|
|
||||||
HANDLE (WINAPI * pCreateToolhelp32Snapshot)(DWORD, DWORD);
|
|
||||||
BOOL (WINAPI * pProcess32First)(HANDLE, PROCESSENTRY32 *);
|
|
||||||
BOOL (WINAPI * pProcess32Next)(HANDLE, PROCESSENTRY32 *);
|
|
||||||
|
|
||||||
// get KERNEL32.DLL handle
|
|
||||||
hKernel = GetModuleHandleW(_T("kernel32.dll"));
|
|
||||||
if(hKernel == NULL) {
|
|
||||||
#ifdef DEBUG_MONITOR
|
|
||||||
_stprintf(buffer, _T("KillProcessTreeNtHelper: wrong parameter"));
|
|
||||||
OutputDebugStringW(buffer);
|
|
||||||
#endif
|
|
||||||
return FALSE;
|
|
||||||
}
|
|
||||||
|
|
||||||
// find necessary entrypoints in KERNEL32.DLL
|
|
||||||
*(FARPROC *)&pCreateToolhelp32Snapshot =
|
|
||||||
GetProcAddress(hKernel, "CreateToolhelp32Snapshot");
|
|
||||||
*(FARPROC *)&pProcess32First =
|
|
||||||
GetProcAddress(hKernel, "Process32First");
|
|
||||||
*(FARPROC *)&pProcess32Next =
|
|
||||||
GetProcAddress(hKernel, "Process32Next");
|
|
||||||
|
|
||||||
if (pCreateToolhelp32Snapshot == NULL ||
|
|
||||||
pProcess32First == NULL ||
|
|
||||||
pProcess32Next == NULL)
|
|
||||||
return ERROR_PROC_NOT_FOUND;
|
|
||||||
|
|
||||||
HANDLE hSnapshot;
|
|
||||||
PROCESSENTRY32 Entry;
|
|
||||||
|
|
||||||
// create a snapshot of all processes
|
|
||||||
hSnapshot = pCreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
|
|
||||||
if (hSnapshot == INVALID_HANDLE_VALUE)
|
|
||||||
return GetLastError();
|
|
||||||
|
|
||||||
Entry.dwSize = sizeof(Entry);
|
|
||||||
if (!pProcess32First(hSnapshot, &Entry))
|
|
||||||
{
|
|
||||||
DWORD dwError = GetLastError();
|
|
||||||
CloseHandle(hSnapshot);
|
|
||||||
return dwError;
|
|
||||||
}
|
|
||||||
|
|
||||||
// terminate children first
|
|
||||||
do
|
|
||||||
{
|
|
||||||
if (Entry.th32ParentProcessID == dwProcessId)
|
|
||||||
KillProcessTreeWinHelper(Entry.th32ProcessID);
|
|
||||||
|
|
||||||
Entry.dwSize = sizeof(Entry);
|
|
||||||
}
|
|
||||||
while (pProcess32Next(hSnapshot, &Entry));
|
|
||||||
|
|
||||||
CloseHandle(hSnapshot);
|
|
||||||
|
|
||||||
// terminate the specified process
|
|
||||||
if (!KillProcess(dwProcessId))
|
|
||||||
return GetLastError();
|
|
||||||
|
|
||||||
return ERROR_SUCCESS;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
@ -1,122 +0,0 @@
|
||||||
/*******************************************************************************
|
|
||||||
* Copyright (c) 2002 - 2005 QNX Software Systems and others.
|
|
||||||
* All rights reserved. This program and the accompanying materials
|
|
||||||
* are made available under the terms of the Eclipse Public License v1.0
|
|
||||||
* which accompanies this distribution, and is available at
|
|
||||||
* http://www.eclipse.org/legal/epl-v10.html
|
|
||||||
*
|
|
||||||
* Contributors:
|
|
||||||
* QNX Software Systems - initial API and implementation
|
|
||||||
*
|
|
||||||
* StdAfx.h
|
|
||||||
*
|
|
||||||
* This is a header file for helper function for the process killing
|
|
||||||
* Implementation based on the article "Terminating Windows Processes"
|
|
||||||
* see http://www.alexfedotov.com/articles/killproc.asp and
|
|
||||||
* http://www.alexfedotov.com/samples/threads.asp
|
|
||||||
***********************************************************************/
|
|
||||||
|
|
||||||
#if !defined(_KILLER_H_INCLUDED_)
|
|
||||||
#define _KILLER_H_INCLUDED_
|
|
||||||
|
|
||||||
#if _MSC_VER > 1000
|
|
||||||
#pragma once
|
|
||||||
#endif // _MSC_VER > 1000
|
|
||||||
|
|
||||||
#include <Ntsecapi.h>
|
|
||||||
|
|
||||||
|
|
||||||
typedef LONG KPRIORITY; // From ntddk.h
|
|
||||||
|
|
||||||
//
|
|
||||||
// Process Virtual Memory Counters
|
|
||||||
// NtQueryInformationProcess using ProcessVmCounters
|
|
||||||
// From ntddk.h
|
|
||||||
|
|
||||||
typedef struct _VM_COUNTERS {
|
|
||||||
SIZE_T PeakVirtualSize;
|
|
||||||
SIZE_T VirtualSize;
|
|
||||||
ULONG PageFaultCount;
|
|
||||||
SIZE_T PeakWorkingSetSize;
|
|
||||||
SIZE_T WorkingSetSize;
|
|
||||||
SIZE_T QuotaPeakPagedPoolUsage;
|
|
||||||
SIZE_T QuotaPagedPoolUsage;
|
|
||||||
SIZE_T QuotaPeakNonPagedPoolUsage;
|
|
||||||
SIZE_T QuotaNonPagedPoolUsage;
|
|
||||||
SIZE_T PagefileUsage;
|
|
||||||
SIZE_T PeakPagefileUsage;
|
|
||||||
} VM_COUNTERS;
|
|
||||||
typedef VM_COUNTERS *PVM_COUNTERS;
|
|
||||||
|
|
||||||
//
|
|
||||||
// ClientId
|
|
||||||
//
|
|
||||||
|
|
||||||
typedef struct _CLIENT_ID {
|
|
||||||
HANDLE UniqueProcess;
|
|
||||||
HANDLE UniqueThread;
|
|
||||||
} CLIENT_ID;
|
|
||||||
typedef CLIENT_ID *PCLIENT_ID;
|
|
||||||
|
|
||||||
typedef struct _SYSTEM_THREAD_INFORMATION {
|
|
||||||
LARGE_INTEGER KernelTime; // time spent in kernel mode
|
|
||||||
LARGE_INTEGER UserTime; // time spent in user mode
|
|
||||||
LARGE_INTEGER CreateTime; // thread creation time
|
|
||||||
ULONG WaitTime; // wait time
|
|
||||||
PVOID StartAddress; // start address
|
|
||||||
CLIENT_ID ClientId; // thread and process IDs
|
|
||||||
KPRIORITY Priority; // dynamic priority
|
|
||||||
KPRIORITY BasePriority; // base priority
|
|
||||||
ULONG ContextSwitchCount; // number of context switches
|
|
||||||
LONG State; // current state
|
|
||||||
LONG WaitReason; // wait reason
|
|
||||||
} SYSTEM_THREAD_INFORMATION, * PSYSTEM_THREAD_INFORMATION;
|
|
||||||
|
|
||||||
typedef struct _SYSTEM_PROCESS_INFORMATION {
|
|
||||||
ULONG NextEntryDelta; // offset to the next entry
|
|
||||||
ULONG ThreadCount; // number of threads
|
|
||||||
ULONG Reserved1[6]; // reserved
|
|
||||||
LARGE_INTEGER CreateTime; // process creation time
|
|
||||||
LARGE_INTEGER UserTime; // time spent in user mode
|
|
||||||
LARGE_INTEGER KernelTime; // time spent in kernel mode
|
|
||||||
UNICODE_STRING ProcessName; // process name
|
|
||||||
KPRIORITY BasePriority; // base process priority
|
|
||||||
ULONG ProcessId; // process identifier
|
|
||||||
ULONG InheritedFromProcessId; // parent process identifier
|
|
||||||
ULONG HandleCount; // number of handles
|
|
||||||
ULONG Reserved2[2]; // reserved
|
|
||||||
VM_COUNTERS VmCounters; // virtual memory counters
|
|
||||||
#if _WIN32_WINNT >= 0x500
|
|
||||||
IO_COUNTERS IoCounters; // i/o counters
|
|
||||||
#endif
|
|
||||||
SYSTEM_THREAD_INFORMATION Threads[1]; // threads
|
|
||||||
} SYSTEM_PROCESS_INFORMATION, * PSYSTEM_PROCESS_INFORMATION;
|
|
||||||
|
|
||||||
|
|
||||||
static BOOL KillProcessTreeNtHelper(
|
|
||||||
IN PSYSTEM_PROCESS_INFORMATION pInfo,
|
|
||||||
IN DWORD dwProcessId);
|
|
||||||
|
|
||||||
static BOOL KillProcessTreeWinHelper(
|
|
||||||
IN DWORD dwProcessId);
|
|
||||||
|
|
||||||
// From ntstatus.h
|
|
||||||
// MessageId: STATUS_INFO_LENGTH_MISMATCH
|
|
||||||
//
|
|
||||||
// MessageText:
|
|
||||||
//
|
|
||||||
// The specified information record length does not match the length required for the specified information class.
|
|
||||||
//
|
|
||||||
#define STATUS_INFO_LENGTH_MISMATCH ((NTSTATUS)0xC0000004L)
|
|
||||||
|
|
||||||
// From ntstatus.h
|
|
||||||
// Generic test for success on any status value (non-negative numbers
|
|
||||||
// indicate success).
|
|
||||||
//
|
|
||||||
|
|
||||||
#define NT_SUCCESS(Status) ((NTSTATUS)(Status) >= 0)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#endif // _KILLER_H_INCLUDED_
|
|
||||||
|
|
@ -29,7 +29,7 @@
|
||||||
int copyTo(_TCHAR * target, const _TCHAR * source, int cpyLength, int availSpace);
|
int copyTo(_TCHAR * target, const _TCHAR * source, int cpyLength, int availSpace);
|
||||||
void DisplayErrorMessage();
|
void DisplayErrorMessage();
|
||||||
|
|
||||||
BOOL KillProcessEx(DWORD dwProcessId); // Handle of the process
|
//BOOL KillProcessEx(DWORD dwProcessId); // Handle of the process
|
||||||
|
|
||||||
///////////////////////////////////////////////////////////////////////////////
|
///////////////////////////////////////////////////////////////////////////////
|
||||||
BOOL WINAPI HandlerRoutine( DWORD dwCtrlType) // control signal type
|
BOOL WINAPI HandlerRoutine( DWORD dwCtrlType) // control signal type
|
||||||
|
|
@ -214,7 +214,7 @@ extern "C" int _tmain(int argc, _TCHAR * argv[]) {
|
||||||
OutputDebugStringW(buffer);
|
OutputDebugStringW(buffer);
|
||||||
#endif
|
#endif
|
||||||
// Create job object if it is possible
|
// Create job object if it is possible
|
||||||
HMODULE hKernel = GetModuleHandle("kernel32.dll");
|
HMODULE hKernel = GetModuleHandle(L"kernel32.dll");
|
||||||
HANDLE hJob = NULL;
|
HANDLE hJob = NULL;
|
||||||
HANDLE (WINAPI * pCreateJobObject)(LPSECURITY_ATTRIBUTES lpJobAttributes,
|
HANDLE (WINAPI * pCreateJobObject)(LPSECURITY_ATTRIBUTES lpJobAttributes,
|
||||||
char * lpName);
|
char * lpName);
|
||||||
|
|
@ -297,12 +297,6 @@ extern "C" int _tmain(int argc, _TCHAR * argv[]) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else
|
} else
|
||||||
if(!KillProcessEx(pi.dwProcessId)) {
|
|
||||||
#ifdef DEBUG_MONITOR
|
|
||||||
_stprintf(buffer, _T("Cannot kill process (PID %i) tree\n"), pi.dwProcessId);
|
|
||||||
OutputDebugStringW(buffer);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
exitProc = TRUE;
|
exitProc = TRUE;
|
||||||
break;
|
break;
|
||||||
default:
|
default:
|
||||||
|
|
|
||||||
Binary file not shown.
Loading…
Add table
Reference in a new issue