]> git.saurik.com Git - apple/xnu.git/blobdiff - osfmk/kern/thread_policy.c
xnu-2050.7.9.tar.gz
[apple/xnu.git] / osfmk / kern / thread_policy.c
index b04cf8f01d8d806ab87135cc0d5a54c760929059..8108514f547f3d0d3c530dfb82b96eef45208b30 100644 (file)
 /*
- * Copyright (c) 2000 Apple Computer, Inc. All rights reserved.
+ * Copyright (c) 2000-2007 Apple Inc. All rights reserved.
  *
- * @APPLE_LICENSE_HEADER_START@
+ * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
  * 
- * The contents of this file constitute Original Code as defined in and
- * are subject to the Apple Public Source License Version 1.1 (the
- * "License").  You may not use this file except in compliance with the
- * License.  Please obtain a copy of the License at
- * http://www.apple.com/publicsource and read it before using this file.
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. The rights granted to you under the License
+ * may not be used to create, or enable the creation or redistribution of,
+ * unlawful or unlicensed copies of an Apple operating system, or to
+ * circumvent, violate, or enable the circumvention or violation of, any
+ * terms of an Apple operating system software license agreement.
  * 
- * This Original Code and all software distributed under the License are
- * distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this file.
+ * 
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
- * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT.  Please see the
- * License for the specific language governing rights and limitations
- * under the License.
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
  * 
- * @APPLE_LICENSE_HEADER_END@
- */
-/*
- * Copyright (c) 2000 Apple Computer, Inc.  All rights reserved.
- *
- * HISTORY
- *
- * 15 October 2000 (debo)
- *  Created.
+ * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  */
 
+#include <mach/mach_types.h>
+#include <mach/thread_act_server.h>
+
+#include <kern/kern_types.h>
+#include <kern/processor.h>
 #include <kern/thread.h>
