mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-09 01:50:50 +00:00
335 lines
8.0 KiB
C++
335 lines
8.0 KiB
C++
#include <Access/ExtendedRoleSet.h>
|
|
#include <Access/AccessControlManager.h>
|
|
#include <Access/User.h>
|
|
#include <Access/Role.h>
|
|
#include <Parsers/ASTExtendedRoleSet.h>
|
|
#include <Parsers/formatAST.h>
|
|
#include <IO/ReadHelpers.h>
|
|
#include <IO/WriteHelpers.h>
|
|
#include <boost/range/algorithm/set_algorithm.hpp>
|
|
#include <boost/range/algorithm/sort.hpp>
|
|
#include <boost/range/algorithm_ext/push_back.hpp>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int LOGICAL_ERROR;
|
|
}
|
|
|
|
|
|
ExtendedRoleSet::ExtendedRoleSet() = default;
|
|
ExtendedRoleSet::ExtendedRoleSet(const ExtendedRoleSet & src) = default;
|
|
ExtendedRoleSet & ExtendedRoleSet::operator =(const ExtendedRoleSet & src) = default;
|
|
ExtendedRoleSet::ExtendedRoleSet(ExtendedRoleSet && src) = default;
|
|
ExtendedRoleSet & ExtendedRoleSet::operator =(ExtendedRoleSet && src) = default;
|
|
|
|
|
|
ExtendedRoleSet::ExtendedRoleSet(AllTag)
|
|
{
|
|
all = true;
|
|
}
|
|
|
|
ExtendedRoleSet::ExtendedRoleSet(const UUID & id)
|
|
{
|
|
add(id);
|
|
}
|
|
|
|
|
|
ExtendedRoleSet::ExtendedRoleSet(const std::vector<UUID> & ids_)
|
|
{
|
|
add(ids_);
|
|
}
|
|
|
|
|
|
ExtendedRoleSet::ExtendedRoleSet(const boost::container::flat_set<UUID> & ids_)
|
|
{
|
|
add(ids_);
|
|
}
|
|
|
|
|
|
ExtendedRoleSet::ExtendedRoleSet(const ASTExtendedRoleSet & ast)
|
|
{
|
|
init(ast, nullptr, nullptr);
|
|
}
|
|
|
|
ExtendedRoleSet::ExtendedRoleSet(const ASTExtendedRoleSet & ast, const UUID & current_user_id)
|
|
{
|
|
init(ast, nullptr, ¤t_user_id);
|
|
}
|
|
|
|
ExtendedRoleSet::ExtendedRoleSet(const ASTExtendedRoleSet & ast, const AccessControlManager & manager)
|
|
{
|
|
init(ast, &manager, nullptr);
|
|
}
|
|
|
|
ExtendedRoleSet::ExtendedRoleSet(const ASTExtendedRoleSet & ast, const AccessControlManager & manager, const UUID & current_user_id)
|
|
{
|
|
init(ast, &manager, ¤t_user_id);
|
|
}
|
|
|
|
void ExtendedRoleSet::init(const ASTExtendedRoleSet & ast, const AccessControlManager * manager, const UUID * current_user_id)
|
|
{
|
|
all = ast.all;
|
|
|
|
auto name_to_id = [id_mode{ast.id_mode}, manager](const String & name) -> UUID
|
|
{
|
|
if (id_mode)
|
|
return parse<UUID>(name);
|
|
assert(manager);
|
|
auto id = manager->find<User>(name);
|
|
if (id)
|
|
return *id;
|
|
return manager->getID<Role>(name);
|
|
};
|
|
|
|
if (!ast.names.empty() && !all)
|
|
{
|
|
ids.reserve(ast.names.size());
|
|
for (const String & name : ast.names)
|
|
ids.insert(name_to_id(name));
|
|
}
|
|
|
|
if (ast.current_user && !all)
|
|
{
|
|
assert(current_user_id);
|
|
ids.insert(*current_user_id);
|
|
}
|
|
|
|
if (!ast.except_names.empty())
|
|
{
|
|
except_ids.reserve(ast.except_names.size());
|
|
for (const String & except_name : ast.except_names)
|
|
except_ids.insert(name_to_id(except_name));
|
|
}
|
|
|
|
if (ast.except_current_user)
|
|
{
|
|
assert(current_user_id);
|
|
except_ids.insert(*current_user_id);
|
|
}
|
|
|
|
for (const UUID & except_id : except_ids)
|
|
ids.erase(except_id);
|
|
}
|
|
|
|
|
|
std::shared_ptr<ASTExtendedRoleSet> ExtendedRoleSet::toAST() const
|
|
{
|
|
auto ast = std::make_shared<ASTExtendedRoleSet>();
|
|
ast->id_mode = true;
|
|
ast->all = all;
|
|
|
|
if (!ids.empty())
|
|
{
|
|
ast->names.reserve(ids.size());
|
|
for (const UUID & id : ids)
|
|
ast->names.emplace_back(::DB::toString(id));
|
|
}
|
|
|
|
if (!except_ids.empty())
|
|
{
|
|
ast->except_names.reserve(except_ids.size());
|
|
for (const UUID & except_id : except_ids)
|
|
ast->except_names.emplace_back(::DB::toString(except_id));
|
|
}
|
|
|
|
return ast;
|
|
}
|
|
|
|
|
|
String ExtendedRoleSet::toString() const
|
|
{
|
|
auto ast = toAST();
|
|
return serializeAST(*ast);
|
|
}
|
|
|
|
|
|
Strings ExtendedRoleSet::toStrings() const
|
|
{
|
|
if (all || !except_ids.empty())
|
|
return {toString()};
|
|
|
|
Strings names;
|
|
names.reserve(ids.size());
|
|
for (const UUID & id : ids)
|
|
names.emplace_back(::DB::toString(id));
|
|
return names;
|
|
}
|
|
|
|
|
|
std::shared_ptr<ASTExtendedRoleSet> ExtendedRoleSet::toASTWithNames(const AccessControlManager & manager) const
|
|
{
|
|
auto ast = std::make_shared<ASTExtendedRoleSet>();
|
|
ast->all = all;
|
|
|
|
if (!ids.empty())
|
|
{
|
|
ast->names.reserve(ids.size());
|
|
for (const UUID & id : ids)
|
|
{
|
|
auto name = manager.tryReadName(id);
|
|
if (name)
|
|
ast->names.emplace_back(std::move(*name));
|
|
}
|
|
boost::range::sort(ast->names);
|
|
}
|
|
|
|
if (!except_ids.empty())
|
|
{
|
|
ast->except_names.reserve(except_ids.size());
|
|
for (const UUID & except_id : except_ids)
|
|
{
|
|
auto except_name = manager.tryReadName(except_id);
|
|
if (except_name)
|
|
ast->except_names.emplace_back(std::move(*except_name));
|
|
}
|
|
boost::range::sort(ast->except_names);
|
|
}
|
|
|
|
return ast;
|
|
}
|
|
|
|
|
|
String ExtendedRoleSet::toStringWithNames(const AccessControlManager & manager) const
|
|
{
|
|
auto ast = toASTWithNames(manager);
|
|
return serializeAST(*ast);
|
|
}
|
|
|
|
|
|
Strings ExtendedRoleSet::toStringsWithNames(const AccessControlManager & manager) const
|
|
{
|
|
if (all || !except_ids.empty())
|
|
return {toStringWithNames(manager)};
|
|
|
|
Strings names;
|
|
names.reserve(ids.size());
|
|
for (const UUID & id : ids)
|
|
{
|
|
auto name = manager.tryReadName(id);
|
|
if (name)
|
|
names.emplace_back(std::move(*name));
|
|
}
|
|
boost::range::sort(names);
|
|
return names;
|
|
}
|
|
|
|
|
|
bool ExtendedRoleSet::empty() const
|
|
{
|
|
return ids.empty() && !all;
|
|
}
|
|
|
|
|
|
void ExtendedRoleSet::clear()
|
|
{
|
|
ids.clear();
|
|
all = false;
|
|
except_ids.clear();
|
|
}
|
|
|
|
|
|
void ExtendedRoleSet::add(const UUID & id)
|
|
{
|
|
ids.insert(id);
|
|
}
|
|
|
|
|
|
void ExtendedRoleSet::add(const std::vector<UUID> & ids_)
|
|
{
|
|
for (const auto & id : ids_)
|
|
add(id);
|
|
}
|
|
|
|
|
|
void ExtendedRoleSet::add(const boost::container::flat_set<UUID> & ids_)
|
|
{
|
|
for (const auto & id : ids_)
|
|
add(id);
|
|
}
|
|
|
|
|
|
bool ExtendedRoleSet::match(const UUID & id) const
|
|
{
|
|
return (all || ids.contains(id)) && !except_ids.contains(id);
|
|
}
|
|
|
|
|
|
bool ExtendedRoleSet::match(const UUID & user_id, const std::vector<UUID> & enabled_roles) const
|
|
{
|
|
if (!all && !ids.contains(user_id))
|
|
{
|
|
bool found_enabled_role = std::any_of(
|
|
enabled_roles.begin(), enabled_roles.end(), [this](const UUID & enabled_role) { return ids.contains(enabled_role); });
|
|
if (!found_enabled_role)
|
|
return false;
|
|
}
|
|
|
|
if (except_ids.contains(user_id))
|
|
return false;
|
|
|
|
bool in_except_list = std::any_of(
|
|
enabled_roles.begin(), enabled_roles.end(), [this](const UUID & enabled_role) { return except_ids.contains(enabled_role); });
|
|
return !in_except_list;
|
|
}
|
|
|
|
|
|
bool ExtendedRoleSet::match(const UUID & user_id, const boost::container::flat_set<UUID> & enabled_roles) const
|
|
{
|
|
if (!all && !ids.contains(user_id))
|
|
{
|
|
bool found_enabled_role = std::any_of(
|
|
enabled_roles.begin(), enabled_roles.end(), [this](const UUID & enabled_role) { return ids.contains(enabled_role); });
|
|
if (!found_enabled_role)
|
|
return false;
|
|
}
|
|
|
|
if (except_ids.contains(user_id))
|
|
return false;
|
|
|
|
bool in_except_list = std::any_of(
|
|
enabled_roles.begin(), enabled_roles.end(), [this](const UUID & enabled_role) { return except_ids.contains(enabled_role); });
|
|
return !in_except_list;
|
|
}
|
|
|
|
|
|
std::vector<UUID> ExtendedRoleSet::getMatchingIDs() const
|
|
{
|
|
if (all)
|
|
throw Exception("getAllMatchingIDs() can't get ALL ids without manager", ErrorCodes::LOGICAL_ERROR);
|
|
std::vector<UUID> res;
|
|
boost::range::set_difference(ids, except_ids, std::back_inserter(res));
|
|
return res;
|
|
}
|
|
|
|
|
|
std::vector<UUID> ExtendedRoleSet::getMatchingIDs(const AccessControlManager & manager) const
|
|
{
|
|
if (!all)
|
|
return getMatchingIDs();
|
|
|
|
std::vector<UUID> res;
|
|
for (const UUID & id : manager.findAll<User>())
|
|
{
|
|
if (match(id))
|
|
res.push_back(id);
|
|
}
|
|
for (const UUID & id : manager.findAll<Role>())
|
|
{
|
|
if (match(id))
|
|
res.push_back(id);
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
|
|
bool operator ==(const ExtendedRoleSet & lhs, const ExtendedRoleSet & rhs)
|
|
{
|
|
return (lhs.all == rhs.all) && (lhs.ids == rhs.ids) && (lhs.except_ids == rhs.except_ids);
|
|
}
|
|
|
|
}
|