LXR 6.1/​testsuites/​smptests/​smpschedaffinity01/​init.c	Source navigation Diff markup Identifier search General search
	Version: 6.1 Revert   Change

File indexing completed on 2025-05-11 08:24:43

 /* SPDX-License-Identifier: BSD-2-Clause */
 
 /*
  *  COPYRIGHT (c) 2014.
  *  On-Line Applications Research Corporation (OAR).
  *
  * 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, 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 COPYRIGHT HOLDERS AND CONTRIBUTORS "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 COPYRIGHT OWNER OR CONTRIBUTORS 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.
  */
 
 /*
  * Start 4 tasks with affinity for each of the 4 cpus.
  * Allow tasks to set their actual cpu value and delete themselves.
  * Verify the actual cpu values match the expected cpu values.
  *
  * Init task is at a lower priority 8 and the threads
  * with affinity are at priority 4, so the affinity task
  * on the core init is running on will preempt it.
  *
  * Test tasks run and delete themselves. 
  * Init task never blocks.
  */
 
 #ifdef HAVE_CONFIG_H
 #include "config.h"
 #endif
 
 #include <rtems.h>
 
 #include "tmacros.h"
 
 const char rtems_test_name[] = "SMPSCHEDAFFINITY 1";
 
 #define NUM_CPUS   4
 #define TASK_COUNT 5
 
 struct task_data_t {
   rtems_id   id;
   int        expected_cpu;
   cpu_set_t  cpuset;
   bool       ran;
   int        actual_cpu;
 };
 
 static struct task_data_t task_data[NUM_CPUS] = {
   {0x0, 2, {{0x4}}, false, -1},
   {0x0, 0, {{0x1}}, false, -1},
   {0x0, 3, {{0x8}}, false, -1},
   {0x0, 1, {{0x2}}, false, -1}
 };
   
 /*
  * Spin loop to allow tasks to delay without yeilding the
  * processor.
  */
 static void test_delay(int ticks)
 { 
   rtems_interval start, stop;
   start = rtems_clock_get_ticks_since_boot();
   do {
     stop = rtems_clock_get_ticks_since_boot();
   } while ( (stop - start) < ticks );
 }
 
 static void task(rtems_task_argument arg)
 {
   uint32_t cpu;
   cpu_set_t cpuset;
 
   cpu = rtems_scheduler_get_processor();
 
   rtems_task_get_affinity( rtems_task_self(), sizeof(cpuset), &cpuset );
 
   task_data[arg].ran = true;
   task_data[arg].actual_cpu = cpu;
 
   rtems_task_exit();
 }
 
 static void test(void)
 {
   rtems_status_code   sc;
   rtems_task_argument i;
   size_t              size;
   uint32_t            cpu_count;
 
   /* Get the number of processors that we are using. */
   cpu_count = rtems_scheduler_get_processor_maximum();
  
   size = sizeof(cpu_set_t);
 
   /* Create and start tasks on each cpu with the appropriate affinity. */
   for (i = 0; i < NUM_CPUS; i++) {
 
       /* Skip if this cpu doesn't exist, don't create task */
       if ( task_data[i].expected_cpu >= cpu_count ) {
         printf(
           "Skipping TA0%" PRIdrtems_task_argument
             " because on a core we do not have\n",
           i
         );
         continue;
       }
 
       sc = rtems_task_create(
         rtems_build_name('T', 'A', '0', '0'+i),
         4,
         RTEMS_MINIMUM_STACK_SIZE,
         RTEMS_DEFAULT_MODES,
         RTEMS_DEFAULT_ATTRIBUTES,
         &task_data[ i ].id
       );
       rtems_test_assert(sc == RTEMS_SUCCESSFUL);
 
       printf(
         "Set TA%" PRIdrtems_task_argument " affinity to cpu %d\n",
         i,
         task_data[i].expected_cpu
       );
       sc = rtems_task_set_affinity( task_data[ i ].id, size, &task_data[i].cpuset );
       rtems_test_assert(sc == RTEMS_SUCCESSFUL);
 
       printf(
         "Start TA%" PRIdrtems_task_argument " on cpu %d\n",
         i,
         task_data[i].expected_cpu
       );
       sc = rtems_task_start( task_data[ i ].id, task, i );
       rtems_test_assert(sc == RTEMS_SUCCESSFUL);
   }
 
   /* spin for 100 ticks */
   test_delay(100);
 
   printf("Verify Tasks Ran\n");
   for (i = 0; i < NUM_CPUS; i++) {
 
     /* Skip if this cpu doesn't exist, task doesn't exist */
     if ( task_data[i].expected_cpu >= cpu_count ) {
       printf(
         "Skipping TA0%" PRIdrtems_task_argument
           " because on a core we do not have\n",
         i
       );
       continue;
     }
 
     /* print the expected and actual values */
     printf(
       "TA0%" PRIdrtems_task_argument ": ran=%d expected=%d actual=%d\n",
       i,
       task_data[i].ran,
       task_data[i].expected_cpu,
       task_data[i].actual_cpu
     );
 
     /*  Abort test if values are not as expected */
     rtems_test_assert( task_data[i].ran == true );
     rtems_test_assert( task_data[i].expected_cpu == task_data[i].actual_cpu );
   }
 }
 
 static void Init(rtems_task_argument arg)
 {
   TEST_BEGIN();
 
   test();
 
   TEST_END();
   rtems_test_exit(0);
 }
 
 #define CONFIGURE_APPLICATION_NEEDS_CLOCK_DRIVER
 #define CONFIGURE_APPLICATION_NEEDS_SIMPLE_CONSOLE_DRIVER
 
 #define CONFIGURE_SCHEDULER_PRIORITY_AFFINITY_SMP
 
 #define CONFIGURE_MAXIMUM_PROCESSORS NUM_CPUS
 
 #define CONFIGURE_MAXIMUM_TASKS          TASK_COUNT
 
 #define CONFIGURE_INITIAL_EXTENSIONS RTEMS_TEST_INITIAL_EXTENSION
 
   #define CONFIGURE_INIT_TASK_PRIORITY      8
 #define CONFIGURE_RTEMS_INIT_TASKS_TABLE
 
 #define CONFIGURE_INIT
 
 #include <rtems/confdefs.h>

This page was automatically generated by the 2.3.7 LXR engine. The LXR team