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
|
/*
* affinity5.c
*
*
* --------------------------------------------------------------------------
*
* Pthreads-win32 - POSIX Threads Library for Win32
* Copyright(C) 1998 John E. Bossom
* Copyright(C) 1999,2012 Pthreads-win32 contributors
*
* Homepage1: http://sourceware.org/pthreads-win32/
* Homepage2: http://sourceforge.net/projects/pthreads4w/
*
* The current list of contributors is contained
* in the file CONTRIBUTORS included with the source
* code distribution. The list can also be seen at the
* following World Wide Web location:
* http://sources.redhat.com/pthreads-win32/contributors.html
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library in the file COPYING.LIB;
* if not, write to the Free Software Foundation, Inc.,
* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
*
* --------------------------------------------------------------------------
*
* Test thread CPU affinity inheritance.
*
*/
#include "test.h"
typedef union
{
/* Violates opacity */
cpu_set_t cpuset;
unsigned long int bits; /* To stop GCC complaining about %lx args to printf */
} cpuset_to_ulint;
void *
mythread(void * arg)
{
HANDLE threadH = GetCurrentThread();
cpu_set_t *parentCpus = (cpu_set_t*) arg;
cpu_set_t threadCpus;
DWORD_PTR vThreadMask;
cpuset_to_ulint a, b;
assert(pthread_getaffinity_np(pthread_self(), sizeof(cpu_set_t), &threadCpus) == 0);
assert(CPU_EQUAL(parentCpus, &threadCpus));
vThreadMask = SetThreadAffinityMask(threadH, (*(PDWORD_PTR)&threadCpus) /* Violating Opacity */);
assert(vThreadMask != 0);
assert(memcmp(&vThreadMask, &threadCpus, sizeof(DWORD_PTR)) == 0);
a.cpuset = *parentCpus;
b.cpuset = threadCpus;
/* Violates opacity */
printf("CPU affinity: Parent/Thread = 0x%lx/0x%lx\n", a.bits, b.bits);
return (void*) 0;
}
int
main()
{
unsigned int cpu;
pthread_t tid;
cpu_set_t threadCpus;
DWORD_PTR vThreadMask;
cpu_set_t keepCpus;
pthread_t self = pthread_self();
CPU_ZERO(&keepCpus);
for (cpu = 1; cpu < sizeof(cpu_set_t)*8; cpu += 2)
{
CPU_SET(cpu, &keepCpus); /* 0b10101010101010101010101010101010 */
}
assert(pthread_getaffinity_np(self, sizeof(cpu_set_t), &threadCpus) == 0);
if (CPU_COUNT(&threadCpus) > 1)
{
assert(pthread_create(&tid, NULL, mythread, (void*)&threadCpus) == 0);
assert(pthread_join(tid, NULL) == 0);
CPU_AND(&threadCpus, &threadCpus, &keepCpus);
assert(pthread_setaffinity_np(self, sizeof(cpu_set_t), &threadCpus) == 0);
vThreadMask = SetThreadAffinityMask(GetCurrentThread(), (*(PDWORD_PTR)&threadCpus) /* Violating Opacity */);
assert(vThreadMask != 0);
assert(memcmp(&vThreadMask, &threadCpus, sizeof(DWORD_PTR)) == 0);
assert(pthread_create(&tid, NULL, mythread, (void*)&threadCpus) == 0);
assert(pthread_join(tid, NULL) == 0);
}
return 0;
}
|
