Enable xTaskGetCurrentTaskHandleForCore() for single core builds (#978)

Enable xTaskGetCurrentTaskHandleForCore() for single core builds --------- Co-authored-by: Paul Bartell <pbartell@amazon.com> Co-authored-by: Ching-Hsin Lee <chinglee@amazon.com>
FreeRTOS · Feb 2, 2024 · 1c35cb3 · 1c35cb3
1 parent 1189198
commit 1c35cb3
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 14 deletions.
diff --git a/include/task.h b/include/task.h
@@ -3574,9 +3574,7 @@ TaskHandle_t xTaskGetCurrentTaskHandle( void ) PRIVILEGED_FUNCTION;
 /*
  * Return the handle of the task running on specified core.
  */
-#if ( configNUMBER_OF_CORES > 1 )
-    TaskHandle_t xTaskGetCurrentTaskHandleForCore( BaseType_t xCoreID ) PRIVILEGED_FUNCTION;
-#endif
+TaskHandle_t xTaskGetCurrentTaskHandleForCore( BaseType_t xCoreID ) PRIVILEGED_FUNCTION;
 
 /*
  * Shortcut used by the queue implementation to prevent unnecessary call to

diff --git a/tasks.c b/tasks.c
@@ -6559,23 +6559,27 @@ static void prvResetNextTaskUnblockTime( void )
 
             return xReturn;
         }
+    #endif /* #if ( configNUMBER_OF_CORES == 1 ) */
 
-        TaskHandle_t xTaskGetCurrentTaskHandleForCore( BaseType_t xCoreID )
-        {
-            TaskHandle_t xReturn = NULL;
+    TaskHandle_t xTaskGetCurrentTaskHandleForCore( BaseType_t xCoreID )
+    {
+        TaskHandle_t xReturn = NULL;
 
-            traceENTER_xTaskGetCurrentTaskHandleForCore( xCoreID );
+        traceENTER_xTaskGetCurrentTaskHandleForCore( xCoreID );
 
-            if( taskVALID_CORE_ID( xCoreID ) != pdFALSE )
-            {
+        if( taskVALID_CORE_ID( xCoreID ) != pdFALSE )
+        {
+            #if ( configNUMBER_OF_CORES == 1 )
+                xReturn = pxCurrentTCB;
+            #else /* #if ( configNUMBER_OF_CORES == 1 ) */
                 xReturn = pxCurrentTCBs[ xCoreID ];
-            }
+            #endif /* #if ( configNUMBER_OF_CORES == 1 ) */
+        }
 
-            traceRETURN_xTaskGetCurrentTaskHandleForCore( xReturn );
+        traceRETURN_xTaskGetCurrentTaskHandleForCore( xReturn );
 
-            return xReturn;
-        }
-    #endif /* #if ( configNUMBER_OF_CORES == 1 ) */
+        return xReturn;
+    }
 
 #endif /* ( ( INCLUDE_xTaskGetCurrentTaskHandle == 1 ) || ( configUSE_MUTEXES == 1 ) ) */
 /*-----------------------------------------------------------*/