mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 11:22:12 +00:00
38 lines
1.2 KiB
C++
38 lines
1.2 KiB
C++
#pragma once
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <mutex>
|
|
#include <unordered_set>
|
|
|
|
|
|
namespace Poco { class URI; }
|
|
namespace Poco { namespace Util { class AbstractConfiguration; } }
|
|
|
|
namespace DB
|
|
{
|
|
class RemoteHostFilter
|
|
{
|
|
/**
|
|
* This class checks if URL is allowed.
|
|
* If primary_hosts and regexp_hosts are empty all urls are allowed.
|
|
*/
|
|
public:
|
|
void checkURL(const Poco::URI & uri) const; /// If URL not allowed in config.xml throw UNACCEPTABLE_URL Exception
|
|
|
|
void setValuesFromConfig(const Poco::Util::AbstractConfiguration & config);
|
|
|
|
void checkHostAndPort(const std::string & host, const std::string & port) const; /// Does the same as checkURL, but for host and port.
|
|
|
|
private:
|
|
std::atomic_bool is_initialized = false;
|
|
|
|
mutable std::mutex hosts_mutex;
|
|
std::unordered_set<std::string> primary_hosts; /// Allowed primary (<host>) URL from config.xml
|
|
std::vector<std::string> regexp_hosts; /// Allowed regexp (<hots_regexp>) URL from config.xml
|
|
|
|
/// Checks if the primary_hosts and regexp_hosts contain str. If primary_hosts and regexp_hosts are empty return true.
|
|
bool checkForDirectEntry(const std::string & str) const;
|
|
};
|
|
}
|