]> git.saurik.com Git - apple/xnu.git/blobdiff - bsd/kern/kern_sfi.c
xnu-6153.81.5.tar.gz
[apple/xnu.git] / bsd / kern / kern_sfi.c
index a9787dfb97577ae8526925da8b5223f0f09ae26a..55e421bf1e6e801d64b56bdf501420284f964cfc 100644 (file)
@@ -2,7 +2,7 @@
  * Copyright (c) 2013 Apple Inc. All rights reserved.
  *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
- * 
+ *
  * 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
  * 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.
- * 
+ *
  * 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,
@@ -22,7 +22,7 @@
  * 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_OSREFERENCE_LICENSE_HEADER_END@
  *
  */
@@ -41,6 +41,7 @@
 #include <sys/sfi.h>
 #include <sys/kdebug.h>
 #include <sys/priv.h>
+#include <kern/policy_internal.h>
 
 /*
  * This file provides the syscall-based configuration facility
 
 static int proc_apply_sfi_managed(proc_t p, void * arg);
 
-int sfi_ctl(struct proc *p __unused, struct sfi_ctl_args *uap, int32_t *retval __unused)
+int
+sfi_ctl(struct proc *p __unused, struct sfi_ctl_args *uap, int32_t *retval __unused)
 {
-       uint32_t        operation = uap->operation;
-       int                     error = 0;
-       kern_return_t   kret = KERN_SUCCESS;
-       uint64_t        out_time = 0;
+       uint32_t        operation = uap->operation;
+       int                     error = 0;
+       kern_return_t   kret = KERN_SUCCESS;
+       uint64_t        out_time = 0;
 
        switch (operation) {
-               case SFI_CTL_OPERATION_SFI_SET_WINDOW:
-                       if (uap->out_time != USER_ADDR_NULL) {
-                               return EINVAL;
-                       }
-                       if (uap->sfi_class != SFI_CLASS_UNSPECIFIED) {
-                               return EINVAL;
-                       }
-
-                       error = priv_check_cred(kauth_cred_get(), PRIV_SELECTIVE_FORCED_IDLE, 0);
-                       if (error) {
-                               dprintf("%s failed privilege check for sfi_ctl: %d\n", p->p_comm, error);
-                               return (error);
-                       } else {
-                               dprintf("%s succeeded privilege check for sfi_ctl\n", p->p_comm);
-                       }
-
-                       if (uap->time == 0) {
-                               /* actually a cancel */
-                               kret = sfi_window_cancel();
-                       } else {
-                               kret = sfi_set_window(uap->time);
-                       }
-
-                       if (kret) {
-                               error = EINVAL;
-                       }
-
-                       break;
-               case SFI_CTL_OPERATION_SFI_GET_WINDOW:
-                       if (uap->time != 0) {
-                               return EINVAL;
-                       }
-                       if (uap->sfi_class != SFI_CLASS_UNSPECIFIED) {
-                               return EINVAL;
-                       }
-
-                       kret = sfi_get_window(&out_time);
-                       if (kret == KERN_SUCCESS) {
-                               error = copyout(&out_time, uap->out_time, sizeof(out_time));
-                       } else {
-                               error = EINVAL;
-                       }
-
-                       break;
-               case SFI_CTL_OPERATION_SET_CLASS_OFFTIME:
-                       if (uap->out_time != USER_ADDR_NULL) {
-                               return EINVAL;
-                       }
-
-                       error = priv_check_cred(kauth_cred_get(), PRIV_SELECTIVE_FORCED_IDLE, 0);
-                       if (error) {
-                               dprintf("%s failed privilege check for sfi_ctl: %d\n", p->p_comm, error);
-                               return (error);
-                       } else {
-                               dprintf("%s succeeded privilege check for sfi_ctl\n", p->p_comm);
-                       }
-
-                       if (uap->time == 0) {
-                               /* actually a cancel */
-                               kret = sfi_class_offtime_cancel(uap->sfi_class);
-                       } else {
-                               kret = sfi_set_class_offtime(uap->sfi_class, uap->time);
-                       }
-
-                       if (kret) {
-                               error = EINVAL;
-                       }
-
-                       break;
-               case SFI_CTL_OPERATION_GET_CLASS_OFFTIME:
-                       if (uap->time != 0) {
-                               return EINVAL;
-                       }
-
-                       kret = sfi_get_class_offtime(uap->sfi_class, &out_time);
-                       if (kret == KERN_SUCCESS) {
-                               error = copyout(&out_time, uap->out_time, sizeof(out_time));
-                       } else {
-                               error = EINVAL;
-                       }
-
-                       break;
-               default:
-                       error = ENOTSUP;
-                       break;
-       }                       
+       case SFI_CTL_OPERATION_SFI_SET_WINDOW:
+               if (uap->out_time != USER_ADDR_NULL) {
+                       return EINVAL;
+               }
+               if (uap->sfi_class != SFI_CLASS_UNSPECIFIED) {
+                       return EINVAL;
+               }
+
+               error = priv_check_cred(kauth_cred_get(), PRIV_SELECTIVE_FORCED_IDLE, 0);
+               if (error) {
+                       dprintf("%s failed privilege check for sfi_ctl: %d\n", p->p_comm, error);
+                       return error;
+               } else {
+                       dprintf("%s succeeded privilege check for sfi_ctl\n", p->p_comm);
+               }
+
+               if (uap->time == 0) {
+                       /* actually a cancel */
+                       kret = sfi_window_cancel();
+               } else {
+                       kret = sfi_set_window(uap->time);
+               }
+
+               if (kret) {
+                       error = EINVAL;
+               }
+
+               break;
+       case SFI_CTL_OPERATION_SFI_GET_WINDOW:
+               if (uap->time != 0) {
+                       return EINVAL;
+               }
+               if (uap->sfi_class != SFI_CLASS_UNSPECIFIED) {
+                       return EINVAL;
+               }
+
+               kret = sfi_get_window(&out_time);
+               if (kret == KERN_SUCCESS) {
+                       error = copyout(&out_time, uap->out_time, sizeof(out_time));
+               } else {
+                       error = EINVAL;
+               }
+
+               break;
+       case SFI_CTL_OPERATION_SET_CLASS_OFFTIME:
+               if (uap->out_time != USER_ADDR_NULL) {
+                       return EINVAL;
+               }
+
+               error = priv_check_cred(kauth_cred_get(), PRIV_SELECTIVE_FORCED_IDLE, 0);
+               if (error) {
+                       dprintf("%s failed privilege check for sfi_ctl: %d\n", p->p_comm, error);
+                       return error;
+               } else {
+                       dprintf("%s succeeded privilege check for sfi_ctl\n", p->p_comm);
+               }
+
+               if (uap->time == 0) {
+                       /* actually a cancel */
+                       kret = sfi_class_offtime_cancel(uap->sfi_class);
+               } else {
+                       kret = sfi_set_class_offtime(uap->sfi_class, uap->time);
+               }
+
+               if (kret) {
+                       error = EINVAL;
+               }
+
+               break;
+       case SFI_CTL_OPERATION_GET_CLASS_OFFTIME:
+               if (uap->time != 0) {
+                       return EINVAL;
+               }
+
+               kret = sfi_get_class_offtime(uap->sfi_class, &out_time);
+               if (kret == KERN_SUCCESS) {
+                       error = copyout(&out_time, uap->out_time, sizeof(out_time));
+               } else {
+                       error = EINVAL;
+               }
+
+               break;
+       default:
+               error = ENOTSUP;
+               break;
+       }
 
        return error;
 }
 
