Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 1 | /* Copyright (C) 2002-2005 RealVNC Ltd. All Rights Reserved. |
Constantin Kaplinsky | 2c01983 | 2008-05-30 11:02:04 +0000 | [diff] [blame] | 2 | * Copyright (C) 2004-2008 Constantin Kaplinsky. All Rights Reserved. |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 3 | * |
| 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 Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 32 | #include <rfb/Timer.h> |
| 33 | #include <network/TcpSocket.h> |
| 34 | #include <tx/TXWindow.h> |
| 35 | |
Constantin Kaplinsky | a3b60c4 | 2006-06-02 04:07:49 +0000 | [diff] [blame] | 36 | #include <vncconfig/QueryConnectDialog.h> |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 37 | |
| 38 | #include <signal.h> |
| 39 | #include <X11/X.h> |
| 40 | #include <X11/Xlib.h> |
| 41 | #include <X11/Xutil.h> |
Peter Korsgaard | 5b7ff37 | 2017-07-13 00:36:01 +0200 | [diff] [blame] | 42 | #include <X11/XKBlib.h> |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 43 | #ifdef HAVE_XTEST |
| 44 | #include <X11/extensions/XTest.h> |
| 45 | #endif |
Pierre Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 46 | #ifdef HAVE_XDAMAGE |
| 47 | #include <X11/extensions/Xdamage.h> |
| 48 | #endif |
Alan Coopersmith | 9a739d4 | 2017-08-05 15:16:29 -0700 | [diff] [blame] | 49 | #ifdef HAVE_XFIXES |
| 50 | #include <X11/extensions/Xfixes.h> |
| 51 | #endif |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 52 | |
| 53 | #include <x0vncserver/Geometry.h> |
| 54 | #include <x0vncserver/Image.h> |
Constantin Kaplinsky | 614c7b5 | 2007-12-26 18:17:09 +0000 | [diff] [blame] | 55 | #include <x0vncserver/XPixelBuffer.h> |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 56 | #include <x0vncserver/PollingScheduler.h> |
| 57 | |
Peter Korsgaard | 8fe5090 | 2017-07-13 00:35:57 +0200 | [diff] [blame] | 58 | extern const unsigned short code_map_qnum_to_xorgevdev[]; |
| 59 | extern const unsigned int code_map_qnum_to_xorgevdev_len; |
| 60 | |
| 61 | extern const unsigned short code_map_qnum_to_xorgkbd[]; |
| 62 | extern const unsigned int code_map_qnum_to_xorgkbd_len; |
| 63 | |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 64 | // XXX Lynx/OS 2.3: protos for select(), bzero() |
| 65 | #ifdef Lynx |
| 66 | #include <sys/proto.h> |
| 67 | #endif |
| 68 | |
Constantin Kaplinsky | b963270 | 2006-12-01 10:54:55 +0000 | [diff] [blame] | 69 | extern char buildtime[]; |
| 70 | |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 71 | using namespace rfb; |
| 72 | using namespace network; |
| 73 | |
Peter Korsgaard | 5b7ff37 | 2017-07-13 00:36:01 +0200 | [diff] [blame] | 74 | // number of XKb indicator leds to handle |
| 75 | static const int N_LEDS = 3; |
| 76 | |
| 77 | // order is important as it must match RFB extension |
| 78 | static const char * ledNames[N_LEDS] = { |
| 79 | "Scroll Lock", "Num Lock", "Caps Lock" |
| 80 | }; |
| 81 | |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 82 | static LogWriter vlog("Main"); |
| 83 | |
| 84 | IntParameter pollingCycle("PollingCycle", "Milliseconds per one polling " |
| 85 | "cycle; actual interval may be dynamically " |
| 86 | "adjusted to satisfy MaxProcessorUsage setting", 30); |
| 87 | IntParameter maxProcessorUsage("MaxProcessorUsage", "Maximum percentage of " |
| 88 | "CPU time to be consumed", 35); |
| 89 | BoolParameter useShm("UseSHM", "Use MIT-SHM extension if available", true); |
| 90 | BoolParameter useOverlay("OverlayMode", "Use overlay mode under " |
| 91 | "IRIX or Solaris", true); |
| 92 | StringParameter displayname("display", "The X display", ""); |
| 93 | IntParameter rfbport("rfbport", "TCP port to listen for RFB protocol",5900); |
| 94 | IntParameter queryConnectTimeout("QueryConnectTimeout", |
| 95 | "Number of seconds to show the Accept Connection dialog before " |
| 96 | "rejecting the connection", |
| 97 | 10); |
| 98 | StringParameter hostsFile("HostsFile", "File with IP access control rules", ""); |
Peter Korsgaard | 8fe5090 | 2017-07-13 00:35:57 +0200 | [diff] [blame] | 99 | BoolParameter rawKeyboard("RawKeyboard", |
| 100 | "Send keyboard events straight through and " |
| 101 | "avoid mapping them to the current keyboard " |
| 102 | "layout", false); |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 103 | |
| 104 | // |
| 105 | // Allow the main loop terminate itself gracefully on receiving a signal. |
| 106 | // |
| 107 | |
| 108 | static bool caughtSignal = false; |
| 109 | |
| 110 | static void CleanupSignalHandler(int sig) |
| 111 | { |
| 112 | caughtSignal = true; |
| 113 | } |
| 114 | |
| 115 | |
| 116 | class QueryConnHandler : public VNCServerST::QueryConnectionHandler, |
| 117 | public QueryResultCallback { |
| 118 | public: |
| 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 Tkac | d36b626 | 2009-09-04 10:57:20 +0000 | [diff] [blame] | 128 | *reason = strDup("Another connection is currently being queried."); |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 129 | 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 | } |
| 152 | private: |
| 153 | Display* display; |
| 154 | VNCServerST* server; |
| 155 | QueryConnectDialog* queryConnectDialog; |
| 156 | network::Socket* queryConnectSock; |
| 157 | }; |
| 158 | |
| 159 | |
Pierre Ossman | b6b4dc6 | 2014-01-20 15:05:21 +0100 | [diff] [blame] | 160 | class XDesktop : public SDesktop, public TXGlobalEventHandler |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 161 | { |
| 162 | public: |
| 163 | XDesktop(Display* dpy_, Geometry *geometry_) |
Constantin Kaplinsky | 303433a | 2008-06-04 05:57:06 +0000 | [diff] [blame] | 164 | : dpy(dpy_), geometry(geometry_), pb(0), server(0), |
Pierre Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 165 | oldButtonMask(0), haveXtest(false), haveDamage(false), |
Pierre Ossman | d092f05 | 2017-09-15 13:10:54 +0200 | [diff] [blame^] | 166 | haveXfixes(false), maxButtons(0), running(false), |
| 167 | ledMasks(), ledState(0), codeMap(0), codeMapLen(0) |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 168 | { |
Peter Korsgaard | 5b7ff37 | 2017-07-13 00:36:01 +0200 | [diff] [blame] | 169 | 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 Korsgaard | 8fe5090 | 2017-07-13 00:35:57 +0200 | [diff] [blame] | 200 | // 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 Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 226 | #ifdef HAVE_XTEST |
| 227 | int xtestEventBase; |
| 228 | int xtestErrorBase; |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 229 | |
| 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 Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 243 | #ifdef HAVE_XDAMAGE |
| 244 | int xdamageErrorBase; |
| 245 | |
| 246 | if (XDamageQueryExtension(dpy, &xdamageEventBase, &xdamageErrorBase)) { |
Pierre Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 247 | 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 Korsgaard | 5b7ff37 | 2017-07-13 00:36:01 +0200 | [diff] [blame] | 255 | |
Alan Coopersmith | 9a739d4 | 2017-08-05 15:16:29 -0700 | [diff] [blame] | 256 | #ifdef HAVE_XFIXES |
| 257 | int xfixesErrorBase; |
| 258 | |
| 259 | if (XFixesQueryExtension(dpy, &xfixesEventBase, &xfixesErrorBase)) { |
Alan Coopersmith | 9a739d4 | 2017-08-05 15:16:29 -0700 | [diff] [blame] | 260 | haveXfixes = true; |
| 261 | } else { |
| 262 | #endif |
| 263 | vlog.info("XFIXES extension not present"); |
| 264 | vlog.info("Will not be able to display cursors"); |
| 265 | #ifdef HAVE_XFIXES |
| 266 | } |
| 267 | #endif |
Pierre Ossman | d092f05 | 2017-09-15 13:10:54 +0200 | [diff] [blame^] | 268 | |
Peter Korsgaard | 5b7ff37 | 2017-07-13 00:36:01 +0200 | [diff] [blame] | 269 | TXWindow::setGlobalEventHandler(this); |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 270 | } |
| 271 | virtual ~XDesktop() { |
| 272 | stop(); |
| 273 | } |
| 274 | |
Constantin Kaplinsky | dc873dc | 2008-06-04 09:46:57 +0000 | [diff] [blame] | 275 | inline void poll() { |
Pierre Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 276 | if (pb and not haveDamage) |
Constantin Kaplinsky | dc873dc | 2008-06-04 09:46:57 +0000 | [diff] [blame] | 277 | pb->poll(server); |
| 278 | } |
| 279 | |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 280 | // -=- SDesktop interface |
| 281 | |
| 282 | virtual void start(VNCServer* vs) { |
| 283 | |
| 284 | // Determine actual number of buttons of the X pointer device. |
| 285 | unsigned char btnMap[8]; |
| 286 | int numButtons = XGetPointerMapping(dpy, btnMap, 8); |
| 287 | maxButtons = (numButtons > 8) ? 8 : numButtons; |
| 288 | vlog.info("Enabling %d button%s of X pointer device", |
| 289 | maxButtons, (maxButtons != 1) ? "s" : ""); |
| 290 | |
Constantin Kaplinsky | e0c80c5 | 2008-06-04 03:10:05 +0000 | [diff] [blame] | 291 | // Create an ImageFactory instance for producing Image objects. |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 292 | ImageFactory factory((bool)useShm, (bool)useOverlay); |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 293 | |
Constantin Kaplinsky | dc873dc | 2008-06-04 09:46:57 +0000 | [diff] [blame] | 294 | // Create pixel buffer and provide it to the server object. |
Pierre Ossman | b6b4dc6 | 2014-01-20 15:05:21 +0100 | [diff] [blame] | 295 | pb = new XPixelBuffer(dpy, factory, geometry->getRect()); |
Constantin Kaplinsky | e0c80c5 | 2008-06-04 03:10:05 +0000 | [diff] [blame] | 296 | vlog.info("Allocated %s", pb->getImage()->classDesc()); |
| 297 | |
Constantin Kaplinsky | c341ac6 | 2008-08-21 03:35:08 +0000 | [diff] [blame] | 298 | server = (VNCServerST *)vs; |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 299 | server->setPixelBuffer(pb); |
| 300 | |
Pierre Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 301 | #ifdef HAVE_XDAMAGE |
| 302 | if (haveDamage) { |
| 303 | damage = XDamageCreate(dpy, DefaultRootWindow(dpy), |
| 304 | XDamageReportRawRectangles); |
| 305 | } |
| 306 | #endif |
| 307 | |
Alan Coopersmith | 9a739d4 | 2017-08-05 15:16:29 -0700 | [diff] [blame] | 308 | #ifdef HAVE_XFIXES |
| 309 | if (haveXfixes) { |
| 310 | XFixesSelectCursorInput(dpy, DefaultRootWindow(dpy), |
| 311 | XFixesDisplayCursorNotifyMask); |
| 312 | } |
| 313 | #endif |
| 314 | |
Peter Korsgaard | 5b7ff37 | 2017-07-13 00:36:01 +0200 | [diff] [blame] | 315 | server->setLEDState(ledState); |
| 316 | |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 317 | running = true; |
| 318 | } |
| 319 | |
| 320 | virtual void stop() { |
| 321 | running = false; |
| 322 | |
Pierre Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 323 | #ifdef HAVE_XDAMAGE |
| 324 | if (haveDamage) |
| 325 | XDamageDestroy(dpy, damage); |
| 326 | #endif |
| 327 | |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 328 | delete pb; |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 329 | pb = 0; |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 330 | } |
| 331 | |
| 332 | inline bool isRunning() { |
| 333 | return running; |
| 334 | } |
| 335 | |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 336 | virtual void pointerEvent(const Point& pos, int buttonMask) { |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 337 | #ifdef HAVE_XTEST |
| 338 | if (!haveXtest) return; |
| 339 | XTestFakeMotionEvent(dpy, DefaultScreen(dpy), |
| 340 | geometry->offsetLeft() + pos.x, |
| 341 | geometry->offsetTop() + pos.y, |
| 342 | CurrentTime); |
| 343 | if (buttonMask != oldButtonMask) { |
| 344 | for (int i = 0; i < maxButtons; i++) { |
| 345 | if ((buttonMask ^ oldButtonMask) & (1<<i)) { |
| 346 | if (buttonMask & (1<<i)) { |
| 347 | XTestFakeButtonEvent(dpy, i+1, True, CurrentTime); |
| 348 | } else { |
| 349 | XTestFakeButtonEvent(dpy, i+1, False, CurrentTime); |
| 350 | } |
| 351 | } |
| 352 | } |
| 353 | } |
| 354 | oldButtonMask = buttonMask; |
| 355 | #endif |
| 356 | } |
| 357 | |
Pierre Ossman | ac73038 | 2017-08-16 15:20:20 +0200 | [diff] [blame] | 358 | #ifdef HAVE_XTEST |
| 359 | KeyCode XkbKeysymToKeycode(Display* dpy, KeySym keysym) { |
| 360 | XkbDescPtr xkb; |
| 361 | XkbStateRec state; |
| 362 | unsigned keycode; |
| 363 | |
| 364 | xkb = XkbGetMap(dpy, XkbAllComponentsMask, XkbUseCoreKbd); |
| 365 | if (!xkb) |
| 366 | return 0; |
| 367 | |
| 368 | XkbGetState(dpy, XkbUseCoreKbd, &state); |
| 369 | |
| 370 | for (keycode = xkb->min_key_code; |
| 371 | keycode <= xkb->max_key_code; |
| 372 | keycode++) { |
| 373 | KeySym cursym; |
| 374 | unsigned int mods; |
| 375 | XkbTranslateKeyCode(xkb, keycode, state.compat_state, &mods, &cursym); |
| 376 | if (cursym == keysym) |
| 377 | break; |
| 378 | } |
| 379 | |
| 380 | if (keycode > xkb->max_key_code) |
| 381 | keycode = 0; |
| 382 | |
| 383 | XkbFreeKeyboard(xkb, XkbAllComponentsMask, True); |
| 384 | |
| 385 | return keycode; |
| 386 | } |
| 387 | #endif |
| 388 | |
Pierre Ossman | 5ae2821 | 2017-05-16 14:30:38 +0200 | [diff] [blame] | 389 | virtual void keyEvent(rdr::U32 keysym, rdr::U32 xtcode, bool down) { |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 390 | #ifdef HAVE_XTEST |
Peter Korsgaard | 8fe5090 | 2017-07-13 00:35:57 +0200 | [diff] [blame] | 391 | int keycode = 0; |
Pierre Ossman | ac73038 | 2017-08-16 15:20:20 +0200 | [diff] [blame] | 392 | |
| 393 | if (!haveXtest) |
| 394 | return; |
Peter Korsgaard | 8fe5090 | 2017-07-13 00:35:57 +0200 | [diff] [blame] | 395 | |
| 396 | // Use scan code if provided and mapping exists |
| 397 | if (codeMap && rawKeyboard && xtcode < codeMapLen) |
| 398 | keycode = codeMap[xtcode]; |
| 399 | |
Pierre Ossman | 63cc07e | 2017-09-15 11:17:02 +0200 | [diff] [blame] | 400 | if (!keycode) { |
| 401 | if (!down || (pressedKeys.find(keysym) != pressedKeys.end())) |
| 402 | keycode = pressedKeys[keysym]; |
Pierre Ossman | ac73038 | 2017-08-16 15:20:20 +0200 | [diff] [blame] | 403 | else { |
| 404 | // XKeysymToKeycode() doesn't respect state, so we have to use |
| 405 | // something slightly more complex |
Pierre Ossman | 63cc07e | 2017-09-15 11:17:02 +0200 | [diff] [blame] | 406 | keycode = XkbKeysymToKeycode(dpy, keysym); |
Pierre Ossman | ac73038 | 2017-08-16 15:20:20 +0200 | [diff] [blame] | 407 | } |
Pierre Ossman | ac73038 | 2017-08-16 15:20:20 +0200 | [diff] [blame] | 408 | } |
Peter Korsgaard | 8fe5090 | 2017-07-13 00:35:57 +0200 | [diff] [blame] | 409 | |
Pierre Ossman | 63cc07e | 2017-09-15 11:17:02 +0200 | [diff] [blame] | 410 | if (!keycode) |
| 411 | return; |
| 412 | |
| 413 | if (down) |
| 414 | pressedKeys[keysym] = keycode; |
| 415 | else |
| 416 | pressedKeys.erase(keysym); |
| 417 | |
Pierre Ossman | ac73038 | 2017-08-16 15:20:20 +0200 | [diff] [blame] | 418 | XTestFakeKeyEvent(dpy, keycode, down, CurrentTime); |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 419 | #endif |
| 420 | } |
| 421 | |
| 422 | virtual void clientCutText(const char* str, int len) { |
| 423 | } |
| 424 | |
| 425 | virtual Point getFbSize() { |
| 426 | return Point(pb->width(), pb->height()); |
| 427 | } |
| 428 | |
Pierre Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 429 | // -=- TXGlobalEventHandler interface |
| 430 | |
| 431 | virtual bool handleGlobalEvent(XEvent* ev) { |
Peter Korsgaard | 5b7ff37 | 2017-07-13 00:36:01 +0200 | [diff] [blame] | 432 | if (ev->type == xkbEventBase + XkbEventCode) { |
| 433 | XkbEvent *kb = (XkbEvent *)ev; |
Pierre Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 434 | |
Peter Korsgaard | 5b7ff37 | 2017-07-13 00:36:01 +0200 | [diff] [blame] | 435 | if (kb->any.xkb_type != XkbIndicatorStateNotify) |
| 436 | return false; |
Pierre Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 437 | |
Peter Korsgaard | 5b7ff37 | 2017-07-13 00:36:01 +0200 | [diff] [blame] | 438 | vlog.debug("Got indicator update, mask is now 0x%x", kb->indicators.state); |
| 439 | |
| 440 | ledState = 0; |
| 441 | for (int i = 0; i < N_LEDS; i++) { |
| 442 | if (kb->indicators.state & ledMasks[i]) |
| 443 | ledState |= 1u << i; |
| 444 | } |
| 445 | |
| 446 | if (running) |
| 447 | server->setLEDState(ledState); |
| 448 | |
Pierre Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 449 | return true; |
Peter Korsgaard | 5b7ff37 | 2017-07-13 00:36:01 +0200 | [diff] [blame] | 450 | #ifdef HAVE_XDAMAGE |
| 451 | } else if (ev->type == xdamageEventBase) { |
| 452 | XDamageNotifyEvent* dev; |
| 453 | Rect rect; |
Pierre Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 454 | |
Peter Korsgaard | 5b7ff37 | 2017-07-13 00:36:01 +0200 | [diff] [blame] | 455 | if (!running) |
| 456 | return true; |
Pierre Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 457 | |
Peter Korsgaard | 5b7ff37 | 2017-07-13 00:36:01 +0200 | [diff] [blame] | 458 | dev = (XDamageNotifyEvent*)ev; |
| 459 | rect.setXYWH(dev->area.x, dev->area.y, dev->area.width, dev->area.height); |
| 460 | server->add_changed(rect); |
| 461 | |
| 462 | return true; |
Pierre Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 463 | #endif |
Alan Coopersmith | 9a739d4 | 2017-08-05 15:16:29 -0700 | [diff] [blame] | 464 | #ifdef HAVE_XFIXES |
Pierre Ossman | d092f05 | 2017-09-15 13:10:54 +0200 | [diff] [blame^] | 465 | } else if (ev->type == xfixesEventBase + XFixesCursorNotify) { |
Alan Coopersmith | 9a739d4 | 2017-08-05 15:16:29 -0700 | [diff] [blame] | 466 | XFixesCursorNotifyEvent* cev; |
| 467 | XFixesCursorImage *cim; |
| 468 | |
| 469 | if (!running) |
| 470 | return true; |
| 471 | |
| 472 | cev = (XFixesCursorNotifyEvent*)ev; |
| 473 | |
| 474 | if (cev->subtype != XFixesDisplayCursorNotify) |
| 475 | return false; |
| 476 | |
| 477 | cim = XFixesGetCursorImage(dpy); |
| 478 | if (cim == NULL) |
| 479 | return false; |
| 480 | |
| 481 | // Copied from XserverDesktop::setCursor() in |
| 482 | // unix/xserver/hw/vnc/XserverDesktop.cc and adapted to |
| 483 | // handle long -> U32 conversion for 64-bit Xlib |
| 484 | rdr::U8* cursorData; |
| 485 | rdr::U8 *out; |
| 486 | const unsigned long *pixels; |
| 487 | |
| 488 | cursorData = new rdr::U8[cim->width * cim->height * 4]; |
| 489 | |
| 490 | // Un-premultiply alpha |
| 491 | pixels = cim->pixels; |
| 492 | out = cursorData; |
| 493 | for (int y = 0; y < cim->height; y++) { |
| 494 | for (int x = 0; x < cim->width; x++) { |
| 495 | rdr::U8 alpha; |
| 496 | rdr::U32 pixel = *pixels++; |
| 497 | rdr::U8 *in = (rdr::U8 *) &pixel; |
| 498 | |
| 499 | alpha = in[3]; |
| 500 | if (alpha == 0) |
| 501 | alpha = 1; // Avoid division by zero |
| 502 | |
| 503 | *out++ = (unsigned)*in++ * 255/alpha; |
| 504 | *out++ = (unsigned)*in++ * 255/alpha; |
| 505 | *out++ = (unsigned)*in++ * 255/alpha; |
| 506 | *out++ = *in++; |
| 507 | } |
| 508 | } |
| 509 | |
| 510 | try { |
| 511 | server->setCursor(cim->width, cim->height, Point(cim->xhot, cim->yhot), |
| 512 | cursorData); |
| 513 | } catch (rdr::Exception& e) { |
| 514 | vlog.error("XserverDesktop::setCursor: %s",e.str()); |
| 515 | } |
| 516 | |
| 517 | delete [] cursorData; |
| 518 | XFree(cim); |
| 519 | return true; |
Alan Coopersmith | 9a739d4 | 2017-08-05 15:16:29 -0700 | [diff] [blame] | 520 | #endif |
Peter Korsgaard | 5b7ff37 | 2017-07-13 00:36:01 +0200 | [diff] [blame] | 521 | } |
| 522 | |
| 523 | return false; |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 524 | } |
| 525 | |
| 526 | protected: |
| 527 | Display* dpy; |
| 528 | Geometry* geometry; |
Constantin Kaplinsky | 2c01983 | 2008-05-30 11:02:04 +0000 | [diff] [blame] | 529 | XPixelBuffer* pb; |
Constantin Kaplinsky | c341ac6 | 2008-08-21 03:35:08 +0000 | [diff] [blame] | 530 | VNCServerST* server; |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 531 | int oldButtonMask; |
| 532 | bool haveXtest; |
Pierre Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 533 | bool haveDamage; |
Alan Coopersmith | 9a739d4 | 2017-08-05 15:16:29 -0700 | [diff] [blame] | 534 | bool haveXfixes; |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 535 | int maxButtons; |
Pierre Ossman | ac73038 | 2017-08-16 15:20:20 +0200 | [diff] [blame] | 536 | std::map<KeySym, KeyCode> pressedKeys; |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 537 | bool running; |
Pierre Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 538 | #ifdef HAVE_XDAMAGE |
| 539 | Damage damage; |
| 540 | int xdamageEventBase; |
| 541 | #endif |
Peter Korsgaard | 5b7ff37 | 2017-07-13 00:36:01 +0200 | [diff] [blame] | 542 | int xkbEventBase; |
Alan Coopersmith | 9a739d4 | 2017-08-05 15:16:29 -0700 | [diff] [blame] | 543 | #ifdef HAVE_XFIXES |
| 544 | int xfixesEventBase; |
| 545 | #endif |
Peter Korsgaard | 5b7ff37 | 2017-07-13 00:36:01 +0200 | [diff] [blame] | 546 | int ledMasks[N_LEDS]; |
| 547 | unsigned ledState; |
Peter Korsgaard | 8fe5090 | 2017-07-13 00:35:57 +0200 | [diff] [blame] | 548 | const unsigned short *codeMap; |
| 549 | unsigned codeMapLen; |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 550 | }; |
| 551 | |
| 552 | |
| 553 | class FileTcpFilter : public TcpFilter |
| 554 | { |
| 555 | |
| 556 | public: |
| 557 | |
| 558 | FileTcpFilter(const char *fname) |
| 559 | : TcpFilter("-"), fileName(NULL), lastModTime(0) |
| 560 | { |
| 561 | if (fname != NULL) |
| 562 | fileName = strdup((char *)fname); |
| 563 | } |
| 564 | |
| 565 | virtual ~FileTcpFilter() |
| 566 | { |
| 567 | if (fileName != NULL) |
| 568 | free(fileName); |
| 569 | } |
| 570 | |
| 571 | virtual bool verifyConnection(Socket* s) |
| 572 | { |
| 573 | if (!reloadRules()) { |
| 574 | vlog.error("Could not read IP filtering rules: rejecting all clients"); |
| 575 | filter.clear(); |
| 576 | filter.push_back(parsePattern("-")); |
| 577 | return false; |
| 578 | } |
| 579 | |
| 580 | return TcpFilter::verifyConnection(s); |
| 581 | } |
| 582 | |
| 583 | protected: |
| 584 | |
| 585 | bool reloadRules() |
| 586 | { |
| 587 | if (fileName == NULL) |
| 588 | return true; |
| 589 | |
| 590 | struct stat st; |
| 591 | if (stat(fileName, &st) != 0) |
| 592 | return false; |
| 593 | |
| 594 | if (st.st_mtime != lastModTime) { |
| 595 | // Actually reload only if the file was modified |
| 596 | FILE *fp = fopen(fileName, "r"); |
| 597 | if (fp == NULL) |
| 598 | return false; |
| 599 | |
| 600 | // Remove all the rules from the parent class |
| 601 | filter.clear(); |
| 602 | |
| 603 | // Parse the file contents adding rules to the parent class |
| 604 | char buf[32]; |
| 605 | while (readLine(buf, 32, fp)) { |
| 606 | if (buf[0] && strchr("+-?", buf[0])) { |
| 607 | filter.push_back(parsePattern(buf)); |
| 608 | } |
| 609 | } |
| 610 | |
| 611 | fclose(fp); |
| 612 | lastModTime = st.st_mtime; |
| 613 | } |
| 614 | return true; |
| 615 | } |
| 616 | |
| 617 | protected: |
| 618 | |
| 619 | char *fileName; |
| 620 | time_t lastModTime; |
| 621 | |
| 622 | private: |
| 623 | |
| 624 | // |
| 625 | // NOTE: we silently truncate long lines in this function. |
| 626 | // |
| 627 | |
| 628 | bool readLine(char *buf, int bufSize, FILE *fp) |
| 629 | { |
| 630 | if (fp == NULL || buf == NULL || bufSize == 0) |
| 631 | return false; |
| 632 | |
| 633 | if (fgets(buf, bufSize, fp) == NULL) |
| 634 | return false; |
| 635 | |
| 636 | char *ptr = strchr(buf, '\n'); |
| 637 | if (ptr != NULL) { |
| 638 | *ptr = '\0'; // remove newline at the end |
| 639 | } else { |
| 640 | if (!feof(fp)) { |
| 641 | int c; |
| 642 | do { // skip the rest of a long line |
| 643 | c = getc(fp); |
| 644 | } while (c != '\n' && c != EOF); |
| 645 | } |
| 646 | } |
| 647 | return true; |
| 648 | } |
| 649 | |
| 650 | }; |
| 651 | |
| 652 | char* programName; |
| 653 | |
Constantin Kaplinsky | 2039d7b | 2008-06-04 10:43:10 +0000 | [diff] [blame] | 654 | static void printVersion(FILE *fp) |
| 655 | { |
Peter Ã…strand | 4eacc02 | 2009-02-27 10:12:14 +0000 | [diff] [blame] | 656 | fprintf(fp, "TigerVNC Server version %s, built %s\n", |
Constantin Kaplinsky | ea7b650 | 2008-09-28 05:08:48 +0000 | [diff] [blame] | 657 | PACKAGE_VERSION, buildtime); |
Constantin Kaplinsky | 2039d7b | 2008-06-04 10:43:10 +0000 | [diff] [blame] | 658 | } |
| 659 | |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 660 | static void usage() |
| 661 | { |
Constantin Kaplinsky | 2039d7b | 2008-06-04 10:43:10 +0000 | [diff] [blame] | 662 | printVersion(stderr); |
| 663 | fprintf(stderr, "\nUsage: %s [<parameters>]\n", programName); |
| 664 | fprintf(stderr, " %s --version\n", programName); |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 665 | fprintf(stderr,"\n" |
| 666 | "Parameters can be turned on with -<param> or off with -<param>=0\n" |
| 667 | "Parameters which take a value can be specified as " |
| 668 | "-<param> <value>\n" |
| 669 | "Other valid forms are <param>=<value> -<param>=<value> " |
| 670 | "--<param>=<value>\n" |
| 671 | "Parameter names are case-insensitive. The parameters are:\n\n"); |
| 672 | Configuration::listParams(79, 14); |
| 673 | exit(1); |
| 674 | } |
| 675 | |
| 676 | int main(int argc, char** argv) |
| 677 | { |
| 678 | initStdIOLoggers(); |
| 679 | LogWriter::setLogParams("*:stderr:30"); |
| 680 | |
| 681 | programName = argv[0]; |
| 682 | Display* dpy; |
| 683 | |
Adam Tkac | c58b3d1 | 2010-04-23 13:55:10 +0000 | [diff] [blame] | 684 | Configuration::enableServerParams(); |
| 685 | |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 686 | for (int i = 1; i < argc; i++) { |
| 687 | if (Configuration::setParam(argv[i])) |
| 688 | continue; |
| 689 | |
| 690 | if (argv[i][0] == '-') { |
| 691 | if (i+1 < argc) { |
| 692 | if (Configuration::setParam(&argv[i][1], argv[i+1])) { |
| 693 | i++; |
| 694 | continue; |
| 695 | } |
| 696 | } |
Constantin Kaplinsky | 2039d7b | 2008-06-04 10:43:10 +0000 | [diff] [blame] | 697 | if (strcmp(argv[i], "-v") == 0 || |
| 698 | strcmp(argv[i], "-version") == 0 || |
| 699 | strcmp(argv[i], "--version") == 0) { |
| 700 | printVersion(stdout); |
| 701 | return 0; |
| 702 | } |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 703 | usage(); |
| 704 | } |
| 705 | |
| 706 | usage(); |
| 707 | } |
| 708 | |
| 709 | CharArray dpyStr(displayname.getData()); |
| 710 | if (!(dpy = XOpenDisplay(dpyStr.buf[0] ? dpyStr.buf : 0))) { |
Constantin Kaplinsky | 7bdccd7 | 2008-08-20 06:22:28 +0000 | [diff] [blame] | 711 | // FIXME: Why not vlog.error(...)? |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 712 | fprintf(stderr,"%s: unable to open display \"%s\"\r\n", |
Constantin Kaplinsky | 7bdccd7 | 2008-08-20 06:22:28 +0000 | [diff] [blame] | 713 | programName, XDisplayName(dpyStr.buf)); |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 714 | exit(1); |
| 715 | } |
| 716 | |
| 717 | signal(SIGHUP, CleanupSignalHandler); |
| 718 | signal(SIGINT, CleanupSignalHandler); |
| 719 | signal(SIGTERM, CleanupSignalHandler); |
| 720 | |
Pierre Ossman | f7aa3f9 | 2015-09-29 15:40:49 +0200 | [diff] [blame] | 721 | std::list<TcpListener*> listeners; |
Tim Waugh | 892d10a | 2015-03-11 13:12:07 +0000 | [diff] [blame] | 722 | |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 723 | try { |
| 724 | TXWindow::init(dpy,"x0vncserver"); |
| 725 | Geometry geo(DisplayWidth(dpy, DefaultScreen(dpy)), |
| 726 | DisplayHeight(dpy, DefaultScreen(dpy))); |
Constantin Kaplinsky | 23c6022 | 2008-06-04 03:58:07 +0000 | [diff] [blame] | 727 | if (geo.getRect().is_empty()) { |
| 728 | vlog.error("Exiting with error"); |
| 729 | return 1; |
| 730 | } |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 731 | XDesktop desktop(dpy, &geo); |
Constantin Kaplinsky | 8232831 | 2008-04-24 08:44:24 +0000 | [diff] [blame] | 732 | |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 733 | VNCServerST server("x0vncserver", &desktop); |
| 734 | QueryConnHandler qcHandler(dpy, &server); |
| 735 | server.setQueryConnectionHandler(&qcHandler); |
| 736 | |
Tim Waugh | 892d10a | 2015-03-11 13:12:07 +0000 | [diff] [blame] | 737 | createTcpListeners(&listeners, 0, (int)rfbport); |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 738 | vlog.info("Listening on port %d", (int)rfbport); |
| 739 | |
Constantin Kaplinsky | 7bdccd7 | 2008-08-20 06:22:28 +0000 | [diff] [blame] | 740 | const char *hostsData = hostsFile.getData(); |
| 741 | FileTcpFilter fileTcpFilter(hostsData); |
| 742 | if (strlen(hostsData) != 0) |
Pierre Ossman | f7aa3f9 | 2015-09-29 15:40:49 +0200 | [diff] [blame] | 743 | for (std::list<TcpListener*>::iterator i = listeners.begin(); |
Tim Waugh | 892d10a | 2015-03-11 13:12:07 +0000 | [diff] [blame] | 744 | i != listeners.end(); |
| 745 | i++) |
Pierre Ossman | f7aa3f9 | 2015-09-29 15:40:49 +0200 | [diff] [blame] | 746 | (*i)->setFilter(&fileTcpFilter); |
Constantin Kaplinsky | 7bdccd7 | 2008-08-20 06:22:28 +0000 | [diff] [blame] | 747 | delete[] hostsData; |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 748 | |
| 749 | PollingScheduler sched((int)pollingCycle, (int)maxProcessorUsage); |
| 750 | |
| 751 | while (!caughtSignal) { |
Pierre Ossman | 278e420 | 2016-04-29 14:28:54 +0200 | [diff] [blame] | 752 | int wait_ms; |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 753 | struct timeval tv; |
Pierre Ossman | 16419cc | 2016-04-29 14:29:43 +0200 | [diff] [blame] | 754 | fd_set rfds, wfds; |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 755 | std::list<Socket*> sockets; |
| 756 | std::list<Socket*>::iterator i; |
| 757 | |
| 758 | // Process any incoming X events |
| 759 | TXWindow::handleXEvents(dpy); |
| 760 | |
| 761 | FD_ZERO(&rfds); |
Pierre Ossman | 16419cc | 2016-04-29 14:29:43 +0200 | [diff] [blame] | 762 | FD_ZERO(&wfds); |
| 763 | |
Pierre Ossman | a7b728a | 2014-06-13 10:56:59 +0000 | [diff] [blame] | 764 | FD_SET(ConnectionNumber(dpy), &rfds); |
Pierre Ossman | f7aa3f9 | 2015-09-29 15:40:49 +0200 | [diff] [blame] | 765 | for (std::list<TcpListener*>::iterator i = listeners.begin(); |
Tim Waugh | 892d10a | 2015-03-11 13:12:07 +0000 | [diff] [blame] | 766 | i != listeners.end(); |
| 767 | i++) |
Pierre Ossman | f7aa3f9 | 2015-09-29 15:40:49 +0200 | [diff] [blame] | 768 | FD_SET((*i)->getFd(), &rfds); |
Tim Waugh | 892d10a | 2015-03-11 13:12:07 +0000 | [diff] [blame] | 769 | |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 770 | server.getSockets(&sockets); |
| 771 | int clients_connected = 0; |
| 772 | for (i = sockets.begin(); i != sockets.end(); i++) { |
| 773 | if ((*i)->isShutdown()) { |
| 774 | server.removeSocket(*i); |
| 775 | delete (*i); |
| 776 | } else { |
| 777 | FD_SET((*i)->getFd(), &rfds); |
Pierre Ossman | 16419cc | 2016-04-29 14:29:43 +0200 | [diff] [blame] | 778 | if ((*i)->outStream().bufferUsage() > 0) |
| 779 | FD_SET((*i)->getFd(), &wfds); |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 780 | clients_connected++; |
| 781 | } |
| 782 | } |
| 783 | |
Constantin Kaplinsky | 813dbb4 | 2006-12-05 08:03:18 +0000 | [diff] [blame] | 784 | if (!clients_connected) |
| 785 | sched.reset(); |
| 786 | |
Pierre Ossman | 278e420 | 2016-04-29 14:28:54 +0200 | [diff] [blame] | 787 | wait_ms = 0; |
| 788 | |
Constantin Kaplinsky | 813dbb4 | 2006-12-05 08:03:18 +0000 | [diff] [blame] | 789 | if (sched.isRunning()) { |
Pierre Ossman | 278e420 | 2016-04-29 14:28:54 +0200 | [diff] [blame] | 790 | wait_ms = sched.millisRemaining(); |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 791 | if (wait_ms > 500) { |
| 792 | wait_ms = 500; |
| 793 | } |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 794 | } |
Pierre Ossman | 278e420 | 2016-04-29 14:28:54 +0200 | [diff] [blame] | 795 | |
| 796 | soonestTimeout(&wait_ms, server.checkTimeouts()); |
| 797 | |
| 798 | tv.tv_sec = wait_ms / 1000; |
| 799 | tv.tv_usec = (wait_ms % 1000) * 1000; |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 800 | |
| 801 | // Do the wait... |
| 802 | sched.sleepStarted(); |
Pierre Ossman | 16419cc | 2016-04-29 14:29:43 +0200 | [diff] [blame] | 803 | int n = select(FD_SETSIZE, &rfds, &wfds, 0, |
Pierre Ossman | 278e420 | 2016-04-29 14:28:54 +0200 | [diff] [blame] | 804 | wait_ms ? &tv : NULL); |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 805 | sched.sleepFinished(); |
| 806 | |
| 807 | if (n < 0) { |
| 808 | if (errno == EINTR) { |
| 809 | vlog.debug("Interrupted select() system call"); |
| 810 | continue; |
| 811 | } else { |
| 812 | throw rdr::SystemException("select", errno); |
| 813 | } |
| 814 | } |
| 815 | |
| 816 | // Accept new VNC connections |
Pierre Ossman | f7aa3f9 | 2015-09-29 15:40:49 +0200 | [diff] [blame] | 817 | for (std::list<TcpListener*>::iterator i = listeners.begin(); |
Tim Waugh | 892d10a | 2015-03-11 13:12:07 +0000 | [diff] [blame] | 818 | i != listeners.end(); |
| 819 | i++) { |
Pierre Ossman | f7aa3f9 | 2015-09-29 15:40:49 +0200 | [diff] [blame] | 820 | if (FD_ISSET((*i)->getFd(), &rfds)) { |
| 821 | Socket* sock = (*i)->accept(); |
Tim Waugh | 892d10a | 2015-03-11 13:12:07 +0000 | [diff] [blame] | 822 | if (sock) { |
Pierre Ossman | 16419cc | 2016-04-29 14:29:43 +0200 | [diff] [blame] | 823 | sock->outStream().setBlocking(false); |
Tim Waugh | 892d10a | 2015-03-11 13:12:07 +0000 | [diff] [blame] | 824 | server.addSocket(sock); |
| 825 | } else { |
| 826 | vlog.status("Client connection rejected"); |
| 827 | } |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 828 | } |
| 829 | } |
| 830 | |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 831 | server.checkTimeouts(); |
| 832 | |
| 833 | // Client list could have been changed. |
| 834 | server.getSockets(&sockets); |
| 835 | |
| 836 | // Nothing more to do if there are no client connections. |
| 837 | if (sockets.empty()) |
| 838 | continue; |
| 839 | |
| 840 | // Process events on existing VNC connections |
| 841 | for (i = sockets.begin(); i != sockets.end(); i++) { |
| 842 | if (FD_ISSET((*i)->getFd(), &rfds)) |
Pierre Ossman | d408ca5 | 2016-04-29 14:26:05 +0200 | [diff] [blame] | 843 | server.processSocketReadEvent(*i); |
Pierre Ossman | 16419cc | 2016-04-29 14:29:43 +0200 | [diff] [blame] | 844 | if (FD_ISSET((*i)->getFd(), &wfds)) |
| 845 | server.processSocketWriteEvent(*i); |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 846 | } |
| 847 | |
| 848 | if (desktop.isRunning() && sched.goodTimeToPoll()) { |
| 849 | sched.newPass(); |
| 850 | desktop.poll(); |
| 851 | } |
| 852 | } |
| 853 | |
| 854 | } catch (rdr::Exception &e) { |
Pierre Ossman | ad8609a | 2012-04-26 09:04:14 +0000 | [diff] [blame] | 855 | vlog.error("%s", e.str()); |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 856 | return 1; |
| 857 | } |
| 858 | |
Constantin Kaplinsky | 0c4306c | 2008-09-05 07:13:55 +0000 | [diff] [blame] | 859 | TXWindow::handleXEvents(dpy); |
| 860 | |
Constantin Kaplinsky | b30ae7f | 2006-05-25 05:04:46 +0000 | [diff] [blame] | 861 | vlog.info("Terminated"); |
| 862 | return 0; |
| 863 | } |