score: Add and use _Scheduler_Get_context()

3 files changed, 6 insertions, 5 deletions

diff --git a/cpukit/score/src/schedulerprioritysmp.c b/cpukit/score/src/schedulerprioritysmp.c
index 98ff454564..8133956fbd 100644
--- a/cpukit/score/src/schedulerprioritysmp.c
+++ b/cpukit/score/src/schedulerprioritysmp.c
@@ -32,7 +32,7 @@
 static Scheduler_priority_SMP_Context *
 _Scheduler_priority_SMP_Get_context( const Scheduler_Control *scheduler )
 {
-  return (Scheduler_priority_SMP_Context *) scheduler->context;
+  return (Scheduler_priority_SMP_Context *) _Scheduler_Get_context( scheduler );
 }
 
 static Scheduler_priority_SMP_Context *
diff --git a/cpukit/score/src/schedulersimplesmp.c b/cpukit/score/src/schedulersimplesmp.c
index 9861cb877d..8436659df4 100644
--- a/cpukit/score/src/schedulersimplesmp.c
+++ b/cpukit/score/src/schedulersimplesmp.c
@@ -25,7 +25,7 @@
 static Scheduler_simple_SMP_Context *
 _Scheduler_simple_SMP_Get_context( const Scheduler_Control *scheduler )
 {
-  return (Scheduler_simple_SMP_Context *) scheduler->context;
+  return (Scheduler_simple_SMP_Context *) _Scheduler_Get_context( scheduler );
 }
 
 static Scheduler_simple_SMP_Context *
diff --git a/cpukit/score/src/smp.c b/cpukit/score/src/smp.c
index af0eb183e1..b9d4b0a6cc 100644
--- a/cpukit/score/src/smp.c
+++ b/cpukit/score/src/smp.c
@@ -58,10 +58,11 @@ static void _SMP_Start_processors( uint32_t cpu_count )
     cpu->started = started;
 
     if ( started ) {
-      Scheduler_Context *scheduler_context = assignment->scheduler->context;
+      Scheduler_Context *context =
+        _Scheduler_Get_context( assignment->scheduler );
 
-      ++scheduler_context->processor_count;
-      cpu->scheduler_context = scheduler_context;
+      ++context->processor_count;
+      cpu->scheduler_context = context;
     }
   }
 }