1*b1cdbd2cSJim Jagielski /**************************************************************
2*b1cdbd2cSJim Jagielski *
3*b1cdbd2cSJim Jagielski * Licensed to the Apache Software Foundation (ASF) under one
4*b1cdbd2cSJim Jagielski * or more contributor license agreements. See the NOTICE file
5*b1cdbd2cSJim Jagielski * distributed with this work for additional information
6*b1cdbd2cSJim Jagielski * regarding copyright ownership. The ASF licenses this file
7*b1cdbd2cSJim Jagielski * to you under the Apache License, Version 2.0 (the
8*b1cdbd2cSJim Jagielski * "License"); you may not use this file except in compliance
9*b1cdbd2cSJim Jagielski * with the License. You may obtain a copy of the License at
10*b1cdbd2cSJim Jagielski *
11*b1cdbd2cSJim Jagielski * http://www.apache.org/licenses/LICENSE-2.0
12*b1cdbd2cSJim Jagielski *
13*b1cdbd2cSJim Jagielski * Unless required by applicable law or agreed to in writing,
14*b1cdbd2cSJim Jagielski * software distributed under the License is distributed on an
15*b1cdbd2cSJim Jagielski * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
16*b1cdbd2cSJim Jagielski * KIND, either express or implied. See the License for the
17*b1cdbd2cSJim Jagielski * specific language governing permissions and limitations
18*b1cdbd2cSJim Jagielski * under the License.
19*b1cdbd2cSJim Jagielski *
20*b1cdbd2cSJim Jagielski *************************************************************/
21*b1cdbd2cSJim Jagielski
22*b1cdbd2cSJim Jagielski
23*b1cdbd2cSJim Jagielski
24*b1cdbd2cSJim Jagielski // MARKER(update_precomp.py): autogen include statement, do not remove
25*b1cdbd2cSJim Jagielski #include "precompiled_sal.hxx"
26*b1cdbd2cSJim Jagielski
27*b1cdbd2cSJim Jagielski #include <tchar.h>
28*b1cdbd2cSJim Jagielski
29*b1cdbd2cSJim Jagielski #ifdef _MSC_VER
30*b1cdbd2cSJim Jagielski #pragma warning(push,1) // disable warnings within system headers
31*b1cdbd2cSJim Jagielski #endif
32*b1cdbd2cSJim Jagielski #define WIN32_LEAN_AND_MEAN
33*b1cdbd2cSJim Jagielski #include <windows.h>
34*b1cdbd2cSJim Jagielski #include <tlhelp32.h>
35*b1cdbd2cSJim Jagielski #include <psapi.h>
36*b1cdbd2cSJim Jagielski #ifdef _MSC_VER
37*b1cdbd2cSJim Jagielski #pragma warning(pop)
38*b1cdbd2cSJim Jagielski #endif
39*b1cdbd2cSJim Jagielski
40*b1cdbd2cSJim Jagielski #include <signal.h>
41*b1cdbd2cSJim Jagielski #include <stdarg.h>
42*b1cdbd2cSJim Jagielski #include <stdlib.h>
43*b1cdbd2cSJim Jagielski #include <stdio.h>
44*b1cdbd2cSJim Jagielski
45*b1cdbd2cSJim Jagielski #ifndef SIGNULL
46*b1cdbd2cSJim Jagielski #define SIGNULL 0
47*b1cdbd2cSJim Jagielski #endif
48*b1cdbd2cSJim Jagielski
49*b1cdbd2cSJim Jagielski #ifndef SIGKILL
50*b1cdbd2cSJim Jagielski #define SIGKILL 9
51*b1cdbd2cSJim Jagielski #endif
52*b1cdbd2cSJim Jagielski
53*b1cdbd2cSJim Jagielski #include <signal.h>
54*b1cdbd2cSJim Jagielski
55*b1cdbd2cSJim Jagielski #define MAX_MODULES 1024
56*b1cdbd2cSJim Jagielski
57*b1cdbd2cSJim Jagielski /////////////////////////////////////////////////////////////////////////////
58*b1cdbd2cSJim Jagielski // Determines if a returned handle value is valid
59*b1cdbd2cSJim Jagielski /////////////////////////////////////////////////////////////////////////////
60*b1cdbd2cSJim Jagielski
IsValidHandle(HANDLE handle)61*b1cdbd2cSJim Jagielski static inline bool IsValidHandle( HANDLE handle )
62*b1cdbd2cSJim Jagielski {
63*b1cdbd2cSJim Jagielski return INVALID_HANDLE_VALUE != handle && NULL != handle;
64*b1cdbd2cSJim Jagielski }
65*b1cdbd2cSJim Jagielski
66*b1cdbd2cSJim Jagielski
67*b1cdbd2cSJim Jagielski #define elementsof( a ) (sizeof(a) / sizeof( (a)[0] ))
68*b1cdbd2cSJim Jagielski
69*b1cdbd2cSJim Jagielski /////////////////////////////////////////////////////////////////////////////
70*b1cdbd2cSJim Jagielski // Retrieves function adress in another process
71*b1cdbd2cSJim Jagielski /////////////////////////////////////////////////////////////////////////////
72*b1cdbd2cSJim Jagielski
73*b1cdbd2cSJim Jagielski #if 1
74*b1cdbd2cSJim Jagielski #define GetProcAddressEx( hProcess, hModule, lpProcName ) GetProcAddress( hModule, lpProcName )
75*b1cdbd2cSJim Jagielski #else
GetProcAddressEx(HANDLE hProcess,HMODULE hModule,LPCSTR lpProcName)76*b1cdbd2cSJim Jagielski FARPROC WINAPI GetProcAddressEx( HANDLE hProcess, HMODULE hModule, LPCSTR lpProcName )
77*b1cdbd2cSJim Jagielski {
78*b1cdbd2cSJim Jagielski FARPROC lpfnProcAddress = GetProcAddress( hModule, lpProcName );
79*b1cdbd2cSJim Jagielski
80*b1cdbd2cSJim Jagielski if ( lpfnProcAddress )
81*b1cdbd2cSJim Jagielski {
82*b1cdbd2cSJim Jagielski DWORD dwProcessId = GetProcessId( hProcess );
83*b1cdbd2cSJim Jagielski
84*b1cdbd2cSJim Jagielski if ( GetCurrentProcessId() != dwProcessId )
85*b1cdbd2cSJim Jagielski {
86*b1cdbd2cSJim Jagielski FARPROC lpfnRemoteProcAddress = NULL;
87*b1cdbd2cSJim Jagielski TCHAR szBaseName[MAX_PATH];
88*b1cdbd2cSJim Jagielski
89*b1cdbd2cSJim Jagielski if ( GetModuleBaseName( GetCurrentProcess(), hModule, szBaseName, elementsof(szBaseName) ) )
90*b1cdbd2cSJim Jagielski {
91*b1cdbd2cSJim Jagielski HMODULE ahModules[MAX_MODULES];
92*b1cdbd2cSJim Jagielski DWORD cbNeeded = 0;
93*b1cdbd2cSJim Jagielski
94*b1cdbd2cSJim Jagielski if ( EnumProcessModules( hProcess, ahModules, sizeof(ahModules), &cbNeeded ) )
95*b1cdbd2cSJim Jagielski {
96*b1cdbd2cSJim Jagielski ULONG nModules = cbNeeded / sizeof(ahModules[0]);
97*b1cdbd2cSJim Jagielski
98*b1cdbd2cSJim Jagielski for ( ULONG n = 0; n < nModules; n++ )
99*b1cdbd2cSJim Jagielski {
100*b1cdbd2cSJim Jagielski TCHAR szRemoteBaseName[MAX_PATH];
101*b1cdbd2cSJim Jagielski
102*b1cdbd2cSJim Jagielski if ( GetModuleBaseName(
103*b1cdbd2cSJim Jagielski hProcess, ahModules[n], szRemoteBaseName, elementsof(szRemoteBaseName) ) &&
104*b1cdbd2cSJim Jagielski 0 == lstrcmpi( szRemoteBaseName, szBaseName )
105*b1cdbd2cSJim Jagielski )
106*b1cdbd2cSJim Jagielski {
107*b1cdbd2cSJim Jagielski lpfnRemoteProcAddress = lpfnProcAddress;
108*b1cdbd2cSJim Jagielski
109*b1cdbd2cSJim Jagielski if ( ahModules[n] != hModule )
110*b1cdbd2cSJim Jagielski *(LPBYTE*)&lpfnRemoteProcAddress += (LPBYTE)ahModules[n] - (LPBYTE)hModule;
111*b1cdbd2cSJim Jagielski break;
112*b1cdbd2cSJim Jagielski }
113*b1cdbd2cSJim Jagielski }
114*b1cdbd2cSJim Jagielski }
115*b1cdbd2cSJim Jagielski }
116*b1cdbd2cSJim Jagielski
117*b1cdbd2cSJim Jagielski lpfnProcAddress = lpfnRemoteProcAddress;
118*b1cdbd2cSJim Jagielski }
119*b1cdbd2cSJim Jagielski }
120*b1cdbd2cSJim Jagielski
121*b1cdbd2cSJim Jagielski return lpfnProcAddress;
122*b1cdbd2cSJim Jagielski }
123*b1cdbd2cSJim Jagielski #endif
124*b1cdbd2cSJim Jagielski
125*b1cdbd2cSJim Jagielski /////////////////////////////////////////////////////////////////////////////
126*b1cdbd2cSJim Jagielski // Raises a signal in an other process
127*b1cdbd2cSJim Jagielski /////////////////////////////////////////////////////////////////////////////
128*b1cdbd2cSJim Jagielski
SignalToExceptionCode(int signal)129*b1cdbd2cSJim Jagielski static DWORD SignalToExceptionCode( int signal )
130*b1cdbd2cSJim Jagielski {
131*b1cdbd2cSJim Jagielski switch ( signal )
132*b1cdbd2cSJim Jagielski {
133*b1cdbd2cSJim Jagielski case SIGSEGV:
134*b1cdbd2cSJim Jagielski return EXCEPTION_ACCESS_VIOLATION;
135*b1cdbd2cSJim Jagielski case SIGFPE:
136*b1cdbd2cSJim Jagielski return EXCEPTION_FLT_INVALID_OPERATION;
137*b1cdbd2cSJim Jagielski case SIGILL:
138*b1cdbd2cSJim Jagielski return EXCEPTION_ILLEGAL_INSTRUCTION;
139*b1cdbd2cSJim Jagielski case SIGINT:
140*b1cdbd2cSJim Jagielski return CONTROL_C_EXIT;
141*b1cdbd2cSJim Jagielski case SIGBREAK:
142*b1cdbd2cSJim Jagielski return CONTROL_C_EXIT;
143*b1cdbd2cSJim Jagielski default:
144*b1cdbd2cSJim Jagielski return 0;
145*b1cdbd2cSJim Jagielski }
146*b1cdbd2cSJim Jagielski }
147*b1cdbd2cSJim Jagielski
RaiseSignalEx(HANDLE hProcess,int sig)148*b1cdbd2cSJim Jagielski static BOOL RaiseSignalEx( HANDLE hProcess, int sig )
149*b1cdbd2cSJim Jagielski {
150*b1cdbd2cSJim Jagielski DWORD dwProcessId = GetProcessId( hProcess );
151*b1cdbd2cSJim Jagielski
152*b1cdbd2cSJim Jagielski HANDLE hSnapshot = CreateToolhelp32Snapshot( TH32CS_SNAPTHREAD, 0 );
153*b1cdbd2cSJim Jagielski HANDLE hThread = 0;
154*b1cdbd2cSJim Jagielski BOOL fSuccess = FALSE;
155*b1cdbd2cSJim Jagielski
156*b1cdbd2cSJim Jagielski if ( IsValidHandle(hSnapshot) )
157*b1cdbd2cSJim Jagielski {
158*b1cdbd2cSJim Jagielski THREADENTRY32 te;
159*b1cdbd2cSJim Jagielski
160*b1cdbd2cSJim Jagielski te.dwSize = sizeof(te);
161*b1cdbd2cSJim Jagielski fSuccess = Thread32First( hSnapshot, &te );
162*b1cdbd2cSJim Jagielski while ( fSuccess )
163*b1cdbd2cSJim Jagielski {
164*b1cdbd2cSJim Jagielski if ( te.th32OwnerProcessID == dwProcessId )
165*b1cdbd2cSJim Jagielski {
166*b1cdbd2cSJim Jagielski hThread = OpenThread(
167*b1cdbd2cSJim Jagielski THREAD_SUSPEND_RESUME | THREAD_QUERY_INFORMATION |
168*b1cdbd2cSJim Jagielski THREAD_GET_CONTEXT | THREAD_SET_CONTEXT,
169*b1cdbd2cSJim Jagielski FALSE, te.th32ThreadID );
170*b1cdbd2cSJim Jagielski if ( IsValidHandle(hThread) )
171*b1cdbd2cSJim Jagielski break;
172*b1cdbd2cSJim Jagielski }
173*b1cdbd2cSJim Jagielski
174*b1cdbd2cSJim Jagielski fSuccess = Thread32Next( hSnapshot, &te );
175*b1cdbd2cSJim Jagielski }
176*b1cdbd2cSJim Jagielski
177*b1cdbd2cSJim Jagielski CloseHandle( hSnapshot );
178*b1cdbd2cSJim Jagielski }
179*b1cdbd2cSJim Jagielski
180*b1cdbd2cSJim Jagielski if ( fSuccess )
181*b1cdbd2cSJim Jagielski {
182*b1cdbd2cSJim Jagielski CONTEXT aContext;
183*b1cdbd2cSJim Jagielski
184*b1cdbd2cSJim Jagielski if ( SuspendThread( hThread ) != (DWORD)-1 )
185*b1cdbd2cSJim Jagielski {
186*b1cdbd2cSJim Jagielski ZeroMemory( &aContext, sizeof(aContext) );
187*b1cdbd2cSJim Jagielski aContext.ContextFlags = CONTEXT_FULL;
188*b1cdbd2cSJim Jagielski
189*b1cdbd2cSJim Jagielski fSuccess = GetThreadContext( hThread, &aContext );
190*b1cdbd2cSJim Jagielski
191*b1cdbd2cSJim Jagielski if ( fSuccess )
192*b1cdbd2cSJim Jagielski {
193*b1cdbd2cSJim Jagielski if ( sig )
194*b1cdbd2cSJim Jagielski {
195*b1cdbd2cSJim Jagielski DWORD dwStackBuffer[] =
196*b1cdbd2cSJim Jagielski {
197*b1cdbd2cSJim Jagielski aContext.Eip,
198*b1cdbd2cSJim Jagielski SignalToExceptionCode( sig ),
199*b1cdbd2cSJim Jagielski EXCEPTION_NONCONTINUABLE,
200*b1cdbd2cSJim Jagielski 0,
201*b1cdbd2cSJim Jagielski 0
202*b1cdbd2cSJim Jagielski };
203*b1cdbd2cSJim Jagielski
204*b1cdbd2cSJim Jagielski aContext.Esp -= sizeof(dwStackBuffer);
205*b1cdbd2cSJim Jagielski WriteProcessMemory( hProcess, (LPVOID)aContext.Esp, dwStackBuffer, sizeof(dwStackBuffer), NULL );
206*b1cdbd2cSJim Jagielski aContext.Eip = (DWORD)GetProcAddressEx( hProcess, GetModuleHandleA("KERNEL32"), "RaiseException" );
207*b1cdbd2cSJim Jagielski }
208*b1cdbd2cSJim Jagielski else
209*b1cdbd2cSJim Jagielski {
210*b1cdbd2cSJim Jagielski aContext.Ecx = aContext.Eax = aContext.Ebx = aContext.Edx = aContext.Esi = aContext.Edi = 0;
211*b1cdbd2cSJim Jagielski }
212*b1cdbd2cSJim Jagielski
213*b1cdbd2cSJim Jagielski fSuccess = SetThreadContext( hThread, &aContext );
214*b1cdbd2cSJim Jagielski }
215*b1cdbd2cSJim Jagielski
216*b1cdbd2cSJim Jagielski fSuccess = ResumeThread( hThread ) && fSuccess;
217*b1cdbd2cSJim Jagielski
218*b1cdbd2cSJim Jagielski DWORD dwLastError = GetLastError();
219*b1cdbd2cSJim Jagielski CloseHandle( hThread );
220*b1cdbd2cSJim Jagielski SetLastError( dwLastError );
221*b1cdbd2cSJim Jagielski
222*b1cdbd2cSJim Jagielski return fSuccess;
223*b1cdbd2cSJim Jagielski }
224*b1cdbd2cSJim Jagielski }
225*b1cdbd2cSJim Jagielski
226*b1cdbd2cSJim Jagielski return FALSE;
227*b1cdbd2cSJim Jagielski }
228*b1cdbd2cSJim Jagielski /////////////////////////////////////////////////////////////////////////////
229*b1cdbd2cSJim Jagielski // Command line parameter parsing
230*b1cdbd2cSJim Jagielski /////////////////////////////////////////////////////////////////////////////
231*b1cdbd2cSJim Jagielski
ParseCommandArgs(LPDWORD lpProcesses,LPDWORD lpdwNumProcesses,int * pSig)232*b1cdbd2cSJim Jagielski static void ParseCommandArgs( LPDWORD lpProcesses, LPDWORD lpdwNumProcesses, int *pSig )
233*b1cdbd2cSJim Jagielski {
234*b1cdbd2cSJim Jagielski typedef struct _SignalEntry
235*b1cdbd2cSJim Jagielski {
236*b1cdbd2cSJim Jagielski LPCTSTR lpSignalName;
237*b1cdbd2cSJim Jagielski int iSignalValue;
238*b1cdbd2cSJim Jagielski } SignalEntry;
239*b1cdbd2cSJim Jagielski
240*b1cdbd2cSJim Jagielski #define SIG_ENTRY( signal ) { TEXT(#signal), SIG##signal }
241*b1cdbd2cSJim Jagielski
242*b1cdbd2cSJim Jagielski static SignalEntry SupportedSignals[] =
243*b1cdbd2cSJim Jagielski {
244*b1cdbd2cSJim Jagielski SIG_ENTRY( NULL ),
245*b1cdbd2cSJim Jagielski SIG_ENTRY( SEGV ),
246*b1cdbd2cSJim Jagielski SIG_ENTRY( ILL ),
247*b1cdbd2cSJim Jagielski SIG_ENTRY( FPE ),
248*b1cdbd2cSJim Jagielski SIG_ENTRY( INT ),
249*b1cdbd2cSJim Jagielski SIG_ENTRY( BREAK ),
250*b1cdbd2cSJim Jagielski SIG_ENTRY( TERM ),
251*b1cdbd2cSJim Jagielski SIG_ENTRY( ABRT ),
252*b1cdbd2cSJim Jagielski SIG_ENTRY( KILL )
253*b1cdbd2cSJim Jagielski };
254*b1cdbd2cSJim Jagielski
255*b1cdbd2cSJim Jagielski const int NumSupportedSignals = elementsof(SupportedSignals);
256*b1cdbd2cSJim Jagielski
257*b1cdbd2cSJim Jagielski DWORD dwMaxProcesses = *lpdwNumProcesses;
258*b1cdbd2cSJim Jagielski int argc = __argc;
259*b1cdbd2cSJim Jagielski TCHAR **argv = __targv;
260*b1cdbd2cSJim Jagielski
261*b1cdbd2cSJim Jagielski *lpdwNumProcesses = 0;
262*b1cdbd2cSJim Jagielski
263*b1cdbd2cSJim Jagielski for ( int argn = 1; argn < argc; argn++ )
264*b1cdbd2cSJim Jagielski {
265*b1cdbd2cSJim Jagielski if ( 0 == lstrcmpi( argv[argn], TEXT("-l") ) ||
266*b1cdbd2cSJim Jagielski 0 == lstrcmpi( argv[argn], TEXT("/l") ) )
267*b1cdbd2cSJim Jagielski
268*b1cdbd2cSJim Jagielski {
269*b1cdbd2cSJim Jagielski for ( int n = 0; n < NumSupportedSignals; n++ )
270*b1cdbd2cSJim Jagielski {
271*b1cdbd2cSJim Jagielski _tprintf( _T("%s "), SupportedSignals[n].lpSignalName );
272*b1cdbd2cSJim Jagielski }
273*b1cdbd2cSJim Jagielski _tprintf( _T("\n") );
274*b1cdbd2cSJim Jagielski ExitProcess( 0 );
275*b1cdbd2cSJim Jagielski }
276*b1cdbd2cSJim Jagielski else if ( 0 == lstrcmpi( argv[argn], TEXT("-?") ) ||
277*b1cdbd2cSJim Jagielski 0 == lstrcmpi( argv[argn], TEXT("/?") ) ||
278*b1cdbd2cSJim Jagielski 0 == lstrcmpi( argv[argn], TEXT("-h") ) ||
279*b1cdbd2cSJim Jagielski 0 == lstrcmpi( argv[argn], TEXT("/h") ) ||
280*b1cdbd2cSJim Jagielski 0 == lstrcmpi( argv[argn], TEXT("--help") ) )
281*b1cdbd2cSJim Jagielski {
282*b1cdbd2cSJim Jagielski _tprintf(
283*b1cdbd2cSJim Jagielski _T("Terminates a process by sending a signal.\n\n")
284*b1cdbd2cSJim Jagielski _T("Usage: kill [ -l ] [ -signal ] pid ...\n\n")
285*b1cdbd2cSJim Jagielski _T("-l Lists supported signals\n")
286*b1cdbd2cSJim Jagielski _T("-signal Sends the specified signal to the given processes.\n")
287*b1cdbd2cSJim Jagielski _T(" signal can be a numeric value specifying the signal number\n")
288*b1cdbd2cSJim Jagielski _T(" or a string listed by the -l parameter. If no signal is\n")
289*b1cdbd2cSJim Jagielski _T(" given SIGTERM (-TERM) is used.\n")
290*b1cdbd2cSJim Jagielski _T("pid Process id(s) or executables names(s) of processes to \n")
291*b1cdbd2cSJim Jagielski _T(" signal or terminate.\n\n")
292*b1cdbd2cSJim Jagielski );
293*b1cdbd2cSJim Jagielski ExitProcess( 0 );
294*b1cdbd2cSJim Jagielski }
295*b1cdbd2cSJim Jagielski else if ( argv[argn] && ( *argv[argn] == '-' || *argv[argn] == '/' ) )
296*b1cdbd2cSJim Jagielski {
297*b1cdbd2cSJim Jagielski LPCTSTR argsig = CharNext( argv[argn] );
298*b1cdbd2cSJim Jagielski
299*b1cdbd2cSJim Jagielski int n;
300*b1cdbd2cSJim Jagielski for ( n = 0; n < NumSupportedSignals; n++ )
301*b1cdbd2cSJim Jagielski {
302*b1cdbd2cSJim Jagielski _TCHAR *endptr = NULL;
303*b1cdbd2cSJim Jagielski
304*b1cdbd2cSJim Jagielski if ( 0 == lstrcmpi( SupportedSignals[n].lpSignalName, argsig ) ||
305*b1cdbd2cSJim Jagielski _tcstoul( argsig, &endptr, 0 ) == static_cast< unsigned >(SupportedSignals[n].iSignalValue) && (!endptr || !*endptr) )
306*b1cdbd2cSJim Jagielski {
307*b1cdbd2cSJim Jagielski *pSig = SupportedSignals[n].iSignalValue;
308*b1cdbd2cSJim Jagielski break;
309*b1cdbd2cSJim Jagielski }
310*b1cdbd2cSJim Jagielski }
311*b1cdbd2cSJim Jagielski
312*b1cdbd2cSJim Jagielski if ( n >= NumSupportedSignals )
313*b1cdbd2cSJim Jagielski {
314*b1cdbd2cSJim Jagielski _ftprintf( stderr,
315*b1cdbd2cSJim Jagielski _T("kill: Illegal argument %s\n")
316*b1cdbd2cSJim Jagielski _T("Type 'kill --help' to show allowed syntax.\n")
317*b1cdbd2cSJim Jagielski _T("Type 'kill -l' to show supported signals.\n"),
318*b1cdbd2cSJim Jagielski argv[argn] );
319*b1cdbd2cSJim Jagielski ExitProcess( 0 );
320*b1cdbd2cSJim Jagielski }
321*b1cdbd2cSJim Jagielski }
322*b1cdbd2cSJim Jagielski else
323*b1cdbd2cSJim Jagielski {
324*b1cdbd2cSJim Jagielski unsigned long value = 0;
325*b1cdbd2cSJim Jagielski _TCHAR *endptr = NULL;
326*b1cdbd2cSJim Jagielski
327*b1cdbd2cSJim Jagielski value = _tcstoul( argv[argn], &endptr, 0 );
328*b1cdbd2cSJim Jagielski
329*b1cdbd2cSJim Jagielski if ( !endptr || !*endptr )
330*b1cdbd2cSJim Jagielski {
331*b1cdbd2cSJim Jagielski if ( *lpdwNumProcesses < dwMaxProcesses )
332*b1cdbd2cSJim Jagielski {
333*b1cdbd2cSJim Jagielski *(lpProcesses++) = value;
334*b1cdbd2cSJim Jagielski (*lpdwNumProcesses)++;
335*b1cdbd2cSJim Jagielski }
336*b1cdbd2cSJim Jagielski }
337*b1cdbd2cSJim Jagielski else
338*b1cdbd2cSJim Jagielski {
339*b1cdbd2cSJim Jagielski HANDLE hSnapshot = CreateToolhelp32Snapshot( TH32CS_SNAPPROCESS, 0 );
340*b1cdbd2cSJim Jagielski
341*b1cdbd2cSJim Jagielski if ( IsValidHandle( hSnapshot ) )
342*b1cdbd2cSJim Jagielski {
343*b1cdbd2cSJim Jagielski PROCESSENTRY32 pe;
344*b1cdbd2cSJim Jagielski
345*b1cdbd2cSJim Jagielski pe.dwSize = sizeof(pe);
346*b1cdbd2cSJim Jagielski BOOL fSuccess = Process32First( hSnapshot, &pe );
347*b1cdbd2cSJim Jagielski
348*b1cdbd2cSJim Jagielski while ( fSuccess )
349*b1cdbd2cSJim Jagielski {
350*b1cdbd2cSJim Jagielski if ( 0 == lstrcmpi( argv[argn], pe.szExeFile ) )
351*b1cdbd2cSJim Jagielski {
352*b1cdbd2cSJim Jagielski if ( *lpdwNumProcesses < dwMaxProcesses )
353*b1cdbd2cSJim Jagielski {
354*b1cdbd2cSJim Jagielski *(lpProcesses++) = pe.th32ProcessID;
355*b1cdbd2cSJim Jagielski (*lpdwNumProcesses)++;
356*b1cdbd2cSJim Jagielski }
357*b1cdbd2cSJim Jagielski }
358*b1cdbd2cSJim Jagielski fSuccess = Process32Next( hSnapshot, &pe );
359*b1cdbd2cSJim Jagielski }
360*b1cdbd2cSJim Jagielski
361*b1cdbd2cSJim Jagielski CloseHandle( hSnapshot );
362*b1cdbd2cSJim Jagielski }
363*b1cdbd2cSJim Jagielski }
364*b1cdbd2cSJim Jagielski }
365*b1cdbd2cSJim Jagielski }
366*b1cdbd2cSJim Jagielski
367*b1cdbd2cSJim Jagielski if ( !*lpdwNumProcesses )
368*b1cdbd2cSJim Jagielski {
369*b1cdbd2cSJim Jagielski _ftprintf( stderr,
370*b1cdbd2cSJim Jagielski _T("kill: No process specified.\n")
371*b1cdbd2cSJim Jagielski _T("Use kill --help to show allowed syntax.\n")
372*b1cdbd2cSJim Jagielski );
373*b1cdbd2cSJim Jagielski ExitProcess( 0 );
374*b1cdbd2cSJim Jagielski }
375*b1cdbd2cSJim Jagielski
376*b1cdbd2cSJim Jagielski }
377*b1cdbd2cSJim Jagielski
OutputSystemMessage(DWORD dwErrorCode)378*b1cdbd2cSJim Jagielski void OutputSystemMessage( DWORD dwErrorCode )
379*b1cdbd2cSJim Jagielski {
380*b1cdbd2cSJim Jagielski LPVOID lpMsgBuf;
381*b1cdbd2cSJim Jagielski FormatMessageA(
382*b1cdbd2cSJim Jagielski FORMAT_MESSAGE_ALLOCATE_BUFFER |
383*b1cdbd2cSJim Jagielski FORMAT_MESSAGE_FROM_SYSTEM |
384*b1cdbd2cSJim Jagielski FORMAT_MESSAGE_IGNORE_INSERTS,
385*b1cdbd2cSJim Jagielski NULL,
386*b1cdbd2cSJim Jagielski dwErrorCode,
387*b1cdbd2cSJim Jagielski MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
388*b1cdbd2cSJim Jagielski (LPSTR)&lpMsgBuf,
389*b1cdbd2cSJim Jagielski 0,
390*b1cdbd2cSJim Jagielski NULL
391*b1cdbd2cSJim Jagielski );
392*b1cdbd2cSJim Jagielski
393*b1cdbd2cSJim Jagielski printf( (LPSTR)lpMsgBuf );
394*b1cdbd2cSJim Jagielski LocalFree( lpMsgBuf );
395*b1cdbd2cSJim Jagielski }
396*b1cdbd2cSJim Jagielski
_tmain()397*b1cdbd2cSJim Jagielski int _tmain()
398*b1cdbd2cSJim Jagielski {
399*b1cdbd2cSJim Jagielski DWORD dwProcessIds[1024];
400*b1cdbd2cSJim Jagielski DWORD nProcesses = elementsof(dwProcessIds);
401*b1cdbd2cSJim Jagielski int sig = SIGTERM;
402*b1cdbd2cSJim Jagielski
403*b1cdbd2cSJim Jagielski
404*b1cdbd2cSJim Jagielski ParseCommandArgs( dwProcessIds, &nProcesses, &sig );
405*b1cdbd2cSJim Jagielski
406*b1cdbd2cSJim Jagielski for ( ULONG n = 0; n < nProcesses; n++ )
407*b1cdbd2cSJim Jagielski {
408*b1cdbd2cSJim Jagielski HANDLE hProcess;
409*b1cdbd2cSJim Jagielski
410*b1cdbd2cSJim Jagielski _tprintf( _T("Sending signal to process id %d..."), dwProcessIds[n] );
411*b1cdbd2cSJim Jagielski hProcess = OpenProcess( PROCESS_TERMINATE | PROCESS_CREATE_THREAD | SYNCHRONIZE |
412*b1cdbd2cSJim Jagielski PROCESS_QUERY_INFORMATION | PROCESS_VM_OPERATION | PROCESS_VM_WRITE | PROCESS_VM_READ,
413*b1cdbd2cSJim Jagielski FALSE, dwProcessIds[n] );
414*b1cdbd2cSJim Jagielski
415*b1cdbd2cSJim Jagielski if ( IsValidHandle( hProcess ) )
416*b1cdbd2cSJim Jagielski {
417*b1cdbd2cSJim Jagielski if ( SIGKILL == sig )
418*b1cdbd2cSJim Jagielski TerminateProcess( hProcess, 255 );
419*b1cdbd2cSJim Jagielski else
420*b1cdbd2cSJim Jagielski {
421*b1cdbd2cSJim Jagielski if ( RaiseSignalEx( hProcess, sig ) )
422*b1cdbd2cSJim Jagielski _tprintf( _T("OK\n") );
423*b1cdbd2cSJim Jagielski else
424*b1cdbd2cSJim Jagielski {
425*b1cdbd2cSJim Jagielski OutputSystemMessage( GetLastError() );
426*b1cdbd2cSJim Jagielski }
427*b1cdbd2cSJim Jagielski }
428*b1cdbd2cSJim Jagielski
429*b1cdbd2cSJim Jagielski CloseHandle( hProcess );
430*b1cdbd2cSJim Jagielski }
431*b1cdbd2cSJim Jagielski else
432*b1cdbd2cSJim Jagielski {
433*b1cdbd2cSJim Jagielski OutputSystemMessage( GetLastError() );
434*b1cdbd2cSJim Jagielski }
435*b1cdbd2cSJim Jagielski }
436*b1cdbd2cSJim Jagielski
437*b1cdbd2cSJim Jagielski return 0;
438*b1cdbd2cSJim Jagielski }
439*b1cdbd2cSJim Jagielski
440