taskkill.c File Reference

#include <stdlib.h>
#include <windows.h>
#include <psapi.h>
#include <wine/debug.h>
#include <wine/unicode.h>
#include "taskkill.h"

Include dependency graph for taskkill.c:

Go to the source code of this file.

Classes
struct	pid_close_info

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (taskkill)
static int	taskkill_vprintfW (const WCHAR *msg, __ms_va_list va_args)
static int WINAPIV	taskkill_printfW (const WCHAR *msg,...)
static int WINAPIV	taskkill_message_printfW (int msg,...)
static int	taskkill_message (int msg)
static BOOL CALLBACK	pid_enum_proc (HWND hwnd, LPARAM lParam)
static DWORD *	enumerate_processes (DWORD *list_count)
static BOOL	get_process_name_from_pid (DWORD pid, WCHAR *buf, DWORD chars)
static int	send_close_messages (void)
static int	terminate_processes (void)
static BOOL	add_to_task_list (WCHAR *name)
static BOOL	process_arguments (int argc, WCHAR *argv[])
int	wmain (int argc, WCHAR *argv[])

Variables
static BOOL	force_termination = FALSE
static WCHAR **	task_list
static unsigned int	task_count

Function Documentation

add_to_task_list()

static BOOL add_to_task_list ( WCHAR *  name )

static

Definition at line 488 of file taskkill.c.

{
    static unsigned int list_size = 16;
 
    if (!task_list)
    {
        task_list = HeapAlloc(GetProcessHeap(), 0,
                                   list_size * sizeof(*task_list));
        if (!task_list)
            return FALSE;
    }
    else if (task_count == list_size)
    {
        void *realloc_list;
 
        list_size *= 2;
        realloc_list = HeapReAlloc(GetProcessHeap(), 0, task_list,
                                   list_size * sizeof(*task_list));
        if (!realloc_list)
            return FALSE;
 
        task_list = realloc_list;
    }
 
    task_list[task_count++] = name;
    return TRUE;
}

Referenced by process_arguments().

enumerate_processes()

static DWORD * enumerate_processes ( DWORD *  list_count )

static

Definition at line 152 of file taskkill.c.

{
    DWORD *pid_list, alloc_bytes = 1024 * sizeof(*pid_list), needed_bytes;
 
    pid_list = HeapAlloc(GetProcessHeap(), 0, alloc_bytes);
    if (!pid_list)
        return NULL;
 
    for (;;)
    {
        DWORD *realloc_list;
 
        if (!EnumProcesses(pid_list, alloc_bytes, &needed_bytes))
        {
            HeapFree(GetProcessHeap(), 0, pid_list);
            return NULL;
        }
 
        /* EnumProcesses can't signal an insufficient buffer condition, so the
         * only way to possibly determine whether a larger buffer is required
         * is to see whether the written number of bytes is the same as the
         * buffer size. If so, the buffer will be reallocated to twice the
         * size. */
        if (alloc_bytes != needed_bytes)
            break;
 
        alloc_bytes *= 2;
        realloc_list = HeapReAlloc(GetProcessHeap(), 0, pid_list, alloc_bytes);
        if (!realloc_list)
        {
            HeapFree(GetProcessHeap(), 0, pid_list);
            return NULL;
        }
        pid_list = realloc_list;
    }
 
    *list_count = needed_bytes / sizeof(*pid_list);
    return pid_list;
}

Referenced by send_close_messages(), and terminate_processes().

get_process_name_from_pid()

static BOOL get_process_name_from_pid ( DWORD  pid, WCHAR *  buf, DWORD  chars  )

static

Definition at line 192 of file taskkill.c.

{
    HANDLE process;
    HMODULE module;
    DWORD required_size;
 
    process = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, pid);
    if (!process)
        return FALSE;
 
    if (!EnumProcessModules(process, &module, sizeof(module), &required_size))
    {
        CloseHandle(process);
        return FALSE;
    }
 
    if (!GetModuleBaseNameW(process, module, buf, chars))
    {
        CloseHandle(process);
        return FALSE;
    }
 
    CloseHandle(process);
    return TRUE;
}

Referenced by send_close_messages(), and terminate_processes().

pid_enum_proc()

static BOOL CALLBACK pid_enum_proc ( HWND  hwnd, LPARAM  lParam  )

static

Definition at line 136 of file taskkill.c.

{
    struct pid_close_info *info = (struct pid_close_info *)lParam;
    DWORD hwnd_pid;
 
    GetWindowThreadProcessId(hwnd, &hwnd_pid);
 
    if (hwnd_pid == info->pid)
    {
        PostMessageW(hwnd, WM_CLOSE, 0, 0);
        info->found = TRUE;
    }
 
    return TRUE;
}

