xnu-1228.15.4.tar.gz

[apple/xnu.git] / bsd / kern / kern_control.c

diff --git a/bsd/kern/kern_control.c b/bsd/kern/kern_control.c
index e742a0debe7aefa9a4fa1f886602a6fc8c20f333..877fe67785b8021ab45ec4dbfb7e54f67711964f 100644 (file)
--- a/bsd/kern/kern_control.c
+++ b/bsd/kern/kern_control.c
@@ -1,5 +1,5 @@
 /*
 /*

- * Copyright (c) 1999-2004 Apple Computer, Inc. All rights reserved.
+ * Copyright (c) 1999-2006 Apple Computer, Inc. All rights reserved.

  *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
  * 
  *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
  * 


@@ -48,6 +48,7 @@
 #include <sys/sys_domain.h>
 #include <sys/kern_event.h>
 #include <sys/kern_control.h>
 #include <sys/sys_domain.h>
 #include <sys/kern_event.h>
 #include <sys/kern_control.h>

+#include <sys/kauth.h>

 #include <net/if_var.h>
 
 #include <mach/vm_types.h>
 #include <net/if_var.h>
 
 #include <mach/vm_types.h>


@@ -66,8 +67,6 @@
  * Definitions and vars for we support
  */
 
  * Definitions and vars for we support
  */
 

-static u_int32_t               ctl_last_id = 0;
-static u_int32_t               ctl_max = 256;

 static u_int32_t               ctl_maxunit = 65536;
 static lck_grp_attr_t  *ctl_lck_grp_attr = 0;
 static lck_attr_t              *ctl_lck_attr = 0;
 static u_int32_t               ctl_maxunit = 65536;
 static lck_grp_attr_t  *ctl_lck_grp_attr = 0;
 static lck_attr_t              *ctl_lck_attr = 0;


@@ -76,7 +75,7 @@ static lck_mtx_t              *ctl_mtx;
 
 
 /* all the controllers are chained */
 
 
 /* all the controllers are chained */

-TAILQ_HEAD(, kctl)     ctl_head;
+TAILQ_HEAD(kctl_list, kctl)    ctl_head;

 
 static int ctl_attach(struct socket *, int, struct proc *);
 static int ctl_detach(struct socket *);
 
 static int ctl_attach(struct socket *, int, struct proc *);
 static int ctl_detach(struct socket *);


@@ -90,7 +89,6 @@ static int ctl_send(struct socket *, int, struct mbuf *,
 static int ctl_ctloutput(struct socket *, struct sockopt *);
 static int ctl_peeraddr(struct socket *so, struct sockaddr **nam);
 
 static int ctl_ctloutput(struct socket *, struct sockopt *);
 static int ctl_peeraddr(struct socket *so, struct sockaddr **nam);
 

-static struct kctl *ctl_find_by_id(u_int32_t);

 static struct kctl *ctl_find_by_name(const char *);
 static struct kctl *ctl_find_by_id_unit(u_int32_t id, u_int32_t unit);
 
 static struct kctl *ctl_find_by_name(const char *);
 static struct kctl *ctl_find_by_id_unit(u_int32_t id, u_int32_t unit);
 


@@ -262,6 +260,7 @@ ctl_sofreelastref(struct socket *so)
        }
        kcb_delete(kcb);
     }
        }
        kcb_delete(kcb);
     }

+    sofreelastref(so, 1);

     return 0;
 }
 
     return 0;
 }
 


@@ -286,6 +285,7 @@ ctl_connect(struct socket *so, struct sockaddr *nam, __unused struct proc *p)
     int                                        error = 0;
     struct sockaddr_ctl        sa;
     struct ctl_cb              *kcb = (struct ctl_cb *)so->so_pcb;
     int                                        error = 0;
     struct sockaddr_ctl        sa;
     struct ctl_cb              *kcb = (struct ctl_cb *)so->so_pcb;

+    struct ctl_cb              *kcb_next = NULL;

     
     if (kcb == 0)
        panic("ctl_connect so_pcb null\n");
     
     if (kcb == 0)
        panic("ctl_connect so_pcb null\n");


@@ -313,9 +313,9 @@ ctl_connect(struct socket *so, struct sockaddr *nam, __unused struct proc *p)
             lck_mtx_unlock(ctl_mtx);
             return(EINVAL);
         }
             lck_mtx_unlock(ctl_mtx);
             return(EINVAL);
         }

