1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
#include <EXTERN.h>
#include <perl.h>
#include <XSUB.h>
#include <pthread.h>
#include <sched.h>
MODULE = Sys::CpuAffinity PACKAGE = Sys::CpuAffinity
int
xs_pthread_self_getaffinity(dummy)
int dummy
CODE:
/*
* Retrieves the CPU affinity of the current thread.
* For use with NetBSD, but it also might work on Linux and FreeBSD.
* On NetBSD, may require super-user.
*
* Return < 0 on error.
* Return current thread affinity on success.
*/
#ifndef DEBIAN
/* Ubuntu 15.10 build log filter will abort on the implicit *
* int-to-pointer conversion below, even though this xs file *
* won't compile -- hide all this code under Debian *
* -- http://aunchpadlibrarian.net/222688017 *
/buildlog_ubuntu-wily-amd64.libsys-cpuaffinity-perl_1.06-1ubuntu1~wily1_BUILDING.txt.gz */
cpuset_t *cset;
pthread_t pth;
cpuid_t icpu;
int error, affinity;
pth = pthread_self();
affinity = 0;
cset = cpuset_create();
if (cset == NULL) {
fprintf(stderr, "pthread_getaffinity_np: failed to create cpuset\n");
affinity = -1;
}
error = pthread_getaffinity_np(pth, cpuset_size(cset), cset);
if (error) {
fprintf(stderr, "pthread_getaffinity_np: %d %s\n",
error, strerror(error));
affinity = -1;
}
if (affinity >= 0) {
for (icpu = 0; icpu < 8 * cpuset_size(cset); icpu++) {
int n = cpuset_isset(icpu, cset);
if (n < 0) {
break;
} else if (n > 0) {
affinity |= 1 << icpu;
}
}
/*
* What does it mean if affinity is still 0 here?
* Does that mean that pthread_getaffinity_np didn't work?
* Or does it mean that the thread affinity is in a default
* (i.e., affinitied to all processors)?
*/
}
if (cset != NULL) {
cpuset_destroy(cset);
}
#endif
RETVAL = affinity;
OUTPUT:
RETVAL
int
xs_pthread_self_setaffinity(affinity)
int affinity
CODE:
/*
* Sets the CPU affinity for the current thread.
* For use with NetBSD. Might need to be run as super-user.
* Returns 0 on error, 1 on success.
*/
#ifndef DEBIAN
cpuset_t *cset;
pthread_t pth;
cpuid_t icpu;
int error, result;
pth = pthread_self();
result = 2;
cset = cpuset_create();
if (cset == NULL) {
fprintf(stderr, "xs_set_pthread_self_affinity: failed to create cpu set\n");
result = 0;
} else {
for (icpu = 0; icpu < 8 * cpuset_size(cset); icpu++) {
if (affinity & (1 << icpu)) {
cpuset_set(icpu, cset);
} else {
cpuset_clr(icpu, cset);
}
}
error = pthread_setaffinity_np(pth, cpuset_size(cset), cset);
if (error) {
fprintf(stderr, "xs_set_pthread_self_affinity: %d %s\n",
error, strerror(error));
result = 0;
} else {
result = 1;
}
}
if (cset != NULL) {
cpuset_destroy(cset);
}
#endif
RETVAL = result;
OUTPUT:
RETVAL
|
