mirror of
git://source.winehq.org/git/wine.git
synced 2024-09-06 16:48:03 +00:00
taskkill: Implement basic command-line parsing.
This commit is contained in:
parent
8e540ed26f
commit
db90b2d004
34
programs/taskkill/En.rc
Normal file
34
programs/taskkill/En.rc
Normal file
|
@ -0,0 +1,34 @@
|
||||||
|
/*
|
||||||
|
* Task termination utility
|
||||||
|
* English language support
|
||||||
|
*
|
||||||
|
* Copyright 2010 Andrew Nguyen
|
||||||
|
*
|
||||||
|
* This library is free software; you can redistribute it and/or
|
||||||
|
* modify it under the terms of the GNU Lesser General Public
|
||||||
|
* License as published by the Free Software Foundation; either
|
||||||
|
* version 2.1 of the License, or (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This library is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||||
|
* Lesser General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU Lesser General Public
|
||||||
|
* License along with this library; if not, write to the Free Software
|
||||||
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "taskkill.h"
|
||||||
|
|
||||||
|
LANGUAGE LANG_ENGLISH, SUBLANG_DEFAULT
|
||||||
|
|
||||||
|
STRINGTABLE
|
||||||
|
{
|
||||||
|
STRING_USAGE, "Usage: taskkill [/?] [/f] [/im ProcessName | /pid ProcessID]\n"
|
||||||
|
STRING_INVALID_OPTION, "Error: Unknown or invalid command line option specified.\n"
|
||||||
|
STRING_INVALID_PARAM, "Error: Invalid command line parameter specified.\n"
|
||||||
|
STRING_MISSING_OPTION, "Error: One of options /im or /pid must be specified.\n"
|
||||||
|
STRING_MISSING_PARAM, "Error: Option %s expects a command line parameter.\n"
|
||||||
|
STRING_MUTUAL_EXCLUSIVE, "Error: Options /im and /pid are mutually exclusive.\n"
|
||||||
|
}
|
|
@ -1,7 +1,11 @@
|
||||||
EXTRADEFS = -DWINE_NO_UNICODE_MACROS
|
EXTRADEFS = -DWINE_NO_UNICODE_MACROS
|
||||||
MODULE = taskkill.exe
|
MODULE = taskkill.exe
|
||||||
APPMODE = -mconsole -municode
|
APPMODE = -mconsole -municode
|
||||||
|
IMPORTS = user32
|
||||||
|
|
||||||
C_SRCS = taskkill.c
|
C_SRCS = taskkill.c
|
||||||
|
|
||||||
|
RC_SRCS = \
|
||||||
|
En.rc
|
||||||
|
|
||||||
@MAKE_PROG_RULES@
|
@MAKE_PROG_RULES@
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
/*
|
/*
|
||||||
* Task termination utility
|
* Task termination utility
|
||||||
*
|
*
|
||||||
|
* Copyright 2008 Andrew Riedi
|
||||||
* Copyright 2010 Andrew Nguyen
|
* Copyright 2010 Andrew Nguyen
|
||||||
*
|
*
|
||||||
* This library is free software; you can redistribute it and/or
|
* This library is free software; you can redistribute it and/or
|
||||||
|
@ -22,10 +23,193 @@
|
||||||
#include <wine/debug.h>
|
#include <wine/debug.h>
|
||||||
#include <wine/unicode.h>
|
#include <wine/unicode.h>
|
||||||
|
|
||||||
|
#include "taskkill.h"
|
||||||
|
|
||||||
WINE_DEFAULT_DEBUG_CHANNEL(taskkill);
|
WINE_DEFAULT_DEBUG_CHANNEL(taskkill);
|
||||||
|
|
||||||
|
int force_termination;
|
||||||
|
|
||||||
|
WCHAR **task_list;
|
||||||
|
unsigned int task_count;
|
||||||
|
|
||||||
|
static int taskkill_vprintfW(const WCHAR *msg, va_list va_args)
|
||||||
|
{
|
||||||
|
int wlen;
|
||||||
|
DWORD count, ret;
|
||||||
|
WCHAR msg_buffer[8192];
|
||||||
|
|
||||||
|
wlen = vsprintfW(msg_buffer, msg, va_args);
|
||||||
|
|
||||||
|
ret = WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE), msg_buffer, wlen, &count, NULL);
|
||||||
|
if (!ret)
|
||||||
|
{
|
||||||
|
DWORD len;
|
||||||
|
char *msgA;
|
||||||
|
|
||||||
|
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;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int taskkill_printfW(const WCHAR *msg, ...)
|
||||||
|
{
|
||||||
|
va_list va_args;
|
||||||
|
int len;
|
||||||
|
|
||||||
|
va_start(va_args, msg);
|
||||||
|
len = taskkill_vprintfW(msg, va_args);
|
||||||
|
va_end(va_args);
|
||||||
|
|
||||||
|
return len;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int taskkill_message_printfW(int msg, ...)
|
||||||
|
{
|
||||||
|
va_list va_args;
|
||||||
|
WCHAR msg_buffer[8192];
|
||||||
|
int len;
|
||||||
|
|
||||||
|
LoadStringW(GetModuleHandleW(NULL), msg, msg_buffer,
|
||||||
|
sizeof(msg_buffer)/sizeof(WCHAR));
|
||||||
|
|
||||||
|
va_start(va_args, msg);
|
||||||
|
len = taskkill_vprintfW(msg_buffer, va_args);
|
||||||
|
va_end(va_args);
|
||||||
|
|
||||||
|
return len;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int taskkill_message(int msg)
|
||||||
|
{
|
||||||
|
static const WCHAR formatW[] = {'%','s',0};
|
||||||
|
WCHAR msg_buffer[8192];
|
||||||
|
|
||||||
|
LoadStringW(GetModuleHandleW(NULL), msg, msg_buffer,
|
||||||
|
sizeof(msg_buffer)/sizeof(WCHAR));
|
||||||
|
|
||||||
|
return taskkill_printfW(formatW, msg_buffer);
|
||||||
|
}
|
||||||
|
|
||||||
|
static BOOL add_to_task_list(WCHAR *name)
|
||||||
|
{
|
||||||
|
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;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* FIXME Argument processing does not match behavior observed on Windows.
|
||||||
|
* Stringent argument counting and processing is performed, and unrecognized
|
||||||
|
* options are detected as parameters when placed after options that accept one. */
|
||||||
|
static BOOL process_arguments(int argc, WCHAR *argv[])
|
||||||
|
{
|
||||||
|
static const WCHAR slashForceTerminate[] = {'/','f',0};
|
||||||
|
static const WCHAR slashImage[] = {'/','i','m',0};
|
||||||
|
static const WCHAR slashPID[] = {'/','p','i','d',0};
|
||||||
|
static const WCHAR slashHelp[] = {'/','?',0};
|
||||||
|
|
||||||
|
if (argc > 1)
|
||||||
|
{
|
||||||
|
int i;
|
||||||
|
BOOL has_im = 0, has_pid = 0;
|
||||||
|
|
||||||
|
/* Only the lone help option is recognized. */
|
||||||
|
if (argc == 2 && !strcmpW(slashHelp, argv[1]))
|
||||||
|
{
|
||||||
|
taskkill_message(STRING_USAGE);
|
||||||
|
exit(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (i = 1; i < argc; i++)
|
||||||
|
{
|
||||||
|
int got_im = 0, got_pid = 0;
|
||||||
|
|
||||||
|
if (!strcmpiW(slashForceTerminate, argv[i]))
|
||||||
|
force_termination = 1;
|
||||||
|
/* 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(slashImage, argv[i])) ||
|
||||||
|
(got_pid = !strcmpiW(slashPID, argv[i])))
|
||||||
|
{
|
||||||
|
if (!argv[i + 1])
|
||||||
|
{
|
||||||
|
taskkill_message_printfW(STRING_MISSING_PARAM, argv[i]);
|
||||||
|
taskkill_message(STRING_USAGE);
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (got_im) has_im = 1;
|
||||||
|
if (got_pid) has_pid = 1;
|
||||||
|
|
||||||
|
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
|
||||||
|
{
|
||||||
|
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;
|
||||||
|
}
|
||||||
|
|
||||||
int wmain(int argc, WCHAR *argv[])
|
int wmain(int argc, WCHAR *argv[])
|
||||||
{
|
{
|
||||||
WINE_FIXME("taskkill.exe is not implemented\n");
|
if (!process_arguments(argc, argv))
|
||||||
|
{
|
||||||
|
HeapFree(GetProcessHeap(), 0, task_list);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
WINE_FIXME("taskkill.exe functionality is not implemented\n");
|
||||||
|
|
||||||
|
HeapFree(GetProcessHeap(), 0, task_list);
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
29
programs/taskkill/taskkill.h
Normal file
29
programs/taskkill/taskkill.h
Normal file
|
@ -0,0 +1,29 @@
|
||||||
|
/*
|
||||||
|
* Task termination utility private definitions
|
||||||
|
*
|
||||||
|
* Copyright 2010 Andrew Nguyen
|
||||||
|
*
|
||||||
|
* This library is free software; you can redistribute it and/or
|
||||||
|
* modify it under the terms of the GNU Lesser General Public
|
||||||
|
* License as published by the Free Software Foundation; either
|
||||||
|
* version 2.1 of the License, or (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This library is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||||
|
* Lesser General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU Lesser General Public
|
||||||
|
* License along with this library; if not, write to the Free Software
|
||||||
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <windef.h>
|
||||||
|
|
||||||
|
/* Translation IDs. */
|
||||||
|
#define STRING_USAGE 101
|
||||||
|
#define STRING_INVALID_OPTION 102
|
||||||
|
#define STRING_INVALID_PARAM 103
|
||||||
|
#define STRING_MISSING_OPTION 104
|
||||||
|
#define STRING_MISSING_PARAM 105
|
||||||
|
#define STRING_MUTUAL_EXCLUSIVE 106
|
Loading…
Reference in a new issue