1
mirror of https://invent.kde.org/network/falkon.git synced 2024-11-13 10:32:11 +01:00
falkonOfficial/src/lib/cookies/cookiejar.cpp

258 lines
6.5 KiB
C++
Raw Normal View History

2011-03-03 18:29:20 +01:00
/* ============================================================
* QupZilla - WebKit based browser
* Copyright (C) 2010-2013 David Rosca <nowrep@gmail.com>
2011-03-03 18:29:20 +01:00
*
* 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 3 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.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
* ============================================================ */
2011-03-02 16:57:41 +01:00
#include "cookiejar.h"
#include "qupzilla.h"
#include "mainapplication.h"
#include "settings.h"
#include <QNetworkCookie>
#include <QWebSettings>
#include <QDateTime>
#include <QDebug>
#include <QWebPage> // QTWEBKIT_VERSION_CHECK macro
//#define COOKIE_DEBUG
2011-03-02 16:57:41 +01:00
#if QTWEBKIT_TO_2_3
static bool blockThirdParty(QString string, QString domain)
{
if (string.isEmpty()) {
// Some cookies have empty domain() ... bug?
return false;
}
if (string.startsWith(QLatin1String("www."))) {
string = string.mid(3);
}
if (domain.startsWith(QLatin1String("www."))) {
domain = domain.mid(4);
}
return !domain.endsWith(string);
}
#endif
static bool matchDomain(const QString &domain, const QString &filter)
{
// According to RFC 6265
if (domain == filter) {
return true;
}
if (!domain.endsWith(filter)) {
return false;
}
int index = domain.indexOf(filter);
return (index > 0 && filter[0] == QLatin1Char('.'));
}
static int listContainsDomain(const QStringList &list, const QString &domain)
{
if (domain.isEmpty()) {
return -1;
}
foreach(const QString & d, list) {
if (matchDomain(domain, d)) {
return 1;
}
}
return 0;
}
CookieJar::CookieJar(QupZilla* mainClass, QObject* parent)
: QNetworkCookieJar(parent)
, p_QupZilla(mainClass)
2011-03-02 16:57:41 +01:00
{
m_activeProfil = mApp->currentProfilePath();
loadSettings();
2011-03-02 16:57:41 +01:00
}
void CookieJar::loadSettings()
{
Settings settings;
settings.beginGroup("Cookie-Settings");
m_allowCookies = settings.value("allowCookies", true).toBool();
m_blockThirdParty = settings.value("allowCookiesFromVisitedDomainOnly", false).toBool();
m_filterTrackingCookie = settings.value("filterTrackingCookie", false).toBool();
2011-03-02 16:57:41 +01:00
m_deleteOnClose = settings.value("deleteCookiesOnClose", false).toBool();
m_whitelist = settings.value("whitelist", QStringList()).toStringList();
m_blacklist = settings.value("blacklist", QStringList()).toStringList();
settings.endGroup();
#if QTWEBKIT_FROM_2_3
mApp->webSettings()->setThirdPartyCookiePolicy(m_blockThirdParty ?
QWebSettings::AlwaysBlockThirdPartyCookies :
QWebSettings::AlwaysAllowThirdPartyCookies);
#endif
2011-03-02 16:57:41 +01:00
}
void CookieJar::setAllowCookies(bool allow)
{
m_allowCookies = allow;
}
bool CookieJar::rejectCookie(const QString &domain, const QNetworkCookie &cookie) const
2011-03-02 16:57:41 +01:00
{
Q_UNUSED(domain)
const QString &cookieDomain = cookie.domain();
2011-03-02 16:57:41 +01:00
if (!m_allowCookies) {
int result = listContainsDomain(m_whitelist, cookieDomain);
if (result != 1) {
2011-03-02 16:57:41 +01:00
#ifdef COOKIE_DEBUG
qDebug() << "not in whitelist" << cookie;
2011-03-02 16:57:41 +01:00
#endif
return true;
2011-03-02 16:57:41 +01:00
}
}
if (m_allowCookies) {
int result = listContainsDomain(m_blacklist, cookieDomain);
if (result == 1) {
2011-03-02 16:57:41 +01:00
#ifdef COOKIE_DEBUG
qDebug() << "found in blacklist" << cookie;
2011-03-02 16:57:41 +01:00
#endif
return true;
}
}
// This feature is now natively in QtWebKit 2.3
#if QTWEBKIT_TO_2_3
if (m_blockThirdParty) {
bool result = blockThirdParty(cookieDomain, domain);
if (result) {
#ifdef COOKIE_DEBUG
qDebug() << "purged for domain mismatch" << cookie << cookieDomain << domain;
#endif
return true;
}
}
#endif
if (m_filterTrackingCookie && cookie.name().startsWith("__utm")) {
#ifdef COOKIE_DEBUG
qDebug() << "purged as tracking " << cookie;
#endif
return true;
}
return false;
}
bool CookieJar::setCookiesFromUrl(const QList<QNetworkCookie> &cookieList, const QUrl &url)
{
QList<QNetworkCookie> newList = cookieList;
foreach(const QNetworkCookie & cookie, newList) {
if (rejectCookie(url.host(), cookie)) {
newList.removeOne(cookie);
2011-03-02 16:57:41 +01:00
continue;
}
}
2011-03-02 16:57:41 +01:00
return QNetworkCookieJar::setCookiesFromUrl(newList, url);
}
void CookieJar::saveCookies()
{
if (mApp->isPrivateSession()) {
2011-03-02 16:57:41 +01:00
return;
}
2011-03-02 16:57:41 +01:00
QList<QNetworkCookie> allCookies;
// If we are deleting cookies on close, let's just save empty cookie list
if (!m_deleteOnClose) {
allCookies = getAllCookies();
}
QFile file(m_activeProfil + "cookies.dat");
2011-03-02 16:57:41 +01:00
file.open(QIODevice::WriteOnly);
QDataStream stream(&file);
int count = allCookies.count();
stream << count;
for (int i = 0; i < count; i++) {
const QNetworkCookie &cookie = allCookies.at(i);
if (cookie.isSessionCookie()) {
continue;
}
stream << cookie.toRawForm();
2011-03-02 16:57:41 +01:00
}
file.close();
}
void CookieJar::restoreCookies()
{
if (!QFile::exists(m_activeProfil + "cookies.dat") || mApp->isPrivateSession()) {
2011-03-02 16:57:41 +01:00
return;
}
2011-03-02 16:57:41 +01:00
QDateTime now = QDateTime::currentDateTime();
QList<QNetworkCookie> restoredCookies;
QFile file(m_activeProfil + "cookies.dat");
2011-03-02 16:57:41 +01:00
file.open(QIODevice::ReadOnly);
QDataStream stream(&file);
int count;
stream >> count;
for (int i = 0; i < count; i++) {
2011-03-02 16:57:41 +01:00
QByteArray rawForm;
stream >> rawForm;
const QList<QNetworkCookie> &cookieList = QNetworkCookie::parseCookies(rawForm);
if (cookieList.isEmpty()) {
2011-03-02 16:57:41 +01:00
continue;
}
const QNetworkCookie &cookie = cookieList.at(0);
if (cookie.expirationDate() < now) {
2011-03-02 16:57:41 +01:00
continue;
}
restoredCookies.append(cookie);
2011-03-02 16:57:41 +01:00
}
file.close();
setAllCookies(restoredCookies);
}
void CookieJar::clearCookies()
{
setAllCookies(QList<QNetworkCookie>());
}
2011-03-02 16:57:41 +01:00
QList<QNetworkCookie> CookieJar::getAllCookies()
{
return QNetworkCookieJar::allCookies();
}
void CookieJar::setAllCookies(const QList<QNetworkCookie> &cookieList)
{
QNetworkCookieJar::setAllCookies(cookieList);
}