-static int proc_apply_sfi_managed(proc_t p, void * arg)
+static int
+proc_apply_sfi_managed(proc_t p, void * arg)
 {
        uint32_t flags = *(uint32_t *)arg;
        pid_t pid = p->p_pid;
        boolean_t managed_enabled = (flags == SFI_PROCESS_SET_MANAGED)? TRUE : FALSE;
-       
-       if (pid == 0) {         /* ignore setting on kernproc */
+
+       if (pid == 0) {         /* ignore setting on kernproc */
                return PROC_RETURNED;
        }
 
@@ -173,86 +176,88 @@ static int proc_apply_sfi_managed(proc_t p, void * arg)
        } else {
                KERNEL_DEBUG_CONSTANT(MACHDBG_CODE(DBG_MACH_SFI, SFI_PID_CLEAR_MANAGED) | DBG_FUNC_NONE, pid, 0, 0, 0, 0);
        }
-       
-       proc_set_task_policy(p->task, THREAD_NULL,
-                                                                TASK_POLICY_ATTRIBUTE, TASK_POLICY_SFI_MANAGED,
-                                                                managed_enabled ? TASK_POLICY_ENABLE : TASK_POLICY_DISABLE);
+
+       proc_set_task_policy(p->task,
+           TASK_POLICY_ATTRIBUTE, TASK_POLICY_SFI_MANAGED,
+           managed_enabled ? TASK_POLICY_ENABLE : TASK_POLICY_DISABLE);
+
        return PROC_RETURNED;
 }
 
