diff --git a/cpukit/rtems/src/workspacegreedy.c b/cpukit/rtems/src/workspacegreedy.c index ad46df2561..ecd4b86577 100644 --- a/cpukit/rtems/src/workspacegreedy.c +++ b/cpukit/rtems/src/workspacegreedy.c @@ -24,6 +24,7 @@ #endif #include +#include #include #include #include @@ -35,9 +36,9 @@ void *rtems_workspace_greedy_allocate( { void *opaque; - _Thread_Disable_dispatch(); + _RTEMS_Lock_allocator(); opaque = _Heap_Greedy_allocate( &_Workspace_Area, block_sizes, block_count ); - _Thread_Enable_dispatch(); + _RTEMS_Unlock_allocator(); return opaque; } @@ -48,19 +49,19 @@ void *rtems_workspace_greedy_allocate_all_except_largest( { void *opaque; - _Thread_Disable_dispatch(); + _RTEMS_Lock_allocator(); opaque = _Heap_Greedy_allocate_all_except_largest( &_Workspace_Area, allocatable_size ); - _Thread_Enable_dispatch(); + _RTEMS_Unlock_allocator(); return opaque; } void rtems_workspace_greedy_free( void *opaque ) { - _Thread_Disable_dispatch(); + _RTEMS_Lock_allocator(); _Heap_Greedy_free( &_Workspace_Area, opaque ); - _Thread_Enable_dispatch(); + _RTEMS_Unlock_allocator(); }