score: Add cpuset support to Score.

This new Score Handler provides a structure to manage a
cpu_set_t plus helper routines to validate the contents
against the current system configuration.

3 files changed, 184 insertions, 1 deletions

diff --git a/cpukit/score/src/cpuset.c b/cpukit/score/src/cpuset.c
new file mode 100644
index 0000000000..4cf24288b1
--- /dev/null
+++ b/cpukit/score/src/cpuset.c
@@ -0,0 +1,94 @@
+/**
+ * @file
+ *
+ * @ingroup ScoreCpuset
+ *
+ * @brief Routines to Control a CPU Set.
+ */
+
+/*
+ *  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
+
+#ifdef __RTEMS_HAVE_SYS_CPUSET_H__
+#include <sys/cpuset.h>
+#include <rtems/score/cpusetimpl.h>
+#include <rtems/score/assert.h>
+#include <rtems/score/percpu.h>
+
+static CPU_set_Control cpuset_default;
+
+/*
+ * _CPU_set_Handler_initialization
+ */
+void _CPU_set_Handler_initialization()
+{
+  int i;
+  int max_cpus;
+
+  /* We do not support a cpu count over CPU_SETSIZE  */
+  max_cpus = _SMP_Get_processor_count();
+
+  /* This should never happen */
+  _Assert( max_cpus <= CPU_SETSIZE );
+
+  /*  Initialize the affinity to be the set of all available CPU's   */
+  cpuset_default.set     = &cpuset_default.preallocated;
+  cpuset_default.setsize = sizeof( *cpuset_default.set );
+  CPU_ZERO_S( cpuset_default.setsize, &cpuset_default.preallocated );
+
+  for (i=0; i<max_cpus; i++)  
+    CPU_SET_S(i, cpuset_default.setsize, cpuset_default.set );
+}
+
+/**
+ * _CPU_set_Is_valid
+ *
+ * This routine validates a cpuset size corresponds to
+ * the system correct size, that at least one
+ * valid cpu is set and that no invalid cpus are set.
+ */
+int _CPU_set_Is_valid( const cpu_set_t *cpuset, size_t setsize )
+{
+  cpu_set_t             temp;
+
+  if ( !cpuset )
+    return -1;
+
+  if (setsize != cpuset_default.setsize )
+    return -1;
+
+  /* Validate at least 1 valid cpu is set in cpuset */
+  CPU_AND_S( cpuset_default.setsize, &temp, cpuset, cpuset_default.set );
+
+  if ( CPU_COUNT_S( setsize, &temp ) == 0 )
+    return -1;
+
+  /* Validate that no invalid cpu's are set in cpuset */
+  if ( !CPU_EQUAL_S( setsize, &temp, cpuset ) )
+    return -1;
+
+  return 0;
+}
+
+/**
+ * _CPU_set_Default
+ *
+ * This routine returns the default cpuset.
+ */
+const CPU_set_Control *_CPU_set_Default() 
+{
+  return &cpuset_default;
+}
+#endif
+
+
diff --git a/cpukit/score/src/cpusetprintsupport.c b/cpukit/score/src/cpusetprintsupport.c
new file mode 100644
index 0000000000..a56305e443
--- /dev/null
+++ b/cpukit/score/src/cpusetprintsupport.c
@@ -0,0 +1,84 @@
+/**
+ * @file
+ *
+ * @brief CPU Set Print Support Routines
+ * @ingroup ScoreCpuset
+ */
+
+/*
+ *  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.
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include <string.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <ctype.h>
+#include <inttypes.h>
+#include <rtems/bspIo.h>
+#include <rtems/score/cpusetimpl.h>
+
+#ifdef __RTEMS_HAVE_SYS_CPUSET_H__
+
+  void _CPU_set_Show_with_plugin(
+    void                  *context,
+    rtems_printk_plugin_t  print,
+    const char            *description, 
+    const cpu_set_t       *cpuset
+  );
+
+  /*
+   * _CPU_set_Show_with_plugin
+   *
+   * This routine shows cpuset cpuset using a
+   * print plugin .
+   */
+  void _CPU_set_Show_with_plugin(
+    void                  *context,
+    rtems_printk_plugin_t  print,
+    const char            *description, 
+    const cpu_set_t       *cpuset
+  )
+  {
+    int i;
+
+    if ( !print )
+      return;
+
+    (*print)(context ,"%s: ", description);
+    for(i=0; i<_NCPUWORDS; i++)
+      (*print)(context ,"%x", cpuset->__bits[i]);
+    (*print)(context ,"\n");
+  }
+
+  /*
+   * _CPU_set_Show
+   *
+   * This routine shows a cpuset using the
+   * printk plugin.
+   */
+  void _CPU_set_Show( const char *description, const cpu_set_t   *cpuset)
+  {
+    _CPU_set_Show_with_plugin( NULL, printk_plugin, description, cpuset );
+  }
+
+  /*
+   * _CPU_set_Show_default
+   *
+   * This routine shows the default cpuset.
+   */
+ void _CPU_set_Show_default( const char *description )
+  {
+    const CPU_set_Control *ctl;
+    ctl = _CPU_set_Default();
+    _CPU_set_Show( description, ctl->set );  
+  }
+#endif
diff --git a/cpukit/score/src/threadinitialize.c b/cpukit/score/src/threadinitialize.c
index 94ff8650c2..acd6636ac8 100644
--- a/cpukit/score/src/threadinitialize.c
+++ b/cpukit/score/src/threadinitialize.c
@@ -5,7 +5,7 @@
  *  @ingroup ScoreThread
  */
 /*
- *  COPYRIGHT (c) 1989-2011.
+ *  COPYRIGHT (c) 1989-2014.
  *  On-Line Applications Research Corporation (OAR).
  *
  *  The license and distribution terms for this file may be
@@ -24,6 +24,7 @@
 #include <rtems/score/userextimpl.h>
 #include <rtems/score/watchdogimpl.h>
 #include <rtems/score/wkspace.h>
+#include <rtems/score/cpusetimpl.h>
 #include <rtems/config.h>
 
 bool _Thread_Initialize(
@@ -206,6 +207,10 @@ bool _Thread_Initialize(
 
   /* Initialize the cpu field for the non-SMP schedulers */
   the_thread->cpu                     = _Per_CPU_Get_by_index( 0 );
+#if __RTEMS_HAVE_SYS_CPUSET_H__
+   the_thread->affinity               = *(_CPU_set_Default());
+   the_thread->affinity.set           = &the_thread->affinity.preallocated;
+#endif
 #endif
 
   the_thread->current_state           = STATES_DORMANT;