X-Git-Url: https://git.quassel-irc.org/?p=quassel.git;a=blobdiff_plain;f=src%2Fcore%2Fcoresession.cpp;h=d258df7f03cc7f63329bae6dba7d384e00203a10;hp=bb7b8c0f75e32dccd27cdc6e76927e70cfef3f09;hb=6eefdfc697067d184a589fc8a231b16316c09106;hpb=e9e9f28438f4e11995e6b444928da1c0f8487804 diff --git a/src/core/coresession.cpp b/src/core/coresession.cpp index bb7b8c0f..d258df7f 100644 --- a/src/core/coresession.cpp +++ b/src/core/coresession.cpp @@ -1,5 +1,5 @@ /*************************************************************************** - * Copyright (C) 2005-09 by the Quassel Project * + * Copyright (C) 2005-2018 by the Quassel Project * * devel@quassel-irc.org * * * * This program is free software; you can redistribute it and/or modify * @@ -15,7 +15,7 @@ * You should have received a copy of the GNU General Public License * * along with this program; if not, write to the * * Free Software Foundation, Inc., * - * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. * + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. * ***************************************************************************/ #include "coresession.h" @@ -27,527 +27,765 @@ #include "corebuffersyncer.h" #include "corebacklogmanager.h" #include "corebufferviewmanager.h" +#include "coredccconfig.h" +#include "coreeventmanager.h" #include "coreidentity.h" #include "coreignorelistmanager.h" +#include "coreinfo.h" #include "coreirclisthelper.h" #include "corenetwork.h" #include "corenetworkconfig.h" #include "coresessioneventprocessor.h" +#include "coretransfermanager.h" #include "coreusersettings.h" -#include "eventmanager.h" +#include "ctcpparser.h" +#include "eventstringifier.h" +#include "internalpeer.h" #include "ircchannel.h" +#include "ircparser.h" #include "ircuser.h" -#include "logger.h" -#include "signalproxy.h" +#include "logmessage.h" +#include "messageevent.h" +#include "remotepeer.h" #include "storage.h" #include "util.h" -class ProcessMessagesEvent : public QEvent { + +class ProcessMessagesEvent : public QEvent +{ public: - ProcessMessagesEvent() : QEvent(QEvent::User) {} + ProcessMessagesEvent() : QEvent(QEvent::User) {} }; -CoreSession::CoreSession(UserId uid, bool restoreState, QObject *parent) - : QObject(parent), + +CoreSession::CoreSession(UserId uid, bool restoreState, bool strictIdentEnabled, QObject *parent) + : QObject(parent), _user(uid), - _signalProxy(new SignalProxy(SignalProxy::Server, 0, this)), + _strictIdentEnabled(strictIdentEnabled), + _signalProxy(new SignalProxy(SignalProxy::Server, this)), _aliasManager(this), _bufferSyncer(new CoreBufferSyncer(this)), _backlogManager(new CoreBacklogManager(this)), _bufferViewManager(new CoreBufferViewManager(_signalProxy, this)), + _dccConfig(new CoreDccConfig(this)), _ircListHelper(new CoreIrcListHelper(this)), _networkConfig(new CoreNetworkConfig("GlobalNetworkConfig", this)), - _coreInfo(this), - _eventManager(new EventManager(this)), - _eventProcessor(new CoreSessionEventProcessor(this)), + _coreInfo(new CoreInfo(this)), + _transferManager(new CoreTransferManager(this)), + _eventManager(new CoreEventManager(this)), + _eventStringifier(new EventStringifier(this)), + _sessionEventProcessor(new CoreSessionEventProcessor(this)), + _ctcpParser(new CtcpParser(this)), + _ircParser(new IrcParser(this)), scriptEngine(new QScriptEngine(this)), _processMessages(false), - _ignoreListManager(this) + _ignoreListManager(this), + _highlightRuleManager(this) { - SignalProxy *p = signalProxy(); - p->setHeartBeatInterval(30); - p->setMaxHeartBeatCount(60); // 30 mins until we throw a dead socket out - - connect(p, SIGNAL(peerRemoved(QIODevice *)), this, SLOT(removeClient(QIODevice *))); + SignalProxy *p = signalProxy(); + p->setHeartBeatInterval(30); + p->setMaxHeartBeatCount(60); // 30 mins until we throw a dead socket out + + connect(p, &SignalProxy::peerRemoved, this, &CoreSession::removeClient); + + connect(p, &SignalProxy::connected, this, &CoreSession::clientsConnected); + connect(p, &SignalProxy::disconnected, this, &CoreSession::clientsDisconnected); + + p->attachSlot(SIGNAL(sendInput(BufferInfo, QString)), this, SLOT(msgFromClient(BufferInfo, QString))); + p->attachSignal(this, SIGNAL(displayMsg(Message))); + p->attachSignal(this, SIGNAL(displayStatusMsg(QString, QString))); + + p->attachSignal(this, SIGNAL(identityCreated(const Identity &))); + p->attachSignal(this, SIGNAL(identityRemoved(IdentityId))); + p->attachSlot(SIGNAL(createIdentity(const Identity &, const QVariantMap &)), this, SLOT(createIdentity(const Identity &, const QVariantMap &))); + p->attachSlot(SIGNAL(removeIdentity(IdentityId)), this, SLOT(removeIdentity(IdentityId))); + + p->attachSignal(this, SIGNAL(networkCreated(NetworkId))); + p->attachSignal(this, SIGNAL(networkRemoved(NetworkId))); + p->attachSlot(SIGNAL(createNetwork(const NetworkInfo &, const QStringList &)), this, SLOT(createNetwork(const NetworkInfo &, const QStringList &))); + p->attachSlot(SIGNAL(removeNetwork(NetworkId)), this, SLOT(removeNetwork(NetworkId))); + + p->attachSlot(SIGNAL(changePassword(PeerPtr,QString,QString,QString)), this, SLOT(changePassword(PeerPtr,QString,QString,QString))); + p->attachSignal(this, SIGNAL(passwordChanged(PeerPtr,bool))); + + p->attachSlot(SIGNAL(kickClient(int)), this, SLOT(kickClient(int))); + p->attachSignal(this, SIGNAL(disconnectFromCore())); + + QVariantMap data; + data["quasselVersion"] = Quassel::buildInfo().fancyVersionString; + data["quasselBuildDate"] = Quassel::buildInfo().commitDate; // "BuildDate" for compatibility + data["startTime"] = Core::instance()->startTime(); + data["sessionConnectedClients"] = 0; + _coreInfo->setCoreData(data); + + loadSettings(); + initScriptEngine(); + + eventManager()->registerObject(ircParser(), EventManager::NormalPriority); + eventManager()->registerObject(sessionEventProcessor(), EventManager::HighPriority); // needs to process events *before* the stringifier! + eventManager()->registerObject(ctcpParser(), EventManager::NormalPriority); + eventManager()->registerObject(eventStringifier(), EventManager::NormalPriority); + eventManager()->registerObject(this, EventManager::LowPriority); // for sending MessageEvents to the client + // some events need to be handled after msg generation + eventManager()->registerObject(sessionEventProcessor(), EventManager::LowPriority, "lateProcess"); + eventManager()->registerObject(ctcpParser(), EventManager::LowPriority, "send"); + + // periodically save our session state + connect(Core::instance()->syncTimer(), &QTimer::timeout, this, &CoreSession::saveSessionState); + + p->synchronize(_bufferSyncer); + p->synchronize(&aliasManager()); + p->synchronize(_backlogManager); + p->synchronize(dccConfig()); + p->synchronize(ircListHelper()); + p->synchronize(networkConfig()); + p->synchronize(_coreInfo); + p->synchronize(&_ignoreListManager); + p->synchronize(&_highlightRuleManager); + // Listen to network removed events + connect(this, &CoreSession::networkRemoved, + &_highlightRuleManager, &HighlightRuleManager::networkRemoved); + p->synchronize(transferManager()); + // Restore session state + if (restoreState) + restoreSessionState(); + + emit initialized(); +} - connect(p, SIGNAL(connected()), this, SLOT(clientsConnected())); - connect(p, SIGNAL(disconnected()), this, SLOT(clientsDisconnected())); - p->attachSlot(SIGNAL(sendInput(BufferInfo, QString)), this, SLOT(msgFromClient(BufferInfo, QString))); - p->attachSignal(this, SIGNAL(displayMsg(Message))); - p->attachSignal(this, SIGNAL(displayStatusMsg(QString, QString))); +void CoreSession::shutdown() +{ + saveSessionState(); + + // Request disconnect from all connected networks in parallel, and wait until every network + // has emitted the disconnected() signal before deleting the session itself + for (CoreNetwork *net : _networks.values()) { + if (net->socketState() != QAbstractSocket::UnconnectedState) { + _networksPendingDisconnect.insert(net->networkId()); + connect(net, &CoreNetwork::disconnected, this, &CoreSession::onNetworkDisconnected); + net->shutdown(); + } + } - p->attachSignal(this, SIGNAL(identityCreated(const Identity &))); - p->attachSignal(this, SIGNAL(identityRemoved(IdentityId))); - p->attachSlot(SIGNAL(createIdentity(const Identity &, const QVariantMap &)), this, SLOT(createIdentity(const Identity &, const QVariantMap &))); - p->attachSlot(SIGNAL(removeIdentity(IdentityId)), this, SLOT(removeIdentity(IdentityId))); + if (_networksPendingDisconnect.isEmpty()) { + // Nothing to do, suicide so the core can shut down + deleteLater(); + } +} - p->attachSignal(this, SIGNAL(networkCreated(NetworkId))); - p->attachSignal(this, SIGNAL(networkRemoved(NetworkId))); - p->attachSlot(SIGNAL(createNetwork(const NetworkInfo &, const QStringList &)), this, SLOT(createNetwork(const NetworkInfo &, const QStringList &))); - p->attachSlot(SIGNAL(removeNetwork(NetworkId)), this, SLOT(removeNetwork(NetworkId))); - loadSettings(); - initScriptEngine(); +void CoreSession::onNetworkDisconnected(NetworkId networkId) +{ + _networksPendingDisconnect.remove(networkId); + if (_networksPendingDisconnect.isEmpty()) { + // We're done, suicide so the core can shut down + deleteLater(); + } +} - eventManager()->registerObject(eventProcessor(), EventManager::Prepend, "process"); - // periodically save our session state - connect(&(Core::instance()->syncTimer()), SIGNAL(timeout()), this, SLOT(saveSessionState())); +CoreNetwork *CoreSession::network(NetworkId id) const +{ + if (_networks.contains(id)) return _networks[id]; + return nullptr; +} - p->synchronize(_bufferSyncer); - p->synchronize(&aliasManager()); - p->synchronize(_backlogManager); - p->synchronize(ircListHelper()); - p->synchronize(networkConfig()); - p->synchronize(&_coreInfo); - p->synchronize(&_ignoreListManager); - // Restore session state - if(restoreState) - restoreSessionState(); - emit initialized(); +CoreIdentity *CoreSession::identity(IdentityId id) const +{ + if (_identities.contains(id)) return _identities[id]; + return nullptr; } -CoreSession::~CoreSession() { - saveSessionState(); - foreach(CoreNetwork *net, _networks.values()) { - delete net; - } -} -CoreNetwork *CoreSession::network(NetworkId id) const { - if(_networks.contains(id)) return _networks[id]; - return 0; +void CoreSession::loadSettings() +{ + CoreUserSettings s(user()); + + // migrate to db + QList ids = s.identityIds(); + QList networkInfos = Core::networks(user()); + foreach(IdentityId id, ids) { + CoreIdentity identity(s.identity(id)); + IdentityId newId = Core::createIdentity(user(), identity); + QList::iterator networkIter = networkInfos.begin(); + while (networkIter != networkInfos.end()) { + if (networkIter->identity == id) { + networkIter->identity = newId; + Core::updateNetwork(user(), *networkIter); + networkIter = networkInfos.erase(networkIter); + } + else { + ++networkIter; + } + } + s.removeIdentity(id); + } + // end of migration + + foreach(CoreIdentity identity, Core::identities(user())) { + createIdentity(identity); + } + + foreach(NetworkInfo info, Core::networks(user())) { + createNetwork(info); + } } -CoreIdentity *CoreSession::identity(IdentityId id) const { - if(_identities.contains(id)) return _identities[id]; - return 0; + +void CoreSession::saveSessionState() const +{ + _bufferSyncer->storeDirtyIds(); + _bufferViewManager->saveBufferViews(); + _networkConfig->save(); } -void CoreSession::loadSettings() { - CoreUserSettings s(user()); - // migrate to db - QList ids = s.identityIds(); - QList networkInfos = Core::networks(user()); - foreach(IdentityId id, ids) { - CoreIdentity identity(s.identity(id)); - IdentityId newId = Core::createIdentity(user(), identity); - QList::iterator networkIter = networkInfos.begin(); - while(networkIter != networkInfos.end()) { - if(networkIter->identity == id) { - networkIter->identity = newId; - Core::updateNetwork(user(), *networkIter); - networkIter = networkInfos.erase(networkIter); - } else { - networkIter++; - } +void CoreSession::restoreSessionState() +{ + QList nets = Core::connectedNetworks(user()); + CoreNetwork *net = nullptr; + foreach(NetworkId id, nets) { + net = network(id); + Q_ASSERT(net); + net->connectToIrc(); } - s.removeIdentity(id); - } - // end of migration +} - foreach(CoreIdentity identity, Core::identities(user())) { - createIdentity(identity); - } - foreach(NetworkInfo info, Core::networks(user())) { - createNetwork(info); - } -} +void CoreSession::addClient(RemotePeer *peer) +{ + signalProxy()->setTargetPeer(peer); + + peer->dispatch(sessionState()); + signalProxy()->addPeer(peer); + _coreInfo->setConnectedClientData(signalProxy()->peerCount(), signalProxy()->peerData()); -void CoreSession::saveSessionState() const { - _bufferSyncer->storeDirtyIds(); - _bufferViewManager->saveBufferViews(); - _networkConfig->save(); + signalProxy()->setTargetPeer(nullptr); } -void CoreSession::restoreSessionState() { - QList nets = Core::connectedNetworks(user()); - CoreNetwork *net = 0; - foreach(NetworkId id, nets) { - net = network(id); - Q_ASSERT(net); - net->connectToIrc(); - } + +void CoreSession::addClient(InternalPeer *peer) +{ + signalProxy()->addPeer(peer); + emit sessionStateReceived(sessionState()); } -void CoreSession::addClient(QIODevice *device) { - if(!device) { - qCritical() << "Invoking CoreSession::addClient with a QObject that is not a QIODevice!"; - } else { - // if the socket is an orphan, the signalProxy adopts it. - // -> we don't need to care about it anymore - device->setParent(0); - signalProxy()->addPeer(device); - QVariantMap reply; - reply["MsgType"] = "SessionInit"; - reply["SessionState"] = sessionState(); - SignalProxy::writeDataToDevice(device, reply); - } + +void CoreSession::removeClient(Peer *peer) +{ + auto *p = qobject_cast(peer); + if (p) + quInfo() << qPrintable(tr("Client")) << p->description() << qPrintable(tr("disconnected (UserId: %1).").arg(user().toInt())); + _coreInfo->setConnectedClientData(signalProxy()->peerCount(), signalProxy()->peerData()); } -void CoreSession::addClient(SignalProxy *proxy) { - signalProxy()->addPeer(proxy); - emit sessionState(sessionState()); + +QHash CoreSession::persistentChannels(NetworkId id) const +{ + return Core::persistentChannels(user(), id); } -void CoreSession::removeClient(QIODevice *iodev) { - QTcpSocket *socket = qobject_cast(iodev); - if(socket) - quInfo() << qPrintable(tr("Client")) << qPrintable(socket->peerAddress().toString()) << qPrintable(tr("disconnected (UserId: %1).").arg(user().toInt())); + +QHash CoreSession::bufferCiphers(NetworkId id) const +{ + return Core::bufferCiphers(user(), id); } -QHash CoreSession::persistentChannels(NetworkId id) const { - return Core::persistentChannels(user(), id); +void CoreSession::setBufferCipher(NetworkId id, const QString &bufferName, const QByteArray &cipher) const +{ + Core::setBufferCipher(user(), id, bufferName, cipher); } + // FIXME switch to BufferId -void CoreSession::msgFromClient(BufferInfo bufinfo, QString msg) { - CoreNetwork *net = network(bufinfo.networkId()); - if(net) { - net->userInput(bufinfo, msg); - } else { - qWarning() << "Trying to send to unconnected network:" << msg; - } +void CoreSession::msgFromClient(BufferInfo bufinfo, QString msg) +{ + CoreNetwork *net = network(bufinfo.networkId()); + if (net) { + net->userInput(bufinfo, msg); + } + else { + qWarning() << "Trying to send to unconnected network:" << msg; + } } + // ALL messages coming pass through these functions before going to the GUI. // So this is the perfect place for storing the backlog and log stuff. void CoreSession::recvMessageFromServer(NetworkId networkId, Message::Type type, BufferInfo::Type bufferType, - const QString &target, const QString &text_, const QString &sender, Message::Flags flags) { - - // U+FDD0 and U+FDD1 are special characters for Qt's text engine, specifically they mark the boundaries of - // text frames in a QTextDocument. This might lead to problems in widgets displaying QTextDocuments (such as - // KDE's notifications), hence we remove those just to be safe. - QString text = text_; - text.remove(QChar(0xfdd0)).remove(QChar(0xfdd1)); - RawMessage rawMsg(networkId, type, bufferType, target, text, sender, flags); + const QString &target, const QString &text_, const QString &sender, Message::Flags flags) +{ + // U+FDD0 and U+FDD1 are special characters for Qt's text engine, specifically they mark the boundaries of + // text frames in a QTextDocument. This might lead to problems in widgets displaying QTextDocuments (such as + // KDE's notifications), hence we remove those just to be safe. + QString text = text_; + text.remove(QChar(0xfdd0)).remove(QChar(0xfdd1)); + RawMessage rawMsg(networkId, type, bufferType, target, text, sender, flags); + + // check for HardStrictness ignore + CoreNetwork *currentNetwork = network(networkId); + QString networkName = currentNetwork ? currentNetwork->networkName() : QString(""); + if (_ignoreListManager.match(rawMsg, networkName) == IgnoreListManager::HardStrictness) + return; + + + if (currentNetwork && _highlightRuleManager.match(rawMsg, currentNetwork->myNick(), currentNetwork->identityPtr()->nicks())) + rawMsg.flags |= Message::Flag::Highlight; + + _messageQueue << rawMsg; + if (!_processMessages) { + _processMessages = true; + QCoreApplication::postEvent(this, new ProcessMessagesEvent()); + } +} - // check for HardStrictness ignore - CoreNetwork *currentNetwork = network(networkId); - QString networkName = currentNetwork ? currentNetwork->networkName() : QString(""); - if(_ignoreListManager.match(rawMsg, networkName) == IgnoreListManager::HardStrictness) - return; - _messageQueue << rawMsg; - if(!_processMessages) { - _processMessages = true; - QCoreApplication::postEvent(this, new ProcessMessagesEvent()); - } +void CoreSession::recvStatusMsgFromServer(QString msg) +{ + auto *net = qobject_cast(sender()); + Q_ASSERT(net); + emit displayStatusMsg(net->networkName(), msg); } -void CoreSession::recvStatusMsgFromServer(QString msg) { - CoreNetwork *net = qobject_cast(sender()); - Q_ASSERT(net); - emit displayStatusMsg(net->networkName(), msg); + +void CoreSession::processMessageEvent(MessageEvent *event) +{ + recvMessageFromServer(event->networkId(), event->msgType(), event->bufferType(), + event->target().isNull() ? "" : event->target(), + event->text().isNull() ? "" : event->text(), + event->sender().isNull() ? "" : event->sender(), + event->msgFlags()); } -QList CoreSession::buffers() const { - return Core::requestBuffers(user()); + +QList CoreSession::buffers() const +{ + return Core::requestBuffers(user()); } -void CoreSession::customEvent(QEvent *event) { - if(event->type() != QEvent::User) - return; - processMessages(); - event->accept(); +void CoreSession::customEvent(QEvent *event) +{ + if (event->type() != QEvent::User) + return; + + processMessages(); + event->accept(); } -void CoreSession::processMessages() { - if(_messageQueue.count() == 1) { - const RawMessage &rawMsg = _messageQueue.first(); - bool createBuffer = !(rawMsg.flags & Message::Redirected); - BufferInfo bufferInfo = Core::bufferInfo(user(), rawMsg.networkId, rawMsg.bufferType, rawMsg.target, createBuffer); - if(!bufferInfo.isValid()) { - Q_ASSERT(!createBuffer); - bufferInfo = Core::bufferInfo(user(), rawMsg.networkId, BufferInfo::StatusBuffer, ""); - } - Message msg(bufferInfo, rawMsg.type, rawMsg.text, rawMsg.sender, rawMsg.flags); - Core::storeMessage(msg); - emit displayMsg(msg); - } else { - QHash > bufferInfoCache; - MessageList messages; - QList redirectedMessages; // list of Messages which don't enforce a buffer creation - BufferInfo bufferInfo; - for(int i = 0; i < _messageQueue.count(); i++) { - const RawMessage &rawMsg = _messageQueue.at(i); - if(bufferInfoCache.contains(rawMsg.networkId) && bufferInfoCache[rawMsg.networkId].contains(rawMsg.target)) { - bufferInfo = bufferInfoCache[rawMsg.networkId][rawMsg.target]; - } else { + +void CoreSession::processMessages() +{ + if (_messageQueue.count() == 1) { + const RawMessage &rawMsg = _messageQueue.first(); bool createBuffer = !(rawMsg.flags & Message::Redirected); - bufferInfo = Core::bufferInfo(user(), rawMsg.networkId, rawMsg.bufferType, rawMsg.target, createBuffer); - if(!bufferInfo.isValid()) { - Q_ASSERT(!createBuffer); - redirectedMessages << rawMsg; - continue; + BufferInfo bufferInfo = Core::bufferInfo(user(), rawMsg.networkId, rawMsg.bufferType, rawMsg.target, createBuffer); + if (!bufferInfo.isValid()) { + Q_ASSERT(!createBuffer); + bufferInfo = Core::bufferInfo(user(), rawMsg.networkId, BufferInfo::StatusBuffer, ""); } - bufferInfoCache[rawMsg.networkId][rawMsg.target] = bufferInfo; - } - Message msg(bufferInfo, rawMsg.type, rawMsg.text, rawMsg.sender, rawMsg.flags); - messages << msg; + Message msg(bufferInfo, rawMsg.type, rawMsg.text, rawMsg.sender, senderPrefixes(rawMsg.sender, bufferInfo), + realName(rawMsg.sender, rawMsg.networkId), avatarUrl(rawMsg.sender, rawMsg.networkId), + rawMsg.flags); + if(Core::storeMessage(msg)) + emit displayMsg(msg); } + else { + QHash > bufferInfoCache; + MessageList messages; + QList redirectedMessages; // list of Messages which don't enforce a buffer creation + BufferInfo bufferInfo; + for (int i = 0; i < _messageQueue.count(); i++) { + const RawMessage &rawMsg = _messageQueue.at(i); + if (bufferInfoCache.contains(rawMsg.networkId) && bufferInfoCache[rawMsg.networkId].contains(rawMsg.target)) { + bufferInfo = bufferInfoCache[rawMsg.networkId][rawMsg.target]; + } + else { + bool createBuffer = !(rawMsg.flags & Message::Redirected); + bufferInfo = Core::bufferInfo(user(), rawMsg.networkId, rawMsg.bufferType, rawMsg.target, createBuffer); + if (!bufferInfo.isValid()) { + Q_ASSERT(!createBuffer); + redirectedMessages << rawMsg; + continue; + } + bufferInfoCache[rawMsg.networkId][rawMsg.target] = bufferInfo; + } + Message msg(bufferInfo, rawMsg.type, rawMsg.text, rawMsg.sender, senderPrefixes(rawMsg.sender, bufferInfo), + realName(rawMsg.sender, rawMsg.networkId), avatarUrl(rawMsg.sender, rawMsg.networkId), + rawMsg.flags); + messages << msg; + } - // recheck if there exists a buffer to store a redirected message in - for(int i = 0; i < redirectedMessages.count(); i++) { - const RawMessage &rawMsg = _messageQueue.at(i); - if(bufferInfoCache.contains(rawMsg.networkId) && bufferInfoCache[rawMsg.networkId].contains(rawMsg.target)) { - bufferInfo = bufferInfoCache[rawMsg.networkId][rawMsg.target]; - } else { - // no luck -> we store them in the StatusBuffer - bufferInfo = Core::bufferInfo(user(), rawMsg.networkId, BufferInfo::StatusBuffer, ""); - // add the StatusBuffer to the Cache in case there are more Messages for the original target - bufferInfoCache[rawMsg.networkId][rawMsg.target] = bufferInfo; - } - Message msg(bufferInfo, rawMsg.type, rawMsg.text, rawMsg.sender, rawMsg.flags); - messages << msg; - } + // recheck if there exists a buffer to store a redirected message in + for (int i = 0; i < redirectedMessages.count(); i++) { + const RawMessage &rawMsg = redirectedMessages.at(i); + if (bufferInfoCache.contains(rawMsg.networkId) && bufferInfoCache[rawMsg.networkId].contains(rawMsg.target)) { + bufferInfo = bufferInfoCache[rawMsg.networkId][rawMsg.target]; + } + else { + // no luck -> we store them in the StatusBuffer + bufferInfo = Core::bufferInfo(user(), rawMsg.networkId, BufferInfo::StatusBuffer, ""); + // add the StatusBuffer to the Cache in case there are more Messages for the original target + bufferInfoCache[rawMsg.networkId][rawMsg.target] = bufferInfo; + } + Message msg(bufferInfo, rawMsg.type, rawMsg.text, rawMsg.sender, senderPrefixes(rawMsg.sender, bufferInfo), + realName(rawMsg.sender, rawMsg.networkId), avatarUrl(rawMsg.sender, rawMsg.networkId), + rawMsg.flags); + messages << msg; + } - Core::storeMessages(messages); - // FIXME: extend protocol to a displayMessages(MessageList) - for(int i = 0; i < messages.count(); i++) { - emit displayMsg(messages[i]); + if(Core::storeMessages(messages)) { + // FIXME: extend protocol to a displayMessages(MessageList) + for (int i = 0; i < messages.count(); i++) { + emit displayMsg(messages[i]); + } + } } - } - _processMessages = false; - _messageQueue.clear(); + _processMessages = false; + _messageQueue.clear(); } -QVariant CoreSession::sessionState() { - QVariantMap v; +QString CoreSession::senderPrefixes(const QString &sender, const BufferInfo &bufferInfo) const +{ + CoreNetwork *currentNetwork = network(bufferInfo.networkId()); + if (!currentNetwork) { + return {}; + } + + if (bufferInfo.type() != BufferInfo::ChannelBuffer) { + return {}; + } + + IrcChannel *currentChannel = currentNetwork->ircChannel(bufferInfo.bufferName()); + if (!currentChannel) { + return {}; + } - v["CoreFeatures"] = (int)Quassel::features(); + const QString modes = currentChannel->userModes(nickFromMask(sender).toLower()); + return currentNetwork->modesToPrefixes(modes); +} - QVariantList bufs; - foreach(BufferInfo id, buffers()) bufs << qVariantFromValue(id); - v["BufferInfos"] = bufs; - QVariantList networkids; - foreach(NetworkId id, _networks.keys()) networkids << qVariantFromValue(id); - v["NetworkIds"] = networkids; +QString CoreSession::realName(const QString &sender, NetworkId networkId) const +{ + CoreNetwork *currentNetwork = network(networkId); + if (!currentNetwork) { + return {}; + } - quint32 ircusercount = 0; - quint32 ircchannelcount = 0; - foreach(Network *net, _networks.values()) { - ircusercount += net->ircUserCount(); - ircchannelcount += net->ircChannelCount(); - } - v["IrcUserCount"] = ircusercount; - v["IrcChannelCount"] = ircchannelcount; + IrcUser *currentUser = currentNetwork->ircUser(nickFromMask(sender)); + if (!currentUser) { + return {}; + } - QList idlist; - foreach(Identity *i, _identities.values()) idlist << qVariantFromValue(*i); - v["Identities"] = idlist; + return currentUser->realName(); +} + +QString CoreSession::avatarUrl(const QString &sender, NetworkId networkId) const +{ + Q_UNUSED(sender); + Q_UNUSED(networkId); + // Currently we do not have a way to retrieve this value yet. + // + // This likely will require implementing IRCv3's METADATA spec. + // See https://ircv3.net/irc/ + // And https://blog.irccloud.com/avatars/ + return ""; +} - //v["Payload"] = QByteArray(100000000, 'a'); // for testing purposes - return v; +Protocol::SessionState CoreSession::sessionState() const +{ + QVariantList bufferInfos; + QVariantList networkIds; + QVariantList identities; + + foreach(const BufferInfo &id, buffers()) + bufferInfos << QVariant::fromValue(id); + foreach(const NetworkId &id, _networks.keys()) + networkIds << QVariant::fromValue(id); + foreach(const Identity *i, _identities.values()) + identities << QVariant::fromValue(*i); + + return Protocol::SessionState(identities, bufferInfos, networkIds); } -void CoreSession::initScriptEngine() { - signalProxy()->attachSlot(SIGNAL(scriptRequest(QString)), this, SLOT(scriptRequest(QString))); - signalProxy()->attachSignal(this, SIGNAL(scriptResult(QString))); - // FIXME - //QScriptValue storage_ = scriptEngine->newQObject(storage); - //scriptEngine->globalObject().setProperty("storage", storage_); +void CoreSession::initScriptEngine() +{ + signalProxy()->attachSlot(SIGNAL(scriptRequest(QString)), this, SLOT(scriptRequest(QString))); + signalProxy()->attachSignal(this, SIGNAL(scriptResult(QString))); + + // FIXME + //QScriptValue storage_ = scriptEngine->newQObject(storage); + //scriptEngine->globalObject().setProperty("storage", storage_); } -void CoreSession::scriptRequest(QString script) { - emit scriptResult(scriptEngine->evaluate(script).toString()); + +void CoreSession::scriptRequest(QString script) +{ + emit scriptResult(scriptEngine->evaluate(script).toString()); } + /*** Identity Handling ***/ -void CoreSession::createIdentity(const Identity &identity, const QVariantMap &additional) { +void CoreSession::createIdentity(const Identity &identity, const QVariantMap &additional) +{ #ifndef HAVE_SSL - Q_UNUSED(additional) + Q_UNUSED(additional) #endif - CoreIdentity coreIdentity(identity); + CoreIdentity coreIdentity(identity); #ifdef HAVE_SSL - if(additional.contains("KeyPem")) - coreIdentity.setSslKey(additional["KeyPem"].toByteArray()); - if(additional.contains("CertPem")) - coreIdentity.setSslCert(additional["CertPem"].toByteArray()); + if (additional.contains("KeyPem")) + coreIdentity.setSslKey(additional["KeyPem"].toByteArray()); + if (additional.contains("CertPem")) + coreIdentity.setSslCert(additional["CertPem"].toByteArray()); #endif - qDebug() << Q_FUNC_INFO; - IdentityId id = Core::createIdentity(user(), coreIdentity); - if(!id.isValid()) - return; - else - createIdentity(coreIdentity); + qDebug() << Q_FUNC_INFO; + IdentityId id = Core::createIdentity(user(), coreIdentity); + if (!id.isValid()) + return; + else + createIdentity(coreIdentity); } -void CoreSession::createIdentity(const CoreIdentity &identity) { - CoreIdentity *coreIdentity = new CoreIdentity(identity, this); - _identities[identity.id()] = coreIdentity; - // CoreIdentity has it's own synchronize method since it's "private" sslManager needs to be synced aswell - coreIdentity->synchronize(signalProxy()); - connect(coreIdentity, SIGNAL(updated()), this, SLOT(updateIdentityBySender())); - emit identityCreated(*coreIdentity); +const QString CoreSession::strictCompliantIdent(const CoreIdentity *identity) { + if (_strictIdentEnabled) { + // Strict mode enabled: only allow the user's Quassel username as an ident + return Core::instance()->strictSysIdent(_user); + } else { + // Strict mode disabled: allow any identity specified + return identity->ident(); + } } -void CoreSession::updateIdentityBySender() { - CoreIdentity *identity = qobject_cast(sender()); - if(!identity) - return; - Core::updateIdentity(user(), *identity); +void CoreSession::createIdentity(const CoreIdentity &identity) +{ + auto *coreIdentity = new CoreIdentity(identity, this); + _identities[identity.id()] = coreIdentity; + // CoreIdentity has its own synchronize method since its "private" sslManager needs to be synced as well + coreIdentity->synchronize(signalProxy()); + connect(coreIdentity, &SyncableObject::updated, this, &CoreSession::updateIdentityBySender); + emit identityCreated(*coreIdentity); } -void CoreSession::removeIdentity(IdentityId id) { - CoreIdentity *identity = _identities.take(id); - if(identity) { - emit identityRemoved(id); - Core::removeIdentity(user(), id); - identity->deleteLater(); - } + +void CoreSession::updateIdentityBySender() +{ + auto *identity = qobject_cast(sender()); + if (!identity) + return; + Core::updateIdentity(user(), *identity); } + +void CoreSession::removeIdentity(IdentityId id) +{ + CoreIdentity *identity = _identities.take(id); + if (identity) { + emit identityRemoved(id); + Core::removeIdentity(user(), id); + identity->deleteLater(); + } +} + + /*** Network Handling ***/ -void CoreSession::createNetwork(const NetworkInfo &info_, const QStringList &persistentChans) { - NetworkInfo info = info_; - int id; - - if(!info.networkId.isValid()) - Core::createNetwork(user(), info); - - if(!info.networkId.isValid()) { - qWarning() << qPrintable(tr("CoreSession::createNetwork(): Got invalid networkId from Core when trying to create network %1!").arg(info.networkName)); - return; - } - - id = info.networkId.toInt(); - if(!_networks.contains(id)) { - - // create persistent chans - QRegExp rx("\\s*(\\S+)(?:\\s*(\\S+))?\\s*"); - foreach(QString channel, persistentChans) { - if(!rx.exactMatch(channel)) { - qWarning() << QString("Invalid persistent channel declaration: %1").arg(channel); - continue; - } - Core::bufferInfo(user(), info.networkId, BufferInfo::ChannelBuffer, rx.cap(1), true); - Core::setChannelPersistent(user(), info.networkId, rx.cap(1), true); - if(!rx.cap(2).isEmpty()) - Core::setPersistentChannelKey(user(), info.networkId, rx.cap(1), rx.cap(2)); +void CoreSession::createNetwork(const NetworkInfo &info_, const QStringList &persistentChans) +{ + NetworkInfo info = info_; + int id; + + if (!info.networkId.isValid()) + Core::createNetwork(user(), info); + + if (!info.networkId.isValid()) { + qWarning() << qPrintable(tr("CoreSession::createNetwork(): Got invalid networkId from Core when trying to create network %1!").arg(info.networkName)); + return; + } + + id = info.networkId.toInt(); + if (!_networks.contains(id)) { + // create persistent chans + QRegExp rx(R"(\s*(\S+)(?:\s*(\S+))?\s*)"); + foreach(QString channel, persistentChans) { + if (!rx.exactMatch(channel)) { + qWarning() << QString("Invalid persistent channel declaration: %1").arg(channel); + continue; + } + Core::bufferInfo(user(), info.networkId, BufferInfo::ChannelBuffer, rx.cap(1), true); + Core::setChannelPersistent(user(), info.networkId, rx.cap(1), true); + if (!rx.cap(2).isEmpty()) + Core::setPersistentChannelKey(user(), info.networkId, rx.cap(1), rx.cap(2)); + } + + CoreNetwork *net = new CoreNetwork(id, this); + connect(net, &CoreNetwork::displayMsg, this, &CoreSession::recvMessageFromServer); + connect(net, &CoreNetwork::displayStatusMsg, this, &CoreSession::recvStatusMsgFromServer); + connect(net, &CoreNetwork::disconnected, this, &CoreSession::networkDisconnected); + + net->setNetworkInfo(info); + net->setProxy(signalProxy()); + _networks[id] = net; + signalProxy()->synchronize(net); + emit networkCreated(id); + } + else { + qWarning() << qPrintable(tr("CoreSession::createNetwork(): Trying to create a network that already exists, updating instead!")); + _networks[info.networkId]->requestSetNetworkInfo(info); + } +} + + +void CoreSession::removeNetwork(NetworkId id) +{ + // Make sure the network is disconnected! + CoreNetwork *net = network(id); + if (!net) + return; + + if (net->connectionState() != Network::Disconnected) { + // make sure we no longer receive data from the tcp buffer + disconnect(net, &CoreNetwork::displayMsg, this, nullptr); + disconnect(net, &CoreNetwork::displayStatusMsg, this, nullptr); + connect(net, &CoreNetwork::disconnected, this, &CoreSession::destroyNetwork); + net->disconnectFromIrc(); } + else { + destroyNetwork(id); + } +} - CoreNetwork *net = new CoreNetwork(id, this); - connect(net, SIGNAL(displayMsg(NetworkId, Message::Type, BufferInfo::Type, const QString &, const QString &, const QString &, Message::Flags)), - SLOT(recvMessageFromServer(NetworkId, Message::Type, BufferInfo::Type, const QString &, const QString &, const QString &, Message::Flags))); - connect(net, SIGNAL(displayStatusMsg(QString)), SLOT(recvStatusMsgFromServer(QString))); - - net->setNetworkInfo(info); - net->setProxy(signalProxy()); - _networks[id] = net; - signalProxy()->synchronize(net); - emit networkCreated(id); - } else { - qWarning() << qPrintable(tr("CoreSession::createNetwork(): Trying to create a network that already exists, updating instead!")); - _networks[info.networkId]->requestSetNetworkInfo(info); - } -} - -void CoreSession::removeNetwork(NetworkId id) { - // Make sure the network is disconnected! - CoreNetwork *net = network(id); - if(!net) - return; - - if(net->connectionState() != Network::Disconnected) { - connect(net, SIGNAL(disconnected(NetworkId)), this, SLOT(destroyNetwork(NetworkId))); - net->disconnectFromIrc(); - } else { - destroyNetwork(id); - } -} - -void CoreSession::destroyNetwork(NetworkId id) { - QList removedBuffers = Core::requestBufferIdsForNetwork(user(), id); - Network *net = _networks.take(id); - if(net && Core::removeNetwork(user(), id)) { - foreach(BufferId bufferId, removedBuffers) { - _bufferSyncer->removeBuffer(bufferId); + +void CoreSession::destroyNetwork(NetworkId id) +{ + QList removedBuffers = Core::requestBufferIdsForNetwork(user(), id); + Network *net = _networks.take(id); + if (net && Core::removeNetwork(user(), id)) { + // make sure that all unprocessed RawMessages from this network are removed + QList::iterator messageIter = _messageQueue.begin(); + while (messageIter != _messageQueue.end()) { + if (messageIter->networkId == id) { + messageIter = _messageQueue.erase(messageIter); + } + else { + ++messageIter; + } + } + // remove buffers from syncer + foreach(BufferId bufferId, removedBuffers) { + _bufferSyncer->removeBuffer(bufferId); + } + emit networkRemoved(id); + net->deleteLater(); } - emit networkRemoved(id); - net->deleteLater(); - } -} - -void CoreSession::renameBuffer(const NetworkId &networkId, const QString &newName, const QString &oldName) { - BufferInfo bufferInfo = Core::bufferInfo(user(), networkId, BufferInfo::QueryBuffer, oldName, false); - if(bufferInfo.isValid()) { - _bufferSyncer->renameBuffer(bufferInfo.bufferId(), newName); - } -} - -void CoreSession::clientsConnected() { - QHash::iterator netIter = _networks.begin(); - Identity *identity = 0; - CoreNetwork *net = 0; - IrcUser *me = 0; - while(netIter != _networks.end()) { - net = *netIter; - netIter++; - - if(!net->isConnected()) - continue; - identity = net->identityPtr(); - if(!identity) - continue; - me = net->me(); - if(!me) - continue; - - if(identity->detachAwayEnabled() && me->isAway()) { - net->userInputHandler()->handleAway(BufferInfo(), QString()); +} + + +void CoreSession::renameBuffer(const NetworkId &networkId, const QString &newName, const QString &oldName) +{ + BufferInfo bufferInfo = Core::bufferInfo(user(), networkId, BufferInfo::QueryBuffer, oldName, false); + if (bufferInfo.isValid()) { + _bufferSyncer->renameBuffer(bufferInfo.bufferId(), newName); } - } -} - -void CoreSession::clientsDisconnected() { - QHash::iterator netIter = _networks.begin(); - Identity *identity = 0; - CoreNetwork *net = 0; - IrcUser *me = 0; - QString awayReason; - while(netIter != _networks.end()) { - net = *netIter; - netIter++; - - if(!net->isConnected()) - continue; - - identity = net->identityPtr(); - if(!identity) - continue; - me = net->me(); - if(!me) - continue; - - if(identity->detachAwayEnabled() && !me->isAway()) { - if(!identity->detachAwayReason().isEmpty()) - awayReason = identity->detachAwayReason(); - net->setAutoAwayActive(true); - net->userInputHandler()->handleAway(BufferInfo(), awayReason); +} + + +void CoreSession::clientsConnected() +{ + QHash::iterator netIter = _networks.begin(); + Identity *identity = nullptr; + CoreNetwork *net = nullptr; + IrcUser *me = nullptr; + while (netIter != _networks.end()) { + net = *netIter; + ++netIter; + + if (!net->isConnected()) + continue; + identity = net->identityPtr(); + if (!identity) + continue; + me = net->me(); + if (!me) + continue; + + if (identity->detachAwayEnabled() && me->isAway()) { + net->userInputHandler()->handleAway(BufferInfo(), QString()); + } } - } } -void CoreSession::globalAway(const QString &msg) { - QHash::iterator netIter = _networks.begin(); - CoreNetwork *net = 0; - while(netIter != _networks.end()) { - net = *netIter; - netIter++; +void CoreSession::clientsDisconnected() +{ + QHash::iterator netIter = _networks.begin(); + Identity *identity = nullptr; + CoreNetwork *net = nullptr; + IrcUser *me = nullptr; + QString awayReason; + while (netIter != _networks.end()) { + net = *netIter; + ++netIter; + + if (!net->isConnected()) + continue; + + identity = net->identityPtr(); + if (!identity) + continue; + me = net->me(); + if (!me) + continue; + + if (identity->detachAwayEnabled() && !me->isAway()) { + if (!identity->detachAwayReason().isEmpty()) + awayReason = identity->detachAwayReason(); + net->setAutoAwayActive(true); + // Allow handleAway() to format the current date/time in the string. + net->userInputHandler()->handleAway(BufferInfo(), awayReason); + } + } +} + + +void CoreSession::globalAway(const QString &msg, const bool skipFormatting) +{ + QHash::iterator netIter = _networks.begin(); + CoreNetwork *net = nullptr; + while (netIter != _networks.end()) { + net = *netIter; + ++netIter; - if(!net->isConnected()) - continue; + if (!net->isConnected()) + continue; - net->userInputHandler()->issueAway(msg, false /* no force away */); - } + net->userInputHandler()->issueAway(msg, false /* no force away */, skipFormatting); + } +} + +void CoreSession::changePassword(PeerPtr peer, const QString &userName, const QString &oldPassword, const QString &newPassword) { + Q_UNUSED(peer); + + bool success = false; + UserId uid = Core::validateUser(userName, oldPassword); + if (uid.isValid() && uid == user()) + success = Core::changeUserPassword(uid, newPassword); + + signalProxy()->restrictTargetPeers(signalProxy()->sourcePeer(), [&]{ + emit passwordChanged(nullptr, success); + }); +} + +void CoreSession::kickClient(int peerId) { + auto peer = signalProxy()->peerById(peerId); + if (peer == nullptr) { + qWarning() << "Invalid peer Id: " << peerId; + return; + } + signalProxy()->restrictTargetPeers(peer, [&]{ + emit disconnectFromCore(); + }); }