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

Support CPU affinity on FreeBSD.

Browse Source
This commit is contained in:
levlam 2022-09-17 14:45:42 +03:00
parent 9d5c151a3f
commit b47fce0d95
1 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -14,8 +14,14 @@ char disable_linker_warning_about_empty_file_thread_pthread_cpp TD_UNUSED;
#include "td/utils/port/detail/skip_eintr.h" #include "td/utils/port/detail/skip_eintr.h"
#include <pthread.h> #include <pthread.h>
#if TD_FREEBSD
#include <pthread_np.h>
#endif
#include <sched.h> #include <sched.h>
#include <signal.h> #include <signal.h>
#if TD_FREEBSD
#include <sys/cpuset.h>
#endif
#if TD_FREEBSD || TD_OPENBSD || TD_NETBSD #if TD_FREEBSD || TD_OPENBSD || TD_NETBSD
#include <sys/sysctl.h> #include <sys/sysctl.h>
#endif #endif
@ -96,8 +102,12 @@ int ThreadPthread::do_pthread_create(pthread_t *thread, const pthread_attr_t *at
} }
Status ThreadPthread::set_affinity_mask(id thread_id, uint64 mask) { Status ThreadPthread::set_affinity_mask(id thread_id, uint64 mask) {
#if TD_LINUX #if TD_LINUX || TD_FREEBSD
#if TD_FREEBSD
cpuset_t cpuset;
#else
cpu_set_t cpuset; cpu_set_t cpuset;
#endif
CPU_ZERO(&cpuset); CPU_ZERO(&cpuset);
for (int j = 0; j < 64 && j < CPU_SETSIZE; j++) { for (int j = 0; j < 64 && j < CPU_SETSIZE; j++) {
if ((mask >> j) & 1) { if ((mask >> j) & 1) {
@ -105,7 +115,7 @@ Status ThreadPthread::set_affinity_mask(id thread_id, uint64 mask) {
} }
} }
auto res = skip_eintr([&] { return pthread_setaffinity_np(thread_id, sizeof(cpu_set_t), &cpuset); }); auto res = skip_eintr([&] { return pthread_setaffinity_np(thread_id, sizeof(cpuset), &cpuset); });
if (res) { if (res) {
return OS_ERROR("Failed to set thread affinity mask"); return OS_ERROR("Failed to set thread affinity mask");
} }
@ -116,10 +126,14 @@ Status ThreadPthread::set_affinity_mask(id thread_id, uint64 mask) {
} }
uint64 ThreadPthread::get_affinity_mask(id thread_id) { uint64 ThreadPthread::get_affinity_mask(id thread_id) {
#if TD_LINUX #if TD_LINUX || TD_FREEBSD
#if TD_FREEBSD
cpuset_t cpuset;
#else
cpu_set_t cpuset; cpu_set_t cpuset;
#endif
CPU_ZERO(&cpuset); CPU_ZERO(&cpuset);
auto res = skip_eintr([&] { return pthread_getaffinity_np(thread_id, sizeof(cpu_set_t), &cpuset); }); auto res = skip_eintr([&] { return pthread_getaffinity_np(thread_id, sizeof(cpuset), &cpuset); });
if (res) { if (res) {
return 0; return 0;
} }