Implement a metrics daemon to expose Prometheus metrics
[quassel.git] / src / core / corenetwork.h
index 4e6544e..01d71ed 100644 (file)
@@ -1,5 +1,5 @@
 /***************************************************************************
- *   Copyright (C) 2005-2015 by the Quassel Project                        *
+ *   Copyright (C) 2005-2019 by the Quassel Project                        *
  *   devel@quassel-irc.org                                                 *
  *                                                                         *
  *   This program is free software; you can redistribute it and/or modify  *
  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
  ***************************************************************************/
 
-#ifndef CORENETWORK_H
-#define CORENETWORK_H
+#pragma once
 
-#include "network.h"
-#include "coreircchannel.h"
-#include "coreircuser.h"
+#include <functional>
 
 #include <QTimer>
 
 #ifdef HAVE_SSL
-# include <QSslSocket>
-# include <QSslError>
+#    include <QSslError>
+#    include <QSslSocket>
 #else
-# include <QTcpSocket>
+#    include <QTcpSocket>
 #endif
 
 #ifdef HAVE_QCA2
-#  include "cipher.h"
+#    include "cipher.h"
 #endif
 
+#include "coreircchannel.h"
+#include "coreircuser.h"
 #include "coresession.h"
-
-#include <functional>
+#include "irccap.h"
+#include "irctag.h"
+#include "network.h"
 
 class CoreIdentity;
 class CoreUserInputHandler;
