From f88b9710087dfae24a065020116215db6bc34d68 Mon Sep 17 00:00:00 2001 From: davidxu Date: Mon, 3 Mar 2008 09:16:29 +0000 Subject: Implement functions pthread_getaffinity_np and pthread_setaffinity_np to get and set thread's cpu affinity mask. --- lib/libthr/pthread.map | 4 ++- lib/libthr/thread/Makefile.inc | 1 + lib/libthr/thread/thr_affinity.c | 74 ++++++++++++++++++++++++++++++++++++++++ 3 files changed, 78 insertions(+), 1 deletion(-) create mode 100644 lib/libthr/thread/thr_affinity.c (limited to 'lib/libthr') diff --git a/lib/libthr/pthread.map b/lib/libthr/pthread.map index 0da4f67..b2ef7e8 100644 --- a/lib/libthr/pthread.map +++ b/lib/libthr/pthread.map @@ -117,8 +117,8 @@ FBSD_1.0 { pthread_rwlockattr_getpshared; pthread_rwlockattr_init; pthread_rwlockattr_setpshared; - pthread_self; pthread_set_name_np; + pthread_self; pthread_setcancelstate; pthread_setcanceltype; pthread_setconcurrency; @@ -388,6 +388,8 @@ FBSDprivate_1.0 { }; FBSD_1.1 { + pthread_getaffinity_np; + pthread_setaffinity_np; pthread_mutex_getspinloops_np; pthread_mutex_getyieldloops_np; pthread_mutex_isowned_np; diff --git a/lib/libthr/thread/Makefile.inc b/lib/libthr/thread/Makefile.inc index bdae92e..6a5105f 100644 --- a/lib/libthr/thread/Makefile.inc +++ b/lib/libthr/thread/Makefile.inc @@ -4,6 +4,7 @@ .PATH: ${.CURDIR}/thread SRCS+= \ + thr_affinity.c \ thr_attr.c \ thr_barrier.c \ thr_barrierattr.c \ diff --git a/lib/libthr/thread/thr_affinity.c b/lib/libthr/thread/thr_affinity.c new file mode 100644 index 0000000..240dd74 --- /dev/null +++ b/lib/libthr/thread/thr_affinity.c @@ -0,0 +1,74 @@ +/* + * Copyright (c) 2008, David Xu + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice unmodified, this list of conditions, and the following + * disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR + * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES + * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. + * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, + * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT + * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, + * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY + * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF + * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + * + * $FreeBSD$ + * + */ + +#include "namespace.h" +#include +#include +#include +#include "un-namespace.h" + +#include "thr_private.h" + +__weak_reference(_pthread_getaffinity_np, pthread_getaffinity_np); +__weak_reference(_pthread_setaffinity_np, pthread_setaffinity_np); + +int +_pthread_setaffinity_np(pthread_t td, size_t cpusetsize, const cpuset_t *cpuset) +{ + struct pthread *curthread = _get_curthread(); + lwpid_t tid; + int error; + + THR_THREAD_LOCK(curthread, td); + if (td->state == PS_DEAD) { + THR_THREAD_UNLOCK(curthread, td); + return (EINVAL); + } + tid = TID(td); + error = cpuset_setaffinity(CPU_LEVEL_WHICH, CPU_WHICH_TID, tid, + cpusetsize, cpuset); + if (error == -1) + error = errno; + THR_THREAD_UNLOCK(curthread, td); + return (error); +} + +int +_pthread_getaffinity_np(pthread_t td, size_t cpusetsize, cpuset_t *cpuset) +{ + lwpid_t tid; + int error; + + tid = TID(td); + error = cpuset_getaffinity(CPU_LEVEL_WHICH, CPU_WHICH_TID, tid, + cpusetsize, cpuset); + if (error == -1) + error = errno; + return (error); +} -- cgit v1.1