aboutsummaryrefslogtreecommitdiff
path: root/lib/web/urlfilter/filterdomain.cpp
blob: 99cc71c6f7641b759eb294a4cfb034054ac1bf8d (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
#include "filterdomain.h"
#include <QVector>

bool isMatchingDomain(const QString &domain, const QString &filter)
{
    // domain and filter are the same
    if(domain == filter) {
        return true;
    }

    // domain can't be matched by filter if it doesn't end with filter
    // ex. example2.com isn't matched by example.com
    if(!domain.endsWith(filter)) {
        return false;
    }

    // match with subdomains
    // ex. subdomain.example.com is matched by example.com
    int index = domain.indexOf(filter);

    // match if (domain ends with filter) && (filter has been found) and (character before filter is '.')
    return index > 0 && domain[index - 1] == QLatin1Char('.');
}

FilterDomain::FilterDomain(QObject *parent)
    : QObject(parent)
{
}

void FilterDomain::addDomain(const QString &domain, bool isException)
{
    if(domain.isEmpty())
        return;

    if(!isException)
        m_allowedOnDomains.append(domain);
    else
        m_blockedOnDomains.append(domain);
}

void FilterDomain::addRule(FilterRule_ptr &rule)
{
    Q_ASSERT(rule);
    m_rules.emplace_back(std::move(rule));
}

bool FilterDomain::hasMatch(const QString &host) const
{
    for(const auto &domain : qAsConst(m_blockedOnDomains)) {
        if(isMatchingDomain(host, domain))
            return false;
    }

    for(const auto &domain : qAsConst(m_allowedOnDomains)) {
        if(isMatchingDomain(host, domain))
            return true;
    }

    return false;
}

void FilterDomain::process(QWebEngineUrlRequestInfo &info) const
{
    for(const FilterRule_ptr &rule : m_rules) {
        if(rule->matchRequestUrl(info.requestUrl().toString(), info.resourceType()))
            rule->process(info);
    }
}