From 1f3472f71ae867d9fc5e5482a355f12465c23ece Mon Sep 17 00:00:00 2001
From: Jeremy Lainé <jeremy.laine@m4x.org>
Date: Sun, 29 Aug 2010 12:01:32 +0000
Subject: fix SASL authentication :

- on the client side, check the second challenge we receive
- on the server side, send second challenge
---
 src/QXmppIncomingClient.cpp | 77 +++++++++++++++++++++++++++------------------
 1 file changed, 47 insertions(+), 30 deletions(-)

(limited to 'src/QXmppIncomingClient.cpp')

diff --git a/src/QXmppIncomingClient.cpp b/src/QXmppIncomingClient.cpp
index f70d6edf..eb23d7bc 100644
--- a/src/QXmppIncomingClient.cpp
+++ b/src/QXmppIncomingClient.cpp
@@ -29,6 +29,7 @@
 #include "QXmppBindIq.h"
 #include "QXmppConstants.h"
 #include "QXmppMessage.h"
+#include "QXmppSaslAuth.h"
 #include "QXmppSessionIq.h"
 #include "QXmppStreamFeatures.h"
 #include "QXmppUtils.h"
@@ -44,7 +45,8 @@ public:
     QString username;
     QString resource;
     QXmppPasswordChecker *passwordChecker;
-    QByteArray saslNonce;
+    QXmppSaslDigestMd5 saslDigest;
+    int saslStep;
 };
 
 /// Constructs a new incoming client stream.
@@ -60,6 +62,7 @@ QXmppIncomingClient::QXmppIncomingClient(QSslSocket *socket, const QString &doma
 {
     d->passwordChecker = 0;
     d->domain = domain;
+    d->saslStep = 0;
 
     setObjectName("C2S-in");
     setSocket(socket);
@@ -119,6 +122,7 @@ void QXmppIncomingClient::setPasswordChecker(QXmppPasswordChecker *checker)
 void QXmppIncomingClient::handleStream(const QDomElement &streamElement)
 {
     d->idleTimer->start();
+    d->saslStep = 0;
 
     // start stream
     const QByteArray sessionId = generateStanzaHash().toAscii();
@@ -208,14 +212,13 @@ void QXmppIncomingClient::handleStanza(const QDomElement &nodeRecv)
             else if (mechanism == "DIGEST-MD5")
             {
                 // generate nonce
-                QByteArray nonce(32, 'm');
-                for(int n = 0; n < nonce.size(); ++n)
-                    nonce[n] = (char)(256.0*qrand()/(RAND_MAX+1.0));
-                d->saslNonce = nonce.toBase64();
+                d->saslDigest.setNonce(QXmppSaslDigestMd5::generateNonce());
+                d->saslDigest.setRealm(d->domain.toUtf8());
+                d->saslStep = 1;
 
                 QMap<QByteArray, QByteArray> challenge;
-                challenge["nonce"] = d->saslNonce;
-                challenge["realm"] = d->domain.toUtf8();
+                challenge["nonce"] = d->saslDigest.nonce();
+                challenge["realm"] = d->saslDigest.realm();
                 challenge["qop"] = "auth";
                 challenge["charset"] = "utf-8";
                 challenge["algorithm"] = "md5-sess";
@@ -236,32 +239,45 @@ void QXmppIncomingClient::handleStanza(const QDomElement &nodeRecv)
             const QByteArray raw = QByteArray::fromBase64(nodeRecv.text().toAscii());
             QMap<QByteArray, QByteArray> response = parseDigestMd5(raw);
 
-            // check credentials
-            const QString username = QString::fromUtf8(response.value("username"));
-            QString password;
-            if (!d->passwordChecker || !d->passwordChecker->getPassword(username, password))
+            if (d->saslStep == 1)
             {
-                sendData("<failure xmlns='urn:ietf:params:xml:ns:xmpp-sasl'><not-authorized/></failure>");
-                disconnectFromHost();
-                return;
+                // check credentials
+                const QString username = QString::fromUtf8(response.value("username"));
+                QString password;
+                if (!d->passwordChecker || !d->passwordChecker->getPassword(username, password))
+                {
+                    sendData("<failure xmlns='urn:ietf:params:xml:ns:xmpp-sasl'><not-authorized/></failure>");
+                    disconnectFromHost();
+                    return;
+                }
+                d->saslDigest.setUsername(username.toUtf8());
+                d->saslDigest.setPassword(password.toUtf8());
+                d->saslDigest.setDigestUri(response.value("digest-uri"));
+                d->saslDigest.setNc(response.value("nc"));
+                d->saslDigest.setCnonce(response.value("cnonce"));
+                if (response["response"] != d->saslDigest.calculateDigest(
+                        QByteArray("AUTHENTICATE:") + d->saslDigest.digestUri()))
+                {
+                    sendData("<failure xmlns='urn:ietf:params:xml:ns:xmpp-sasl'><not-authorized/></failure>");
+                    disconnectFromHost();
+                    return;
+                }
+
+                // send new challenge
+                d->username = username;
+                d->saslStep = 2;
+                QMap<QByteArray, QByteArray> challenge;
+                challenge["rspauth"] = d->saslDigest.calculateDigest(
+                    QByteArray(":") + d->saslDigest.digestUri());
+                const QByteArray data = serializeDigestMd5(challenge).toBase64();
+                sendData("<challenge xmlns='urn:ietf:params:xml:ns:xmpp-sasl'>" + data +"</challenge>");
             }
-            const QByteArray a1 = username.toUtf8() + ':' + d->domain.toUtf8() + ':' + password.toUtf8();
-            const QByteArray remote = QByteArray::fromHex(response["response"]);
-            if (remote != calculateDigestMd5(a1,
-                d->saslNonce,
-                response.value("nc"),
-                response.value("cnonce"),
-                response.value("digest-uri"),
-                QByteArray()))
+            else if (d->saslStep == 2)
             {
-                sendData("<failure xmlns='urn:ietf:params:xml:ns:xmpp-sasl'><not-authorized/></failure>");
-                disconnectFromHost();
-                return;
+                // authentication succeeded
+                d->saslStep = 3;
+                sendData("<success xmlns='urn:ietf:params:xml:ns:xmpp-sasl'/>");
             }
-
-            // authentication succeeded
-            d->username = username;
-            sendData("<success xmlns='urn:ietf:params:xml:ns:xmpp-sasl'/>");
         }
     }
     else if (ns == ns_client)
@@ -290,9 +306,10 @@ void QXmppIncomingClient::handleStanza(const QDomElement &nodeRecv)
                 QXmppSessionIq sessionSet;
                 sessionSet.parse(nodeRecv);
 
-                QXmppSessionIq sessionResult;
+                QXmppIq sessionResult;
                 sessionResult.setType(QXmppIq::Result);
                 sessionResult.setId(sessionSet.id());
+                sessionResult.setTo(jid());
                 sendPacket(sessionResult);
                 return;
             }
-- 
cgit v1.2.3