From dd0017c13510c51d979618c499d501417599abcc Mon Sep 17 00:00:00 2001
From: Jennifer Averett <jennifer.averett@oarcorp.com>
Date: Mon, 10 Feb 2014 10:01:31 -0600
Subject: posix: Add dynamic pthread get and set affinity.

This patch adds the following methods:

  + pthread_get_affinity_np
  + pthread_set_affinity_np
---
 cpukit/posix/Makefile.am                |  3 +-
 cpukit/posix/src/pthreadgetaffinitynp.c | 68 ++++++++++++++++++++++++++++++++
 cpukit/posix/src/pthreadsetaffinitynp.c | 69 +++++++++++++++++++++++++++++++++
 3 files changed, 139 insertions(+), 1 deletion(-)
 create mode 100644 cpukit/posix/src/pthreadgetaffinitynp.c
 create mode 100644 cpukit/posix/src/pthreadsetaffinitynp.c

diff --git a/cpukit/posix/Makefile.am b/cpukit/posix/Makefile.am
index 13f698c876..9dc8820507 100644
--- a/cpukit/posix/Makefile.am
+++ b/cpukit/posix/Makefile.am
@@ -145,7 +145,8 @@ libposix_a_SOURCES += src/pthreadattrcompare.c
 if HAS_SMP
 ## PTHREAD_AFFINITY_C_FILES 
 libposix_a_SOURCES += src/pthreadattrsetaffinitynp.c \
-    src/pthreadattrgetaffinitynp.c
+    src/pthreadattrgetaffinitynp.c  src/pthreadgetaffinitynp.c   \
+    src/pthreadsetaffinitynp.c
 endif
 
 ## PSIGNAL_C_FILES
diff --git a/cpukit/posix/src/pthreadgetaffinitynp.c b/cpukit/posix/src/pthreadgetaffinitynp.c
new file mode 100644
index 0000000000..e92f800a1d
--- /dev/null
+++ b/cpukit/posix/src/pthreadgetaffinitynp.c
@@ -0,0 +1,68 @@
+/**
+ * @file
+ *
+ * @brief Pthread Get Affinity
+ * @ingroup POSIXAPI
+ */
+
+/*
+ *  COPYRIGHT (c) 2014.
+ *  On-Line Applications Research Corporation (OAR).
+ *
+ *  The license and distribution terms for this file may be
+ *  found in the file LICENSE in this distribution or at
+ *  http://www.rtems.org/license/LICENSE.
+ */
+
+
+#if HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#if HAVE_DECL_PTHREAD_GETAFFINITY_NP
+
+#define  _GNU_SOURCE
+#include <pthread.h>
+#include <errno.h>
+
+#include <rtems/posix/pthreadimpl.h>
+#include <rtems/posix/priorityimpl.h>
+#include <rtems/score/threadimpl.h>
+
+int pthread_getaffinity_np(
+  const pthread_t       id,
+  size_t                cpusetsize,
+  cpu_set_t            *cpuset
+)
+{
+  Objects_Locations        location;
+  Thread_Control          *the_thread;
+  int                      error;
+
+  if ( !cpuset )
+    return EFAULT;
+ 
+  the_thread = _Thread_Get( id, &location );
+  switch ( location ) {
+
+    case OBJECTS_LOCAL:
+      error = 0;
+      if ( cpusetsize != the_thread->affinity.setsize )
+        error = EINVAL;
+      else
+        CPU_COPY( cpuset, the_thread->affinity.set );
+
+      _Objects_Put( &the_thread->Object );
+      return error;
+      break;
+
+#if defined(RTEMS_MULTIPROCESSING)
+    case OBJECTS_REMOTE:
+#endif
+    case OBJECTS_ERROR:
+      break;
+  }
+  return ESRCH;
+}
+
+#endif
diff --git a/cpukit/posix/src/pthreadsetaffinitynp.c b/cpukit/posix/src/pthreadsetaffinitynp.c
new file mode 100644
index 0000000000..f186935637
--- /dev/null
+++ b/cpukit/posix/src/pthreadsetaffinitynp.c
@@ -0,0 +1,69 @@
+/**
+ * @file
+ *
+ * @brief Pthread Set Affinity
+ * @ingroup POSIXAPI
+ */
+
+/*
+ *  COPYRIGHT (c) 2014.
+ *  On-Line Applications Research Corporation (OAR).
+ *
+ *  The license and distribution terms for this file may be
+ *  found in the file LICENSE in this distribution or at
+ *  http://www.rtems.org/license/LICENSE.
+ */
+
+#if HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#if HAVE_DECL_PTHREAD_SETAFFINITY_NP
+
+#define  _GNU_SOURCE
+#include <pthread.h>
+#include <errno.h>
+
+#include <rtems/posix/pthreadimpl.h>
+#include <rtems/posix/priorityimpl.h>
+#include <rtems/score/threadimpl.h>
+#include <rtems/score/cpusetimpl.h>
+
+int pthread_setaffinity_np(
+  pthread_t          id,
+  size_t             cpusetsize,
+  const cpu_set_t   *cpuset)
+{
+  Objects_Locations        location;
+  POSIX_API_Control       *api;
+  Thread_Control          *the_thread;
+  int                      error;
+
+  if ( !cpuset )
+    return EFAULT;
+
+  error = _CPU_set_Is_valid( cpuset, cpusetsize );
+  if ( error != 0 )
+    return EINVAL;
+   
+  the_thread = _Thread_Get( id, &location );
+  switch ( location ) {
+
+    case OBJECTS_LOCAL:
+      api = the_thread->API_Extensions[ THREAD_API_POSIX ];
+      CPU_COPY( the_thread->affinity.set, cpuset );
+      CPU_COPY( api->Attributes.affinityset, cpuset );
+      _Objects_Put( &the_thread->Object );
+      return 0;
+      break;
+
+#if defined(RTEMS_MULTIPROCESSING)
+    case OBJECTS_REMOTE:
+#endif
+    case OBJECTS_ERROR:
+      break;
+  }
+
+  return ESRCH;
+}
+#endif
-- 
cgit v1.2.3