Implement a metrics daemon to expose Prometheus metrics
[quassel.git] / src / core / corenetwork.h
index 6ad77b6..01d71ed 100644 (file)
@@ -1,5 +1,5 @@
 /***************************************************************************
- *   Copyright (C) 2005-2018 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"
-
-// IRCv3 capabilities
-#include "irccap.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;
@@ -52,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(); }
 
@@ -129,7 +130,9 @@ public:
      */
     inline bool isPongReplyPending() const { return _pongReplyPending; }
 
-    QList<QList<QByteArray>> splitMessage(const QString &cmd, const QString &message, std::function<QList<QByteArray>(QString &)> cmdGenerator);
+    QList<QList<QByteArray>> splitMessage(const QString& cmd,
+                                          const QString& message,
+                                          const std::function<QList<QByteArray>(QString&)>& cmdGenerator);
 
     // IRCv3 capability negotiation
 
@@ -138,8 +141,7 @@ public:
      *
      * @returns True if in progress, otherwise false
      */
-    inline bool capNegotiationInProgress() const { return (!_capsQueuedIndividual.empty() ||
-                                                           !_capsQueuedBundled.empty()); }
+    inline bool capNegotiationInProgress() const { return (!_capsQueuedIndividual.empty() || !_capsQueuedBundled.empty()); }
 
     /**
      * Queues a capability to be requested.
@@ -149,7 +151,7 @@ public:
      *
      * @param[in] capability Name of the capability
      */
-    void queueCap(const QString &capability);
+    void queueCap(const QStringcapability);
 
     /**
      * Begins capability negotiation if capabilities are queued, otherwise returns.
@@ -192,20 +194,18 @@ public:
      *
      * See: http://ircv3.net/specs/extensions/sasl-3.2.html
      */
-    const QStringList capsRequiringConfiguration = QStringList {
-        IrcCap::SASL
-    };
+    const QStringList capsRequiringConfiguration = QStringList{IrcCap::SASL};
 
 public slots:
-    virtual void setMyNick(const QString &mynick);
+    void setMyNick(const QString& mynick) override;
 
-    virtual void requestConnect() const;
-    virtual void requestDisconnect() const;
-    virtual void requestSetNetworkInfo(const NetworkInfo &info);
+    void requestConnect() const override;
+    void requestDisconnect() const override;
+    void requestSetNetworkInfo(const NetworkInfo& info) override;
 
-    virtual void setUseAutoReconnect(bool);
-    virtual void setAutoReconnectInterval(quint32);
-    virtual void setAutoReconnectRetries(quint16);
+    void setUseAutoReconnect(bool) override;
+    void setAutoReconnectInterval(quint32) override;
+    void setAutoReconnectRetries(quint16) override;
 
     void setPingInterval(int interval);
 
@@ -220,6 +220,13 @@ public slots:
      */
     void setPongTimestampValid(bool validTimestamp);
 
+    /**
+     * Indicates that the CoreSession is shutting down.
+     *
+     * Disconnects the network if connected, and sets a flag that prevents reconnections.
+     */
+    void shutdown();
+
     void connectToIrc(bool reconnecting = false);
     /**
      * Disconnect from the IRC server.
@@ -229,23 +236,21 @@ public slots:
      * @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)
-     * @param forceImmediate  Immediately disconnect from network, skipping queue of other commands
      */
-    void disconnectFromIrc(bool requested = true, const QString &reason = QString(),
-                           bool withReconnect = false, bool forceImmediate = false);
+    void disconnectFromIrc(bool requested = true, const QString& reason = QString(), bool withReconnect = false);
 
     /**
      * Forcibly close the IRC server socket, waiting for it to close.
      *
      * Call CoreNetwork::disconnectFromIrc() first, allow the event loop to run, then if you need to
-     * be sure the network's disconencted (e.g. clean-up), call this.
+     * be sure the network's disconnected (e.g. clean-up), call this.
      *
      * @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
      */
     bool forceDisconnect(int msecs = 1000);
 
