blob: 363d3baa3ae5b3f353735a4f6dfe7144f24af2a4 (
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
|
/*
* This file is part of smolbote. It's copyrighted by the contributors recorded
* in the version control history of the file, available from its original
* location: git://neueland.iserlohn-fortress.net/smolbote.git
*
* SPDX-License-Identifier: GPL-3.0
*/
#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;
}
|
