blob: cbf1c868e45f70ca99f5d722815b24972d740831 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
/*******************************************************************************
**
** smolbote: yet another qute browser
** Copyright (C) 2017 Xian Nox
**
** 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/>.
**
******************************************************************************/
#include "urlinterceptor.h"
#include <QFile>
#include <QTextStream>
UrlRequestInterceptor::UrlRequestInterceptor(const QString &path, QObject *parent) :
QWebEngineUrlRequestInterceptor(parent)
{
#ifdef QT_DEBUG
qDebug("Reading request blocklist: %s", qUtf8Printable(path));
#endif
int n_rules = 0;
QFile filter(path);
if(filter.open(QIODevice::ReadOnly | QIODevice::Text)) {
QTextStream out(&filter);
while(!out.atEnd()) {
const QString &line = out.readLine();
if(!line.startsWith('!')) {
FilterRule r(line);
// check if valid
m_rules.push_back(std::move(r));
++n_rules;
}
}
filter.close();
}
qDebug("Added %i rules", n_rules);
}
void UrlRequestInterceptor::interceptRequest(QWebEngineUrlRequestInfo &info)
{
for(const FilterRule &test : m_rules) {
const QUrl &url = info.requestUrl();
if(test.shouldBlock(url)) {
info.block(true);
#ifdef QT_DEBUG
qDebug("blocked [%s] %s", qUtf8Printable(info.firstPartyUrl().toString()), qUtf8Printable(url.toString()));
#endif
return;
}
}
}
bool shouldBlock(const QUrl &url, const QString &test)
{
if(url.toString().contains(test)) {
return true;
}
return false;
}
|
