From 2290d0385e0b6753b279c3fe7e9020cca552488b Mon Sep 17 00:00:00 2001 From: Angus Gratton Date: Mon, 19 Apr 2021 15:59:44 +1000 Subject: [PATCH] freertos: Fix race condition returning incorrect TCB on unpinned tasks Noted as a problem with thread local storage returning a different task's pointers, but some other were APIs also accessing current task unsafely. Regression in FreeRTOS 10 update a3c90bf59add307d3a41f5fb4c17385ee67dcf20 --- components/freertos/tasks.c | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/components/freertos/tasks.c b/components/freertos/tasks.c index 07f1e2b387..95180b79ae 100644 --- a/components/freertos/tasks.c +++ b/components/freertos/tasks.c @@ -231,7 +231,12 @@ count overflows. */ * task should be used in place of the parameter. This macro simply checks to * see if the parameter is NULL and returns a pointer to the appropriate TCB. */ -#define prvGetTCBFromHandle( pxHandle ) ( ( ( pxHandle ) == NULL ) ? (TaskHandle_t)pxCurrentTCB[xPortGetCoreID()] : ( (TaskHandle_t)pxHandle ) ) +#if portNUM_PROCESSORS > 1 +/* In SMP, we need to disable interrupts if getting the current task handle outside a critical section. Calling xTaskGetCurrentTaskHandle() ensures this. */ +#define prvGetTCBFromHandle( pxHandle ) ( ( ( pxHandle ) == NULL ) ? xTaskGetCurrentTaskHandle() : ( (TaskHandle_t)pxHandle ) ) +#else +#define prvGetTCBFromHandle( pxHandle ) ( ( ( pxHandle ) == NULL ) ? (TaskHandle_t) pxCurrentTCB[0] : ( (TaskHandle_t)pxHandle ) ) +#endif /* The item value of the event list item is normally used to hold the priority of the task to which it belongs (coded to allow it to be held in reverse @@ -4465,7 +4470,7 @@ TCB_t *pxTCB; } /*-----------------------------------------------------------*/ -#if ( ( INCLUDE_xTaskGetCurrentTaskHandle == 1 ) || ( configUSE_MUTEXES == 1 ) ) +#if ( ( INCLUDE_xTaskGetCurrentTaskHandle == 1 ) || ( configUSE_MUTEXES == 1 ) || (portNUM_PROCESSORS > 1) ) TaskHandle_t xTaskGetCurrentTaskHandle( void ) {