krfb/krfbserver.cpp

   1 /* This file is part of the KDE project
   2    Copyright (C) 2007 Alessandro Praduroux <pradu@pradu.it>
   3              (C) 2001-2003 by Tim Jansen <tim@tjansen.de>
   4
   5    This program is free software; you can redistribute it and/or
   6    modify it under the terms of the GNU General Public
   7    License as published by the Free Software Foundation; either
   8    version 2 of the License, or (at your option) any later version.
   9 */
  10
  11
  12 #include "krfbserver.h"
  13 #include "krfbserver.moc"
  14
  15 #include <QTcpServer>
  16 #include <QTcpSocket>
  17 #include <QTimer>
  18 #include <QHostInfo>
  19 #include <QApplication>
  20 #include <QDesktopWidget>
  21 #include <QPointer>
  22
  23 #include <KGlobal>
  24 #include <KUser>
  25 #include <KLocale>
  26 #include <KDebug>
  27 #include <KMessageBox>
  28 #include <dnssd/publicservice.h>
  29
  30 #include "connectioncontroller.h"
  31 #include "framebuffer.h"
  32 #include "krfbconfig.h"
  33 #include "invitationmanager.h"
  34
  35 #include <X11/Xutil.h>
  36 #include <X11/extensions/XTest.h>
  37
  38
  39 static const char* cur=
  40         "                   "
  41         " x                 "
  42         " xx                "
  43         " xxx               "
  44         " xxxx              "
  45         " xxxxx             "
  46         " xxxxxx            "
  47         " xxxxxxx           "
  48         " xxxxxxxx          "
  49         " xxxxxxxxx         "
  50         " xxxxxxxxxx        "
  51         " xxxxx             "
  52         " xx xxx            "
  53         " x  xxx            "
  54         "     xxx           "
  55         "     xxx           "
  56         "      xxx          "
  57         "      xxx          "
  58         "                   ";
  59
  60 static const char* mask=
  61         "xx                 "
  62         "xxx                "
  63         "xxxx               "
  64         "xxxxx              "
  65         "xxxxxx             "
  66         "xxxxxxx            "
  67         "xxxxxxxx           "
  68         "xxxxxxxxx          "
  69         "xxxxxxxxxx         "
  70         "xxxxxxxxxxx        "
  71         "xxxxxxxxxxxx       "
  72         "xxxxxxxxxx         "
  73         "xxxxxxxx           "
  74         "xxxxxxxx           "
  75         "xx  xxxxx          "
  76         "    xxxxx          "
  77         "     xxxxx         "
  78         "     xxxxx         "
  79         "      xxx          ";
  80
  81 static rfbCursorPtr myCursor;
  82
  83
  84 static enum rfbNewClientAction newClientHook(struct _rfbClientRec *cl)
  85 {
  86     KrfbServer *server = KrfbServer::self();
  87     return server->handleNewClient(cl);
  88 }
  89
  90 static rfbBool passwordCheck(rfbClientPtr cl,
  91                              const char* encryptedPassword,
  92                              int len)
  93 {
  94     ConnectionController *cc = static_cast<ConnectionController *>(cl->clientData);
  95     return cc->handleCheckPassword(cl, encryptedPassword, len);
  96 }
  97
  98 static void keyboardHook(rfbBool down, rfbKeySym keySym, rfbClientPtr cl)
  99 {
 100     ConnectionController *cc = static_cast<ConnectionController *>(cl->clientData);
 101     cc->handleKeyEvent(down ? true : false, keySym);
 102 }
 103
 104 static void pointerHook(int bm, int x, int y, rfbClientPtr cl)
 105 {
 106     ConnectionController *cc = static_cast<ConnectionController *>(cl->clientData);
 107     cc->handlePointerEvent(bm, x, y);
 108 }
 109
 110 static void clipboardHook(char* str,int len, rfbClientPtr cl)
 111 {
 112     ConnectionController *cc = static_cast<ConnectionController *>(cl->clientData);
 113     cc->clipboardToServer(QString::fromUtf8(str, len));
 114 }
 115
 116
 117 class KrfbServer::KrfbServerP {
 118
 119     public:
 120         KrfbServerP() : fb(0), screen(0), numClients(0) {};
 121
 122         FrameBuffer *fb;
 123         QList< QPointer<ConnectionController> > controllers;
 124         rfbScreenInfoPtr screen;
 125         bool running;
 126         int numClients;
 127         QByteArray desktopName;
 128 };
 129
 130 class KrfbServerPrivate
 131 {
 132 public:
 133     KrfbServer instance;
 134 };
 135
 136 K_GLOBAL_STATIC(KrfbServerPrivate, krfbServerPrivate)
 137
 138 KrfbServer * KrfbServer::self() {
 139     return &krfbServerPrivate->instance;
 140 }
 141
 142
 143 KrfbServer::KrfbServer()
 144     :d(new KrfbServerP)
 145 {
 146     kDebug() << "starting ";
 147     d->running = true;
 148     d->fb = FrameBuffer::getFrameBuffer(QApplication::desktop()->winId(), this);
 149     QTimer::singleShot(0, this, SLOT(startListening()));
 150     connect(InvitationManager::self(), SIGNAL(invitationNumChanged(int)),SLOT(updatePassword()));
 151 }
 152
 153 KrfbServer::~KrfbServer()
 154 {
 155     delete d;
 156 }
 157
 158
 159 void KrfbServer::startListening()
 160 {
 161     rfbScreenInfoPtr screen;
 162
 163     int port = KrfbConfig::port();
 164
 165     int w = d->fb->width();
 166     int h = d->fb->height();
 167     int depth = d->fb->depth();
 168
 169     rfbLogEnable(0);
 170     screen = rfbGetScreen(0, 0, w, h, 8, 3,depth / 8);
 171     screen->paddedWidthInBytes = d->fb->paddedWidth();
 172
 173     d->fb->getServerFormat(screen->serverFormat);
 174
 175     screen->frameBuffer = d->fb->data();
 176     d->screen = screen;
 177     screen->autoPort = 0;
 178     screen->port = port;
 179
 180     // server hooks
 181     screen->newClientHook = newClientHook;
 182
 183     screen->kbdAddEvent = keyboardHook;
 184     screen->ptrAddEvent = pointerHook;
 185     screen->newClientHook = newClientHook;
 186     screen->passwordCheck = passwordCheck;
 187     screen->setXCutText = clipboardHook;
 188
 189     d->desktopName = i18n("%1@%2 (shared desktop)", KUser().loginName(), QHostInfo::localHostName()).toLatin1();
 190     screen->desktopName = d->desktopName.constData();
 191
 192     if (!myCursor) {
 193         myCursor = rfbMakeXCursor(19, 19, (char*) cur, (char*) mask);
 194     }
 195     screen->cursor = myCursor;
 196
 197     // configure passwords and desktop control here
 198     updateSettings();
 199
 200     rfbInitServer(screen);
 201     if (!rfbIsActive(screen)) {
 202         KMessageBox::error(0,i18n("Address already in use"),"krfb");
 203         disconnectAndQuit();
 204     };
 205
 206     if (KrfbConfig::publishService()) {
 207         DNSSD::PublicService *service = new DNSSD::PublicService(i18n("%1@%2 (shared desktop)", KUser().loginName(), QHostInfo::localHostName()),"_rfb._tcp",port);
 208         service->publishAsync();
 209     }
 210
 211     while (d->running) {
 212         foreach(const QRect &r, d->fb->modifiedTiles()) {
 213             rfbMarkRectAsModified(screen, r.x(), r.y(), r.right(), r.bottom());
 214         }
 215         rfbProcessEvents(screen, 100);
 216         qApp->processEvents();
 217     }
 218     rfbShutdownServer(screen, true);
 219     // framebuffer has to be deleted before X11 connection goes down
 220     delete d->fb;
 221     d->fb = 0;
 222
 223     emit quitApp();
 224 }
 225
 226
 227 void KrfbServer::enableDesktopControl(bool enable)
 228 {
 229     foreach (QPointer<ConnectionController> ptr, d->controllers) {
 230         if (ptr) {
 231             ptr->setControlEnabled(enable);
 232         }
 233     }
 234 }
 235
 236 void KrfbServer::disconnectAndQuit()
 237 {
 238     d->running = false;
 239 }
 240
 241 enum rfbNewClientAction KrfbServer::handleNewClient(struct _rfbClientRec * cl)
 242 {
 243     ConnectionController *cc = new ConnectionController(cl, this);
 244     if (d->numClients++ == 0)
 245         d->fb->startMonitor();
 246
 247     d->controllers.append(cc);
 248     cc->setControlEnabled(KrfbConfig::allowDesktopControl());
 249
 250     connect(cc, SIGNAL(sessionEstablished(QString)), SIGNAL(sessionEstablished(QString)));
 251     connect(cc, SIGNAL(clientDisconnected(ConnectionController *)),SLOT(clientDisconnected(ConnectionController *)));
 252
 253     return cc->handleNewClient();
 254 }
 255
 256 void KrfbServer::updateSettings()
 257 {
 258     enableDesktopControl(KrfbConfig::allowDesktopControl());
 259     updatePassword();
 260 }
 261
 262 void KrfbServer::updatePassword()
 263 {
 264
 265     if (!d->screen) return;
 266     QString pw = KrfbConfig::uninvitedConnectionPassword();
 267     kDebug() << "password: " << pw << " allow " <<
 268             KrfbConfig::allowUninvitedConnections() <<
 269             " invitations " << InvitationManager::self()->activeInvitations() << endl;
 270
 271     if (pw.isEmpty() && InvitationManager::self()->activeInvitations() == 0) {
 272         kDebug() << "no password from now on";
 273         d->screen->authPasswdData = (void *)0;
 274     } else {
 275         kDebug() << "Ask for password to accept connections";
 276         d->screen->authPasswdData = (void *)1;
 277     }
 278 }
 279
 280 bool KrfbServer::checkX11Capabilities() {
 281     int bp1, bp2, majorv, minorv;
 282     Bool r = XTestQueryExtension(QX11Info::display(), &bp1, &bp2,
 283                                  &majorv, &minorv);
 284     if ((!r) || (((majorv*1000)+minorv) < 2002)) {
 285         KMessageBox::error(0,
 286                            i18n("Your X11 Server does not support the required XTest extension version 2.2. Sharing your desktop is not possible."),
 287                                 i18n("Desktop Sharing Error"));
 288         return false;
 289     }
 290
 291     return true;
 292 }
 293
 294 void KrfbServer::clientDisconnected(ConnectionController *cc)
 295 {
 296     kDebug() << "clients--: " << d->numClients;
 297     if (!--d->numClients) {
 298         d->fb->stopMonitor();
 299     }
 300     disconnect(cc, SIGNAL(clientDisconnected(ConnectionController)),this, SLOT(clientDisconnected(ConnectionController)));
 301 }
 302
 303