12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502 |
- /* Copyright (C) 2002-2004 RealVNC Ltd. All Rights Reserved.
- *
- * This 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 software 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 software; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
- * USA.
- */
- #define WIN32_LEAN_AND_MEAN
- #if (_WIN32_WINNT < 0x0400)
- #define _WIN32_WINNT 0x0400
- #endif
- #include <windows.h>
- #include <winsock2.h>
- #include <tchar.h>
- #include <commctrl.h>
-
- #include <network/TcpSocket.h>
-
- #include <vncviewer/CView.h>
- #include <vncviewer/UserPasswdDialog.h>
- #include <vncviewer/resource.h>
-
- #include <rfb/encodings.h>
- #include <rfb/secTypes.h>
- #include <rfb/CSecurityNone.h>
- #include <rfb/CSecurityVncAuth.h>
- #include <rfb/CMsgWriter.h>
- #include <rfb/Configuration.h>
- #include <rfb/LogWriter.h>
-
- #include <rfb_win32/WMShatter.h>
-
- using namespace rfb;
- using namespace rfb::win32;
- using namespace rdr;
-
- // - Statics & consts
-
- static LogWriter vlog("CView");
-
- const int IDM_FULLSCREEN = 1;
- const int IDM_SEND_MENU_KEY = 2;
- const int IDM_SEND_CAD = 3;
- const int IDM_ABOUT = 4;
- const int IDM_OPTIONS = 5;
- const int IDM_INFO = 6;
- const int IDM_NEWCONN = 7;
- const int IDM_REQUEST_REFRESH = 9;
- const int IDM_CTRL_KEY = 10;
- const int IDM_ALT_KEY = 11;
-
- const int TIMER_BUMPSCROLL = 1;
- const int TIMER_POINTER_INTERVAL = 2;
- const int TIMER_POINTER_3BUTTON = 3;
-
-
- IntParameter debugDelay("DebugDelay","Milliseconds to display inverted "
- "pixel data - a debugging feature", 0);
-
-
- //
- // -=- CViewClass
-
- //
- // Window class used as the basis for all CView instances
- //
-
- class CViewClass {
- public:
- CViewClass();
- ~CViewClass();
- ATOM classAtom;
- HINSTANCE instance;
- };
-
- LRESULT CALLBACK CViewProc(HWND wnd, UINT msg, WPARAM wParam, LPARAM lParam) {
- LRESULT result;
-
- // *** vlog.debug("CViewMsg %x->(%x, %x, %x)", wnd, msg, wParam, lParam);
-
- if (msg == WM_CREATE)
- SetWindowLong(wnd, GWL_USERDATA, (long)((CREATESTRUCT*)lParam)->lpCreateParams);
- else if (msg == WM_DESTROY)
- SetWindowLong(wnd, GWL_USERDATA, 0);
- CView* _this = (CView*) GetWindowLong(wnd, GWL_USERDATA);
- if (!_this) {
- vlog.info("null _this in %x, message %u", wnd, msg);
- return rfb::win32::SafeDefWindowProc(wnd, msg, wParam, lParam);
- }
-
- try {
- result = _this->processMessage(msg, wParam, lParam);
- } catch (rdr::Exception& e) {
- vlog.error("untrapped: %s", e.str());
- }
-
- return result;
- };
-
- HCURSOR dotCursor = (HCURSOR)LoadImage(GetModuleHandle(0), MAKEINTRESOURCE(IDC_DOT_CURSOR), IMAGE_CURSOR, 0, 0, LR_SHARED);
- HCURSOR arrowCursor = (HCURSOR)LoadImage(NULL, IDC_ARROW, IMAGE_CURSOR, 0, 0, LR_SHARED);
-
- CViewClass::CViewClass() : classAtom(0) {
- WNDCLASS wndClass;
- wndClass.style = 0;
- wndClass.lpfnWndProc = CViewProc;
- wndClass.cbClsExtra = 0;
- wndClass.cbWndExtra = 0;
- wndClass.hInstance = instance = GetModuleHandle(0);
- wndClass.hIcon = (HICON)LoadImage(GetModuleHandle(0), MAKEINTRESOURCE(IDI_ICON), IMAGE_ICON, 0, 0, LR_SHARED);
- if (!wndClass.hIcon)
- printf("unable to load icon:%ld", GetLastError());
- wndClass.hCursor = NULL;
- wndClass.hbrBackground = NULL;
- wndClass.lpszMenuName = 0;
- wndClass.lpszClassName = _T("rfb::win32::CViewClass");
- classAtom = RegisterClass(&wndClass);
- if (!classAtom) {
- throw rdr::SystemException("unable to register CView window class", GetLastError());
- }
- }
-
- CViewClass::~CViewClass() {
- if (classAtom) {
- UnregisterClass((const TCHAR*)classAtom, instance);
- }
- }
-
- CViewClass baseClass;
-
-
- //
- // -=- CView instance implementation
- //
-
- RegKey CView::userConfigKey;
-
-
- CView::CView()
- : quit_on_destroy(false), buffer(0), sock(0), readyToRead(false),
- client_size(0, 0, 16, 16), window_size(0, 0, 32, 32),
- cursorVisible(false), cursorAvailable(false), cursorInBuffer(false),
- systemCursorVisible(true), trackingMouseLeave(false),
- hwnd(0), requestUpdate(false), has_focus(false), palette_changed(false),
- sameMachine(false), encodingChange(false), formatChange(false),
- lastUsedEncoding_(encodingRaw), fullScreenActive(false),
- bumpScroll(false), manager(0) {
-
- // Create the window
- const TCHAR* name = _T("VNC Viewer 4.0b");
- hwnd = CreateWindow((const TCHAR*)baseClass.classAtom, name, WS_OVERLAPPEDWINDOW,
- 0, 0, 10, 10, 0, 0, baseClass.instance, this);
- if (!hwnd) {
- throw rdr::SystemException("unable to create WMNotifier window instance", GetLastError());
- }
- vlog.debug("created window \"%s\" (%x)", (const char*)CStr(name), hwnd);
-
- // Initialise the CPointer pointer handler
- ptr.setHWND(getHandle());
- ptr.setIntervalTimerId(TIMER_POINTER_INTERVAL);
- ptr.set3ButtonTimerId(TIMER_POINTER_3BUTTON);
-
- // Initialise the bumpscroll timer
- bumpScrollTimer.setHWND(getHandle());
- bumpScrollTimer.setId(TIMER_BUMPSCROLL);
-
- // Hook the clipboard
- clipboard.setNotifier(this);
-
- // Create the backing buffer
- buffer = new win32::DIBSectionBuffer(getHandle());
- }
-
- CView::~CView() {
- vlog.debug("~CView");
- showSystemCursor();
- if (hwnd) {
- setVisible(false);
- DestroyWindow(hwnd);
- hwnd = 0;
- }
- delete buffer;
- vlog.debug("~CView done");
- }
-
- bool CView::initialise(network::Socket* s) {
- // Update the window menu
- HMENU wndmenu = GetSystemMenu(hwnd, FALSE);
- AppendMenu(wndmenu, MF_SEPARATOR, 0, 0);
- AppendMenu(wndmenu, MF_STRING, IDM_FULLSCREEN, _T("&Full screen"));
- AppendMenu(wndmenu, MF_SEPARATOR, 0, 0);
- AppendMenu(wndmenu, MF_STRING, IDM_CTRL_KEY, _T("Ctr&l"));
- AppendMenu(wndmenu, MF_STRING, IDM_ALT_KEY, _T("Al&t"));
- AppendMenu(wndmenu, MF_STRING, IDM_SEND_CAD, _T("Send Ctrl-Alt-&Del"));
- AppendMenu(wndmenu, MF_STRING, IDM_REQUEST_REFRESH, _T("Refres&h Screen"));
- AppendMenu(wndmenu, MF_SEPARATOR, 0, 0);
- if (manager) AppendMenu(wndmenu, MF_STRING, IDM_NEWCONN, _T("Ne&w Connection..."));
- AppendMenu(wndmenu, MF_STRING, IDM_OPTIONS, _T("&Options..."));
- AppendMenu(wndmenu, MF_STRING, IDM_INFO, _T("Connection &Info..."));
- AppendMenu(wndmenu, MF_STRING, IDM_ABOUT, _T("&About..."));
-
- // Set the server's name for MRU purposes
- CharArray endpoint(s->getPeerEndpoint());
- setServerName(endpoint.buf);
- if (!options.host.buf)
- options.setHost(endpoint.buf);
-
- // Initialise the underlying CConnection
- setStreams(&s->inStream(), &s->outStream());
-
- // Enable processing of window messages while blocked on I/O
- s->inStream().setBlockCallback(this);
-
- // Initialise the viewer options
- applyOptions(options);
-
- // - Set which auth schemes we support
- addSecType(secTypeNone);
- addSecType(secTypeVncAuth);
-
- initialiseProtocol();
- WSAAsyncSelect(s->getFd(), getHandle(), WM_USER, FD_READ | FD_CLOSE);
- sock = s;
-
- return true;
- }
-
-
- void
- CView::applyOptions(CViewOptions& opt) {
- // *** CHANGE THIS TO USE CViewOptions::operator= ***
-
- // - Take the username, password, config filename, and host spec
- options.setUserName(opt.userName.buf);
- options.setPassword(opt.password.buf);
- options.setHost(opt.host.buf);
- options.setConfigFileName(opt.configFileName.buf);
- options.setMonitor(opt.monitor.buf);
-
- // - Set optional features in ConnParams
- encodingChange |= ((options.useLocalCursor != opt.useLocalCursor) ||
- (options.useDesktopResize != opt.useDesktopResize));
- cp.supportsLocalCursor = options.useLocalCursor = opt.useLocalCursor;
- cp.supportsDesktopResize = options.useDesktopResize = opt.useDesktopResize;
-
- encodingChange |= ((options.customCompressLevel != opt.customCompressLevel) ||
- (options.compressLevel != opt.compressLevel) ||
- (options.noJpeg != opt.noJpeg) ||
- (options.qualityLevel != opt.qualityLevel));
- cp.customCompressLevel = options.customCompressLevel = opt.customCompressLevel;
- cp.compressLevel = options.compressLevel = opt.compressLevel;
- cp.noJpeg = options.noJpeg = opt.noJpeg;
- cp.qualityLevel = options.qualityLevel = opt.qualityLevel;
-
- if (cursorAvailable)
- hideLocalCursor();
- cursorAvailable = cursorAvailable && options.useLocalCursor;
-
- // - Switch full-screen mode on/off
- options.fullScreen = opt.fullScreen;
- setFullscreen(options.fullScreen);
-
- // - Handle format/encoding options
- encodingChange |= (options.preferredEncoding != opt.preferredEncoding);
- options.preferredEncoding = opt.preferredEncoding;
-
- formatChange |= (options.fullColour != opt.fullColour);
- options.fullColour = opt.fullColour;
-
- if (!options.fullColour)
- formatChange |= (options.lowColourLevel != opt.lowColourLevel);
- options.lowColourLevel = opt.lowColourLevel;
-
- options.autoSelect = opt.autoSelect;
-
- // - Sharing
- options.shared = opt.shared;
- setShared(options.shared);
-
- // - Inputs
- options.sendPtrEvents = opt.sendPtrEvents;
- options.sendKeyEvents = opt.sendKeyEvents;
- options.clientCutText = opt.clientCutText;
- options.serverCutText = opt.serverCutText;
- options.emulate3 = opt.emulate3;
- ptr.enableEmulate3(opt.emulate3);
- options.pointerEventInterval = opt.pointerEventInterval;
- ptr.enableInterval(opt.pointerEventInterval);
- options.menuKey = opt.menuKey;
-
- // - Protocol version override
- options.protocol3_3 = opt.protocol3_3;
- setProtocol3_3(options.protocol3_3);
-
- // - Bell
- options.acceptBell = opt.acceptBell;
- }
-
- void
- CView::setFullscreen(bool fs) {
- // Set the menu fullscreen option tick
- CheckMenuItem(GetSystemMenu(getHandle(), FALSE), IDM_FULLSCREEN,
- (options.fullScreen ? MF_CHECKED : 0) | MF_BYCOMMAND);
-
- // If the window is not visible then we ignore the request.
- // setVisible() will call us to correct the full-screen state when
- // the window is visible, to keep things consistent.
- if (!IsWindowVisible(getHandle()))
- return;
-
- if (fs && !fullScreenActive) {
- fullScreenActive = bumpScroll = true;
-
- // Un-minimize the window if required
- if (GetWindowLong(getHandle(), GWL_STYLE) & WS_MINIMIZE)
- ShowWindow(getHandle(), SW_RESTORE);
-
- // Save the non-fullscreen window position
- RECT wrect;
- GetWindowRect(getHandle(), &wrect);
- fullScreenOldRect = Rect(wrect.left, wrect.top, wrect.right, wrect.bottom);
-
- // Find the size of the display the window is on
- MonitorInfo mi(getHandle());
-
- // Set the window full-screen
- DWORD flags = GetWindowLong(getHandle(), GWL_STYLE);
- fullScreenOldFlags = flags;
- flags = flags & ~(WS_CAPTION | WS_THICKFRAME | WS_MAXIMIZE | WS_MINIMIZE);
- vlog.debug("flags=%x", flags);
-
- SetWindowLong(getHandle(), GWL_STYLE, flags);
- SetWindowPos(getHandle(), HWND_TOP, mi.rcMonitor.left, mi.rcMonitor.top,
- mi.rcMonitor.right-mi.rcMonitor.left,
- mi.rcMonitor.bottom-mi.rcMonitor.top,
- SWP_FRAMECHANGED);
- } else if (!fs && fullScreenActive) {
- fullScreenActive = bumpScroll = false;
-
- // Set the window non-fullscreen
- SetWindowLong(getHandle(), GWL_STYLE, fullScreenOldFlags);
- SetWindowPos(getHandle(), HWND_NOTOPMOST,
- fullScreenOldRect.tl.x, fullScreenOldRect.tl.y,
- fullScreenOldRect.width(), fullScreenOldRect.height(),
- SWP_FRAMECHANGED);
- }
-
- // Adjust the viewport offset to cope with change in size between FS
- // and previous window state.
- setViewportOffset(scrolloffset);
- }
-
-
- bool CView::setViewportOffset(const Point& tl) {
- /* ***
- Point np = Point(max(0, min(maxscrolloffset.x, tl.x)),
- max(0, min(maxscrolloffset.y, tl.y)));
- */
- Point np = Point(max(0, min(tl.x, buffer->width()-client_size.width())),
- max(0, min(tl.y, buffer->height()-client_size.height())));
- Point delta = np.translate(scrolloffset.negate());
- if (!np.equals(scrolloffset)) {
- scrolloffset = np;
- ScrollWindowEx(getHandle(), -delta.x, -delta.y, 0, 0, 0, 0, SW_INVALIDATE);
- UpdateWindow(getHandle());
- return true;
- }
- return false;
- }
-
-
- bool CView::processBumpScroll(const Point& pos)
- {
- if (!bumpScroll) return false;
- int bumpScrollPixels = 20;
- bumpScrollDelta = Point();
-
- if (pos.x == client_size.width()-1)
- bumpScrollDelta.x = bumpScrollPixels;
- else if (pos.x == 0)
- bumpScrollDelta.x = -bumpScrollPixels;
- if (pos.y == client_size.height()-1)
- bumpScrollDelta.y = bumpScrollPixels;
- else if (pos.y == 0)
- bumpScrollDelta.y = -bumpScrollPixels;
-
- if (bumpScrollDelta.x || bumpScrollDelta.y) {
- if (bumpScrollTimer.isActive()) return true;
- if (setViewportOffset(scrolloffset.translate(bumpScrollDelta))) {
- bumpScrollTimer.start(25);
- return true;
- }
- }
-
- bumpScrollTimer.stop();
- return false;
- }
-
-
- LRESULT
- CView::processMessage(UINT msg, WPARAM wParam, LPARAM lParam) {
- switch (msg) {
-
- // -=- Process standard window messages
-
- case WM_DISPLAYCHANGE:
- // Display has changed - use new pixel format
- calculateFullColourPF();
- break;
-
- case WM_PAINT:
- {
- PAINTSTRUCT ps;
- HDC paintDC = BeginPaint(getHandle(), &ps);
- if (!paintDC)
- throw SystemException("unable to BeginPaint", GetLastError());
- Rect pr = Rect(ps.rcPaint.left, ps.rcPaint.top, ps.rcPaint.right, ps.rcPaint.bottom);
-
- if (!pr.is_empty()) {
-
- // Draw using the correct palette
- PaletteSelector pSel(paintDC, windowPalette.getHandle());
-
- if (buffer->bitmap) {
- // Update the bitmap's palette
- if (palette_changed) {
- palette_changed = false;
- buffer->refreshPalette();
- }
-
- // Get device context
- BitmapDC bitmapDC(paintDC, buffer->bitmap);
-
- // Blit the border if required
- Rect bufpos = bufferToClient(buffer->getRect());
- if (!pr.enclosed_by(bufpos)) {
- vlog.debug("draw border");
- HBRUSH black = (HBRUSH) GetStockObject(BLACK_BRUSH);
- RECT r;
- SetRect(&r, 0, 0, bufpos.tl.x, client_size.height()); FillRect(paintDC, &r, black);
- SetRect(&r, bufpos.tl.x, 0, bufpos.br.x, bufpos.tl.y); FillRect(paintDC, &r, black);
- SetRect(&r, bufpos.br.x, 0, client_size.width(), client_size.height()); FillRect(paintDC, &r, black);
- SetRect(&r, bufpos.tl.x, bufpos.br.y, bufpos.br.x, client_size.height()); FillRect(paintDC, &r, black);
- }
-
- // Do the blit
- Point buf_pos = clientToBuffer(pr.tl);
- if (!BitBlt(paintDC, pr.tl.x, pr.tl.y, pr.width(), pr.height(),
- bitmapDC, buf_pos.x, buf_pos.y, SRCCOPY))
- throw SystemException("unable to BitBlt to window", GetLastError());
-
- } else {
- // Blit a load of black
- if (!BitBlt(paintDC, pr.tl.x, pr.tl.y, pr.width(), pr.height(),
- 0, 0, 0, BLACKNESS))
- throw SystemException("unable to BitBlt to blank window", GetLastError());
- }
- }
-
- EndPaint(getHandle(), &ps);
-
- // - Request the next update from the server, if required
- requestNewUpdate();
- }
- return 0;
-
- // -=- Palette management
-
- case WM_PALETTECHANGED:
- vlog.debug("WM_PALETTECHANGED");
- if ((HWND)wParam == getHandle()) {
- vlog.debug("ignoring");
- break;
- }
- case WM_QUERYNEWPALETTE:
- vlog.debug("re-selecting palette");
- {
- WindowDC wdc(getHandle());
- PaletteSelector pSel(wdc, windowPalette.getHandle());
- if (pSel.isRedrawRequired()) {
- InvalidateRect(getHandle(), 0, FALSE);
- UpdateWindow(getHandle());
- }
- }
- return TRUE;
-
- // -=- Window position
-
- // Prevent the window from being resized to be too large if in normal mode.
- // If maximized or fullscreen the allow oversized windows.
-
- case WM_WINDOWPOSCHANGING:
- {
- WINDOWPOS* wpos = (WINDOWPOS*)lParam;
- if (wpos->flags & SWP_NOSIZE)
- break;
-
- // Work out how big the window should ideally be
- DWORD current_style = GetWindowLong(getHandle(), GWL_STYLE);
- DWORD style = current_style & ~(WS_VSCROLL | WS_HSCROLL);
- RECT r;
- SetRect(&r, 0, 0, buffer->width(), buffer->height());
- AdjustWindowRect(&r, style, FALSE);
- Rect reqd_size = Rect(r.left, r.top, r.right, r.bottom);
- if (current_style & WS_VSCROLL)
- reqd_size.br.x += GetSystemMetrics(SM_CXVSCROLL);
- if (current_style & WS_HSCROLL)
- reqd_size.br.y += GetSystemMetrics(SM_CXHSCROLL);
- RECT current;
- GetWindowRect(getHandle(), ¤t);
-
- // Ensure that the window isn't resized too large
- // If the window is maximized or full-screen then any size is allowed
- if (!(GetWindowLong(getHandle(), GWL_STYLE) & WS_MAXIMIZE) && !fullScreenActive) {
- if (wpos->cx > reqd_size.width()) {
- wpos->cx = reqd_size.width();
- wpos->x = current.left;
- }
- if (wpos->cy > reqd_size.height()) {
- wpos->cy = reqd_size.height();
- wpos->y = current.top;
- }
- }
-
- }
- break;
-
- // Add scrollbars if required and update window size info we have cached.
-
- case WM_SIZE:
- {
- Point old_offset = bufferToClient(Point(0, 0));
-
- // Update the cached sizing information
- RECT r;
- GetWindowRect(getHandle(), &r);
- window_size = Rect(r.left, r.top, r.right, r.bottom);
- GetClientRect(getHandle(), &r);
- client_size = Rect(r.left, r.top, r.right, r.bottom);
-
- // Determine whether scrollbars are required
- calculateScrollBars();
-
- // Redraw if required
- if (!old_offset.equals(bufferToClient(Point(0, 0))))
- InvalidateRect(getHandle(), 0, TRUE);
- }
- break;
-
- case WM_VSCROLL:
- case WM_HSCROLL:
- {
- Point delta;
- int newpos = (msg == WM_VSCROLL) ? scrolloffset.y : scrolloffset.x;
-
- switch (LOWORD(wParam)) {
- case SB_PAGEUP: newpos -= 50; break;
- case SB_PAGEDOWN: newpos += 50; break;
- case SB_LINEUP: newpos -= 5; break;
- case SB_LINEDOWN: newpos += 5; break;
- case SB_THUMBTRACK:
- case SB_THUMBPOSITION: newpos = HIWORD(wParam); break;
- default: vlog.info("received unknown scroll message");
- };
-
- if (msg == WM_HSCROLL)
- setViewportOffset(Point(newpos, scrolloffset.y));
- else
- setViewportOffset(Point(scrolloffset.x, newpos));
-
- SCROLLINFO si;
- si.cbSize = sizeof(si);
- si.fMask = SIF_POS;
- si.nPos = newpos;
- SetScrollInfo(getHandle(), (msg == WM_VSCROLL) ? SB_VERT : SB_HORZ, &si, TRUE);
- }
- break;
-
- // -=- Bump-scrolling
-
- case WM_TIMER:
- switch (wParam) {
- case TIMER_BUMPSCROLL:
- if (!setViewportOffset(scrolloffset.translate(bumpScrollDelta)))
- bumpScrollTimer.stop();
- break;
- case TIMER_POINTER_INTERVAL:
- case TIMER_POINTER_3BUTTON:
- try {
- ptr.handleTimer(writer(), wParam);
- } catch (rdr::Exception& e) {
- close(e.str());
- }
- break;
- }
- break;
-
- // -=- Cursor shape/visibility handling
-
- case WM_SETCURSOR:
- if (LOWORD(lParam) != HTCLIENT)
- break;
- SetCursor(cursorInBuffer ? dotCursor : arrowCursor);
- return TRUE;
-
- case WM_MOUSELEAVE:
- trackingMouseLeave = false;
- cursorOutsideBuffer();
- return 0;
-
- // -=- Mouse input handling
-
- case WM_MOUSEMOVE:
- case WM_LBUTTONUP:
- case WM_MBUTTONUP:
- case WM_RBUTTONUP:
- case WM_LBUTTONDOWN:
- case WM_MBUTTONDOWN:
- case WM_RBUTTONDOWN:
- case WM_MOUSEWHEEL:
- if (has_focus)
- {
- if (!trackingMouseLeave) {
- TRACKMOUSEEVENT tme;
- tme.cbSize = sizeof(TRACKMOUSEEVENT);
- tme.dwFlags = TME_LEAVE;
- tme.hwndTrack = hwnd;
- _TrackMouseEvent(&tme);
- trackingMouseLeave = true;
- }
- int mask = 0;
- if (LOWORD(wParam) & MK_LBUTTON) mask |= 1;
- if (LOWORD(wParam) & MK_MBUTTON) mask |= 2;
- if (LOWORD(wParam) & MK_RBUTTON) mask |= 4;
-
- if (msg == WM_MOUSEWHEEL) {
- int delta = (short)HIWORD(wParam);
- int repeats = (abs(delta)+119) / 120;
- int wheelMask = (delta > 0) ? 8 : 16;
- vlog.debug("repeats %d, mask %d\n",repeats,wheelMask);
- for (int i=0; i<repeats; i++) {
- writePointerEvent(oldpos.x, oldpos.y, mask | wheelMask);
- writePointerEvent(oldpos.x, oldpos.y, mask);
- }
- } else {
- Point clientPos = Point(LOWORD(lParam), HIWORD(lParam));
- Point p = clientToBuffer(clientPos);
-
- // If the mouse is not within the server buffer area, do nothing
- cursorInBuffer = buffer->getRect().contains(p);
- if (!cursorInBuffer) {
- cursorOutsideBuffer();
- break;
- }
-
- // If we're locally rendering the cursor then redraw it
- if (cursorAvailable) {
- // - Render the cursor!
- if (!p.equals(cursorPos)) {
- hideLocalCursor();
- cursorPos = p;
- showLocalCursor();
- if (cursorVisible)
- hideSystemCursor();
- }
- }
-
- // If we are doing bump-scrolling then try that first...
- if (processBumpScroll(clientPos))
- break;
-
- // Send a pointer event to the server
- writePointerEvent(p.x, p.y, mask);
- oldpos = p;
- }
- } else {
- cursorOutsideBuffer();
- }
- break;
-
- // -=- Track whether or not the window has focus
-
- case WM_SETFOCUS:
- has_focus = true;
- break;
- case WM_KILLFOCUS:
- has_focus = false;
- cursorOutsideBuffer();
- // Restore the remote keys to consistent states
- try {
- kbd.releaseAllKeys(writer());
- } catch (rdr::Exception& e) {
- close(e.str());
- }
- break;
-
- // -=- Handle the extra window menu items
-
- // Process the items added to the system menu
- case WM_SYSCOMMAND:
-
- // - First check whether it's one of our messages
- switch (wParam) {
- case IDM_FULLSCREEN:
- options.fullScreen = !options.fullScreen;
- setFullscreen(options.fullScreen);
- return 0;
- case IDM_CTRL_KEY:
- writeKeyEvent(VK_CONTROL, 0, !kbd.keyPressed(VK_CONTROL));
- return 0;
- case IDM_ALT_KEY:
- writeKeyEvent(VK_MENU, 0, !kbd.keyPressed(VK_MENU));
- return 0;
- case IDM_SEND_MENU_KEY:
- writeKeyEvent(options.menuKey, 0, true);
- writeKeyEvent(options.menuKey, 0, false);
- return 0;
- case IDM_SEND_CAD:
- writeKeyEvent(VK_CONTROL, 0, true);
- writeKeyEvent(VK_MENU, 0, true);
- writeKeyEvent(VK_DELETE, 0, true);
- writeKeyEvent(VK_DELETE, 0, false);
- writeKeyEvent(VK_MENU, 0, false);
- writeKeyEvent(VK_CONTROL, 0, false);
- return 0;
- case IDM_REQUEST_REFRESH:
- try {
- writer()->writeFramebufferUpdateRequest(Rect(0,0,cp.width,cp.height), false);
- requestUpdate = false;
- } catch (rdr::Exception& e) {
- close(e.str());
- }
- return 0;
- case IDM_NEWCONN:
- manager->addClient(0);
- return 0;
- case IDM_OPTIONS:
- // Update the monitor device name in the CViewOptions instance
- {
- MonitorInfo mi(getHandle());
- options.setMonitor(mi.szDevice);
- optionsDialog.showDialog(this);
- return 0;
- }
- case IDM_INFO:
- infoDialog.showDialog(this);
- return 0;
- case IDM_ABOUT:
- AboutDialog::instance.showDialog();
- return 0;
- };
-
- // - Not one of our messages, so process it as a system message
- switch (wParam & 0xfff0) {
-
- // When restored, ensure that full-screen mode is re-enabled if required.
- case SC_RESTORE:
- rfb::win32::SafeDefWindowProc(getHandle(), msg, wParam, lParam);
- setFullscreen(options.fullScreen);
- return 0;
-
- // If we are maximized or minimized then that cancels full-screen mode.
- case SC_MINIMIZE:
- case SC_MAXIMIZE:
- setFullscreen(false);
- break;
-
- // If the system menu is shown then make sure it's up to date
- case SC_KEYMENU:
- case SC_MOUSEMENU:
- updateF8Menu(false);
- break;
-
- };
- break;
-
- // Treat all menu commands as system menu commands
- case WM_COMMAND:
- SendMessage(getHandle(), WM_SYSCOMMAND, wParam, lParam);
- return 0;
-
- case WM_MENUCHAR:
- vlog.debug("menuchar");
- break;
-
- // -=- Handle keyboard input
-
- case WM_KEYUP:
- case WM_KEYDOWN:
- // Hook the MenuKey to pop-up the window menu
- if (options.menuKey && (wParam == options.menuKey)) {
-
- bool ctrlDown = (GetAsyncKeyState(VK_CONTROL) & 0x8000) != 0;
- bool altDown = (GetAsyncKeyState(VK_MENU) & 0x8000) != 0;
- bool shiftDown = (GetAsyncKeyState(VK_SHIFT) & 0x8000) != 0;
- if (!(ctrlDown || altDown || shiftDown)) {
-
- // If MenuKey is being released then pop-up the menu
- if ((msg == WM_KEYDOWN)) {
- // Make sure it's up to date
- updateF8Menu(true);
-
- // Show it under the pointer
- POINT pt;
- GetCursorPos(&pt);
- cursorInBuffer = false;
- TrackPopupMenu(GetSystemMenu(getHandle(), FALSE),
- TPM_CENTERALIGN | TPM_VCENTERALIGN, pt.x, pt.y, 0, getHandle(), 0);
- }
-
- // Ignore the MenuKey keypress for both press & release events
- return 0;
- }
- }
- case WM_SYSKEYDOWN:
- case WM_SYSKEYUP:
- writeKeyEvent(wParam, lParam, (msg == WM_KEYDOWN) || (msg == WM_SYSKEYDOWN));
- return 0;
-
- // -=- Handle the window closing
-
- case WM_CLOSE:
- vlog.debug("WM_CLOSE %x", getHandle());
- if (quit_on_destroy) {
- vlog.debug("posting WM_QUIT");
- PostQuitMessage(0);
- } else {
- vlog.debug("not posting WM_QUIT");
- }
- break;
-
- // -=- Process incoming socket data
-
- case WM_USER:
- readyToRead = true;
- break;
-
- }
-
- return rfb::win32::SafeDefWindowProc(getHandle(), msg, wParam, lParam);
- }
-
- void CView::blockCallback() {
- // - An InStream has blocked on I/O while processing an RFB message
- // We re-enable socket event notifications, so we'll know when more
- // data is available, then we sit and dispatch window events until
- // the notification arrives.
- readyToRead = false;
- WSAAsyncSelect(sock->getFd(), getHandle(), WM_USER, FD_READ | FD_CLOSE);
- MSG msg;
- while (true) {
- if (readyToRead) {
- // - Network event notification. Return control to I/O routine.
- WSAAsyncSelect(sock->getFd(), getHandle(), WM_USER, 0);
- return;
- }
-
- DWORD result = GetMessage(&msg, NULL, 0, 0);
- if (result == 0) {
- vlog.debug("WM_QUIT");
- throw QuitMessage(msg.wParam);
- } else if (result < 0) {
- throw rdr::SystemException("GetMessage error", GetLastError());
- }
-
- // IMPORTANT: We mustn't call TranslateMessage() here, because instead we
- // call ToAscii() in CKeyboard::keyEvent(). ToAscii() stores dead key
- // state from one call to the next, which would be messed up by calls to
- // TranslateMessage() (actually it looks like TranslateMessage() calls
- // ToAscii() internally).
- DispatchMessage(&msg);
- }
- }
-
-
- void
- CView::hideLocalCursor() {
- // - Blit the cursor backing store over the cursor
- // *** ALWAYS call this BEFORE changing buffer PF!!!
- if (cursorVisible) {
- cursorVisible = false;
- buffer->imageRect(cursorBackingRect, cursorBacking.data);
- invalidateBufferRect(cursorBackingRect);
- }
- }
-
- void
- CView::showLocalCursor() {
- if (cursorAvailable && !cursorVisible && cursorInBuffer) {
- if (!cp.pf().equal(cursor.getPF()) ||
- cursor.getRect().is_empty()) {
- vlog.info("attempting to render invalid local cursor");
- cursorAvailable = false;
- showSystemCursor();
- return;
- }
- cursorVisible = true;
-
- cursorBackingRect = cursor.getRect().translate(cursorPos).translate(cursor.hotspot.negate());
- cursorBackingRect = cursorBackingRect.intersect(buffer->getRect());
- buffer->getImage(cursorBacking.data, cursorBackingRect);
-
- renderLocalCursor();
-
- invalidateBufferRect(cursorBackingRect);
- }
- }
-
- void CView::cursorOutsideBuffer()
- {
- cursorInBuffer = false;
- hideLocalCursor();
- showSystemCursor();
- }
-
- void
- CView::renderLocalCursor()
- {
- Rect r = cursor.getRect();
- r = r.translate(cursorPos).translate(cursor.hotspot.negate());
- buffer->maskRect(r, cursor.data, cursor.mask.buf);
- }
-
- void
- CView::hideSystemCursor() {
- if (systemCursorVisible) {
- vlog.debug("hide system cursor");
- systemCursorVisible = false;
- ShowCursor(FALSE);
- }
- }
-
- void
- CView::showSystemCursor() {
- if (!systemCursorVisible) {
- vlog.debug("show system cursor");
- systemCursorVisible = true;
- ShowCursor(TRUE);
- }
- }
-
-
- bool
- CView::invalidateBufferRect(const Rect& crect) {
- Rect rect = bufferToClient(crect);
- if (rect.intersect(client_size).is_empty()) return false;
- RECT invalid = {rect.tl.x, rect.tl.y, rect.br.x, rect.br.y};
- InvalidateRect(getHandle(), &invalid, FALSE);
- return true;
- }
-
-
- void
- CView::notifyClipboardChanged(const char* text, int len) {
- if (!options.clientCutText) return;
- if (state() != RFBSTATE_NORMAL) return;
- try {
- writer()->writeClientCutText(text, len);
- } catch (rdr::Exception& e) {
- close(e.str());
- }
- }
-
-
- CSecurity* CView::getCSecurity(int secType)
- {
- switch (secType) {
- case secTypeNone:
- return new CSecurityNone();
- case secTypeVncAuth:
- return new CSecurityVncAuth(this);
- default:
- throw Exception("Unsupported secType?");
- }
- }
-
-
- void
- CView::setColourMapEntries(int first, int count, U16* rgbs) {
- vlog.debug("setColourMapEntries: first=%d, count=%d", first, count);
- int i;
- for (i=0;i<count;i++) {
- buffer->setColour(i+first, rgbs[i*3], rgbs[i*3+1], rgbs[i*3+2]);
- }
- // *** change to 0, 256?
- refreshWindowPalette(first, count);
- palette_changed = true;
- InvalidateRect(getHandle(), 0, FALSE);
- }
-
- void
- CView::bell() {
- if (options.acceptBell)
- MessageBeep(-1);
- }
-
-
- void
- CView::setDesktopSize(int w, int h) {
- vlog.debug("setDesktopSize %dx%d", w, h);
-
- // If the locally-rendered cursor is visible then remove it
- hideLocalCursor();
-
- // Resize the backing buffer
- buffer->setSize(w, h);
-
- // If the window is not maximised or full-screen then resize it
- if (!(GetWindowLong(getHandle(), GWL_STYLE) & WS_MAXIMIZE) && !fullScreenActive) {
- // Resize the window to the required size
- RECT r = {0, 0, w, h};
- AdjustWindowRect(&r, GetWindowLong(getHandle(), GWL_STYLE), FALSE);
- SetWindowPos(getHandle(), 0, 0, 0, r.right-r.left, r.bottom-r.top,
- SWP_NOZORDER | SWP_NOMOVE | SWP_NOACTIVATE | SWP_NOOWNERZORDER);
-
- // Move the window to the desired monitor
- if (options.monitor.buf)
- moveToMonitor(getHandle(), options.monitor.buf);
-
- // Clip to the system work area
- centerWindow(getHandle(), 0, true);
- } else {
- // Ensure the screen contents are consistent
- InvalidateRect(getHandle(), 0, FALSE);
- }
-
- // Tell the underlying CConnection
- CConnection::setDesktopSize(w, h);
-
- // Enable/disable scrollbars as appropriate
- calculateScrollBars();
- }
-
- void
- CView::setCursor(const Point& hotspot, const Point& size, void* data, void* mask) {
- if (!options.useLocalCursor) return;
- hideLocalCursor();
-
- cursor.hotspot = hotspot;
-
- cursor.setSize(size.x, size.y);
- cursor.setPF(cp.pf());
- cursor.imageRect(cursor.getRect(), data);
- memcpy(cursor.mask.buf, mask, cursor.maskLen());
- cursor.crop();
-
- cursorBacking.setSize(size.x, size.y);
- cursorBacking.setPF(cp.pf());
-
- cursorAvailable = true;
-
- showLocalCursor();
- }
-
- PixelFormat
- CView::getNativePF() const {
- vlog.debug("getNativePF()");
- return WindowDC(getHandle()).getPF();
- }
-
- void
- CView::setVisible(bool visible) {
- ShowWindow(getHandle(), visible ? SW_SHOW : SW_HIDE);
- if (visible) {
- // When the window becomes visible, make it active
- SetForegroundWindow(getHandle());
- SetActiveWindow(getHandle());
- // If the window should be full-screen, then do so
- setFullscreen(options.fullScreen);
- } else {
- // Disable full-screen mode
- setFullscreen(false);
- }
- }
-
- void
- CView::close(const char* reason) {
- setVisible(false);
- if (reason) {
- vlog.info("closing - %s", reason);
- MsgBox(NULL, TStr(reason), MB_ICONINFORMATION | MB_OK);
- }
- SendMessage(getHandle(), WM_CLOSE, 0, 0);
- }
-
-
- void
- CView::framebufferUpdateEnd() {
- if (debugDelay != 0) {
- vlog.debug("debug delay %d",(int)debugDelay);
- UpdateWindow(getHandle());
- Sleep(debugDelay);
- std::list<rfb::Rect>::iterator i;
- for (i = debugRects.begin(); i != debugRects.end(); i++) {
- invertRect(*i);
- }
- debugRects.clear();
- }
- if (options.autoSelect)
- autoSelectFormatAndEncoding();
-
- // Always request the next update
- requestUpdate = true;
-
- // Check that at least part of the window has changed
- if (!GetUpdateRect(getHandle(), 0, FALSE)) {
- if (!(GetWindowLong(getHandle(), GWL_STYLE) & WS_MINIMIZE))
- requestNewUpdate();
- }
-
- showLocalCursor();
- }
-
-
- // Note: The method below is duplicated in vncviewer_unix/CConn.cxx!
-
- // autoSelectFormatAndEncoding() chooses the format and encoding appropriate
- // to the connection speed:
- //
- // Above 16Mbps (timing for at least a second), same machine, switch to raw
- // Above 3Mbps, switch to hextile
- // Otherwise, switch to Tight
- //
- // Above 256Kbps, use full colour mode
- //
- void
- CView::autoSelectFormatAndEncoding() {
- int kbitsPerSecond = sock->inStream().kbitsPerSecond();
- unsigned int newEncoding = options.preferredEncoding;
- bool newFullColour = options.fullColour;
- unsigned int timeWaited = sock->inStream().timeWaited();
-
- // Select best encoding
- if (kbitsPerSecond > 16000 && sameMachine && timeWaited >= 10000) {
- newEncoding = encodingRaw;
- } else if (kbitsPerSecond > 3000 && timeWaited >= 10000) {
- newEncoding = encodingHextile;
- } else {
- newEncoding = encodingTight;
- }
-
- if (newEncoding != options.preferredEncoding) {
- vlog.info("Throughput %d kbit/s - changing to %s encoding",
- kbitsPerSecond, encodingName(newEncoding));
- options.preferredEncoding = newEncoding;
- encodingChange = true;
- }
-
- if (kbitsPerSecond == 0) {
- return;
- }
-
- if (cp.majorVersion <= 3 && cp.minorVersion <= 7) {
- // Xvnc from TightVNC 1.2.9 sends out FramebufferUpdates with
- // cursors "asynchronously". If this happens in the middle of a
- // pixel format change, the server will encode the cursor with
- // the old format, but the client will try to decode it
- // according to the new format. This will lead to a
- // crash. Therefore, we do not allow automatic format change for
- // old servers.
- return;
- }
-
- // Select best color level
- newFullColour = (kbitsPerSecond > 256);
- if (newFullColour != options.fullColour) {
- vlog.info("Throughput %d kbit/s - full color is now %s",
- kbitsPerSecond,
- newFullColour ? "enabled" : "disabled");
- options.fullColour = newFullColour;
- formatChange = true;
- }
- }
-
- void
- CView::requestNewUpdate() {
- if (!requestUpdate) return;
-
- if (formatChange) {
- // Hide the rendered cursor, if any, to prevent
- // the backing buffer being used in the wrong format
- hideLocalCursor();
-
- // Select the required pixel format
- if (options.fullColour) {
- buffer->setPF(fullColourPF);
- } else {
- switch (options.lowColourLevel) {
- case 0:
- buffer->setPF(PixelFormat(8,3,0,1,1,1,1,2,1,0));
- break;
- case 1:
- buffer->setPF(PixelFormat(8,6,0,1,3,3,3,4,2,0));
- break;
- case 2:
- buffer->setPF(PixelFormat(8,8,0,0,0,0,0,0,0,0));
- break;
- }
- }
-
- // Print the current pixel format
- char str[256];
- buffer->getPF().print(str, 256);
- vlog.info("Using pixel format %s",str);
-
- // Save the connection pixel format and tell server to use it
- cp.setPF(buffer->getPF());
- writer()->writeSetPixelFormat(cp.pf());
-
- // Correct the local window's palette
- if (!getNativePF().trueColour)
- refreshWindowPalette(0, 1 << cp.pf().depth);
- }
-
- if (encodingChange) {
- vlog.info("Using %s encoding",encodingName(options.preferredEncoding));
- writer()->writeSetEncodings(options.preferredEncoding, true);
- }
-
- writer()->writeFramebufferUpdateRequest(Rect(0, 0, cp.width, cp.height),
- !formatChange);
-
- encodingChange = formatChange = requestUpdate = false;
- }
-
-
- void
- CView::writeKeyEvent(rdr::U8 vkey, rdr::U32 flags, bool down) {
- if (!options.sendKeyEvents) return;
- try {
- kbd.keyEvent(writer(), vkey, flags, down);
- } catch (rdr::Exception& e) {
- close(e.str());
- }
- }
-
- void
- CView::writePointerEvent(int x, int y, int buttonMask) {
- if (!options.sendPtrEvents) return;
- try {
- ptr.pointerEvent(writer(), x, y, buttonMask);
- } catch (rdr::Exception& e) {
- close(e.str());
- }
- }
-
-
- void
- CView::refreshWindowPalette(int start, int count) {
- vlog.debug("refreshWindowPalette(%d, %d)", start, count);
-
- Colour colours[256];
- if (count > 256) {
- vlog.debug("%d palette entries", count);
- throw rdr::Exception("too many palette entries");
- }
-
- // Copy the palette from the DIBSectionBuffer
- ColourMap* cm = buffer->getColourMap();
- if (!cm) return;
- for (int i=0; i<count; i++) {
- int r, g, b;
- cm->lookup(i, &r, &g, &b);
- colours[i].r = r;
- colours[i].g = g;
- colours[i].b = b;
- }
-
- // Set the window palette
- windowPalette.setEntries(start, count, colours);
-
- // Cause the window to be redrawn
- InvalidateRect(getHandle(), 0, 0);
- }
-
-
- void CView::calculateScrollBars() {
- // Calculate the required size of window
- DWORD current_style = GetWindowLong(getHandle(), GWL_STYLE);
- DWORD style = current_style & ~(WS_VSCROLL | WS_HSCROLL);
- DWORD old_style;
- RECT r;
- SetRect(&r, 0, 0, buffer->width(), buffer->height());
- AdjustWindowRect(&r, style, FALSE);
- Rect reqd_size = Rect(r.left, r.top, r.right, r.bottom);
-
- if (!bumpScroll) {
- // We only enable scrollbars if bump-scrolling is not active.
- // Effectively, this means if full-screen is not active,
- // but I think it's better to make these things explicit.
-
- // Work out whether scroll bars are required
- do {
- old_style = style;
-
- if (!(style & WS_HSCROLL) && (reqd_size.width() > window_size.width())) {
- style |= WS_HSCROLL;
- reqd_size.br.y += GetSystemMetrics(SM_CXHSCROLL);
- }
- if (!(style & WS_VSCROLL) && (reqd_size.height() > window_size.height())) {
- style |= WS_VSCROLL;
- reqd_size.br.x += GetSystemMetrics(SM_CXVSCROLL);
- }
- } while (style != old_style);
- }
-
- // Tell Windows to update the window style & cached settings
- if (style != current_style) {
- SetWindowLong(getHandle(), GWL_STYLE, style);
- SetWindowPos(getHandle(), NULL, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE | SWP_NOZORDER | SWP_FRAMECHANGED);
- }
-
- // Update the scroll settings
- SCROLLINFO si;
- if (style & WS_VSCROLL) {
- si.cbSize = sizeof(si);
- si.fMask = SIF_RANGE | SIF_PAGE | SIF_POS;
- si.nMin = 0;
- si.nMax = buffer->height();
- si.nPage = buffer->height() - (reqd_size.height() - window_size.height());
- maxscrolloffset.y = max(0, si.nMax-si.nPage);
- scrolloffset.y = min(maxscrolloffset.y, scrolloffset.y);
- si.nPos = scrolloffset.y;
- SetScrollInfo(getHandle(), SB_VERT, &si, TRUE);
- }
- if (style & WS_HSCROLL) {
- si.cbSize = sizeof(si);
- si.fMask = SIF_RANGE | SIF_PAGE | SIF_POS;
- si.nMin = 0;
- si.nMax = buffer->width();
- si.nPage = buffer->width() - (reqd_size.width() - window_size.width());
- maxscrolloffset.x = max(0, si.nMax-si.nPage);
- scrolloffset.x = min(maxscrolloffset.x, scrolloffset.x);
- si.nPos = scrolloffset.x;
- SetScrollInfo(getHandle(), SB_HORZ, &si, TRUE);
- }
- }
-
-
- void
- CView::calculateFullColourPF() {
- // If the server is palette based then use palette locally
- // Also, don't bother doing bgr222
- if (!serverDefaultPF.trueColour || (serverDefaultPF.depth < 6)) {
- fullColourPF = serverDefaultPF;
- options.fullColour = true;
- } else {
- // If server is trueColour, use lowest depth PF
- PixelFormat native = getNativePF();
- if ((serverDefaultPF.bpp < native.bpp) ||
- ((serverDefaultPF.bpp == native.bpp) &&
- (serverDefaultPF.depth < native.depth)))
- fullColourPF = serverDefaultPF;
- else
- fullColourPF = getNativePF();
- }
- formatChange = true;
- }
-
-
- void
- CView::updateF8Menu(bool hideSystemCommands) {
- HMENU menu = GetSystemMenu(getHandle(), FALSE);
-
- if (hideSystemCommands) {
- // Gray out menu items that might cause a World Of Pain
- HMENU menu = GetSystemMenu(getHandle(), FALSE);
- EnableMenuItem(menu, SC_SIZE, MF_BYCOMMAND | MF_GRAYED);
- EnableMenuItem(menu, SC_MOVE, MF_BYCOMMAND | MF_GRAYED);
- EnableMenuItem(menu, SC_RESTORE, MF_BYCOMMAND | MF_ENABLED);
- EnableMenuItem(menu, SC_MINIMIZE, MF_BYCOMMAND | MF_ENABLED);
- EnableMenuItem(menu, SC_MAXIMIZE, MF_BYCOMMAND | MF_ENABLED);
- }
-
- // Update the modifier key menu items
- UINT ctrlCheckFlags = kbd.keyPressed(VK_CONTROL) ? MF_CHECKED : MF_UNCHECKED;
- UINT altCheckFlags = kbd.keyPressed(VK_MENU) ? MF_CHECKED : MF_UNCHECKED;
- CheckMenuItem(menu, IDM_CTRL_KEY, MF_BYCOMMAND | ctrlCheckFlags);
- CheckMenuItem(menu, IDM_ALT_KEY, MF_BYCOMMAND | altCheckFlags);
-
- // Ensure that the Send <MenuKey> menu item has the correct text
- if (options.menuKey) {
- TCharArray menuKeyStr(options.menuKeyName());
- TCharArray tmp(_tcslen(menuKeyStr.buf) + 6);
- _stprintf(tmp.buf, _T("Send %s"), menuKeyStr.buf);
- if (!ModifyMenu(menu, IDM_SEND_MENU_KEY, MF_BYCOMMAND | MF_STRING, IDM_SEND_MENU_KEY, tmp.buf))
- InsertMenu(menu, IDM_SEND_CAD, MF_BYCOMMAND | MF_STRING, IDM_SEND_MENU_KEY, tmp.buf);
- } else {
- RemoveMenu(menu, IDM_SEND_MENU_KEY, MF_BYCOMMAND);
- }
- }
-
-
- void
- CView::setName(const char* name) {
- vlog.debug("setName %s", name);
- ::SetWindowText(getHandle(), TStr(name));
- CConnection::setName(name);
- }
-
-
- void CView::serverInit() {
- CConnection::serverInit();
-
- // Save the server's current format
- serverDefaultPF = cp.pf();
-
- // Calculate the full-colour format to use
- calculateFullColourPF();
-
- // Request the initial update
- vlog.info("requesting initial update");
- formatChange = encodingChange = requestUpdate = true;
- requestNewUpdate();
-
- // Show the window
- setVisible(true);
- }
-
- void
- CView::serverCutText(const char* str, int len) {
- if (!options.serverCutText) return;
- CharArray t(len+1);
- memcpy(t.buf, str, len);
- t.buf[len] = 0;
- clipboard.setClipText(t.buf);
- }
-
-
- void CView::beginRect(const Rect& r, unsigned int encoding) {
- sock->inStream().startTiming();
- }
-
- void CView::endRect(const Rect& r, unsigned int encoding) {
- sock->inStream().stopTiming();
- lastUsedEncoding_ = encoding;
- if (debugDelay != 0) {
- invertRect(r);
- debugRects.push_back(r);
- }
- }
-
- void CView::fillRect(const Rect& r, Pixel pix) {
- if (cursorBackingRect.overlaps(r)) hideLocalCursor();
- buffer->fillRect(r, pix);
- invalidateBufferRect(r);
- }
- void CView::imageRect(const Rect& r, void* pixels) {
- if (cursorBackingRect.overlaps(r)) hideLocalCursor();
- buffer->imageRect(r, pixels);
- invalidateBufferRect(r);
- }
- void CView::copyRect(const Rect& r, int srcX, int srcY) {
- if (cursorBackingRect.overlaps(r) ||
- cursorBackingRect.overlaps(Rect(srcX, srcY, srcX+r.width(), srcY+r.height())))
- hideLocalCursor();
- buffer->copyRect(r, Point(r.tl.x-srcX, r.tl.y-srcY));
- invalidateBufferRect(r);
- }
-
- void CView::invertRect(const Rect& r) {
- int stride;
- rdr::U8* p = buffer->getPixelsRW(r, &stride);
- for (int y = 0; y < r.height(); y++) {
- for (int x = 0; x < r.width(); x++) {
- switch (buffer->getPF().bpp) {
- case 8: ((rdr::U8* )p)[x+y*stride] ^= 0xff; break;
- case 16: ((rdr::U16*)p)[x+y*stride] ^= 0xffff; break;
- case 32: ((rdr::U32*)p)[x+y*stride] ^= 0xffffffff; break;
- }
- }
- }
- invalidateBufferRect(r);
- }
-
- bool CView::getUserPasswd(char** user, char** password) {
- if (user && options.userName.buf)
- *user = strDup(options.userName.buf);
- if (password && options.password.buf)
- *password = strDup(options.password.buf);
- if ((user && !*user) || (password && !*password)) {
- // Missing username or password - prompt the user
- UserPasswdDialog userPasswdDialog;
- userPasswdDialog.setCSecurity(getCurrentCSecurity());
- if (!userPasswdDialog.getUserPasswd(user, password))
- return false;
- }
- if (user) options.setUserName(*user);
- if (password) options.setPassword(*password);
- return true;
- }
|