/***************************************************************************
- * Copyright (C) 2005-2014 by the Quassel Project *
+ * Copyright (C) 2005-2016 by the Quassel Project *
* devel@quassel-irc.org *
* *
* This program is free software; you can redistribute it and/or modify *
// TODO: support system application proxy (new in Qt 4.6)
+#include <QtEndian>
+
#ifdef HAVE_SSL
#include <QSslSocket>
#else
#include "client.h"
#include "clientsettings.h"
+#include "peerfactory.h"
-#include "protocols/legacy/legacypeer.h"
+#if QT_VERSION < 0x050000
+# include "../../3rdparty/sha512/sha512.h"
+#endif
using namespace Protocol;
ClientAuthHandler::ClientAuthHandler(CoreAccount account, QObject *parent)
: AuthHandler(parent),
_peer(0),
- _account(account)
+ _account(account),
+ _probing(false),
+ _legacy(false),
+ _connectionFeatures(0)
{
}
#endif
setSocket(socket);
- // handled by the base class for now; may need to rethink for protocol detection
- //connect(socket, SIGNAL(error(QAbstractSocket::SocketError)), SLOT(onSocketError(QAbstractSocket::SocketError)));
- //connect(socket, SIGNAL(stateChanged(QAbstractSocket::SocketState)), SLOT(onSocketStateChanged(QAbstractSocket::SocketState)));
+ connect(socket, SIGNAL(stateChanged(QAbstractSocket::SocketState)), SLOT(onSocketStateChanged(QAbstractSocket::SocketState)));
+ connect(socket, SIGNAL(readyRead()), SLOT(onReadyRead()));
connect(socket, SIGNAL(connected()), SLOT(onSocketConnected()));
emit statusMessage(tr("Connecting to %1...").arg(_account.accountName()));
}
-// TODO: handle protocol detection
-// This method might go away anyway, unless we really need our own states...
-/*
void ClientAuthHandler::onSocketStateChanged(QAbstractSocket::SocketState socketState)
{
- qDebug() << Q_FUNC_INFO << socketState;
QString text;
- AuthHandler::State state = UnconnectedState;
switch(socketState) {
- case QAbstractSocket::UnconnectedState:
- text = tr("Disconnected");
- state = UnconnectedState;
- break;
case QAbstractSocket::HostLookupState:
- text = tr("Looking up %1...").arg(_account.hostName());
- state = HostLookupState;
+ if (!_legacy)
+ text = tr("Looking up %1...").arg(_account.hostName());
break;
case QAbstractSocket::ConnectingState:
- text = tr("Connecting to %1...").arg(_account.hostName());
- state = ConnectingState;
+ if (!_legacy)
+ text = tr("Connecting to %1...").arg(_account.hostName());
break;
case QAbstractSocket::ConnectedState:
text = tr("Connected to %1").arg(_account.hostName());
- state = ConnectedState;
break;
case QAbstractSocket::ClosingState:
- text = tr("Disconnecting from %1...").arg(_account.hostName());
- state = ClosingState;
+ if (!_probing)
+ text = tr("Disconnecting from %1...").arg(_account.hostName());
+ break;
+ case QAbstractSocket::UnconnectedState:
+ if (!_probing) {
+ text = tr("Disconnected");
+ // Ensure the disconnected() signal is sent even if we haven't reached the Connected state yet.
+ // The baseclass implementation will make sure to only send the signal once.
+ // However, we do want to prefer a potential socket error signal that may be on route already, so
+ // give this a chance to overtake us by spinning the loop...
+ QTimer::singleShot(0, this, SLOT(onSocketDisconnected()));
+ }
break;
default:
break;
}
if (!text.isEmpty()) {
- setState(state);
emit statusMessage(text);
}
}
-*/
-// TODO: handle protocol detection
-/*
void ClientAuthHandler::onSocketError(QAbstractSocket::SocketError error)
{
- emit socketError(error, socket()->errorString());
+ if (_probing && error == QAbstractSocket::RemoteHostClosedError) {
+ _legacy = true;
+ return;
+ }
+
+ _probing = false; // all other errors are unrelated to probing and should be handled
+ AuthHandler::onSocketError(error);
}
-*/
-void ClientAuthHandler::onSocketConnected()
+
+void ClientAuthHandler::onSocketDisconnected()
{
- // TODO: protocol detection
+ if (_probing && _legacy) {
+ // Remote host has closed the connection while probing
+ _probing = false;
+ disconnect(socket(), SIGNAL(readyRead()), this, SLOT(onReadyRead()));
+ emit statusMessage(tr("Reconnecting in compatibility mode..."));
+ socket()->connectToHost(_account.hostName(), _account.port());
+ return;
+ }
+
+ AuthHandler::onSocketDisconnected();
+}
+
+void ClientAuthHandler::onSocketConnected()
+{
if (_peer) {
qWarning() << Q_FUNC_INFO << "Peer already exists!";
return;
}
- _peer = new LegacyPeer(this, socket(), this);
+ socket()->setSocketOption(QAbstractSocket::KeepAliveOption, true);
- connect(_peer, SIGNAL(transferProgress(int,int)), SIGNAL(transferProgress(int,int)));
-
- // compat only
- connect(_peer, SIGNAL(protocolVersionMismatch(int,int)), SLOT(onProtocolVersionMismatch(int,int)));
+ if (!_legacy) {
+ // First connection attempt, try probing for a capable core
+ _probing = true;
- emit statusMessage(tr("Synchronizing to core..."));
+ QDataStream stream(socket()); // stream handles the endianness for us
+ stream.setVersion(QDataStream::Qt_4_2);
- bool useSsl = false;
+ quint32 magic = Protocol::magic;
#ifdef HAVE_SSL
- useSsl = _account.useSsl();
+ if (_account.useSsl())
+ magic |= Protocol::Encryption;
#endif
+ magic |= Protocol::Compression;
+
+ stream << magic;
+
+ // here goes the list of protocols we support, in order of preference
+ PeerFactory::ProtoList protos = PeerFactory::supportedProtocols();
+ for (int i = 0; i < protos.count(); ++i) {
+ quint32 reply = protos[i].first;
+ reply |= protos[i].second<<8;
+ if (i == protos.count() - 1)
+ reply |= 0x80000000; // end list
+ stream << reply;
+ }
- _peer->dispatch(RegisterClient(Quassel::buildInfo().fancyVersionString, useSsl));
+ socket()->flush(); // make sure the probing data is sent immediately
+ return;
+ }
+
+ // If we arrive here, it's the second connection attempt, meaning probing was not successful -> enable legacy support
+
+ qDebug() << "Legacy core detected, switching to compatibility mode";
+
+ RemotePeer *peer = PeerFactory::createPeer(PeerFactory::ProtoDescriptor(Protocol::LegacyProtocol, 0), this, socket(), Compressor::NoCompression, this);
+ // Only needed for the legacy peer, as all others check the protocol version before instantiation
+ connect(peer, SIGNAL(protocolVersionMismatch(int,int)), SLOT(onProtocolVersionMismatch(int,int)));
+
+ setPeer(peer);
+}
+
+
+void ClientAuthHandler::onReadyRead()
+{
+ if (socket()->bytesAvailable() < 4)
+ return;
+
+ if (!_probing)
+ return; // make sure to not read more data than needed
+
+ _probing = false;
+ disconnect(socket(), SIGNAL(readyRead()), this, SLOT(onReadyRead()));
+
+ quint32 reply;
+ socket()->read((char *)&reply, 4);
+ reply = qFromBigEndian<quint32>(reply);
+
+ Protocol::Type type = static_cast<Protocol::Type>(reply & 0xff);
+ quint16 protoFeatures = static_cast<quint16>(reply>>8 & 0xffff);
+ _connectionFeatures = static_cast<quint8>(reply>>24);
+
+ Compressor::CompressionLevel level;
+ if (_connectionFeatures & Protocol::Compression)
+ level = Compressor::BestCompression;
+ else
+ level = Compressor::NoCompression;
+
+ RemotePeer *peer = PeerFactory::createPeer(PeerFactory::ProtoDescriptor(type, protoFeatures), this, socket(), level, this);
+ if (!peer) {
+ qWarning() << "No valid protocol supported for this core!";
+ emit errorPopup(tr("<b>Incompatible Quassel Core!</b><br>"
+ "None of the protocols this client speaks are supported by the core you are trying to connect to."));
+
+ requestDisconnect(tr("Core speaks none of the protocols we support"));
+ return;
+ }
+
+ if (peer->protocol() == Protocol::LegacyProtocol) {
+ connect(peer, SIGNAL(protocolVersionMismatch(int,int)), SLOT(onProtocolVersionMismatch(int,int)));
+ _legacy = true;
+ }
+
+ setPeer(peer);
}
}
-void ClientAuthHandler::handle(const ClientDenied &msg)
+void ClientAuthHandler::setPeer(RemotePeer *peer)
{
- emit errorPopup(msg.errorString);
- requestDisconnect(tr("The core refused connection from this client"));
+ qDebug().nospace() << "Using " << qPrintable(peer->protocolName()) << "...";
+
+ _peer = peer;
+ connect(_peer, SIGNAL(transferProgress(int,int)), SIGNAL(transferProgress(int,int)));
+
+ // The legacy protocol enables SSL later, after registration
+ if (!_account.useSsl() || _legacy)
+ startRegistration();
+ // otherwise, do it now
+ else
+ checkAndEnableSsl(_connectionFeatures & Protocol::Encryption);
}
-void ClientAuthHandler::handle(const ClientRegistered &msg)
+void ClientAuthHandler::startRegistration()
{
- _coreConfigured = msg.coreConfigured;
- _backendInfo = msg.backendInfo;
-
- Client::setCoreFeatures(static_cast<Quassel::Features>(msg.coreFeatures));
+ emit statusMessage(tr("Synchronizing to core..."));
+ // useSsl will be ignored by non-legacy peers
+ bool useSsl = false;
#ifdef HAVE_SSL
- CoreAccountSettings s;
- if (_account.useSsl()) {
- if (msg.sslSupported) {
- // Make sure the warning is shown next time we don't have SSL in the core
- s.setAccountValue("ShowNoCoreSslWarning", true);
-
- QSslSocket *sslSocket = qobject_cast<QSslSocket *>(socket());
- Q_ASSERT(sslSocket);
- connect(sslSocket, SIGNAL(encrypted()), SLOT(onSslSocketEncrypted()));
- connect(sslSocket, SIGNAL(sslErrors(QList<QSslError>)), SLOT(onSslErrors()));
- qDebug() << "Starting encryption...";
- sslSocket->flush();
- sslSocket->startClientEncryption();
- }
- else {
- if (s.accountValue("ShowNoCoreSslWarning", true).toBool()) {
- bool accepted = false;
- emit handleNoSslInCore(&accepted);
- if (!accepted) {
- requestDisconnect(tr("Unencrypted connection cancelled"));
- return;
- }
- s.setAccountValue("ShowNoCoreSslWarning", false);
- s.setAccountValue("SslCert", QString());
- }
- onConnectionReady();
- }
- return;
- }
+ useSsl = _account.useSsl();
#endif
- // if we use SSL we wait for the next step until every SSL warning has been cleared
- onConnectionReady();
-}
+ _peer->dispatch(RegisterClient(Quassel::buildInfo().fancyVersionString, Quassel::buildInfo().commitDate, useSsl));
+}
-#ifdef HAVE_SSL
-void ClientAuthHandler::onSslSocketEncrypted()
+void ClientAuthHandler::handle(const ClientDenied &msg)
{
- QSslSocket *socket = qobject_cast<QSslSocket *>(sender());
- Q_ASSERT(socket);
-
- if (!socket->sslErrors().count()) {
- // Cert is valid, so we don't want to store it as known
- // That way, a warning will appear in case it becomes invalid at some point
- CoreAccountSettings s;
- s.setAccountValue("SSLCert", QString());
- }
-
- emit encrypted(true);
- onConnectionReady();
+ emit errorPopup(msg.errorString);
+ requestDisconnect(tr("The core refused connection from this client"));
}
-void ClientAuthHandler::onSslErrors()
+void ClientAuthHandler::handle(const ClientRegistered &msg)
{
- QSslSocket *socket = qobject_cast<QSslSocket *>(sender());
- Q_ASSERT(socket);
-
- CoreAccountSettings s;
- QByteArray knownDigest = s.accountValue("SslCert").toByteArray();
-
- if (knownDigest != socket->peerCertificate().digest()) {
- bool accepted = false;
- bool permanently = false;
- emit handleSslErrors(socket, &accepted, &permanently);
-
- if (!accepted) {
- requestDisconnect(tr("Unencrypted connection canceled"));
- return;
- }
+ _coreConfigured = msg.coreConfigured;
+ _backendInfo = msg.backendInfo;
+ _authenticatorInfo = msg.authenticatorInfo;
- if (permanently)
- s.setAccountValue("SslCert", socket->peerCertificate().digest());
- else
- s.setAccountValue("SslCert", QString());
- }
+ Client::setCoreFeatures(static_cast<Quassel::Features>(msg.coreFeatures));
- socket->ignoreSslErrors();
+ // The legacy protocol enables SSL at this point
+ if(_legacy && _account.useSsl())
+ checkAndEnableSsl(msg.sslSupported);
+ else
+ onConnectionReady();
}
-#endif /* HAVE_SSL */
-
void ClientAuthHandler::onConnectionReady()
{
if (!_coreConfigured) {
// start wizard
- emit startCoreSetup(_backendInfo);
+ emit startCoreSetup(_backendInfo, _authenticatorInfo);
}
else // TODO: check if we need LoginEnabled
login();
_peer->setParent(0);
emit handshakeComplete(_peer, msg);
}
+
+
+/*** SSL Stuff ***/
+
+void ClientAuthHandler::checkAndEnableSsl(bool coreSupportsSsl)
+{
+#ifndef HAVE_SSL
+ Q_UNUSED(coreSupportsSsl);
+#else
+ CoreAccountSettings s;
+ if (coreSupportsSsl && _account.useSsl()) {
+ // Make sure the warning is shown next time we don't have SSL in the core
+ s.setAccountValue("ShowNoCoreSslWarning", true);
+
+ QSslSocket *sslSocket = qobject_cast<QSslSocket *>(socket());
+ Q_ASSERT(sslSocket);
+ connect(sslSocket, SIGNAL(encrypted()), SLOT(onSslSocketEncrypted()));
+ connect(sslSocket, SIGNAL(sslErrors(QList<QSslError>)), SLOT(onSslErrors()));
+ qDebug() << "Starting encryption...";
+ sslSocket->flush();
+ sslSocket->startClientEncryption();
+ }
+ else {
+ if (s.accountValue("ShowNoCoreSslWarning", true).toBool()) {
+ bool accepted = false;
+ emit handleNoSslInCore(&accepted);
+ if (!accepted) {
+ requestDisconnect(tr("Unencrypted connection cancelled"));
+ return;
+ }
+ s.setAccountValue("ShowNoCoreSslWarning", false);
+ s.setAccountValue("SslCert", QString());
+ s.setAccountValue("SslCertDigestVersion", QVariant(QVariant::Int));
+ }
+ if (_legacy)
+ onConnectionReady();
+ else
+ startRegistration();
+ }
+#endif
+}
+
+#ifdef HAVE_SSL
+
+void ClientAuthHandler::onSslSocketEncrypted()
+{
+ QSslSocket *socket = qobject_cast<QSslSocket *>(sender());
+ Q_ASSERT(socket);
+
+ if (!socket->sslErrors().count()) {
+ // Cert is valid, so we don't want to store it as known
+ // That way, a warning will appear in case it becomes invalid at some point
+ CoreAccountSettings s;
+ s.setAccountValue("SSLCert", QString());
+ s.setAccountValue("SslCertDigestVersion", QVariant(QVariant::Int));
+ }
+
+ emit encrypted(true);
+
+ if (_legacy)
+ onConnectionReady();
+ else
+ startRegistration();
+}
+
+
+void ClientAuthHandler::onSslErrors()
+{
+ QSslSocket *socket = qobject_cast<QSslSocket *>(sender());
+ Q_ASSERT(socket);
+
+ CoreAccountSettings s;
+ QByteArray knownDigest = s.accountValue("SslCert").toByteArray();
+ ClientAuthHandler::DigestVersion knownDigestVersion = static_cast<ClientAuthHandler::DigestVersion>(s.accountValue("SslCertDigestVersion").toInt());
+
+ QByteArray calculatedDigest;
+ switch (knownDigestVersion) {
+ case ClientAuthHandler::DigestVersion::Md5:
+ calculatedDigest = socket->peerCertificate().digest(QCryptographicHash::Md5);
+ break;
+
+ case ClientAuthHandler::DigestVersion::Sha2_512:
+#if QT_VERSION >= 0x050000
+ calculatedDigest = socket->peerCertificate().digest(QCryptographicHash::Sha512);
+#else
+ calculatedDigest = sha2_512(socket->peerCertificate().toDer());
+#endif
+ break;
+
+ default:
+ qWarning() << "Certificate digest version" << QString(knownDigestVersion) << "is not supported";
+ }
+
+ if (knownDigest != calculatedDigest) {
+ bool accepted = false;
+ bool permanently = false;
+ emit handleSslErrors(socket, &accepted, &permanently);
+
+ if (!accepted) {
+ requestDisconnect(tr("Unencrypted connection canceled"));
+ return;
+ }
+
+ if (permanently) {
+#if QT_VERSION >= 0x050000
+ s.setAccountValue("SslCert", socket->peerCertificate().digest(QCryptographicHash::Sha512));
+#else
+ s.setAccountValue("SslCert", sha2_512(socket->peerCertificate().toDer()));
+#endif
+ s.setAccountValue("SslCertDigestVersion", ClientAuthHandler::DigestVersion::Latest);
+ }
+ else {
+ s.setAccountValue("SslCert", QString());
+ s.setAccountValue("SslCertDigestVersion", QVariant(QVariant::Int));
+ }
+ }
+ else if (knownDigestVersion != ClientAuthHandler::DigestVersion::Latest) {
+#if QT_VERSION >= 0x050000
+ s.setAccountValue("SslCert", socket->peerCertificate().digest(QCryptographicHash::Sha512));
+#else
+ s.setAccountValue("SslCert", sha2_512(socket->peerCertificate().toDer()));
+#endif
+ s.setAccountValue("SslCertDigestVersion", ClientAuthHandler::DigestVersion::Latest);
+ }
+
+ socket->ignoreSslErrors();
+}
+
+#if QT_VERSION < 0x050000
+QByteArray ClientAuthHandler::sha2_512(const QByteArray &input) {
+ unsigned char output[64];
+ sha512((unsigned char*) input.constData(), input.size(), output, false);
+ // QByteArray::fromRawData() cannot be used here because that constructor
+ // does not copy "output" and the data is clobbered when the variable goes
+ // out of scope.
+ QByteArray result;
+ result.append((char*) output, 64);
+ return result;
+}
+#endif
+
+#endif /* HAVE_SSL */