Add support for away-notify capability
[quassel.git] / src / core / corenetwork.h
index fd96ff6..88ea9bd 100644 (file)
@@ -1,5 +1,5 @@
 /***************************************************************************
- *   Copyright (C) 2005-08 by the Quassel Project                          *
+ *   Copyright (C) 2005-2015 by the Quassel Project                        *
  *   devel@quassel-irc.org                                                 *
  *                                                                         *
  *   This program is free software; you can redistribute it and/or modify  *
  *   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.         *
  ***************************************************************************/
 
 #ifndef CORENETWORK_H
 #define CORENETWORK_H
 
 #include "network.h"
+#include "coreircchannel.h"
+#include "coreircuser.h"
 
-class CoreSession;
+#include <QTimer>
 
-class CoreNetwork : public Network {
-  Q_OBJECT
+#ifdef HAVE_SSL
+# include <QSslSocket>
+# include <QSslError>
+#else
+# include <QTcpSocket>
+#endif
+
+#ifdef HAVE_QCA2
+#  include "cipher.h"
+#endif
+
+#include "coresession.h"
+
+#include <functional>
+
+class CoreIdentity;
+class CoreUserInputHandler;
+class CoreIgnoreListManager;
+class Event;
+
+class CoreNetwork : public Network
+{
+    SYNCABLE_OBJECT
+        Q_OBJECT
 
 public:
-  CoreNetwork(const NetworkId &networkid, CoreSession *session);
+    CoreNetwork(const NetworkId &networkid, CoreSession *session);
+    ~CoreNetwork();
+    inline virtual const QMetaObject *syncMetaObject() const { return &Network::staticMetaObject; }
+
+    inline CoreIdentity *identityPtr() const { return coreSession()->identity(identity()); }
+    inline CoreSession *coreSession() const { return _coreSession; }
+    inline CoreNetworkConfig *networkConfig() const { return coreSession()->networkConfig(); }
+
+    inline CoreUserInputHandler *userInputHandler() const { return _userInputHandler; }
+    inline CoreIgnoreListManager *ignoreListManager() { return coreSession()->ignoreListManager(); }
+
+    //! Decode a string using the server (network) decoding.
+    inline QString serverDecode(const QByteArray &string) 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;
+
+    //! 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;
+
+    //! Encode a string using the server (network) encoding.
+    inline QByteArray serverEncode(const QString &string) 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;
+
+    //! 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;
+
+    inline QString channelKey(const QString &channel) 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 bool isAutoWhoInProgress(const QString &channel) const { return _autoWhoPending.value(channel.toLower(), 0); }
+
+    inline UserId userId() const { return _coreSession->user(); }
 
-  inline virtual const QMetaObject *syncMetaObject() const { return &Network::staticMetaObject; }
+    inline QAbstractSocket::SocketState socketState() const { return socket.state(); }
+    inline bool socketConnected() const { return socket.state() == QAbstractSocket::ConnectedState; }
+    inline QHostAddress localAddress() const { return socket.localAddress(); }
+    inline QHostAddress peerAddress() const { return socket.peerAddress(); }
+    inline quint16 localPort() const { return socket.localPort(); }
+    inline quint16 peerPort() const { return socket.peerPort(); }
 
-  inline CoreSession *coreSession() const { return _coreSession; }
+    QList<QList<QByteArray>> splitMessage(const QString &cmd, const QString &message, std::function<QList<QByteArray>(QString &)> cmdGenerator);
+
+    // IRCv3 capability negotiation
+
+    /**
+     * Checks if a given capability is enabled.
+     *
+     * @returns True if enabled, otherwise false
+     */
+    inline bool capEnabled(const QString &capability) const { return _capsSupported.contains(capability); }
+
+    /**
+     * Checks if capability negotiation is currently ongoing.
+     *
+     * @returns True if in progress, otherwise false
+     */
+    inline bool capNegotiationInProgress() const { return !_capsQueued.empty(); }
+
+    /**
+     * Gets the value of an enabled or pending capability, e.g. sasl=plain.
+     *
+     * @returns Value of capability if one was specified, otherwise empty string
+     */
+    QString capValue(const QString &capability) const;
+
+    /**
+     * Gets the next capability to request, removing it from the queue.
+     *
+     * @returns Name of capability to request
+     */
+    QString takeQueuedCap();
+
+    // Specific capabilities for easy reference
+
+    /**
+     * Gets the status of the sasl authentication capability.
+     *
+     * http://ircv3.net/specs/extensions/sasl-3.2.html
+     *
+     * @returns True if SASL authentication is enabled, otherwise false
+     */
+    inline bool useCapSASL() const { return capEnabled("sasl"); }
+
+    /**
+     * Gets the status of the away-notify capability.
+     *
+     * http://ircv3.net/specs/extensions/away-notify-3.1.html
+     *
+     * @returns True if away-notify is enabled, otherwise false
+     */
+    inline bool useCapAwayNotify() const { return capEnabled("away-notify"); }
 
 public slots:
-  virtual void requestConnect() const;
-  virtual void requestDisconnect() const;
-  virtual void requestSetNetworkInfo(const NetworkInfo &info);
+    virtual void setMyNick(const QString &mynick);
+
+    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);
+
+    void connectToIrc(bool reconnecting = false);
+    void disconnectFromIrc(bool requested = true, const QString &reason = QString(), bool withReconnect = 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());
+
+    void setChannelJoined(const QString &channel);
+    void setChannelParted(const QString &channel);
+    void addChannelKey(const QString &channel, const QString &key);
+    void removeChannelKey(const QString &channel);
+
+    // 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);
+#endif
+
+    // IRCv3 capability negotiation (can be connected to signals)
+
+    /**
+     * Marks a capability as accepted, providing an optional value.
+     *
+     * Removes it from queue of pending capabilities and triggers any capability-specific
+     * activation.
+     *
+     * @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());
+
+    /**
+     * Marks a capability as denied.
+     *
+     * Removes it from the queue of pending capabilities and triggers any capability-specific
+     * deactivation.
+     *
+     * @param[in] capability Name of the capability
+     */
+    void removeCap(const QString &capability);
+
+    /**
+     * Queues a capability as available but not yet accepted or denied.
+     *
+     * Capabilities should be queued when registration pauses for CAP LS for capabilities are only
+     * requested during login.
+     *
+     * @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 setAutoWhoEnabled(bool enabled);
+    void setAutoWhoInterval(int interval);
+    void setAutoWhoDelay(int delay);
+
+    /**
+     * Appends the given channel/nick to the front of the AutoWho queue.
+     *
+     * 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
+     */
+    void queueAutoWhoOneshot(const QString &channelOrNick);
+
+    bool setAutoWhoDone(const QString &channel);
+
+    void updateIssuedModes(const QString &requestedModes);
+    void updatePersistentModes(QString addModes, QString removeModes);
+    void resetPersistentModes();
+
+    Server usedServer() const;
+
+    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)
+    {
+        emit displayMsg(networkId(), msgType, bufferType, target, text, sender, flags);
+    }
+
+
+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 disconnected(NetworkId networkId);
+    void connectionError(const QString &errorMsg);
+
+    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);
+
+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); }
+
+protected slots:
+    // TODO: remove cached cipher keys, when appropriate
+    //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 networkInitialized();
+
+    void sendPerform();
+    void restoreUserModes();
+    void doAutoReconnect();
+    void sendPing();
+    void enablePingTimeout(bool enable = true);
+    void disablePingTimeout();
+    void sendAutoWho();
+    void startAutoWhoCycle();
+
+#ifdef HAVE_SSL
+    void sslErrors(const QList<QSslError> &errors);
+#endif
+
+    void fillBucketAndProcessQueue();
+
+    void writeToSocket(const QByteArray &data);
 
 private:
-  CoreSession *_coreSession;
+    CoreSession *_coreSession;
+
+#ifdef HAVE_SSL
+    QSslSocket socket;
+#else
+    QTcpSocket socket;
+#endif
+
+    CoreUserInputHandler *_userInputHandler;
+
+    QHash<QString, QString> _channelKeys; // stores persistent channels and their passwords, if any
+
+    QTimer _autoReconnectTimer;
+    int _autoReconnectCount;
+
+    QTimer _socketCloseTimer;
+
+    /* this flag triggers quitRequested() once the socket is closed
+     * it is needed to determine whether or not the connection needs to be
+     * in the automatic session restore. */
+    bool _quitRequested;
+    QString _quitReason;
+
+    bool _previousConnectionAttemptFailed;
+    int _lastUsedServerIndex;
+
+    QTimer _pingTimer;
+    uint _lastPingTime;
+    uint _pingCount;
+    bool _sendPings;
+
+    QStringList _autoWhoQueue;
+    QHash<QString, int> _autoWhoPending;
+    QTimer _autoWhoTimer, _autoWhoCycleTimer;
+
+    // CAPs may have parameter values
+    // 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'
+
+    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;
+
+    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