ModSecurity/src/utils/ip_tree.h
2016-11-04 14:45:01 -03:00

51 lines
1.3 KiB
C++

/*
* ModSecurity, http://www.modsecurity.org/
* Copyright (c) 2015 Trustwave Holdings, Inc. (http://www.trustwave.com/)
*
* You may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* If any of the files related to licensing are missing or if you have any
* other questions related to licensing please contact Trustwave Holdings, Inc.
* directly using the email address security@modsecurity.org.
*
*/
#include <iostream>
#include <fstream>
#include <string>
#include <functional>
#ifndef SRC_UTILS_IP_TREE_H_
#define SRC_UTILS_IP_TREE_H_
#include "modsecurity/transaction.h"
#include "src/utils/msc_tree.h"
namespace modsecurity {
namespace Utils {
class IpTree {
public:
IpTree();
~IpTree();
bool contains(const std::string &ip);
void postOrderTraversal(TreeNode *node);
bool addFromBuffer(std::istream *ss, std::string *error);
bool addFromBuffer(const std::string& buffer, std::string *error);
bool addFromFile(const std::string& file, std::string *error);
bool addFromUrl(const std::string& url, std::string *error);
private:
TreeRoot *m_tree;
};
} // namespace Utils
} // namespace modsecurity
#endif // SRC_UTILS_IP_TREE_H_