/* Copyright (C) 2002-2005 RealVNC Ltd.  All Rights Reserved. *  * This is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. *  * This software is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the * GNU General Public License for more details. *  * You should have received a copy of the GNU General Public License * along with this software; if not, write to the Free Software * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, * USA. *///// XserverDesktop.cxx//#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"extern "C" {#define public c_public#define class c_class  // windowTable is in globals.h in XFree 4, but not in XFree 3 unfortunatelyextern WindowPtr *WindowTable;extern char *display;#include "inputstr.h"#include "servermd.h"#include "colormapst.h"#include "resource.h"#include "cursorstr.h"#include "windowstr.h"#define XK_CYRILLIC#include "keysym.h"#undef public#undef class}using namespace rfb;using namespace network;static LogWriter vlog("XserverDesktop");rfb::IntParameter deferUpdateTime("DeferUpdate",                                  "Time in milliseconds to defer updates",40);rfb::BoolParameter alwaysSetDeferUpdateTimer("AlwaysSetDeferUpdateTimer",                  "Always reset the defer update timer on every change",false);IntParameter queryConnectTimeout("QueryConnectTimeout",                                 "Number of seconds to show the Accept Connection dialog before "                                 "rejecting the connection",                                 10);static KeyCode KeysymToKeycode(KeySymsPtr keymap, KeySym ks, int* col);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, void* fbptr)  : pScreen(pScreen_), deferredUpdateTimer(0), dummyTimer(0),    server(0), httpServer(0),    listener(listener_), httpListener(httpListener_),    cmap(0), deferredUpdateTimerSet(false),    grabbing(false), ignoreHooks_(false), directFbptr(fbptr != 0),    oldButtonMask(0),    queryConnectId(0){  int i;  format.depth = pScreen->rootDepth;  for (i = 0; i < screenInfo.numPixmapFormats; i++) {    if (screenInfo.formats[i].depth == format.depth) {      format.bpp = screenInfo.formats[i].bitsPerPixel;      break;    }  }  if (i == screenInfo.numPixmapFormats) {    fprintf(stderr,"no pixmap format for root depth???\n");    abort();  }  format.bigEndian = (screenInfo.imageByteOrder == MSBFirst);  VisualPtr vis;  for (i = 0; i < pScreen->numVisuals; i++) {    if (pScreen->visuals[i].vid == pScreen->rootVisual) {      vis = &pScreen->visuals[i];      break;    }  }  if (i == pScreen->numVisuals) {    fprintf(stderr,"no visual rec for root visual???\n");    abort();  }  format.trueColour = (vis->c_class == TrueColor);  if (!format.trueColour && format.bpp != 8)    throw rfb::Exception("X server uses unsupported visual");  format.redShift   = ffs(vis->redMask) - 1;  format.greenShift = ffs(vis->greenMask) - 1;  format.blueShift  = ffs(vis->blueMask) - 1;  format.redMax     = vis->redMask   >> format.redShift;  format.greenMax   = vis->greenMask >> format.greenShift;  format.blueMax    = vis->blueMask  >> format.blueShift;  width_ = pScreen->width;  height_ = pScreen->height;  if (fbptr)    data = (rdr::U8*)fbptr;  else    data = new rdr::U8[pScreen->width * pScreen->height * (format.bpp/8)];  colourmap = this;  serverReset(pScreen);  server = new VNCServerST(name, this);  server->setPixelBuffer(this);  server->setQueryConnectionHandler(this);  if (httpListener)    httpServer = new FileHTTPServer(this);}XserverDesktop::~XserverDesktop(){  if (!directFbptr)    delete [] data;  TimerFree(deferredUpdateTimer);  TimerFree(dummyTimer);  delete httpServer;  delete server;}void XserverDesktop::serverReset(ScreenPtr pScreen_){  pScreen = pScreen_;  XID* ids = new XID[pScreen->maxInstalledCmaps];  int nmaps = (*pScreen->ListInstalledColormaps)(pScreen, ids);  cmap = (ColormapPtr)LookupIDByType(ids[0], RT_COLORMAP);  delete [] ids;}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() + 32);    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];    strncat(str, uts.nodename, 240);    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::queryResultXserverDesktop::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::setColormap(ColormapPtr cmap_){  if (cmap != cmap_) {    cmap = cmap_;    setColourMapEntries(0, 0);  }}void XserverDesktop::setColourMapEntries(ColormapPtr pColormap, int ndef,                                         xColorItem* pdef){  if (cmap != pColormap || ndef <= 0) return;  int first = pdef[0].pixel;  int n = 1;  for (int i = 1; i < ndef; i++) {    if (first + n == pdef[i].pixel) {      n++;    } else {      setColourMapEntries(first, n);      first = pdef[i].pixel;      n = 1;    }  }  setColourMapEntries(first, n);}void XserverDesktop::setColourMapEntries(int firstColour, int nColours){  try {    server->setColourMapEntries(firstColour, nColours);  } catch (rdr::Exception& e) {    vlog.error("XserverDesktop::setColourMapEntries: %s",e.str());  }}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::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)];    xColorItem fg, bg;    fg.red   = cursor->foreRed;    fg.green = cursor->foreGreen;    fg.blue  = cursor->foreBlue;    FakeAllocColor(cmap, &fg);    bg.red   = cursor->backRed;    bg.green = cursor->backGreen;    bg.blue  = cursor->backBlue;    FakeAllocColor(cmap, &bg);    FakeFreeColor(cmap, fg.pixel);    FakeFreeColor(cmap, bg.pixel);    int xMaskBytesPerRow = BitmapBytePad(w);    for (int y = 0; y < h; y++) {      for (int x = 0; x < w; x++) {        int byte = y * xMaskBytesPerRow + x / 8;#if (BITMAP_BIT_ORDER == MSBFirst)        int bit = 7 - x % 8;#else        int bit = x % 8;#endif        switch (getPF().bpp) {        case 8:          ((rdr::U8*)cursorData)[y * w + x]            = (cursor->bits->source[byte] & (1 << bit)) ? fg.pixel : bg.pixel;          break;        case 16:          ((rdr::U16*)cursorData)[y * w + x]            = (cursor->bits->source[byte] & (1 << bit)) ? fg.pixel : bg.pixel;          break;        case 32:          ((rdr::U32*)cursorData)[y * w + x]            = (cursor->bits->source[byte] & (1 << bit)) ? fg.pixel : bg.pixel;          break;