add a skeleton for QXmppServer

3 files changed, 529 insertions, 0 deletions

diff --git a/src/QXmppServer.cpp b/src/QXmppServer.cpp
new file mode 100644
index 00000000..36b1ee58
--- /dev/null
+++ b/src/QXmppServer.cpp
@@ -0,0 +1,419 @@
+/*
+ * Copyright (C) 2008-2010 The QXmpp developers
+ *
+ * Author:
+ *  Jeremy Lainé
+ *
+ * Source:
+ *  http://code.google.com/p/qxmpp
+ *
+ * This file is a part of QXmpp library.
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ */
+
+#include <QDomElement>
+#include <QSslSocket>
+
+#include "QXmppDialback.h"
+#include "QXmppIncomingClient.h"
+#include "QXmppIncomingServer.h"
+#include "QXmppOutgoingServer.h"
+#include "QXmppServer.h"
+#include "QXmppUtils.h"
+
+static QString jidToDomain(const QString &to)
+{
+    return jidToBareJid(to).split("@").last();
+}
+
+class QXmppServerPrivate
+{
+public:
+    QXmppServerPrivate();
+
+    QString domain;
+    QMap<QString, QSet<QString> > subscribers;
+    QXmppLogger *logger;
+    QXmppPasswordChecker *passwordChecker;
+
+    // client-to-server
+    QList<QXmppIncomingClient*> incomingClients;
+
+    // server-to-server
+    QList<QXmppOutgoingServer*> outgoingServers;
+};
+
+QXmppServerPrivate::QXmppServerPrivate()
+    : logger(0),
+    passwordChecker(0)
+{
+}
+
+/// Constructs a new XMPP server instance.
+///
+/// \param parent
+
+QXmppServer::QXmppServer(QObject *parent)
+    : QObject(parent),
+    d(new QXmppServerPrivate)
+{
+}
+
+/// Destroys an XMPP server instance.
+///
+
+QXmppServer::~QXmppServer()
+{
+    delete d;
+}
+
+/// Returns the server's domain.
+///
+
+QString QXmppServer::domain() const
+{
+    return d->domain;
+}
+
+/// Sets the server's domain.
+///
+/// \param domain
+
+void QXmppServer::setDomain(const QString &domain)
+{
+    d->domain = domain;
+}
+
+/// Returns the QXmppLogger associated with the server.
+
+QXmppLogger *QXmppServer::logger()
+{
+    return d->logger;
+}
+
+/// Sets the QXmppLogger associated with the server.
+
+void QXmppServer::setLogger(QXmppLogger *logger)
+{
+    d->logger = logger;
+}
+
+/// Returns the password checker used to verify client credentials.
+///
+/// \param checker
+///
+
+QXmppPasswordChecker *QXmppServer::passwordChecker()
+{
+    return d->passwordChecker;
+}
+
+/// Sets the password checker used to verify client credentials.
+///
+/// \param checker
+///
+
+void QXmppServer::setPasswordChecker(QXmppPasswordChecker *checker)
+{
+    d->passwordChecker = checker;
+}
+
+
+QXmppOutgoingServer* QXmppServer::connectToDomain(const QString &domain)
+{
+    // initialise outgoing server-to-server
+    QXmppOutgoingServer *stream = new QXmppOutgoingServer(d->domain, this);
+    stream->setObjectName("S2S-out-" + domain);
+    stream->setLocalStreamKey(generateStanzaHash().toAscii());
+    stream->setLogger(d->logger);
+    stream->configuration().setDomain(domain);
+    stream->configuration().setHost(domain);
+    stream->configuration().setPort(5269);
+    bool check = connect(stream, SIGNAL(elementReceived(QDomElement, bool&)),
+                         this, SLOT(slotElementReceived(QDomElement, bool&)));
+    Q_ASSERT(check);
+    Q_UNUSED(check);
+
+    d->outgoingServers.append(stream);
+
+    // connect to remote server
+    stream->connectToHost();
+    return stream;
+}
+
+/// Returns the XMPP streams for the given recipient.
+///
+/// \param to
+///
+
+QList<QXmppStream*> QXmppServer::getStreams(const QString &to)
+{
+    QList<QXmppStream*> found;
+    if (to.isEmpty())
+        return found;
+    const QString toDomain = jidToDomain(to);
+    if (toDomain == d->domain)
+    {
+        // look for a client connection
+        foreach (QXmppIncomingClient *conn, d->incomingClients)
+        {
+            if (conn->jid() == to || jidToBareJid(conn->jid()) == to)
+                found << conn;
+        }
+    } else {
+        // look for an outgoing S2S connection
+        foreach (QXmppOutgoingServer *conn, d->outgoingServers)
+        {
+            if (conn->configuration().domain() == toDomain && conn->isConnected())
+            {
+                found << conn;
+                break;
+            }
+        }
+
+        // if we did not find an outgoing server,
+        // we need to establish the S2S connection
+        if (found.isEmpty())
+        {
+            connectToDomain(toDomain);
+
+            // FIXME : the current packet will not be delivered
+        }
+    }
+    return found;
+}
+
+/// Route an XMPP stanza.
+///
+/// \param element
+
+bool QXmppServer::sendElement(const QDomElement &element)
+{
+    bool sent = false;
+    const QString to = element.attribute("to");
+    foreach (QXmppStream *conn, getStreams(to))
+    {
+        if (conn->sendElement(element))
+            sent = true;
+    }
+    return sent;
+}
+
+/// Route an XMPP packet.
+///
+/// \param packet
+
+bool QXmppServer::sendPacket(const QXmppStanza &packet)
+{
+    bool sent = false;
+    foreach (QXmppStream *conn, getStreams(packet.to()))
+    {
+        if (conn->sendPacket(packet))
+            sent = true;
+    }
+    return sent;
+}
+
+/// Handle a new incoming TCP connection from a client.
+///
+/// \param socket
+
+void QXmppServer::slotClientConnection(QSslSocket *socket)
+{
+    QXmppIncomingClient *stream = new QXmppIncomingClient(socket, d->domain, this);
+    socket->setParent(stream);
+    stream->setLogger(d->logger);
+    stream->setPasswordChecker(d->passwordChecker);
+
+    bool check = connect(stream, SIGNAL(connected()),
+                         this, SLOT(slotClientConnected()));
+    Q_ASSERT(check);
+
+    check = connect(stream, SIGNAL(disconnected()),
+                    this, SLOT(slotClientDisconnected()));
+    Q_ASSERT(check);
+
+    check = connect(stream, SIGNAL(elementReceived(QDomElement, bool&)),
+                    this, SLOT(slotElementReceived(QDomElement, bool&)));
+    Q_ASSERT(check);
+
+    d->incomingClients.append(stream);
+}
+
+/// Handle a successful client authentication.
+///
+
+void QXmppServer::slotClientConnected()
+{
+    QXmppIncomingClient *stream = qobject_cast<QXmppIncomingClient *>(sender());
+    if (!stream || !d->incomingClients.contains(stream))
+        return;
+
+    // check whether the connection conflicts with another one
+    foreach (QXmppIncomingClient *conn, d->incomingClients)
+    {
+        if (conn != stream && conn->jid() == stream->jid())
+        {
+            conn->sendData("<stream:error><conflict xmlns='urn:ietf:params:xml:ns:xmpp-streams'/><text xmlns='urn:ietf:params:xml:ns:xmpp-streams'>Replaced by new connection</text></stream:error>");
+            conn->disconnectFromHost();
+        }
+    }
+
+    // update statistics
+    updateStatistics();
+}
+
+/// Handle a disconnection from a client.
+///
+
+void QXmppServer::slotClientDisconnected()
+{
+    QXmppIncomingClient *stream = qobject_cast<QXmppIncomingClient *>(sender());
+    if (!stream || !d->incomingClients.contains(stream))
+        return;
+
+    // notify subscribed peers of disconnection
+    if (stream->isConnected())
+    {
+        foreach (QString subscriber, d->subscribers.value(stream->jid()))
+        {
+            QXmppPresence presence;
+            presence.setFrom(stream->jid());
+            presence.setTo(subscriber);
+            presence.setType(QXmppPresence::Unavailable);
+            sendPacket(presence);
+        }
+    }
+    d->incomingClients.removeAll(stream);
+    stream->deleteLater();
+
+    // update statistics
+    updateStatistics();
+}
+
+void QXmppServer::slotDialbackRequestReceived(const QXmppDialback &dialback)
+{
+    QXmppIncomingServer *stream = qobject_cast<QXmppIncomingServer *>(sender());
+    if (!stream)
+        return;
+
+    if (dialback.command() == QXmppDialback::Verify)
+    {
+        // handle a verify request
+        foreach (QXmppOutgoingServer *out, d->outgoingServers)
+        {
+            if (out->configuration().domain() != dialback.from())
+                continue;
+
+            bool isValid = dialback.key() == out->localStreamKey();
+            QXmppDialback verify;
+            verify.setCommand(QXmppDialback::Verify);
+            verify.setId(dialback.id());
+            verify.setTo(dialback.from());
+            verify.setFrom(d->domain);
+            verify.setType(isValid ? "valid" : "invalid");
+            stream->sendPacket(verify);
+            return;
+        }
+    }
+}
+
+/// Handle an incoming XML element.
+
+void QXmppServer::slotElementReceived(const QDomElement &origElement, bool &handled)
+{
+    QXmppStream *incoming = qobject_cast<QXmppStream *>(sender());
+    if (!incoming)
+        return;
+    handleStanza(incoming, origElement);
+    handled = true;
+}
+
+/// Handle a new incoming TCP connection from a server.
+///
+/// \param socket
+
+void QXmppServer::slotServerConnection(QSslSocket *socket)
+{
+    QXmppIncomingServer *stream = new QXmppIncomingServer(socket, d->domain, this);
+    socket->setParent(stream);
+    stream->setLogger(d->logger);
+
+    bool check = connect(stream, SIGNAL(disconnected()),
+                         stream, SLOT(deleteLater()));
+    Q_ASSERT(check);
+
+    check = connect(stream, SIGNAL(dialbackRequestReceived(QXmppDialback)),
+                    this, SLOT(slotDialbackRequestReceived(QXmppDialback)));
+    Q_ASSERT(check);
+
+    check = connect(stream, SIGNAL(elementReceived(QDomElement, bool&)),
+                    this, SLOT(slotElementReceived(QDomElement, bool&)));
+    Q_ASSERT(check);
+}
+
+class QXmppSslServerPrivate
+{
+public:
+    QString caCertificates;
+    QString localCertificate;
+    QString privateKey;
+};
+
+/// Constructs a new SSL server instance.
+///
+/// \param parent
+
+QXmppSslServer::QXmppSslServer(QObject *parent)
+    : QTcpServer(parent),
+    d(new QXmppSslServerPrivate)
+{
+}
+
+/// Destroys an SSL server instance.
+///
+
+QXmppSslServer::~QXmppSslServer()
+{
+    delete d;
+}
+
+void QXmppSslServer::incomingConnection(int socketDescriptor)
+{
+    QSslSocket *socket = new QSslSocket;
+    socket->setSocketDescriptor(socketDescriptor);
+    if (!d->localCertificate.isEmpty() && !d->privateKey.isEmpty())
+    {
+        socket->setProtocol(QSsl::AnyProtocol);
+        socket->addCaCertificates(d->caCertificates);
+        socket->setLocalCertificate(d->localCertificate);
+        socket->setPrivateKey(d->privateKey);
+    }
+    emit newConnection(socket);
+}
+
+void QXmppSslServer::addCaCertificates(const QString &caCertificates)
+{
+    d->caCertificates = caCertificates;
+}
+
+void QXmppSslServer::setLocalCertificate(const QString &localCertificate)
+{
+    d->localCertificate = localCertificate;
+}
+
+void QXmppSslServer::setPrivateKey(const QString &privateKey)
+{
+    d->privateKey = privateKey;
+}
+
diff --git a/src/QXmppServer.h b/src/QXmppServer.h
new file mode 100644
index 00000000..0c686334
--- /dev/null
+++ b/src/QXmppServer.h
@@ -0,0 +1,108 @@
+/*
+ * Copyright (C) 2008-2010 The QXmpp developers
+ *
+ * Author:
+ *  Jeremy Lainé
+ *
+ * Source:
+ *  http://code.google.com/p/qxmpp
+ *
+ * This file is a part of QXmpp library.
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ */
+
+#ifndef QXMPPSERVER_H
+#define QXMPPSERVER_H
+
+#include <QTcpServer>
+
+class QDomElement;
+class QSslSocket;
+
+class QXmppDialback;
+class QXmppLogger;
+class QXmppOutgoingServer;
+class QXmppPasswordChecker;
+class QXmppServerPrivate;
+class QXmppStanza;
+class QXmppStream;
+
+/// \brief The QXmppServer class represents an XMPP server.
+///
+/// \ingroup Core
+
+class QXmppServer : public QObject
+{
+    Q_OBJECT
+
+public:
+    QXmppServer(QObject *parent = 0);
+    ~QXmppServer();
+
+    QString domain() const;
+    void setDomain(const QString &domain);
+
+    QXmppLogger *logger();
+    void setLogger(QXmppLogger *logger);
+
+    QXmppPasswordChecker *passwordChecker();
+    void setPasswordChecker(QXmppPasswordChecker *checker);
+
+    bool sendElement(const QDomElement &element);
+    bool sendPacket(const QXmppStanza &stanza);
+
+private slots:
+    void slotClientConnection(QSslSocket *socket);
+    void slotClientConnected();
+    void slotClientDisconnected();
+    void slotDialbackRequestReceived(const QXmppDialback &dialback);
+    void slotElementReceived(const QDomElement &element, bool &handled);
+    void slotServerConnection(QSslSocket *socket);
+
+protected:
+    QXmppOutgoingServer *connectToDomain(const QString &domain);
+    // overridable methods
+    virtual void handleStanza(QXmppStream *stream, const QDomElement &element) = 0;
+    virtual void updateStatistics() = 0;
+
+private:
+    QList<QXmppStream*> getStreams(const QString &to);
+    QXmppServerPrivate * const d;
+};
+
+class QXmppSslServerPrivate;
+
+/// \brief The QXmppSslServer class represents an SSL-enabled TCP server.
+///
+
+class QXmppSslServer : public QTcpServer
+{
+    Q_OBJECT
+
+public:
+    QXmppSslServer(QObject *parent = 0);
+    ~QXmppSslServer();
+
+    void addCaCertificates(const QString &caCertificates);
+    void setLocalCertificate(const QString &localCertificate);
+    void setPrivateKey(const QString &privateKey);
+
+signals:
+    void newConnection(QSslSocket *socket);
+
+private:
+    void incomingConnection(int socketDescriptor);
+    QXmppSslServerPrivate * const d;
+};
+
+#endif
diff --git a/src/src.pro b/src/src.pro
index b14412c9..c9eb05b3 100644
--- a/src/src.pro
+++ b/src/src.pro
@@ -51,6 +51,7 @@ HEADERS += QXmppUtils.h \
     QXmppRoster.h \
     QXmppRosterIq.h \
     QXmppRosterManager.h \
+    QXmppServer.h \
     QXmppSessionIq.h \
     QXmppSocks.h \
     QXmppStanza.h \
@@ -99,6 +100,7 @@ SOURCES += QXmppUtils.cpp \
     QXmppPresence.cpp \
     QXmppRosterIq.cpp \
     QXmppRosterManager.cpp \
+    QXmppServer.cpp \
     QXmppSessionIq.cpp \
     QXmppSocks.cpp \
     QXmppStanza.cpp \