+#include <kern/affinity.h>
+
+static void
+thread_recompute_priority(
+       thread_t                thread);
+
+#if CONFIG_EMBEDDED
+static void
+thread_throttle(
+       thread_t                thread,
+       integer_t               task_priority);
+
+extern int mach_do_background_thread(thread_t thread, int prio);
+#endif
+
 
 kern_return_t
 thread_policy_set(
-       thread_act_t                    act,
+       thread_t                                thread,
        thread_policy_flavor_t  flavor,
        thread_policy_t                 policy_info,
        mach_msg_type_number_t  count)
 {
-       kern_return_t                   result = KERN_SUCCESS;
-       thread_t                                thread;
-       task_t                                  task;
-       spl_t                                   s;
 
-       if (act == THR_ACT_NULL)
+       if (thread == THREAD_NULL)
                return (KERN_INVALID_ARGUMENT);
 
-       act_lock(act);
-       task = act->task;
-       act_unlock(act);
-
-       task_lock(task);
+       if (thread->static_param)
+               return (KERN_SUCCESS);
 
-       thread = act_lock_thread(act);
-       if (!act->active) {
-               act_unlock_thread(act);
-               task_unlock(task);
+       return (thread_policy_set_internal(thread, flavor, policy_info, count));
+}
 
-               return (KERN_TERMINATED);
-       }
+kern_return_t
+thread_policy_set_internal(
+       thread_t                                thread,
+       thread_policy_flavor_t  flavor,
+       thread_policy_t                 policy_info,
+       mach_msg_type_number_t  count)
+{
+       kern_return_t                   result = KERN_SUCCESS;
+       spl_t                                   s;
 
-       if (thread == THREAD_NULL) {
-               act_unlock_thread(act);
-               task_unlock(task);
+       thread_mtx_lock(thread);
+       if (!thread->active) {
+               thread_mtx_unlock(thread);
 
-               return (KERN_NOT_SUPPORTED);
+               return (KERN_TERMINATED);
        }
-
-#define        thread_priority_set(thread, pri)                                        \
-MACRO_BEGIN                                                                                                    \
-       if ((thread)->depress_priority >= 0)                                    \
-               (thread)->depress_priority = (pri);                                     \
-       else {                                                                                                  \
-               (thread)->priority = (pri);                                                     \
-               compute_priority((thread), TRUE);                                       \
-                                                                                                                       \
-               if ((thread) == current_thread())                                       \
-                       ast_on(AST_BLOCK);                                                              \
-       }                                                                                                               \
-MACRO_END
-
        switch (flavor) {
 
-       case THREAD_STANDARD_POLICY:
+       case THREAD_EXTENDED_POLICY:
        {
-               integer_t                               priority;
+               boolean_t                               timeshare = TRUE;
+
+               if (count >= THREAD_EXTENDED_POLICY_COUNT) {
+                       thread_extended_policy_t        info;
 
+                       info = (thread_extended_policy_t)policy_info;
+                       timeshare = info->timeshare;
+               }
+
+               if (!SCHED(supports_timeshare_mode)())
+                       timeshare = FALSE;
+               
                s = splsched();
                thread_lock(thread);
 
-               thread->sched_mode &=~ TH_MODE_REALTIME;
+               if (!(thread->sched_flags & TH_SFLAG_DEMOTED_MASK)) {
+                       integer_t       oldmode = (thread->sched_mode == TH_MODE_TIMESHARE);
 
-               thread->policy = POLICY_TIMESHARE;
+                       if (timeshare) {
+                               thread->sched_mode = TH_MODE_TIMESHARE;
 
-               if (thread->importance > MAXPRI)
-                       priority = MAXPRI;
-               else
-               if (thread->importance < -MAXPRI)
-                       priority = -MAXPRI;
-               else
-                       priority = thread->importance;
+                               if (!oldmode) {
+                                       if ((thread->state & (TH_RUN|TH_IDLE)) == TH_RUN)
+                                               sched_share_incr();
+                               }
+                       }
+                       else {
+                               thread->sched_mode = TH_MODE_FIXED;
 
-               priority += task->priority;
+                               if (oldmode) {
+                                       if ((thread->state & (TH_RUN|TH_IDLE)) == TH_RUN)
+                                               sched_share_decr();
+                               }
+                       }
 
-               if (priority > thread->max_priority)
-                       priority = thread->max_priority;
-               else
-               if (priority < MINPRI)
-                       priority = MINPRI;
+                       thread_recompute_priority(thread);
+               }
+               else {
 
-               thread_priority_set(thread, priority);
+                       if (timeshare)
+                               thread->saved_mode = TH_MODE_TIMESHARE;
+                       else
+                               thread->saved_mode = TH_MODE_FIXED;
+               }
 
                thread_unlock(thread);
                splx(s);
@@ -126,20 +146,32 @@ MACRO_END
                }
 
                info = (thread_time_constraint_policy_t)policy_info;
+               if (    info->constraint < info->computation    ||
+                               info->computation > max_rt_quantum              ||
+                               info->computation < min_rt_quantum              ) {
+                       result = KERN_INVALID_ARGUMENT;
+                       break;
+               }
 
                s = splsched();
                thread_lock(thread);
 
-               thread->sched_mode |= TH_MODE_REALTIME;
-
                thread->realtime.period = info->period;
                thread->realtime.computation = info->computation;
                thread->realtime.constraint = info->constraint;
                thread->realtime.preemptible = info->preemptible;
 
-               thread->policy = POLICY_RR;
-
-               thread_priority_set(thread, BASEPRI_REALTIME);
+               if (thread->sched_flags & TH_SFLAG_DEMOTED_MASK) {
+                       thread->saved_mode = TH_MODE_REALTIME;
+               }
+               else {
+                       if (thread->sched_mode == TH_MODE_TIMESHARE) {
+                               if ((thread->state & (TH_RUN|TH_IDLE)) == TH_RUN)
+                                       sched_share_decr();
+                       }
+                       thread->sched_mode = TH_MODE_REALTIME;
+                       thread_recompute_priority(thread);
+               }
 
                thread_unlock(thread);
                splx(s);
@@ -155,7 +187,6 @@ MACRO_END
                        result = KERN_INVALID_ARGUMENT;
                        break;
                }
-
                info = (thread_precedence_policy_t)policy_info;
 
                s = splsched();
@@ -163,86 +194,242 @@ MACRO_END
 
                thread->importance = info->importance;
 
-               if (!(thread->sched_mode & TH_MODE_REALTIME)) {
-                       integer_t                                       priority;
+               thread_recompute_priority(thread);
 
-                       if (thread->importance > MAXPRI)
-                               priority = MAXPRI;
-                       else
-                       if (thread->importance < -MAXPRI)
-                               priority = -MAXPRI;
-                       else
-                               priority = thread->importance;
+               thread_unlock(thread);
+               splx(s);
 
-                       priority += task->priority;
+               break;
+       }
 
-                       if (priority > thread->max_priority)
-                               priority = thread->max_priority;
-                       else
-                       if (priority < MINPRI)
-                               priority = MINPRI;
+       case THREAD_AFFINITY_POLICY:
+       {
+               thread_affinity_policy_t        info;
 
-                       thread_priority_set(thread, priority);
+               if (!thread_affinity_is_supported()) {
+                       result = KERN_NOT_SUPPORTED;
+                       break;
+               }
+               if (count < THREAD_AFFINITY_POLICY_COUNT) {
+                       result = KERN_INVALID_ARGUMENT;
+                       break;
                }
 
-               thread_unlock(thread);
-               splx(s);
+               info = (thread_affinity_policy_t) policy_info;
+               /*
+                * Unlock the thread mutex here and
+                * return directly after calling thread_affinity_set().
+                * This is necessary for correct lock ordering because
+                * thread_affinity_set() takes the task lock.
+                */
+               thread_mtx_unlock(thread);
+               return thread_affinity_set(thread, info->affinity_tag);
+       }
 
-               break;
+#if CONFIG_EMBEDDED
+       case THREAD_BACKGROUND_POLICY:
+       {
+               thread_background_policy_t      info;
+
+               info = (thread_background_policy_t) policy_info;
+
+               thread_mtx_unlock(thread);
+               return mach_do_background_thread(thread, info->priority);
        }
+#endif /* CONFIG_EMBEDDED */
 
        default:
                result = KERN_INVALID_ARGUMENT;
                break;
        }
 
-       act_unlock_thread(act);
+       thread_mtx_unlock(thread);
+       return (result);
+}
 