-        if ((error = proc_suser(p))) {
+        if (kauth_cred_issuser(kauth_cred_get()) == 0) {

             lck_mtx_unlock(ctl_mtx);
             lck_mtx_unlock(ctl_mtx);

-            return error;
+            return EPERM;

         }
     }
 
         }
     }
 


@@ -325,25 +325,35 @@ ctl_connect(struct socket *so, struct sockaddr *nam, __unused struct proc *p)
                        return EBUSY;
                }
        } else {
                        return EBUSY;
                }
        } else {

-       u_int32_t       unit = kctl->lastunit + 1;
+               /* Find an unused ID, assumes control IDs are listed in order */
+       u_int32_t       unit = 1;

        
        

-       while (1) {
-           if (unit == ctl_maxunit)
-               unit = 1;
-                   if (kcb_find(kctl, unit) == NULL) {
-                       kctl->lastunit = sa.sc_unit = unit;
-                       break;
-                   }
-               if (unit++ == kctl->lastunit) {
-                   lck_mtx_unlock(ctl_mtx);
-               return EBUSY;
-           }
-           }
+       TAILQ_FOREACH(kcb_next, &kctl->kcb_head, next) {
+               if (kcb_next->unit > unit) {
+                       /* Found a gap, lets fill it in */
+                       break;
+               }
+               unit = kcb_next->unit + 1;
+               if (unit == ctl_maxunit)
+                       break;
+       }
+       
+               if (unit == ctl_maxunit) {
+                       lck_mtx_unlock(ctl_mtx);
+                       return EBUSY;
+               }
+               
+               sa.sc_unit = unit;

     }
 
        kcb->unit = sa.sc_unit;
     kcb->kctl = kctl;
     }
 
        kcb->unit = sa.sc_unit;
     kcb->kctl = kctl;

-    TAILQ_INSERT_TAIL(&kctl->kcb_head, kcb, next);
+    if (kcb_next != NULL) {
+       TAILQ_INSERT_BEFORE(kcb_next, kcb, next);
+    }
+    else {
+               TAILQ_INSERT_TAIL(&kctl->kcb_head, kcb, next);
+       }

     lck_mtx_unlock(ctl_mtx);
 
     error = soreserve(so, kctl->sendbufsize, kctl->recvbufsize);
     lck_mtx_unlock(ctl_mtx);
 
     error = soreserve(so, kctl->sendbufsize, kctl->recvbufsize);


