tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Support CPU affinity on NetBSD.

Browse Source
This commit is contained in:
levlam 2022-09-17 21:48:16 +03:00
parent b47fce0d95
commit b8abc6c19e
1 changed files with 44 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
tdutils/td/utils/port/detail/ThreadPthread.cpp
View File

@ -12,6 +12,9 @@ char disable_linker_warning_about_empty_file_thread_pthread_cpp TD_UNUSED;
#include "td/utils/misc.h"
#include "td/utils/port/detail/skip_eintr.h"
#if TD_NETBSD
#include "td/utils/ScopeGuard.h"
#endif
#include <pthread.h>
#if TD_FREEBSD
@ -120,6 +123,27 @@ Status ThreadPthread::set_affinity_mask(id thread_id, uint64 mask) {
return OS_ERROR("Failed to set thread affinity mask");
}
return Status::OK();
#elif TD_NETBSD
cpuset_t *cpuset = cpuset_create();
if (cpuset == nullptr) {
return OS_ERROR("Failed to create cpuset");
}
SCOPE_EXIT {
cpuset_destroy(cpuset);
};
for (int j = 0; j < 64; j++) {
if ((mask >> j) & 1) {
if (cpuset_set(j, cpuset) != 0) {
return OS_ERROR("Failed to set CPU identifier");
}
}
}
auto res = skip_eintr([&] { return pthread_setaffinity_np(thread_id, cpuset_size(cpuset), cpuset); });
if (res) {
return OS_ERROR("Failed to set thread affinity mask");
}
return Status::OK();
#else
return Status::Error("Unsupported");
#endif
@ -145,6 +169,26 @@ uint64 ThreadPthread::get_affinity_mask(id thread_id) {
}
}
return mask;
#elif TD_NETBSD
cpuset_t *cpuset = cpuset_create();
if (cpuset == nullptr) {
return 0;
}
SCOPE_EXIT {
cpuset_destroy(cpuset);
};
auto res = skip_eintr([&] { return pthread_getaffinity_np(thread_id, cpuset_size(cpuset), cpuset); });
if (res) {
return 0;
}
uint64 mask = 0;
for (int j = 0; j < 64; j++) {
if (cpuset_isset(j, cpuset) > 0) {
mask |= static_cast<uint64>(1) << j;
}
}
return mask;
#else
return 0;
#endif