aboutsummaryrefslogtreecommitdiff
path: root/src/webengine/blockerrule.cpp
blob: 314685448b98e411fdebe52daa672c7db24addbe (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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
/** LICENSE ********************************************************************
 **
 ** 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 "blockerrule.h"

/* AdBlock filter reference
 * https://adblockplus.org/en/filters
 * https://adblockplus.org/en/filter-cheatsheet
 */

BlockerRule::BlockerRule(QString rule, QObject *parent) :
    QObject(parent)
{
    QString pattern = rule;

    // Empty rule or comment
    if(pattern.trimmed().isEmpty() || pattern.startsWith("!")) {
        m_type = RuleType::Invalid;
        return;
    }

    // Exception
    if(pattern.startsWith("@@")) {
        m_exception = true;
        pattern = pattern.remove(0, 2);
    }

    // Domain
    if(pattern.startsWith("||")) {
        m_type = RuleType::DomainMatch;
        m_domain = pattern.mid(2, pattern.indexOf("^")-2);
        return;
    }

    // Regular expression
    if(rule.startsWith("/") && rule.endsWith("/")) {
        m_type = RuleType::RegularExpressionMatch;
        ruleExpression.setPattern(pattern);
        return;
    }

    // Regular rule
    ruleExpression.setPattern(fromWildcardMatch(pattern));
    m_type = RuleType::WildcardMatch;
}

QString BlockerRule::pattern() const
{
    return ruleExpression.pattern();
}

QString BlockerRule::domain() const
{
    return m_domain;
}

bool BlockerRule::match(const QUrl &url)
{
    return ruleExpression.match(url.toString()).hasMatch();
}

bool BlockerRule::isValid()
{
    if(m_type == RuleType::Invalid) {
        return false;
    } else {
        return true;
    }
}
bool BlockerRule::isException()
{
    return m_exception;
}

QString BlockerRule::fromWildcardMatch(const QString &pattern)
{
    QString parsed;

    for(int i=0; i<pattern.length(); i++) {
        const QChar c = pattern.at(i);
        switch (c.toLatin1()) {
        case '*':
            parsed.append(".*");
            break;
        case '^':
            parsed.append("(?:[^\\w\\d\\_\\-\\.\\%]|$)");
            break;
        case '|':
            if(i == 0) {
                // beginning of string
                parsed.append('^');
            } else {
                // end of string
                parsed.append('$');
            }
            break;
        default:
            parsed.append(c);
            break;
        }
    }

    return parsed;
}