@@ -499,7 +509,7 @@ ctl_enqueuedata(void *kctlref, u_int32_t unit, void *data, size_t len, u_int32_t
                return EINVAL;
        
        socket_lock(so, 1);
                return EINVAL;
        
        socket_lock(so, 1);

-       if ((size_t)sbspace(&so->so_rcv) < len) {
+       if (sbspace(&so->so_rcv) < (long)len) {

                error = ENOBUFS;
                goto bye;
        }
                error = ENOBUFS;
                goto bye;
        }


@@ -539,6 +549,7 @@ ctl_getenqueuespace(kern_ctl_ref kctlref, u_int32_t unit, size_t *space)
        struct ctl_cb   *kcb;
        struct kctl             *kctl = (struct kctl *)kctlref;
        struct socket   *so;
        struct ctl_cb   *kcb;
        struct kctl             *kctl = (struct kctl *)kctlref;
        struct socket   *so;

+       long avail;

        
        if (kctlref == NULL || space == NULL)
                return EINVAL;
        
        if (kctlref == NULL || space == NULL)
                return EINVAL;


@@ -552,7 +563,8 @@ ctl_getenqueuespace(kern_ctl_ref kctlref, u_int32_t unit, size_t *space)
                return EINVAL;
        
        socket_lock(so, 1);
                return EINVAL;
        
        socket_lock(so, 1);

-       *space = sbspace(&so->so_rcv);
+       avail = sbspace(&so->so_rcv);
+       *space = (avail < 0) ? 0 : avail;

        socket_unlock(so, 1);
 
        return 0;
        socket_unlock(so, 1);
 
        return 0;


@@ -581,10 +593,14 @@ ctl_ctloutput(struct socket *so, struct sockopt *sopt)
                case SOPT_SET:
                        if (kctl->setopt == NULL)
                                return(ENOTSUP);
                case SOPT_SET:
                        if (kctl->setopt == NULL)
                                return(ENOTSUP);

-                       MALLOC(data, void *, sopt->sopt_valsize, M_TEMP, M_WAITOK);
-                       if (data == NULL)
-                               return(ENOMEM);
-                       error = sooptcopyin(sopt, data, sopt->sopt_valsize, sopt->sopt_valsize);
+                       if (sopt->sopt_valsize == 0) {
+                               data = NULL;
+                       } else {
+                               MALLOC(data, void *, sopt->sopt_valsize, M_TEMP, M_WAITOK);
+                               if (data == NULL)
+                                       return(ENOMEM);
+                               error = sooptcopyin(sopt, data, sopt->sopt_valsize, sopt->sopt_valsize);
+                       }

                        if (error == 0) {
                                socket_unlock(so, 0);
                                error = (*kctl->setopt)(kcb->kctl, kcb->unit, kcb->userdata, sopt->sopt_name, 
                        if (error == 0) {
                                socket_unlock(so, 0);
                                error = (*kctl->setopt)(kcb->kctl, kcb->unit, kcb->userdata, sopt->sopt_name, 


@@ -677,10 +693,10 @@ ctl_ioctl(__unused struct socket *so, u_long cmd, caddr_t data,
  */
 errno_t
 ctl_register(struct kern_ctl_reg *userkctl, kern_ctl_ref *kctlref)
  */
 errno_t
 ctl_register(struct kern_ctl_reg *userkctl, kern_ctl_ref *kctlref)

-{      
-       struct kctl     *kctl = 0;
-       u_int32_t               id = -1;
-       u_int32_t               n;
+{
+       struct kctl     *kctl = NULL;
+       struct kctl     *kctl_next = NULL;
+       u_int32_t               id = 1;

        size_t                  name_len;
        
        if (userkctl == NULL)   /* sanity check */
        size_t                  name_len;
        
        if (userkctl == NULL)   /* sanity check */


@@ -698,29 +714,64 @@ ctl_register(struct kern_ctl_reg *userkctl, kern_ctl_ref *kctlref)
        
        lck_mtx_lock(ctl_mtx);
        
        
        lck_mtx_lock(ctl_mtx);
        

-       if ((userkctl->ctl_flags & CTL_FLAG_REG_ID_UNIT) == 0) {    
+       /*
+        * Kernel Control IDs
+        *
+        * CTL_FLAG_REG_ID_UNIT indicates the control ID and unit number are
+        * static. If they do not exist, add them to the list in order. If the
+        * flag is not set, we must find a new unique value. We assume the
+        * list is in order. We find the last item in the list and add one. If
+        * this leads to wrapping the id around, we start at the front of the
+        * list and look for a gap.
+        */
+       
+       if ((userkctl->ctl_flags & CTL_FLAG_REG_ID_UNIT) == 0) {
+               /* Must dynamically assign an unused ID */
+               
+               /* Verify the same name isn't already registered */

                if (ctl_find_by_name(userkctl->ctl_name) != NULL) {
                        lck_mtx_unlock(ctl_mtx);
                        FREE(kctl, M_TEMP);
                        return(EEXIST);
                }
                if (ctl_find_by_name(userkctl->ctl_name) != NULL) {
                        lck_mtx_unlock(ctl_mtx);
                        FREE(kctl, M_TEMP);
                        return(EEXIST);
                }

-               for (n = 0, id = ctl_last_id + 1; n < ctl_max; id++, n++) {
+               
+               /* Start with 1 in case the list is empty */
+               id = 1;
+               kctl_next = TAILQ_LAST(&ctl_head, kctl_list);
+               
+               if (kctl_next != NULL) {
+                       /* List was not empty, add one to the last item in the list */
+                       id = kctl_next->id + 1;
+                       kctl_next = NULL;
+                       
+                       /*
+                        * If this wrapped the id number, start looking at the front
+                        * of the list for an unused id.
+                        */

                        if (id == 0) {
                        if (id == 0) {

-                               n--;
-                               continue;
+                               /* Find the next unused ID */
+                               id = 1;
+                               
+                               TAILQ_FOREACH(kctl_next, &ctl_head, next) {
+                                       if (kctl_next->id > id) {
+                                               /* We found a gap */
+                                               break;
+                                       }
+                                       
+                                       id = kctl_next->id + 1;
+                               }

                        }
                        }

-                       if (ctl_find_by_id(id) == 0)
-                               break;

                }
                }

-               if (id == ctl_max) {
-                       lck_mtx_unlock(ctl_mtx);
-                       FREE(kctl, M_TEMP);
-                       return(ENOBUFS);
-               }
-               userkctl->ctl_id =id;
+               
+               userkctl->ctl_id = id;

                kctl->id = id;
                kctl->reg_unit = -1;
        } else {
                kctl->id = id;
                kctl->reg_unit = -1;
        } else {

+               TAILQ_FOREACH(kctl_next, &ctl_head, next) {
+                       if (kctl_next->id > userkctl->ctl_id)
+                               break;
+               }
+               

                if (ctl_find_by_id_unit(userkctl->ctl_id, userkctl->ctl_unit) != NULL) {
                        lck_mtx_unlock(ctl_mtx);
                        FREE(kctl, M_TEMP);
                if (ctl_find_by_id_unit(userkctl->ctl_id, userkctl->ctl_unit) != NULL) {
                        lck_mtx_unlock(ctl_mtx);
                        FREE(kctl, M_TEMP);


@@ -729,7 +780,7 @@ ctl_register(struct kern_ctl_reg *userkctl, kern_ctl_ref *kctlref)
                kctl->id = userkctl->ctl_id;
                kctl->reg_unit = userkctl->ctl_unit;
        }
                kctl->id = userkctl->ctl_id;
                kctl->reg_unit = userkctl->ctl_unit;
        }

-       strcpy(kctl->name, userkctl->ctl_name);
+       strlcpy(kctl->name, userkctl->ctl_name, MAX_KCTL_NAME);

        kctl->flags = userkctl->ctl_flags;
 
        /* Let the caller know the default send and receive sizes */
        kctl->flags = userkctl->ctl_flags;
 
        /* Let the caller know the default send and receive sizes */


@@ -749,8 +800,10 @@ ctl_register(struct kern_ctl_reg *userkctl, kern_ctl_ref *kctlref)
        
        TAILQ_INIT(&kctl->kcb_head);
        
        
        TAILQ_INIT(&kctl->kcb_head);
        

-       TAILQ_INSERT_TAIL(&ctl_head, kctl, next);
-       ctl_max++;
+       if (kctl_next)
+               TAILQ_INSERT_BEFORE(kctl_next, kctl, next);
+       else
+               TAILQ_INSERT_TAIL(&ctl_head, kctl, next);

        
        lck_mtx_unlock(ctl_mtx);
        
        
        lck_mtx_unlock(ctl_mtx);
        


@@ -783,7 +836,6 @@ ctl_deregister(void *kctlref)
        }
 
     TAILQ_REMOVE(&ctl_head, kctl, next);
        }
 
     TAILQ_REMOVE(&ctl_head, kctl, next);

-    ctl_max--;

 
     lck_mtx_unlock(ctl_mtx);
     
 
     lck_mtx_unlock(ctl_mtx);
     


@@ -792,21 +844,6 @@ ctl_deregister(void *kctlref)
     return(0);
 }
 
     return(0);
 }
 

-/*
- * Must be called with global lock taked
- */
-static struct kctl *
-ctl_find_by_id(u_int32_t id)
-{      
-    struct kctl        *kctl;
-
-    TAILQ_FOREACH(kctl, &ctl_head, next)
-        if (kctl->id == id)
-            return kctl;
-
-    return NULL;
-}
-

 /*
  * Must be called with global ctl_mtx lock taked
  */
 /*
  * Must be called with global ctl_mtx lock taked
  */


@@ -816,7 +853,7 @@ ctl_find_by_name(const char *name)
     struct kctl        *kctl;
 
     TAILQ_FOREACH(kctl, &ctl_head, next)
     struct kctl        *kctl;
 
     TAILQ_FOREACH(kctl, &ctl_head, next)

-        if (strcmp(kctl->name, name) == 0)
+        if (strncmp(kctl->name, name, sizeof(kctl->name)) == 0)

             return kctl;
 
     return NULL;
             return kctl;
 
     return NULL;


@@ -884,7 +921,7 @@ ctl_post_msg(u_long event_code, u_int32_t id)
 static int
 ctl_lock(struct socket *so, int refcount, int lr)
  {
 static int
 ctl_lock(struct socket *so, int refcount, int lr)
  {

-       int lr_saved;
+       uint32_t lr_saved;

        if (lr == 0) 
                lr_saved = (unsigned int) __builtin_return_address(0);
        else lr_saved = lr;
        if (lr == 0) 
                lr_saved = (unsigned int) __builtin_return_address(0);
        else lr_saved = lr;


@@ -892,18 +929,18 @@ ctl_lock(struct socket *so, int refcount, int lr)
        if (so->so_pcb) {
                lck_mtx_lock(((struct ctl_cb *)so->so_pcb)->mtx);
        } else  {
        if (so->so_pcb) {
                lck_mtx_lock(((struct ctl_cb *)so->so_pcb)->mtx);
        } else  {

-               panic("ctl_lock: so=%x NO PCB! lr=%x\n", so, lr_saved);
+               panic("ctl_lock: so=%p NO PCB! lr=%x\n", so, lr_saved);

                lck_mtx_lock(so->so_proto->pr_domain->dom_mtx);
        }
        
        if (so->so_usecount < 0)
                lck_mtx_lock(so->so_proto->pr_domain->dom_mtx);
        }
        
        if (so->so_usecount < 0)

-               panic("ctl_lock: so=%x so_pcb=%x lr=%x ref=%x\n",
+               panic("ctl_lock: so=%p so_pcb=%p lr=%x ref=%x\n",

                so, so->so_pcb, lr_saved, so->so_usecount);
        
        if (refcount)
                so->so_usecount++;
 
                so, so->so_pcb, lr_saved, so->so_usecount);
        
        if (refcount)
                so->so_usecount++;
 

-       so->lock_lr[so->next_lock_lr] = (void *)lr_saved;
+       so->lock_lr[so->next_lock_lr] = lr_saved;

        so->next_lock_lr = (so->next_lock_lr+1) % SO_LCKDBG_MAX;
        return (0);
 }
        so->next_lock_lr = (so->next_lock_lr+1) % SO_LCKDBG_MAX;
        return (0);
 }


@@ -911,7 +948,7 @@ ctl_lock(struct socket *so, int refcount, int lr)
 static int
 ctl_unlock(struct socket *so, int refcount, int lr)
 {
 static int
 ctl_unlock(struct socket *so, int refcount, int lr)
 {

-       int lr_saved;
+       uint32_t lr_saved;

        lck_mtx_t * mutex_held;
        
        if (lr == 0) 
        lck_mtx_t * mutex_held;
        
        if (lr == 0) 


@@ -926,15 +963,15 @@ ctl_unlock(struct socket *so, int refcount, int lr)
                so->so_usecount--;
        
        if (so->so_usecount < 0)
                so->so_usecount--;
        
        if (so->so_usecount < 0)

-               panic("ctl_unlock: so=%x usecount=%x\n", so, so->so_usecount);
+               panic("ctl_unlock: so=%p usecount=%x\n", so, so->so_usecount);

        if (so->so_pcb == NULL) {
        if (so->so_pcb == NULL) {

-               panic("ctl_unlock: so=%x NO PCB usecount=%x lr=%x\n", so, so->so_usecount, lr_saved);
+               panic("ctl_unlock: so=%p NO PCB usecount=%x lr=%x\n", so, so->so_usecount, lr_saved);

                mutex_held = so->so_proto->pr_domain->dom_mtx;
        } else {
                mutex_held = ((struct ctl_cb *)so->so_pcb)->mtx;
        }
        lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);
                mutex_held = so->so_proto->pr_domain->dom_mtx;
        } else {
                mutex_held = ((struct ctl_cb *)so->so_pcb)->mtx;
        }
        lck_mtx_assert(mutex_held, LCK_MTX_ASSERT_OWNED);

-       so->unlock_lr[so->next_unlock_lr] = (void *)lr_saved;
+       so->unlock_lr[so->next_unlock_lr] = lr_saved;

        so->next_unlock_lr = (so->next_unlock_lr+1) % SO_LCKDBG_MAX;
        lck_mtx_unlock(mutex_held);
        
        so->next_unlock_lr = (so->next_unlock_lr+1) % SO_LCKDBG_MAX;
        lck_mtx_unlock(mutex_held);
        


@@ -951,10 +988,10 @@ ctl_getlock(struct socket *so, __unused int locktype)
        
        if (so->so_pcb)  {
                if (so->so_usecount < 0)
        
        if (so->so_pcb)  {
                if (so->so_usecount < 0)

-                       panic("ctl_getlock: so=%x usecount=%x\n", so, so->so_usecount);
+                       panic("ctl_getlock: so=%p usecount=%x\n", so, so->so_usecount);

                return(kcb->mtx);
        } else {
                return(kcb->mtx);
        } else {

-               panic("ctl_getlock: so=%x NULL so_pcb\n", so);
+               panic("ctl_getlock: so=%p NULL so_pcb\n", so);

                return (so->so_proto->pr_domain->dom_mtx);
        }
 }
                return (so->so_proto->pr_domain->dom_mtx);
        }
 }