taskkill.c File Reference

#include <stdlib.h>
#include <windows.h>
#include <tlhelp32.h>
#include <wine/debug.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, 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 BOOL	enumerate_processes (void)
static void	mark_task_process (const WCHAR *str, int *status_code)
static void	taskkill_message_print_process (int msg, unsigned int index)
static BOOL	find_parent (unsigned int process_index, unsigned int *parent_index)
static void	mark_child_processes (void)
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 BOOL	kill_child_processes = FALSE
static WCHAR **	task_list
static unsigned int	task_count
struct {
PROCESSENTRY32W   p
BOOL   matched
BOOL   is_numeric
} *	process_list
static unsigned int	process_count

Function Documentation

add_to_task_list()

static BOOL add_to_task_list ( WCHAR *  name )

static

Definition at line 574 of file taskkill.c.

{
    static unsigned int list_size = 16;
 
    if (!task_list)
    {
        task_list = malloc(list_size * sizeof(*task_list));
        if (!task_list)
            return FALSE;
    }
    else if (task_count == list_size)
    {
        void *realloc_list;
 
        list_size *= 2;
        realloc_list = realloc(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 BOOL enumerate_processes ( void  )

static

Definition at line 162 of file taskkill.c.

{
    unsigned int alloc_count = 128;
    void *realloc_list;
    HANDLE snapshot;
 
    snapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
    if (snapshot == INVALID_HANDLE_VALUE)
        return FALSE;
 
    process_list = malloc(alloc_count * sizeof(*process_list));
    if (!process_list)
        return FALSE;
 
    process_list[0].p.dwSize = sizeof(process_list[0].p);
    if (!Process32FirstW(snapshot, &process_list[0].p))
        return FALSE;
 
    do
    {
        process_list[process_count].is_numeric = FALSE;
        process_list[process_count++].matched = FALSE;
        if (process_count == alloc_count)
        {
            alloc_count *= 2;
            realloc_list = realloc(process_list, alloc_count * sizeof(*process_list));
            if (!realloc_list)
                return FALSE;
            process_list = realloc_list;
        }
        process_list[process_count].p.dwSize = sizeof(process_list[process_count].p);
    } while (Process32NextW(snapshot, &process_list[process_count].p));
    CloseHandle(snapshot);
    return TRUE;
}

Referenced by is_netmeeting_running(), and wmain().

find_parent()

static BOOL find_parent ( unsigned int  process_index, unsigned int *  parent_index  )

static

Definition at line 310 of file taskkill.c.

{
    DWORD parent_id = process_list[process_index].p.th32ParentProcessID;
    unsigned int i;
 
    if (!parent_id)
        return FALSE;
 
    for (i = 0; i < process_count; ++i)
    {
        if (process_list[i].p.th32ProcessID == parent_id)
        {
            *parent_index = i;
            return TRUE;
        }
    }
    return FALSE;
}

Referenced by mark_child_processes().

mark_child_processes()

static void mark_child_processes ( void  )

static

Definition at line 329 of file taskkill.c.

{
    unsigned int i, parent;
 
    for (i = 0; i < process_count; ++i)
    {
        if (process_list[i].matched)
            continue;
#ifdef __REACTOS__
        // Prevent self-termination if we are in the process tree
        if (process_list[i].p.th32ProcessID == self_pid)
            continue;
#endif
        parent = i;
        while (find_parent(parent, &parent))
        {
            if (process_list[parent].matched)
            {
                WINE_TRACE("Adding child %04lx.\n", process_list[i].p.th32ProcessID);
                process_list[i].matched = TRUE;
#ifdef __REACTOS__
                pkill_list[pkill_size++] = i;
#endif
                break;
            }
        }
    }
}

Referenced by wmain().

mark_task_process()

static void mark_task_process ( const WCHAR *  str, int *  status_code  )

static

Definition at line 198 of file taskkill.c.

{
#ifndef __REACTOS__
    DWORD self_pid = GetCurrentProcessId();
#endif
    const WCHAR *p = str;
    BOOL is_numeric;
    unsigned int i;
    DWORD pid;
 
    is_numeric = TRUE;
    while (*p)
    {
        if (!iswdigit(*p++))
        {
            is_numeric = FALSE;
            break;
        }
    }
 
    if (is_numeric)
    {
        pid = wcstol(str, NULL, 10);
        for (i = 0; i < process_count; ++i)
        {
            if (process_list[i].p.th32ProcessID == pid)
                break;
        }
        if (i == process_count || process_list[i].matched)
            goto not_found;
        process_list[i].matched = TRUE;
        process_list[i].is_numeric = TRUE;
        if (pid == self_pid)
        {
            taskkill_message(STRING_SELF_TERMINATION);
            *status_code = 1;
        }
#ifdef __REACTOS__
        else
        {
            pkill_list[pkill_size++] = i;
        }
#endif
        return;
    }
 
    for (i = 0; i < process_count; ++i)
    {
        if (!wcsicmp(process_list[i].p.szExeFile, str) && !process_list[i].matched)
        {
            process_list[i].matched = TRUE;
            if (process_list[i].p.th32ProcessID == self_pid)
            {
                taskkill_message(STRING_SELF_TERMINATION);
                *status_code = 1;
            }
#ifdef __REACTOS__
            else
            {
                pkill_list[pkill_size++] = i;
                continue;
            }
#endif
            return;
        }
    }
 
#ifdef __REACTOS__
    // Cannot find any process matching the PID or name
    if (pkill_size == 0)
    {
#endif
not_found:
    taskkill_message_printfW(STRING_SEARCH_FAILED, str);
    *status_code = 128;
#ifdef __REACTOS__
    }
#endif
}

Referenced by wmain().

pid_enum_proc()

static BOOL CALLBACK pid_enum_proc ( HWND  hwnd, LPARAM  lParam  )

static

Definition at line 146 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().

process_arguments()

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

static

Definition at line 746 of file taskkill.c.

{
    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 == '-') && !lstrcmpW(L"?", 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 (!wcsicmp(L"t", argdata))
                kill_child_processes = TRUE;
            else if (!wcsicmp(L"f", 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 = !wcsicmp(L"im", argdata)) ||
                     (got_pid = !wcsicmp(L"pid", 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 477 of file taskkill.c.

{
    const WCHAR *process_name;
    struct pid_close_info info;
    unsigned int i;
    int status_code = 0;
 
#ifdef __REACTOS__
    DWORD index;
    for (index = 0; index < pkill_size; ++index)
#else
    for (i = 0; i < process_count; i++)
#endif
    {
#ifdef __REACTOS__
        i = pkill_list[index];
#else
        if (!process_list[i].matched)
            continue;
#endif
 
        info.pid = process_list[i].p.th32ProcessID;
        process_name = process_list[i].p.szExeFile;
        info.found = FALSE;
        WINE_TRACE("Terminating pid %04lx.\n", info.pid);
        EnumWindows(pid_enum_proc, (LPARAM)&info);
        if (info.found)
        {
            if (kill_child_processes)
                taskkill_message_printfW(STRING_CLOSE_CHILD, info.pid, process_list[i].p.th32ParentProcessID);
            else if (process_list[i].is_numeric)
                taskkill_message_printfW(STRING_CLOSE_PID_SEARCH, info.pid);
            else
                taskkill_message_printfW(STRING_CLOSE_PROC_SRCH, process_name, info.pid);
            continue;
        }
        taskkill_message_print_process(STRING_SEARCH_FAILED, i);
        status_code = 128;
    }
 
    return status_code;
}

Referenced by wmain().

taskkill_message()

static int taskkill_message ( int  msg )

static

Definition at line 136 of file taskkill.c.

{
    WCHAR msg_buffer[8192];
 
    LoadStringW(GetModuleHandleW(NULL), msg, msg_buffer, ARRAY_SIZE(msg_buffer));
 
    return taskkill_printfW(L"%1", msg_buffer);
}

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

taskkill_message_print_process()

static void taskkill_message_print_process ( int  msg, unsigned int  index  )

static

Definition at line 278 of file taskkill.c.

{
    WCHAR pid_str[16];
 
    if (!process_list[index].is_numeric)
    {
        taskkill_message_printfW(msg, process_list[index].p.szExeFile);
        return;
    }
    wsprintfW(pid_str, L"%lu", process_list[index].p.th32ProcessID);
    taskkill_message_printfW(msg, pid_str);
}

Referenced by send_close_messages(), and terminate_processes().

taskkill_message_printfW()

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

static

Definition at line 121 of file taskkill.c.

{
    va_list va_args;
    WCHAR msg_buffer[8192];
    int len;
 
    LoadStringW(GetModuleHandleW(NULL), msg, msg_buffer, ARRAY_SIZE(msg_buffer));
 
    va_start(va_args, msg);
    len = taskkill_vprintfW(msg_buffer, va_args);
    va_end(va_args);
 
    return len;
}

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

taskkill_printfW()

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

static

Definition at line 109 of file taskkill.c.

{
    va_list va_args;
    int len;
 
    va_start(va_args, msg);
    len = taskkill_vprintfW(msg, va_args);
    va_end(va_args);
 
    return len;
}

Referenced by taskkill_message().

taskkill_vprintfW()

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

static

Definition at line 78 of file taskkill.c.

{
    int wlen;
    DWORD count;
    WCHAR msg_buffer[8192];
 
    wlen = FormatMessageW(FORMAT_MESSAGE_FROM_STRING, msg, 0, 0, msg_buffer,
                          ARRAY_SIZE(msg_buffer), &va_args);
 
    if (!WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE), msg_buffer, wlen, &count, NULL))
    {
        DWORD len;
        char *msgA;
 
        /* On Windows WriteConsoleW() fails if the output is redirected. So fall
         * back to WriteFile() using OEM code page.
         */
        len = WideCharToMultiByte(GetOEMCP(), 0, msg_buffer, wlen,
            NULL, 0, NULL, NULL);
        msgA = malloc(len);
        if (!msgA)
            return 0;
 
        WideCharToMultiByte(GetOEMCP(), 0, msg_buffer, wlen, msgA, len, NULL, NULL);
        WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), msgA, len, &count, FALSE);
        free(msgA);
    }
 
    return count;
}

Referenced by taskkill_message_printfW(), and taskkill_printfW().

terminate_processes()

static int terminate_processes ( void  )

static

Definition at line 520 of file taskkill.c.

{
    const WCHAR *process_name;
    unsigned int i;
    int status_code = 0;
    HANDLE process;
    DWORD pid;
 
#ifdef __REACTOS__
    DWORD index;
    for (index = 0; index < pkill_size; ++index)
#else
    for (i = 0; i < process_count; i++)
#endif
    {
#ifdef __REACTOS__
        i = pkill_list[index];
#else
        if (!process_list[i].matched)
            continue;
#endif
 
        pid = process_list[i].p.th32ProcessID;
        process_name = process_list[i].p.szExeFile;
        process = OpenProcess(PROCESS_TERMINATE, FALSE, pid);
        if (!process)
        {
            taskkill_message_print_process(STRING_SEARCH_FAILED, i);
            status_code = 128;
            continue;
        }
        if (!TerminateProcess(process, 1))
        {
#ifdef __REACTOS__
            if (kill_child_processes)
                taskkill_message_printfW(STRING_TERM_CHILD_FAILED, pid, process_list[i].p.th32ParentProcessID);
            else
#endif
            taskkill_message_print_process(STRING_TERMINATE_FAILED, i);
            status_code = 1;
            CloseHandle(process);
            continue;
        }
        if (kill_child_processes)
            taskkill_message_printfW(STRING_TERM_CHILD, pid, process_list[i].p.th32ParentProcessID);
        else if (process_list[i].is_numeric)
            taskkill_message_printfW(STRING_TERM_PID_SEARCH, pid);
        else
            taskkill_message_printfW(STRING_TERM_PROC_SEARCH, process_name, pid);
        CloseHandle(process);
    }
    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 825 of file taskkill.c.

{
    int search_status = 0, terminate_status;
    unsigned int i;
 
    if (!process_arguments(argc, argv))
        return 1;
 
    if (!enumerate_processes())
    {
        taskkill_message(STRING_ENUM_FAILED);
        return 1;
    }
 
#ifdef __REACTOS__
    pkill_list = malloc(process_count * sizeof(unsigned int*));
    if (!pkill_list)
        return 1;
    memset(pkill_list, 0, process_count * sizeof(unsigned int*));
    pkill_size = 0;
 
    self_pid = GetCurrentProcessId();
#endif
 
    for (i = 0; i < task_count; ++i)
        mark_task_process(task_list[i], &search_status);
    if (kill_child_processes)
        mark_child_processes();
    if (force_termination)
        terminate_status = terminate_processes();
    else
        terminate_status = send_close_messages();
    return search_status ? search_status : terminate_status;
}

Variable Documentation

force_termination

BOOL force_termination = FALSE

static

Definition at line 37 of file taskkill.c.

Referenced by process_arguments(), and wmain().

is_numeric

BOOL is_numeric

Definition at line 47 of file taskkill.c.

Referenced by mark_task_process(), send_close_messages(), taskkill_message_print_process(), and terminate_processes().

kill_child_processes

BOOL kill_child_processes = FALSE

static

Definition at line 38 of file taskkill.c.

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

matched

BOOL matched

Definition at line 46 of file taskkill.c.

Referenced by compare_and_check(), flush_refs(), ldap_parse_resultA(), ldap_parse_resultW(), mark_child_processes(), mark_task_process(), Tokenizer::match(), matches_domain_pattern(), module_find_cb(), pdb_fetch_file_info(), pdb_init(), pdb_process_internal(), FxPkgPnp::PnpEventHardwareAvailable(), FxPkgPnp::PnpEventRestartHardwareAvailable(), FxPkgPnp::PnpEventStartingFromStopped(), search_domain_for_zone(), SECUR32_findPackageW(), send_close_messages(), terminate_processes(), and test_try_transform().

p

PROCESSENTRY32W p

Definition at line 45 of file taskkill.c.

process_count

unsigned int process_count

static

Definition at line 50 of file taskkill.c.

Referenced by enumerate_processes(), find_parent(), mark_child_processes(), mark_task_process(), send_close_messages(), terminate_processes(), test_GetPerformanceInfo(), and wmain().

struct { ... } * process_list

Referenced by enumerate_processes(), find_parent(), mark_child_processes(), mark_task_process(), send_close_messages(), taskkill_message_print_process(), and terminate_processes().

task_count

unsigned int task_count

static

Definition at line 41 of file taskkill.c.

Referenced by add_to_task_list(), and wmain().

task_list

WCHAR** task_list

static

Definition at line 40 of file taskkill.c.

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