1 files changed, 58 insertions, 0 deletions
diff --git a/plugins/HostlistFilter/filterlist.cpp b/plugins/HostlistFilter/filterlist.cpp
new file mode 100644
index 0000000..a0fd414
--- /dev/null
+++ b/plugins/HostlistFilter/filterlist.cpp
@@ -0,0 +1,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;
+}