reactos/base/applications/mstsc/win32.c

1384 lines
36 KiB
C

/* -*- c-basic-offset: 8 -*-
rdesktop: A Remote Desktop Protocol client.
win32 calls
Copyright (C) Jay Sorg 2006
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.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#include "precomp.h"
#include "uimain.h"
extern char g_username[];
extern char g_hostname[];
extern char g_servername[];
extern char g_password[];
extern char g_shell[];
extern char g_directory[];
extern char g_domain[];
extern int g_width;
extern int g_height;
extern int g_tcp_sck;
extern int g_server_depth;
extern int g_tcp_port_rdp; /* in tcp.c */
extern int pal_entries[];
static HWND g_Wnd = 0;
static HINSTANCE g_Instance = 0;
static HCURSOR g_cursor = 0;
static int g_block = 0;
static int g_xoff = 0; /* offset from window to client coords */
static int g_yoff = 0;
static int g_xscroll = 0; /* current scroll position */
static int g_yscroll = 0;
static int g_screen_width = 0;
static int g_screen_height = 0;
static int g_wnd_cwidth = 0; /* set from WM_SIZE */
static int g_wnd_cheight = 0;
static int g_fullscreen = 0;
static int g_workarea = 0;
static int g_mousex = 0; /* in client coords */
static int g_mousey = 0;
//static int g_width_height_set = 0;
static int g_clip_left = 0;
static int g_clip_top = 0;
static int g_clip_right = 800;
static int g_clip_bottom = 600;
static RECT g_wnd_clip; /* this client area of whats actually visible */
/* set from WM_SIZE */
/*****************************************************************************/
static void
str_to_uni(TCHAR * sizex, char * size1)
{
int len;
int i;
len = strlen(size1);
for (i = 0; i < len; i++)
{
sizex[i] = size1[i];
}
sizex[len] = 0;
}
/*****************************************************************************/
static void
uni_to_str(char * sizex, TCHAR * size1)
{
int len;
int i;
len = lstrlen(size1);
for (i = 0; i < len; i++)
{
sizex[i] = (char)size1[i];
}
sizex[len] = 0;
}
/*****************************************************************************/
/* returns non zero if it processed something */
static int
check_sck(void)
{
fd_set rfds;
struct timeval tm;
int count;
int rv;
rv = 0;
if (g_block == 0)
{
g_block = 1;
/* see if there really is data */
FD_ZERO(&rfds);
FD_SET((unsigned int)g_tcp_sck, &rfds);
ZeroMemory(&tm, sizeof(tm));
count = select(g_tcp_sck + 1, &rfds, 0, 0, &tm);
if (count > 0)
{
if (ui_read_wire())
{
rv = 1;
}
else
{
PostQuitMessage(0);
}
}
g_block = 0;
}
return rv;
}
/*****************************************************************************/
void
mi_error(char * msg)
{
#ifdef WITH_DEBUG
printf(msg);
#else /* WITH_DEBUG */
TCHAR lmsg[512];
TCHAR ltitle[512];
str_to_uni(lmsg, msg);
str_to_uni(ltitle, "Error");
MessageBox(g_Wnd, lmsg, ltitle, MB_OK);
#endif /* WITH_DEBUG */
}
/*****************************************************************************/
static int
get_scan_code_from_ascii(int code)
{
int rv;
rv = 0;
switch (code & 0xff)
{
case 0x30: rv = 0x0b; break; // 0
case 0x31: rv = 0x02; break; // 1
case 0x32: rv = 0x03; break; // 2
case 0x33: rv = 0x04; break; // 3
case 0x34: rv = 0x05; break; // 4
case 0x35: rv = 0x06; break; // 5
case 0x36: rv = 0x07; break; // 6
case 0x37: rv = 0x08; break; // 7
case 0x38: rv = 0x09; break; // 8
case 0x39: rv = 0x0a; break; // 9
case 0xbd: rv = 0x0c; break; // -
case 0xbb: rv = 0x0d; break; // =
case 0x08: rv = 0x0e; break; // backspace
case 0x09: rv = 0x0f; break; // tab
case 0xdb: rv = 0x1b; break; // ]
case 0xdd: rv = 0x1a; break; // [
case 0x14: rv = 0x3a; break; // capslock
case 0xba: rv = 0x27; break; // ;
case 0xde: rv = 0x28; break; // '
case 0x10: rv = 0x2a; break; // shift
case 0xbc: rv = 0x33; break; // ,
case 0xbe: rv = 0x34; break; // .
case 0xbf: rv = 0x35; break; // /
case 0x0d: rv = 0x1c; break; // enter
case 0x27: rv = 0x4d; break; // arrow right
case 0x25: rv = 0x4b; break; // arrow left
case 0x26: rv = 0x48; break; // arrow up
case 0x28: rv = 0x50; break; // arrow down
case 0x20: rv = 0x39; break; // space
case 0xdc: rv = 0x2b; break; // backslash
case 0xc0: rv = 0x29; break; // `
case 0x11: rv = 0x1d; break; // ctl
case 0x41: rv = 0x1e; break; // a
case 0x42: rv = 0x30; break; // b
case 0x43: rv = 0x2e; break; // c
case 0x44: rv = 0x20; break; // d
case 0x45: rv = 0x12; break; // e
case 0x46: rv = 0x21; break; // f
case 0x47: rv = 0x22; break; // g
case 0x48: rv = 0x23; break; // h
case 0x49: rv = 0x17; break; // i
case 0x4a: rv = 0x24; break; // j
case 0x4b: rv = 0x25; break; // k
case 0x4c: rv = 0x26; break; // l
case 0x4d: rv = 0x32; break; // m
case 0x4e: rv = 0x31; break; // n
case 0x4f: rv = 0x18; break; // o
case 0x50: rv = 0x19; break; // p
case 0x51: rv = 0x10; break; // q
case 0x52: rv = 0x13; break; // r
case 0x53: rv = 0x1f; break; // s
case 0x54: rv = 0x14; break; // t
case 0x55: rv = 0x16; break; // u
case 0x56: rv = 0x2f; break; // v
case 0x57: rv = 0x11; break; // w
case 0x58: rv = 0x2d; break; // x
case 0x59: rv = 0x15; break; // y
case 0x5a: rv = 0x2c; break; // z
}
return rv;
}
/*****************************************************************************/
static void
mi_scroll(int dx, int dy)
{
HRGN rgn;
rgn = CreateRectRgn(0, 0, 0, 0);
ScrollWindowEx(g_Wnd, dx, dy, 0, 0, rgn, 0, SW_ERASE);
InvalidateRgn(g_Wnd, rgn, 0);
DeleteObject(rgn);
}
/*****************************************************************************/
int
mi_read_keyboard_state(void)
{
short keydata;
int code;
code = 0;
keydata = GetKeyState(VK_SCROLL);
if (keydata & 0x0001)
{
code |= 1;
}
keydata = GetKeyState(VK_NUMLOCK);
if (keydata & 0x0001)
{
code |= 2;
}
keydata = GetKeyState(VK_CAPITAL);
if (keydata & 0x0001)
{
code |= 4;
}
return code;
}
/*****************************************************************************/
static void
mi_check_modifier(void)
{
int code;
code = mi_read_keyboard_state();
ui_set_modifier_state(code);
}
/*****************************************************************************/
static LRESULT
handle_WM_SETCURSOR(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
if (g_mousex >= g_wnd_clip.left &&
g_mousey >= g_wnd_clip.top &&
g_mousex < g_wnd_clip.right &&
g_mousey < g_wnd_clip.bottom)
{
SetCursor(g_cursor);
}
/* need default behavoir here */
return DefWindowProc(hWnd, message, wParam, lParam);
}
/*****************************************************************************/
static LRESULT
handle_WM_NCHITTEST(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
POINT pt;
pt.x = LOWORD(lParam);
pt.y = HIWORD(lParam);
if (ScreenToClient(g_Wnd, &pt))
{
g_mousex = pt.x;
g_mousey = pt.y;
}
return DefWindowProc(hWnd, message, wParam, lParam);
}
/*****************************************************************************/
static LRESULT
handle_WM_MOUSEMOVE(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
g_mousex = LOWORD(lParam);
g_mousey = HIWORD(lParam);
ui_mouse_move(g_mousex + g_xscroll, g_mousey + g_yscroll);
return 0;
}
/*****************************************************************************/
static LRESULT
handle_WM_LBUTTONDOWN(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
g_mousex = LOWORD(lParam);
g_mousey = HIWORD(lParam);
ui_mouse_button(1, g_mousex + g_xscroll, g_mousey + g_yscroll, 1);
return 0;
}
/*****************************************************************************/
static LRESULT
handle_WM_LBUTTONUP(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
g_mousex = LOWORD(lParam);
g_mousey = HIWORD(lParam);
ui_mouse_button(1, g_mousex + g_xscroll, g_mousey + g_yscroll, 0);
return 0;
}
/*****************************************************************************/
static LRESULT
handle_WM_RBUTTONDOWN(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
g_mousex = LOWORD(lParam);
g_mousey = HIWORD(lParam);
ui_mouse_button(2, g_mousex + g_xscroll, g_mousey + g_yscroll, 1);
return 0;
}
/*****************************************************************************/
static LRESULT
handle_WM_RBUTTONUP(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
g_mousex = LOWORD(lParam);
g_mousey = HIWORD(lParam);
ui_mouse_button(2, g_mousex + g_xscroll, g_mousey + g_yscroll, 0);
return 0;
}
/*****************************************************************************/
static LRESULT
handle_WM_MBUTTONDOWN(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
g_mousex = LOWORD(lParam);
g_mousey = HIWORD(lParam);
ui_mouse_button(3, g_mousex + g_xscroll, g_mousey + g_yscroll, 1);
return 0;
}
/*****************************************************************************/
static LRESULT
handle_WM_MBUTTONUP(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
g_mousex = LOWORD(lParam);
g_mousey = HIWORD(lParam);
ui_mouse_button(3, g_mousex + g_xscroll, g_mousey + g_yscroll, 0);
return 0;
}
/*****************************************************************************/
static LRESULT
handle_WM_MOUSEWHEEL(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
int delta;
delta = ((signed short)HIWORD(wParam)); /* GET_WHEEL_DELTA_WPARAM(wParam); */
if (delta > 0)
{
ui_mouse_button(4, 0, 0, 1);
ui_mouse_button(4, 0, 0, 0);
}
else
{
ui_mouse_button(5, 0, 0, 1);
ui_mouse_button(5, 0, 0, 0);
}
return 0;
}
/*****************************************************************************/
static LRESULT
handle_WM_KEY(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
int scancode;
int ext;
int down;
ext = HIWORD(lParam);
scancode = ext;
down = !(ext & 0x8000);
scancode &= 0xff;
if (scancode == 0)
{
scancode = get_scan_code_from_ascii(wParam);
}
ext &= 0x0100;
if (scancode == 0x0045) /* num lock */
{
ext &= ~0x0100;
}
if (down)
{
ui_key_down(scancode, ext);
}
else
{
ui_key_up(scancode, ext);
}
return 0;
}
/*****************************************************************************/
static LRESULT
handle_WM_PAINT(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
PAINTSTRUCT ps;
RECT rect;
HBRUSH brush;
BeginPaint(hWnd, &ps);
/* paint the area outside the rdp screen with one colour */
rect = ps.rcPaint;
rect.left = UI_MAX(rect.left, g_width);
if (!IsRectEmpty(&rect))
{
brush = CreateSolidBrush(RGB(0, 0, 255));
FillRect(ps.hdc, &rect, brush);
DeleteObject(brush);
}
rect = ps.rcPaint;
rect.top = UI_MAX(rect.top, g_height);
if (!IsRectEmpty(&rect))
{
brush = CreateSolidBrush(RGB(0, 0, 255));
FillRect(ps.hdc, &rect, brush);
DeleteObject(brush);
}
rect = ps.rcPaint;
EndPaint(hWnd, &ps);
ui_invalidate(rect.left + g_xscroll,
rect.top + g_yscroll,
(rect.right - rect.left) + 1,
(rect.bottom - rect.top) + 1);
return 0;
}
/*****************************************************************************/
static LRESULT
handle_WM_SIZE(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
int oldxscroll;
int oldyscroll;
if (wParam == SIZE_MINIMIZED)
{
return DefWindowProc(hWnd, message, wParam, lParam);
}
g_wnd_cwidth = LOWORD(lParam); /* client width / height */
g_wnd_cheight = HIWORD(lParam);
g_wnd_clip.left = 0;
g_wnd_clip.top = 0;
g_wnd_clip.right = g_wnd_clip.left + g_wnd_cwidth;
g_wnd_clip.bottom = g_wnd_clip.top + g_wnd_cheight;
if (g_wnd_cwidth < g_width || g_wnd_cheight < g_height)
{
SetScrollRange(g_Wnd, SB_HORZ, 0, g_width - g_wnd_cwidth, 1);
SetScrollRange(g_Wnd, SB_VERT, 0, g_height - g_wnd_cheight, 1);
}
oldxscroll = g_xscroll;
oldyscroll = g_yscroll;
if (g_wnd_cwidth >= g_width)
{
g_xscroll = 0;
}
else
{
g_xscroll = UI_MIN(g_xscroll, g_width - g_wnd_cwidth);
}
if (g_wnd_cheight >= g_height)
{
g_yscroll = 0;
}
else
{
g_yscroll = UI_MIN(g_yscroll, g_height - g_wnd_cheight);
}
mi_scroll(oldxscroll - g_xscroll, oldyscroll - g_yscroll);
if (wParam == SIZE_RESTORED || wParam == SIZE_MAXIMIZED)
{
/* check the caps, num, and scroll lock here */
mi_check_modifier();
}
return 0;
}
/*****************************************************************************/
static LRESULT
handle_WM_SIZING(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
LPRECT prect;
int width;
int height;
int style;
prect = (LPRECT) lParam; /* total window rect */
width = (prect->right - prect->left) - (g_xoff * 2);
height = (prect->bottom - prect->top) - (g_yoff + g_xoff);
if (height < g_height || width < g_width)
{
style = GetWindowLongPtr(g_Wnd, GWL_STYLE);
if (!(style & WS_HSCROLL))
{
style |= WS_HSCROLL | WS_VSCROLL;
SetWindowLongPtr(g_Wnd, GWL_STYLE, style);
g_xscroll = 0;
g_yscroll = 0;
SetScrollPos(g_Wnd, SB_HORZ, g_xscroll, 1);
SetScrollPos(g_Wnd, SB_VERT, g_yscroll, 1);
}
}
else if (height >= g_height && width >= g_width)
{
style = GetWindowLongPtr(g_Wnd, GWL_STYLE);
if (style & WS_HSCROLL)
{
style &= ~WS_HSCROLL;
style &= ~WS_VSCROLL;
SetWindowLongPtr(g_Wnd, GWL_STYLE, style);
g_xscroll = 0;
g_yscroll = 0;
}
}
return 0;
}
/*****************************************************************************/
static LRESULT
handle_WM_HSCROLL(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
int code;
int oldxscroll;
code = (int) LOWORD(wParam); /* scroll bar value */
if (code == SB_LINELEFT)
{
oldxscroll = g_xscroll;
g_xscroll--;
g_xscroll = UI_MAX(g_xscroll, 0);
SetScrollPos(g_Wnd, SB_HORZ, g_xscroll, 1);
mi_scroll(oldxscroll - g_xscroll, 0);
}
else if (code == SB_LINERIGHT)
{
oldxscroll = g_xscroll;
g_xscroll++;
g_xscroll = UI_MIN(g_xscroll, g_width - g_wnd_cwidth);
SetScrollPos(g_Wnd, SB_HORZ, g_xscroll, 1);
mi_scroll(oldxscroll - g_xscroll, 0);
}
else if (code == SB_PAGELEFT)
{
oldxscroll = g_xscroll;
g_xscroll -= g_wnd_cwidth / 2;
g_xscroll = UI_MAX(g_xscroll, 0);
SetScrollPos(g_Wnd, SB_HORZ, g_xscroll, 1);
mi_scroll(oldxscroll - g_xscroll, 0);
}
else if (code == SB_PAGERIGHT)
{
oldxscroll = g_xscroll;
g_xscroll += g_wnd_cwidth / 2;
g_xscroll = UI_MIN(g_xscroll, g_width - g_wnd_cwidth);
SetScrollPos(g_Wnd, SB_HORZ, g_xscroll, 1);
mi_scroll(oldxscroll - g_xscroll, 0);
}
else if (code == SB_BOTTOM)
{
oldxscroll = g_xscroll;
g_xscroll = g_width - g_wnd_cwidth;
SetScrollPos(g_Wnd, SB_HORZ, g_xscroll, 1);
mi_scroll(oldxscroll - g_xscroll, 0);
}
else if (code == SB_TOP)
{
oldxscroll = g_xscroll;
g_xscroll = 0;
SetScrollPos(g_Wnd, SB_HORZ, g_xscroll, 1);
mi_scroll(oldxscroll - g_xscroll, 0);
}
else if (code == SB_THUMBPOSITION)
{
oldxscroll = g_xscroll;
g_xscroll = (signed short) HIWORD(wParam);
SetScrollPos(g_Wnd, SB_HORZ, g_xscroll, 1);
mi_scroll(oldxscroll - g_xscroll, 0);
}
return 0;
}
/*****************************************************************************/
static LRESULT
handle_WM_VSCROLL(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
int code;
int oldyscroll;
code = (int) LOWORD(wParam); /* scroll bar value */
if (code == SB_LINELEFT)
{
oldyscroll = g_yscroll;
g_yscroll--;
g_yscroll = UI_MAX(g_yscroll, 0);
SetScrollPos(g_Wnd, SB_VERT, g_yscroll, 1);
mi_scroll(0, oldyscroll - g_yscroll);
}
else if (code == SB_LINERIGHT)
{
oldyscroll = g_yscroll;
g_yscroll++;
g_yscroll = UI_MIN(g_yscroll, g_height - g_wnd_cheight);
SetScrollPos(g_Wnd, SB_VERT, g_yscroll, 1);
mi_scroll(0, oldyscroll - g_yscroll);
}
else if (code == SB_PAGELEFT)
{
oldyscroll = g_yscroll;
g_yscroll -= g_wnd_cheight / 2;
g_yscroll = UI_MAX(g_yscroll, 0);
SetScrollPos(g_Wnd, SB_VERT, g_yscroll, 1);
mi_scroll(0, oldyscroll - g_yscroll);
}
else if (code == SB_PAGERIGHT)
{
oldyscroll = g_yscroll;
g_yscroll += g_wnd_cheight / 2;
g_yscroll = UI_MIN(g_yscroll, g_height - g_wnd_cheight);
SetScrollPos(g_Wnd, SB_VERT, g_yscroll, 1);
mi_scroll(0, oldyscroll - g_yscroll);
}
else if (code == SB_BOTTOM)
{
oldyscroll = g_yscroll;
g_yscroll = g_height - g_wnd_cheight;
SetScrollPos(g_Wnd, SB_VERT, g_yscroll, 1);
mi_scroll(0, oldyscroll - g_yscroll);
}
else if (code == SB_TOP)
{
oldyscroll = g_yscroll;
g_yscroll = 0;
SetScrollPos(g_Wnd, SB_VERT, g_yscroll, 1);
mi_scroll(0, oldyscroll - g_yscroll);
}
else if (code == SB_THUMBPOSITION)
{
oldyscroll = g_yscroll;
g_yscroll = (signed short) HIWORD(wParam);
SetScrollPos(g_Wnd, SB_VERT, g_yscroll, 1);
mi_scroll(0, oldyscroll - g_yscroll);
}
return 0;
}
/*****************************************************************************/
LRESULT CALLBACK
WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
switch (message)
{
case WM_SETCURSOR:
return handle_WM_SETCURSOR(hWnd, message, wParam, lParam);
case 0x0084: /* WinCE don't have this WM_NCHITTEST: */
return handle_WM_NCHITTEST(hWnd, message, wParam, lParam);
case WM_MOUSEMOVE:
return handle_WM_MOUSEMOVE(hWnd, message, wParam, lParam);
case WM_LBUTTONDOWN:
return handle_WM_LBUTTONDOWN(hWnd, message, wParam, lParam);
case WM_LBUTTONUP:
return handle_WM_LBUTTONUP(hWnd, message, wParam, lParam);
case WM_RBUTTONDOWN:
return handle_WM_RBUTTONDOWN(hWnd, message, wParam, lParam);
case WM_RBUTTONUP:
return handle_WM_RBUTTONUP(hWnd, message, wParam, lParam);
case WM_MBUTTONDOWN:
return handle_WM_MBUTTONDOWN(hWnd, message, wParam, lParam);
case WM_MBUTTONUP:
return handle_WM_MBUTTONUP(hWnd, message, wParam, lParam);
/* some windows compilers don't have these defined like vc6 */
case 0x020a: /* WM_MOUSEWHEEL: */
return handle_WM_MOUSEWHEEL(hWnd, message, wParam, lParam);
case WM_KEYDOWN:
case WM_KEYUP:
case WM_SYSKEYDOWN:
case WM_SYSKEYUP:
return handle_WM_KEY(hWnd, message, wParam, lParam);
case WM_CHAR:
case WM_DEADCHAR:
case WM_SYSCHAR:
case WM_SYSDEADCHAR:
break;
case WM_PAINT:
return handle_WM_PAINT(hWnd, message, wParam, lParam);
case WM_DESTROY:
PostQuitMessage(0);
break;
case WM_APP + 1:
case WM_TIMER:
check_sck();
break;
case WM_SIZE:
return handle_WM_SIZE(hWnd, message, wParam, lParam);
case 532: /* not defined in wince WM_SIZING: */
return handle_WM_SIZING(hWnd, message, wParam, lParam);
case WM_HSCROLL:
return handle_WM_HSCROLL(hWnd, message, wParam, lParam);
case WM_VSCROLL:
return handle_WM_VSCROLL(hWnd, message, wParam, lParam);
case WM_SETFOCUS:
mi_check_modifier();
return DefWindowProc(hWnd, message, wParam, lParam);
default:
return DefWindowProc(hWnd, message, wParam, lParam);
}
return 0;
}
/*****************************************************************************/
static HRGN
mi_clip(HDC dc)
{
HRGN rgn;
rgn = CreateRectRgn(g_clip_left + g_xoff - g_xscroll,
g_clip_top + g_yoff - g_yscroll,
g_clip_right + g_xoff - g_xscroll,
g_clip_bottom + g_yoff - g_yscroll);
SelectClipRgn(dc, rgn);
IntersectClipRect(dc, g_wnd_clip.left + g_xoff, g_wnd_clip.top + g_yoff,
g_wnd_clip.right + g_xoff, g_wnd_clip.bottom + g_yoff);
return rgn;
}
/*****************************************************************************/
/* returns non zero if ok */
int
mi_create_window(void)
{
RECT rc;
WNDCLASS wc;
TCHAR classname[512];
TCHAR caption[512];
DWORD style;
int x;
int y;
int w;
int h;
if (g_Wnd != 0 || g_Instance != 0)
{
return 0;
}
g_Instance = GetModuleHandle(NULL);
ZeroMemory(&wc, sizeof(wc));
wc.lpfnWndProc = WndProc; /* points to window procedure */
/* name of window class */
str_to_uni(classname, "rdesktop");
wc.lpszClassName = classname;
str_to_uni(caption, "ReactOS Remote Desktop");
wc.hIcon = LoadIcon(g_Instance,
MAKEINTRESOURCE(IDI_MSTSC));
/* Register the window class. */
if (!RegisterClass(&wc))
{
return 0; /* Failed to register window class */
}
rc.left = 0;
rc.right = rc.left + UI_MIN(g_width, g_screen_width);
rc.top = 0;
rc.bottom = rc.top + UI_MIN(g_height, g_screen_height);
if (g_fullscreen)
{
style = WS_POPUP;
}
else
{
style = WS_OVERLAPPED | WS_CAPTION | WS_POPUP | WS_MINIMIZEBOX |
WS_SYSMENU | WS_SIZEBOX | WS_MAXIMIZEBOX;
}
if (g_screen_width < g_width || g_screen_height < g_height)
{
style |= WS_HSCROLL | WS_VSCROLL;
}
AdjustWindowRectEx(&rc, style, 0, 0);
x = CW_USEDEFAULT;
y = CW_USEDEFAULT;
w = rc.right - rc.left;
h = rc.bottom - rc.top;
g_Wnd = CreateWindow(wc.lpszClassName, caption,
style, x, y, w, h,
(HWND) NULL, (HMENU) NULL, g_Instance,
(LPVOID) NULL);
g_clip_left = 0;
g_clip_top = 0;
g_clip_right = g_clip_left + g_width;
g_clip_bottom = g_clip_top + g_height;
if (g_workarea)
{
ShowWindow(g_Wnd, SW_SHOWMAXIMIZED);
}
else
{
ShowWindow(g_Wnd, SW_SHOWNORMAL);
}
UpdateWindow(g_Wnd);
WSAAsyncSelect(g_tcp_sck, g_Wnd, WM_APP + 1, FD_READ);
SetTimer(g_Wnd, 1, 333, 0);
return 1;
}
/*****************************************************************************/
int
mi_main_loop(void)
{
MSG msg;
while (GetMessage(&msg, NULL, 0, 0))
{
TranslateMessage(&msg);
DispatchMessage(&msg);
}
return msg.wParam;
}
/*****************************************************************************/
void
mi_warning(char * msg)
{
}
/*****************************************************************************/
static void
mi_show_error(char * caption)
{
LPVOID lpMsgBuf;
TCHAR lcaption[512];
FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
NULL, GetLastError(), MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
(LPTSTR) &lpMsgBuf, 0, NULL);
#ifdef WITH_DEBUG
printf(lpMsgBuf);
#else /* WITH_DEBUG */
str_to_uni(lcaption, caption);
MessageBox(g_Wnd, (LPTSTR) lpMsgBuf, lcaption,
MB_OK | MB_ICONINFORMATION);
#endif /* WITH_DEBUG */
LocalFree(lpMsgBuf);
}
/*****************************************************************************/
void
mi_paint_rect(char * data, int width, int height, int x, int y, int cx, int cy)
{
HBITMAP bitmap;
BITMAPINFO bi;
HDC dc;
HDC maindc;
HGDIOBJ save;
HRGN rgn;
void * bits;
int i;
int j;
int colour;
int red;
int green;
int blue;
int index;
ZeroMemory(&bi, sizeof(bi));
bi.bmiHeader.biSize = sizeof(bi.bmiHeader);
bi.bmiHeader.biWidth = width;
bi.bmiHeader.biHeight = -height;
bi.bmiHeader.biPlanes = 1;
bi.bmiHeader.biBitCount = 32;
bi.bmiHeader.biCompression = BI_RGB;
maindc = GetWindowDC(g_Wnd);
bitmap = CreateDIBSection(maindc, &bi, DIB_RGB_COLORS, (void **) &bits, 0, 0);
if (bitmap == 0)
{
mi_show_error("CreateDIBSection failed");
}
if (g_server_depth == 8)
{
for (i = cy - 1; i >= 0; i--)
{
for (j = cx - 1; j >= 0; j--)
{
colour = ((unsigned char*)data)[i * cx + j];
red = (pal_entries[colour & 0xff] & 0xff0000) >> 16;
green = (pal_entries[colour & 0xff] & 0xff00) >> 8;
blue = pal_entries[colour & 0xff] & 0xff;
MAKE_COLOUR32(colour, red, green, blue);
((unsigned int*)bits)[i * cx + j] = colour;
}
}
}
else if (g_server_depth == 15)
{
for (i = cy - 1; i >= 0; i--)
{
for (j = cx - 1; j >= 0; j--)
{
colour = ((unsigned short*)data)[i * cx + j];
SPLIT_COLOUR15(colour, red, green, blue);
MAKE_COLOUR32(colour, red, green, blue);
((unsigned int*)bits)[i * cx + j] = colour;
}
}
}
else if (g_server_depth == 16)
{
for (i = cy - 1; i >= 0; i--)
{
for (j = cx - 1; j >= 0; j--)
{
colour = ((unsigned short*)data)[i * cx + j];
SPLIT_COLOUR16(colour, red, green, blue);
MAKE_COLOUR32(colour, red, green, blue);
((unsigned int*)bits)[i * cx + j] = colour;
}
}
}
else if (g_server_depth == 24)
{
for (i = cy - 1; i >= 0; i--)
{
for (j = cx - 1; j >= 0; j--)
{
index = (i * cx + j) * 3;
red = ((unsigned char*)data)[index + 2];
green = ((unsigned char*)data)[index + 1];
blue = ((unsigned char*)data)[index];
MAKE_COLOUR32(colour, red, green, blue);
((unsigned int*)bits)[i * cx + j] = colour;
}
}
}
else if (g_server_depth == 32)
{
memcpy(bits, data, cx*cy*4);
}
dc = CreateCompatibleDC(maindc);
if (dc == 0)
{
mi_show_error("CreateCompatibleDC failed");
}
save = SelectObject(dc, bitmap);
rgn = mi_clip(maindc);
BitBlt(maindc, x + g_xoff - g_xscroll, y + g_yoff - g_yscroll, cx, cy, dc,
0, 0, SRCCOPY);
SelectObject(dc, save);
DeleteObject(bitmap);
DeleteDC(dc);
ReleaseDC(g_Wnd, maindc);
DeleteObject(rgn);
}
static INT
GetPortNumber(PCHAR szAddress)
{
PCHAR szPort;
INT iPort = TCP_PORT_RDP;
szPort = strtok(szAddress, ":");
if (szPort != NULL)
{
szPort = strtok(NULL, ":");
if (szPort != NULL)
{
iPort = atoi(szPort);
if (iPort <= 0 || iPort > 0xFFFF)
iPort = TCP_PORT_RDP;
}
}
return iPort;
}
static VOID
SetDomainAndUsername(PCHAR pName)
{
PCHAR pDomain;
PCHAR pUsername;
strcpy(g_domain, "");
strcpy(g_username, "");
pDomain = strtok(pName, "\\");
if(pDomain == NULL)
return;
pUsername = strtok(NULL, "\\");
if(pUsername == NULL)
{
strcpy(g_username, pDomain);
return;
}
strcpy(g_username, pUsername);
strcpy(g_domain, pDomain);
return;
}
static BOOL
ParseCommandLine(LPWSTR lpCmdLine,
PRDPSETTINGS pRdpSettings,
BOOL *bSkipDlg)
{
LPWSTR lpStr = lpCmdLine;
WCHAR szSeps[] = L"/";
LPWSTR lpToken;
BOOL bRet = TRUE;
*bSkipDlg = TRUE;
if (*lpCmdLine != L'/')
{
LoadRdpSettingsFromFile(pRdpSettings, lpCmdLine);
}
else
{
/* default to screen size, 16bpp */
SetIntegerToSettings(pRdpSettings, L"session bpp", 16);
SetIntegerToSettings(pRdpSettings, L"desktopwidth", GetSystemMetrics(SM_CXSCREEN));
SetIntegerToSettings(pRdpSettings, L"desktopheight", GetSystemMetrics(SM_CYSCREEN));
lpToken = wcstok(lpStr, szSeps);
while (lpToken)
{
if (wcsncmp(lpToken, L"edit", 4) == 0)
{
lpToken += 5;
LoadRdpSettingsFromFile(pRdpSettings, lpToken);
*bSkipDlg = FALSE;
break;
}
if (*lpToken == L'v')
{
lpToken += 2;
SetStringToSettings(pRdpSettings, L"full address", lpToken);
}
else if (*lpToken == L'w')
{
lpToken += 2;
SetIntegerToSettings(pRdpSettings, L"desktopwidth", _wtoi(lpToken));
}
else if (*lpToken == L'h')
{
lpToken += 2;
SetIntegerToSettings(pRdpSettings, L"desktopheight", _wtoi(lpToken));
}
else if (*lpToken == L'f')
{
SetIntegerToSettings(pRdpSettings, L"screen mode id", 2);
}
lpToken = wcstok(NULL, szSeps);
}
}
return bRet;
}
/*****************************************************************************/
int WINAPI
wWinMain(HINSTANCE hInstance,
HINSTANCE hPrevInstance,
LPWSTR lpCmdLine,
int nCmdShow)
{
PRDPSETTINGS pRdpSettings;
WSADATA d;
int ret = 1;
if (WSAStartup(MAKEWORD(2, 0), &d) == 0)
{
pRdpSettings = HeapAlloc(GetProcessHeap(),
0,
sizeof(RDPSETTINGS));
if (pRdpSettings)
{
pRdpSettings->pSettings = NULL;
pRdpSettings->NumSettings = 0;
if (InitRdpSettings(pRdpSettings))
{
BOOL bSkipDlg = FALSE;
if (*lpCmdLine)
ParseCommandLine(lpCmdLine, pRdpSettings,&bSkipDlg);
else
LoadRdpSettingsFromFile(pRdpSettings, NULL);
if (bSkipDlg || OpenRDPConnectDialog(hInstance,
pRdpSettings))
{
char szValue[MAXVALUE];
DWORD dwSize = MAXVALUE;
uni_to_str(szValue, GetStringFromSettings(pRdpSettings, L"full address"));
/* GetPortNumber also removes possible trailing port number from address */
g_tcp_port_rdp = GetPortNumber(szValue);
strcpy(g_servername, szValue);
uni_to_str(szValue, GetStringFromSettings(pRdpSettings, L"username"));
SetDomainAndUsername(szValue);
strcpy(g_password, "");
if (GetComputerNameA(szValue, &dwSize))
strcpy(g_hostname, szValue);
else
strcpy(g_hostname, tcp_get_address());
g_server_depth = GetIntegerFromSettings(pRdpSettings, L"session bpp");
g_screen_width = GetSystemMetrics(SM_CXSCREEN);
g_screen_height = GetSystemMetrics(SM_CYSCREEN);
g_width = GetIntegerFromSettings(pRdpSettings, L"desktopwidth");
g_height = GetIntegerFromSettings(pRdpSettings, L"desktopheight");
if (GetIntegerFromSettings(pRdpSettings, L"screen mode id") == 2)
{
g_fullscreen = 1;
g_xoff = 0;
g_yoff = 0;
}
else
{
g_xoff = GetSystemMetrics(SM_CXEDGE) * 2;
g_yoff = GetSystemMetrics(SM_CYCAPTION) + GetSystemMetrics(SM_CYEDGE) * 2;
}
ui_main();
ret = 0;
}
HeapFree(GetProcessHeap(),
0,
pRdpSettings->pSettings);
}
HeapFree(GetProcessHeap(),
0,
pRdpSettings);
}
WSACleanup();
}
return ret;
}
/*****************************************************************************/
void
mi_begin_update(void)
{
}
/*****************************************************************************/
void
mi_end_update(void)
{
}
/*****************************************************************************/
void
mi_fill_rect(int x, int y, int cx, int cy, int colour)
{
HBRUSH brush;
RECT rect;
HDC maindc;
HRGN rgn;
int red;
int green;
int blue;
if (g_server_depth == 8)
{
red = (pal_entries[colour & 0xff] & 0xff0000) >> 16;
green = (pal_entries[colour & 0xff] & 0xff00) >> 8;
blue = pal_entries[colour & 0xff] & 0xff;
}
else if (g_server_depth == 15)
{
SPLIT_COLOUR15(colour, red, green, blue);
}
else if (g_server_depth == 16)
{
SPLIT_COLOUR16(colour, red, green, blue);
}
else if (g_server_depth == 24 || g_server_depth == 32)
{
red = (colour>>16)&0xff;
green = (colour>>8)&0xff;
blue = colour&0xff;
}
maindc = GetWindowDC(g_Wnd);
rgn = mi_clip(maindc);
brush = CreateSolidBrush(RGB(red, green, blue));
rect.left = x + g_xoff - g_xscroll;
rect.top = y + g_yoff - g_yscroll;
rect.right = rect.left + cx;
rect.bottom = rect.top + cy;
FillRect(maindc, &rect, brush);
DeleteObject(brush);
ReleaseDC(g_Wnd, maindc);
DeleteObject(rgn);
}
/*****************************************************************************/
void
mi_line(int x1, int y1, int x2, int y2, int colour)
{
HPEN pen;
HDC maindc;
HGDIOBJ save;
HRGN rgn;
int red;
int green;
int blue;
if (g_server_depth == 8)
{
red = (pal_entries[colour & 0xff] & 0xff0000) >> 16;
green = (pal_entries[colour & 0xff] & 0xff00) >> 8;
blue = pal_entries[colour & 0xff] & 0xff;
}
else if (g_server_depth == 15)
{
SPLIT_COLOUR15(colour, red, green, blue);
}
else if (g_server_depth == 16)
{
SPLIT_COLOUR16(colour, red, green, blue);
}
else if (g_server_depth == 24 || g_server_depth == 32)
{
red = (colour>>16)&0xff;
green = (colour>>8)&0xff;
blue = colour&0xff;
}
maindc = GetWindowDC(g_Wnd);
rgn = mi_clip(maindc);
pen = CreatePen(PS_SOLID, 0, RGB(red, green, blue));
save = SelectObject(maindc, pen);
MoveToEx(maindc, x1 + g_xoff - g_xscroll, y1 + g_yoff - g_yscroll, 0);
LineTo(maindc, x2 + g_xoff - g_xscroll, y2 + g_yoff - g_yscroll);
SelectObject(maindc, save);
DeleteObject(pen);
ReleaseDC(g_Wnd, maindc);
DeleteObject(rgn);
}
/*****************************************************************************/
void
mi_screen_copy(int x, int y, int cx, int cy, int srcx, int srcy)
{
RECT rect;
RECT clip_rect;
RECT draw_rect;
HRGN rgn;
int ok_to_ScrollWindowEx;
ok_to_ScrollWindowEx = 1;
if (!ok_to_ScrollWindowEx)
{
rgn = CreateRectRgn(x - g_xscroll, y - g_yscroll,
(x - g_xscroll) + cx,
(y - g_yscroll) + cy);
InvalidateRgn(g_Wnd, rgn, 0);
DeleteObject(rgn);
}
else
{
/* this is all in client coords */
rect.left = srcx - g_xscroll;
rect.top = srcy - g_yscroll;
rect.right = rect.left + cx;
rect.bottom = rect.top + cy;
clip_rect.left = g_clip_left - g_xscroll;
clip_rect.top = g_clip_top - g_yscroll;
clip_rect.right = g_clip_right - g_xscroll;
clip_rect.bottom = g_clip_bottom - g_yscroll;
if (IntersectRect(&draw_rect, &clip_rect, &g_wnd_clip))
{
rgn = CreateRectRgn(0, 0, 0, 0);
ScrollWindowEx(g_Wnd, x - srcx, y - srcy, &rect, &draw_rect,
rgn, 0, SW_ERASE);
InvalidateRgn(g_Wnd, rgn, 0);
DeleteObject(rgn);
}
}
}
/*****************************************************************************/
void
mi_set_clip(int x, int y, int cx, int cy)
{
g_clip_left = x;
g_clip_top = y;
g_clip_right = g_clip_left + cx;
g_clip_bottom = g_clip_top + cy;
}
/*****************************************************************************/
void
mi_reset_clip(void)
{
g_clip_left = 0;
g_clip_top = 0;
g_clip_right = g_clip_left + g_width;
g_clip_bottom = g_clip_top + g_height;
}
/*****************************************************************************/
void *
mi_create_cursor(unsigned int x, unsigned int y,
int width, int height,
unsigned char * andmask, unsigned char * xormask)
{
HCURSOR hCur;
hCur = CreateCursor(g_Instance, x, y, width, height, andmask, xormask);
if (hCur == 0)
{
hCur = LoadCursor(NULL, IDC_ARROW);
}
return hCur;
}
/*****************************************************************************/
void
mi_destroy_cursor(void * cursor)
{
if (g_cursor == cursor)
{
g_cursor = 0;
}
DestroyCursor(cursor);
}
/*****************************************************************************/
void
mi_set_cursor(void * cursor)
{
g_cursor = cursor;
SetCursor(g_cursor);
}
/*****************************************************************************/
void
mi_set_null_cursor(void)
{
}