rtems/src/tasksetaffinity.c

0001 /* SPDX-License-Identifier: BSD-2-Clause */
0002
0003 /**
0004  * @file
0005  *
0006  * @ingroup RTEMSImplClassicTask
0007  *
0008  * @brief This source file contains the implementation of
0009  *   rtems_task_set_affinity().
0010  */
0011
0012 /*
0013  *  COPYRIGHT (c) 2014.
0014  *  On-Line Applications Research Corporation (OAR).
0015  *
0016  * Redistribution and use in source and binary forms, with or without
0017  * modification, are permitted provided that the following conditions
0018  * are met:
0019  * 1. Redistributions of source code must retain the above copyright
0020  *    notice, this list of conditions and the following disclaimer.
0021  * 2. Redistributions in binary form must reproduce the above copyright
0022  *    notice, this list of conditions and the following disclaimer in the
0023  *    documentation and/or other materials provided with the distribution.
0024  *
0025  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
0026  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
0027  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
0028  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
0029  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
0030  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
0031  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
0032  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
0033  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
0034  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
0035  * POSSIBILITY OF SUCH DAMAGE.
0036  */
0037
0038 #ifdef HAVE_CONFIG_H
0039 #include "config.h"
0040 #endif
0041
0042 #include <rtems/rtems/tasks.h>
0043 #include <rtems/rtems/statusimpl.h>
0044 #include <rtems/score/threadimpl.h>
0045 #include <rtems/score/schedulerimpl.h>
0046
0047 rtems_status_code rtems_task_set_affinity(
0048   rtems_id         id,
0049   size_t           cpusetsize,
0050   const cpu_set_t *cpuset
0051 )
0052 {
0053   Thread_Control   *the_thread;
0054   ISR_lock_Context  lock_context;
0055   Per_CPU_Control  *cpu_self;
0056   Status_Control    status;
0057
0058   if ( cpuset == NULL ) {
0059     return RTEMS_INVALID_ADDRESS;
0060   }
0061
0062   the_thread = _Thread_Get( id, &lock_context );
0063
0064   if ( the_thread == NULL ) {
0065 #if defined(RTEMS_MULTIPROCESSING)
0066     if ( _Thread_MP_Is_remote( id ) ) {
0067       return RTEMS_ILLEGAL_ON_REMOTE_OBJECT;
0068     }
0069 #endif
0070
0071     return RTEMS_INVALID_ID;
0072   }
0073
0074   cpu_self = _Thread_Dispatch_disable_critical( &lock_context );
0075   _Thread_State_acquire_critical( the_thread, &lock_context );
0076
0077   status = _Scheduler_Set_affinity(
0078     the_thread,
0079     cpusetsize,
0080     cpuset
0081   );
0082
0083   _Thread_State_release( the_thread, &lock_context );
0084   _Thread_Dispatch_enable( cpu_self );
0085   return _Status_Get( status );
0086 }