Referenced by send_close_messages(), and terminate_processes().

process_arguments()

static BOOL process_arguments ( int  argc, WCHAR *  argv[]  )

static

Definition at line 659 of file taskkill.c.

{
    static const WCHAR opForceTerminate[] = {'f',0};
    static const WCHAR opImage[] = {'i','m',0};
    static const WCHAR opPID[] = {'p','i','d',0};
    static const WCHAR opHelp[] = {'?',0};
    static const WCHAR opTerminateChildren[] = {'t',0};
 
    if (argc > 1)
    {
        int i;
        WCHAR *argdata;
        BOOL has_im = FALSE, has_pid = FALSE;
 
        /* Only the lone help option is recognized. */
        if (argc == 2)
        {
            argdata = argv[1];
            if ((*argdata == '/' || *argdata == '-') && !strcmpW(opHelp, argdata + 1))
            {
                taskkill_message(STRING_USAGE);
                exit(0);
            }
        }
 
        for (i = 1; i < argc; i++)
        {
            BOOL got_im = FALSE, got_pid = FALSE;
 
            argdata = argv[i];
            if (*argdata != '/' && *argdata != '-')
                goto invalid;
            argdata++;
 
            if (!strcmpiW(opTerminateChildren, argdata))
                WINE_FIXME("argument T not supported\n");
            if (!strcmpiW(opForceTerminate, argdata))
                force_termination = TRUE;
            /* Options /IM and /PID appear to behave identically, except for
             * the fact that they cannot be specified at the same time. */
            else if ((got_im = !strcmpiW(opImage, argdata)) ||
                     (got_pid = !strcmpiW(opPID, argdata)))
            {
                if (!argv[i + 1])
                {
                    taskkill_message_printfW(STRING_MISSING_PARAM, argv[i]);
                    taskkill_message(STRING_USAGE);
                    return FALSE;
                }
 
                if (got_im) has_im = TRUE;
                if (got_pid) has_pid = TRUE;
 
                if (has_im && has_pid)
                {
                    taskkill_message(STRING_MUTUAL_EXCLUSIVE);
                    taskkill_message(STRING_USAGE);
                    return FALSE;
                }
 
                if (!add_to_task_list(argv[i + 1]))
                    return FALSE;
                i++;
            }
            else
            {
                invalid:
                taskkill_message(STRING_INVALID_OPTION);
                taskkill_message(STRING_USAGE);
                return FALSE;
            }
        }
    }
    else
    {
        taskkill_message(STRING_MISSING_OPTION);
        taskkill_message(STRING_USAGE);
        return FALSE;
    }
 
    return TRUE;
}

Referenced by wmain().

send_close_messages()

static int send_close_messages ( void  )

static

Definition at line 235 of file taskkill.c.

{
    DWORD *pid_list, pid_list_size;
    DWORD self_pid = GetCurrentProcessId();
    unsigned int i;
    int status_code = 0;
 
    pid_list = enumerate_processes(&pid_list_size);
    if (!pid_list)
    {
        taskkill_message(STRING_ENUM_FAILED);
        return 1;
    }
 
    for (i = 0; i < task_count; i++)
    {
        WCHAR *p = task_list[i];
        BOOL is_numeric = TRUE;
 
        /* Determine whether the string is not numeric. */
        while (*p)
        {
            if (!isdigitW(*p++))
            {
                is_numeric = FALSE;
                break;
            }
        }
 
        if (is_numeric)
        {
            DWORD pid = atoiW(task_list[i]);
            struct pid_close_info info = { pid };
 
            if (pid == self_pid)
            {
                taskkill_message(STRING_SELF_TERMINATION);
                status_code = 1;
                continue;
            }
 
            EnumWindows(pid_enum_proc, (LPARAM)&info);
            if (info.found)
                taskkill_message_printfW(STRING_CLOSE_PID_SEARCH, pid);
            else
            {
                taskkill_message_printfW(STRING_SEARCH_FAILED, task_list[i]);
                status_code = 128;
            }
        }
        else
        {
            DWORD index;
            BOOL found_process = FALSE;
 
            for (index = 0; index < pid_list_size; index++)
            {
                WCHAR process_name[MAX_PATH];
 
                if (get_process_name_from_pid(pid_list[index], process_name, MAX_PATH) &&
                    !strcmpiW(process_name, task_list[i]))
                {
                    struct pid_close_info info = { pid_list[index] };
 
                    found_process = TRUE;
                    if (pid_list[index] == self_pid)
                    {
                        taskkill_message(STRING_SELF_TERMINATION);
                        status_code = 1;
                        continue;
                    }
 
                    EnumWindows(pid_enum_proc, (LPARAM)&info);
                    taskkill_message_printfW(STRING_CLOSE_PROC_SRCH, process_name, pid_list[index]);
                }
            }
 
            if (!found_process)
            {
                taskkill_message_printfW(STRING_SEARCH_FAILED, task_list[i]);
                status_code = 128;
            }
        }
    }
 
    HeapFree(GetProcessHeap(), 0, pid_list);
    return status_code;
}

