Author: glebius
Date: Tue Jun  4 11:19:08 2013
New Revision: 251374
URL: http://svnweb.freebsd.org/changeset/base/251374

Log:
  Improve r250890, so that we stop processing of a message with zero
  descriptors as early as possible, and assert that number of descriptors
  is positive in unp_freerights().
  
  Reviewed by:  mjg, pjd, jilles

Modified:
  head/sys/kern/uipc_usrreq.c

Modified: head/sys/kern/uipc_usrreq.c
==============================================================================
--- head/sys/kern/uipc_usrreq.c Tue Jun  4 11:05:57 2013        (r251373)
+++ head/sys/kern/uipc_usrreq.c Tue Jun  4 11:19:08 2013        (r251374)
@@ -1686,8 +1686,8 @@ unp_freerights(struct filedescent **fdep
        struct file *fp;
        int i;
 
-       if (fdcount == 0)
-               return;
+       KASSERT(fdcount > 0, ("%s: fdcount %d", __func__, fdcount));
+
        for (i = 0; i < fdcount; i++) {
                fp = fdep[i]->fde_file;
                filecaps_free(&fdep[i]->fde_caps);
@@ -1725,6 +1725,8 @@ unp_externalize(struct mbuf *control, st
                if (cm->cmsg_level == SOL_SOCKET
                    && cm->cmsg_type == SCM_RIGHTS) {
                        newfds = datalen / sizeof(*fdep);
+                       if (newfds == 0)
+                               goto next;
                        fdep = data;
 
                        /* If we're not outputting the descriptors free them. */
@@ -1770,8 +1772,7 @@ unp_externalize(struct mbuf *control, st
                                unp_externalize_fp(fde->fde_file);
                        }
                        FILEDESC_XUNLOCK(fdesc);
-                       if (newfds != 0)
-                               free(fdep[0], M_FILECAPS);
+                       free(fdep[0], M_FILECAPS);
                } else {
                        /* We can just copy anything else across. */
                        if (error || controlp == NULL)
@@ -1894,6 +1895,8 @@ unp_internalize(struct mbuf **controlp, 
 
                case SCM_RIGHTS:
                        oldfds = datalen / sizeof (int);
+                       if (oldfds == 0)
+                               break;
                        /*
                         * Check that all the FDs passed in refer to legal
                         * files.  If not, reject the entire operation.
@@ -1928,10 +1931,6 @@ unp_internalize(struct mbuf **controlp, 
                                error = E2BIG;
                                goto out;
                        }
-                       if (oldfds == 0) {
-                               FILEDESC_SUNLOCK(fdesc);
-                               break;
-                       }
                        fdp = data;
                        fdep = (struct filedescent **)
                            CMSG_DATA(mtod(*controlp, struct cmsghdr *));
_______________________________________________
[email protected] mailing list
http://lists.freebsd.org/mailman/listinfo/svn-src-head
To unsubscribe, send any mail to "[email protected]"

Reply via email to