-int sfi_pidctl(struct proc *p __unused, struct sfi_pidctl_args *uap, int32_t *retval __unused)
+int
+sfi_pidctl(struct proc *p __unused, struct sfi_pidctl_args *uap, int32_t *retval __unused)
 {
-       uint32_t        operation = uap->operation;
-       pid_t           pid = uap->pid;
-       int                     error = 0;
-       uint32_t        out_flags = 0;
-       boolean_t       managed_enabled;
-       proc_t          targetp;
+       uint32_t        operation = uap->operation;
+       pid_t           pid = uap->pid;
+       int                     error = 0;
+       uint32_t        out_flags = 0;
+       boolean_t       managed_enabled;
+       proc_t          targetp;
 
        switch (operation) {
-               case SFI_PIDCTL_OPERATION_PID_SET_FLAGS:
-                       if (uap->out_sfi_flags != USER_ADDR_NULL
-                           || !(uap->sfi_flags & SFI_PROCESS_SET_MANAGED_MASK)
-                           || uap->sfi_flags == SFI_PROCESS_SET_MANAGED_MASK) {
+       case SFI_PIDCTL_OPERATION_PID_SET_FLAGS:
+               if (uap->out_sfi_flags != USER_ADDR_NULL
+                   || !(uap->sfi_flags & SFI_PROCESS_SET_MANAGED_MASK)
+                   || uap->sfi_flags == SFI_PROCESS_SET_MANAGED_MASK) {
+                       return EINVAL;
+               }
+
+               error = priv_check_cred(kauth_cred_get(), PRIV_SELECTIVE_FORCED_IDLE, 0);
+               if (error) {
+                       dprintf("%s failed privilege check for sfi_pidctl: %d\n", p->p_comm, error);
+                       return error;
+               } else {
+                       dprintf("%s succeeded privilege check for sfi_pidctl\n", p->p_comm);
+               }
+
+               if (uap->pid == 0) {
+                       /* only allow SFI_PROCESS_SET_UNMANAGED for pid 0 */
+                       if (uap->sfi_flags != SFI_PROCESS_SET_UNMANAGED) {
                                return EINVAL;
                        }
 
-                       error = priv_check_cred(kauth_cred_get(), PRIV_SELECTIVE_FORCED_IDLE, 0);
-                       if (error) {
-                               dprintf("%s failed privilege check for sfi_pidctl: %d\n", p->p_comm, error);
-                               return (error);
-                       } else {
-                               dprintf("%s succeeded privilege check for sfi_pidctl\n", p->p_comm);
-                       }
+                       proc_iterate(PROC_ALLPROCLIST, proc_apply_sfi_managed, (void *)&uap->sfi_flags, NULL, NULL);
+                       break;
+               }
 
-                       if (uap->pid == 0) {
-                               /* only allow SFI_PROCESS_SET_UNMANAGED for pid 0 */
-                               if (uap->sfi_flags != SFI_PROCESS_SET_UNMANAGED) {
-                                       return EINVAL;
-                               }
+               targetp = proc_find(pid);
+               if (!targetp) {
+                       error = ESRCH;
+                       break;
+               }
 
-                               proc_iterate(PROC_ALLPROCLIST, proc_apply_sfi_managed, (void *)&uap->sfi_flags, NULL, NULL);
-                               break;
-                       }
+               proc_apply_sfi_managed(targetp, (void *)&uap->sfi_flags);
 
-                       targetp = proc_find(pid);
-                       if (!targetp) {
-                               error = ESRCH;
-                               break;
-                       }
-                       
-                       proc_apply_sfi_managed(targetp, (void *)&uap->sfi_flags);
-                       
-                       proc_rele(targetp);
+               proc_rele(targetp);
 
-                       break;
-               case SFI_PIDCTL_OPERATION_PID_GET_FLAGS:
-                       if (uap->sfi_flags != 0) {
-                               return EINVAL;
-                       }
-                       if (uap->pid == 0) {
-                               return EINVAL;
-                       }
+               break;
+       case SFI_PIDCTL_OPERATION_PID_GET_FLAGS:
+               if (uap->sfi_flags != 0) {
+                       return EINVAL;
+               }
+               if (uap->pid == 0) {
+                       return EINVAL;
+               }
 
-                       targetp = proc_find(pid);
-                       if (!targetp) {
-                               error = ESRCH;
-                               break;
-                       }
+               targetp = proc_find(pid);
+               if (!targetp) {
+                       error = ESRCH;
+                       break;
+               }
 
-                       managed_enabled = proc_get_task_policy(targetp->task, THREAD_NULL, TASK_POLICY_ATTRIBUTE, TASK_POLICY_SFI_MANAGED);
+               managed_enabled = proc_get_task_policy(targetp->task, TASK_POLICY_ATTRIBUTE, TASK_POLICY_SFI_MANAGED);
 
-                       proc_rele(targetp);
+               proc_rele(targetp);
 
-                       out_flags = managed_enabled ? SFI_PROCESS_SET_MANAGED : SFI_PROCESS_SET_UNMANAGED;
+               out_flags = managed_enabled ? SFI_PROCESS_SET_MANAGED : SFI_PROCESS_SET_UNMANAGED;
 
-                       error = copyout(&out_flags, uap->out_sfi_flags, sizeof(out_flags));
+               error = copyout(&out_flags, uap->out_sfi_flags, sizeof(out_flags));
 
-                       break;
-               default:
-                       error = ENOTSUP;
-                       break;
-       }                       
+               break;
+       default:
+               error = ENOTSUP;
+               break;
+       }
 
        return error;
 }