/* 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>