-    void userInput(BufferInfo bufferInfo, QString msg);
+    void userInput(const BufferInfo& bufferInfo, QString msg);
 
     /**
      * Sends the raw (encoded) line, adding to the queue if needed, optionally with higher priority.
@@ -258,7 +263,7 @@ public slots:
      * PING/PONG replies, the other side will close the connection.
      * @endparmblock
      */
-    void putRawLine(const QByteArray input, const bool prepend = false);
+    void putRawLine(const QByteArray& input, bool prepend = false);
 
     /**
      * Sends the command with encoded parameters, with optional prefix or high priority.
@@ -273,7 +278,7 @@ public slots:
      * maintain PING/PONG replies, the other side will close the connection.
      * @endparmblock
      */
-    void putCmd(const QString &cmd, const QList<QByteArray> &params, const QByteArray &prefix = QByteArray(), const bool prepend = false);
+    void putCmd(const QString& cmd, const QList<QByteArray>& params, const QByteArray& prefix = {}, const QHash<IrcTagKey, QString> &tags = {}, bool prepend = false);
 
     /**
      * Sends the command for each set of encoded parameters, with optional prefix or high priority.
@@ -292,19 +297,19 @@ public slots:
      * 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 = QByteArray(), const bool prependAll = false);
+    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)
@@ -330,7 +335,7 @@ public slots:
      * set by the user.  Use with caution and remember to re-enable configured limits when done.
      * @endparmblock
      */
-    void updateRateLimiting(const bool forceUnlimited = false);
+    void updateRateLimiting(bool forceUnlimited = false);
 
     /**
      * Resets the token bucket up to the maximum
@@ -350,7 +355,7 @@ public slots:
      *
      * @param[in] capability Name of the capability
      */
-    void serverCapAdded(const QString &capability);
+    void serverCapAdded(const QStringcapability);
 
     /**
      * Indicates a capability was acknowledged (enabled by the IRC server).
@@ -359,7 +364,7 @@ public slots:
      *
      * @param[in] capability Name of the capability
      */
-    void serverCapAcknowledged(const QString &capability);
+    void serverCapAcknowledged(const QStringcapability);
 
     /**
      * Indicates a capability was removed from the list of available capabilities.
@@ -368,7 +373,7 @@ public slots:
      *
      * @param[in] capability Name of the capability
      */
-    void serverCapRemoved(const QString &capability);
+    void serverCapRemoved(const QStringcapability);
 
     /**
      * Sends the next capability from the queue.
@@ -388,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();
 
@@ -407,43 +418,53 @@ public slots:
      */
     inline void resetPongReplyPending() { _pongReplyPending = false; }
 
-    inline void displayMsg(Message::Type msgType, BufferInfo::Type bufferType, const QString &target, const QString &text, const QString &sender = "", Message::Flags flags = Message::None)
+    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, qint64 socketId);
-    void socketDisconnected(const CoreIdentity *identity, const QHostAddress &localAddress, quint16 localPort, const QHostAddress &peerAddress, quint16 peerPort, qint64 socketId);
+    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();
@@ -456,7 +477,7 @@ private slots:
     void startAutoWhoCycle();
 
 #ifdef HAVE_SSL
-    void sslErrors(const QList<QSslError> &errors);
+    void onSslErrors(const QList<QSslError>& errors);
 #endif
 
     /**
@@ -478,10 +499,19 @@ private slots:
      */
     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;
@@ -490,9 +520,10 @@ private:
 #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;
@@ -509,6 +540,8 @@ private:
     // 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;
 
@@ -519,7 +552,7 @@ private:
     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
+    bool _pongReplyPending = false;  ///< If true, at least one PING sent without a PONG reply
 
     QStringList _autoWhoQueue;
     QHash<QString, int> _autoWhoPending;
@@ -537,9 +570,9 @@ private:
     // 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
+    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
+    bool _capInitialNegotiationEnded;  /// Whether or not initial capability negotiation finished
     // Avoid sending repeat "CAP END" replies when registration is already ended
 
     /**
@@ -564,17 +597,14 @@ private:
 
     QTimer _tokenBucketTimer;
     // 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
+    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