diff options
Diffstat (limited to 'pthreads/pthread_setaffinity.c')
| -rwxr-xr-x | pthreads/pthread_setaffinity.c | 225 |
1 files changed, 225 insertions, 0 deletions
diff --git a/pthreads/pthread_setaffinity.c b/pthreads/pthread_setaffinity.c new file mode 100755 index 000000000..62fc4945a --- /dev/null +++ b/pthreads/pthread_setaffinity.c @@ -0,0 +1,225 @@ +/*
+ * pthread_setaffinity.c
+ *
+ * Description:
+ * This translation unit implements thread cpu affinity setting.
+ *
+ * --------------------------------------------------------------------------
+ *
+ * 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
+ */
+
+#ifdef HAVE_CONFIG_H
+# include <config.h>
+#endif
+
+#include "pthread.h"
+#include "implement.h"
+
+int
+pthread_setaffinity_np (pthread_t thread, size_t cpusetsize,
+ const cpu_set_t *cpuset)
+ /*
+ * ------------------------------------------------------
+ * DOCPUBLIC
+ * The pthread_setaffinity_np() function sets the CPU affinity mask
+ * of the thread thread to the CPU set pointed to by cpuset. If the
+ * call is successful, and the thread is not currently running on one
+ * of the CPUs in cpuset, then it is migrated to one of those CPUs.
+ *
+ * PARAMETERS
+ * thread
+ * The target thread
+ *
+ * cpusetsize
+ * Ignored in pthreads4w.
+ * Usually set to sizeof(cpu_set_t)
+ *
+ * cpuset
+ * The new cpu set mask.
+ *
+ * The set of CPUs on which the thread will actually run
+ * is the intersection of the set specified in the cpuset
+ * argument and the set of CPUs actually present for
+ * the process.
+ *
+ * DESCRIPTION
+ * The pthread_setaffinity_np() function sets the CPU affinity mask
+ * of the thread thread to the CPU set pointed to by cpuset. If the
+ * call is successful, and the thread is not currently running on one
+ * of the CPUs in cpuset, then it is migrated to one of those CPUs.
+ *
+ * RESULTS
+ * 0 Success
+ * ESRCH Thread does not exist
+ * EFAULT pcuset is NULL
+ * EAGAIN The thread affinity could not be set
+ *
+ * ------------------------------------------------------
+ */
+{
+ int result = 0;
+ ptw32_thread_t * tp;
+ ptw32_mcs_local_node_t node;
+ cpu_set_t processCpuset;
+
+ ptw32_mcs_lock_acquire (&ptw32_thread_reuse_lock, &node);
+
+ tp = (ptw32_thread_t *) thread.p;
+
+ if (NULL == tp || thread.x != tp->ptHandle.x || NULL == tp->threadH)
+ {
+ result = ESRCH;
+ }
+ else
+ {
+ if (cpuset)
+ {
+ if (sched_getaffinity(0, sizeof(cpu_set_t), &processCpuset))
+ {
+ result = PTW32_GET_ERRNO();
+ }
+ else
+ {
+ /*
+ * Result is the intersection of available CPUs and the mask.
+ */
+ cpu_set_t newMask;
+
+ CPU_AND(&newMask, &processCpuset, cpuset);
+
+ if (((_sched_cpu_set_vector_*)&newMask)->_cpuset)
+ {
+ if (SetThreadAffinityMask (tp->threadH, ((_sched_cpu_set_vector_*)&newMask)->_cpuset))
+ {
+ /*
+ * We record the intersection of the process affinity
+ * and the thread affinity cpusets so that
+ * pthread_getaffinity_np() returns the actual thread
+ * CPU set.
+ */
+ tp->cpuset = ((_sched_cpu_set_vector_*)&newMask)->_cpuset;
+ }
+ else
+ {
+ result = EAGAIN;
+ }
+ }
+ else
+ {
+ result = EINVAL;
+ }
+ }
+ }
+ else
+ {
+ result = EFAULT;
+ }
+ }
+
+ ptw32_mcs_lock_release (&node);
+
+ return result;
+}
+
+int
+pthread_getaffinity_np (pthread_t thread, size_t cpusetsize, cpu_set_t *cpuset)
+ /*
+ * ------------------------------------------------------
+ * DOCPUBLIC
+ * The pthread_getaffinity_np() function returns the CPU affinity mask
+ * of the thread thread in the CPU set pointed to by cpuset.
+ *
+ * PARAMETERS
+ * thread
+ * The target thread
+ *
+ * cpusetsize
+ * Ignored in pthreads4w.
+ * Usually set to sizeof(cpu_set_t)
+ *
+ * cpuset
+ * The location where the current cpu set
+ * will be returned.
+ *
+ *
+ * DESCRIPTION
+ * The pthread_getaffinity_np() function returns the CPU affinity mask
+ * of the thread thread in the CPU set pointed to by cpuset.
+ *
+ * RESULTS
+ * 0 Success
+ * ESRCH thread does not exist
+ * EFAULT cpuset is NULL
+ *
+ * ------------------------------------------------------
+ */
+{
+ int result = 0;
+ ptw32_thread_t * tp;
+ ptw32_mcs_local_node_t node;
+
+ ptw32_mcs_lock_acquire(&ptw32_thread_reuse_lock, &node);
+
+ tp = (ptw32_thread_t *) thread.p;
+
+ if (NULL == tp || thread.x != tp->ptHandle.x || NULL == tp->threadH)
+ {
+ result = ESRCH;
+ }
+ else
+ {
+ if (cpuset)
+ {
+ if (tp->cpuset)
+ {
+ /*
+ * The application may have set thread affinity independently
+ * via SetThreadAffinityMask(). If so, we adjust our record of the threads
+ * affinity and try to do so in a reasonable way.
+ */
+ DWORD_PTR vThreadMask = SetThreadAffinityMask(tp->threadH, tp->cpuset);
+ if (vThreadMask && vThreadMask != tp->cpuset)
+ {
+ (void) SetThreadAffinityMask(tp->threadH, vThreadMask);
+ tp->cpuset = vThreadMask;
+ }
+ }
+ ((_sched_cpu_set_vector_*)cpuset)->_cpuset = tp->cpuset;
+ }
+ else
+ {
+ result = EFAULT;
+ }
+ }
+
+ ptw32_mcs_lock_release(&node);
+
+ return result;
+}
|