/* * Copyright (C) by Olivier Goffart * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program 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 General Public License * for more details. */ #include #include #include #include #include "account.h" #include "creds/oauth.h" #include #include #include "theme.h" #include "networkjobs.h" namespace OCC { Q_LOGGING_CATEGORY(lcOauth, "sync.credentials.oauth", QtInfoMsg) OAuth::~OAuth() { } static void httpReplyAndClose(QTcpSocket *socket, const char *code, const char *html, const char *moreHeaders = nullptr) { if (!socket) return; // socket can have been deleted if the browser was closed socket->write("HTTP/1.1 "); socket->write(code); socket->write("\r\nContent-Type: text/html\r\nConnection: close\r\nContent-Length: "); socket->write(QByteArray::number(qstrlen(html))); if (moreHeaders) { socket->write("\r\n"); socket->write(moreHeaders); } socket->write("\r\n\r\n"); socket->write(html); socket->disconnectFromHost(); // We don't want that deleting the server too early prevent queued data to be sent on this socket. // The socket will be deleted after disconnection because disconnected is connected to deleteLater socket->setParent(nullptr); } void OAuth::start() { // Listen on the socket to get a port which will be used in the redirect_uri if (!_server.listen(QHostAddress::LocalHost)) { emit result(NotSupported, QString()); return; } if (!openBrowser()) return; QObject::connect(&_server, &QTcpServer::newConnection, this, [this] { while (QPointer socket = _server.nextPendingConnection()) { QObject::connect(socket.data(), &QTcpSocket::disconnected, socket.data(), &QTcpSocket::deleteLater); QObject::connect(socket.data(), &QIODevice::readyRead, this, [this, socket] { QByteArray peek = socket->peek(qMin(socket->bytesAvailable(), 4000LL)); //The code should always be within the first 4K if (peek.indexOf('\n') < 0) return; // wait until we find a \n QRegExp rx("^GET /\\?code=([a-zA-Z0-9]+)[& ]"); // Match a /?code=... URL if (rx.indexIn(peek) != 0) { httpReplyAndClose(socket, "404 Not Found", "404 Not Found

404 Not Found

"); return; } QString code = rx.cap(1); // The 'code' is the first capture of the regexp QUrl requestToken = Utility::concatUrlPath(_account->url().toString(), QLatin1String("/index.php/apps/oauth2/api/v1/token")); QNetworkRequest req; req.setHeader(QNetworkRequest::ContentTypeHeader, "application/x-www-form-urlencoded"); QString basicAuth = QString("%1:%2").arg( Theme::instance()->oauthClientId(), Theme::instance()->oauthClientSecret()); req.setRawHeader("Authorization", "Basic " + basicAuth.toUtf8().toBase64()); auto requestBody = new QBuffer; QUrlQuery arguments(QString( "grant_type=authorization_code&code=%1&redirect_uri=http://localhost:%2") .arg(code, QString::number(_server.serverPort()))); requestBody->setData(arguments.query(QUrl::FullyEncoded).toLatin1()); auto job = _account->sendRequest("POST", requestToken, req, requestBody); job->setTimeout(qMin(30 * 1000ll, job->timeoutMsec())); QObject::connect(job, &SimpleNetworkJob::finishedSignal, this, [this, socket](QNetworkReply *reply) { auto jsonData = reply->readAll(); QJsonParseError jsonParseError; QJsonObject json = QJsonDocument::fromJson(jsonData, &jsonParseError).object(); QString accessToken = json["access_token"].toString(); QString refreshToken = json["refresh_token"].toString(); QString user = json["user_id"].toString(); QUrl messageUrl = json["message_url"].toString(); if (reply->error() != QNetworkReply::NoError || jsonParseError.error != QJsonParseError::NoError || json.isEmpty() || refreshToken.isEmpty() || accessToken.isEmpty() || json["token_type"].toString() != QLatin1String("Bearer")) { QString errorReason; QString errorFromJson = json["error"].toString(); if (!errorFromJson.isEmpty()) { errorReason = tr("Error returned from the server: %1") .arg(errorFromJson.toHtmlEscaped()); } else if (reply->error() != QNetworkReply::NoError) { errorReason = tr("There was an error accessing the 'token' endpoint:
%1") .arg(reply->errorString().toHtmlEscaped()); } else if (jsonParseError.error != QJsonParseError::NoError) { errorReason = tr("Could not parse the JSON returned from the server:
%1") .arg(jsonParseError.errorString()); } else { errorReason = tr("The reply from the server did not contain all expected fields"); } qCWarning(lcOauth) << "Error when getting the accessToken" << json << errorReason; httpReplyAndClose(socket, "500 Internal Server Error", tr("

Login Error

").arg(errorReason).toUtf8().constData()); emit result(Error); return; } if (!_expectedUser.isNull() && user != _expectedUser) { // Connected with the wrong user QString message = tr("

Wrong user

" "

You logged-in with user %1, but must login with user %2.
" "Please log out of %3 in another tab, then click here " "and log in as user %2

") .arg(user, _expectedUser, Theme::instance()->appNameGUI(), authorisationLink().toString(QUrl::FullyEncoded)); httpReplyAndClose(socket, "200 OK", message.toUtf8().constData()); // We are still listening on the socket so we will get the new connection return; } const char *loginSuccessfullHtml = "

Login Successful

You can close this window.

"; if (messageUrl.isValid()) { httpReplyAndClose(socket, "303 See Other", loginSuccessfullHtml, QByteArray("Location: " + messageUrl.toEncoded()).constData()); } else { httpReplyAndClose(socket, "200 OK", loginSuccessfullHtml); } emit result(LoggedIn, user, accessToken, refreshToken); }); }); } }); QTimer::singleShot(5 * 60 * 1000, this, [this] { result(Error); }); } QUrl OAuth::authorisationLink() const { Q_ASSERT(_server.isListening()); QUrlQuery query; query.setQueryItems({ { QLatin1String("response_type"), QLatin1String("code") }, { QLatin1String("client_id"), Theme::instance()->oauthClientId() }, { QLatin1String("redirect_uri"), QLatin1String("http://localhost:") + QString::number(_server.serverPort()) } }); if (!_expectedUser.isNull()) query.addQueryItem("user", _expectedUser); QUrl url = Utility::concatUrlPath(_account->url(), QLatin1String("/index.php/apps/oauth2/authorize"), query); return url; } bool OAuth::openBrowser() { if (!QDesktopServices::openUrl(authorisationLink())) { // We cannot open the browser, then we claim we don't support OAuth. emit result(NotSupported, QString()); return false; } return true; } } // namespace OCC