From ff8057d34bb4e9cf12f5d39b5c1b79ac092b387f Mon Sep 17 00:00:00 2001 From: Tannin Date: Sun, 9 Aug 2015 12:40:57 +0200 Subject: - MO now validates session cookie on startup - It also retrieves account status - rewrote how MO decides when to log-in/when to give up on logging in --- src/nxmaccessmanager.cpp | 171 +++++++++++++++++++++++++++++++++++++++-------- 1 file changed, 144 insertions(+), 27 deletions(-) (limited to 'src/nxmaccessmanager.cpp') diff --git a/src/nxmaccessmanager.cpp b/src/nxmaccessmanager.cpp index c99d2d8d..0786a254 100644 --- a/src/nxmaccessmanager.cpp +++ b/src/nxmaccessmanager.cpp @@ -24,6 +24,8 @@ along with Mod Organizer. If not, see . #include "selfupdater.h" #include "persistentcookiejar.h" #include "settings.h" +#include +#include #include #include #include @@ -32,24 +34,29 @@ along with Mod Organizer. If not, see . #include #include #include -#include - -#if QT_VERSION >= QT_VERSION_CHECK(5,0,0) #include -#endif +#include +#include +#include + using namespace MOBase; using namespace MOShared; +// unfortunately Nexus doesn't seem to document these states, all I know is all these listed +// are considered premium (27 should be lifetime premium) +const std::set NXMAccessManager::s_PremiumAccountStates { 4, 6, 13, 27, 31, 32 }; + + NXMAccessManager::NXMAccessManager(QObject *parent, const QString &moVersion) : QNetworkAccessManager(parent) , m_LoginReply(nullptr) , m_ProgressDialog() , m_MOVersion(moVersion) - , m_LoginAttempted(false) { - + m_LoginTimeout.setSingleShot(true); + m_LoginTimeout.setInterval(30000); setCookieJar(new PersistentCookieJar( QDir::fromNativeSeparators(Settings::instance().getCacheDirectory() + "/nexus_cookies.dat"))); } @@ -88,20 +95,96 @@ QNetworkReply *NXMAccessManager::createRequest( } -void NXMAccessManager::showCookies() +void NXMAccessManager::showCookies() const { - QList cookies = cookieJar()->cookiesForUrl(QUrl(ToQString(GameInfo::instance().getNexusPage()) + "/")); - foreach (QNetworkCookie cookie, cookies) { - qDebug("%s - %s", cookie.name().constData(), cookie.value().constData()); + QUrl url(ToQString(GameInfo::instance().getNexusPage()) + "/"); + + for (const QNetworkCookie &cookie : cookieJar()->cookiesForUrl(url)) { + qDebug("%s - %s (expires: %s)", + cookie.name().constData(), cookie.value().constData(), + qPrintable(cookie.expirationDate().toString())); + } +} + + +void NXMAccessManager::startLoginCheck() +{ + if (hasLoginCookies()) { + QNetworkRequest request(ToQString(GameInfo::instance().getNexusPage()) + "/Sessions/?Validate"); + request.setHeader(QNetworkRequest::ContentTypeHeader, "application/x-www-form-urlencoded"); + request.setRawHeader("User-Agent", userAgent().toUtf8()); + + m_LoginReply = get(request); + m_LoginTimeout.start(); + m_LoginState = LOGIN_CHECKING; + connect(m_LoginReply, SIGNAL(finished()), this, SLOT(loginChecked())); + connect(m_LoginReply, SIGNAL(error(QNetworkReply::NetworkError)), + this, SLOT(loginError(QNetworkReply::NetworkError))); } } +void NXMAccessManager::retrieveCredentials() +{ + QNetworkRequest request(ToQString(GameInfo::instance().getNexusPage()) + + QString("/Core/Libs/Flamework/Entities/User?GetCredentials&game_id=%1" + ).arg(GameInfo::instance().getNexusGameID())); + request.setHeader(QNetworkRequest::ContentTypeHeader, "application/x-www-form-urlencoded"); + request.setRawHeader("User-Agent", userAgent().toUtf8()); + + QNetworkReply *reply = get(request); + QTimer timeout; + connect(&timeout, &QTimer::timeout, [reply] () { + reply->deleteLater(); + }); + timeout.start(); + + connect(reply, &QNetworkReply::finished, [reply, this] () { + QJsonDocument jdoc = QJsonDocument::fromJson(reply->readAll()); + QJsonArray credentialsData = jdoc.array(); + emit credentialsReceived(credentialsData.at(2).toString(), + s_PremiumAccountStates.find(credentialsData.at(1).toInt()) + != s_PremiumAccountStates.end()); + reply->deleteLater(); + }); + + connect(reply, static_cast(&QNetworkReply::error), + [=] (QNetworkReply::NetworkError) { + qDebug("failed to retrieve account credentials: %s", qPrintable(reply->errorString())); + reply->deleteLater(); + }); +} + + bool NXMAccessManager::loggedIn() const { - return hasLoginCookies(); + if (m_LoginState == LOGIN_CHECKING) { + QProgressDialog progress; + progress.setLabelText(tr("Verifying Nexus login")); + progress.show(); + while (m_LoginState == LOGIN_CHECKING) { + QCoreApplication::processEvents(); + QThread::msleep(100); + } + progress.hide(); + } + + return m_LoginState == LOGIN_VALID; +} + + +void NXMAccessManager::refuseLogin() +{ + m_LoginState = LOGIN_REFUSED; +} + + +bool NXMAccessManager::loginAttempted() const +{ + return m_LoginState != LOGIN_NOT_CHECKED; } + bool NXMAccessManager::loginWaiting() const { return m_LoginReply != nullptr; @@ -114,19 +197,29 @@ void NXMAccessManager::login(const QString &username, const QString &password) return; } - if (hasLoginCookies()) { + if (m_LoginState == LOGIN_VALID) { emit loginSuccessful(false); return; } - m_LoginAttempted = true; - m_Username = username; m_Password = password; pageLogin(); } +QString NXMAccessManager::userAgent(const QString &subModule) const +{ + QStringList comments; + comments << "compatible to Nexus Client v" + m_NMMVersion; + if (!subModule.isEmpty()) { + comments << "module: " + subModule; + } + + return QString("Mod Organizer v%1 (%2)").arg(m_MOVersion, comments.join("; ")); +} + + void NXMAccessManager::pageLogin() { QString requestString = (ToQString(GameInfo::instance().getNexusPage()) + "/Sessions/?Login&uri=%1") @@ -148,8 +241,7 @@ void NXMAccessManager::pageLogin() postDataQuery = postData.encodedQuery(); #endif - QString userAgent = QString("Mod Organizer v%1 (compatible to Nexus Client v%2)").arg(m_MOVersion).arg(m_NMMVersion); - request.setRawHeader("User-Agent", userAgent.toUtf8()); + request.setRawHeader("User-Agent", userAgent().toUtf8()); m_ProgressDialog.setLabelText(tr("Logging into Nexus")); QList buttons = m_ProgressDialog.findChildren(); @@ -159,6 +251,7 @@ void NXMAccessManager::pageLogin() m_LoginReply = post(request, postDataQuery); m_LoginTimeout.start(); + m_LoginState = LOGIN_CHECKING; connect(m_LoginReply, SIGNAL(finished()), this, SLOT(loginFinished())); connect(m_LoginReply, SIGNAL(error(QNetworkReply::NetworkError)), this, SLOT(loginError(QNetworkReply::NetworkError))); } @@ -166,37 +259,41 @@ void NXMAccessManager::pageLogin() void NXMAccessManager::loginTimeout() { - emit loginFailed(tr("timeout")); m_LoginReply->deleteLater(); m_LoginReply = nullptr; m_LoginAttempted = false; // this usually means we might have success later - m_LoginTimeout.stop(); m_Username.clear(); m_Password.clear(); + m_LoginState = LOGIN_NOT_VALID; + + emit loginFailed(tr("timeout")); } void NXMAccessManager::loginError(QNetworkReply::NetworkError) { + qDebug("login error"); m_ProgressDialog.hide(); - m_LoginTimeout.stop(); + m_Username.clear(); + m_Password.clear(); + m_LoginState = LOGIN_NOT_VALID; + if (m_LoginReply != nullptr) { - emit loginFailed(m_LoginReply->errorString()); m_LoginReply->deleteLater(); m_LoginReply = nullptr; + emit loginFailed(m_LoginReply->errorString()); } else { emit loginFailed(tr("Unknown error")); } - m_Username.clear(); - m_Password.clear(); } bool NXMAccessManager::hasLoginCookies() const { bool sidCookie = false; - QList cookies = cookieJar()->cookiesForUrl(QUrl(ToQString(GameInfo::instance().getNexusPage()) + "/")); - foreach (QNetworkCookie cookie, cookies) { + QUrl url(ToQString(GameInfo::instance().getNexusPage()) + "/"); + QList cookies = cookieJar()->cookiesForUrl(url); + for (const QNetworkCookie &cookie : cookies) { if (cookie.name() == "sid") { sidCookie = true; } @@ -208,15 +305,35 @@ bool NXMAccessManager::hasLoginCookies() const void NXMAccessManager::loginFinished() { m_ProgressDialog.hide(); + + m_LoginReply->deleteLater(); + m_LoginReply = nullptr; + m_Username.clear(); + m_Password.clear(); + if (hasLoginCookies()) { + m_LoginState = LOGIN_VALID; + retrieveCredentials(); emit loginSuccessful(true); } else { + m_LoginState = LOGIN_NOT_VALID; emit loginFailed(tr("Please check your password")); } +} + - m_LoginTimeout.stop(); +void NXMAccessManager::loginChecked() +{ + QNetworkReply *reply = static_cast(sender()); + QByteArray data = reply->readAll(); + m_LoginState = data == "null" ? LOGIN_NOT_VALID + : LOGIN_VALID; + if (m_LoginState == LOGIN_VALID) { + retrieveCredentials(); + } else { + qDebug("cookies seem to be invalid"); + } m_LoginReply->deleteLater(); m_LoginReply = nullptr; - m_Username.clear(); - m_Password.clear(); } + -- cgit v1.3.1