src/core/core.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-09 by the Quassel Project                          *
   3  *   devel@quassel-irc.org                                                 *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) version 3.                                           *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  19  ***************************************************************************/
  20
  21 #include <QMetaObject>
  22 #include <QMetaMethod>
  23 #include <QCoreApplication>
  24
  25 #include "core.h"
  26 #include "coresession.h"
  27 #include "coresettings.h"
  28 #include "quassel.h"
  29 #include "signalproxy.h"
  30 #include "sqlitestorage.h"
  31 #include "network.h"
  32 #include "logger.h"
  33
  34 #include "util.h"
  35
  36 Core *Core::instanceptr = 0;
  37
  38 Core *Core::instance() {
  39   if(instanceptr) return instanceptr;
  40   instanceptr = new Core();
  41   instanceptr->init();
  42   return instanceptr;
  43 }
  44
  45 void Core::destroy() {
  46   delete instanceptr;
  47   instanceptr = 0;
  48 }
  49
  50 Core::Core() : storage(0) {
  51   _startTime = QDateTime::currentDateTime().toUTC();  // for uptime :)
  52
  53   loadTranslation(QLocale::system());
  54
  55   // Register storage backends here!
  56   registerStorageBackend(new SqliteStorage(this));
  57
  58   if(!_storageBackends.count()) {
  59     qWarning() << qPrintable(tr("Could not initialize any storage backend! Exiting..."));
  60     qWarning() << qPrintable(tr("Currently, Quassel only supports SQLite3. You need to build your\n"
  61                                 "Qt library with the sqlite plugin enabled in order for quasselcore\n"
  62                                 "to work."));
  63     exit(1); // TODO make this less brutal (especially for mono client -> popup)
  64   }
  65   connect(&_storageSyncTimer, SIGNAL(timeout()), this, SLOT(syncStorage()));
  66   _storageSyncTimer.start(10 * 60 * 1000); // 10 minutes
  67 }
  68
  69 void Core::init() {
  70   configured = false;
  71
  72   CoreSettings cs;
  73
  74   if(!(configured = initStorage(cs.storageSettings().toMap()))) {
  75     qWarning() << "Core is currently not configured! Please connect with a Quassel Client for basic setup.";
  76
  77     // try to migrate old settings
  78     QVariantMap old = cs.oldDbSettings().toMap();
  79     if(old.count() && old["Type"].toString().toUpper() == "SQLITE") {
  80       QVariantMap newSettings;
  81       newSettings["Backend"] = "SQLite";
  82       if((configured = initStorage(newSettings))) {
  83         qWarning() << "...but thankfully I found some old settings to migrate!";
  84         cs.setStorageSettings(newSettings);
  85       }
  86     }
  87   }
  88
  89   connect(&_server, SIGNAL(newConnection()), this, SLOT(incomingConnection()));
  90   connect(&_v6server, SIGNAL(newConnection()), this, SLOT(incomingConnection()));
  91   if(!startListening()) exit(1); // TODO make this less brutal
  92 }
  93
  94 Core::~Core() {
  95   foreach(QTcpSocket *socket, blocksizes.keys()) {
  96     socket->disconnectFromHost();  // disconnect non authed clients
  97   }
  98   qDeleteAll(sessions);
  99   qDeleteAll(_storageBackends);
 100 }
 101
 102 /*** Session Restore ***/
 103
 104 void Core::saveState() {
 105   CoreSettings s;
 106   QVariantMap state;
 107   QVariantList activeSessions;
 108   foreach(UserId user, instance()->sessions.keys()) activeSessions << QVariant::fromValue<UserId>(user);
 109   state["CoreStateVersion"] = 1;
 110   state["ActiveSessions"] = activeSessions;
 111   s.setCoreState(state);
 112 }
 113
 114 void Core::restoreState() {
 115   if(!instance()->configured) {
 116     // qWarning() << qPrintable(tr("Cannot restore a state for an unconfigured core!"));
 117     return;
 118   }
 119   if(instance()->sessions.count()) {
 120     qWarning() << qPrintable(tr("Calling restoreState() even though active sessions exist!"));
 121     return;
 122   }
 123   CoreSettings s;
 124   /* We don't check, since we are at the first version since switching to Git
 125   uint statever = s.coreState().toMap()["CoreStateVersion"].toUInt();
 126   if(statever < 1) {
 127     qWarning() << qPrintable(tr("Core state too old, ignoring..."));
 128     return;
 129   }
 130   */
 131   QVariantList activeSessions = s.coreState().toMap()["ActiveSessions"].toList();
 132   if(activeSessions.count() > 0) {
 133     quInfo() << "Restoring previous core state...";
 134     foreach(QVariant v, activeSessions) {
 135       UserId user = v.value<UserId>();
 136       instance()->createSession(user, true);
 137     }
 138   }
 139 }
 140
 141 /*** Core Setup ***/
 142 QString Core::setupCoreForInternalUsage() {
 143   Q_ASSERT(!_storageBackends.isEmpty());
 144   QVariantMap setupData;
 145   qsrand(QDateTime::currentDateTime().toTime_t());
 146   int pass = 0;
 147   for(int i = 0; i < 10; i++) {
 148     pass *= 10;
 149     pass += qrand() % 10;
 150   }
 151   setupData["AdminUser"] = "AdminUser";
 152   setupData["AdminPasswd"] = QString::number(pass);
 153   setupData["Backend"] = _storageBackends[_storageBackends.keys().first()]->displayName();
 154   return setupCore(setupData);
 155 }
 156
 157 QString Core::setupCore(QVariantMap setupData) {
 158   QString user = setupData.take("AdminUser").toString();
 159   QString password = setupData.take("AdminPasswd").toString();
 160   if(user.isEmpty() || password.isEmpty()) {
 161     return tr("Admin user or password not set.");
 162   }
 163   if(!initStorage(setupData, true)) {
 164     return tr("Could not setup storage!");
 165   }
 166   CoreSettings s;
 167   s.setStorageSettings(setupData);
 168   quInfo() << qPrintable(tr("Creating admin user..."));
 169   storage->addUser(user, password);
 170   startListening();  // TODO check when we need this
 171   return QString();
 172 }
 173
 174 /*** Storage Handling ***/
 175
 176 bool Core::registerStorageBackend(Storage *backend) {
 177   if(backend->isAvailable()) {
 178     _storageBackends[backend->displayName()] = backend;
 179     return true;
 180   } else {
 181     backend->deleteLater();
 182     return false;
 183   }
 184 }
 185
 186 void Core::unregisterStorageBackend(Storage *backend) {
 187   _storageBackends.remove(backend->displayName());
 188   backend->deleteLater();
 189 }
 190
 191 // old db settings:
 192 // "Type" => "sqlite"
 193 bool Core::initStorage(QVariantMap dbSettings, bool setup) {
 194   QString backend = dbSettings["Backend"].toString();
 195   if(backend.isEmpty()) {
 196     //qWarning() << "No storage backend selected!";
 197     return configured = false;
 198   }
 199
 200   if(_storageBackends.contains(backend)) {
 201     storage = _storageBackends[backend];
 202   } else {
 203     qCritical() << "Selected storage backend is not available:" << backend;
 204     return configured = false;
 205   }
 206   if(!storage->init(dbSettings)) {
 207     if(!setup || !(storage->setup(dbSettings) && storage->init(dbSettings))) {
 208       qCritical() << "Could not init storage!";
 209       storage = 0;
 210       return configured = false;
 211     }
 212   }
 213   // delete all other backends
 214   foreach(Storage *s, _storageBackends.values()) {
 215     if(s != storage) s->deleteLater();
 216   }
 217   _storageBackends.clear();
 218
 219   connect(storage, SIGNAL(bufferInfoUpdated(UserId, const BufferInfo &)), this, SIGNAL(bufferInfoUpdated(UserId, const BufferInfo &)));
 220   return configured = true;
 221 }
 222
 223 void Core::syncStorage() {
 224   if(storage) storage->sync();
 225 }
 226
 227 /*** Storage Access ***/
 228 bool Core::createNetwork(UserId user, NetworkInfo &info) {
 229   NetworkId networkId = instance()->storage->createNetwork(user, info);
 230   if(!networkId.isValid())
 231     return false;
 232
 233   info.networkId = networkId;
 234   return true;
 235 }
 236
 237 /*** Network Management ***/
 238
 239 bool Core::startListening() {
 240   // in mono mode we only start a local port if a port is specified in the cli call
 241   if(Quassel::runMode() == Quassel::Monolithic && !Quassel::isOptionSet("port"))
 242     return true;
 243
 244   bool success = false;
 245   uint port = Quassel::optionValue("port").toUInt();
 246
 247   const QString listen = Quassel::optionValue("listen");
 248   const QStringList listen_list = listen.split(",", QString::SkipEmptyParts);
 249   if (listen_list.size() > 0) {
 250     foreach (const QString listen_term, listen_list) {
 251       QHostAddress addr;
 252       if (! addr.setAddress(listen_term)) {
 253         qCritical() << qPrintable(
 254           QString("Invalid listen address %1")
 255             .arg(listen_term)
 256         );
 257       } else {
 258         switch (addr.protocol()) {
 259           case QAbstractSocket::IPv4Protocol:
 260             if (_server.listen(addr, port)) {
 261               quInfo() << qPrintable(
 262                 QString("Listening for GUI clients on IPv4 %1 port %2 using protocol version %3")
 263                   .arg(addr.toString())
 264                   .arg(_server.serverPort())
 265                   .arg(Quassel::buildInfo().protocolVersion)
 266               );
 267               success = true;
 268             }
 269             break;
 270           case QAbstractSocket::IPv6Protocol:
 271             if (_v6server.listen(addr, port)) {
 272               quInfo() << qPrintable(
 273                 QString("Listening for GUI clients on IPv6 %1 port %2 using protocol version %3")
 274                   .arg(addr.toString())
 275                   .arg(_server.serverPort())
 276                   .arg(Quassel::buildInfo().protocolVersion)
 277               );
 278               success = true;
 279             }
 280             break;
 281           default:
 282             qCritical() << qPrintable(
 283               QString("Invalid listen address %1, unknown network protocol")
 284                 .arg(listen_term)
 285             );
 286             break;
 287         }
 288       }
 289     }
 290   }
 291
 292   return success;
 293 }
 294
 295 void Core::stopListening(const QString &reason) {
 296   bool wasListening = false;
 297   if(_server.isListening()) {
 298     wasListening = true;
 299     _server.close();
 300   }
 301   if(_v6server.isListening()) {
 302     wasListening = true;
 303     _v6server.close();
 304   }
 305   if(wasListening) {
 306     if(reason.isEmpty())
 307       quInfo() << "No longer listening for GUI clients.";
 308     else
 309       quInfo() << qPrintable(reason);
 310   }
 311 }
 312
 313 void Core::incomingConnection() {
 314   QTcpServer *server = qobject_cast<QTcpServer *>(sender());
 315   Q_ASSERT(server);
 316   while(server->hasPendingConnections()) {
 317     QTcpSocket *socket = server->nextPendingConnection();
 318     connect(socket, SIGNAL(disconnected()), this, SLOT(clientDisconnected()));
 319     connect(socket, SIGNAL(readyRead()), this, SLOT(clientHasData()));
 320     connect(socket, SIGNAL(error(QAbstractSocket::SocketError)), this, SLOT(socketError(QAbstractSocket::SocketError)));
 321
 322     QVariantMap clientInfo;
 323     blocksizes.insert(socket, (quint32)0);
 324     quInfo() << qPrintable(tr("Client connected from"))  << qPrintable(socket->peerAddress().toString());
 325
 326     if(!configured) {
 327       stopListening(tr("Closing server for basic setup."));
 328     }
 329   }
 330 }
 331
 332 void Core::clientHasData() {
 333   QTcpSocket *socket = dynamic_cast<QTcpSocket*>(sender());
 334   Q_ASSERT(socket && blocksizes.contains(socket));
 335   QVariant item;
 336   while(SignalProxy::readDataFromDevice(socket, blocksizes[socket], item)) {
 337     QVariantMap msg = item.toMap();
 338     processClientMessage(socket, msg);
 339     if(!blocksizes.contains(socket)) break;  // this socket is no longer ours to handle!
 340   }
 341 }
 342
 343 void Core::processClientMessage(QTcpSocket *socket, const QVariantMap &msg) {
 344   if(!msg.contains("MsgType")) {
 345     // Client is way too old, does not even use the current init format
 346     qWarning() << qPrintable(tr("Antique client trying to connect... refusing."));
 347     socket->close();
 348     return;
 349   }
 350   // OK, so we have at least an init message format we can understand
 351   if(msg["MsgType"] == "ClientInit") {
 352     QVariantMap reply;
 353
 354     // Just version information -- check it!
 355     uint ver = msg["ProtocolVersion"].toUInt();
 356     if(ver < Quassel::buildInfo().coreNeedsProtocol) {
 357       reply["MsgType"] = "ClientInitReject";
 358       reply["Error"] = tr("<b>Your Quassel Client is too old!</b><br>"
 359       "This core needs at least client/core protocol version %1.<br>"
 360       "Please consider upgrading your client.").arg(Quassel::buildInfo().coreNeedsProtocol);
 361       SignalProxy::writeDataToDevice(socket, reply);
 362       qWarning() << qPrintable(tr("Client")) << qPrintable(socket->peerAddress().toString()) << qPrintable(tr("too old, rejecting."));
 363       socket->close(); return;
 364     }
 365
 366     reply["CoreVersion"] = Quassel::buildInfo().fancyVersionString;
 367     reply["CoreDate"] = Quassel::buildInfo().buildDate;
 368     reply["ProtocolVersion"] = Quassel::buildInfo().protocolVersion;
 369     // TODO: Make the core info configurable
 370     int uptime = startTime().secsTo(QDateTime::currentDateTime().toUTC());
 371     int updays = uptime / 86400; uptime %= 86400;
 372     int uphours = uptime / 3600; uptime %= 3600;
 373     int upmins = uptime / 60;
 374     reply["CoreInfo"] = tr("<b>Quassel Core Version %1</b><br>"
 375                            "Built: %2<br>"
 376                            "Up %3d%4h%5m (since %6)").arg(Quassel::buildInfo().fancyVersionString)
 377                                                      .arg(Quassel::buildInfo().buildDate)
 378       .arg(updays).arg(uphours,2,10,QChar('0')).arg(upmins,2,10,QChar('0')).arg(startTime().toString(Qt::TextDate));
 379
 380 #ifdef HAVE_SSL
 381     SslServer *sslServer = qobject_cast<SslServer *>(&_server);
 382     QSslSocket *sslSocket = qobject_cast<QSslSocket *>(socket);
 383     bool supportSsl = (bool)sslServer && (bool)sslSocket && sslServer->certIsValid();
 384 #else
 385     bool supportSsl = false;
 386 #endif
 387
 388 #ifndef QT_NO_COMPRESS
 389     bool supportsCompression = true;
 390 #else
 391     bool supportsCompression = false;
 392 #endif
 393
 394     reply["SupportSsl"] = supportSsl;
 395     reply["SupportsCompression"] = supportsCompression;
 396     // switch to ssl/compression after client has been informed about our capabilities (see below)
 397
 398     reply["LoginEnabled"] = true;
 399
 400     // check if we are configured, start wizard otherwise
 401     if(!configured) {
 402       reply["Configured"] = false;
 403       QList<QVariant> backends;
 404       foreach(Storage *backend, _storageBackends.values()) {
 405         QVariantMap v;
 406         v["DisplayName"] = backend->displayName();
 407         v["Description"] = backend->description();
 408         backends.append(v);
 409       }
 410       reply["StorageBackends"] = backends;
 411       reply["LoginEnabled"] = false;
 412     } else {
 413       reply["Configured"] = true;
 414     }
 415     clientInfo[socket] = msg; // store for future reference
 416     reply["MsgType"] = "ClientInitAck";
 417     SignalProxy::writeDataToDevice(socket, reply);
 418
 419 #ifdef HAVE_SSL
 420     // after we told the client that we are ssl capable we switch to ssl mode
 421     if(supportSsl && msg["UseSsl"].toBool()) {
 422       qDebug() << qPrintable(tr("Starting TLS for Client:"))  << qPrintable(socket->peerAddress().toString());
 423       connect(sslSocket, SIGNAL(sslErrors(const QList<QSslError> &)), this, SLOT(sslErrors(const QList<QSslError> &)));
 424       sslSocket->startServerEncryption();
 425     }
 426 #endif
 427
 428 #ifndef QT_NO_COMPRESS
 429     if(supportsCompression && msg["UseCompression"].toBool()) {
 430       socket->setProperty("UseCompression", true);
 431       qDebug() << "Using compression for Client:" << qPrintable(socket->peerAddress().toString());
 432     }
 433 #endif
 434
 435   } else {
 436     // for the rest, we need an initialized connection
 437     if(!clientInfo.contains(socket)) {
 438       QVariantMap reply;
 439       reply["MsgType"] = "ClientLoginReject";
 440       reply["Error"] = tr("<b>Client not initialized!</b><br>You need to send an init message before trying to login.");
 441       SignalProxy::writeDataToDevice(socket, reply);
 442       qWarning() << qPrintable(tr("Client")) << qPrintable(socket->peerAddress().toString()) << qPrintable(tr("did not send an init message before trying to login, rejecting."));
 443       socket->close(); return;
 444     }
 445     if(msg["MsgType"] == "CoreSetupData") {
 446       QVariantMap reply;
 447       QString result = setupCore(msg["SetupData"].toMap());
 448       if(!result.isEmpty()) {
 449         reply["MsgType"] = "CoreSetupReject";
 450         reply["Error"] = result;
 451       } else {
 452         reply["MsgType"] = "CoreSetupAck";
 453       }
 454       SignalProxy::writeDataToDevice(socket, reply);
 455     } else if(msg["MsgType"] == "ClientLogin") {
 456       QVariantMap reply;
 457       UserId uid = storage->validateUser(msg["User"].toString(), msg["Password"].toString());
 458       if(uid == 0) {
 459         reply["MsgType"] = "ClientLoginReject";
 460         reply["Error"] = tr("<b>Invalid username or password!</b><br>The username/password combination you supplied could not be found in the database.");
 461         SignalProxy::writeDataToDevice(socket, reply);
 462         return;
 463       }
 464       reply["MsgType"] = "ClientLoginAck";
 465       SignalProxy::writeDataToDevice(socket, reply);
 466       quInfo() << qPrintable(tr("Client")) << qPrintable(socket->peerAddress().toString()) << qPrintable(tr("initialized and authenticated successfully as \"%1\" (UserId: %2).").arg(msg["User"].toString()).arg(uid.toInt()));
 467       setupClientSession(socket, uid);
 468     }
 469   }
 470 }
 471
 472 // Potentially called during the initialization phase (before handing the connection off to the session)
 473 void Core::clientDisconnected() {
 474   QTcpSocket *socket = qobject_cast<QTcpSocket *>(sender());
 475   if(socket) {
 476     // here it's safe to call methods on socket!
 477     quInfo() << qPrintable(tr("Non-authed client disconnected.")) << qPrintable(socket->peerAddress().toString());
 478     blocksizes.remove(socket);
 479     clientInfo.remove(socket);
 480     socket->deleteLater();
 481   } else {
 482     // we have to crawl through the hashes and see if we find a victim to remove
 483     qDebug() << qPrintable(tr("Non-authed client disconnected. (socket allready destroyed)"));
 484
 485     // DO NOT CALL ANY METHODS ON socket!!
 486     socket = static_cast<QTcpSocket *>(sender());
 487
 488     QHash<QTcpSocket *, quint32>::iterator blockSizeIter = blocksizes.begin();
 489     while(blockSizeIter != blocksizes.end()) {
 490       if(blockSizeIter.key() == socket) {
 491         blockSizeIter = blocksizes.erase(blockSizeIter);
 492       } else {
 493         blockSizeIter++;
 494       }
 495     }
 496
 497     QHash<QTcpSocket *, QVariantMap>::iterator clientInfoIter = clientInfo.begin();
 498     while(clientInfoIter != clientInfo.end()) {
 499       if(clientInfoIter.key() == socket) {
 500         clientInfoIter = clientInfo.erase(clientInfoIter);
 501       } else {
 502         clientInfoIter++;
 503       }
 504     }
 505   }
 506
 507
 508   // make server listen again if still not configured
 509   if (!configured) {
 510     startListening();
 511   }
 512
 513   // TODO remove unneeded sessions - if necessary/possible...
 514   // Suggestion: kill sessions if they are not connected to any network and client.
 515 }
 516
 517 void Core::setupClientSession(QTcpSocket *socket, UserId uid) {
 518   // Find or create session for validated user
 519   SessionThread *sess;
 520   if(sessions.contains(uid)) sess = sessions[uid];
 521   else sess = createSession(uid);
 522   // Hand over socket, session then sends state itself
 523   disconnect(socket, 0, this, 0);
 524   blocksizes.remove(socket);
 525   clientInfo.remove(socket);
 526   if(!sess) {
 527     qWarning() << qPrintable(tr("Could not initialize session for client:")) << qPrintable(socket->peerAddress().toString());
 528     socket->close();
 529   }
 530   sess->addClient(socket);
 531 }
 532
 533 void Core::setupInternalClientSession(SignalProxy *proxy) {
 534   if(!configured) {
 535     stopListening();
 536     setupCoreForInternalUsage();
 537   }
 538
 539   UserId uid = storage->internalUser();
 540
 541   // Find or create session for validated user
 542   SessionThread *sess;
 543   if(sessions.contains(uid))
 544     sess = sessions[uid];
 545   else
 546     sess = createSession(uid);
 547   sess->addClient(proxy);
 548 }
 549
 550 SessionThread *Core::createSession(UserId uid, bool restore) {
 551   if(sessions.contains(uid)) {
 552     qWarning() << "Calling createSession() when a session for the user already exists!";
 553     return 0;
 554   }
 555   SessionThread *sess = new SessionThread(uid, restore, this);
 556   sessions[uid] = sess;
 557   sess->start();
 558   return sess;
 559 }
 560
 561 #ifdef HAVE_SSL
 562 void Core::sslErrors(const QList<QSslError> &errors) {
 563   Q_UNUSED(errors);
 564   QSslSocket *socket = qobject_cast<QSslSocket *>(sender());
 565   if(socket)
 566     socket->ignoreSslErrors();
 567 }
 568 #endif
 569
 570 void Core::socketError(QAbstractSocket::SocketError err) {
 571   QAbstractSocket *socket = qobject_cast<QAbstractSocket *>(sender());
 572   if(socket && err != QAbstractSocket::RemoteHostClosedError)
 573     qWarning() << "Core::socketError()" << socket << err << socket->errorString();
 574 }