1 /*
2 * Copyright (c) 1995-1998 John Birrell <jb@cimlogic.com.au>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. Neither the name of the author nor the names of any co-contributors
14 * may be used to endorse or promote products derived from this software
15 * without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY JOHN BIRRELL AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 */
29
30 #include "namespace.h"
31 #include <sys/lwp.h>
32 #include <errno.h>
33 #include <string.h>
34 #include <pthread.h>
35 #include <pthread_np.h>
36 #include "un-namespace.h"
37
38 #include "thr_private.h"
39
40 int
_pthread_getaffinity_np(pthread_t thread,size_t cpusetsize,cpu_set_t * mask)41 _pthread_getaffinity_np(pthread_t thread, size_t cpusetsize,
42 cpu_set_t *mask)
43 {
44 pthread_t curthread = tls_get_curthread();
45 cpu_set_t mask1;
46 int ret;
47
48 if (curthread == thread) {
49 ret = 0;
50 if (lwp_getaffinity(0, thread->tid, &mask1) < 0)
51 ret = errno;
52 } else {
53 ret = ESRCH;
54 if (_thr_ref_add(curthread, thread, 0) == 0) {
55 THR_THREAD_LOCK(curthread, thread);
56 if (thread->state != PS_DEAD) {
57 ret = 0;
58 if (lwp_getaffinity(0, thread->tid, &mask1) < 0)
59 ret = errno;
60 }
61 THR_THREAD_UNLOCK(curthread, thread);
62 _thr_ref_delete(curthread, thread);
63 }
64 }
65 if (ret != 0)
66 return (ret);
67
68 if (cpusetsize > sizeof(mask1)) {
69 memset(mask, 0, cpusetsize);
70 memcpy(mask, &mask1, sizeof(mask1));
71 } else {
72 memcpy(mask, &mask1, cpusetsize);
73 }
74 return (0);
75 }
76 __strong_reference(_pthread_getaffinity_np, pthread_getaffinity_np);
77
78 int
_pthread_setaffinity_np(pthread_t thread,size_t cpusetsize,const cpu_set_t * mask)79 _pthread_setaffinity_np(pthread_t thread, size_t cpusetsize,
80 const cpu_set_t *mask)
81 {
82 pthread_t curthread = tls_get_curthread();
83 size_t cplen = cpusetsize;
84 cpu_set_t mask1;
85 int ret;
86
87 if (cplen > sizeof(mask1))
88 cplen = sizeof(mask1);
89 CPU_ZERO(&mask1);
90 memcpy(&mask1, mask, cplen);
91
92 if (curthread == thread) {
93 ret = 0;
94 if (lwp_setaffinity(0, thread->tid, &mask1) < 0)
95 ret = errno;
96 } else {
97 ret = ESRCH;
98 if (_thr_ref_add(curthread, thread, 0) == 0) {
99 THR_THREAD_LOCK(curthread, thread);
100 if (thread->state != PS_DEAD) {
101 ret = 0;
102 if (lwp_setaffinity(0, thread->tid, &mask1) < 0)
103 ret = errno;
104 }
105 THR_THREAD_UNLOCK(curthread, thread);
106 _thr_ref_delete(curthread, thread);
107 }
108 }
109 return (ret);
110 }
111 __strong_reference(_pthread_setaffinity_np, pthread_setaffinity_np);
112