143 lines
4.3 KiB
C++
143 lines
4.3 KiB
C++
/*
|
|
Copyright <2018-2025> <scott.e.graves@protonmail.com>
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in all
|
|
copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
SOFTWARE.
|
|
*/
|
|
#if !defined(_WIN32)
|
|
|
|
#include "utils/unix.hpp"
|
|
|
|
#include "utils/collection.hpp"
|
|
#include "utils/error.hpp"
|
|
|
|
namespace {
|
|
[[nodiscard]] auto get_group_list(auto *pass) -> std::vector<gid_t> {
|
|
REPERTORY_USES_FUNCTION_NAME();
|
|
|
|
std::vector<gid_t> groups{};
|
|
#if defined(__APPLE__)
|
|
constexpr const int buffer_count{8};
|
|
constexpr const int max_group_count{1024};
|
|
groups.resize(buffer_count);
|
|
|
|
std::size_t orig_count{0U};
|
|
while (true) {
|
|
auto group_count{static_cast<int>(groups.size())};
|
|
if (group_count > max_group_count) {
|
|
repertory::utils::error::handle_error(function_name,
|
|
"group list has too many groups");
|
|
break;
|
|
}
|
|
|
|
auto res{
|
|
getgrouplist(pass->pw_name, static_cast<int>(pass->pw_gid),
|
|
reinterpret_cast<int *>(groups.data()), &group_count),
|
|
};
|
|
if (res < 0) {
|
|
if (orig_count == 0U) {
|
|
repertory::utils::error::handle_error(
|
|
function_name, std::string{"failed to get group list|error|"} +
|
|
std::to_string(errno));
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
groups.resize(static_cast<std::size_t>(group_count));
|
|
if (groups.size() == orig_count) {
|
|
break;
|
|
}
|
|
|
|
orig_count = groups.size();
|
|
}
|
|
#else // !defined(__APPLE__)
|
|
int group_count{};
|
|
auto res = getgrouplist(pass->pw_name, pass->pw_gid, nullptr, &group_count);
|
|
if (res >= 0) {
|
|
repertory::utils::error::handle_error(
|
|
function_name, std::string{"failed to get group list count|error|"} +
|
|
std::to_string(errno));
|
|
}
|
|
#endif // defined(__APPLE__)
|
|
|
|
#if !defined(__APPLE__)
|
|
groups.resize(static_cast<std::size_t>(group_count));
|
|
res = getgrouplist(pass->pw_name, pass->pw_gid, groups.data(), &group_count);
|
|
if (res >= 0) {
|
|
repertory::utils::error::handle_error(
|
|
function_name,
|
|
std::string{"failed to get group list|error|"} + std::to_string(errno));
|
|
}
|
|
#endif // !defined(__APPLE__)
|
|
|
|
return groups;
|
|
}
|
|
} // namespace
|
|
|
|
namespace repertory::utils {
|
|
#if !defined(__APPLE__)
|
|
auto convert_to_uint64(const pthread_t &thread) -> std::uint64_t {
|
|
return static_cast<std::uint64_t>(thread);
|
|
}
|
|
#endif // !defined(__APPLE__)
|
|
|
|
auto get_last_error_code() -> int { return errno; }
|
|
|
|
auto get_thread_id() -> std::uint64_t {
|
|
return convert_to_uint64(pthread_self());
|
|
}
|
|
|
|
auto is_uid_member_of_group(uid_t uid, gid_t gid) -> bool {
|
|
REPERTORY_USES_FUNCTION_NAME();
|
|
|
|
std::vector<gid_t> groups{};
|
|
auto res = use_getpwuid(
|
|
uid, [&groups](struct passwd *pass) { groups = get_group_list(pass); });
|
|
if (not res) {
|
|
throw utils::error::create_exception(res.function_name,
|
|
{"use_getpwuid failed", res.reason});
|
|
}
|
|
|
|
return collection::includes(groups, gid);
|
|
}
|
|
|
|
auto use_getpwuid(uid_t uid, passwd_callback_t callback) -> result {
|
|
REPERTORY_USES_FUNCTION_NAME();
|
|
|
|
static std::mutex mtx{};
|
|
mutex_lock lock{mtx};
|
|
|
|
auto *temp_pw = getpwuid(uid);
|
|
if (temp_pw == nullptr) {
|
|
return {
|
|
.function_name = std::string{function_name},
|
|
.ok = false,
|
|
.reason = "'getpwuid' returned nullptr",
|
|
};
|
|
}
|
|
|
|
callback(temp_pw);
|
|
return {
|
|
.function_name = std::string{function_name},
|
|
};
|
|
}
|
|
} // namespace repertory::utils
|
|
|
|
#endif // !defined(_WIN32)
|