From: richardbarry <richardbarry@1d2547de-c912-0410-9cb9-b8ca96c0e9e2>
Date: Mon, 25 Feb 2008 18:54:28 +0000 (+0000)
Subject: Add vTaskEnterCritical() and vTaskExitCritical() functions.
X-Git-Tag: V4.8.0~60
X-Git-Url: https://git.sur5r.net/?a=commitdiff_plain;h=134514ed58151f700aae3059c25e3f453f5f7106;p=freertos

Add vTaskEnterCritical() and vTaskExitCritical() functions.

git-svn-id: https://svn.code.sf.net/p/freertos/code/trunk@221 1d2547de-c912-0410-9cb9-b8ca96c0e9e2
---

diff --git a/Source/include/FreeRTOS.h b/Source/include/FreeRTOS.h
index ee963a4fa..f9139ffca 100644
--- a/Source/include/FreeRTOS.h
+++ b/Source/include/FreeRTOS.h
@@ -134,6 +134,10 @@
 	#define configUSE_ALTERNATIVE_API 0
 #endif
 
+#ifndef portCRITICAL_NESTING_IN_TCB
+	#define portCRITICAL_NESTING_IN_TCB 0
+#endif
+
 #if ( configUSE_MUTEXES == 1 )
 	/* xTaskGetCurrentTaskHandle is used by the priority inheritance mechanism
 	within the mutex implementation so must be available if mutexes are used. */
diff --git a/Source/tasks.c b/Source/tasks.c
index cc43e7e5b..1c1013a1c 100644
--- a/Source/tasks.c
+++ b/Source/tasks.c
@@ -266,6 +266,10 @@ typedef struct tskTaskControlBlock
 	portSTACK_TYPE			*pxStack;			/*< Points to the start of the stack. */
 	signed portCHAR			pcTaskName[ configMAX_TASK_NAME_LEN ];/*< Descriptive name given to the task when created.  Facilitates debugging only. */
 
+	#if ( portCRITICAL_NESTING_IN_TCB == 1 )
+		unsigned portBASE_TYPE uxCriticalNesting;
+	#endif
+
 	#if ( configUSE_TRACE_FACILITY == 1 )
 		unsigned portBASE_TYPE	uxTCBNumber;		/*< This is used for tracing the scheduler and making debugging easier only. */
 	#endif	
@@ -1755,6 +1759,12 @@ static void prvInitialiseTCBVariables( tskTCB *pxTCB, const signed portCHAR * co
 	strncpy( ( char * ) pxTCB->pcTaskName, ( const char * ) pcName, ( unsigned portSHORT ) configMAX_TASK_NAME_LEN );
 	pxTCB->pcTaskName[ ( unsigned portSHORT ) configMAX_TASK_NAME_LEN - ( unsigned portSHORT ) 1 ] = '\0';
 
+	#if ( portCRITICAL_NESTING_IN_TCB == 1 )
+	{
+		pxTCB->uxCriticalNesting = ( unsigned portBASE_TYPE ) 0;
+	}
+	#endif
+
 	/* This is used as an array index so must ensure it's not too large. */
 	if( uxPriority >= configMAX_PRIORITIES )
 	{
@@ -1983,6 +1993,7 @@ tskTCB *pxNewTCB;
 	}
 
 #endif
+/*-----------------------------------------------------------*/
 
 #if ( configUSE_MUTEXES == 1 )
 	
@@ -2014,6 +2025,7 @@ tskTCB *pxNewTCB;
 	}
 
 #endif
+/*-----------------------------------------------------------*/
 
 #if ( configUSE_MUTEXES == 1 )	
 
@@ -2039,8 +2051,44 @@ tskTCB *pxNewTCB;
 	}
 
 #endif
+/*-----------------------------------------------------------*/
+
+#if ( portCRITICAL_NESTING_IN_TCB == 1 )
+
+	void vTaskEnterCritical( void )
+	{
+		portDISABLE_INTERRUPTS();
+
+		if( xSchedulerRunning != pdFALSE )
+		{
+			pxCurrentTCB->uxCriticalNesting++;
+		}
+	}
+
+#endif
+/*-----------------------------------------------------------*/
+
+#if ( portCRITICAL_NESTING_IN_TCB == 1 )
+
+void vTaskExitCritical( void )
+{
+	if( xSchedulerRunning != pdFALSE )
+	{
+		if( pxCurrentTCB->uxCriticalNesting > 0 )
+		{
+			pxCurrentTCB->uxCriticalNesting--;
+
+			if( pxCurrentTCB->uxCriticalNesting == 0 )
+			{
+				portENABLE_INTERRUPTS();
+			}
+		}
+	}
+}
+
+#endif
+/*-----------------------------------------------------------*/
 
-