xnu-4570.20.62.tar.gz

[apple/xnu.git] / osfmk / kern / syscall_subr.c

diff --git a/osfmk/kern/syscall_subr.c b/osfmk/kern/syscall_subr.c
index a0c1d38e32911ceea1f56c7b665bcb56cb1c20c3..e944c79d94956f64150b36fb3fa67025a689a17e 100644 (file)
--- a/osfmk/kern/syscall_subr.c
+++ b/osfmk/kern/syscall_subr.c
@@ -109,11 +109,11 @@ static void
 swtch_continue(void)
 {
        processor_t     myprocessor;
-    boolean_t                          result;
+       boolean_t       result;
 
-    disable_preemption();
+       disable_preemption();
        myprocessor = current_processor();
-       result = !SCHED(processor_queue_empty)(myprocessor) || rt_runq.count > 0;
+       result = SCHED(thread_should_yield)(myprocessor, current_thread());
        enable_preemption();
 
        thread_syscall_return(result);
@@ -129,7 +129,7 @@ swtch(
 
        disable_preemption();
        myprocessor = current_processor();
-       if (SCHED(processor_queue_empty)(myprocessor) &&        rt_runq.count == 0) {
+       if (!SCHED(thread_should_yield)(myprocessor, current_thread())) {
                mp_enable_preemption();
 
                return (FALSE);
@@ -142,7 +142,7 @@ swtch(
 
        disable_preemption();
        myprocessor = current_processor();
-       result = !SCHED(processor_queue_empty)(myprocessor) || rt_runq.count > 0;
+       result = SCHED(thread_should_yield)(myprocessor, current_thread());
        enable_preemption();
 
        return (result);
@@ -152,13 +152,13 @@ static void
 swtch_pri_continue(void)
 {
        processor_t     myprocessor;
-    boolean_t                          result;
+       boolean_t       result;
 
        thread_depress_abort_internal(current_thread());
 
-    disable_preemption();
+       disable_preemption();
        myprocessor = current_processor();
-       result = !SCHED(processor_queue_empty)(myprocessor) || rt_runq.count > 0;
+       result = SCHED(thread_should_yield)(myprocessor, current_thread());
        mp_enable_preemption();
 
        thread_syscall_return(result);
@@ -174,7 +174,7 @@ __unused    struct swtch_pri_args *args)
 
        disable_preemption();
        myprocessor = current_processor();
-       if (SCHED(processor_queue_empty)(myprocessor) && rt_runq.count == 0) {
+       if (!SCHED(thread_should_yield)(myprocessor, current_thread())) {
                mp_enable_preemption();
 
                return (FALSE);
@@ -191,7 +191,7 @@ __unused    struct swtch_pri_args *args)
 
        disable_preemption();
        myprocessor = current_processor();
-       result = !SCHED(processor_queue_empty)(myprocessor) || rt_runq.count > 0;
+       result = SCHED(thread_should_yield)(myprocessor, current_thread());
        enable_preemption();
 
        return (result);
@@ -496,6 +496,7 @@ thread_depress_abstime(
                                      0);
 
                myprocessor->current_pri = self->sched_pri;
+               myprocessor->current_perfctl_class = thread_get_perfcontrol_class(self);
                self->sched_flags |= TH_SFLAG_DEPRESS;
 
                if (interval != 0) {
@@ -599,6 +600,7 @@ thread_poll_yield(
                                                      0);
 
                                myprocessor->current_pri = self->sched_pri;
+                               myprocessor->current_perfctl_class = thread_get_perfcontrol_class(self);
                        }
                        self->computation_epoch = abstime;
                        self->computation_metered = 0;
@@ -626,7 +628,7 @@ thread_yield_internal(
 
        disable_preemption();
        myprocessor = current_processor();
-       if (SCHED(processor_queue_empty)(myprocessor) && rt_runq.count == 0) {
+       if (!SCHED(thread_should_yield)(myprocessor, current_thread())) {
                mp_enable_preemption();
 
                return;