@@ -49,45 +49,49 @@ class Event;
 
 class CoreNetwork : public Network
 {
-    SYNCABLE_OBJECT
-        Q_OBJECT
+    Q_OBJECT
 
 public:
-    CoreNetwork(const NetworkId &networkid, CoreSession *session);
-    ~CoreNetwork();
-    inline virtual const QMetaObject *syncMetaObject() const { return &Network::staticMetaObject; }
+    CoreNetwork(const NetworkId& networkid, CoreSession* session);
+    ~CoreNetwork() override;
 
-    inline CoreIdentity *identityPtr() const { return coreSession()->identity(identity()); }
-    inline CoreSession *coreSession() const { return _coreSession; }
-    inline CoreNetworkConfig *networkConfig() const { return coreSession()->networkConfig(); }
+    inline CoreIdentityidentityPtr() const { return coreSession()->identity(identity()); }
+    inline CoreSessioncoreSession() const { return _coreSession; }
+    inline CoreNetworkConfignetworkConfig() const { return coreSession()->networkConfig(); }
 
-    inline CoreUserInputHandler *userInputHandler() const { return _userInputHandler; }
-    inline CoreIgnoreListManager *ignoreListManager() { return coreSession()->ignoreListManager(); }
+    inline CoreUserInputHandleruserInputHandler() const { return _userInputHandler; }
+    inline CoreIgnoreListManagerignoreListManager() { return coreSession()->ignoreListManager(); }
 
     //! Decode a string using the server (network) decoding.
-    inline QString serverDecode(const QByteArray &string) const { return decodeServerString(string); }
+    inline QString serverDecode(const QByteArraystring) const { return decodeServerString(string); }
 
     //! Decode a string using a channel-specific encoding if one is set (and use the standard encoding else).
-    QString channelDecode(const QString &channelName, const QByteArray &string) const;
+    QString channelDecode(const QString& channelName, const QByteArray& string) const;
 
     //! Decode a string using an IrcUser-specific encoding, if one exists (using the standaed encoding else).
-    QString userDecode(const QString &userNick, const QByteArray &string) const;
+    QString userDecode(const QString& userNick, const QByteArray& string) const;
 
     //! Encode a string using the server (network) encoding.
-    inline QByteArray serverEncode(const QString &string) const { return encodeServerString(string); }
+    inline QByteArray serverEncode(const QStringstring) const { return encodeServerString(string); }
 
     //! Encode a string using the channel-specific encoding, if set, and use the standard encoding else.
-    QByteArray channelEncode(const QString &channelName, const QString &string) const;
+    QByteArray channelEncode(const QString& channelName, const QString& string) const;
 
     //! Encode a string using the user-specific encoding, if set, and use the standard encoding else.
-    QByteArray userEncode(const QString &userNick, const QString &string) const;
+    QByteArray userEncode(const QString& userNick, const QString& string) const;
 
-    inline QString channelKey(const QString &channel) const { return _channelKeys.value(channel.toLower(), QString()); }
+    inline QString channelKey(const QStringchannel) const { return _channelKeys.value(channel.toLower(), QString()); }
 
-    inline QByteArray readChannelCipherKey(const QString &channel) const { return _cipherKeys.value(channel.toLower()); }
-    inline void storeChannelCipherKey(const QString &channel, const QByteArray &key) { _cipherKeys[channel.toLower()] = key; }
+    inline QByteArray readChannelCipherKey(const QStringchannel) const { return _cipherKeys.value(channel.toLower()); }
+    inline void storeChannelCipherKey(const QString& channel, const QByteArray& key) { _cipherKeys[channel.toLower()] = key; }
 
-    inline bool isAutoWhoInProgress(const QString &channel) const { return _autoWhoPending.value(channel.toLower(), 0); }
+    /**
+     * Checks if the given target has an automatic WHO in progress
+     *
+     * @param name Channel or nickname
+     * @return True if an automatic WHO is in progress, otherwise false
+     */
+    inline bool isAutoWhoInProgress(const QString& name) const { return _autoWhoPending.value(name.toLower(), 0); }
 
     inline UserId userId() const { return _coreSession->user(); }
 
@@ -98,167 +102,286 @@ public:
     inline quint16 localPort() const { return socket.localPort(); }
     inline quint16 peerPort() const { return socket.peerPort(); }
 
-    QList<QList<QByteArray>> splitMessage(const QString &cmd, const QString &message, std::function<QList<QByteArray>(QString &)> cmdGenerator);
+    /**
+     * Gets whether or not a disconnect was expected.
+     *
+     * Distinguishes desired quits from unexpected disconnections such as socket errors or timeouts.
+     *
+     * @return True if disconnect was requested, otherwise false.
+     */
+    inline bool disconnectExpected() const { return _disconnectExpected; }
 
-    // IRCv3 capability negotiation
+    /**
+     * Gets whether or not the server replies to automated PINGs with a valid timestamp
+     *
+     * Distinguishes between servers that reply by quoting the text sent, and those that respond
+     * with whatever they want.
+     *
+     * @return True if a valid timestamp has been received as a PONG, otherwise false.
+     */
+    inline bool isPongTimestampValid() const { return _pongTimestampValid; }
 
     /**
-     * Checks if a given capability is enabled.
+     * Gets whether or not an automated PING has been sent without any PONG received
      *
-     * @param[in] capability Name of capability
-     * @returns True if enabled, otherwise false
+     * Reset whenever any PONG is received, not just the automated one sent.
+     *
+     * @return True if a PING has been sent without a PONG received, otherwise false.
      */
-    inline bool capEnabled(const QString &capability) const { return _capsSupported.contains(capability); }
+    inline bool isPongReplyPending() const { return _pongReplyPending; }
+
+    QList<QList<QByteArray>> splitMessage(const QString& cmd,
+                                          const QString& message,
+                                          const std::function<QList<QByteArray>(QString&)>& cmdGenerator);
+
+    // IRCv3 capability negotiation
 
     /**
      * Checks if capability negotiation is currently ongoing.
      *
      * @returns True if in progress, otherwise false
      */
-    inline bool capNegotiationInProgress() const { return !_capsQueued.empty(); }
+    inline bool capNegotiationInProgress() const { return (!_capsQueuedIndividual.empty() || !_capsQueuedBundled.empty()); }
 
     /**
-     * Gets the value of an enabled or pending capability, e.g. sasl=plain.
+     * Queues a capability to be requested.
+     *
+     * Adds to the list of capabilities being requested.  If non-empty, CAP REQ messages are sent
+     * to the IRC server.  This may happen at login or if capabilities are announced via CAP NEW.
      *
-     * @param[in] capability Name of capability
-     * @returns Value of capability if one was specified, otherwise empty string
+     * @param[in] capability Name of the capability
      */
-    QString capValue(const QString &capability) const;
+    void queueCap(const QString& capability);
 
     /**
-     * Gets the next capability to request, removing it from the queue.
+     * Begins capability negotiation if capabilities are queued, otherwise returns.
      *
-     * @returns Name of capability to request
+     * If any capabilities are queued, this will begin the cycle of taking each capability and
+     * requesting it.  When no capabilities remain, capability negotiation is suitably ended.
      */
-    QString takeQueuedCap();
+    void beginCapNegotiation();
 
-    // Specific capabilities for easy reference
+    /**
+     * Ends capability negotiation.
+     *
+     * This won't have effect if other CAP commands are in the command queue before calling this
+     * command.  It should only be called when capability negotiation is complete.
+     */
+    void endCapNegotiation();
 
     /**
-     * Gets the status of the sasl authentication capability.
+     * Queues the most recent capability set for retrying individually.
+     *
+     * Retries the most recent bundle of capabilities one at a time instead of as a group, working
+     * around the issue that IRC servers can deny a group of requested capabilities without
+     * indicating which capabilities failed.
      *
-     * http://ircv3.net/specs/extensions/sasl-3.2.html
+     * See: http://ircv3.net/specs/core/capability-negotiation-3.1.html
      *
-     * @returns True if SASL authentication is enabled, otherwise false
+     * This does NOT call CoreNetwork::sendNextCap().  Call that when ready afterwards.  Does
+     * nothing if the last capability tried was individual instead of a set.
      */
-    inline bool useCapSASL() const { return capEnabled("sasl"); }
+    void retryCapsIndividually();
 
     /**
-     * Gets the status of the away-notify capability.
+     * List of capabilities requiring further core<->server messages to configure.
      *
-     * http://ircv3.net/specs/extensions/away-notify-3.1.html
+     * For example, SASL requires the back-and-forth of AUTHENTICATE, so the next capability cannot
+     * be immediately sent.
      *
-     * @returns True if away-notify is enabled, otherwise false
+     * Any capabilities in this list must call CoreNetwork::sendNextCap() on their own and they will
+     * not be batched together with other capabilities.
+     *
+     * See: http://ircv3.net/specs/extensions/sasl-3.2.html
      */
-    inline bool useCapAwayNotify() const { return capEnabled("away-notify"); }
+    const QStringList capsRequiringConfiguration = QStringList{IrcCap::SASL};
+
+public slots:
+    void setMyNick(const QString& mynick) override;
+
+    void requestConnect() const override;
+    void requestDisconnect() const override;
+    void requestSetNetworkInfo(const NetworkInfo& info) override;
+
+    void setUseAutoReconnect(bool) override;
+    void setAutoReconnectInterval(quint32) override;
+    void setAutoReconnectRetries(quint16) override;
+
+    void setPingInterval(int interval);
 
     /**
-     * Gets the status of the account-notify capability.
+     * Sets whether or not the IRC server has replied to PING with a valid timestamp
      *
-     * http://ircv3.net/specs/extensions/account-notify-3.1.html
+     * This allows determining whether or not an IRC server responds to PING with a PONG that quotes
+     * what was sent, or if it does something else (and therefore PONGs should be more aggressively
+     * hidden).
      *
-     * @returns True if account-notify is enabled, otherwise false
+     * @param timestampValid If true, a valid timestamp has been received via PONG reply
      */
-    inline bool useCapAccountNotify() const { return capEnabled("account-notify"); }
+    void setPongTimestampValid(bool validTimestamp);
 
     /**
-     * Gets the status of the extended-join capability.
+     * Indicates that the CoreSession is shutting down.
      *
-     * http://ircv3.net/specs/extensions/extended-join-3.1.html
-     *
-     * @returns True if extended-join is enabled, otherwise false
+     * Disconnects the network if connected, and sets a flag that prevents reconnections.
      */
-    inline bool useCapExtendedJoin() const { return capEnabled("extended-join"); }
+    void shutdown();
 
+    void connectToIrc(bool reconnecting = false);
     /**
-     * Gets the status of the userhost-in-names capability.
+     * Disconnect from the IRC server.
      *
-     * http://ircv3.net/specs/extensions/userhost-in-names-3.2.html
+     * Begin disconnecting from the IRC server, including optionally reconnecting.
      *
-     * @returns True if userhost-in-names is enabled, otherwise false
+     * @param requested       If true, user requested this disconnect; don't try to reconnect
+     * @param reason          Reason for quitting, defaulting to the user-configured quit reason
+     * @param withReconnect   Reconnect to the network after disconnecting (e.g. ping timeout)
      */
-    inline bool useCapUserhostInNames() const { return capEnabled("userhost-in-names"); }
+    void disconnectFromIrc(bool requested = true, const QString& reason = QString(), bool withReconnect = false);
 
     /**
-     * Gets the status of the multi-prefix capability.
+     * Forcibly close the IRC server socket, waiting for it to close.
      *
-     * http://ircv3.net/specs/extensions/multi-prefix-3.1.html
+     * Call CoreNetwork::disconnectFromIrc() first, allow the event loop to run, then if you need to
+     * be sure the network's disconnected (e.g. clean-up), call this.
      *
-     * @returns True if multi-prefix is enabled, otherwise false
+     * @param msecs  Maximum time to wait for socket to close, in milliseconds.
+     * @return True if socket closes successfully; false if error occurs or timeout reached
      */
-    inline bool useCapMultiPrefix() const { return capEnabled("multi-prefix"); }
+    bool forceDisconnect(int msecs = 1000);
 
-public slots:
-    virtual void setMyNick(const QString &mynick);
+    void userInput(const BufferInfo& bufferInfo, QString msg);
 
-    virtual void requestConnect() const;
-    virtual void requestDisconnect() const;
-    virtual void requestSetNetworkInfo(const NetworkInfo &info);
-
-    virtual void setUseAutoReconnect(bool);
-    virtual void setAutoReconnectInterval(quint32);
-    virtual void setAutoReconnectRetries(quint16);
-
-    void setPingInterval(int interval);
+    /**
+     * Sends the raw (encoded) line, adding to the queue if needed, optionally with higher priority.
+     *
+     * @param[in] input   QByteArray of encoded characters
+     * @param[in] prepend
+     * @parmblock
+     * If true, the line is prepended into the start of the queue, otherwise, it's appended to the
+     * end.  This should be used sparingly, for if either the core or the IRC server cannot maintain
+     * PING/PONG replies, the other side will close the connection.
+     * @endparmblock
+     */
+    void putRawLine(const QByteArray& input, bool prepend = false);
 
-    void connectToIrc(bool reconnecting = false);
-    void disconnectFromIrc(bool requested = true, const QString &reason = QString(), bool withReconnect = false);
+    /**
+     * Sends the command with encoded parameters, with optional prefix or high priority.
+     *
+     * @param[in] cmd      Command to send, ignoring capitalization
+     * @param[in] params   Parameters for the command, encoded within a QByteArray
+     * @param[in] prefix   Optional command prefix
+     * @param[in] prepend
+     * @parmblock
+     * If true, the command is prepended into the start of the queue, otherwise, it's appended to
+     * the end.  This should be used sparingly, for if either the core or the IRC server cannot
+     * maintain PING/PONG replies, the other side will close the connection.
+     * @endparmblock
+     */
+    void putCmd(const QString& cmd, const QList<QByteArray>& params, const QByteArray& prefix = {}, const QHash<IrcTagKey, QString> &tags = {}, bool prepend = false);
 
-    void userInput(BufferInfo bufferInfo, QString msg);
-    void putRawLine(QByteArray input);
-    void putCmd(const QString &cmd, const QList<QByteArray> &params, const QByteArray &prefix = QByteArray());
-    void putCmd(const QString &cmd, const QList<QList<QByteArray>> &params, const QByteArray &prefix = QByteArray());
+    /**
+     * Sends the command for each set of encoded parameters, with optional prefix or high priority.
+     *
+     * @param[in] cmd         Command to send, ignoring capitalization
+     * @param[in] params
+     * @parmblock
+     * List of parameter lists for the command, encoded within a QByteArray.  The command will be
+     * sent multiple times, once for each set of params stored within the outer list.
+     * @endparmblock
+     * @param[in] prefix      Optional command prefix
+     * @param[in] prependAll
+     * @parmblock
+     * If true, ALL of the commands are prepended into the start of the queue, otherwise, they're
+     * appended to the end.  This should be used sparingly, for if either the core or the IRC server
+     * cannot maintain PING/PONG replies, the other side will close the connection.
+     * @endparmblock
+     */
+    void putCmd(const QString& cmd, const QList<QList<QByteArray>>& params, const QByteArray& prefix = {}, const QHash<IrcTagKey, QString> &tags = {}, bool prependAll = false);
 
-    void setChannelJoined(const QString &channel);
-    void setChannelParted(const QString &channel);
-    void addChannelKey(const QString &channel, const QString &key);
-    void removeChannelKey(const QString &channel);
+    void setChannelJoined(const QStringchannel);
+    void setChannelParted(const QStringchannel);
+    void addChannelKey(const QString& channel, const QString& key);
+    void removeChannelKey(const QStringchannel);
 
     // Blowfish stuff
 #ifdef HAVE_QCA2
-    Cipher *cipher(const QString &recipient);
-    QByteArray cipherKey(const QString &recipient) const;
-    void setCipherKey(const QString &recipient, const QByteArray &key);
-    bool cipherUsesCBC(const QString &target);
+    Cipher* cipher(const QString& recipient);
+    QByteArray cipherKey(const QStringrecipient) const;
+    void setCipherKey(const QString& recipient, const QByteArray& key);
+    bool cipherUsesCBC(const QStringtarget);
 #endif
 
+    // Custom rate limiting (can be connected to signals)
+
+    /**
+     * Update rate limiting according to Network configuration
+     *
+     * Updates the token bucket and message queue timer according to the network configuration, such
+     * as on first load, or after changing settings.
+     *
+     * Calling this will reset any ongoing queue delays.  If messages exist in the queue when rate
+     * limiting is disabled, messages will be quickly sent (100 ms) with new messages queued to send
+     * until the queue is cleared.
+     *
+     * @see Network::useCustomMessageRate()
+     * @see Network::messageRateBurstSize()
+     * @see Network::messageRateDelay()
+     * @see Network::unlimitedMessageRate()
+     *
+     * @param[in] forceUnlimited
+     * @parmblock
+     * If true, override user settings to disable message rate limiting, otherwise apply rate limits
+     * set by the user.  Use with caution and remember to re-enable configured limits when done.
+     * @endparmblock
+     */
+    void updateRateLimiting(bool forceUnlimited = false);
+
+    /**
+     * Resets the token bucket up to the maximum
+     *
+     * Call this if the connection's been reset after calling updateRateLimiting() if needed.
+     *
+     * @see CoreNetwork::updateRateLimiting()
+     */
+    void resetTokenBucket();
+
     // IRCv3 capability negotiation (can be connected to signals)
 
     /**
-     * Marks a capability as accepted, providing an optional value.
+     * Indicates a capability is now available, with optional value in Network::capValue().
      *
-     * Removes it from queue of pending capabilities and triggers any capability-specific
-     * activation.
+     * @see Network::addCap()
      *
      * @param[in] capability Name of the capability
-     * @param[in] value
-     * @parblock
-     * Optional value of the capability, e.g. sasl=plain.  If left empty, will be copied from the
-     * pending capability.
-     * @endparblock
      */
-    void addCap(const QString &capability, const QString &value = QString());
+    void serverCapAdded(const QString& capability);
 
     /**
-     * Marks a capability as denied.
+     * Indicates a capability was acknowledged (enabled by the IRC server).
      *
-     * Removes it from the queue of pending capabilities and triggers any capability-specific
-     * deactivation.
+     * @see Network::acknowledgeCap()
      *
      * @param[in] capability Name of the capability
      */
-    void removeCap(const QString &capability);
+    void serverCapAcknowledged(const QString& capability);
 
     /**
-     * Queues a capability as available but not yet accepted or denied.
+     * Indicates a capability was removed from the list of available capabilities.
      *
-     * Capabilities should be queued when registration pauses for CAP LS for capabilities are only
-     * requested during login.
+     * @see Network::removeCap()
      *
      * @param[in] capability Name of the capability
-     * @param[in] value      Optional value of the capability, e.g. sasl=plain
      */
-    void queuePendingCap(const QString &capability, const QString &value = QString());
+    void serverCapRemoved(const QString& capability);
+
+    /**
+     * Sends the next capability from the queue.
+     *
+     * During nick registration if any capabilities remain queued, this will take the next and
+     * request it.  When no capabilities remain, capability negotiation is ended.
+     */
+    void sendNextCap();
 
     void setAutoWhoEnabled(bool enabled);
     void setAutoWhoInterval(int interval);
@@ -270,13 +393,19 @@ public slots:
      * When 'away-notify' is enabled, this will trigger an immediate AutoWho since regular
      * who-cycles are disabled as per IRCv3 specifications.
      *
-     * @param[in] channelOrNick Channel or nickname to WHO
+     * @param[in] name Channel or nickname
      */
-    void queueAutoWhoOneshot(const QString &channelOrNick);
+    void queueAutoWhoOneshot(const QString& name);
 
-    bool setAutoWhoDone(const QString &channel);
+    /**
+     * Checks if the given target has an automatic WHO in progress, and sets it as done if so
+     *
+     * @param name Channel or nickname
+     * @return True if an automatic WHO is in progress (and should be silenced), otherwise false
+     */
+    bool setAutoWhoDone(const QString& name);
 
-    void updateIssuedModes(const QString &requestedModes);
+    void updateIssuedModes(const QStringrequestedModes);
     void updatePersistentModes(QString addModes, QString removeModes);
     void resetPersistentModes();
 
@@ -284,43 +413,58 @@ public slots:
 
     inline void resetPingTimeout() { _pingCount = 0; }
 
-    inline void displayMsg(Message::Type msgType, BufferInfo::Type bufferType, const QString &target, const QString &text, const QString &sender = "", Message::Flags flags = Message::None)
+    /**
+     * Marks the network as no longer having a pending reply to an automated PING
+     */
+    inline void resetPongReplyPending() { _pongReplyPending = false; }
+
+    void onDisplayMsg(const NetworkInternalMessage& msg)
     {
-        emit displayMsg(networkId(), msgType, bufferType, target, text, sender, flags);
+        emit displayMsg(RawMessage(networkId(), msg));
     }
 
-
 signals:
-    void recvRawServerMsg(QString);
-    void displayStatusMsg(QString);
-    void displayMsg(NetworkId, Message::Type, BufferInfo::Type, const QString &target, const QString &text, const QString &sender = "", Message::Flags flags = Message::None);
+    void recvRawServerMsg(const QString&);
+    void displayStatusMsg(const QString&);
+    void displayMsg(const RawMessage& msg);
     void disconnected(NetworkId networkId);
-    void connectionError(const QString &errorMsg);
+    void connectionError(const QStringerrorMsg);
 
     void quitRequested(NetworkId networkId);
-    void sslErrors(const QVariant &errorData);
-
-    void newEvent(Event *event);
-    void socketInitialized(const CoreIdentity *identity, const QHostAddress &localAddress, quint16 localPort, const QHostAddress &peerAddress, quint16 peerPort);
-    void socketDisconnected(const CoreIdentity *identity, const QHostAddress &localAddress, quint16 localPort, const QHostAddress &peerAddress, quint16 peerPort);
+    void sslErrors(const QVariant& errorData);
+
+    void newEvent(Event* event);
+    void socketInitialized(const CoreIdentity* identity,
+                           const QHostAddress& localAddress,
+                           quint16 localPort,
+                           const QHostAddress& peerAddress,
+                           quint16 peerPort,
+                           qint64 socketId);
+    void socketDisconnected(const CoreIdentity* identity,
+                            const QHostAddress& localAddress,
+                            quint16 localPort,
+                            const QHostAddress& peerAddress,
+                            quint16 peerPort,
+                            qint64 socketId);
 
 protected:
-    inline virtual IrcChannel *ircChannelFactory(const QString &channelname) { return new CoreIrcChannel(channelname, this); }
-    inline virtual IrcUser *ircUserFactory(const QString &hostmask) { return new CoreIrcUser(hostmask, this); }
+    inline IrcChannel* ircChannelFactory(const QString& channelname) override { return new CoreIrcChannel(channelname, this); }
+    inline IrcUser* ircUserFactory(const QString& hostmask) override { return new CoreIrcUser(hostmask, this); }
 
 protected slots:
     // TODO: remove cached cipher keys, when appropriate
-    //virtual void removeIrcUser(IrcUser *ircuser);
-    //virtual void removeIrcChannel(IrcChannel *ircChannel);
-    //virtual void removeChansAndUsers();
+    // virtual void removeIrcUser(IrcUser *ircuser);
+    // virtual void removeIrcChannel(IrcChannel *ircChannel);
+    // virtual void removeChansAndUsers();
 
 private slots:
-    void socketHasData();
-    void socketError(QAbstractSocket::SocketError);
-    void socketInitialized();
-    inline void socketCloseTimeout() { socket.abort(); }
-    void socketDisconnected();
-    void socketStateChanged(QAbstractSocket::SocketState);
+    void onSocketHasData();
+    void onSocketError(QAbstractSocket::SocketError);
+    void onSocketInitialized();
+    void onSocketCloseTimeout();
+    void onSocketDisconnected();
+    void onSocketStateChanged(QAbstractSocket::SocketState);
+
     void networkInitialized();
 
     void sendPerform();
@@ -333,25 +477,53 @@ private slots:
     void startAutoWhoCycle();
 
 #ifdef HAVE_SSL
-    void sslErrors(const QList<QSslError> &errors);
+    void onSslErrors(const QList<QSslError>& errors);
 #endif
 
+    /**
+     * Check the message token bucket
+     *
+     * If rate limiting is disabled and the message queue is empty, this disables the token bucket
+     * timer.  Otherwise, a queued message will be sent.
+     *
+     * @see CoreNetwork::fillBucketAndProcessQueue()
+     */
+    void checkTokenBucket();
+
+    /**
+     * Top up token bucket and send as many queued messages as possible
+     *
+     * If there's any room for more tokens, add to the token bucket.  Separately, if there's any
+     * messages to send, send until there's no more tokens or the queue is empty, whichever comes
+     * first.
+     */
     void fillBucketAndProcessQueue();
 
-    void writeToSocket(const QByteArray &data);
+    void writeToSocket(const QByteArraydata);
 
 private:
-    CoreSession *_coreSession;
+    void showMessage(const NetworkInternalMessage& msg)
+    {
+        emit displayMsg(RawMessage(networkId(), msg));
+    }
+
+private:
+    CoreSession* _coreSession;
+
+    bool _debugLogRawIrc;      ///< If true, include raw IRC socket messages in the debug log
+    qint32 _debugLogRawNetId;  ///< Network ID for logging raw IRC socket messages, or -1 for all
 
 #ifdef HAVE_SSL
     QSslSocket socket;
 #else
     QTcpSocket socket;
 #endif
+    qint64 _socketId{0};
 
-    CoreUserInputHandler *_userInputHandler;
+    CoreUserInputHandler* _userInputHandler;
+    MetricsServer* _metricsServer;
 
-    QHash<QString, QString> _channelKeys; // stores persistent channels and their passwords, if any
+    QHash<QString, QString> _channelKeys;  // stores persistent channels and their passwords, if any
 
     QTimer _autoReconnectTimer;
     int _autoReconnectCount;
@@ -364,35 +536,75 @@ private:
     bool _quitRequested;
     QString _quitReason;
 
+    bool _disconnectExpected;  /// If true, connection is quitting, expect a socket close
+    // This avoids logging a spurious RemoteHostClosedError whenever disconnect is called without
+    // specifying a permanent (saved to core session) disconnect.
+
+    bool _shuttingDown{false};  ///< If true, we're shutting down and ignore requests to (dis)connect networks
+
     bool _previousConnectionAttemptFailed;
     int _lastUsedServerIndex;
 
     QTimer _pingTimer;
-    uint _lastPingTime;
-    uint _pingCount;
-    bool _sendPings;
+    qint64 _lastPingTime = 0;          ///< Unix time of most recently sent automatic ping
+    uint _pingCount = 0;               ///< Unacknowledged automatic pings
+    bool _sendPings = false;           ///< If true, pings should be periodically sent to server
+    bool _pongTimestampValid = false;  ///< If true, IRC server responds to PING by quoting in PONG
+    // This tracks whether or not a server responds to PING with a PONG of what was sent, or if it
+    // does something else.  If false, PING reply hiding should be more aggressive.
+    bool _pongReplyPending = false;  ///< If true, at least one PING sent without a PONG reply
 
     QStringList _autoWhoQueue;
     QHash<QString, int> _autoWhoPending;
     QTimer _autoWhoTimer, _autoWhoCycleTimer;
 
-    // CAPs may have parameter values
+    // Maintain a list of CAPs that are being checked; if empty, negotiation finished
     // See http://ircv3.net/specs/core/capability-negotiation-3.2.html
-    QStringList _capsQueued;                /// Capabilities to be checked
-    QHash<QString, QString> _capsPending;   /// Capabilities pending 'CAP ACK' from server
-    QHash<QString, QString> _capsSupported; /// Enabled capabilities that received 'CAP ACK'
+    QStringList _capsQueuedIndividual;  /// Capabilities to check that require one at a time requests
+    QStringList _capsQueuedBundled;     /// Capabilities to check that can be grouped together
+    QStringList _capsQueuedLastBundle;  /// Most recent capability bundle requested (no individuals)
+    // Some capabilities, such as SASL, require follow-up messages to be fully enabled.  These
+    // capabilities should not be grouped with others to avoid requesting new capabilities while the
+    // previous capability is still being set up.
+    // Additionally, IRC servers can choose to send a 'NAK' to any set of requested capabilities.
+    // If this happens, we need a way to retry each capability individually in order to avoid having
+    // one failing capability (e.g. SASL) block all other capabilities.
+
+    bool _capNegotiationActive;  /// Whether or not full capability negotiation was started
+    // Avoid displaying repeat "negotiation finished" messages
+    bool _capInitialNegotiationEnded;  /// Whether or not initial capability negotiation finished
+    // Avoid sending repeat "CAP END" replies when registration is already ended
+
+    /**
+     * Gets the next set of capabilities to request, removing them from the queue.
+     *
+     * May return one or multiple space-separated capabilities, depending on queue.
+     *
+     * @returns Space-separated names of capabilities to request, or empty string if none remain
+     */
+    QString takeQueuedCaps();
+
+    /**
+     * Maximum length of a single 'CAP REQ' command.
+     *
+     * To be safe, 100 chars.  Higher numbers should be possible; this is following the conservative
+     * minimum number of characters that IRC servers must return in CAP NAK replies.  This also
+     * means CAP NAK replies will contain the full list of denied capabilities.
+     *
+     * See: http://ircv3.net/specs/core/capability-negotiation-3.1.html
+     */
+    const int maxCapRequestLength = 100;
 
     QTimer _tokenBucketTimer;
-    int _messageDelay;      // token refill speed in ms
-    int _burstSize;         // size of the token bucket
-    int _tokenBucket;       // the virtual bucket that holds the tokens
-    QList<QByteArray> _msgQueue;
+    // No need for int type as one cannot travel into the past (at least not yet, Doc)
+    quint32 _messageDelay;        /// Token refill speed in ms
+    quint32 _burstSize;           /// Size of the token bucket
+    quint32 _tokenBucket;         /// The virtual bucket that holds the tokens
+    QList<QByteArray> _msgQueue;  /// Queue of messages waiting to be sent
+    bool _skipMessageRates;       /// If true, skip all message rate limits
 
-    QString _requestedUserModes; // 2 strings separated by a '-' character. first part are requested modes to add, the second to remove
+    QString _requestedUserModes;  // 2 strings separated by a '-' character. first part are requested modes to add, the second to remove
 
     // List of blowfish keys for channels
     QHash<QString, QByteArray> _cipherKeys;
 };
-
-
-#endif //CORENETWORK_H