aboutsummaryrefslogtreecommitdiff
path: root/plugins/HostlistFilter/filterlist.cpp
blob: a0fd414b4e085651b709486bcb95318f4a072fb4 (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
/*
 * 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://library.iserlohn-fortress.net/aqua/smolbote.git
 *
 * SPDX-License-Identifier: GPL-3.0
 */

#include "filterlist.h"
#include <QIODevice>
#include <QTextStream>

using namespace Hostlist;

#ifdef FUZZER
extern "C" int LLVMFuzzerTestOneInput(const char *Data, long long Size)
{
    Filterlist::parseRule(QString::fromLatin1(Data, Size));
    return 0;
}
#endif

std::map<Filterlist::DomainHash, Filterlist::Rule> Filterlist::parseRule(const QString &line)
{
    if(line.isEmpty() || line.at(0) == '#') {
        return {};
    }

    auto parts = line.trimmed().split(' ');
    if(parts.size() < 2) {
        return {};
    }

    const auto redirect = (parts[0] == "0.0.0.0") ? QString() : parts[0];

    std::map<DomainHash, Rule> r;
    for(int i = 1; i < parts.size(); ++i) {
        r.emplace(qHash(parts[i], 0), Filterlist::Rule{ parts[i], redirect });
    }
    return r;
}

bool Filterlist::load(QIODevice &from)
{
    if(!from.isReadable() || !from.isTextModeEnabled()) {
        return false;
    }

    while(from.bytesAvailable() > 0) {
        const auto line = from.readLine(512).trimmed();
        auto r = parseRule(line);
        if(!r.empty()) {
            qDebug("merging in %lu rules", r.size());
            rules.merge(r);
        }
    }
    return true;
}