src/core/core.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-07 by The Quassel IRC Development Team             *
   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) any later version.                                   *
   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 "core.h"
  22 #include "server.h"
  23 #include "global.h"
  24 #include "util.h"
  25 #include "coreproxy.h"
  26 #include "sqlitestorage.h"
  27
  28 #include <QtSql>
  29 #include <QSettings>
  30
  31 Core *Core::instanceptr = 0;
  32
  33 Core * Core::instance() {
  34   if(instanceptr) return instanceptr;
  35   instanceptr = new Core();
  36   instanceptr->init();
  37   return instanceptr;
  38 }
  39
  40 void Core::destroy() {
  41   delete instanceptr;
  42   instanceptr = 0;
  43 }
  44
  45 Core::Core() {
  46
  47 }
  48
  49 void Core::init() {
  50   if(!SqliteStorage::isAvailable()) {
  51     qFatal("Sqlite is currently required! Please make sure your Qt library has sqlite support enabled.");
  52   }
  53   //SqliteStorage::init();
  54   storage = new SqliteStorage();
  55   connect(Global::instance(), SIGNAL(dataPutLocally(UserId, QString)), this, SLOT(updateGlobalData(UserId, QString)));
  56   connect(&server, SIGNAL(newConnection()), this, SLOT(incomingConnection()));
  57   //startListening(); // FIXME
  58   guiUser = 0;
  59   /*
  60   if(Global::runMode == Global::Monolithic) {  // TODO Make GUI user configurable
  61     try {
  62       guiUser = storage->validateUser("Default", "password");
  63     } catch(Storage::AuthError) {
  64       guiUser = storage->addUser("Default", "password");
  65     }
  66     Q_ASSERT(guiUser);
  67     Global::setGuiUser(guiUser);
  68     createSession(guiUser);
  69   } else guiUser = 0;
  70   */
  71   // Read global settings from config file
  72   QSettings s;
  73   s.beginGroup("Global");
  74   foreach(QString unum, s.childGroups()) {
  75     UserId uid = unum.toUInt();
  76     s.beginGroup(unum);
  77     foreach(QString key, s.childKeys()) {
  78       Global::updateData(uid, key, s.value(key));
  79     }
  80     s.endGroup();
  81   }
  82   s.endGroup();
  83 }
  84
  85 Core::~Core() {
  86   foreach(QTcpSocket *sock, validClients.keys()) {
  87     delete sock;
  88   }
  89   qDeleteAll(sessions);
  90   delete storage;
  91 }
  92
  93 CoreSession *Core::session(UserId uid) {
  94   Core *core = instance();
  95   if(core->sessions.contains(uid)) return core->sessions[uid];
  96   else return 0;
  97 }
  98
  99 CoreSession *Core::localSession() {
 100   Core *core = instance();
 101   if(core->guiUser && core->sessions.contains(core->guiUser)) return core->sessions[core->guiUser];
 102   else return 0;
 103 }
 104
 105 CoreSession *Core::createSession(UserId uid) {
 106   Core *core = instance();
 107   Q_ASSERT(!core->sessions.contains(uid));
 108   CoreSession *sess = new CoreSession(uid, core->storage);
 109   core->sessions[uid] = sess;
 110   connect(sess, SIGNAL(proxySignal(CoreSignal, QVariant, QVariant, QVariant)), core, SLOT(recvProxySignal(CoreSignal, QVariant, QVariant, QVariant)));
 111   return sess;
 112 }
 113
 114
 115 bool Core::startListening(uint port) {
 116   if(!server.listen(QHostAddress::Any, port)) {
 117     qWarning(QString(QString("Could not open GUI client port %1: %2").arg(port).arg(server.errorString())).toAscii());
 118     return false;
 119   }
 120   qDebug() << "Listening for GUI clients on port" << server.serverPort();
 121   return true;
 122 }
 123
 124 void Core::stopListening() {
 125   server.close();
 126   qDebug() << "No longer listening for GUI clients.";
 127 }
 128
 129 void Core::incomingConnection() {
 130   // TODO implement SSL
 131   QTcpSocket *socket = server.nextPendingConnection();
 132   connect(socket, SIGNAL(disconnected()), this, SLOT(clientDisconnected()));
 133   connect(socket, SIGNAL(readyRead()), this, SLOT(clientHasData()));
 134   blockSizes.insert(socket, (quint32)0);
 135   qDebug() << "Client connected from " << socket->peerAddress().toString();
 136 }
 137
 138 void Core::clientHasData() {
 139   QTcpSocket *socket = dynamic_cast<QTcpSocket*>(sender());
 140   Q_ASSERT(socket && blockSizes.contains(socket));
 141   quint32 bsize = blockSizes.value(socket);
 142   QVariant item;
 143   while(readDataFromDevice(socket, bsize, item)) {
 144     if(validClients.contains(socket)) {
 145       QList<QVariant> sigdata = item.toList();
 146       if((ClientSignal)sigdata[0].toInt() == GS_UPDATE_GLOBAL_DATA) {
 147         processClientUpdate(socket, sigdata[1].toString(), sigdata[2]);
 148       } else {
 149         sessions[validClients[socket]]->processSignal((ClientSignal)sigdata[0].toInt(), sigdata[1], sigdata[2], sigdata[3]);
 150       }
 151     } else {
 152       // we need to auth the client
 153       try {
 154         processClientInit(socket, item);
 155       } catch(Storage::AuthError) {
 156         qWarning() << "Authentification error!";  // FIXME
 157         socket->close();
 158         return;
 159       } catch(Exception e) {
 160         qWarning() << "Client init error:" << e.msg();
 161         socket->close();
 162         return;
 163       }
 164     }
 165     blockSizes[socket] = bsize = 0;
 166   }
 167   blockSizes[socket] = bsize;
 168 }
 169
 170 void Core::clientDisconnected() {
 171   QTcpSocket *socket = dynamic_cast<QTcpSocket*>(sender());
 172   blockSizes.remove(socket);
 173   validClients.remove(socket);
 174   qDebug() << "Client disconnected.";
 175   // TODO remove unneeded sessions - if necessary/possible...
 176 }
 177
 178 QVariant Core::connectLocalClient(QString user, QString passwd) {
 179   UserId uid = instance()->storage->validateUser(user, passwd);
 180   QVariant reply = instance()->initSession(uid);
 181   instance()->guiUser = uid;
 182   Global::setGuiUser(uid);
 183   qDebug() << "Local client connected.";
 184   return reply;
 185 }
 186
 187 QVariant Core::disconnectLocalClient() {
 188   qDebug() << "Local client disconnected.";
 189   instance()->guiUser = 0;
 190   Global::setGuiUser(0);
 191 }
 192
 193 void Core::processClientInit(QTcpSocket *socket, const QVariant &v) {
 194   VarMap msg = v.toMap();
 195   if(msg["GUIProtocol"].toUInt() != GUI_PROTOCOL) {
 196     //qWarning() << "Client version mismatch.";
 197     throw Exception("GUI client version mismatch");
 198   }
 199     // Auth
 200   UserId uid = storage->validateUser(msg["User"].toString(), msg["Password"].toString());  // throws exception if this failed
 201   VarMap reply = initSession(uid).toMap();
 202   validClients[socket] = uid;
 203   QList<QVariant> sigdata;
 204   sigdata.append(CS_CORE_STATE); sigdata.append(QVariant(reply)); sigdata.append(QVariant()); sigdata.append(QVariant());
 205   writeDataToDevice(socket, QVariant(sigdata));
 206 }
 207
 208 QVariant Core::initSession(UserId uid) {
 209   // Find or create session for validated user
 210   CoreSession *sess;
 211   if(sessions.contains(uid)) sess = sessions[uid];
 212   else {
 213     sess = createSession(uid);
 214     //validClients[socket] = uid;
 215   }
 216   VarMap reply;
 217   VarMap coreData;
 218   QStringList dataKeys = Global::keys(uid);
 219   foreach(QString key, dataKeys) {
 220     coreData[key] = Global::data(uid, key);
 221   }
 222   reply["CoreData"] = coreData;
 223   reply["SessionState"] = sess->sessionState();
 224   return reply;
 225 }
 226
 227 void Core::processClientUpdate(QTcpSocket *socket, QString key, const QVariant &data) {
 228   UserId uid = validClients[socket];
 229   Global::updateData(uid, key, data);
 230   QList<QVariant> sigdata;
 231   sigdata.append(CS_UPDATE_GLOBAL_DATA); sigdata.append(key); sigdata.append(data); sigdata.append(QVariant());
 232   foreach(QTcpSocket *s, validClients.keys()) {
 233     if(validClients[s] == uid && s != socket) writeDataToDevice(s, QVariant(sigdata));
 234   }
 235 }
 236
 237 void Core::updateGlobalData(UserId uid, QString key) {
 238   QVariant data = Global::data(uid, key);
 239   QList<QVariant> sigdata;
 240   sigdata.append(CS_UPDATE_GLOBAL_DATA); sigdata.append(key); sigdata.append(data); sigdata.append(QVariant());
 241   foreach(QTcpSocket *socket, validClients.keys()) {
 242     if(validClients[socket] == uid) writeDataToDevice(socket, QVariant(sigdata));
 243   }
 244 }
 245
 246 void Core::recvProxySignal(CoreSignal sig, QVariant arg1, QVariant arg2, QVariant arg3) {
 247   CoreSession *sess = qobject_cast<CoreSession*>(sender());
 248   Q_ASSERT(sess);
 249   UserId uid = sess->userId();
 250   QList<QVariant> sigdata;
 251   sigdata.append(sig); sigdata.append(arg1); sigdata.append(arg2); sigdata.append(arg3);
 252   //qDebug() << "Sending signal: " << sigdata;
 253   foreach(QTcpSocket *socket, validClients.keys()) {
 254     if(validClients[socket] == uid) writeDataToDevice(socket, QVariant(sigdata));
 255   }
 256 }
 257
 258 /*
 259   // Read global settings from config file
 260   QSettings s;
 261   s.beginGroup("Global");
 262   QString key;
 263   foreach(key, s.childKeys()) {
 264     global->updateData(key, s.value(key));
 265   }
 266
 267   global->updateData("CoreReady", true);
 268   // Now that we are in sync, we can connect signals to automatically store further updates.
 269   // I don't think we care if global data changed locally or if it was updated by a client.
 270   connect(global, SIGNAL(dataUpdatedRemotely(QString)), SLOT(globalDataUpdated(QString)));
 271   connect(global, SIGNAL(dataPutLocally(QString)), SLOT(globalDataUpdated(QString)));
 272
 273 }
 274   */
 275
 276 CoreSession::CoreSession(UserId uid, Storage *_storage) : user(uid), storage(_storage) {
 277   coreProxy = new CoreProxy();
 278
 279   connect(coreProxy, SIGNAL(send(CoreSignal, QVariant, QVariant, QVariant)), this, SIGNAL(proxySignal(CoreSignal, QVariant, QVariant, QVariant)));
 280   connect(coreProxy, SIGNAL(requestServerStates()), this, SIGNAL(serverStateRequested()));
 281   connect(coreProxy, SIGNAL(gsRequestConnect(QStringList)), this, SLOT(connectToIrc(QStringList)));
 282   connect(coreProxy, SIGNAL(gsUserInput(BufferId, QString)), this, SLOT(msgFromGui(BufferId, QString)));
 283   connect(coreProxy, SIGNAL(gsImportBacklog()), storage, SLOT(importOldBacklog()));
 284   connect(coreProxy, SIGNAL(gsRequestBacklog(BufferId, QVariant, QVariant)), this, SLOT(sendBacklog(BufferId, QVariant, QVariant)));
 285   connect(coreProxy, SIGNAL(gsRequestNetworkStates()), this, SLOT(sendServerStates()));
 286   connect(this, SIGNAL(displayMsg(Message)), coreProxy, SLOT(csDisplayMsg(Message)));
 287   connect(this, SIGNAL(displayStatusMsg(QString, QString)), coreProxy, SLOT(csDisplayStatusMsg(QString, QString)));
 288   connect(this, SIGNAL(backlogData(BufferId, QList<QVariant>, bool)), coreProxy, SLOT(csBacklogData(BufferId, QList<QVariant>, bool)));
 289   connect(this, SIGNAL(bufferIdUpdated(BufferId)), coreProxy, SLOT(csUpdateBufferId(BufferId)));
 290   connect(storage, SIGNAL(bufferIdUpdated(BufferId)), coreProxy, SLOT(csUpdateBufferId(BufferId)));
 291   connect(Global::instance(), SIGNAL(dataUpdatedRemotely(UserId, QString)), this, SLOT(globalDataUpdated(UserId, QString)));
 292   connect(Global::instance(), SIGNAL(dataPutLocally(UserId, QString)), this, SLOT(globalDataUpdated(UserId, QString)));
 293 }
 294
 295 CoreSession::~CoreSession() {
 296
 297 }
 298
 299 UserId CoreSession::userId() {
 300   return user;
 301 }
 302
 303 void CoreSession::processSignal(ClientSignal sig, QVariant arg1, QVariant arg2, QVariant arg3) {
 304   coreProxy->recv(sig, arg1, arg2, arg3);
 305 }
 306
 307 void CoreSession::globalDataUpdated(UserId uid, QString key) {
 308   Q_ASSERT(uid == userId());
 309   QVariant data = Global::data(userId(), key);
 310   QSettings s;
 311   s.setValue(QString("Global/%1/").arg(userId())+key, data);
 312 }
 313
 314 void CoreSession::connectToIrc(QStringList networks) {
 315   foreach(QString net, networks) {
 316     if(servers.contains(net)) {
 317
 318     } else {
 319       Server *server = new Server(userId(), net);
 320       connect(this, SIGNAL(serverStateRequested()), server, SLOT(sendState()));
 321       connect(this, SIGNAL(connectToIrc(QString)), server, SLOT(connectToIrc(QString)));
 322       connect(this, SIGNAL(disconnectFromIrc(QString)), server, SLOT(disconnectFromIrc(QString)));
 323       connect(this, SIGNAL(msgFromGui(QString, QString, QString)), server, SLOT(userInput(QString, QString, QString)));
 324
 325       connect(server, SIGNAL(connected(QString)), this, SLOT(serverConnected(QString)));
 326       connect(server, SIGNAL(disconnected(QString)), this, SLOT(serverDisconnected(QString)));
 327
 328       connect(server, SIGNAL(serverState(QString, VarMap)), coreProxy, SLOT(csServerState(QString, VarMap)));
 329       //connect(server, SIGNAL(displayMsg(Message)), this, SLOT(recvMessageFromServer(Message)));
 330       connect(server, SIGNAL(displayMsg(Message::Type, QString, QString, QString, quint8)), this, SLOT(recvMessageFromServer(Message::Type, QString, QString, QString, quint8)));
 331       connect(server, SIGNAL(displayStatusMsg(QString)), this, SLOT(recvStatusMsgFromServer(QString)));
 332       connect(server, SIGNAL(modeSet(QString, QString, QString)), coreProxy, SLOT(csModeSet(QString, QString, QString)));
 333       connect(server, SIGNAL(topicSet(QString, QString, QString)), coreProxy, SLOT(csTopicSet(QString, QString, QString)));
 334       connect(server, SIGNAL(nickAdded(QString, QString, VarMap)), coreProxy, SLOT(csNickAdded(QString, QString, VarMap)));
 335       connect(server, SIGNAL(nickRenamed(QString, QString, QString)), coreProxy, SLOT(csNickRenamed(QString, QString, QString)));
 336       connect(server, SIGNAL(nickRemoved(QString, QString)), coreProxy, SLOT(csNickRemoved(QString, QString)));
 337       connect(server, SIGNAL(nickUpdated(QString, QString, VarMap)), coreProxy, SLOT(csNickUpdated(QString, QString, VarMap)));
 338       connect(server, SIGNAL(ownNickSet(QString, QString)), coreProxy, SLOT(csOwnNickSet(QString, QString)));
 339       connect(server, SIGNAL(queryRequested(QString, QString)), coreProxy, SLOT(csQueryRequested(QString, QString)));
 340       // TODO add error handling
 341       connect(server, SIGNAL(connected(QString)), coreProxy, SLOT(csServerConnected(QString)));
 342       connect(server, SIGNAL(disconnected(QString)), coreProxy, SLOT(csServerDisconnected(QString)));
 343
 344       server->start();
 345       servers[net] = server;
 346     }
 347     emit connectToIrc(net);
 348   }
 349 }
 350
 351 void CoreSession::serverConnected(QString net) {
 352   storage->getBufferId(userId(), net); // create status buffer
 353 }
 354
 355 void CoreSession::serverDisconnected(QString net) {
 356   delete servers[net];
 357   servers.remove(net);
 358   coreProxy->csServerDisconnected(net);
 359 }
 360
 361 void CoreSession::msgFromGui(BufferId bufid, QString msg) {
 362   emit msgFromGui(bufid.network(), bufid.buffer(), msg);
 363 }
 364
 365 // ALL messages coming pass through these functions before going to the GUI.
 366 // So this is the perfect place for storing the backlog and log stuff.
 367
 368 void CoreSession::recvMessageFromServer(Message::Type type, QString target, QString text, QString sender, quint8 flags) {
 369   Server *s = qobject_cast<Server*>(this->sender());
 370   Q_ASSERT(s);
 371   BufferId buf;
 372   if((flags & Message::PrivMsg) && !(flags & Message::Self)) {
 373     buf = storage->getBufferId(user, s->getNetwork(), nickFromMask(sender));
 374   } else {
 375     buf = storage->getBufferId(user, s->getNetwork(), target);
 376   }
 377   Message msg(buf, type, text, sender, flags);
 378   msg.msgId = storage->logMessage(msg); //qDebug() << msg.msgId;
 379   Q_ASSERT(msg.msgId);
 380   emit displayMsg(msg);
 381 }
 382
 383 void CoreSession::recvStatusMsgFromServer(QString msg) {
 384   Server *s = qobject_cast<Server*>(sender());
 385   Q_ASSERT(s);
 386   emit displayStatusMsg(s->getNetwork(), msg);
 387 }
 388
 389
 390 QList<BufferId> CoreSession::buffers() const {
 391   return storage->requestBuffers(user);
 392 }
 393
 394
 395 QVariant CoreSession::sessionState() {
 396   VarMap v;
 397   QList<QVariant> bufs;
 398   foreach(BufferId id, storage->requestBuffers(user)) { bufs.append(QVariant::fromValue(id)); }
 399   v["Buffers"] = bufs;
 400
 401   return v;
 402 }
 403
 404 void CoreSession::sendServerStates() {
 405   emit serverStateRequested();
 406 }
 407
 408 void CoreSession::sendBacklog(BufferId id, QVariant v1, QVariant v2) {
 409   QList<QVariant> log;
 410   QList<Message> msglist;
 411   if(v1.type() == QVariant::DateTime) {
 412
 413
 414   } else {
 415     msglist = storage->requestMsgs(id, v1.toInt(), v2.toInt());
 416   }
 417
 418   // Send messages out in smaller packages - we don't want to make the signal data too large!
 419   for(int i = 0; i < msglist.count(); i++) {
 420     log.append(QVariant::fromValue(msglist[i]));
 421     if(log.count() >= 5) {
 422       emit backlogData(id, log, i >= msglist.count() - 1);
 423       log.clear();
 424     }
 425   }
 426   if(log.count() > 0) emit backlogData(id, log, true);
 427 }
 428
 429
 430 //Core *core = 0;