-       task_unlock(task);
+static void
+thread_recompute_priority(
+       thread_t                thread)
+{
+       integer_t               priority;
 
-       return (result);
+       if (thread->sched_mode == TH_MODE_REALTIME)
+               priority = BASEPRI_RTQUEUES;
+       else {
+               if (thread->importance > MAXPRI)
+                       priority = MAXPRI;
+               else
+               if (thread->importance < -MAXPRI)
+                       priority = -MAXPRI;
+               else
+                       priority = thread->importance;
+
+               priority += thread->task_priority;
+
+               if (priority > thread->max_priority)
+                       priority = thread->max_priority;
+               else
+               if (priority < MINPRI)
+                       priority = MINPRI;
+#if CONFIG_EMBEDDED
+               /* No one can have a base priority less than MAXPRI_THROTTLE */
+               if (priority < MAXPRI_THROTTLE) 
+                       priority = MAXPRI_THROTTLE;
+#endif /* CONFIG_EMBEDDED */
+       }
+
+       set_priority(thread, priority);
+}
+
+#if CONFIG_EMBEDDED
+static void
+thread_throttle(
+       thread_t                thread,
+       integer_t               task_priority)
+{
+       if ((!(thread->sched_flags & TH_SFLAG_THROTTLED)
+                || (thread->sched_flags & TH_SFLAG_PENDING_THROTTLE_PROMOTION))
+                && (task_priority <= MAXPRI_THROTTLE)) {
+
+               /* Kill a promotion if it was in flight */
+               thread->sched_flags &= ~TH_SFLAG_PENDING_THROTTLE_PROMOTION;
+
+               if (!(thread->sched_flags & TH_SFLAG_THROTTLED)) {
+                       /*
+                        * Set the pending bit so that we can switch runqueues
+                        * (potentially) at a later time safely
+                        */
+                       thread->sched_flags |= TH_SFLAG_PENDING_THROTTLE_DEMOTION;
+               }
+       }
+       else if (((thread->sched_flags & TH_SFLAG_THROTTLED)
+                         || (thread->sched_flags & TH_SFLAG_PENDING_THROTTLE_DEMOTION))
+                         && (task_priority > MAXPRI_THROTTLE)) {
+
+               /* Kill a demotion if it was in flight */
+               thread->sched_flags &= ~TH_SFLAG_PENDING_THROTTLE_DEMOTION;
+
+               if (thread->sched_flags & TH_SFLAG_THROTTLED) {
+                       thread->sched_flags |= TH_SFLAG_PENDING_THROTTLE_PROMOTION;
+               }
+       }       
+}
+#endif
+
+void
+thread_task_priority(
+       thread_t                thread,
+       integer_t               priority,
+       integer_t               max_priority)
+{
+       spl_t                           s;
+
+       assert(thread != THREAD_NULL);
+
+       s = splsched();
+       thread_lock(thread);
+
+#if CONFIG_EMBEDDED
+       thread_throttle(thread, priority);
+#endif
+
+       thread->task_priority = priority;
+       thread->max_priority = max_priority;
+
+       thread_recompute_priority(thread);
+
+       thread_unlock(thread);
+       splx(s);
+}
+
+void
+thread_policy_reset(
+       thread_t                thread)
+{
+       spl_t           s;
+
+       s = splsched();
+       thread_lock(thread);
+
+       if (!(thread->sched_flags & TH_SFLAG_DEMOTED_MASK)) {
+               sched_mode_t oldmode = thread->sched_mode;
+               
+               thread->sched_mode = SCHED(initial_thread_sched_mode)(thread->task);
+
+               if ((oldmode != TH_MODE_TIMESHARE) && (thread->sched_mode == TH_MODE_TIMESHARE)) {
+
+                       if ((thread->state & (TH_RUN|TH_IDLE)) == TH_RUN)
+                               sched_share_incr();
+               }
+       }
+       else {
+               thread->sched_mode = thread->saved_mode;
+               thread->saved_mode = TH_MODE_NONE;
+               thread->sched_flags &= ~TH_SFLAG_DEMOTED_MASK;
+       }
+
+       thread->importance = 0;
+
+       thread_recompute_priority(thread);
+
+       thread_unlock(thread);
+       splx(s);
 }
 
 kern_return_t
 thread_policy_get(
-       thread_act_t                    act,
+       thread_t                                thread,
        thread_policy_flavor_t  flavor,
        thread_policy_t                 policy_info,
        mach_msg_type_number_t  *count,
        boolean_t                               *get_default)
 {
        kern_return_t                   result = KERN_SUCCESS;
-       thread_t                                thread;
        spl_t                                   s;
 
-       if (act == THR_ACT_NULL)
+       if (thread == THREAD_NULL)
                return (KERN_INVALID_ARGUMENT);
 
-       thread = act_lock_thread(act);
-       if (!act->active) {
-               act_unlock_thread(act);
+       thread_mtx_lock(thread);
+       if (!thread->active) {
+               thread_mtx_unlock(thread);
 
                return (KERN_TERMINATED);
        }
 
-       if (thread == THREAD_NULL) {
-               act_unlock_thread(act);
+       switch (flavor) {
 
-               return (KERN_NOT_SUPPORTED);
-       }
+       case THREAD_EXTENDED_POLICY:
+       {
+               boolean_t               timeshare = TRUE;
 
-       switch (flavor) {
+               if (!(*get_default)) {
+                       s = splsched();
+                       thread_lock(thread);
 
-       case THREAD_STANDARD_POLICY:
-               s = splsched();
-               thread_lock(thread);
+                       if (     (thread->sched_mode != TH_MODE_REALTIME)       &&
+                                        (thread->saved_mode != TH_MODE_REALTIME)                       ) {
+                               if (!(thread->sched_flags & TH_SFLAG_DEMOTED_MASK))
+                                       timeshare = (thread->sched_mode == TH_MODE_TIMESHARE) != 0;
+                               else
+                                       timeshare = (thread->saved_mode == TH_MODE_TIMESHARE) != 0;
+                       }
+                       else
+                               *get_default = TRUE;
 
-               if (thread->sched_mode & TH_MODE_REALTIME)
-                       *get_default = TRUE;
+                       thread_unlock(thread);
+                       splx(s);
+               }
+
+               if (*count >= THREAD_EXTENDED_POLICY_COUNT) {
+                       thread_extended_policy_t        info;
+
+                       info = (thread_extended_policy_t)policy_info;
+                       info->timeshare = timeshare;
+               }
 
-               thread_unlock(thread);
-               splx(s);
                break;
+       }
 
        case THREAD_TIME_CONSTRAINT_POLICY:
        {
@@ -255,29 +442,31 @@ thread_policy_get(
 
                info = (thread_time_constraint_policy_t)policy_info;
 
-               s = splsched();
-               thread_lock(thread);
+               if (!(*get_default)) {
+                       s = splsched();
+                       thread_lock(thread);
 
-               if ((thread->sched_mode & TH_MODE_REALTIME) && !(*get_default)) {
-                       info->period = thread->realtime.period;
-                       info->computation = thread->realtime.computation;
-                       info->constraint = thread->realtime.constraint;
-                       info->preemptible = thread->realtime.preemptible;
-               }
-               else {
-                       extern natural_t                min_quantum_abstime;
+                       if (    (thread->sched_mode == TH_MODE_REALTIME)        ||
+                                       (thread->saved_mode == TH_MODE_REALTIME)                ) {
+                               info->period = thread->realtime.period;
+                               info->computation = thread->realtime.computation;
+                               info->constraint = thread->realtime.constraint;
+                               info->preemptible = thread->realtime.preemptible;
+                       }
+                       else
+                               *get_default = TRUE;
 
-                       *get_default = TRUE;
+                       thread_unlock(thread);
+                       splx(s);
+               }
 
+               if (*get_default) {
                        info->period = 0;
-                       info->computation = min_quantum_abstime / 2;
-                       info->constraint = min_quantum_abstime;
+                       info->computation = default_timeshare_computation;
+                       info->constraint = default_timeshare_constraint;
                        info->preemptible = TRUE;
                }
 
-               thread_unlock(thread);
-               splx(s);
-
                break;
        }
 
@@ -292,9 +481,7 @@ thread_policy_get(
 
                info = (thread_precedence_policy_t)policy_info;
 
-               if (*get_default)
-                       info->importance = 0;
-               else {
+               if (!(*get_default)) {
                        s = splsched();
                        thread_lock(thread);
 
@@ -303,6 +490,31 @@ thread_policy_get(
                        thread_unlock(thread);
                        splx(s);
                }
+               else
+                       info->importance = 0;
+
+               break;
+       }
+
+       case THREAD_AFFINITY_POLICY:
+       {
+               thread_affinity_policy_t                info;
+
+               if (!thread_affinity_is_supported()) {
+                       result = KERN_NOT_SUPPORTED;
+                       break;
+               }
+               if (*count < THREAD_AFFINITY_POLICY_COUNT) {
+                       result = KERN_INVALID_ARGUMENT;
+                       break;
+               }
+
+               info = (thread_affinity_policy_t)policy_info;
+
+               if (!(*get_default))
+                       info->affinity_tag = thread_affinity_get(thread);
+               else
+                       info->affinity_tag = THREAD_AFFINITY_TAG_NULL;
 
                break;
        }
@@ -312,7 +524,7 @@ thread_policy_get(
                break;
        }
 
-       act_unlock_thread(act);
+       thread_mtx_unlock(thread);
 
        return (result);
 }