diff options
Diffstat (limited to 'testsuites/smptests/smpschedaffinity05/init.c')
-rw-r--r-- | testsuites/smptests/smpschedaffinity05/init.c | 243 |
1 files changed, 243 insertions, 0 deletions
diff --git a/testsuites/smptests/smpschedaffinity05/init.c b/testsuites/smptests/smpschedaffinity05/init.c new file mode 100644 index 0000000000..754c226687 --- /dev/null +++ b/testsuites/smptests/smpschedaffinity05/init.c @@ -0,0 +1,243 @@ +/* + * 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. + */ + +/* + * 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 5"; + +#define NUM_CPUS 4 +#define TASK_COUNT 5 + +struct task_data_t { + rtems_id id; + cpu_set_t cpuset; + rtems_task_priority priority; + bool ran; + int expected_cpu; + int actual_cpu; + int migrate_cpu; +}; + +static struct task_data_t task_data[TASK_COUNT] = { + {0x0, {{0xc}}, 7, false, 3, -1, 2}, + {0x0, {{0xf}}, 8, false, 2, -1, -1}, + {0x0, {{0x3}}, 5, false, 1, -1, 0}, + {0x0, {{0x9}}, 6, false, 0, -1, 3}, + {0x0, {{0x2}}, 9, false, -1, -1, 1} +}; + +rtems_id task_sem; + +static void verify_tasks(void); + +/* + * 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) +{ + rtems_status_code sc; + + while (true) { + sc = rtems_semaphore_obtain (task_sem, RTEMS_NO_WAIT, 0); + if (sc == RTEMS_SUCCESSFUL) { + task_data[arg].ran = true; + task_data[arg].actual_cpu = rtems_get_current_processor(); + rtems_semaphore_release(task_sem); + } + } +} +static void verify_tasks(void) +{ + int i; + + printf("Verify Tasks Ran\n"); + + while( rtems_semaphore_obtain (task_sem, RTEMS_NO_WAIT, 0) != RTEMS_SUCCESSFUL ); + + /* Set Init task data */ + task_data[0].ran = true; + task_data[0].actual_cpu = rtems_get_current_processor(); + + /* Verify all tasks */ + for (i = 0; i < NUM_CPUS; i++) { + if (i==0) + printf("Init(%d): ran=%d expected=%d actual=%d\n", + task_data[i].priority, + task_data[i].ran, + task_data[i].expected_cpu, + task_data[i].actual_cpu + ); + else + printf( "TA0%d(%d): ran=%d expected=%d actual=%d\n", + i, + task_data[i].priority, + task_data[i].ran, + task_data[i].expected_cpu, + task_data[i].actual_cpu + ); + + /* Abort test if values are not as expected */ + if ( task_data[i].expected_cpu == -1 ) + rtems_test_assert( task_data[i].ran == false ); + else { + rtems_test_assert( task_data[i].ran == true ); + rtems_test_assert( task_data[i].expected_cpu == task_data[i].actual_cpu ); + } + } + + rtems_semaphore_release(task_sem); +} + +static void test(void) +{ + rtems_status_code sc; + rtems_task_argument i; + size_t size; + uint32_t cpu_count; + rtems_task_priority priority; + + /* Get the number of processors that we are using. */ + cpu_count = rtems_get_processor_count(); + if (cpu_count != 4) { + printf("Test requires a minimum of 4 cores\n"); + return; + } + + size = sizeof(cpu_set_t); + task_data[0].id = rtems_task_self(); + printf("Create Semaphore\n"); + + sc = rtems_semaphore_create( + rtems_build_name('S', 'E', 'M', '0'), + 1, /* initial count = 1 */ + RTEMS_LOCAL | + RTEMS_SIMPLE_BINARY_SEMAPHORE | + RTEMS_NO_INHERIT_PRIORITY | + RTEMS_NO_PRIORITY_CEILING | + RTEMS_FIFO, + 0, + &task_sem + ); + rtems_test_assert(sc == RTEMS_SUCCESSFUL); + + + /* Create and start tasks on each cpu with the appropriate affinity. */ + for (i = 1; i < TASK_COUNT; i++) { + + sc = rtems_task_create( + rtems_build_name('T', 'A', '0', '0'+i), + task_data[ i ].priority, + RTEMS_MINIMUM_STACK_SIZE, + RTEMS_DEFAULT_MODES, + RTEMS_DEFAULT_ATTRIBUTES, + &task_data[ i ].id + ); + rtems_test_assert(sc == RTEMS_SUCCESSFUL); + + sc = rtems_task_set_affinity( + task_data[ i ].id, + size, + &task_data[i].cpuset + ); + rtems_test_assert(sc == RTEMS_SUCCESSFUL); + + printf( + "Start TA%d at priority %d on cpu %d\n", + i, + task_data[i].priority, + 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); + + verify_tasks(); + + i = TASK_COUNT - 1; + task_data[ i ].priority = 4; + printf("Set TA%d priority %d\n", i,task_data[i].priority ); + sc = rtems_task_set_priority( + task_data[ i ].id, + task_data[ i ].priority, + &priority + ); + test_delay(25); + + while( rtems_semaphore_obtain (task_sem, RTEMS_NO_WAIT, 0) != RTEMS_SUCCESSFUL ); + for (i = 0; i < TASK_COUNT; i++) { + task_data[ i ].expected_cpu = task_data[ i ].migrate_cpu; + task_data[ i ].actual_cpu = -1; + task_data[ i ].ran = false; + } + rtems_semaphore_release(task_sem); + test_delay(25); + verify_tasks(); +} + +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_CONSOLE_DRIVER + +#define CONFIGURE_SMP_APPLICATION + +#define CONFIGURE_SCHEDULER_PRIORITY_AFFINITY_SMP + +#define CONFIGURE_SMP_MAXIMUM_PROCESSORS NUM_CPUS + +#define CONFIGURE_MAXIMUM_TASKS TASK_COUNT + +#define CONFIGURE_INITIAL_EXTENSIONS RTEMS_TEST_INITIAL_EXTENSION + +#define CONFIGURE_INIT_TASK_PRIORITY 7 +#define CONFIGURE_RTEMS_INIT_TASKS_TABLE + +#define CONFIGURE_INIT + +#include <rtems/confdefs.h> |