Referenced by wmain().

taskkill_message()

static int taskkill_message ( int  msg )

static

Definition at line 125 of file taskkill.c.

{
    static const WCHAR formatW[] = {'%','1',0};
    WCHAR msg_buffer[8192];
 
    LoadStringW(GetModuleHandleW(NULL), msg, msg_buffer, ARRAY_SIZE(msg_buffer));
 
    return taskkill_printfW(formatW, msg_buffer);
}

Referenced by process_arguments(), send_close_messages(), and terminate_processes().

taskkill_message_printfW()

static int WINAPIV taskkill_message_printfW ( int  msg,   ...  )

static

Definition at line 110 of file taskkill.c.

{
    __ms_va_list va_args;
    WCHAR msg_buffer[8192];
    int len;
 
    LoadStringW(GetModuleHandleW(NULL), msg, msg_buffer, ARRAY_SIZE(msg_buffer));
 
    __ms_va_start(va_args, msg);
    len = taskkill_vprintfW(msg_buffer, va_args);
    __ms_va_end(va_args);
 
    return len;
}

Referenced by process_arguments(), send_close_messages(), and terminate_processes().

taskkill_printfW()

static int WINAPIV taskkill_printfW ( const WCHAR *  msg,   ...  )

static

Definition at line 98 of file taskkill.c.

{
    __ms_va_list va_args;
    int len;
 
    __ms_va_start(va_args, msg);
    len = taskkill_vprintfW(msg, va_args);
    __ms_va_end(va_args);
 
    return len;
}

Referenced by taskkill_message().

taskkill_vprintfW()

static int taskkill_vprintfW ( const WCHAR *  msg, __ms_va_list  va_args  )

static

Definition at line 64 of file taskkill.c.

{
    int wlen;
    DWORD count, ret;
    WCHAR msg_buffer[8192];
 
    wlen = FormatMessageW(FORMAT_MESSAGE_FROM_STRING, msg, 0, 0, msg_buffer,
                          ARRAY_SIZE(msg_buffer), &va_args);
 
    ret = WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE), msg_buffer, wlen, &count, NULL);
    if (!ret)
    {
        DWORD len;
        char *msgA;
 
        /* On Windows WriteConsoleW() fails if the output is redirected. So fall
         * back to WriteFile(), assuming the console encoding is still the right
         * one in that case.
         */
        len = WideCharToMultiByte(GetConsoleOutputCP(), 0, msg_buffer, wlen,
            NULL, 0, NULL, NULL);
        msgA = HeapAlloc(GetProcessHeap(), 0, len);
        if (!msgA)
            return 0;
 
        WideCharToMultiByte(GetConsoleOutputCP(), 0, msg_buffer, wlen, msgA, len,
            NULL, NULL);
        WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), msgA, len, &count, FALSE);
        HeapFree(GetProcessHeap(), 0, msgA);
    }
 
    return count;
}

Referenced by taskkill_message_printfW(), and taskkill_printfW().

terminate_processes()

static int terminate_processes ( void  )

static

Definition at line 329 of file taskkill.c.

