X-Git-Url: https://git.quassel-irc.org/?p=quassel.git;a=blobdiff_plain;f=src%2Fcore%2Fcoreauthhandler.cpp;h=b28952006cc7c4b295eecaf1bd151b26b0f934bd;hp=3514ecd15580e0c4b4faa0193f014bdfae721bb7;hb=c0d6dc0dec628f2e143e37ecc95cec45e636f8a5;hpb=8c6448c2e0048389fbac9e2e9daf22b5b050d5b6 diff --git a/src/core/coreauthhandler.cpp b/src/core/coreauthhandler.cpp index 3514ecd1..b2895200 100644 --- a/src/core/coreauthhandler.cpp +++ b/src/core/coreauthhandler.cpp @@ -1,5 +1,5 @@ /*************************************************************************** - * Copyright (C) 2005-2018 by the Quassel Project * + * Copyright (C) 2005-2020 by the Quassel Project * * devel@quassel-irc.org * * * * This program is free software; you can redistribute it and/or modify * @@ -20,17 +20,19 @@ #include "coreauthhandler.h" -#ifdef HAVE_SSL -# include -#endif +#include -#include "core.h" +#include -using namespace Protocol; +#include "core.h" -CoreAuthHandler::CoreAuthHandler(QTcpSocket* socket, QObject* parent) +CoreAuthHandler::CoreAuthHandler(QSslSocket* socket, QObject* parent) : AuthHandler(parent) , _peer(nullptr) + , _metricsServer(Core::instance()->metricsServer()) + , _proxyReceived(false) + , _proxyLine({}) + , _useProxyLine(false) , _magicReceived(false) , _legacy(false) , _clientRegistered(false) @@ -44,13 +46,37 @@ CoreAuthHandler::CoreAuthHandler(QTcpSocket* socket, QObject* parent) void CoreAuthHandler::onReadyRead() { - if (socket()->bytesAvailable() < 4) - return; - // once we have selected a peer, we certainly don't want to read more data! if (_peer) return; + if (!_proxyReceived) { + quint32 magic; + socket()->peek((char*) &magic, 4); + magic = qFromBigEndian(magic); + + if (magic == Protocol::proxyMagic) { + if (!socket()->canReadLine()) { + return; + } + QByteArray line = socket()->readLine(108); + _proxyLine = ProxyLine::parseProxyLine(line); + if (_proxyLine.protocol != QAbstractSocket::UnknownNetworkLayerProtocol) { + QList subnets = Quassel::optionValue("proxy-cidr").split(","); + for (const QString& subnet : subnets) { + if (socket()->peerAddress().isInSubnet(QHostAddress::parseSubnet(subnet))) { + _useProxyLine = true; + break; + } + } + } + } + _proxyReceived = true; + } + + if (socket()->bytesAvailable() < 4) + return; + if (!_magicReceived) { quint32 magic; socket()->peek((char*)&magic, 4); @@ -100,7 +126,7 @@ void CoreAuthHandler::onReadyRead() RemotePeer* peer = PeerFactory::createPeer(_supportedProtos, this, socket(), level, this); if (!peer) { - qWarning() << "Received invalid handshake data from client" << socket()->peerAddress().toString(); + qWarning() << "Received invalid handshake data from client" << hostAddress().toString(); close(); return; } @@ -129,6 +155,9 @@ void CoreAuthHandler::setPeer(RemotePeer* peer) qDebug().nospace() << "Using " << qPrintable(peer->protocolName()) << "..."; _peer = peer; + if (_proxyLine.protocol != QAbstractSocket::UnknownNetworkLayerProtocol) { + _peer->setProxyLine(_proxyLine); + } disconnect(socket(), &QIODevice::readyRead, this, &CoreAuthHandler::onReadyRead); } @@ -140,24 +169,24 @@ void CoreAuthHandler::onProtocolVersionMismatch(int actual, int expected) "This core needs at least client/core protocol version %1 (got: %2).
" "Please consider upgrading your client.") .arg(expected, actual); - _peer->dispatch(ClientDenied(errorString)); + _peer->dispatch(Protocol::ClientDenied(errorString)); _peer->close(); } bool CoreAuthHandler::checkClientRegistered() { if (!_clientRegistered) { - qWarning() << qPrintable(tr("Client")) << qPrintable(socket()->peerAddress().toString()) + qWarning() << qPrintable(tr("Client")) << qPrintable(hostAddress().toString()) << qPrintable(tr("did not send a registration message before trying to login, rejecting.")); _peer->dispatch( - ClientDenied(tr("Client not initialized!
You need to send a registration message before trying to login."))); + Protocol::ClientDenied(tr("Client not initialized!
You need to send a registration message before trying to login."))); _peer->close(); return false; } return true; } -void CoreAuthHandler::handle(const RegisterClient& msg) +void CoreAuthHandler::handle(const Protocol::RegisterClient& msg) { bool useSsl; if (_legacy) @@ -166,8 +195,8 @@ void CoreAuthHandler::handle(const RegisterClient& msg) useSsl = _connectionFeatures & Protocol::Encryption; if (Quassel::isOptionSet("require-ssl") && !useSsl && !_peer->isLocal()) { - qInfo() << qPrintable(tr("SSL required but non-SSL connection attempt from %1").arg(socket()->peerAddress().toString())); - _peer->dispatch(ClientDenied(tr("SSL is required!
You need to use SSL in order to connect to this core."))); + qInfo() << qPrintable(tr("SSL required but non-SSL connection attempt from %1").arg(hostAddress().toString())); + _peer->dispatch(Protocol::ClientDenied(tr("SSL is required!
You need to use SSL in order to connect to this core."))); _peer->close(); return; } @@ -186,7 +215,7 @@ void CoreAuthHandler::handle(const RegisterClient& msg) } } - _peer->dispatch(ClientRegistered(Quassel::Features{}, configured, backends, authenticators, useSsl)); + _peer->dispatch(Protocol::ClientRegistered(Quassel::Features{}, configured, backends, authenticators, useSsl)); // useSsl is only used for the legacy protocol if (_legacy && useSsl) @@ -195,7 +224,7 @@ void CoreAuthHandler::handle(const RegisterClient& msg) _clientRegistered = true; } -void CoreAuthHandler::handle(const SetupData& msg) +void CoreAuthHandler::handle(const Protocol::SetupData& msg) { if (!checkClientRegistered()) return; @@ -210,41 +239,57 @@ void CoreAuthHandler::handle(const SetupData& msg) QString result = Core::setup(msg.adminUser, msg.adminPassword, msg.backend, msg.setupData, authenticator, msg.authSetupData); if (!result.isEmpty()) - _peer->dispatch(SetupFailed(result)); + _peer->dispatch(Protocol::SetupFailed(result)); else - _peer->dispatch(SetupDone()); + _peer->dispatch(Protocol::SetupDone()); } -void CoreAuthHandler::handle(const Login& msg) +void CoreAuthHandler::handle(const Protocol::Login& msg) { if (!checkClientRegistered()) return; if (!Core::isConfigured()) { - qWarning() << qPrintable(tr("Client")) << qPrintable(socket()->peerAddress().toString()) + qWarning() << qPrintable(tr("Client")) << qPrintable(hostAddress().toString()) << qPrintable(tr("attempted to login before the core was configured, rejecting.")); - _peer->dispatch(ClientDenied( + _peer->dispatch(Protocol::ClientDenied( tr("Attempted to login before core was configured!
The core must be configured before attempting to login."))); return; } // First attempt local auth using the real username and password. // If that fails, move onto the auth provider. - UserId uid = Core::validateUser(msg.user, msg.password); + + // Check to see if the user has the "Database" authenticator configured. + UserId uid = 0; + if (Core::getUserAuthenticator(msg.user) == "Database") { + uid = Core::validateUser(msg.user, msg.password); + } + + // If they did not, *or* if the database login fails, try to use a different authenticator. + // TODO: this logic should likely be moved into Core::authenticateUser in the future. + // Right now a core can only have one authenticator configured; this might be something + // to change in the future. if (uid == 0) { uid = Core::authenticateUser(msg.user, msg.password); } if (uid == 0) { - qInfo() << qPrintable(tr("Invalid login attempt from %1 as \"%2\"").arg(socket()->peerAddress().toString(), msg.user)); - _peer->dispatch(LoginFailed(tr( + qInfo() << qPrintable(tr("Invalid login attempt from %1 as \"%2\"").arg(hostAddress().toString(), msg.user)); + _peer->dispatch(Protocol::LoginFailed(tr( "Invalid username or password!
The username/password combination you supplied could not be found in the database."))); + if (_metricsServer) { + _metricsServer->addLoginAttempt(msg.user, false); + } return; } - _peer->dispatch(LoginSuccess()); + _peer->dispatch(Protocol::LoginSuccess()); + if (_metricsServer) { + _metricsServer->addLoginAttempt(uid, true); + } qInfo() << qPrintable(tr("Client %1 initialized and authenticated successfully as \"%2\" (UserId: %3).") - .arg(socket()->peerAddress().toString(), msg.user, QString::number(uid.toInt()))); + .arg(_peer->address(), msg.user, QString::number(uid.toInt()))); const auto& clientFeatures = _peer->features(); auto unsupported = clientFeatures.toStringList(false); @@ -267,26 +312,35 @@ void CoreAuthHandler::handle(const Login& msg) emit handshakeComplete(_peer, uid); } +QHostAddress CoreAuthHandler::hostAddress() const +{ + if (_useProxyLine) { + return _proxyLine.sourceHost; + } + else if (socket()) { + return socket()->peerAddress(); + } + + return {}; +} + +bool CoreAuthHandler::isLocal() const +{ + return hostAddress() == QHostAddress::LocalHost || + hostAddress() == QHostAddress::LocalHostIPv6; +} + /*** SSL Stuff ***/ void CoreAuthHandler::startSsl() { -#ifdef HAVE_SSL - auto* sslSocket = qobject_cast(socket()); - Q_ASSERT(sslSocket); - qDebug() << qPrintable(tr("Starting encryption for Client:")) << _peer->description(); - connect(sslSocket, selectOverload&>(&QSslSocket::sslErrors), this, &CoreAuthHandler::onSslErrors); - sslSocket->flush(); // ensure that the write cache is flushed before we switch to ssl (bug 682) - sslSocket->startServerEncryption(); -#endif /* HAVE_SSL */ + connect(socket(), selectOverload&>(&QSslSocket::sslErrors), this, &CoreAuthHandler::onSslErrors); + socket()->flush(); // ensure that the write cache is flushed before we switch to ssl (bug 682) + socket()->startServerEncryption(); } -#ifdef HAVE_SSL void CoreAuthHandler::onSslErrors() { - auto* sslSocket = qobject_cast(socket()); - Q_ASSERT(sslSocket); - sslSocket->ignoreSslErrors(); + socket()->ignoreSslErrors(); } -#endif