blob: 9e5ea0f77f4d52abeda963faad55174144b8a952 [file] [log] [blame]
/* 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 unfortunately
extern 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::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::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;
}
}
}
int rfbMaskBytesPerRow = (w + 7) / 8;
rdr::U8* cursorMask = new rdr::U8[rfbMaskBytesPerRow * h];
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
}
server->setCursor(cursor->bits->width, cursor->bits->height,
Point(cursor->bits->xhot, cursor->bits->yhot),
cursorData, cursorMask);
server->tryUpdate();
delete [] cursorData;
delete [] cursorMask;
} catch (rdr::Exception& e) {
vlog.error("XserverDesktop::setCursor: %s",e.str());
}
}
static void printRegion(RegionPtr reg)
{
int nrects = REGION_NUM_RECTS(reg);
fprintf(stderr,"Region num rects %2d extents %3d,%3d %3dx%3d\n",nrects,
(REGION_EXTENTS(pScreen,reg))->x1,
(REGION_EXTENTS(pScreen,reg))->y1,
(REGION_EXTENTS(pScreen,reg))->x2-(REGION_EXTENTS(pScreen,reg))->x1,
(REGION_EXTENTS(pScreen,reg))->y2-(REGION_EXTENTS(pScreen,reg))->y1);
for (int i = 0; i < nrects; i++) {
fprintf(stderr," rect %3d,%3d %3dx%3d\n",
REGION_RECTS(reg)[i].x1,
REGION_RECTS(reg)[i].y1,
REGION_RECTS(reg)[i].x2-REGION_RECTS(reg)[i].x1,
REGION_RECTS(reg)[i].y2-REGION_RECTS(reg)[i].y1);
}
}
CARD32 XserverDesktop::deferredUpdateTimerCallback(OsTimerPtr timer,
CARD32 now, pointer arg)
{
XserverDesktop* desktop = (XserverDesktop*)arg;
desktop->deferredUpdateTimerSet = false;
try {
desktop->server->tryUpdate();
} catch (rdr::Exception& e) {
vlog.error("XserverDesktop::deferredUpdateTimerCallback: %s",e.str());
}
return 0;
}
void XserverDesktop::deferUpdate()
{
if (deferUpdateTime != 0) {
if (!deferredUpdateTimerSet || alwaysSetDeferUpdateTimer) {
deferredUpdateTimerSet = true;
deferredUpdateTimer = TimerSet(deferredUpdateTimer, 0,
deferUpdateTime,
deferredUpdateTimerCallback, this);
}
} else {
server->tryUpdate();
}
}
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);
deferUpdate();
} 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));
deferUpdate();
} catch (rdr::Exception& e) {
vlog.error("XserverDesktop::add_copied: %s",e.str());
}
}
void XserverDesktop::positionCursor()
{
if (!cursorPos.equals(oldCursorPos)) {
oldCursorPos = cursorPos;
(*pScreen->SetCursorPosition) (pScreen, cursorPos.x, cursorPos.y, FALSE);
server->setCursorPos(cursorPos);
server->tryUpdate();
}
}
void XserverDesktop::blockHandler(fd_set* fds)
{
try {
ScreenPtr screenWithCursor = GetCurrentRootWindow()->drawable.pScreen;
if (screenWithCursor == pScreen) {
int x, y;
GetSpritePosition(&x, &y);
if (x != cursorPos.x || y != cursorPos.y) {
cursorPos = oldCursorPos = Point(x, y);
server->setCursorPos(cursorPos);
server->tryUpdate();
}
}
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);
}
}
}
} catch (rdr::Exception& e) {
vlog.error("XserverDesktop::blockHandler: %s",e.str());
}
}
static CARD32 dummyTimerCallback(OsTimerPtr timer, CARD32 now, pointer arg) {
return 0;
}
void XserverDesktop::wakeupHandler(fd_set* fds, int nfds)
{
try {
if (nfds >= 1) {
if (listener) {
if (FD_ISSET(listener->getFd(), fds)) {
FD_CLR(listener->getFd(), fds);
Socket* sock = listener->accept();
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();
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);
}
}
}
positionCursor();
}
int timeout = server->checkTimeouts();
if (timeout > 0) {
// set a dummy timer just so we are guaranteed be called again next time.
dummyTimer = TimerSet(dummyTimer, 0, timeout,
dummyTimerCallback, 0);
}
} catch (rdr::Exception& e) {
vlog.error("XserverDesktop::wakeupHandler: %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)
{
xEvent ev;
DevicePtr dev = LookupPointerDevice();
// SetCursorPosition seems to be very expensive (at least on XFree86 3.3.6
// for S3), so we delay calling it until positionCursor() is called at the
// end of processing a load of RFB.
//(*pScreen->SetCursorPosition) (pScreen, pos.x, pos.y, FALSE);
NewCurrentScreen(pScreen, pos.x, pos.y);
ev.u.u.type = MotionNotify;
ev.u.u.detail = 0;
ev.u.keyButtonPointer.rootX = pos.x;
ev.u.keyButtonPointer.rootY = pos.y;
ev.u.keyButtonPointer.time = GetTimeInMillis();
if (!pos.equals(cursorPos))
(*dev->processInputProc)(&ev, (DeviceIntPtr)dev, 1);
for (int i = 0; i < 5; i++) {
if ((buttonMask ^ oldButtonMask) & (1<<i)) {
// Do not use the pointer mapping. Treat VNC buttons as logical
// buttons.
ev.u.u.detail = i + 1;
ev.u.u.type = (buttonMask & (1<<i)) ? ButtonPress : ButtonRelease;
(*dev->processInputProc)(&ev, (DeviceIntPtr)dev, 1);
}
}
cursorPos = pos;
oldButtonMask = buttonMask;
}
void XserverDesktop::clientCutText(const char* str, int len)
{
vncClientCutText(str, len);
}
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++) {
(*pScreen->GetImage) ((DrawablePtr)WindowTable[pScreen->myNum],
i->tl.x, y, i->width(), 1,
ZPixmap, (unsigned long)~0L,
((char*)data
+ y * bytesPerRow + i->tl.x * bytesPerPixel));
}
}
grabbing = false;
}
void XserverDesktop::lookup(int index, int* r, int* g, int* b)
{
if ((cmap->c_class | DynamicClass) == DirectColor) {
VisualPtr v = cmap->pVisual;
*r = cmap->red [(index & v->redMask ) >> v->offsetRed ].co.local.red;
*g = cmap->green[(index & v->greenMask) >> v->offsetGreen].co.local.green;
*b = cmap->blue [(index & v->blueMask ) >> v->offsetBlue ].co.local.blue;
} else {
EntryPtr pent;
pent = (EntryPtr)&cmap->red[index];
if (pent->fShared) {
*r = pent->co.shco.red->color;
*g = pent->co.shco.green->color;
*b = pent->co.shco.blue->color;
} else {
*r = pent->co.local.red;
*g = pent->co.local.green;
*b = pent->co.local.blue;
}
}
}
//
// Keyboard handling
//
#define IS_PRESSED(keyc, keycode) \
((keyc)->down[(keycode) >> 3] & (1 << ((keycode) & 7)))
// ModifierState is a class which helps simplify generating a "fake" press
// or release of shift, ctrl, alt, etc. An instance of the class is created
// for every modifier which may need to be pressed or released. Then either
// press() or release() may be called to make sure that the corresponding keys
// are in the right state. The destructor of the class automatically reverts
// to the previous state. Each modifier may have multiple keys associated with
// it, so in the case of a fake release, this may involve releasing more than
// one key.
class ModifierState {
public:
ModifierState(DeviceIntPtr dev_, int modIndex_)
: dev(dev_), modIndex(modIndex_), nKeys(0), keys(0), pressed(false)
{
}
~ModifierState() {
for (int i = 0; i < nKeys; i++)
generateXKeyEvent(keys[i], !pressed);
delete [] keys;
}
void press() {
KeyClassPtr keyc = dev->key;
if (!(keyc->state & (1<<modIndex))) {
tempKeyEvent(keyc->modifierKeyMap[modIndex * keyc->maxKeysPerModifier],
true);
pressed = true;
}
}
void release() {
KeyClassPtr keyc = dev->key;
if (keyc->state & (1<<modIndex)) {
for (int k = 0; k < keyc->maxKeysPerModifier; k++) {
int keycode
= keyc->modifierKeyMap[modIndex * keyc->maxKeysPerModifier + k];
if (keycode && IS_PRESSED(keyc, keycode))
tempKeyEvent(keycode, false);
}
}
}
private:
void tempKeyEvent(int keycode, bool down) {
if (keycode) {
if (!keys) keys = new int[dev->key->maxKeysPerModifier];
keys[nKeys++] = keycode;
generateXKeyEvent(keycode, down);
}
}
void generateXKeyEvent(int keycode, bool down) {
xEvent ev;
ev.u.u.type = down ? KeyPress : KeyRelease;
ev.u.u.detail = keycode;
ev.u.keyButtonPointer.time = GetTimeInMillis();
(*dev->c_public.processInputProc)(&ev, dev, 1);
vlog.debug("fake keycode %d %s", keycode, down ? "down" : "up");
}
DeviceIntPtr dev;
int modIndex;
int nKeys;
int* keys;
bool pressed;
};
// altKeysym is a table of alternative keysyms which have the same meaning.
struct altKeysym_t {
KeySym a, b;
};
altKeysym_t altKeysym[] = {
{ XK_Shift_L, XK_Shift_R },
{ XK_Control_L, XK_Control_R },
{ XK_Meta_L, XK_Meta_R },
{ XK_Alt_L, XK_Alt_R },
{ XK_Super_L, XK_Super_R },
{ XK_Hyper_L, XK_Hyper_R },
{ XK_KP_Space, XK_space },
{ XK_KP_Tab, XK_Tab },
{ XK_KP_Enter, XK_Return },
{ XK_KP_F1, XK_F1 },
{ XK_KP_F2, XK_F2 },
{ XK_KP_F3, XK_F3 },
{ XK_KP_F4, XK_F4 },
{ XK_KP_Home, XK_Home },
{ XK_KP_Left, XK_Left },
{ XK_KP_Up, XK_Up },
{ XK_KP_Right, XK_Right },
{ XK_KP_Down, XK_Down },
{ XK_KP_Page_Up, XK_Page_Up },
{ XK_KP_Page_Down, XK_Page_Down },
{ XK_KP_End, XK_End },
{ XK_KP_Begin, XK_Begin },
{ XK_KP_Insert, XK_Insert },
{ XK_KP_Delete, XK_Delete },
{ XK_KP_Equal, XK_equal },
{ XK_KP_Multiply, XK_asterisk },
{ XK_KP_Add, XK_plus },
{ XK_KP_Separator, XK_comma },
{ XK_KP_Subtract, XK_minus },
{ XK_KP_Decimal, XK_period },
{ XK_KP_Divide, XK_slash },
{ XK_KP_0, XK_0 },
{ XK_KP_1, XK_1 },
{ XK_KP_2, XK_2 },
{ XK_KP_3, XK_3 },
{ XK_KP_4, XK_4 },
{ XK_KP_5, XK_5 },
{ XK_KP_6, XK_6 },
{ XK_KP_7, XK_7 },
{ XK_KP_8, XK_8 },
{ XK_KP_9, XK_9 },
};
// keyEvent() - work out the best keycode corresponding to the keysym sent by
// the viewer. This is non-trivial because we can't assume much about the
// local keyboard layout. We must also find out which column of the keyboard
// mapping the keysym is in, and alter the shift state appropriately. Column 0
// means both shift and "mode_switch" (AltGr) must be released, column 1 means
// shift must be pressed and mode_switch released, column 2 means shift must be
// released and mode_switch pressed, and column 3 means both shift and
// mode_switch must be pressed.
void XserverDesktop::keyEvent(rdr::U32 keysym, bool down)
{
if (keysym == XK_Caps_Lock) {
vlog.debug("Ignoring caps lock");
return;
}
DeviceIntPtr dev = (DeviceIntPtr)LookupKeyboardDevice();
KeyClassPtr keyc = dev->key;
KeySymsPtr keymap = &keyc->curKeySyms;
// find which modifier Mode_switch is on.
int modeSwitchMapIndex = 0;
for (int i = 3; i < 8; i++) {
for (int k = 0; k < keyc->maxKeysPerModifier; k++) {
int keycode = keyc->modifierKeyMap[i * keyc->maxKeysPerModifier + k];
for (int j = 0; j < keymap->mapWidth; j++) {
if (keycode != 0 &&
keymap->map[(keycode - keymap->minKeyCode)
* keymap->mapWidth + j] == XK_Mode_switch)
{
modeSwitchMapIndex = i;
break;
}
}
}
}
int col = 0;
if (keyc->state & (1<<ShiftMapIndex)) col |= 1;
if (modeSwitchMapIndex && (keyc->state & (1<<modeSwitchMapIndex))) col |= 2;
int kc = KeysymToKeycode(keymap, keysym, &col);
// Sort out the "shifted Tab" mess. If we are sent a shifted Tab, generate a
// local shifted Tab regardless of what the "shifted Tab" keysym is on the
// local keyboard (it might be Tab, ISO_Left_Tab or HP's private BackTab
// keysym, and quite possibly some others too). We never get ISO_Left_Tab
// here because it's already been translated in VNCSConnectionST.
if (keysym == XK_Tab && (keyc->state & (1<<ShiftMapIndex)))
col |= 1;
if (kc == 0) {
// Not a direct match in the local keyboard mapping. Check for alternative
// keysyms with the same meaning.
for (int i = 0; i < sizeof(altKeysym) / sizeof(altKeysym_t); i++) {
if (keysym == altKeysym[i].a)
kc = KeysymToKeycode(keymap, altKeysym[i].b, &col);
else if (keysym == altKeysym[i].b)
kc = KeysymToKeycode(keymap, altKeysym[i].a, &col);
if (kc) break;
}
}
if (kc == 0) {
// Last resort - dynamically add a new key to the keyboard mapping.
for (kc = keymap->maxKeyCode; kc >= keymap->minKeyCode; kc--) {
if (!keymap->map[(kc - keymap->minKeyCode) * keymap->mapWidth]) {
keymap->map[(kc - keymap->minKeyCode) * keymap->mapWidth] = keysym;
col = 0;
SendMappingNotify(MappingKeyboard, kc, 1, serverClient);
vlog.info("Added unknown keysym 0x%x to keycode %d",keysym,kc);
break;
}
}
if (kc < keymap->minKeyCode) {
vlog.info("Keyboard mapping full - ignoring unknown keysym 0x%x",keysym);
return;
}
}
// See if it's a modifier key. If so, then don't do any auto-repeat, because
// the X server will translate each press into a release followed by a press.
for (int i = 0; i < 8; i++) {
for (int k = 0; k < keyc->maxKeysPerModifier; k++) {
if (kc == keyc->modifierKeyMap[i * keyc->maxKeysPerModifier + k] &&
IS_PRESSED(keyc,kc) && down)
return;
}
}
ModifierState shift(dev, ShiftMapIndex);
ModifierState modeSwitch(dev, modeSwitchMapIndex);
if (down) {
if (col & 1)
shift.press();
else
shift.release();
if (modeSwitchMapIndex) {
if (col & 2)
modeSwitch.press();
else
modeSwitch.release();
}
}
vlog.debug("keycode %d %s", kc, down ? "down" : "up");
xEvent ev;
ev.u.u.type = down ? KeyPress : KeyRelease;
ev.u.u.detail = kc;
ev.u.keyButtonPointer.time = GetTimeInMillis();
(*dev->c_public.processInputProc)(&ev, dev, 1);
}
void XConvertCase(KeySym sym, KeySym *lower, KeySym *upper)
{
*lower = sym;
*upper = sym;
switch(sym >> 8) {
case 0: /* Latin 1 */
if ((sym >= XK_A) && (sym <= XK_Z))
*lower += (XK_a - XK_A);
else if ((sym >= XK_a) && (sym <= XK_z))
*upper -= (XK_a - XK_A);
else if ((sym >= XK_Agrave) && (sym <= XK_Odiaeresis))
*lower += (XK_agrave - XK_Agrave);
else if ((sym >= XK_agrave) && (sym <= XK_odiaeresis))
*upper -= (XK_agrave - XK_Agrave);
else if ((sym >= XK_Ooblique) && (sym <= XK_Thorn))
*lower += (XK_oslash - XK_Ooblique);
else if ((sym >= XK_oslash) && (sym <= XK_thorn))
*upper -= (XK_oslash - XK_Ooblique);
break;
case 1: /* Latin 2 */
/* Assume the KeySym is a legal value (ignore discontinuities) */
if (sym == XK_Aogonek)
*lower = XK_aogonek;
else if (sym >= XK_Lstroke && sym <= XK_Sacute)
*lower += (XK_lstroke - XK_Lstroke);
else if (sym >= XK_Scaron && sym <= XK_Zacute)
*lower += (XK_scaron - XK_Scaron);
else if (sym >= XK_Zcaron && sym <= XK_Zabovedot)
*lower += (XK_zcaron - XK_Zcaron);
else if (sym == XK_aogonek)
*upper = XK_Aogonek;
else if (sym >= XK_lstroke && sym <= XK_sacute)
*upper -= (XK_lstroke - XK_Lstroke);
else if (sym >= XK_scaron && sym <= XK_zacute)
*upper -= (XK_scaron - XK_Scaron);
else if (sym >= XK_zcaron && sym <= XK_zabovedot)
*upper -= (XK_zcaron - XK_Zcaron);
else if (sym >= XK_Racute && sym <= XK_Tcedilla)
*lower += (XK_racute - XK_Racute);
else if (sym >= XK_racute && sym <= XK_tcedilla)
*upper -= (XK_racute - XK_Racute);
break;
case 2: /* Latin 3 */
/* Assume the KeySym is a legal value (ignore discontinuities) */
if (sym >= XK_Hstroke && sym <= XK_Hcircumflex)
*lower += (XK_hstroke - XK_Hstroke);
else if (sym >= XK_Gbreve && sym <= XK_Jcircumflex)
*lower += (XK_gbreve - XK_Gbreve);
else if (sym >= XK_hstroke && sym <= XK_hcircumflex)
*upper -= (XK_hstroke - XK_Hstroke);
else if (sym >= XK_gbreve && sym <= XK_jcircumflex)
*upper -= (XK_gbreve - XK_Gbreve);
else if (sym >= XK_Cabovedot && sym <= XK_Scircumflex)
*lower += (XK_cabovedot - XK_Cabovedot);
else if (sym >= XK_cabovedot && sym <= XK_scircumflex)
*upper -= (XK_cabovedot - XK_Cabovedot);
break;
case 3: /* Latin 4 */
/* Assume the KeySym is a legal value (ignore discontinuities) */
if (sym >= XK_Rcedilla && sym <= XK_Tslash)
*lower += (XK_rcedilla - XK_Rcedilla);
else if (sym >= XK_rcedilla && sym <= XK_tslash)
*upper -= (XK_rcedilla - XK_Rcedilla);
else if (sym == XK_ENG)
*lower = XK_eng;
else if (sym == XK_eng)
*upper = XK_ENG;
else if (sym >= XK_Amacron && sym <= XK_Umacron)
*lower += (XK_amacron - XK_Amacron);
else if (sym >= XK_amacron && sym <= XK_umacron)
*upper -= (XK_amacron - XK_Amacron);
break;
case 6: /* Cyrillic */
/* Assume the KeySym is a legal value (ignore discontinuities) */
if (sym >= XK_Serbian_DJE && sym <= XK_Serbian_DZE)
*lower -= (XK_Serbian_DJE - XK_Serbian_dje);
else if (sym >= XK_Serbian_dje && sym <= XK_Serbian_dze)
*upper += (XK_Serbian_DJE - XK_Serbian_dje);
else if (sym >= XK_Cyrillic_YU && sym <= XK_Cyrillic_HARDSIGN)
*lower -= (XK_Cyrillic_YU - XK_Cyrillic_yu);
else if (sym >= XK_Cyrillic_yu && sym <= XK_Cyrillic_hardsign)
*upper += (XK_Cyrillic_YU - XK_Cyrillic_yu);
break;
case 7: /* Greek */
/* Assume the KeySym is a legal value (ignore discontinuities) */
if (sym >= XK_Greek_ALPHAaccent && sym <= XK_Greek_OMEGAaccent)
*lower += (XK_Greek_alphaaccent - XK_Greek_ALPHAaccent);
else if (sym >= XK_Greek_alphaaccent && sym <= XK_Greek_omegaaccent &&
sym != XK_Greek_iotaaccentdieresis &&
sym != XK_Greek_upsilonaccentdieresis)
*upper -= (XK_Greek_alphaaccent - XK_Greek_ALPHAaccent);
else if (sym >= XK_Greek_ALPHA && sym <= XK_Greek_OMEGA)
*lower += (XK_Greek_alpha - XK_Greek_ALPHA);
else if (sym >= XK_Greek_alpha && sym <= XK_Greek_omega &&
sym != XK_Greek_finalsmallsigma)
*upper -= (XK_Greek_alpha - XK_Greek_ALPHA);
break;
}
}
static KeySym KeyCodetoKeySym(KeySymsPtr keymap, int keycode, int col)
{
register int per = keymap->mapWidth;
register KeySym *syms;
KeySym lsym, usym;
if ((col < 0) || ((col >= per) && (col > 3)) ||
(keycode < keymap->minKeyCode) || (keycode > keymap->maxKeyCode))
return NoSymbol;
syms = &keymap->map[(keycode - keymap->minKeyCode) * per];
if (col < 4) {
if (col > 1) {
while ((per > 2) && (syms[per - 1] == NoSymbol))
per--;
if (per < 3)
col -= 2;
}
if ((per <= (col|1)) || (syms[col|1] == NoSymbol)) {
XConvertCase(syms[col&~1], &lsym, &usym);
if (!(col & 1))
return lsym;
// I'm commenting out this logic because it's incorrect even though it
// was copied from the Xlib sources. The X protocol book quite clearly
// states that where a group consists of element 1 being a non-alphabetic
// keysym and element 2 being NoSymbol that you treat the second element
// as being the same as the first. This also tallies with the behaviour
// produced by the installed Xlib on my linux box (I believe this is
// because it uses some XKB code rather than the original Xlib code -
// compare XKBBind.c with KeyBind.c in lib/X11).
// else if (usym == lsym)
// return NoSymbol;
else
return usym;
}
}
return syms[col];
}
// KeysymToKeycode() - find the keycode and column corresponding to the given
// keysym. The value of col passed in should be the column determined from the
// current shift state. If the keysym can be found in that column we prefer
// that to finding it in a different column (which would require fake events to
// alter the shift state).
static KeyCode KeysymToKeycode(KeySymsPtr keymap, KeySym ks, int* col)
{
register int i, j;
j = *col;
for (i = keymap->minKeyCode; i <= keymap->maxKeyCode; i++) {
if (KeyCodetoKeySym(keymap, i, j) == ks)
return i;
}
for (j = 0; j < keymap->mapWidth; j++) {
for (i = keymap->minKeyCode; i <= keymap->maxKeyCode; i++) {
if (KeyCodetoKeySym(keymap, i, j) == ks) {
*col = j;
return i;
}
}
}
return 0;
}