File indexing completed on 2024-05-12 04:57:50

0001 /* ============================================================
0002 * Falkon - Qt web browser
0003 * Copyright (C) 2013-2014  David Rosca <nowrep@gmail.com>
0004 *
0005 * This program is free software: you can redistribute it and/or modify
0006 * it under the terms of the GNU General Public License as published by
0007 * the Free Software Foundation, either version 3 of the License, or
0008 * (at your option) any later version.
0009 *
0010 * This program is distributed in the hope that it will be useful,
0011 * but WITHOUT ANY WARRANTY; without even the implied warranty of
0012 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
0013 * GNU General Public License for more details.
0014 *
0015 * You should have received a copy of the GNU General Public License
0016 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
0017 * ============================================================ */
0018 #ifndef ADBLOCKSEARCHTREE_H
0019 #define ADBLOCKSEARCHTREE_H
0020 
0021 #include <QChar>
0022 #include <QHash>
0023 
0024 #include "qzcommon.h"
0025 
0026 class QWebEngineUrlRequestInfo;
0027 
0028 class AdBlockRule;
0029 
0030 class FALKON_EXPORT AdBlockSearchTree
0031 {
0032 public:
0033     explicit AdBlockSearchTree();
0034     ~AdBlockSearchTree();
0035 
0036     void clear();
0037 
0038     bool add(const AdBlockRule* rule);
0039     const AdBlockRule* find(const QWebEngineUrlRequestInfo &request, const QString &domain, const QString &urlString) const;
0040 
0041 private:
0042     struct Node {
0043         QChar c;
0044         const AdBlockRule* rule;
0045         QHash<QChar, Node*> children;
0046 
0047         Node() : c(0) , rule(nullptr) { }
0048     };
0049 
0050     const AdBlockRule* prefixSearch(const QWebEngineUrlRequestInfo &request, const QString &domain,
0051                                     const QString &urlString, const QChar* string, int len) const;
0052 
0053     void deleteNode(Node* node);
0054 
0055     Node* m_root;
0056 };
0057 
0058 #endif // ADBLOCKSEARCHTREE_H