Backlogdata is now made persistent every 10 minutes.
[quassel.git] / src / core / core.cpp
index ee93bcd..4826c3e 100644 (file)
@@ -1,5 +1,5 @@
 /***************************************************************************
- *   Copyright (C) 2005-07 by the Quassel IRC Team                         *
+ *   Copyright (C) 2005-08 by the Quassel Project                          *
  *   devel@quassel-irc.org                                                 *
  *                                                                         *
  *   This program is free software; you can redistribute it and/or modify  *
  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  ***************************************************************************/
 
+#include <QMetaObject>
+#include <QMetaMethod>
+#include <QMutexLocker>
+#include <QCoreApplication>
+
 #include "core.h"
 #include "coresession.h"
 #include "coresettings.h"
+#include "signalproxy.h"
 #include "sqlitestorage.h"
-#include "util.h"
-
-#include <QMetaObject>
-#include <QMetaMethod>
+#include "network.h"
 
 Core *Core::instanceptr = 0;
+QMutex Core::mutex;
 
 Core *Core::instance() {
   if(instanceptr) return instanceptr;
@@ -41,106 +45,145 @@ void Core::destroy() {
   instanceptr = 0;
 }
 
-Core::Core() {
-  storage = NULL;
+Core::Core()
+  : storage(0)
+{
+  startTime = QDateTime::currentDateTime();  // for uptime :)
+
+  connect(&_storageSyncTimer, SIGNAL(timeout()),
+         this, SLOT(syncStorage()));
+  _storageSyncTimer.start(10 * 60 * 1000); // in msecs
 }
 
 void Core::init() {
-  CoreSettings s;
   configured = false;
 
-  QVariantMap dbSettings = s.databaseSettings().toMap();
-  QString hname = dbSettings["Type"].toString().toLower();
-  hname[0] = hname[0].toUpper();
-  hname = "initStorage" + hname;
-  if (!QMetaObject::invokeMethod(this, hname.toAscii(), Q_RETURN_ARG(bool, configured),  Q_ARG(QVariantMap, dbSettings), Q_ARG(bool, false))) {
-    qWarning("No database backend configured.");
-  }
-  
-  if (!configured) {
+  CoreSettings cs;
+  if(!(configured = initStorage(cs.databaseSettings().toMap()))) {
     qWarning("Core is currently not configured!");
   }
-    
+
   connect(&server, SIGNAL(newConnection()), this, SLOT(incomingConnection()));
-  startListening(s.port());
+  startListening(cs.port());
   guiUser = 0;
 
 }
 
-bool Core::initStorageSqlite(QVariantMap dbSettings, bool setup) {
-  if (!SqliteStorage::isAvailable()) {
-    qFatal("Sqlite is currently required! Please make sure your Qt library has sqlite support enabled.");
-  }
-  if (storage) {
+bool Core::initStorage(QVariantMap dbSettings, bool setup) {
+  QString engine = dbSettings["Type"].toString().toLower();
+
+  if(storage) {
     qDebug() << "Deleting old storage object.";
-    delete storage;
-    storage = NULL;
+    storage->deleteLater();
+    storage = 0;
   }
-  
-  storage = new SqliteStorage();
-  if (setup && !storage->setup(dbSettings)) {
-    return false;
+
+  // FIXME register new storageProviders here
+  if(engine == "sqlite" && SqliteStorage::isAvailable()) {
+    storage = new SqliteStorage(this);
+    connect(storage, SIGNAL(bufferInfoUpdated(UserId, const BufferInfo &)), this, SIGNAL(bufferInfoUpdated(UserId, const BufferInfo &)));
+  } else {
+    qWarning() << "Selected StorageBackend is not available:" << dbSettings["Type"].toString();
+    return configured = false;
   }
-  
-  return storage->init(dbSettings);
+
+  if(setup && !storage->setup(dbSettings)) {
+    return configured = false;
+  }
+
+  return configured = storage->init(dbSettings);
 }
 
 Core::~Core() {
+  // FIXME properly shutdown the sessions
   qDeleteAll(sessions);
-  if (storage) {
-    delete storage;
-    storage = NULL;
-  }
+}
+
+void Core::syncStorage() {
+  QMutexLocker locker(&mutex);
+  return instance()->storage->sync();
 }
 
 void Core::restoreState() {
-  Q_ASSERT(!instance()->sessions.count());
+  if(instance()->sessions.count()) {
+    qWarning() << qPrintable(tr("Calling restoreState() even though active sessions exist!"));
+    return;
+  }
   CoreSettings s;
-  QList<QVariant> users = s.coreState().toList();
-  if(users.count() > 0) {
+  uint build = s.coreState().toMap()["CoreBuild"].toUInt();
+  if(build < 362) {
+    qWarning() << qPrintable(tr("Core state too old, ignoring..."));
+    return;
+  }
+  QVariantList activeSessions = s.coreState().toMap()["ActiveSessions"].toList();
+  if(activeSessions.count() > 0) {
     qDebug() << "Restoring previous core state...";
-    foreach(QVariant v, users) {
-      QVariantMap m = v.toMap();
-      if(m.contains("UserId")) {
-        CoreSession *sess = createSession(m["UserId"].toUInt());
-        sess->restoreState(m["State"]);
-      }
+    foreach(QVariant v, activeSessions) {
+      UserId user = v.value<UserId>();
+      instance()->createSession(user, true);
     }
+    qDebug() << "...done.";
   }
 }
 
 void Core::saveState() {
   CoreSettings s;
-  QList<QVariant> users;
-  foreach(CoreSession *sess, instance()->sessions.values()) {
-    QVariantMap m;
-    m["UserId"] = sess->userId();
-    m["State"] = sess->state();
-    users << m;
-  }
-  s.setCoreState(users);
+  QVariantMap state;
+  QVariantList activeSessions;
+  foreach(UserId user, instance()->sessions.keys()) activeSessions << QVariant::fromValue<UserId>(user);
+  state["CoreBuild"] = Global::quasselBuild;
+  state["ActiveSessions"] = activeSessions;
+  s.setCoreState(state);
 }
 
-CoreSession *Core::session(UserId uid) {
-  Core *core = instance();
-  if(core->sessions.contains(uid)) return core->sessions[uid];
-  else return 0;
+/*** Storage Access ***/
+bool Core::createNetworkId(UserId user, NetworkInfo &info) {
+  QMutexLocker locker(&mutex);
+  NetworkId networkId = instance()->storage->createNetworkId(user, info);
+  if(!networkId.isValid())
+    return false;
+  
+  info.networkId = networkId;
+  return true;
 }
 
-CoreSession *Core::localSession() {
-  Core *core = instance();
-  if(core->guiUser && core->sessions.contains(core->guiUser)) return core->sessions[core->guiUser];
-  else return 0;
+NetworkId Core::networkId(UserId user, const QString &network) {
+  QMutexLocker locker(&mutex);
+  return instance()->storage->getNetworkId(user, network);
 }
 
-CoreSession *Core::createSession(UserId uid) {
-  Core *core = instance();
-  Q_ASSERT(!core->sessions.contains(uid));
-  CoreSession *sess = new CoreSession(uid, core->storage);
-  core->sessions[uid] = sess;
-  return sess;
+BufferInfo Core::bufferInfo(UserId user, const NetworkId &networkId, const QString &buffer) {
+  //QMutexLocker locker(&mutex);
+  return instance()->storage->getBufferInfo(user, networkId, buffer);
+}
+
+MsgId Core::storeMessage(const Message &message) {
+  QMutexLocker locker(&mutex);
+  return instance()->storage->logMessage(message);
+}
+
+QList<Message> Core::requestMsgs(BufferInfo buffer, int lastmsgs, int offset) {
+  QMutexLocker locker(&mutex);
+  return instance()->storage->requestMsgs(buffer, lastmsgs, offset);
+}
+
+QList<Message> Core::requestMsgs(BufferInfo buffer, QDateTime since, int offset) {
+  QMutexLocker locker(&mutex);
+  return instance()->storage->requestMsgs(buffer, since, offset);
 }
 
+QList<Message> Core::requestMsgRange(BufferInfo buffer, int first, int last) {
+  QMutexLocker locker(&mutex);
+  return instance()->storage->requestMsgRange(buffer, first, last);
+}
+
+QList<BufferInfo> Core::requestBuffers(UserId user, QDateTime since) {
+  QMutexLocker locker(&mutex);
+  return instance()->storage->requestBuffers(user, since);
+}
+
+/*** Network Management ***/
+
 bool Core::startListening(uint port) {
   if(!server.listen(QHostAddress::Any, port)) {
     qWarning(QString(QString("Could not open GUI client port %1: %2").arg(port).arg(server.errorString())).toAscii());
@@ -161,9 +204,10 @@ void Core::incomingConnection() {
     QTcpSocket *socket = server.nextPendingConnection();
     connect(socket, SIGNAL(disconnected()), this, SLOT(clientDisconnected()));
     connect(socket, SIGNAL(readyRead()), this, SLOT(clientHasData()));
-    blockSizes.insert(socket, (quint32)0);
-    qDebug() << "Client connected from " << socket->peerAddress().toString();
-    
+    QVariantMap clientInfo;
+    blocksizes.insert(socket, (quint32)0);
+    qDebug() << "Client connected from"  << qPrintable(socket->peerAddress().toString());
+
     if (!configured) {
       server.close();
       qDebug() << "Closing server for basic setup.";
@@ -173,10 +217,73 @@ void Core::incomingConnection() {
 
 void Core::clientHasData() {
   QTcpSocket *socket = dynamic_cast<QTcpSocket*>(sender());
-  Q_ASSERT(socket && blockSizes.contains(socket));
-  quint32 bsize = blockSizes.value(socket);
+  Q_ASSERT(socket && blocksizes.contains(socket));
   QVariant item;
-  if(readDataFromDevice(socket, bsize, item)) {
+  while(SignalProxy::readDataFromDevice(socket, blocksizes[socket], item)) {
+    QVariantMap msg = item.toMap();
+    if(!msg.contains("MsgType")) {
+      // Client is way too old, does not even use the current init format
+      qWarning() << qPrintable(tr("Antique client trying to connect... refusing."));
+      socket->close();
+      return;
+    }
+    // OK, so we have at least an init message format we can understand
+    if(msg["MsgType"] == "ClientInit") {
+      QVariantMap reply;
+      reply["CoreVersion"] = Global::quasselVersion;
+      reply["CoreDate"] = Global::quasselDate;
+      reply["CoreBuild"] = Global::quasselBuild;
+      // TODO: Make the core info configurable
+      int uptime = startTime.secsTo(QDateTime::currentDateTime());
+      int updays = uptime / 86400; uptime %= 86400;
+      int uphours = uptime / 3600; uptime %= 3600;
+      int upmins = uptime / 60;
+      reply["CoreInfo"] = tr("<b>Quassel Core Version %1 (Build >= %2)</b><br>"
+                             "Up %3d%4h%5m (since %6)").arg(Global::quasselVersion).arg(Global::quasselBuild)
+                             .arg(updays).arg(uphours,2,10,QChar('0')).arg(upmins,2,10,QChar('0')).arg(startTime.toString(Qt::TextDate));
+
+      reply["SupportSsl"] = false;
+      reply["LoginEnabled"] = true;
+      // TODO: check if we are configured, start wizard otherwise
+
+      // Just version information -- check it!
+      if(msg["ClientBuild"].toUInt() < Global::clientBuildNeeded) {
+        reply["MsgType"] = "ClientInitReject";
+        reply["Error"] = tr("<b>Your Quassel Client is too old!</b><br>"
+                            "This core needs at least client version %1 (Build >= %2).<br>"
+                            "Please consider upgrading your client.").arg(Global::quasselVersion).arg(Global::quasselBuild);
+        SignalProxy::writeDataToDevice(socket, reply);
+        qWarning() << qPrintable(tr("Client %1 too old, rejecting.").arg(socket->peerAddress().toString()));
+        socket->close(); return;
+      }
+      clientInfo[socket] = msg; // store for future reference
+      reply["MsgType"] = "ClientInitAck";
+      SignalProxy::writeDataToDevice(socket, reply);
+    } else if(msg["MsgType"] == "ClientLogin") {
+      QVariantMap reply;
+      if(!clientInfo.contains(socket)) {
+        reply["MsgType"] = "ClientLoginReject";
+        reply["Error"] = tr("<b>Client not initialized!</b><br>You need to send an init message before trying to login.");
+        SignalProxy::writeDataToDevice(socket, reply);
+        qWarning() << qPrintable(tr("Client %1 did not send an init message before trying to login, rejecting.").arg(socket->peerAddress().toString()));
+        socket->close(); return;
+      }
+      mutex.lock();
+      UserId uid = storage->validateUser(msg["User"].toString(), msg["Password"].toString());
+      mutex.unlock();
+      if(uid == 0) {
+        reply["MsgType"] = "ClientLoginReject";
+        reply["Error"] = tr("<b>Invalid username or password!</b><br>The username/password combination you supplied could not be found in the database.");
+        SignalProxy::writeDataToDevice(socket, reply);
+        continue;
+      }
+      reply["MsgType"] = "ClientLoginAck";
+      SignalProxy::writeDataToDevice(socket, reply);
+      qDebug() << qPrintable(tr("Client %1 initialized and authentificated successfully as \"%2\".").arg(socket->peerAddress().toString(), msg["User"].toString()));
+      setupClientSession(socket, uid);
+    }
+    //socket->close(); return;
+    /*
     // we need to auth the client
     try {
       QVariantMap msg = item.toMap();
@@ -196,49 +303,26 @@ void Core::clientHasData() {
       qWarning() << "Client init error:" << e.msg();
       socket->close();
       return;
-    }
+    } */
   }
-  blockSizes[socket] = bsize = 0;  // FIXME blockSizes aufr�um0rn!
 }
 
-// FIXME: no longer called, since connection handling is now in SignalProxy
-// No, it is called as long as core is not configured. (kaffeedoktor)
+// Potentially called during the initialization phase (before handing the connection off to the session)
 void Core::clientDisconnected() {
   QTcpSocket *socket = dynamic_cast<QTcpSocket*>(sender());
-  blockSizes.remove(socket);
-  qDebug() << "Client disconnected.";
-  
-  // make server listen again if still not configured
+  blocksizes.remove(socket);
+  clientInfo.remove(socket);
+  qDebug() << qPrintable(tr("Client %1 disconnected.").arg(socket->peerAddress().toString()));
+  socket->deleteLater();
+  socket = 0;
+
+  // make server listen again if still not configured  FIXME
   if (!configured) {
     startListening();
   }
-  
-  // TODO remove unneeded sessions - if necessary/possible...
-}
-
-QVariant Core::connectLocalClient(QString user, QString passwd) {
-  UserId uid = instance()->storage->validateUser(user, passwd);
-  QVariant reply = instance()->initSession(uid);
-  instance()->guiUser = uid;
-  qDebug() << "Local client connected.";
-  return reply;
-}
-
-void Core::disconnectLocalClient() {
-  qDebug() << "Local client disconnected.";
-  instance()->guiUser = 0;
-}
 
-void Core::processClientInit(QTcpSocket *socket, const QVariantMap &msg) {
-  // Auth
-  QVariantMap reply;
-  UserId uid = storage->validateUser(msg["User"].toString(), msg["Password"].toString()); // throws exception if this failed
-  reply["StartWizard"] = false;
-  reply["Reply"] = initSession(uid);
-  disconnect(socket, 0, this, 0);
-  sessions[uid]->addClient(socket);
-  qDebug() << "Client initialized successfully.";
-  writeDataToDevice(socket, reply);
+  // TODO remove unneeded sessions - if necessary/possible...
+  // Suggestion: kill sessions if they are not connected to any network and client.
 }
 
 void Core::processCoreSetup(QTcpSocket *socket, QVariantMap &msg) {
@@ -249,19 +333,14 @@ void Core::processCoreSetup(QTcpSocket *socket, QVariantMap &msg) {
     msg.remove("User");
     msg.remove("Password");
     qDebug() << "Initializing storage provider" << msg["Type"].toString();
-    QString hname = msg["Type"].toString().toLower();
-    hname[0] = hname[0].toUpper();
-    hname = "initStorage" + hname;
-    if (!QMetaObject::invokeMethod(this, hname.toAscii(), Q_RETURN_ARG(bool, configured),  Q_ARG(QVariantMap, msg), Q_ARG(bool, true))) {
-      qWarning("No database backend configured.");
-    }
-    if (!configured) {
+
+    if(!initStorage(msg, true)) {
       // notify client to start wizard again
       qWarning("Core is currently not configured!");
       QVariantMap reply;
       reply["StartWizard"] = true;
       reply["StorageProviders"] = availableStorageProviders();
-      writeDataToDevice(socket, reply);
+      SignalProxy::writeDataToDevice(socket, reply);
     } else {
       // write coresettings
       CoreSettings s;
@@ -270,27 +349,40 @@ void Core::processCoreSetup(QTcpSocket *socket, QVariantMap &msg) {
       storage->addUser(auth["User"].toString(), auth["Password"].toString());
       startListening();
       // continue the normal procedure
-      processClientInit(socket, auth);
+      //processClientInit(socket, auth);
     }
   } else {
     // notify client to start wizard
     QVariantMap reply;
     reply["StartWizard"] = true;
     reply["StorageProviders"] = availableStorageProviders();
-    writeDataToDevice(socket, reply);
+    SignalProxy::writeDataToDevice(socket, reply);
   }
 }
 
-QVariant Core::initSession(UserId uid) {
+void Core::setupClientSession(QTcpSocket *socket, UserId uid) {
   // Find or create session for validated user
-  CoreSession *sess;
-  if(sessions.contains(uid))
-    sess = sessions[uid];
-  else
-    sess = createSession(uid);
-  QVariantMap reply;
-  reply["SessionState"] = sess->sessionState();
-  return reply;
+  SessionThread *sess;
+  if(sessions.contains(uid)) sess = sessions[uid];
+  else sess = createSession(uid);
+  // Hand over socket, session then sends state itself
+  disconnect(socket, 0, this, 0);
+  if(!sess) {
+    qWarning() << qPrintable(tr("Could not initialize session for client %1!").arg(socket->peerAddress().toString()));
+    socket->close();
+  }
+  sess->addClient(socket);
+}
+
+SessionThread *Core::createSession(UserId uid, bool restore) {
+  if(sessions.contains(uid)) {
+    qWarning() << "Calling createSession() when a session for the user already exists!";
+    return 0;
+  }
+  SessionThread *sess = new SessionThread(uid, restore, this);
+  sessions[uid] = sess;
+  sess->start();
+  return sess;
 }
 
 QStringList Core::availableStorageProviders() {
@@ -299,7 +391,7 @@ QStringList Core::availableStorageProviders() {
     storageProviders.append(SqliteStorage::displayName());
   }
   // TODO: temporary
-  storageProviders.append("MySQL");
+  // storageProviders.append("MySQL");
   
   return storageProviders;
 }