1 files changed, 85 insertions, 0 deletions
diff --git a/schedsim/shell/shared/main_tasksetaffinity.c b/schedsim/shell/shared/main_tasksetaffinity.c
new file mode 100644
index 0000000..5d4eb41
--- /dev/null
+++ b/schedsim/shell/shared/main_tasksetaffinity.c
@@ -0,0 +1,85 @@
+/**
+ *  @file
+ *
+ *  Task Set Affinity Shell Command Implmentation
+ */
+
+/*
+ *  COPYRIGHT (c) 1989-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.com/license/LICENSE.
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#define _GNU_SOURCE
+#include <sys/cpuset.h>
+
+#include <stdio.h>
+
+#include <rtems.h>
+#include "shell.h"
+#include <rtems/stringto.h>
+#include <schedsim_shell.h>
+#include <rtems/error.h>
+
+int rtems_shell_main_task_set_affinity(
+  int   argc,
+  char *argv[]
+)
+{
+  rtems_id             id;
+  rtems_status_code    status;
+  unsigned long        tmp;
+  cpu_set_t            cpuset;
+ 
+  CHECK_RTEMS_IS_UP();
+
+  if (argc != 3) {
+    fprintf( stderr, "%s: Usage [name|id] affinity\n", argv[0] );
+    return -1;
+  }
+
+  if ( lookup_task( argv[1], &id ) )
+    return -1;
+
+  if ( rtems_string_to_unsigned_long( argv[2], &tmp, NULL, 0) ) {
+    fprintf( stderr, "Argument (%s) is not a number\n", argv[2] );
+    return 1;
+  }
+
+  CPU_ZERO( &cpuset );
+  cpuset.__bits[0] = tmp;
+
+  /*
+   *  Now change the affinity of the task
+   */
+  status = rtems_task_set_affinity( id, sizeof(cpuset), &cpuset );
+  if ( status != RTEMS_SUCCESSFUL ) {
+    fprintf(
+      stderr,
+      "Task Set Affinity(%s) returned %s\n",
+      argv[1],
+      rtems_status_text( status )
+    ); 
+    return -1;
+  }
+
+  printf("Task (0x%08x) Set affinity=0x%08x\n", id, cpuset.__bits[0] );
+  
+  return 0;
+}
+
+rtems_shell_cmd_t rtems_shell_TASK_SET_AFFINITY_Command = {
+  "task_set_affinity",                 /* name */
+  "task_set_affinity name ",           /* usage */
+  "rtems",                             /* topic */
+  rtems_shell_main_task_set_affinity,  /* command */
+  NULL,                                /* alias */
+  NULL                                 /* next */
+};