X-Git-Url: https://git.quassel-irc.org/?p=quassel.git;a=blobdiff_plain;f=src%2Fcore%2Fcore.cpp;h=4826c3ecc60ef7bbd8449d5af182562085a8dc97;hp=ee93bcd8da2395588b31974e62d5545d3067bef7;hb=59579d82e3a16a815a197a4300c8ef279275ccfc;hpb=9ac6ae4ee26819874d8843ba09f38a24088d663d diff --git a/src/core/core.cpp b/src/core/core.cpp index ee93bcd8..4826c3ec 100644 --- a/src/core/core.cpp +++ b/src/core/core.cpp @@ -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 * @@ -18,16 +18,20 @@ * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. * ***************************************************************************/ +#include +#include +#include +#include + #include "core.h" #include "coresession.h" #include "coresettings.h" +#include "signalproxy.h" #include "sqlitestorage.h" -#include "util.h" - -#include -#include +#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 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(); + instance()->createSession(user, true); } + qDebug() << "...done."; } } void Core::saveState() { CoreSettings s; - QList 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(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 Core::requestMsgs(BufferInfo buffer, int lastmsgs, int offset) { + QMutexLocker locker(&mutex); + return instance()->storage->requestMsgs(buffer, lastmsgs, offset); +} + +QList Core::requestMsgs(BufferInfo buffer, QDateTime since, int offset) { + QMutexLocker locker(&mutex); + return instance()->storage->requestMsgs(buffer, since, offset); } +QList Core::requestMsgRange(BufferInfo buffer, int first, int last) { + QMutexLocker locker(&mutex); + return instance()->storage->requestMsgRange(buffer, first, last); +} + +QList 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(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("Quassel Core Version %1 (Build >= %2)
" + "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("Your Quassel Client is too old!
" + "This core needs at least client version %1 (Build >= %2).
" + "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("Client not initialized!
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("Invalid username or password!
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(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; }