blob: ff652cfc323454d31c055dfd99d1c97ac3f19cee (
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
75
76
77
78
79
|
/*
* 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: https://neueland.iserlohn-fortress.net/gitea/aqua/smolbote
*
* SPDX-License-Identifier: GPL-3.0
*/
#include "hostlist.h"
#include <QIODevice>
#include <QTextStream>
#include <QDebug>
HostList::HostList(QIODevice *device)
{
Q_ASSERT(device->isOpen());
QTextStream list(device);
while (!list.atEnd()) {
parseLine(list.readLine());
}
qDebug() << m_metadata;
}
QString HostList::metadata(const QString& key) const
{
return m_metadata.value(key);
}
int HostList::ruleCount() const
{
return rules.size();
}
std::pair<UrlFilter::MatchResult, QString> HostList::match(const QUrl& firstParty, const QUrl& requestUrl, QWebEngineUrlRequestInfo::ResourceType type) const
{
Q_UNUSED(firstParty);
Q_UNUSED(type);
const QString domain = requestUrl.host();
const uint domainHash = qHash(domain);
for(const Rule &r : rules) {
if(r.domainHash == domainHash)
return std::make_pair(r.action, r.redirect);
}
return std::make_pair(UrlFilter::NotMatched, QString());
}
void HostList::parseLine(const QString& line)
{
// check comment
if(line.startsWith(QLatin1Literal("#")))
return;
QString parsedLine = line.trimmed();
// malformed rule
if(!parsedLine.contains(QLatin1Literal(" ")))
return;
const QStringList parts = parsedLine.split(QLatin1Literal(" "));
const QString &redirect = parts.at(0);
const auto action = (redirect == QLatin1Literal("0.0.0.0")) ? UrlFilter::Block : UrlFilter::Redirect;
for(int i = 1; i < parts.size(); i++) {
const QString &domain = parts.at(i);
Rule r;
r.action = action;
r.domainHash = qHash(domain);
if(action == UrlFilter::Redirect)
r.redirect = redirect;
rules.emplace_back(std::move(r));
}
}
|