blob: a60a72779c5062ae3dee40e1e64a58d4c81beb3e [file] [log] [blame]
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +00001/* Copyright (C) 2002-2005 RealVNC Ltd. All Rights Reserved.
Constantin Kaplinsky2c019832008-05-30 11:02:04 +00002 * Copyright (C) 2004-2008 Constantin Kaplinsky. All Rights Reserved.
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +00003 *
4 * This is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
8 *
9 * This software is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this software; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
17 * USA.
18 */
19
20// FIXME: Check cases when screen width/height is not a multiply of 32.
21// e.g. 800x600.
22
23#include <strings.h>
24#include <sys/types.h>
25#include <sys/stat.h>
26#include <unistd.h>
27#include <errno.h>
28#include <rfb/Logger_stdio.h>
29#include <rfb/LogWriter.h>
30#include <rfb/VNCServerST.h>
31#include <rfb/Configuration.h>
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +000032#include <rfb/Timer.h>
33#include <network/TcpSocket.h>
34#include <tx/TXWindow.h>
35
Constantin Kaplinskya3b60c42006-06-02 04:07:49 +000036#include <vncconfig/QueryConnectDialog.h>
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +000037
38#include <signal.h>
39#include <X11/X.h>
40#include <X11/Xlib.h>
41#include <X11/Xutil.h>
Peter Korsgaard5b7ff372017-07-13 00:36:01 +020042#include <X11/XKBlib.h>
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +000043#ifdef HAVE_XTEST
44#include <X11/extensions/XTest.h>
45#endif
Pierre Ossmana7b728a2014-06-13 10:56:59 +000046#ifdef HAVE_XDAMAGE
47#include <X11/extensions/Xdamage.h>
48#endif
Alan Coopersmith9a739d42017-08-05 15:16:29 -070049#ifdef HAVE_XFIXES
50#include <X11/extensions/Xfixes.h>
51#endif
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +000052
53#include <x0vncserver/Geometry.h>
54#include <x0vncserver/Image.h>
Constantin Kaplinsky614c7b52007-12-26 18:17:09 +000055#include <x0vncserver/XPixelBuffer.h>
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +000056#include <x0vncserver/PollingScheduler.h>
57
Peter Korsgaard8fe50902017-07-13 00:35:57 +020058extern const unsigned short code_map_qnum_to_xorgevdev[];
59extern const unsigned int code_map_qnum_to_xorgevdev_len;
60
61extern const unsigned short code_map_qnum_to_xorgkbd[];
62extern const unsigned int code_map_qnum_to_xorgkbd_len;
63
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +000064// XXX Lynx/OS 2.3: protos for select(), bzero()
65#ifdef Lynx
66#include <sys/proto.h>
67#endif
68
Constantin Kaplinskyb9632702006-12-01 10:54:55 +000069extern char buildtime[];
70
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +000071using namespace rfb;
72using namespace network;
73
Peter Korsgaard5b7ff372017-07-13 00:36:01 +020074// number of XKb indicator leds to handle
75static const int N_LEDS = 3;
76
77// order is important as it must match RFB extension
78static const char * ledNames[N_LEDS] = {
79 "Scroll Lock", "Num Lock", "Caps Lock"
80};
81
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +000082static LogWriter vlog("Main");
83
84IntParameter pollingCycle("PollingCycle", "Milliseconds per one polling "
85 "cycle; actual interval may be dynamically "
86 "adjusted to satisfy MaxProcessorUsage setting", 30);
87IntParameter maxProcessorUsage("MaxProcessorUsage", "Maximum percentage of "
88 "CPU time to be consumed", 35);
89BoolParameter useShm("UseSHM", "Use MIT-SHM extension if available", true);
90BoolParameter useOverlay("OverlayMode", "Use overlay mode under "
91 "IRIX or Solaris", true);
92StringParameter displayname("display", "The X display", "");
93IntParameter rfbport("rfbport", "TCP port to listen for RFB protocol",5900);
94IntParameter queryConnectTimeout("QueryConnectTimeout",
95 "Number of seconds to show the Accept Connection dialog before "
96 "rejecting the connection",
97 10);
98StringParameter hostsFile("HostsFile", "File with IP access control rules", "");
Peter Korsgaard8fe50902017-07-13 00:35:57 +020099BoolParameter rawKeyboard("RawKeyboard",
100 "Send keyboard events straight through and "
101 "avoid mapping them to the current keyboard "
102 "layout", false);
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000103
104//
105// Allow the main loop terminate itself gracefully on receiving a signal.
106//
107
108static bool caughtSignal = false;
109
110static void CleanupSignalHandler(int sig)
111{
112 caughtSignal = true;
113}
114
115
116class QueryConnHandler : public VNCServerST::QueryConnectionHandler,
117 public QueryResultCallback {
118public:
119 QueryConnHandler(Display* dpy, VNCServerST* vs)
120 : display(dpy), server(vs), queryConnectDialog(0), queryConnectSock(0) {}
121 ~QueryConnHandler() { delete queryConnectDialog; }
122
123 // -=- VNCServerST::QueryConnectionHandler interface
124 virtual VNCServerST::queryResult queryConnection(network::Socket* sock,
125 const char* userName,
126 char** reason) {
127 if (queryConnectSock) {
Adam Tkacd36b6262009-09-04 10:57:20 +0000128 *reason = strDup("Another connection is currently being queried.");
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000129 return VNCServerST::REJECT;
130 }
131 if (!userName) userName = "(anonymous)";
132 queryConnectSock = sock;
133 CharArray address(sock->getPeerAddress());
134 delete queryConnectDialog;
135 queryConnectDialog = new QueryConnectDialog(display, address.buf,
136 userName, queryConnectTimeout,
137 this);
138 queryConnectDialog->map();
139 return VNCServerST::PENDING;
140 }
141
142 // -=- QueryResultCallback interface
143 virtual void queryApproved() {
144 server->approveConnection(queryConnectSock, true, 0);
145 queryConnectSock = 0;
146 }
147 virtual void queryRejected() {
148 server->approveConnection(queryConnectSock, false,
149 "Connection rejected by local user");
150 queryConnectSock = 0;
151 }
152private:
153 Display* display;
154 VNCServerST* server;
155 QueryConnectDialog* queryConnectDialog;
156 network::Socket* queryConnectSock;
157};
158
159
Pierre Ossmanb6b4dc62014-01-20 15:05:21 +0100160class XDesktop : public SDesktop, public TXGlobalEventHandler
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000161{
162public:
163 XDesktop(Display* dpy_, Geometry *geometry_)
Constantin Kaplinsky303433a2008-06-04 05:57:06 +0000164 : dpy(dpy_), geometry(geometry_), pb(0), server(0),
Pierre Ossmana7b728a2014-06-13 10:56:59 +0000165 oldButtonMask(0), haveXtest(false), haveDamage(false),
Pierre Ossman61a75ef2017-09-15 14:23:34 +0200166 maxButtons(0), running(false), ledMasks(), ledState(0),
167 codeMap(0), codeMapLen(0)
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000168 {
Peter Korsgaard5b7ff372017-07-13 00:36:01 +0200169 int major, minor;
170
171 int xkbOpcode, xkbErrorBase;
172
173 major = XkbMajorVersion;
174 minor = XkbMinorVersion;
175 if (!XkbQueryExtension(dpy, &xkbOpcode, &xkbEventBase,
176 &xkbErrorBase, &major, &minor)) {
177 vlog.error("XKEYBOARD extension not present");
178 throw Exception();
179 }
180
181 XkbSelectEvents(dpy, XkbUseCoreKbd, XkbIndicatorStateNotifyMask,
182 XkbIndicatorStateNotifyMask);
183
184 // figure out bit masks for the indicators we are interested in
185 for (int i = 0; i < N_LEDS; i++) {
186 Atom a;
187 int shift;
188 Bool on;
189
190 a = XInternAtom(dpy, ledNames[i], True);
191 if (!a || !XkbGetNamedIndicator(dpy, a, &shift, &on, NULL, NULL))
192 continue;
193
194 ledMasks[i] = 1u << shift;
195 vlog.debug("Mask for '%s' is 0x%x", ledNames[i], ledMasks[i]);
196 if (on)
197 ledState |= 1u << i;
198 }
199
Peter Korsgaard8fe50902017-07-13 00:35:57 +0200200 // X11 unfortunately uses keyboard driver specific keycodes and provides no
201 // direct way to query this, so guess based on the keyboard mapping
202 XkbDescPtr desc = XkbGetKeyboard(dpy, XkbAllComponentsMask, XkbUseCoreKbd);
203 if (desc && desc->names) {
204 char *keycodes = XGetAtomName(dpy, desc->names->keycodes);
205
206 if (keycodes) {
207 if (strncmp("evdev", keycodes, strlen("evdev")) == 0) {
208 codeMap = code_map_qnum_to_xorgevdev;
209 codeMapLen = code_map_qnum_to_xorgevdev_len;
210 vlog.info("Using evdev codemap\n");
211 } else if (strncmp("xfree86", keycodes, strlen("xfree86")) == 0) {
212 codeMap = code_map_qnum_to_xorgkbd;
213 codeMapLen = code_map_qnum_to_xorgkbd_len;
214 vlog.info("Using xorgkbd codemap\n");
215 } else {
216 vlog.info("Unknown keycode '%s', no codemap\n", keycodes);
217 }
218 XFree(keycodes);
219 } else {
220 vlog.debug("Unable to get keycode map\n");
221 }
222
223 XkbFreeKeyboard(desc, XkbAllComponentsMask, True);
224 }
225
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000226#ifdef HAVE_XTEST
227 int xtestEventBase;
228 int xtestErrorBase;
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000229
230 if (XTestQueryExtension(dpy, &xtestEventBase,
231 &xtestErrorBase, &major, &minor)) {
232 XTestGrabControl(dpy, True);
233 vlog.info("XTest extension present - version %d.%d",major,minor);
234 haveXtest = true;
235 } else {
236#endif
237 vlog.info("XTest extension not present");
238 vlog.info("Unable to inject events or display while server is grabbed");
239#ifdef HAVE_XTEST
240 }
241#endif
242
Pierre Ossmana7b728a2014-06-13 10:56:59 +0000243#ifdef HAVE_XDAMAGE
244 int xdamageErrorBase;
245
246 if (XDamageQueryExtension(dpy, &xdamageEventBase, &xdamageErrorBase)) {
Pierre Ossmana7b728a2014-06-13 10:56:59 +0000247 haveDamage = true;
248 } else {
249#endif
250 vlog.info("DAMAGE extension not present");
251 vlog.info("Will have to poll screen for changes");
252#ifdef HAVE_XDAMAGE
253 }
254#endif
Peter Korsgaard5b7ff372017-07-13 00:36:01 +0200255
Alan Coopersmith9a739d42017-08-05 15:16:29 -0700256#ifdef HAVE_XFIXES
257 int xfixesErrorBase;
258
259 if (XFixesQueryExtension(dpy, &xfixesEventBase, &xfixesErrorBase)) {
Pierre Ossman61a75ef2017-09-15 14:23:34 +0200260 XFixesSelectCursorInput(dpy, DefaultRootWindow(dpy),
261 XFixesDisplayCursorNotifyMask);
Alan Coopersmith9a739d42017-08-05 15:16:29 -0700262 } else {
263#endif
264 vlog.info("XFIXES extension not present");
265 vlog.info("Will not be able to display cursors");
266#ifdef HAVE_XFIXES
267 }
268#endif
Pierre Ossmand092f052017-09-15 13:10:54 +0200269
Peter Korsgaard5b7ff372017-07-13 00:36:01 +0200270 TXWindow::setGlobalEventHandler(this);
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000271 }
272 virtual ~XDesktop() {
273 stop();
274 }
275
Constantin Kaplinskydc873dc2008-06-04 09:46:57 +0000276 inline void poll() {
Pierre Ossmana7b728a2014-06-13 10:56:59 +0000277 if (pb and not haveDamage)
Constantin Kaplinskydc873dc2008-06-04 09:46:57 +0000278 pb->poll(server);
Pierre Ossman00dec782017-09-15 14:52:21 +0200279 if (running) {
280 Window root, child;
281 int x, y, wx, wy;
282 unsigned int mask;
283 XQueryPointer(dpy, DefaultRootWindow(dpy), &root, &child,
284 &x, &y, &wx, &wy, &mask);
285 server->setCursorPos(rfb::Point(x, y));
286 }
Constantin Kaplinskydc873dc2008-06-04 09:46:57 +0000287 }
288
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000289 // -=- SDesktop interface
290
291 virtual void start(VNCServer* vs) {
292
293 // Determine actual number of buttons of the X pointer device.
294 unsigned char btnMap[8];
295 int numButtons = XGetPointerMapping(dpy, btnMap, 8);
296 maxButtons = (numButtons > 8) ? 8 : numButtons;
297 vlog.info("Enabling %d button%s of X pointer device",
298 maxButtons, (maxButtons != 1) ? "s" : "");
299
Constantin Kaplinskye0c80c52008-06-04 03:10:05 +0000300 // Create an ImageFactory instance for producing Image objects.
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000301 ImageFactory factory((bool)useShm, (bool)useOverlay);
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000302
Constantin Kaplinskydc873dc2008-06-04 09:46:57 +0000303 // Create pixel buffer and provide it to the server object.
Pierre Ossmanb6b4dc62014-01-20 15:05:21 +0100304 pb = new XPixelBuffer(dpy, factory, geometry->getRect());
Constantin Kaplinskye0c80c52008-06-04 03:10:05 +0000305 vlog.info("Allocated %s", pb->getImage()->classDesc());
306
Constantin Kaplinskyc341ac62008-08-21 03:35:08 +0000307 server = (VNCServerST *)vs;
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000308 server->setPixelBuffer(pb);
309
Pierre Ossmana7b728a2014-06-13 10:56:59 +0000310#ifdef HAVE_XDAMAGE
311 if (haveDamage) {
312 damage = XDamageCreate(dpy, DefaultRootWindow(dpy),
313 XDamageReportRawRectangles);
314 }
315#endif
316
Peter Korsgaard5b7ff372017-07-13 00:36:01 +0200317 server->setLEDState(ledState);
318
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000319 running = true;
320 }
321
322 virtual void stop() {
323 running = false;
324
Pierre Ossmana7b728a2014-06-13 10:56:59 +0000325#ifdef HAVE_XDAMAGE
326 if (haveDamage)
327 XDamageDestroy(dpy, damage);
328#endif
329
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000330 delete pb;
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000331 pb = 0;
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000332 }
333
334 inline bool isRunning() {
335 return running;
336 }
337
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000338 virtual void pointerEvent(const Point& pos, int buttonMask) {
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000339#ifdef HAVE_XTEST
340 if (!haveXtest) return;
341 XTestFakeMotionEvent(dpy, DefaultScreen(dpy),
342 geometry->offsetLeft() + pos.x,
343 geometry->offsetTop() + pos.y,
344 CurrentTime);
345 if (buttonMask != oldButtonMask) {
346 for (int i = 0; i < maxButtons; i++) {
347 if ((buttonMask ^ oldButtonMask) & (1<<i)) {
348 if (buttonMask & (1<<i)) {
349 XTestFakeButtonEvent(dpy, i+1, True, CurrentTime);
350 } else {
351 XTestFakeButtonEvent(dpy, i+1, False, CurrentTime);
352 }
353 }
354 }
355 }
356 oldButtonMask = buttonMask;
357#endif
358 }
359
Pierre Ossmanac730382017-08-16 15:20:20 +0200360#ifdef HAVE_XTEST
361 KeyCode XkbKeysymToKeycode(Display* dpy, KeySym keysym) {
362 XkbDescPtr xkb;
363 XkbStateRec state;
364 unsigned keycode;
365
366 xkb = XkbGetMap(dpy, XkbAllComponentsMask, XkbUseCoreKbd);
367 if (!xkb)
368 return 0;
369
370 XkbGetState(dpy, XkbUseCoreKbd, &state);
371
372 for (keycode = xkb->min_key_code;
373 keycode <= xkb->max_key_code;
374 keycode++) {
375 KeySym cursym;
376 unsigned int mods;
377 XkbTranslateKeyCode(xkb, keycode, state.compat_state, &mods, &cursym);
378 if (cursym == keysym)
379 break;
380 }
381
382 if (keycode > xkb->max_key_code)
383 keycode = 0;
384
385 XkbFreeKeyboard(xkb, XkbAllComponentsMask, True);
386
387 return keycode;
388 }
389#endif
390
Pierre Ossman5ae28212017-05-16 14:30:38 +0200391 virtual void keyEvent(rdr::U32 keysym, rdr::U32 xtcode, bool down) {
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000392#ifdef HAVE_XTEST
Peter Korsgaard8fe50902017-07-13 00:35:57 +0200393 int keycode = 0;
Pierre Ossmanac730382017-08-16 15:20:20 +0200394
395 if (!haveXtest)
396 return;
Peter Korsgaard8fe50902017-07-13 00:35:57 +0200397
398 // Use scan code if provided and mapping exists
399 if (codeMap && rawKeyboard && xtcode < codeMapLen)
400 keycode = codeMap[xtcode];
401
Pierre Ossman63cc07e2017-09-15 11:17:02 +0200402 if (!keycode) {
403 if (!down || (pressedKeys.find(keysym) != pressedKeys.end()))
404 keycode = pressedKeys[keysym];
Pierre Ossmanac730382017-08-16 15:20:20 +0200405 else {
406 // XKeysymToKeycode() doesn't respect state, so we have to use
407 // something slightly more complex
Pierre Ossman63cc07e2017-09-15 11:17:02 +0200408 keycode = XkbKeysymToKeycode(dpy, keysym);
Pierre Ossmanac730382017-08-16 15:20:20 +0200409 }
Pierre Ossmanac730382017-08-16 15:20:20 +0200410 }
Peter Korsgaard8fe50902017-07-13 00:35:57 +0200411
Pierre Ossman63cc07e2017-09-15 11:17:02 +0200412 if (!keycode)
413 return;
414
415 if (down)
416 pressedKeys[keysym] = keycode;
417 else
418 pressedKeys.erase(keysym);
419
Pierre Ossmanac730382017-08-16 15:20:20 +0200420 XTestFakeKeyEvent(dpy, keycode, down, CurrentTime);
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000421#endif
422 }
423
424 virtual void clientCutText(const char* str, int len) {
425 }
426
427 virtual Point getFbSize() {
428 return Point(pb->width(), pb->height());
429 }
430
Pierre Ossmana7b728a2014-06-13 10:56:59 +0000431 // -=- TXGlobalEventHandler interface
432
433 virtual bool handleGlobalEvent(XEvent* ev) {
Peter Korsgaard5b7ff372017-07-13 00:36:01 +0200434 if (ev->type == xkbEventBase + XkbEventCode) {
435 XkbEvent *kb = (XkbEvent *)ev;
Pierre Ossmana7b728a2014-06-13 10:56:59 +0000436
Peter Korsgaard5b7ff372017-07-13 00:36:01 +0200437 if (kb->any.xkb_type != XkbIndicatorStateNotify)
438 return false;
Pierre Ossmana7b728a2014-06-13 10:56:59 +0000439
Peter Korsgaard5b7ff372017-07-13 00:36:01 +0200440 vlog.debug("Got indicator update, mask is now 0x%x", kb->indicators.state);
441
442 ledState = 0;
443 for (int i = 0; i < N_LEDS; i++) {
444 if (kb->indicators.state & ledMasks[i])
445 ledState |= 1u << i;
446 }
447
448 if (running)
449 server->setLEDState(ledState);
450
Pierre Ossmana7b728a2014-06-13 10:56:59 +0000451 return true;
Peter Korsgaard5b7ff372017-07-13 00:36:01 +0200452#ifdef HAVE_XDAMAGE
453 } else if (ev->type == xdamageEventBase) {
454 XDamageNotifyEvent* dev;
455 Rect rect;
Pierre Ossmana7b728a2014-06-13 10:56:59 +0000456
Peter Korsgaard5b7ff372017-07-13 00:36:01 +0200457 if (!running)
458 return true;
Pierre Ossmana7b728a2014-06-13 10:56:59 +0000459
Peter Korsgaard5b7ff372017-07-13 00:36:01 +0200460 dev = (XDamageNotifyEvent*)ev;
461 rect.setXYWH(dev->area.x, dev->area.y, dev->area.width, dev->area.height);
462 server->add_changed(rect);
463
464 return true;
Pierre Ossmana7b728a2014-06-13 10:56:59 +0000465#endif
Alan Coopersmith9a739d42017-08-05 15:16:29 -0700466#ifdef HAVE_XFIXES
Pierre Ossmand092f052017-09-15 13:10:54 +0200467 } else if (ev->type == xfixesEventBase + XFixesCursorNotify) {
Alan Coopersmith9a739d42017-08-05 15:16:29 -0700468 XFixesCursorNotifyEvent* cev;
469 XFixesCursorImage *cim;
470
471 if (!running)
472 return true;
473
474 cev = (XFixesCursorNotifyEvent*)ev;
475
476 if (cev->subtype != XFixesDisplayCursorNotify)
477 return false;
478
479 cim = XFixesGetCursorImage(dpy);
480 if (cim == NULL)
481 return false;
482
483 // Copied from XserverDesktop::setCursor() in
484 // unix/xserver/hw/vnc/XserverDesktop.cc and adapted to
485 // handle long -> U32 conversion for 64-bit Xlib
486 rdr::U8* cursorData;
487 rdr::U8 *out;
488 const unsigned long *pixels;
489
490 cursorData = new rdr::U8[cim->width * cim->height * 4];
491
492 // Un-premultiply alpha
493 pixels = cim->pixels;
494 out = cursorData;
495 for (int y = 0; y < cim->height; y++) {
496 for (int x = 0; x < cim->width; x++) {
497 rdr::U8 alpha;
498 rdr::U32 pixel = *pixels++;
499 rdr::U8 *in = (rdr::U8 *) &pixel;
500
501 alpha = in[3];
502 if (alpha == 0)
503 alpha = 1; // Avoid division by zero
504
505 *out++ = (unsigned)*in++ * 255/alpha;
506 *out++ = (unsigned)*in++ * 255/alpha;
507 *out++ = (unsigned)*in++ * 255/alpha;
508 *out++ = *in++;
509 }
510 }
511
512 try {
513 server->setCursor(cim->width, cim->height, Point(cim->xhot, cim->yhot),
514 cursorData);
515 } catch (rdr::Exception& e) {
516 vlog.error("XserverDesktop::setCursor: %s",e.str());
517 }
518
519 delete [] cursorData;
520 XFree(cim);
521 return true;
Alan Coopersmith9a739d42017-08-05 15:16:29 -0700522#endif
Peter Korsgaard5b7ff372017-07-13 00:36:01 +0200523 }
524
525 return false;
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000526 }
527
528protected:
529 Display* dpy;
530 Geometry* geometry;
Constantin Kaplinsky2c019832008-05-30 11:02:04 +0000531 XPixelBuffer* pb;
Constantin Kaplinskyc341ac62008-08-21 03:35:08 +0000532 VNCServerST* server;
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000533 int oldButtonMask;
534 bool haveXtest;
Pierre Ossmana7b728a2014-06-13 10:56:59 +0000535 bool haveDamage;
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000536 int maxButtons;
Pierre Ossmanac730382017-08-16 15:20:20 +0200537 std::map<KeySym, KeyCode> pressedKeys;
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000538 bool running;
Pierre Ossmana7b728a2014-06-13 10:56:59 +0000539#ifdef HAVE_XDAMAGE
540 Damage damage;
541 int xdamageEventBase;
542#endif
Peter Korsgaard5b7ff372017-07-13 00:36:01 +0200543 int xkbEventBase;
Alan Coopersmith9a739d42017-08-05 15:16:29 -0700544#ifdef HAVE_XFIXES
545 int xfixesEventBase;
546#endif
Peter Korsgaard5b7ff372017-07-13 00:36:01 +0200547 int ledMasks[N_LEDS];
548 unsigned ledState;
Peter Korsgaard8fe50902017-07-13 00:35:57 +0200549 const unsigned short *codeMap;
550 unsigned codeMapLen;
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000551};
552
553
554class FileTcpFilter : public TcpFilter
555{
556
557public:
558
559 FileTcpFilter(const char *fname)
560 : TcpFilter("-"), fileName(NULL), lastModTime(0)
561 {
562 if (fname != NULL)
563 fileName = strdup((char *)fname);
564 }
565
566 virtual ~FileTcpFilter()
567 {
568 if (fileName != NULL)
569 free(fileName);
570 }
571
572 virtual bool verifyConnection(Socket* s)
573 {
574 if (!reloadRules()) {
575 vlog.error("Could not read IP filtering rules: rejecting all clients");
576 filter.clear();
577 filter.push_back(parsePattern("-"));
578 return false;
579 }
580
581 return TcpFilter::verifyConnection(s);
582 }
583
584protected:
585
586 bool reloadRules()
587 {
588 if (fileName == NULL)
589 return true;
590
591 struct stat st;
592 if (stat(fileName, &st) != 0)
593 return false;
594
595 if (st.st_mtime != lastModTime) {
596 // Actually reload only if the file was modified
597 FILE *fp = fopen(fileName, "r");
598 if (fp == NULL)
599 return false;
600
601 // Remove all the rules from the parent class
602 filter.clear();
603
604 // Parse the file contents adding rules to the parent class
605 char buf[32];
606 while (readLine(buf, 32, fp)) {
607 if (buf[0] && strchr("+-?", buf[0])) {
608 filter.push_back(parsePattern(buf));
609 }
610 }
611
612 fclose(fp);
613 lastModTime = st.st_mtime;
614 }
615 return true;
616 }
617
618protected:
619
620 char *fileName;
621 time_t lastModTime;
622
623private:
624
625 //
626 // NOTE: we silently truncate long lines in this function.
627 //
628
629 bool readLine(char *buf, int bufSize, FILE *fp)
630 {
631 if (fp == NULL || buf == NULL || bufSize == 0)
632 return false;
633
634 if (fgets(buf, bufSize, fp) == NULL)
635 return false;
636
637 char *ptr = strchr(buf, '\n');
638 if (ptr != NULL) {
639 *ptr = '\0'; // remove newline at the end
640 } else {
641 if (!feof(fp)) {
642 int c;
643 do { // skip the rest of a long line
644 c = getc(fp);
645 } while (c != '\n' && c != EOF);
646 }
647 }
648 return true;
649 }
650
651};
652
653char* programName;
654
Constantin Kaplinsky2039d7b2008-06-04 10:43:10 +0000655static void printVersion(FILE *fp)
656{
Peter Ã…strand4eacc022009-02-27 10:12:14 +0000657 fprintf(fp, "TigerVNC Server version %s, built %s\n",
Constantin Kaplinskyea7b6502008-09-28 05:08:48 +0000658 PACKAGE_VERSION, buildtime);
Constantin Kaplinsky2039d7b2008-06-04 10:43:10 +0000659}
660
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000661static void usage()
662{
Constantin Kaplinsky2039d7b2008-06-04 10:43:10 +0000663 printVersion(stderr);
664 fprintf(stderr, "\nUsage: %s [<parameters>]\n", programName);
665 fprintf(stderr, " %s --version\n", programName);
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000666 fprintf(stderr,"\n"
667 "Parameters can be turned on with -<param> or off with -<param>=0\n"
668 "Parameters which take a value can be specified as "
669 "-<param> <value>\n"
670 "Other valid forms are <param>=<value> -<param>=<value> "
671 "--<param>=<value>\n"
672 "Parameter names are case-insensitive. The parameters are:\n\n");
673 Configuration::listParams(79, 14);
674 exit(1);
675}
676
677int main(int argc, char** argv)
678{
679 initStdIOLoggers();
680 LogWriter::setLogParams("*:stderr:30");
681
682 programName = argv[0];
683 Display* dpy;
684
Adam Tkacc58b3d12010-04-23 13:55:10 +0000685 Configuration::enableServerParams();
686
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000687 for (int i = 1; i < argc; i++) {
688 if (Configuration::setParam(argv[i]))
689 continue;
690
691 if (argv[i][0] == '-') {
692 if (i+1 < argc) {
693 if (Configuration::setParam(&argv[i][1], argv[i+1])) {
694 i++;
695 continue;
696 }
697 }
Constantin Kaplinsky2039d7b2008-06-04 10:43:10 +0000698 if (strcmp(argv[i], "-v") == 0 ||
699 strcmp(argv[i], "-version") == 0 ||
700 strcmp(argv[i], "--version") == 0) {
701 printVersion(stdout);
702 return 0;
703 }
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000704 usage();
705 }
706
707 usage();
708 }
709
710 CharArray dpyStr(displayname.getData());
711 if (!(dpy = XOpenDisplay(dpyStr.buf[0] ? dpyStr.buf : 0))) {
Constantin Kaplinsky7bdccd72008-08-20 06:22:28 +0000712 // FIXME: Why not vlog.error(...)?
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000713 fprintf(stderr,"%s: unable to open display \"%s\"\r\n",
Constantin Kaplinsky7bdccd72008-08-20 06:22:28 +0000714 programName, XDisplayName(dpyStr.buf));
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000715 exit(1);
716 }
717
718 signal(SIGHUP, CleanupSignalHandler);
719 signal(SIGINT, CleanupSignalHandler);
720 signal(SIGTERM, CleanupSignalHandler);
721
Pierre Ossmanf7aa3f92015-09-29 15:40:49 +0200722 std::list<TcpListener*> listeners;
Tim Waugh892d10a2015-03-11 13:12:07 +0000723
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000724 try {
725 TXWindow::init(dpy,"x0vncserver");
726 Geometry geo(DisplayWidth(dpy, DefaultScreen(dpy)),
727 DisplayHeight(dpy, DefaultScreen(dpy)));
Constantin Kaplinsky23c60222008-06-04 03:58:07 +0000728 if (geo.getRect().is_empty()) {
729 vlog.error("Exiting with error");
730 return 1;
731 }
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000732 XDesktop desktop(dpy, &geo);
Constantin Kaplinsky82328312008-04-24 08:44:24 +0000733
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000734 VNCServerST server("x0vncserver", &desktop);
735 QueryConnHandler qcHandler(dpy, &server);
736 server.setQueryConnectionHandler(&qcHandler);
737
Tim Waugh892d10a2015-03-11 13:12:07 +0000738 createTcpListeners(&listeners, 0, (int)rfbport);
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000739 vlog.info("Listening on port %d", (int)rfbport);
740
Constantin Kaplinsky7bdccd72008-08-20 06:22:28 +0000741 const char *hostsData = hostsFile.getData();
742 FileTcpFilter fileTcpFilter(hostsData);
743 if (strlen(hostsData) != 0)
Pierre Ossmanf7aa3f92015-09-29 15:40:49 +0200744 for (std::list<TcpListener*>::iterator i = listeners.begin();
Tim Waugh892d10a2015-03-11 13:12:07 +0000745 i != listeners.end();
746 i++)
Pierre Ossmanf7aa3f92015-09-29 15:40:49 +0200747 (*i)->setFilter(&fileTcpFilter);
Constantin Kaplinsky7bdccd72008-08-20 06:22:28 +0000748 delete[] hostsData;
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000749
750 PollingScheduler sched((int)pollingCycle, (int)maxProcessorUsage);
751
752 while (!caughtSignal) {
Pierre Ossman278e4202016-04-29 14:28:54 +0200753 int wait_ms;
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000754 struct timeval tv;
Pierre Ossman16419cc2016-04-29 14:29:43 +0200755 fd_set rfds, wfds;
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000756 std::list<Socket*> sockets;
757 std::list<Socket*>::iterator i;
758
759 // Process any incoming X events
760 TXWindow::handleXEvents(dpy);
761
762 FD_ZERO(&rfds);
Pierre Ossman16419cc2016-04-29 14:29:43 +0200763 FD_ZERO(&wfds);
764
Pierre Ossmana7b728a2014-06-13 10:56:59 +0000765 FD_SET(ConnectionNumber(dpy), &rfds);
Pierre Ossmanf7aa3f92015-09-29 15:40:49 +0200766 for (std::list<TcpListener*>::iterator i = listeners.begin();
Tim Waugh892d10a2015-03-11 13:12:07 +0000767 i != listeners.end();
768 i++)
Pierre Ossmanf7aa3f92015-09-29 15:40:49 +0200769 FD_SET((*i)->getFd(), &rfds);
Tim Waugh892d10a2015-03-11 13:12:07 +0000770
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000771 server.getSockets(&sockets);
772 int clients_connected = 0;
773 for (i = sockets.begin(); i != sockets.end(); i++) {
774 if ((*i)->isShutdown()) {
775 server.removeSocket(*i);
776 delete (*i);
777 } else {
778 FD_SET((*i)->getFd(), &rfds);
Pierre Ossman16419cc2016-04-29 14:29:43 +0200779 if ((*i)->outStream().bufferUsage() > 0)
780 FD_SET((*i)->getFd(), &wfds);
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000781 clients_connected++;
782 }
783 }
784
Constantin Kaplinsky813dbb42006-12-05 08:03:18 +0000785 if (!clients_connected)
786 sched.reset();
787
Pierre Ossman278e4202016-04-29 14:28:54 +0200788 wait_ms = 0;
789
Constantin Kaplinsky813dbb42006-12-05 08:03:18 +0000790 if (sched.isRunning()) {
Pierre Ossman278e4202016-04-29 14:28:54 +0200791 wait_ms = sched.millisRemaining();
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000792 if (wait_ms > 500) {
793 wait_ms = 500;
794 }
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000795 }
Pierre Ossman278e4202016-04-29 14:28:54 +0200796
797 soonestTimeout(&wait_ms, server.checkTimeouts());
798
799 tv.tv_sec = wait_ms / 1000;
800 tv.tv_usec = (wait_ms % 1000) * 1000;
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000801
802 // Do the wait...
803 sched.sleepStarted();
Pierre Ossman16419cc2016-04-29 14:29:43 +0200804 int n = select(FD_SETSIZE, &rfds, &wfds, 0,
Pierre Ossman278e4202016-04-29 14:28:54 +0200805 wait_ms ? &tv : NULL);
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000806 sched.sleepFinished();
807
808 if (n < 0) {
809 if (errno == EINTR) {
810 vlog.debug("Interrupted select() system call");
811 continue;
812 } else {
813 throw rdr::SystemException("select", errno);
814 }
815 }
816
817 // Accept new VNC connections
Pierre Ossmanf7aa3f92015-09-29 15:40:49 +0200818 for (std::list<TcpListener*>::iterator i = listeners.begin();
Tim Waugh892d10a2015-03-11 13:12:07 +0000819 i != listeners.end();
820 i++) {
Pierre Ossmanf7aa3f92015-09-29 15:40:49 +0200821 if (FD_ISSET((*i)->getFd(), &rfds)) {
822 Socket* sock = (*i)->accept();
Tim Waugh892d10a2015-03-11 13:12:07 +0000823 if (sock) {
Pierre Ossman16419cc2016-04-29 14:29:43 +0200824 sock->outStream().setBlocking(false);
Tim Waugh892d10a2015-03-11 13:12:07 +0000825 server.addSocket(sock);
826 } else {
827 vlog.status("Client connection rejected");
828 }
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000829 }
830 }
831
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000832 server.checkTimeouts();
833
834 // Client list could have been changed.
835 server.getSockets(&sockets);
836
837 // Nothing more to do if there are no client connections.
838 if (sockets.empty())
839 continue;
840
841 // Process events on existing VNC connections
842 for (i = sockets.begin(); i != sockets.end(); i++) {
843 if (FD_ISSET((*i)->getFd(), &rfds))
Pierre Ossmand408ca52016-04-29 14:26:05 +0200844 server.processSocketReadEvent(*i);
Pierre Ossman16419cc2016-04-29 14:29:43 +0200845 if (FD_ISSET((*i)->getFd(), &wfds))
846 server.processSocketWriteEvent(*i);
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000847 }
848
849 if (desktop.isRunning() && sched.goodTimeToPoll()) {
850 sched.newPass();
851 desktop.poll();
852 }
853 }
854
855 } catch (rdr::Exception &e) {
Pierre Ossmanad8609a2012-04-26 09:04:14 +0000856 vlog.error("%s", e.str());
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000857 return 1;
858 }
859
Constantin Kaplinsky0c4306c2008-09-05 07:13:55 +0000860 TXWindow::handleXEvents(dpy);
861
Constantin Kaplinskyb30ae7f2006-05-25 05:04:46 +0000862 vlog.info("Terminated");
863 return 0;
864}