X-Git-Url: https://git.saurik.com/apple/xnu.git/blobdiff_plain/813fb2f63a553c957e917ede5f119b021d6ce391..cc8bc92ae4a8e9f1a1ab61bf83d34ad8150b3405:/osfmk/kern/syscall_subr.c?ds=inline

diff --git a/osfmk/kern/syscall_subr.c b/osfmk/kern/syscall_subr.c
index a0c1d38e3..e944c79d9 100644
--- 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;