mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 19:32:07 +00:00
120 lines
3.2 KiB
C++
120 lines
3.2 KiB
C++
#include <Interpreters/SecurityManager.h>
|
|
|
|
#include <Poco/Net/IPAddress.h>
|
|
#include <Poco/Util/AbstractConfiguration.h>
|
|
#include <Poco/String.h>
|
|
|
|
#include <Common/Exception.h>
|
|
#include <IO/HexWriteBuffer.h>
|
|
#include <IO/WriteBufferFromString.h>
|
|
#include <IO/WriteHelpers.h>
|
|
|
|
#include <openssl/sha.h>
|
|
|
|
#include <common/logger_useful.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int DNS_ERROR;
|
|
extern const int UNKNOWN_ADDRESS_PATTERN_TYPE;
|
|
extern const int UNKNOWN_USER;
|
|
extern const int REQUIRED_PASSWORD;
|
|
extern const int WRONG_PASSWORD;
|
|
extern const int IP_ADDRESS_NOT_ALLOWED;
|
|
extern const int BAD_ARGUMENTS;
|
|
}
|
|
|
|
using UserPtr = SecurityManager::UserPtr;
|
|
|
|
void SecurityManager::loadFromConfig(Poco::Util::AbstractConfiguration & config)
|
|
{
|
|
Container new_users;
|
|
|
|
Poco::Util::AbstractConfiguration::Keys config_keys;
|
|
config.keys("users", config_keys);
|
|
|
|
for (const std::string & key : config_keys)
|
|
{
|
|
auto user = std::make_shared<const User>(key, "users." + key, config);
|
|
new_users.emplace(key, std::move(user));
|
|
}
|
|
|
|
users = std::move(new_users);
|
|
}
|
|
|
|
UserPtr SecurityManager::authorizeAndGetUser(
|
|
const String & user_name,
|
|
const String & password,
|
|
const Poco::Net::IPAddress & address) const
|
|
{
|
|
auto it = users.find(user_name);
|
|
|
|
if (users.end() == it)
|
|
throw Exception("Unknown user " + user_name, ErrorCodes::UNKNOWN_USER);
|
|
|
|
if (!it->second->addresses.contains(address))
|
|
throw Exception("User " + user_name + " is not allowed to connect from address " + address.toString(), ErrorCodes::IP_ADDRESS_NOT_ALLOWED);
|
|
|
|
auto on_wrong_password = [&]()
|
|
{
|
|
if (password.empty())
|
|
throw Exception("Password required for user " + user_name, ErrorCodes::REQUIRED_PASSWORD);
|
|
else
|
|
throw Exception("Wrong password for user " + user_name, ErrorCodes::WRONG_PASSWORD);
|
|
};
|
|
|
|
if (!it->second->password_sha256_hex.empty())
|
|
{
|
|
unsigned char hash[32];
|
|
|
|
SHA256_CTX ctx;
|
|
SHA256_Init(&ctx);
|
|
SHA256_Update(&ctx, reinterpret_cast<const unsigned char *>(password.data()), password.size());
|
|
SHA256_Final(hash, &ctx);
|
|
|
|
String hash_hex;
|
|
{
|
|
WriteBufferFromString buf(hash_hex);
|
|
HexWriteBuffer hex_buf(buf);
|
|
hex_buf.write(reinterpret_cast<const char *>(hash), sizeof(hash));
|
|
}
|
|
|
|
Poco::toLowerInPlace(hash_hex);
|
|
|
|
if (hash_hex != it->second->password_sha256_hex)
|
|
on_wrong_password();
|
|
}
|
|
else if (password != it->second->password)
|
|
{
|
|
on_wrong_password();
|
|
}
|
|
|
|
return it->second;
|
|
}
|
|
|
|
UserPtr SecurityManager::getUser(const String & user_name) const
|
|
{
|
|
auto it = users.find(user_name);
|
|
|
|
if (users.end() == it)
|
|
throw Exception("Unknown user " + user_name, ErrorCodes::UNKNOWN_USER);
|
|
|
|
return it->second;
|
|
}
|
|
|
|
bool SecurityManager::hasAccessToDatabase(const std::string & user_name, const std::string & database_name) const
|
|
{
|
|
auto it = users.find(user_name);
|
|
|
|
if (users.end() == it)
|
|
throw Exception("Unknown user " + user_name, ErrorCodes::UNKNOWN_USER);
|
|
|
|
auto user = it->second;
|
|
return user->databases.empty() || user->databases.count(database_name);
|
|
}
|
|
|
|
}
|