123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053 |
- /* Copyright (C) 2002-2005 RealVNC Ltd. All Rights Reserved.
- * Copyright 2009-2011 Pierre Ossman for Cendio AB
- * Copyright 2014 Brian P. Hinz
- *
- * 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.
- */
- //
- // XserverDesktop.cxx
- //
-
- #ifdef HAVE_DIX_CONFIG_H
- #include <dix-config.h>
- #endif
-
- #include <assert.h>
- #include <stdio.h>
- #include <strings.h>
- #include <unistd.h>
- #include <pwd.h>
- #include <sys/types.h>
- #include <sys/stat.h>
- #include <fcntl.h>
- #include <sys/utsname.h>
- #include <network/TcpSocket.h>
- #include <rfb/Exception.h>
- #include <rfb/VNCServerST.h>
- #include <rfb/HTTPServer.h>
- #include <rfb/LogWriter.h>
- #include <rfb/Configuration.h>
- #include "XserverDesktop.h"
- #include "vncExtInit.h"
- #include "xorg-version.h"
- #include "Input.h"
-
- extern "C" {
- #define public c_public
- #define class c_class
-
- extern char *display;
-
- #ifdef RANDR
- #include "randrstr.h"
- #endif
- #include "cursorstr.h"
- #undef public
- #undef class
- }
-
- using namespace rfb;
- using namespace network;
-
- static LogWriter vlog("XserverDesktop");
-
- IntParameter queryConnectTimeout("QueryConnectTimeout",
- "Number of seconds to show the Accept Connection dialog before "
- "rejecting the connection",
- 10);
-
- static rdr::U8 reverseBits[] = {
- 0x00, 0x80, 0x40, 0xc0, 0x20, 0xa0, 0x60, 0xe0, 0x10, 0x90, 0x50, 0xd0,
- 0x30, 0xb0, 0x70, 0xf0, 0x08, 0x88, 0x48, 0xc8, 0x28, 0xa8, 0x68, 0xe8,
- 0x18, 0x98, 0x58, 0xd8, 0x38, 0xb8, 0x78, 0xf8, 0x04, 0x84, 0x44, 0xc4,
- 0x24, 0xa4, 0x64, 0xe4, 0x14, 0x94, 0x54, 0xd4, 0x34, 0xb4, 0x74, 0xf4,
- 0x0c, 0x8c, 0x4c, 0xcc, 0x2c, 0xac, 0x6c, 0xec, 0x1c, 0x9c, 0x5c, 0xdc,
- 0x3c, 0xbc, 0x7c, 0xfc, 0x02, 0x82, 0x42, 0xc2, 0x22, 0xa2, 0x62, 0xe2,
- 0x12, 0x92, 0x52, 0xd2, 0x32, 0xb2, 0x72, 0xf2, 0x0a, 0x8a, 0x4a, 0xca,
- 0x2a, 0xaa, 0x6a, 0xea, 0x1a, 0x9a, 0x5a, 0xda, 0x3a, 0xba, 0x7a, 0xfa,
- 0x06, 0x86, 0x46, 0xc6, 0x26, 0xa6, 0x66, 0xe6, 0x16, 0x96, 0x56, 0xd6,
- 0x36, 0xb6, 0x76, 0xf6, 0x0e, 0x8e, 0x4e, 0xce, 0x2e, 0xae, 0x6e, 0xee,
- 0x1e, 0x9e, 0x5e, 0xde, 0x3e, 0xbe, 0x7e, 0xfe, 0x01, 0x81, 0x41, 0xc1,
- 0x21, 0xa1, 0x61, 0xe1, 0x11, 0x91, 0x51, 0xd1, 0x31, 0xb1, 0x71, 0xf1,
- 0x09, 0x89, 0x49, 0xc9, 0x29, 0xa9, 0x69, 0xe9, 0x19, 0x99, 0x59, 0xd9,
- 0x39, 0xb9, 0x79, 0xf9, 0x05, 0x85, 0x45, 0xc5, 0x25, 0xa5, 0x65, 0xe5,
- 0x15, 0x95, 0x55, 0xd5, 0x35, 0xb5, 0x75, 0xf5, 0x0d, 0x8d, 0x4d, 0xcd,
- 0x2d, 0xad, 0x6d, 0xed, 0x1d, 0x9d, 0x5d, 0xdd, 0x3d, 0xbd, 0x7d, 0xfd,
- 0x03, 0x83, 0x43, 0xc3, 0x23, 0xa3, 0x63, 0xe3, 0x13, 0x93, 0x53, 0xd3,
- 0x33, 0xb3, 0x73, 0xf3, 0x0b, 0x8b, 0x4b, 0xcb, 0x2b, 0xab, 0x6b, 0xeb,
- 0x1b, 0x9b, 0x5b, 0xdb, 0x3b, 0xbb, 0x7b, 0xfb, 0x07, 0x87, 0x47, 0xc7,
- 0x27, 0xa7, 0x67, 0xe7, 0x17, 0x97, 0x57, 0xd7, 0x37, 0xb7, 0x77, 0xf7,
- 0x0f, 0x8f, 0x4f, 0xcf, 0x2f, 0xaf, 0x6f, 0xef, 0x1f, 0x9f, 0x5f, 0xdf,
- 0x3f, 0xbf, 0x7f, 0xff
- };
-
-
- class FileHTTPServer : public rfb::HTTPServer {
- public:
- FileHTTPServer(XserverDesktop* d) : desktop(d) {}
- virtual ~FileHTTPServer() {}
-
- virtual rdr::InStream* getFile(const char* name, const char** contentType,
- int* contentLength, time_t* lastModified)
- {
- if (name[0] != '/' || strstr(name, "..") != 0) {
- vlog.info("http request was for invalid file name");
- return 0;
- }
-
- if (strcmp(name, "/") == 0) name = "/index.vnc";
-
- CharArray httpDirStr(httpDir.getData());
- CharArray fname(strlen(httpDirStr.buf)+strlen(name)+1);
- sprintf(fname.buf, "%s%s", httpDirStr.buf, name);
- int fd = open(fname.buf, O_RDONLY);
- if (fd < 0) return 0;
- rdr::InStream* is = new rdr::FdInStream(fd, -1, 0, true);
- *contentType = guessContentType(name, *contentType);
- if (strlen(name) > 4 && strcasecmp(&name[strlen(name)-4], ".vnc") == 0) {
- is = new rdr::SubstitutingInStream(is, desktop, 20);
- *contentType = "text/html";
- } else {
- struct stat st;
- if (fstat(fd, &st) == 0) {
- *contentLength = st.st_size;
- *lastModified = st.st_mtime;
- }
- }
- return is;
- }
-
- XserverDesktop* desktop;
- };
-
-
- XserverDesktop::XserverDesktop(ScreenPtr pScreen_,
- network::TcpListener* listener_,
- network::TcpListener* httpListener_,
- const char* name, const rfb::PixelFormat &pf,
- void* fbptr, int stride)
- : pScreen(pScreen_),
- server(0), httpServer(0),
- listener(listener_), httpListener(httpListener_),
- deferredUpdateTimerSet(false),
- grabbing(false), ignoreHooks_(false), directFbptr(true),
- queryConnectId(0)
- {
- format = pf;
-
- server = new VNCServerST(name, this);
- setFramebuffer(pScreen->width, pScreen->height, fbptr, stride);
- server->setQueryConnectionHandler(this);
-
- if (httpListener)
- httpServer = new FileHTTPServer(this);
-
- inputDevice = new InputDevice(server);
- }
-
- XserverDesktop::~XserverDesktop()
- {
- if (!directFbptr)
- delete [] data;
- delete inputDevice;
- delete httpServer;
- delete server;
- }
-
- void XserverDesktop::blockUpdates()
- {
- server->blockUpdates();
- }
-
- void XserverDesktop::unblockUpdates()
- {
- server->unblockUpdates();
- }
-
- void XserverDesktop::setFramebuffer(int w, int h, void* fbptr, int stride_)
- {
- ScreenSet layout;
-
- width_ = w;
- height_ = h;
-
- if (!directFbptr) {
- delete [] data;
- directFbptr = true;
- }
-
- if (!fbptr) {
- fbptr = new rdr::U8[w * h * (format.bpp/8)];
- stride_ = w;
- directFbptr = false;
- }
-
- data = (rdr::U8*)fbptr;
- stride = stride_;
-
- layout = computeScreenLayout();
-
- server->setPixelBuffer(this, layout);
- }
-
- void XserverDesktop::refreshScreenLayout()
- {
- server->setScreenLayout(computeScreenLayout());
- }
-
- ScreenSet XserverDesktop::computeScreenLayout()
- {
- ScreenSet layout;
-
- #ifdef RANDR
- rrScrPrivPtr rp = rrGetScrPriv(pScreen);
- OutputIdMap newIdMap;
-
- for (int i = 0;i < rp->numOutputs;i++) {
- RROutputPtr output;
- RRCrtcPtr crtc;
-
- output = rp->outputs[i];
- crtc = output->crtc;
-
- /* Disabled? */
- if ((crtc == NULL) || (crtc->mode == NULL))
- continue;
-
- /* Known output? */
- if (outputIdMap.count(output) == 1)
- newIdMap[output] = outputIdMap[output];
- else {
- rdr::U32 id;
- OutputIdMap::const_iterator iter;
-
- while (true) {
- id = rand();
- for (iter = outputIdMap.begin();iter != outputIdMap.end();++iter) {
- if (iter->second == id)
- break;
- }
- if (iter == outputIdMap.end())
- break;
- }
-
- newIdMap[output] = id;
- }
-
- layout.add_screen(Screen(newIdMap[output], crtc->x, crtc->y,
- crtc->mode->mode.width,
- crtc->mode->mode.height,
- 0));
- }
-
- /* Only keep the entries that are currently active */
- outputIdMap = newIdMap;
- #endif
-
- /*
- * Make sure we have something to display. Hopefully it's just temporary
- * that we have no active outputs...
- */
- if (layout.num_screens() == 0)
- layout.add_screen(Screen(0, 0, 0, pScreen->width, pScreen->height, 0));
-
- return layout;
- }
-
- #ifdef RANDR
-
- extern RRModePtr vncRandRModeGet(int width, int height);
-
- RRModePtr XserverDesktop::findRandRMode(RROutputPtr output, int width, int height)
- {
- RRModePtr mode;
-
- for (int i = 0;i < output->numModes;i++) {
- if ((output->modes[i]->mode.width == width) &&
- (output->modes[i]->mode.height == height))
- return output->modes[i];
- }
-
- mode = vncRandRModeGet(width, height);
- if (mode != NULL)
- return mode;
-
- return NULL;
- }
-
- #endif
-
- char* XserverDesktop::substitute(const char* varName)
- {
- if (strcmp(varName, "$$") == 0) {
- return rfb::strDup("$");
- }
- if (strcmp(varName, "$PORT") == 0) {
- char* str = new char[10];
- sprintf(str, "%d", listener ? listener->getMyPort() : 0);
- return str;
- }
- if (strcmp(varName, "$WIDTH") == 0) {
- char* str = new char[10];
- sprintf(str, "%d", width());
- return str;
- }
- if (strcmp(varName, "$HEIGHT") == 0) {
- char* str = new char[10];
- sprintf(str, "%d", height());
- return str;
- }
- if (strcmp(varName, "$APPLETWIDTH") == 0) {
- char* str = new char[10];
- sprintf(str, "%d", width());
- return str;
- }
- if (strcmp(varName, "$APPLETHEIGHT") == 0) {
- char* str = new char[10];
- sprintf(str, "%d", height());
- return str;
- }
- if (strcmp(varName, "$DESKTOP") == 0) {
- return rfb::strDup(server->getName());
- }
- if (strcmp(varName, "$DISPLAY") == 0) {
- struct utsname uts;
- uname(&uts);
- char* str = new char[256];
- strncpy(str, uts.nodename, 240);
- str[239] = '\0'; /* Ensure string is zero-terminated */
- strcat(str, ":");
- strncat(str, display, 10);
- return str;
- }
- if (strcmp(varName, "$USER") == 0) {
- struct passwd* user = getpwuid(getuid());
- return rfb::strDup(user ? user->pw_name : "?");
- }
- return 0;
- }
-
- rfb::VNCServerST::queryResult
- XserverDesktop::queryConnection(network::Socket* sock,
- const char* userName,
- char** reason) {
- if (queryConnectId) {
- *reason = strDup("Another connection is currently being queried.");
- return rfb::VNCServerST::REJECT;
- }
- queryConnectAddress.replaceBuf(sock->getPeerAddress());
- if (!userName)
- userName = "(anonymous)";
- queryConnectUsername.replaceBuf(strDup(userName));
- queryConnectId = sock;
- vncQueryConnect(this, sock);
- return rfb::VNCServerST::PENDING;
- }
-
- void XserverDesktop::bell()
- {
- server->bell();
- }
-
- void XserverDesktop::serverCutText(const char* str, int len)
- {
- try {
- server->serverCutText(str, len);
- } catch (rdr::Exception& e) {
- vlog.error("XserverDesktop::serverCutText: %s",e.str());
- }
- }
-
- void XserverDesktop::setDesktopName(const char* name)
- {
- try {
- server->setName(name);
- } catch (rdr::Exception& e) {
- vlog.error("XserverDesktop::setDesktopName: %s",e.str());
- }
- }
-
- void XserverDesktop::setCursor(CursorPtr cursor)
- {
- try {
- int w = cursor->bits->width;
- int h = cursor->bits->height;
-
- rdr::U8* cursorData = new rdr::U8[w * h * (getPF().bpp / 8)];
-
- int rfbMaskBytesPerRow = (w + 7) / 8;
-
- rdr::U8* cursorMask = new rdr::U8[rfbMaskBytesPerRow * h];
-
- #ifdef ARGB_CURSOR
- if (cursor->bits->argb) {
- rdr::U8 *out;
- CARD32 *in;
-
- rdr::U8 rgb[3];
-
- memset(cursorMask, 0, rfbMaskBytesPerRow * h);
-
- in = cursor->bits->argb;
- out = cursorData;
- for (int y = 0; y < h; y++) {
- for (int x = 0; x < w; x++) {
- rgb[0] = (*in >> 16) & 0xff;
- rgb[1] = (*in >> 8) & 0xff;
- rgb[2] = (*in >> 0) & 0xff;
-
- getPF().bufferFromRGB(out, rgb, 1);
-
- if (((*in >> 24) & 0xff) > 127)
- cursorMask[y * rfbMaskBytesPerRow + x/8] |= 0x80>>(x%8);
-
- in++;
- out += getPF().bpp/8;
- }
- }
- } else {
- #endif
- rdr::U8 rgb[3];
- rdr::U8 fg[4], bg[4];
-
- rdr::U8* buffer;
-
- rgb[0] = cursor->foreRed;
- rgb[1] = cursor->foreGreen;
- rgb[2] = cursor->foreBlue;
- getPF().bufferFromRGB(fg, rgb, 1);
-
- rgb[0] = cursor->backRed;
- rgb[1] = cursor->backGreen;
- rgb[2] = cursor->backBlue;
- getPF().bufferFromRGB(bg, rgb, 1);
-
- int xMaskBytesPerRow = BitmapBytePad(w);
-
- buffer = cursorData;
-
- for (int y = 0; y < h; y++) {
- for (int x = 0; x < w; x++) {
- rdr::U8 *pixel;
- int byte = y * xMaskBytesPerRow + x / 8;
- #if (BITMAP_BIT_ORDER == MSBFirst)
- int bit = 7 - x % 8;
- #else
- int bit = x % 8;
- #endif
-
- if (cursor->bits->source[byte] & (1 << bit))
- pixel = fg;
- else
- pixel = bg;
-
- memcpy(buffer, pixel, getPF().bpp/8);
- buffer += getPF().bpp/8;
- }
- }
-
- for (int j = 0; j < h; j++) {
- for (int i = 0; i < rfbMaskBytesPerRow; i++)
- #if (BITMAP_BIT_ORDER == MSBFirst)
- cursorMask[j * rfbMaskBytesPerRow + i]
- = cursor->bits->mask[j * xMaskBytesPerRow + i];
- #else
- cursorMask[j * rfbMaskBytesPerRow + i]
- = reverseBits[cursor->bits->mask[j * xMaskBytesPerRow + i]];
- #endif
- }
- #ifdef ARGB_CURSOR
- }
- #endif
-
- server->setCursor(cursor->bits->width, cursor->bits->height,
- Point(cursor->bits->xhot, cursor->bits->yhot),
- cursorData, cursorMask);
- delete [] cursorData;
- delete [] cursorMask;
- } catch (rdr::Exception& e) {
- vlog.error("XserverDesktop::setCursor: %s",e.str());
- }
- }
-
- void XserverDesktop::add_changed(RegionPtr reg)
- {
- if (ignoreHooks_) return;
- if (grabbing) return;
- try {
- rfb::Region rfbReg;
- rfbReg.setExtentsAndOrderedRects((ShortRect*)REGION_EXTENTS(pScreen, reg),
- REGION_NUM_RECTS(reg),
- (ShortRect*)REGION_RECTS(reg));
- server->add_changed(rfbReg);
- } catch (rdr::Exception& e) {
- vlog.error("XserverDesktop::add_changed: %s",e.str());
- }
- }
-
- void XserverDesktop::add_copied(RegionPtr dst, int dx, int dy)
- {
- if (ignoreHooks_) return;
- if (grabbing) return;
- try {
- rfb::Region rfbReg;
- rfbReg.setExtentsAndOrderedRects((ShortRect*)REGION_EXTENTS(pScreen, dst),
- REGION_NUM_RECTS(dst),
- (ShortRect*)REGION_RECTS(dst));
- server->add_copied(rfbReg, rfb::Point(dx, dy));
- } catch (rdr::Exception& e) {
- vlog.error("XserverDesktop::add_copied: %s",e.str());
- }
- }
-
- static struct timeval XserverDesktopTimeout;
-
- void XserverDesktop::blockHandler(fd_set* fds, OSTimePtr timeout)
- {
- // We don't have a good callback for when we can init input devices[1],
- // so we abuse the fact that this routine will be called first thing
- // once the dix is done initialising.
- // [1] Technically Xvnc has InitInput(), but libvnc.so has nothing.
- inputDevice->InitInputDevice();
-
- try {
- int nextTimeout;
-
- // Add all sockets we want read events for, after purging
- // any closed sockets.
- if (listener)
- FD_SET(listener->getFd(), fds);
- if (httpListener)
- FD_SET(httpListener->getFd(), fds);
-
- std::list<Socket*> sockets;
- server->getSockets(&sockets);
- std::list<Socket*>::iterator i;
- for (i = sockets.begin(); i != sockets.end(); i++) {
- int fd = (*i)->getFd();
- if ((*i)->isShutdown()) {
- vlog.debug("client gone, sock %d",fd);
- server->removeSocket(*i);
- vncClientGone(fd);
- delete (*i);
- } else {
- FD_SET(fd, fds);
- }
- }
- if (httpServer) {
- httpServer->getSockets(&sockets);
- for (i = sockets.begin(); i != sockets.end(); i++) {
- int fd = (*i)->getFd();
- if ((*i)->isShutdown()) {
- vlog.debug("http client gone, sock %d",fd);
- httpServer->removeSocket(*i);
- delete (*i);
- } else {
- FD_SET(fd, fds);
- }
- }
- }
-
- // Then check when the next timer will expire.
- // (this unfortunately also triggers any already expired timers)
- nextTimeout = server->checkTimeouts();
- if (nextTimeout > 0) {
- // No timeout specified? Or later timeout than we need?
- if ((*timeout == NULL) ||
- ((*timeout)->tv_sec > (nextTimeout/1000)) ||
- (((*timeout)->tv_sec == (nextTimeout/1000)) &&
- ((*timeout)->tv_usec > ((nextTimeout%1000)*1000)))) {
- XserverDesktopTimeout.tv_sec = nextTimeout/1000;
- XserverDesktopTimeout.tv_usec = (nextTimeout%1000)*1000;
- *timeout = &XserverDesktopTimeout;
- }
- }
-
- } catch (rdr::Exception& e) {
- vlog.error("XserverDesktop::blockHandler: %s",e.str());
- }
- }
-
- void XserverDesktop::wakeupHandler(fd_set* fds, int nfds)
- {
- try {
- // First check for file descriptors with something to do
- if (nfds >= 1) {
-
- if (listener) {
- if (FD_ISSET(listener->getFd(), fds)) {
- FD_CLR(listener->getFd(), fds);
- Socket* sock = listener->accept();
- sock->outStream().setBlocking(false);
- server->addSocket(sock);
- vlog.debug("new client, sock %d",sock->getFd());
- }
- }
-
- if (httpListener) {
- if (FD_ISSET(httpListener->getFd(), fds)) {
- FD_CLR(httpListener->getFd(), fds);
- Socket* sock = httpListener->accept();
- sock->outStream().setBlocking(false);
- httpServer->addSocket(sock);
- vlog.debug("new http client, sock %d",sock->getFd());
- }
- }
-
- std::list<Socket*> sockets;
- server->getSockets(&sockets);
- std::list<Socket*>::iterator i;
- for (i = sockets.begin(); i != sockets.end(); i++) {
- int fd = (*i)->getFd();
- if (FD_ISSET(fd, fds)) {
- FD_CLR(fd, fds);
- server->processSocketEvent(*i);
- }
- }
-
- if (httpServer) {
- httpServer->getSockets(&sockets);
- for (i = sockets.begin(); i != sockets.end(); i++) {
- int fd = (*i)->getFd();
- if (FD_ISSET(fd, fds)) {
- FD_CLR(fd, fds);
- httpServer->processSocketEvent(*i);
- }
- }
- }
-
- inputDevice->PointerSync();
- }
-
- // Then let the timers do some processing. Rescheduling is done in
- // blockHandler().
- server->checkTimeouts();
- } catch (rdr::Exception& e) {
- vlog.error("XserverDesktop::wakeupHandler: %s",e.str());
- }
- }
-
- void XserverDesktop::writeBlockHandler(fd_set* fds)
- {
- try {
- std::list<Socket*> sockets;
- std::list<Socket*>::iterator i;
-
- server->getSockets(&sockets);
- for (i = sockets.begin(); i != sockets.end(); i++) {
- int fd = (*i)->getFd();
- if ((*i)->isShutdown()) {
- vlog.debug("client gone, sock %d",fd);
- server->removeSocket(*i);
- vncClientGone(fd);
- delete (*i);
- } else {
- if ((*i)->outStream().bufferUsage() > 0)
- FD_SET(fd, fds);
- }
- }
-
- if (httpServer) {
- httpServer->getSockets(&sockets);
- for (i = sockets.begin(); i != sockets.end(); i++) {
- int fd = (*i)->getFd();
- if ((*i)->isShutdown()) {
- vlog.debug("http client gone, sock %d",fd);
- httpServer->removeSocket(*i);
- delete (*i);
- } else {
- if ((*i)->outStream().bufferUsage() > 0)
- FD_SET(fd, fds);
- }
- }
- }
- } catch (rdr::Exception& e) {
- vlog.error("XserverDesktop::writeBlockHandler: %s",e.str());
- }
- }
-
- void XserverDesktop::writeWakeupHandler(fd_set* fds, int nfds)
- {
- if (nfds < 1)
- return;
-
- try {
- std::list<Socket*> sockets;
- std::list<Socket*>::iterator i;
-
- server->getSockets(&sockets);
- for (i = sockets.begin(); i != sockets.end(); i++) {
- int fd = (*i)->getFd();
- if (FD_ISSET(fd, fds)) {
- FD_CLR(fd, fds);
- (*i)->outStream().flush();
- }
- }
-
- if (httpServer) {
- httpServer->getSockets(&sockets);
- for (i = sockets.begin(); i != sockets.end(); i++) {
- int fd = (*i)->getFd();
- if (FD_ISSET(fd, fds)) {
- FD_CLR(fd, fds);
- (*i)->outStream().flush();
- }
- }
- }
- } catch (rdr::Exception& e) {
- vlog.error("XserverDesktop::writeWakeupHandler: %s",e.str());
- }
- }
-
- void XserverDesktop::addClient(Socket* sock, bool reverse)
- {
- vlog.debug("new client, sock %d reverse %d",sock->getFd(),reverse);
- server->addSocket(sock, reverse);
- }
-
- void XserverDesktop::disconnectClients()
- {
- vlog.debug("disconnecting all clients");
- return server->closeClients("Disconnection from server end");
- }
-
-
- int XserverDesktop::getQueryTimeout(void* opaqueId,
- const char** address,
- const char** username)
- {
- if (opaqueId && queryConnectId == opaqueId) {
- vlog.info("address=%s, username=%s, timeout=%d",
- queryConnectAddress.buf, queryConnectUsername.buf,
- (int)queryConnectTimeout);
- if (address) *address = queryConnectAddress.buf;
- if (username) *username = queryConnectUsername.buf;
- return queryConnectTimeout;
- }
- return 0;
- }
-
- void XserverDesktop::approveConnection(void* opaqueId, bool accept,
- const char* rejectMsg)
- {
- if (queryConnectId == opaqueId) {
- server->approveConnection((network::Socket*)opaqueId, accept, rejectMsg);
- queryConnectId = 0;
- }
- }
-
- ///////////////////////////////////////////////////////////////////////////
- //
- // SDesktop callbacks
-
-
- void XserverDesktop::pointerEvent(const Point& pos, int buttonMask)
- {
- inputDevice->PointerMove(pos);
- inputDevice->PointerButtonAction(buttonMask);
- }
-
- void XserverDesktop::clientCutText(const char* str, int len)
- {
- vncClientCutText(str, len);
- }
-
- extern RROutputPtr vncRandROutputCreate(ScreenPtr pScreen);
-
- unsigned int XserverDesktop::setScreenLayout(int fb_width, int fb_height,
- const rfb::ScreenSet& layout)
- {
- #ifndef RANDR
- return rfb::resultProhibited;
- #else
- int availableOutputs;
- Bool ret;
-
- rrScrPrivPtr rp = rrGetScrPriv(pScreen);
-
- /*
- * First check that we don't have any active clone modes. That's just
- * too messy to deal with.
- */
- for (int i = 0;i < rp->numCrtcs;i++) {
- if (rp->crtcs[i]->numOutputs > 1) {
- vlog.error("Clone mode active. Refusing to touch screen layout.");
- return rfb::resultInvalid;
- }
- }
-
- /*
- * Next count how many useful outputs we have...
- *
- * This gets slightly complicated because we might need to hook a CRTC
- * up to the output, but also check that we don't try to use the same
- * CRTC for multiple outputs.
- */
- std::set<RRCrtcPtr> usedCrtcs;
- availableOutputs = 0;
- for (int i = 0;i < rp->numOutputs;i++) {
- RROutputPtr output;
-
- output = rp->outputs[i];
-
- if (output->crtc != NULL)
- availableOutputs++;
- else {
- for (int j = 0;j < output->numCrtcs;j++) {
- if (output->crtcs[j]->numOutputs != 0)
- continue;
- if (usedCrtcs.count(output->crtcs[j]) != 0)
- continue;
-
- availableOutputs++;
- usedCrtcs.insert(output->crtcs[j]);
-
- break;
- }
- }
- }
-
- /* Try to create more outputs if needed... (only works on Xvnc) */
- if (layout.num_screens() > availableOutputs) {
- for (int i = 0;i < (layout.num_screens() - availableOutputs);i++) {
- RROutputPtr output;
- output = vncRandROutputCreate(pScreen);
- if (output == NULL) {
- vlog.error("Unable to create more screens, as needed by the new client layout.");
- return rfb::resultInvalid;
- }
- }
- }
-
- /* First we might need to resize the screen */
- if ((fb_width != pScreen->width) || (fb_height != pScreen->height)) {
- /* Try to retain DPI when we resize */
- ret = RRScreenSizeSet(pScreen, fb_width, fb_height,
- pScreen->mmWidth * fb_width / pScreen->width,
- pScreen->mmHeight * fb_height / pScreen->height);
- if (!ret) {
- vlog.error("Failed to resize screen to %dx%d", fb_width, fb_height);
- return rfb::resultInvalid;
- }
- }
-
- /* Next, reconfigure all known outputs, and turn off the other ones */
- for (int i = 0;i < rp->numOutputs;i++) {
- RROutputPtr output;
- RRCrtcPtr crtc;
- RRModePtr mode;
-
- ScreenSet::const_iterator iter;
-
- output = rp->outputs[i];
- crtc = output->crtc;
-
- /* Known? */
- if (outputIdMap.count(output) == 0)
- continue;
-
- /* A known output should have a CRTC, but double check... */
- if (crtc == NULL) {
- vlog.error("Existing output '%s' has unexpectedly been disabled",
- output->name);
- continue;
- }
-
- /* Find the corresponding screen... */
- for (iter = layout.begin();iter != layout.end();++iter) {
- if (iter->id == outputIdMap[output])
- break;
- }
-
- /* Missing? */
- if (iter == layout.end()) {
- /* Disable and move on... */
- ret = RRCrtcSet(crtc, NULL, crtc->x, crtc->y, crtc->rotation, 0, NULL);
- if (!ret) {
- vlog.error("Failed to disable unused CRTC for output '%s'",
- output->name);
- return rfb::resultInvalid;
- }
- outputIdMap.erase(output);
- continue;
- }
-
- /* Need to switch mode? */
- if ((crtc->mode->mode.width == iter->dimensions.width()) &&
- (crtc->mode->mode.height == iter->dimensions.height()))
- mode = crtc->mode;
- else {
- mode = findRandRMode(output, iter->dimensions.width(),
- iter->dimensions.height());
- if (mode == NULL) {
- vlog.error("Failed to find a suitable mode for %dx%d for output '%s'",
- iter->dimensions.width(), iter->dimensions.height(),
- output->name);
- return rfb::resultInvalid;
- }
- }
-
- /* Reconfigure new mode and position */
- ret = RRCrtcSet(crtc, mode, iter->dimensions.tl.x, iter->dimensions.tl.y,
- crtc->rotation, crtc->numOutputs, crtc->outputs);
- if (!ret) {
- vlog.error("Failed to reconfigure output '%s' to %dx%d+%d+%d",
- output->name,
- iter->dimensions.width(), iter->dimensions.height(),
- iter->dimensions.tl.x, iter->dimensions.tl.y);
- return rfb::resultInvalid;
- }
- }
-
- /* Finally, allocate new outputs for new screens */
- ScreenSet::const_iterator iter;
- for (iter = layout.begin();iter != layout.end();++iter) {
- OutputIdMap::const_iterator oi;
-
- RROutputPtr output;
- RRCrtcPtr crtc;
- RRModePtr mode;
-
- int i;
-
- /* Does this screen have an output already? */
- for (oi = outputIdMap.begin();oi != outputIdMap.end();++oi) {
- if (oi->second == iter->id)
- break;
- }
-
- if (oi != outputIdMap.end())
- continue;
-
- /* Find an unused output */
- for (i = 0;i < rp->numOutputs;i++) {
- output = rp->outputs[i];
- crtc = output->crtc;
-
- /* In use? */
- if (outputIdMap.count(output) == 1)
- continue;
-
- /* Need a CRTC? */
- if (crtc == NULL) {
- for (int j = 0;j < output->numCrtcs;j++) {
- if (output->crtcs[j]->numOutputs != 0)
- continue;
-
- crtc = output->crtcs[j];
- break;
- }
-
- /* Couldn't find one... */
- if (crtc == NULL)
- continue;
-
- ret = RRCrtcSet(crtc, NULL, 0, 0, RR_Rotate_0,
- 1, &output);
- if (!ret) {
- vlog.error("Failed to associate a CRTC with output '%s'",
- output->name);
- return rfb::resultInvalid;
- }
- }
-
- break;
- }
-
- /* Shouldn't happen */
- if (i == rp->numOutputs)
- return rfb::resultInvalid;
-
- mode = findRandRMode(output, iter->dimensions.width(),
- iter->dimensions.height());
- if (mode == NULL) {
- vlog.error("Failed to find a suitable mode for %dx%d for output '%s'",
- iter->dimensions.width(), iter->dimensions.height(),
- output->name);
- return rfb::resultInvalid;
- }
-
- /*
- * Make sure we already have an entry for this, or
- * computeScreenLayout() will think it is a brand new output and
- * assign it a random id.
- */
- outputIdMap[output] = iter->id;
-
- /* Reconfigure new mode and position */
- ret = RRCrtcSet(crtc, mode, iter->dimensions.tl.x, iter->dimensions.tl.y,
- crtc->rotation, crtc->numOutputs, crtc->outputs);
- if (!ret) {
- vlog.error("Failed to reconfigure output '%s' to %dx%d+%d+%d",
- output->name,
- iter->dimensions.width(), iter->dimensions.height(),
- iter->dimensions.tl.x, iter->dimensions.tl.y);
- return rfb::resultInvalid;
- }
- }
-
- /*
- * Update timestamp for when screen layout was last changed.
- * This is normally done in the X11 request handlers, which is
- * why we have to deal with it manually here.
- */
- rp->lastSetTime = currentTime;
-
- return rfb::resultSuccess;
- #endif
- }
-
- void XserverDesktop::grabRegion(const rfb::Region& region)
- {
- if (directFbptr) return;
- if (!pScreen->GetImage) {
- vlog.error("VNC error: pScreen->GetImage == 0");
- return;
- }
-
- grabbing = true;
-
- int bytesPerPixel = format.bpp/8;
- int bytesPerRow = pScreen->width * bytesPerPixel;
-
- std::vector<rfb::Rect> rects;
- std::vector<rfb::Rect>::iterator i;
- region.get_rects(&rects);
- for (i = rects.begin(); i != rects.end(); i++) {
- for (int y = i->tl.y; y < i->br.y; y++) {
- DrawablePtr pDrawable;
- #if XORG < 19
- pDrawable = (DrawablePtr) WindowTable[pScreen->myNum];
- #else
- pDrawable = (DrawablePtr) pScreen->root;
- #endif
-
- (*pScreen->GetImage) (pDrawable, i->tl.x, y, i->width(), 1,
- ZPixmap, (unsigned long)~0L,
- ((char*)data
- + y * bytesPerRow + i->tl.x * bytesPerPixel));
- }
- }
- grabbing = false;
- }
-
- void XserverDesktop::keyEvent(rdr::U32 keysym, bool down)
- {
- if (down)
- inputDevice->KeyboardPress(keysym);
- else
- inputDevice->KeyboardRelease(keysym);
- }
|