tildefriends/deps/libuv/test/test-thread-affinity.c

137 lines
3.1 KiB
C

/* Copyright libuv project contributors. All rights reserved.
*/
#include "uv.h"
#include "task.h"
#include <string.h>
#ifndef NO_CPU_AFFINITY
static void check_affinity(void* arg) {
int r;
char* cpumask;
int cpumasksize;
uv_thread_t tid;
cpumask = (char*)arg;
cpumasksize = uv_cpumask_size();
ASSERT(cpumasksize > 0);
tid = uv_thread_self();
r = uv_thread_setaffinity(&tid, cpumask, NULL, cpumasksize);
ASSERT(r == 0);
r = uv_thread_setaffinity(&tid, cpumask + cpumasksize, cpumask, cpumasksize);
ASSERT(r == 0);
}
TEST_IMPL(thread_affinity) {
int t1first;
int t1second;
int t2first;
int t2second;
int cpumasksize;
char* cpumask;
int ncpus;
int r;
int c;
int i;
uv_thread_t threads[3];
#ifdef _WIN32
/* uv_thread_self isn't defined for the main thread on Windows */
threads[0] = GetCurrentThread();
#else
threads[0] = uv_thread_self();
#endif
cpumasksize = uv_cpumask_size();
ASSERT(cpumasksize > 0);
cpumask = calloc(4 * cpumasksize, 1);
ASSERT(cpumask);
r = uv_thread_getaffinity(&threads[0], cpumask, cpumasksize);
ASSERT(r == 0);
ASSERT(cpumask[0] && "test must be run with cpu 0 affinity");
ncpus = 0;
while (cpumask[++ncpus]) { }
memset(cpumask, 0, 4 * cpumasksize);
t1first = cpumasksize * 0;
t1second = cpumasksize * 1;
t2first = cpumasksize * 2;
t2second = cpumasksize * 3;
cpumask[t1second + 0] = 1;
cpumask[t2first + 0] = 1;
cpumask[t1first + (ncpus >= 2)] = 1;
cpumask[t2second + (ncpus >= 2)] = 1;
#ifdef __linux__
cpumask[t1second + 2] = 1;
cpumask[t2first + 2] = 1;
cpumask[t1first + 3] = 1;
cpumask[t2second + 3] = 1;
#else
if (ncpus >= 3) {
cpumask[t1second + 2] = 1;
cpumask[t2first + 2] = 1;
}
if (ncpus >= 4) {
cpumask[t1first + 3] = 1;
cpumask[t2second + 3] = 1;
}
#endif
ASSERT(0 == uv_thread_create(threads + 1,
check_affinity,
&cpumask[t1first]));
ASSERT(0 == uv_thread_create(threads + 2,
check_affinity,
&cpumask[t2first]));
ASSERT(0 == uv_thread_join(threads + 1));
ASSERT(0 == uv_thread_join(threads + 2));
ASSERT(cpumask[t1first + 0] == (ncpus == 1));
ASSERT(cpumask[t1first + 1] == (ncpus >= 2));
ASSERT(cpumask[t1first + 2] == 0);
ASSERT(cpumask[t1first + 3] == (ncpus >= 4));
ASSERT(cpumask[t2first + 0] == 1);
ASSERT(cpumask[t2first + 1] == 0);
ASSERT(cpumask[t2first + 2] == (ncpus >= 3));
ASSERT(cpumask[t2first + 3] == 0);
c = uv_thread_getcpu();
ASSERT_GE(c, 0);
memset(cpumask, 0, cpumasksize);
cpumask[c] = 1;
r = uv_thread_setaffinity(&threads[0], cpumask, NULL, cpumasksize);
ASSERT_EQ(r, 0);
memset(cpumask, 0, cpumasksize);
r = uv_thread_getaffinity(&threads[0], cpumask, cpumasksize);
ASSERT_EQ(r, 0);
for (i = 0; i < cpumasksize; i++) {
if (i == c)
ASSERT_EQ(1, cpumask[i]);
else
ASSERT_EQ(0, cpumask[i]);
}
free(cpumask);
return 0;
}
#else
TEST_IMPL(thread_affinity) {
int cpumasksize;
cpumasksize = uv_cpumask_size();
ASSERT(cpumasksize == UV_ENOTSUP);
return 0;
}
#endif