mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-17 03:42:48 +00:00
1a0b5f33b3
cmake/target.cmake defines macros for the supported platforms, this commit changes predefined system macros to our own macros. __linux__ --> OS_LINUX __APPLE__ --> OS_DARWIN __FreeBSD__ --> OS_FREEBSD
44 lines
915 B
C++
44 lines
915 B
C++
#if defined(OS_LINUX)
|
|
|
|
#include "hasLinuxCapability.h"
|
|
|
|
#include <syscall.h>
|
|
#include <unistd.h>
|
|
#include <linux/capability.h>
|
|
#include <Common/Exception.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int NETLINK_ERROR;
|
|
}
|
|
|
|
static __user_cap_data_struct getCapabilities()
|
|
{
|
|
/// See man getcap.
|
|
__user_cap_header_struct request{};
|
|
request.version = _LINUX_CAPABILITY_VERSION_1; /// It's enough to check just single CAP_NET_ADMIN capability we are interested.
|
|
request.pid = getpid();
|
|
|
|
__user_cap_data_struct response{};
|
|
|
|
/// Avoid dependency on 'libcap'.
|
|
if (0 != syscall(SYS_capget, &request, &response))
|
|
throwFromErrno("Cannot do 'capget' syscall", ErrorCodes::NETLINK_ERROR);
|
|
|
|
return response;
|
|
}
|
|
|
|
bool hasLinuxCapability(int cap)
|
|
{
|
|
static __user_cap_data_struct capabilities = getCapabilities();
|
|
return (1 << cap) & capabilities.effective;
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|