reactos/rostests/tests/wclickat/wclickat.c
Daniel Reimer 622c29bffb Delete all Trailing spaces in code.
svn path=/trunk/; revision=29688
2007-10-19 23:01:40 +00:00

687 lines
21 KiB
C

/*----------------------------------------------------------------------------
** wclickat.c
** Utilty to send clicks to Wine Windows
**
** See usage() for usage instructions.
**
**---------------------------------------------------------------------------
** Copyright 2004 Jozef Stefanka for CodeWeavers, Inc.
** Copyright 2005 Dmitry Timoshkov for CodeWeavers, Inc.
** Copyright 2005 Francois Gouget for CodeWeavers, Inc.
**
** This program is free software; you can redistribute it and/or modify
** it under the terms of the GNU General Public License as published by
** the Free Software Foundation; either version 2 of the License, or
** (at your option) any later version.
**
** This program 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 General Public License for more details.
**
** You should have received a copy of the GNU General Public License
** along with this program; if not, write to the Free Software
** Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
**
**--------------------------------------------------------------------------*/
#include <windows.h>
#include <windowsx.h>
#include <stdio.h>
#include <ctype.h>
#define APP_NAME "wclickat"
#define DEFAULT_DELAY 500
#define DEFAULT_REPEAT 1000
#define ARRAY_LENGTH(array) (sizeof(array)/sizeof((array)[0]))
static const WCHAR STATIC_CLASS[]={'s','t','a','t','i','c','\0'};
/*----------------------------------------------------------------------------
** Global variables
**--------------------------------------------------------------------------*/
#define RC_RUNNING -1
#define RC_SUCCESS 0
#define RC_INVALID_ARGUMENTS 1
#define RC_NODISPLAY 2
#define RC_TIMEOUT 3
static int status;
typedef enum
{
ACTION_INVALID,
ACTION_FIND,
ACTION_LCLICK,
ACTION_MCLICK,
ACTION_RCLICK
} action_type;
static action_type g_action = ACTION_INVALID;
static WCHAR* g_window_class = NULL;
static WCHAR* g_window_title = NULL;
static long g_control_id = 0;
static WCHAR* g_control_class = NULL;
static WCHAR* g_control_caption = NULL;
static long g_x = -1;
static long g_y = -1;
static long g_dragto_x = -1;
static long g_dragto_y = -1;
static long g_disabled = 0;
static long g_delay = DEFAULT_DELAY;
static long g_timeout = 0;
static long g_repeat = 0;
static long g_untildeath = 0;
static UINT timer_id;
/*
* Provide some basic debugging support.
*/
#ifdef __GNUC__
#define __PRINTF_ATTR(fmt,args) __attribute__((format (printf,fmt,args)))
#else
#define __PRINTF_ATTR(fmt,args)
#endif
static int debug_on=0;
static int init_debug()
{
char* str=getenv("CXTEST_DEBUG");
if (str && strstr(str, "+wclickat"))
debug_on=1;
return debug_on;
}
static void cxlog(const char* format, ...) __PRINTF_ATTR(1,2);
static void cxlog(const char* format, ...)
{
va_list valist;
if (debug_on)
{
va_start(valist, format);
vfprintf(stderr, format, valist);
va_end(valist);
}
}
/*----------------------------------------------------------------------------
** usage
**--------------------------------------------------------------------------*/
static void usage(void)
{
fprintf(stderr, "%s - Utility to send clicks to Wine Windows.\n", APP_NAME);
fprintf(stderr, "----------------------------------------------\n");
fprintf(stderr, "Usage:\n");
fprintf(stderr, " %s action --winclass class --wintitle title [--timeout ms]\n",APP_NAME);
fprintf(stderr, " %*.*s [--ctrlclas class] [--ctrlcaption caption] [--ctrlid id]\n", strlen(APP_NAME) + 3, strlen(APP_NAME) + 3, "");
fprintf(stderr, " %*.*s [--position XxY] [--delay ms] [--untildeath] [--repeat ms]\n", strlen(APP_NAME) + 3, strlen(APP_NAME) + 3, "");
fprintf(stderr, "Where action can be one of:\n");
fprintf(stderr, " find Find the specified window or control\n");
fprintf(stderr, " button<n> Send a click with the given X button number\n");
fprintf(stderr, " click|lclick Synonym for button1 (left click)\n");
fprintf(stderr, " mclick Synonym for button2 (middle click)\n");
fprintf(stderr, " rclick Synonym for button3 (right click)\n");
fprintf(stderr, "\n");
fprintf(stderr, "The options are as follows:\n");
fprintf(stderr, " --timeout ms How long to wait before failing with a code of %d\n", RC_TIMEOUT);
fprintf(stderr, " --winclass class Class name of the top-level window of interest\n");
fprintf(stderr, " --wintitle title Title of the top-level window of interest\n");
fprintf(stderr, " --ctrlclass name Class name of the control of interest, if any\n");
fprintf(stderr, " --ctrlcaption cap A substring of the control's caption\n");
fprintf(stderr, " --ctrlid id Id of the control\n");
fprintf(stderr, " --position XxY Coordinates for the click, relative to the window / control\n");
fprintf(stderr, " --dragto If given, then position specifies start click, and\n");
fprintf(stderr, " dragto specifies release coords.\n");
fprintf(stderr, " --allow-disabled Match the window or control even hidden or disabled\n");
fprintf(stderr, " --delay ms Wait ms milliseconds before clicking. The default is %d\n", DEFAULT_DELAY);
fprintf(stderr, " --untildeath Wait until the window disappears\n");
fprintf(stderr, " --repeat ms Click every ms milliseconds. The default is %d\n", DEFAULT_REPEAT);
fprintf(stderr, "\n");
fprintf(stderr, "%s returns %d on success\n", APP_NAME, RC_SUCCESS);
fprintf(stderr, "\n");
fprintf(stderr, "Environment variable overrides:\n");
fprintf(stderr, " CXTEST_TIME_MULTIPLE Specifies a floating multiplier applied to any\n");
fprintf(stderr, " delay and timeout parameters.\n");
}
static const WCHAR* my_strstriW(const WCHAR* haystack, const WCHAR* needle)
{
const WCHAR *h,*n;
WCHAR first;
if (!*needle)
return haystack;
/* Special case the first character because
* we will be doing a lot of comparisons with it.
*/
first=towlower(*needle);
needle++;
while (*haystack)
{
while (towlower(*haystack)!=first && *haystack)
haystack++;
h=haystack+1;
n=needle;
while (towlower(*h)==towlower(*n) && *h)
{
h++;
n++;
}
if (!*n)
return haystack;
haystack++;
}
return NULL;
}
static BOOL CALLBACK find_control(HWND hwnd, LPARAM lParam)
{
WCHAR str[1024];
HWND* pcontrol;
if (!GetClassNameW(hwnd, str, ARRAY_LENGTH(str)) ||
lstrcmpiW(str, g_control_class))
return TRUE;
if (g_control_caption)
{
if (!GetWindowTextW(hwnd, str, ARRAY_LENGTH(str)) ||
!my_strstriW(str, g_control_caption))
return TRUE;
}
if (g_control_id && g_control_id != GetWindowLong(hwnd, GWL_ID))
return TRUE;
/* Check that the control is visible and active */
if (!g_disabled)
{
DWORD style = GetWindowStyle(hwnd);
if (!(style & WS_VISIBLE) || (style & WS_DISABLED))
return TRUE;
}
pcontrol = (HWND*)lParam;
*pcontrol = hwnd;
return FALSE;
}
static BOOL CALLBACK find_top_window(HWND hwnd, LPARAM lParam)
{
WCHAR str[1024];
HWND* pwindow;
if (!GetClassNameW(hwnd, str, ARRAY_LENGTH(str)) ||
lstrcmpiW(str, g_window_class))
return TRUE;
if (!GetWindowTextW(hwnd, str, ARRAY_LENGTH(str)) ||
lstrcmpiW(str, g_window_title))
return TRUE;
/* Check that the window is visible and active */
if (!g_disabled)
{
DWORD style = GetWindowStyle(hwnd);
if (!(style & WS_VISIBLE) || (style & WS_DISABLED))
return TRUE;
}
/* See if we find the control we want */
if (g_control_class)
{
HWND control = NULL;
EnumChildWindows(hwnd, find_control, (LPARAM)&control);
if (!control)
return TRUE;
hwnd=control;
}
pwindow = (HWND*)lParam;
*pwindow = hwnd;
return FALSE;
}
static HWND find_window()
{
HWND hwnd;
hwnd=NULL;
EnumWindows(find_top_window, (LPARAM)&hwnd);
return hwnd;
}
static void do_click(HWND window, DWORD down, DWORD up)
{
WINDOWINFO window_info;
long x, y;
SetForegroundWindow(GetParent(window));
window_info.cbSize=sizeof(window_info);
GetWindowInfo(window, &window_info);
/* The calculations below convert the coordinates so they are absolute
* screen coordinates in 'Mickeys' as required by mouse_event.
* In mickeys the screen size is always 65535x65535.
*/
x=window_info.rcWindow.left+g_x;
if (x<window_info.rcWindow.left || x>=window_info.rcWindow.right)
x=(window_info.rcWindow.right+window_info.rcWindow.left)/2;
x=(x << 16)/GetSystemMetrics(SM_CXSCREEN);
y=window_info.rcWindow.top+g_y;
if (y<window_info.rcWindow.top || y>=window_info.rcWindow.bottom)
y=(window_info.rcWindow.bottom+window_info.rcWindow.top)/2;
y=(y << 16)/GetSystemMetrics(SM_CYSCREEN);
mouse_event(MOUSEEVENTF_ABSOLUTE | MOUSEEVENTF_MOVE, x, y, 0, 0);
if (down) {
mouse_event(MOUSEEVENTF_ABSOLUTE | down, x, y, 0, 0);
if ((g_dragto_x > 0) && (g_dragto_y > 0)) {
int i;
long dx, dy;
long step_per_x, step_per_y;
long dragto_x, dragto_y;
dragto_x=window_info.rcWindow.left+g_dragto_x;
if (dragto_x<window_info.rcWindow.left || dragto_x>=window_info.rcWindow.right)
dragto_x=(window_info.rcWindow.right+window_info.rcWindow.left)/2;
dragto_x=(dragto_x << 16)/GetSystemMetrics(SM_CXSCREEN);
dragto_y=window_info.rcWindow.top+g_dragto_y;
if (dragto_y<window_info.rcWindow.top || dragto_y>=window_info.rcWindow.bottom)
dragto_y=(window_info.rcWindow.bottom+window_info.rcWindow.top)/2;
dragto_y=(dragto_y << 16)/GetSystemMetrics(SM_CYSCREEN);
dx = g_dragto_x - g_x;
dy = g_dragto_y - g_y;
step_per_x = dx / 4;
step_per_y = dy / 4;
for (i = 0; i < 4; i++) {
mouse_event(MOUSEEVENTF_MOVE, step_per_x, step_per_y, 0, 0);
}
x=dragto_x;
y=dragto_y;
}
}
if (up)
mouse_event(MOUSEEVENTF_ABSOLUTE | up, x, y, 0, 0);
}
static void CALLBACK ClickProc(HWND hwnd, UINT uMsg, UINT_PTR idEvent, DWORD dwTime)
{
HWND window = find_window();
if (!window)
{
if (g_untildeath)
{
/* FIXME: The window / control might just be disabled and if
* that's the case we should not exit yet. But I don't expect
* --untildeath to be used at all anyway so fixing this can
* wait until it becomes necessary.
*/
status=RC_SUCCESS;
}
else
cxlog("The window has disappeared!\n");
return;
}
switch (g_action)
{
case ACTION_FIND:
/* Nothing to do */
break;
case ACTION_LCLICK:
cxlog("Sending left click\n");
do_click(window, MOUSEEVENTF_LEFTDOWN, MOUSEEVENTF_LEFTUP);
break;
case ACTION_MCLICK:
cxlog("Sending middle click\n");
do_click(window, MOUSEEVENTF_MIDDLEDOWN, MOUSEEVENTF_MIDDLEUP);
break;
case ACTION_RCLICK:
cxlog("Sending right click\n");
do_click(window, MOUSEEVENTF_RIGHTDOWN, MOUSEEVENTF_RIGHTUP);
default:
fprintf(stderr, "error: unknown action %d\n", g_action);
break;
}
if (!g_repeat)
status=RC_SUCCESS;
}
static void CALLBACK DelayProc(HWND hwnd, UINT uMsg, UINT_PTR idEvent, DWORD dwTime)
{
KillTimer(NULL, timer_id);
timer_id=0;
if (g_repeat)
{
cxlog("Setting up a timer for --repeat\n");
timer_id=SetTimer(NULL, 0, g_repeat, ClickProc);
}
ClickProc(NULL, 0, 0, 0);
}
static void CALLBACK FindWindowProc(HWND hwnd, UINT uMsg, UINT_PTR idEvent, DWORD dwTime)
{
HWND window = find_window();
if (!window)
return;
cxlog("Found the window\n");
if (g_delay)
{
cxlog("Waiting for a bit\n");
KillTimer(NULL, timer_id);
timer_id=SetTimer(NULL, 0, g_delay, DelayProc);
do_click(window, 0,0);
}
else
{
DelayProc(NULL, 0, 0, 0);
}
}
static void CALLBACK TimeoutProc(HWND hwnd, UINT uMsg, UINT_PTR idEvent, DWORD dwTime)
{
status = RC_TIMEOUT;
}
/*----------------------------------------------------------------------------
** parse_arguments
**--------------------------------------------------------------------------*/
static int arg_get_long(const char** *argv, const char* name, long* value)
{
if (!**argv)
{
fprintf(stderr, "error: missing argument for '%s'\n", name);
return 1;
}
*value=atol(**argv);
if (*value < 0)
{
fprintf(stderr, "error: invalid argument '%s' for '%s'\n",
**argv, name);
(*argv)++;
return 1;
}
(*argv)++;
return 0;
}
static int arg_get_utf8(const char** *argv, const char* name, WCHAR* *value)
{
int len;
if (!**argv)
{
fprintf(stderr, "error: missing argument for '%s'\n", name);
return 1;
}
len = MultiByteToWideChar(CP_UTF8, 0, **argv, -1, NULL, 0);
*value = HeapAlloc(GetProcessHeap(), 0, len*sizeof(WCHAR));
if (!*value)
{
fprintf(stderr, "error: memory allocation error\n");
(*argv)++;
return 1;
}
MultiByteToWideChar(CP_UTF8, 0, **argv, -1, *value, len);
(*argv)++;
return 0;
}
static int parse_arguments(int argc, const char** argv)
{
int rc;
const char* arg;
char* p;
rc=0;
argv++;
while (*argv)
{
arg=*argv++;
if (*arg!='-')
{
if (g_action != ACTION_INVALID)
{
fprintf(stderr, "error: '%s' an action has already been specified\n", arg);
rc=1;
}
else if (strcmp(arg, "click") == 0 || strcmp(arg, "lclick") == 0)
{
g_action = ACTION_LCLICK;
}
else if (strcmp(arg, "mclick") == 0)
{
g_action = ACTION_MCLICK;
}
else if (strcmp(arg, "rclick") == 0)
{
g_action = ACTION_RCLICK;
}
else if (strncmp(arg, "button", 6) == 0)
{
int button;
char extra='\0';
int r=sscanf(arg, "button%d%c", &button, &extra);
/* We should always get r==1 but due to a bug in Wine's
* msvcrt.dll implementation (at least up to 20050127)
* we may also get r==2 and extra=='\0'.
*/
if (r!=1 && (r!=2 || extra!='\0'))
{
fprintf(stderr, "error: invalid argument '%s' for '%s'\n",
*argv, arg);
rc=1;
}
else if (button<1 || button>3)
{
fprintf(stderr, "error: unknown button '%s'\n", arg);
rc=1;
}
else
{
/* Just to remain compatible with the enum */
g_action=button+ACTION_LCLICK-1;
}
}
else if (strcmp(arg, "find") == 0)
{
g_action = ACTION_FIND;
}
else
{
fprintf(stderr, "error: unknown action '%s'\n", arg);
rc=1;
}
}
else if (strcmp(arg, "--winclass") == 0)
{
rc|=arg_get_utf8(&argv, arg, &g_window_class);
}
else if (strcmp(arg, "--wintitle") == 0)
{
rc|=arg_get_utf8(&argv,arg, &g_window_title);
}
else if (strcmp(arg, "--ctrlclass") == 0)
{
rc|=arg_get_utf8(&argv, arg, &g_control_class);
}
else if (strcmp(arg, "--ctrlid") == 0)
{
rc|=arg_get_long(&argv, arg, &g_control_id);
}
else if (strcmp(arg, "--ctrlcaption") == 0)
{
rc|=arg_get_utf8(&argv, arg, &g_control_caption);
}
else if (strcmp(arg, "--position") == 0)
{
if (!*argv)
{
fprintf(stderr, "error: missing argument for '%s'\n", arg);
rc=1;
}
else
{
char extra='\0';
int r=sscanf(*argv, "%ldx%ld%c", &g_x, &g_y, &extra);
/* We should always get r==2 but due to a bug in Wine's
* msvcrt.dll implementation (at least up to 20050127)
* we may also get r==3 and extra=='\0'.
*/
if (r!=2 && (r!=3 || extra!='\0'))
{
fprintf(stderr, "error: invalid argument '%s' for '%s'\n",
*argv, arg);
rc=1;
}
argv++;
}
}
else if (strcmp(arg, "--dragto") == 0)
{
if (!*argv)
{
fprintf(stderr, "error: missing argument for '%s'\n", arg);
rc=1;
}
else
{
char extra='\0';
int r=sscanf(*argv, "%ldx%ld%c", &g_dragto_x, &g_dragto_y, &extra);
/* We should always get r==2 but due to a bug in Wine's
* * msvcrt.dll implementation (at least up to 20050127)
* * we may also get r==3 and extra=='\0'.
* */
if (r!=2 && (r!=3 || extra!='\0'))
{
fprintf(stderr, "error: invalid argument '%s' for '%s'\n",
*argv, arg);
rc=1;
}
argv++;
}
}
else if (strcmp(arg, "--allow-disabled") == 0)
{
g_disabled = 1;
}
else if (strcmp(arg, "--delay") == 0)
{
rc|=arg_get_long(&argv, arg, &g_delay);
}
else if (strcmp(arg, "--timeout") == 0)
{
rc|=arg_get_long(&argv, arg, &g_timeout);
}
else if (strcmp(arg, "--repeat") == 0)
{
rc|=arg_get_long(&argv, arg, &g_repeat);
}
else if (strcmp(arg, "--untildeath") == 0)
{
g_untildeath=1;
}
else if (strcmp(arg, "--help") == 0)
{
rc=2;
}
}
if (g_action == ACTION_INVALID)
{
fprintf(stderr, "error: you must specify an action type\n");
rc=1;
}
else
{
/* Adjust the default delay and repeat parameters depending on
* the operating mode so less needs to be specified on the command
* line, and so we can assume them to be set right.
*/
if (g_action == ACTION_FIND)
g_delay=0;
if (!g_untildeath)
g_repeat=0;
else if (!g_repeat)
g_repeat=DEFAULT_REPEAT;
}
if (!g_window_class)
{
fprintf(stderr, "error: you must specify a --winclass parameter\n");
rc=1;
}
if (!g_window_title)
{
fprintf(stderr, "error: you must specify a --wintitle parameter\n");
rc=1;
}
if (g_control_class)
{
if (!g_control_id && !g_control_caption)
{
fprintf(stderr, "error: you must specify either the control id or its caption\n");
rc=1;
}
}
/*------------------------------------------------------------------------
** Process environment variables
**----------------------------------------------------------------------*/
p = getenv("CXTEST_TIME_MULTIPLE");
if (p)
{
float g_multiple = atof(p);
g_delay = (long) (((float) g_delay) * g_multiple);
g_timeout = (long) (((float) g_timeout) * g_multiple);
}
return rc;
}
int main(int argc, const char** argv)
{
MSG msg;
init_debug();
status = parse_arguments(argc, argv);
if (status)
{
if (status == 2)
usage();
else
fprintf(stderr, "Issue %s --help for usage.\n", *argv);
return RC_INVALID_ARGUMENTS;
}
cxlog("Entering message loop. action=%d\n", g_action);
if (g_timeout>0)
SetTimer(NULL, 0, g_timeout, TimeoutProc);
timer_id=SetTimer(NULL, 0, 100, FindWindowProc);
status=RC_RUNNING;
while (status==RC_RUNNING && GetMessage(&msg, NULL, 0, 0)!=0)
{
TranslateMessage(&msg);
DispatchMessage(&msg);
}
return status;
}