{
    DWORD *pid_list, pid_list_size;
    DWORD self_pid = GetCurrentProcessId();
    unsigned int i;
    int status_code = 0;
 
    pid_list = enumerate_processes(&pid_list_size);
    if (!pid_list)
    {
        taskkill_message(STRING_ENUM_FAILED);
        return 1;
    }
 
    for (i = 0; i < task_count; i++)
    {
        WCHAR *p = task_list[i];
        BOOL is_numeric = TRUE;
 
        /* Determine whether the string is not numeric. */
        while (*p)
        {
            if (!isdigitW(*p++))
            {
                is_numeric = FALSE;
                break;
            }
        }
 
        if (is_numeric)
        {
            DWORD pid = atoiW(task_list[i]);
#ifndef __REACTOS__
            HANDLE process;
#endif
 
            if (pid == self_pid)
            {
                taskkill_message(STRING_SELF_TERMINATION);
                status_code = 1;
                continue;
            }
 
#ifdef __REACTOS__
            if (force_termination)
            {
            HANDLE process;
#endif
            process = OpenProcess(PROCESS_TERMINATE, FALSE, pid);
            if (!process)
            {
                taskkill_message_printfW(STRING_SEARCH_FAILED, task_list[i]);
                status_code = 128;
                continue;
            }
 
            if (!TerminateProcess(process, 0))
            {
                taskkill_message_printfW(STRING_TERMINATE_FAILED, task_list[i]);
                status_code = 1;
                CloseHandle(process);
                continue;
            }
 
            taskkill_message_printfW(STRING_TERM_PID_SEARCH, pid);
            CloseHandle(process);
#ifdef __REACTOS__
            }
            else
            {
                struct pid_close_info info = { pid };
 
                EnumWindows(pid_enum_proc, (LPARAM)&info);
                if (info.found)
                    taskkill_message_printfW(STRING_CLOSE_PID_SEARCH, pid);
                else
                {
                    taskkill_message_printfW(STRING_SEARCH_FAILED, task_list[i]);
                    status_code = 128;
                }
            }
#endif
        }
        else
        {
            DWORD index;
            BOOL found_process = FALSE;
 
            for (index = 0; index < pid_list_size; index++)
            {
                WCHAR process_name[MAX_PATH];
 
                if (get_process_name_from_pid(pid_list[index], process_name, MAX_PATH) &&
                    !strcmpiW(process_name, task_list[i]))
                {
#ifdef __REACTOS__
                    found_process = TRUE;
#else
                    HANDLE process;
#endif
 
                    if (pid_list[index] == self_pid)
                    {
                        taskkill_message(STRING_SELF_TERMINATION);
                        status_code = 1;
                        continue;
                    }
 
#ifdef __REACTOS__
                    if (force_termination)
                    {
                    HANDLE process;
#endif
                    process = OpenProcess(PROCESS_TERMINATE, FALSE, pid_list[index]);
                    if (!process)
                    {
                        taskkill_message_printfW(STRING_SEARCH_FAILED, task_list[i]);
                        status_code = 128;
                        continue;
                    }
 
                    if (!TerminateProcess(process, 0))
                    {
                        taskkill_message_printfW(STRING_TERMINATE_FAILED, task_list[i]);
                        status_code = 1;
                        CloseHandle(process);
                        continue;
                    }
 
#ifndef __REACTOS__
                    found_process = TRUE;
#endif
                    taskkill_message_printfW(STRING_TERM_PROC_SEARCH, task_list[i], pid_list[index]);
                    CloseHandle(process);
#ifdef __REACTOS__
                    }
                    else
                    {
                        struct pid_close_info info = { pid_list[index] };
                        EnumWindows(pid_enum_proc, (LPARAM)&info);
                        taskkill_message_printfW(STRING_CLOSE_PROC_SRCH, process_name, pid_list[index]);
                    }
#endif
                }
            }
 
            if (!found_process)
            {
                taskkill_message_printfW(STRING_SEARCH_FAILED, task_list[i]);
                status_code = 128;
            }
        }
    }
 
    HeapFree(GetProcessHeap(), 0, pid_list);
    return status_code;
}

Referenced by wmain().

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

taskkill

)

wmain()

int wmain	(	int	argc,
		WCHAR *	argv[]
	)

Definition at line 744 of file taskkill.c.

{
    int status_code = 0;
 
    if (!process_arguments(argc, argv))
    {
        HeapFree(GetProcessHeap(), 0, task_list);
        return 1;
    }
 
#ifdef __REACTOS__
    status_code = terminate_processes(force_termination);
#else
    if (force_termination)
        status_code = terminate_processes();
    else
        status_code = send_close_messages();
#endif
 
    HeapFree(GetProcessHeap(), 0, task_list);
    return status_code;
}

Variable Documentation

force_termination

BOOL force_termination = FALSE

static

Definition at line 33 of file taskkill.c.

Referenced by process_arguments(), terminate_processes(), and wmain().

task_count

unsigned int task_count

static

Definition at line 36 of file taskkill.c.

Referenced by add_to_task_list(), send_close_messages(), and terminate_processes().

task_list

WCHAR** task_list

static

Definition at line 35 of file taskkill.c.

Referenced by add_to_task_list(), send_close_messages(), terminate_processes(), test_multiselect(), and wmain().