Author: jamie
Date: Mon Apr 25 04:36:54 2016
New Revision: 298567
URL: https://svnweb.freebsd.org/changeset/base/298567

Log:
  Use the new PR_METHOD_REMOVE to clean up jail handling in POSIX
  message queues.

Modified:
  head/sys/kern/uipc_mqueue.c

Modified: head/sys/kern/uipc_mqueue.c
==============================================================================
--- head/sys/kern/uipc_mqueue.c Mon Apr 25 04:27:58 2016        (r298566)
+++ head/sys/kern/uipc_mqueue.c Mon Apr 25 04:36:54 2016        (r298567)
@@ -154,11 +154,6 @@ struct mqfs_node {
 #define        FPTOMQ(fp)      ((struct mqueue *)(((struct mqfs_node *) \
                                (fp)->f_data)->mn_data))
 
-struct mqfs_osd {
-       struct task     mo_task;
-       const void      *mo_pr_root;
-};
-
 TAILQ_HEAD(msgq, mqueue_msg);
 
 struct mqueue;
@@ -244,9 +239,7 @@ static int  mqfs_destroy(struct mqfs_node
 static void    mqfs_fileno_alloc(struct mqfs_info *mi, struct mqfs_node *mn);
 static void    mqfs_fileno_free(struct mqfs_info *mi, struct mqfs_node *mn);
 static int     mqfs_allocv(struct mount *mp, struct vnode **vpp, struct 
mqfs_node *pn);
-static int     mqfs_prison_create(void *obj, void *data);
-static void    mqfs_prison_destructor(void *data);
-static void    mqfs_prison_remove_task(void *context, int pending);
+static int     mqfs_prison_remove(void *obj, void *data);
 
 /*
  * Message queue construction and maniplation
@@ -656,9 +649,8 @@ mqfs_init(struct vfsconf *vfc)
 {
        struct mqfs_node *root;
        struct mqfs_info *mi;
-       struct prison *pr;
        osd_method_t methods[PR_MAXMETHOD] = {
-           [PR_METHOD_CREATE] = mqfs_prison_create,
+           [PR_METHOD_REMOVE] = mqfs_prison_remove,
        };
 
        mqnode_zone = uma_zcreate("mqnode", sizeof(struct mqfs_node),
@@ -686,13 +678,7 @@ mqfs_init(struct vfsconf *vfc)
            EVENTHANDLER_PRI_ANY);
        mq_fdclose = mqueue_fdclose;
        p31b_setcfg(CTL_P1003_1B_MESSAGE_PASSING, _POSIX_MESSAGE_PASSING);
-
-       /* Note current jails. */
-       mqfs_osd_jail_slot = osd_jail_register(mqfs_prison_destructor, methods);
-       sx_slock(&allprison_lock);
-       TAILQ_FOREACH(pr, &allprison, pr_list)
-               (void)mqfs_prison_create(pr, NULL);
-       sx_sunlock(&allprison_lock);
+       mqfs_osd_jail_slot = osd_jail_register(NULL, methods);
        return (0);
 }
 
@@ -702,14 +688,11 @@ mqfs_init(struct vfsconf *vfc)
 static int
 mqfs_uninit(struct vfsconf *vfc)
 {
-       unsigned slot;
        struct mqfs_info *mi;
 
        if (!unloadable)
                return (EOPNOTSUPP);
-       slot = mqfs_osd_jail_slot;
-       mqfs_osd_jail_slot = 0;
-       osd_jail_deregister(slot);
+       osd_jail_deregister(mqfs_osd_jail_slot);
        EVENTHANDLER_DEREGISTER(process_exit, exit_tag);
        mi = &mqfs_data;
        mqfs_destroy(mi->mi_root);
@@ -1563,64 +1546,22 @@ mqfs_rmdir(struct vop_rmdir_args *ap)
 
 #endif /* notyet */
 
-
 /*
- * Set a destructor task with the prison's root
+ * See if this prison root is obsolete, and clean up associated queues if it 
is.
  */
 static int
-mqfs_prison_create(void *obj, void *data __unused)
-{
-       struct prison *pr = obj;
-       struct mqfs_osd *mo;
-       void *rsv;
-
-       if (pr->pr_root == pr->pr_parent->pr_root)
-               return(0);
-
-       mo = malloc(sizeof(struct mqfs_osd), M_PRISON, M_WAITOK);
-       rsv = osd_reserve(mqfs_osd_jail_slot);
-       TASK_INIT(&mo->mo_task, 0, mqfs_prison_remove_task, mo);
-       mtx_lock(&pr->pr_mtx);
-       mo->mo_pr_root = pr->pr_root;
-       (void)osd_jail_set_reserved(pr, mqfs_osd_jail_slot, rsv, mo);
-       mtx_unlock(&pr->pr_mtx);
-       return (0);
-}
-
-/*
- * Queue the task for after jail/OSD locks are released
- */
-static void
-mqfs_prison_destructor(void *data)
-{
-       struct mqfs_osd *mo = data;
-
-       if (mqfs_osd_jail_slot != 0)
-               taskqueue_enqueue(taskqueue_thread, &mo->mo_task);
-       else
-               free(mo, M_PRISON);
-}
-
-/*
- * See if this prison root is obsolete, and clean up associated queues if it is
- */
-static void
-mqfs_prison_remove_task(void *context, int pending)
+mqfs_prison_remove(void *obj, void *data __unused)
 {
-       struct mqfs_osd *mo = context;
+       const struct prison *pr = obj;
+       const struct prison *tpr;
        struct mqfs_node *pn, *tpn;
-       const struct prison *pr;
-       const void *pr_root;
        int found;
 
-       pr_root = mo->mo_pr_root;
        found = 0;
-       sx_slock(&allprison_lock);
-       TAILQ_FOREACH(pr, &allprison, pr_list) {
-               if (pr->pr_root == pr_root)
+       TAILQ_FOREACH(tpr, &allprison, pr_list) {
+               if (tpr->pr_root == pr->pr_root && tpr != pr && tpr->pr_ref > 0)
                        found = 1;
        }
-       sx_sunlock(&allprison_lock);
        if (!found) {
                /*
                 * No jails are rooted in this directory anymore,
@@ -1629,15 +1570,14 @@ mqfs_prison_remove_task(void *context, i
                sx_xlock(&mqfs_data.mi_lock);
                LIST_FOREACH_SAFE(pn, &mqfs_data.mi_root->mn_children,
                    mn_sibling, tpn) {
-                       if (pn->mn_pr_root == pr_root)
+                       if (pn->mn_pr_root == pr->pr_root)
                                (void)do_unlink(pn, curthread->td_ucred);
                }
                sx_xunlock(&mqfs_data.mi_lock);
        }
-       free(mo, M_PRISON);
+       return (0);
 }
 
-
 /*
  * Allocate a message queue
  */
_______________________________________________
svn-src-head@freebsd.org mailing list
https://lists.freebsd.org/mailman/listinfo/svn-src-head
To unsubscribe, send any mail to "svn-src-head-unsubscr...@freebsd.org"

Reply via email to