2 * Copyright (c) 1995-1998 John Birrell <jb@cimlogic.com.au>
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
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.
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
30 #include "namespace.h"
35 #include <pthread_np.h>
36 #include "un-namespace.h"
38 #include "thr_private.h"
41 _pthread_getaffinity_np(pthread_t thread, size_t cpusetsize,
44 struct pthread *curthread = tls_get_curthread();
48 if (curthread == thread) {
50 if (lwp_getaffinity(0, thread->tid, &mask1) < 0)
54 if (_thr_ref_add(curthread, thread, 0) == 0) {
55 THR_THREAD_LOCK(curthread, thread);
56 if (thread->state != PS_DEAD) {
58 if (lwp_getaffinity(0, thread->tid, &mask1) < 0)
61 THR_THREAD_UNLOCK(curthread, thread);
62 _thr_ref_delete(curthread, thread);
68 if (cpusetsize > sizeof(mask1)) {
69 memset(mask, 0, cpusetsize);
70 memcpy(mask, &mask1, sizeof(mask1));
72 memcpy(mask, &mask1, cpusetsize);
76 __strong_reference(_pthread_getaffinity_np, pthread_getaffinity_np);
79 _pthread_setaffinity_np(pthread_t thread, size_t cpusetsize,
80 const cpu_set_t *mask)
82 struct pthread *curthread = tls_get_curthread();
83 size_t cplen = cpusetsize;
87 if (cplen > sizeof(mask1))
88 cplen = sizeof(mask1);
90 memcpy(&mask1, mask, cplen);
92 if (curthread == thread) {
94 if (lwp_setaffinity(0, thread->tid, &mask1) < 0)
98 if (_thr_ref_add(curthread, thread, 0) == 0) {
99 THR_THREAD_LOCK(curthread, thread);
100 if (thread->state != PS_DEAD) {
102 if (lwp_setaffinity(0, thread->tid, &mask1) < 0)
105 THR_THREAD_UNLOCK(curthread, thread);
106 _thr_ref_delete(curthread, thread);
111 __strong_reference(_pthread_setaffinity_np, pthread_setaffinity_np);