123443Smckusick /* 237616Smckusick * Copyright (c) 1982, 1986, 1989 Regents of the University of California. 333288Sbostic * All rights reserved. 423443Smckusick * 533288Sbostic * Redistribution and use in source and binary forms are permitted 634859Sbostic * provided that the above copyright notice and this paragraph are 734859Sbostic * duplicated in all such forms and that any documentation, 834859Sbostic * advertising materials, and other materials related to such 934859Sbostic * distribution and use acknowledge that the software was developed 1034859Sbostic * by the University of California, Berkeley. The name of the 1134859Sbostic * University may not be used to endorse or promote products derived 1234859Sbostic * from this software without specific prior written permission. 1334859Sbostic * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR 1434859Sbostic * IMPLIED WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED 1537616Smckusick * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. 1633288Sbostic * 17*39353Smckusick * @(#)uipc_usrreq.c 7.13 (Berkeley) 10/19/89 1823443Smckusick */ 198925Sroot 2017105Sbloom #include "param.h" 2117105Sbloom #include "user.h" 2217105Sbloom #include "domain.h" 2317105Sbloom #include "protosw.h" 2417105Sbloom #include "socket.h" 2517105Sbloom #include "socketvar.h" 2617105Sbloom #include "unpcb.h" 2717105Sbloom #include "un.h" 2837616Smckusick #include "vnode.h" 2937616Smckusick #include "mount.h" 3017105Sbloom #include "file.h" 3117105Sbloom #include "stat.h" 3237728Smckusick #include "mbuf.h" /* XXX must appear after mount.h */ 338925Sroot 348925Sroot /* 358925Sroot * Unix communications domain. 3612760Ssam * 3712760Ssam * TODO: 3812760Ssam * SEQPACKET, RDM 3913119Ssam * rethink name space problems 4012760Ssam * need a proper out-of-band 418925Sroot */ 4237617Smckusick struct sockaddr sun_noname = { sizeof(sun_noname), AF_UNIX }; 4337616Smckusick ino_t unp_vno; /* prototype for fake vnode numbers */ 448925Sroot 458925Sroot /*ARGSUSED*/ 4612760Ssam uipc_usrreq(so, req, m, nam, rights) 478925Sroot struct socket *so; 488925Sroot int req; 4912760Ssam struct mbuf *m, *nam, *rights; 508925Sroot { 518925Sroot struct unpcb *unp = sotounpcb(so); 528925Sroot register struct socket *so2; 538925Sroot int error = 0; 548925Sroot 5525555Skarels if (req == PRU_CONTROL) 5625555Skarels return (EOPNOTSUPP); 5712760Ssam if (req != PRU_SEND && rights && rights->m_len) { 5812760Ssam error = EOPNOTSUPP; 5912760Ssam goto release; 6012760Ssam } 6112760Ssam if (unp == 0 && req != PRU_ATTACH) { 6212760Ssam error = EINVAL; 6312760Ssam goto release; 6412760Ssam } 658925Sroot switch (req) { 668925Sroot 678925Sroot case PRU_ATTACH: 688925Sroot if (unp) { 699169Ssam error = EISCONN; 708925Sroot break; 718925Sroot } 729028Sroot error = unp_attach(so); 738925Sroot break; 748925Sroot 758925Sroot case PRU_DETACH: 768925Sroot unp_detach(unp); 778925Sroot break; 788925Sroot 799169Ssam case PRU_BIND: 809169Ssam error = unp_bind(unp, nam); 819169Ssam break; 829169Ssam 839169Ssam case PRU_LISTEN: 8437616Smckusick if (unp->unp_vnode == 0) 859169Ssam error = EINVAL; 869169Ssam break; 879169Ssam 888925Sroot case PRU_CONNECT: 899028Sroot error = unp_connect(so, nam); 908925Sroot break; 918925Sroot 9212760Ssam case PRU_CONNECT2: 9326281Skarels error = unp_connect2(so, (struct socket *)nam); 9412760Ssam break; 9512760Ssam 968925Sroot case PRU_DISCONNECT: 978925Sroot unp_disconnect(unp); 988925Sroot break; 998925Sroot 1009169Ssam case PRU_ACCEPT: 10125899Skarels /* 10225899Skarels * Pass back name of connected socket, 10325899Skarels * if it was bound and we are still connected 10425899Skarels * (our peer may have closed already!). 10525899Skarels */ 10625899Skarels if (unp->unp_conn && unp->unp_conn->unp_addr) { 10725632Skarels nam->m_len = unp->unp_conn->unp_addr->m_len; 10825632Skarels bcopy(mtod(unp->unp_conn->unp_addr, caddr_t), 10925632Skarels mtod(nam, caddr_t), (unsigned)nam->m_len); 11025632Skarels } else { 11125632Skarels nam->m_len = sizeof(sun_noname); 11225632Skarels *(mtod(nam, struct sockaddr *)) = sun_noname; 11325632Skarels } 1148925Sroot break; 1158925Sroot 1168925Sroot case PRU_SHUTDOWN: 1178925Sroot socantsendmore(so); 1188925Sroot unp_usrclosed(unp); 1198925Sroot break; 1208925Sroot 1218925Sroot case PRU_RCVD: 1228925Sroot switch (so->so_type) { 1238925Sroot 1248925Sroot case SOCK_DGRAM: 1258925Sroot panic("uipc 1"); 12610139Ssam /*NOTREACHED*/ 1278925Sroot 12810139Ssam case SOCK_STREAM: 1298925Sroot #define rcv (&so->so_rcv) 1308925Sroot #define snd (&so2->so_snd) 1318925Sroot if (unp->unp_conn == 0) 1328925Sroot break; 1338925Sroot so2 = unp->unp_conn->unp_socket; 1348925Sroot /* 13525632Skarels * Adjust backpressure on sender 1368925Sroot * and wakeup any waiting to write. 1378925Sroot */ 13825632Skarels snd->sb_mbmax += unp->unp_mbcnt - rcv->sb_mbcnt; 13925632Skarels unp->unp_mbcnt = rcv->sb_mbcnt; 14025632Skarels snd->sb_hiwat += unp->unp_cc - rcv->sb_cc; 14125632Skarels unp->unp_cc = rcv->sb_cc; 14217543Skarels sowwakeup(so2); 1438925Sroot #undef snd 1448925Sroot #undef rcv 1458925Sroot break; 1468925Sroot 1478925Sroot default: 1488925Sroot panic("uipc 2"); 1498925Sroot } 1508925Sroot break; 1518925Sroot 1528925Sroot case PRU_SEND: 15325632Skarels if (rights) { 15425632Skarels error = unp_internalize(rights); 15525632Skarels if (error) 15625632Skarels break; 15725632Skarels } 1588925Sroot switch (so->so_type) { 1598925Sroot 16025632Skarels case SOCK_DGRAM: { 16125632Skarels struct sockaddr *from; 16225632Skarels 1639028Sroot if (nam) { 1648925Sroot if (unp->unp_conn) { 1658925Sroot error = EISCONN; 1668925Sroot break; 1678925Sroot } 1689028Sroot error = unp_connect(so, nam); 1698925Sroot if (error) 1708925Sroot break; 1718925Sroot } else { 1728925Sroot if (unp->unp_conn == 0) { 1738925Sroot error = ENOTCONN; 1748925Sroot break; 1758925Sroot } 1768925Sroot } 1778925Sroot so2 = unp->unp_conn->unp_socket; 17825632Skarels if (unp->unp_addr) 17925632Skarels from = mtod(unp->unp_addr, struct sockaddr *); 18025632Skarels else 18125632Skarels from = &sun_noname; 18225632Skarels if (sbspace(&so2->so_rcv) > 0 && 18325632Skarels sbappendaddr(&so2->so_rcv, from, m, rights)) { 18425632Skarels sorwakeup(so2); 18525632Skarels m = 0; 18625632Skarels } else 18725632Skarels error = ENOBUFS; 1889028Sroot if (nam) 1899169Ssam unp_disconnect(unp); 1908925Sroot break; 19125632Skarels } 1928925Sroot 1938925Sroot case SOCK_STREAM: 1948925Sroot #define rcv (&so2->so_rcv) 1958925Sroot #define snd (&so->so_snd) 19623524Skarels if (so->so_state & SS_CANTSENDMORE) { 19723524Skarels error = EPIPE; 19823524Skarels break; 19923524Skarels } 2008925Sroot if (unp->unp_conn == 0) 2018925Sroot panic("uipc 3"); 2028925Sroot so2 = unp->unp_conn->unp_socket; 2038925Sroot /* 20425632Skarels * Send to paired receive port, and then reduce 20525632Skarels * send buffer hiwater marks to maintain backpressure. 2068925Sroot * Wake up readers. 2078925Sroot */ 20825632Skarels if (rights) 20926365Skarels (void)sbappendrights(rcv, m, rights); 21025632Skarels else 21125632Skarels sbappend(rcv, m); 21225632Skarels snd->sb_mbmax -= 21325632Skarels rcv->sb_mbcnt - unp->unp_conn->unp_mbcnt; 21425632Skarels unp->unp_conn->unp_mbcnt = rcv->sb_mbcnt; 21525632Skarels snd->sb_hiwat -= rcv->sb_cc - unp->unp_conn->unp_cc; 21625632Skarels unp->unp_conn->unp_cc = rcv->sb_cc; 21717543Skarels sorwakeup(so2); 21817543Skarels m = 0; 2198925Sroot #undef snd 2208925Sroot #undef rcv 2218925Sroot break; 2228925Sroot 2238925Sroot default: 2248925Sroot panic("uipc 4"); 2258925Sroot } 2268925Sroot break; 2278925Sroot 2288925Sroot case PRU_ABORT: 2298925Sroot unp_drop(unp, ECONNABORTED); 2308925Sroot break; 2318925Sroot 2328925Sroot case PRU_SENSE: 23316973Skarels ((struct stat *) m)->st_blksize = so->so_snd.sb_hiwat; 23416973Skarels if (so->so_type == SOCK_STREAM && unp->unp_conn != 0) { 23516973Skarels so2 = unp->unp_conn->unp_socket; 23616973Skarels ((struct stat *) m)->st_blksize += so2->so_rcv.sb_cc; 23716973Skarels } 23821110Skarels ((struct stat *) m)->st_dev = NODEV; 23937616Smckusick if (unp->unp_vno == 0) 24037616Smckusick unp->unp_vno = unp_vno++; 24137616Smckusick ((struct stat *) m)->st_ino = unp->unp_vno; 24216973Skarels return (0); 2438925Sroot 2448925Sroot case PRU_RCVOOB: 24516774Sbloom return (EOPNOTSUPP); 2468925Sroot 2478925Sroot case PRU_SENDOOB: 24817543Skarels error = EOPNOTSUPP; 2498925Sroot break; 2508925Sroot 2518925Sroot case PRU_SOCKADDR: 25237617Smckusick if (unp->unp_addr) { 25337617Smckusick nam->m_len = unp->unp_addr->m_len; 25437617Smckusick bcopy(mtod(unp->unp_addr, caddr_t), 25537617Smckusick mtod(nam, caddr_t), (unsigned)nam->m_len); 25637617Smckusick } else 25737617Smckusick nam->m_len = 0; 2588925Sroot break; 2598925Sroot 26014121Ssam case PRU_PEERADDR: 26128292Skarels if (unp->unp_conn && unp->unp_conn->unp_addr) { 26228292Skarels nam->m_len = unp->unp_conn->unp_addr->m_len; 26328292Skarels bcopy(mtod(unp->unp_conn->unp_addr, caddr_t), 26433287Sbostic mtod(nam, caddr_t), (unsigned)nam->m_len); 26537617Smckusick } else 26637617Smckusick nam->m_len = 0; 26714121Ssam break; 26814121Ssam 2698925Sroot case PRU_SLOWTIMO: 2708925Sroot break; 2718925Sroot 2728925Sroot default: 2738925Sroot panic("piusrreq"); 2748925Sroot } 27512760Ssam release: 27612760Ssam if (m) 27712760Ssam m_freem(m); 27811709Ssam return (error); 2798925Sroot } 2808925Sroot 28116973Skarels /* 28225632Skarels * Both send and receive buffers are allocated PIPSIZ bytes of buffering 28325632Skarels * for stream sockets, although the total for sender and receiver is 28425632Skarels * actually only PIPSIZ. 28516973Skarels * Datagram sockets really use the sendspace as the maximum datagram size, 28616973Skarels * and don't really want to reserve the sendspace. Their recvspace should 28716973Skarels * be large enough for at least one max-size datagram plus address. 28816973Skarels */ 28916973Skarels #define PIPSIZ 4096 29037617Smckusick u_long unpst_sendspace = PIPSIZ; 29137617Smckusick u_long unpst_recvspace = PIPSIZ; 29237617Smckusick u_long unpdg_sendspace = 2*1024; /* really max datagram size */ 29337617Smckusick u_long unpdg_recvspace = 4*1024; 2948925Sroot 29525632Skarels int unp_rights; /* file descriptors in flight */ 29625632Skarels 2979169Ssam unp_attach(so) 2988925Sroot struct socket *so; 2998925Sroot { 3009169Ssam register struct mbuf *m; 3018925Sroot register struct unpcb *unp; 3028925Sroot int error; 3038925Sroot 30437617Smckusick if (so->so_snd.sb_hiwat == 0 || so->so_rcv.sb_hiwat == 0) { 30537617Smckusick switch (so->so_type) { 30616973Skarels 30737617Smckusick case SOCK_STREAM: 30837617Smckusick error = soreserve(so, unpst_sendspace, unpst_recvspace); 30937617Smckusick break; 31016973Skarels 31137617Smckusick case SOCK_DGRAM: 31237617Smckusick error = soreserve(so, unpdg_sendspace, unpdg_recvspace); 31337617Smckusick break; 31437617Smckusick } 31537617Smckusick if (error) 31637617Smckusick return (error); 31716973Skarels } 3189637Ssam m = m_getclr(M_DONTWAIT, MT_PCB); 31910139Ssam if (m == NULL) 32010139Ssam return (ENOBUFS); 3218925Sroot unp = mtod(m, struct unpcb *); 3228925Sroot so->so_pcb = (caddr_t)unp; 3238925Sroot unp->unp_socket = so; 3248925Sroot return (0); 3258925Sroot } 3268925Sroot 3278925Sroot unp_detach(unp) 3289169Ssam register struct unpcb *unp; 3298925Sroot { 3308925Sroot 33137616Smckusick if (unp->unp_vnode) { 33237616Smckusick unp->unp_vnode->v_socket = 0; 33337616Smckusick vrele(unp->unp_vnode); 33437616Smckusick unp->unp_vnode = 0; 3358925Sroot } 3368925Sroot if (unp->unp_conn) 3378925Sroot unp_disconnect(unp); 3388925Sroot while (unp->unp_refs) 3398925Sroot unp_drop(unp->unp_refs, ECONNRESET); 3408925Sroot soisdisconnected(unp->unp_socket); 3418925Sroot unp->unp_socket->so_pcb = 0; 34225632Skarels m_freem(unp->unp_addr); 3439169Ssam (void) m_free(dtom(unp)); 34425632Skarels if (unp_rights) 34525632Skarels unp_gc(); 3468925Sroot } 3478925Sroot 3489169Ssam unp_bind(unp, nam) 3498925Sroot struct unpcb *unp; 3509169Ssam struct mbuf *nam; 3518925Sroot { 3529169Ssam struct sockaddr_un *soun = mtod(nam, struct sockaddr_un *); 35337616Smckusick register struct vnode *vp; 35416695Smckusick register struct nameidata *ndp = &u.u_nd; 35537616Smckusick struct vattr vattr; 3568925Sroot int error; 3578925Sroot 35816695Smckusick ndp->ni_dirp = soun->sun_path; 35937617Smckusick if (unp->unp_vnode != NULL) 36012760Ssam return (EINVAL); 36137617Smckusick if (nam->m_len == MLEN) { 36237617Smckusick if (*(mtod(nam, caddr_t) + nam->m_len - 1) != 0) 36337617Smckusick return (EINVAL); 36437617Smckusick } else 36537617Smckusick *(mtod(nam, caddr_t) + nam->m_len) = 0; 36612760Ssam /* SHOULD BE ABLE TO ADOPT EXISTING AND wakeup() ALA FIFO's */ 36737616Smckusick ndp->ni_nameiop = CREATE | FOLLOW | LOCKPARENT; 36816695Smckusick ndp->ni_segflg = UIO_SYSSPACE; 36937616Smckusick if (error = namei(ndp)) 37037616Smckusick return (error); 37137616Smckusick vp = ndp->ni_vp; 37237616Smckusick if (vp != NULL) { 37337728Smckusick VOP_ABORTOP(ndp); 37410139Ssam return (EADDRINUSE); 3758925Sroot } 37637616Smckusick vattr_null(&vattr); 37737616Smckusick vattr.va_type = VSOCK; 37837616Smckusick vattr.va_mode = 0777; 37937728Smckusick if (error = VOP_CREATE(ndp, &vattr)) 38011828Ssam return (error); 38137616Smckusick vp = ndp->ni_vp; 38237616Smckusick vp->v_socket = unp->unp_socket; 38337616Smckusick unp->unp_vnode = vp; 38425632Skarels unp->unp_addr = m_copy(nam, 0, (int)M_COPYALL); 38537728Smckusick VOP_UNLOCK(vp); 3868925Sroot return (0); 3878925Sroot } 3888925Sroot 3899169Ssam unp_connect(so, nam) 3908925Sroot struct socket *so; 3919169Ssam struct mbuf *nam; 3928925Sroot { 3939169Ssam register struct sockaddr_un *soun = mtod(nam, struct sockaddr_un *); 39437616Smckusick register struct vnode *vp; 39537617Smckusick register struct socket *so2, *so3; 39637617Smckusick register struct nameidata *ndp = &u.u_nd; 39737617Smckusick struct unpcb *unp2, *unp3; 39837616Smckusick int error; 3998925Sroot 40016695Smckusick ndp->ni_dirp = soun->sun_path; 40137617Smckusick if (nam->m_data + nam->m_len == &nam->m_dat[MLEN]) { /* XXX */ 40237617Smckusick if (*(mtod(nam, caddr_t) + nam->m_len - 1) != 0) 40337617Smckusick return (EMSGSIZE); 40437617Smckusick } else 40537617Smckusick *(mtod(nam, caddr_t) + nam->m_len) = 0; 40637728Smckusick ndp->ni_nameiop = LOOKUP | FOLLOW | LOCKLEAF; 40716695Smckusick ndp->ni_segflg = UIO_SYSSPACE; 40837616Smckusick if (error = namei(ndp)) 40937616Smckusick return (error); 41037616Smckusick vp = ndp->ni_vp; 41137616Smckusick if (vp->v_type != VSOCK) { 4128925Sroot error = ENOTSOCK; 4138925Sroot goto bad; 4148925Sroot } 41538396Smckusick if (error = VOP_ACCESS(vp, VWRITE, ndp->ni_cred)) 41638396Smckusick goto bad; 41737616Smckusick so2 = vp->v_socket; 4188925Sroot if (so2 == 0) { 4198925Sroot error = ECONNREFUSED; 4208925Sroot goto bad; 4218925Sroot } 42213115Ssam if (so->so_type != so2->so_type) { 42313115Ssam error = EPROTOTYPE; 42413115Ssam goto bad; 42513115Ssam } 42637617Smckusick if (so->so_proto->pr_flags & PR_CONNREQUIRED) { 42737617Smckusick if ((so2->so_options & SO_ACCEPTCONN) == 0 || 42837617Smckusick (so3 = sonewconn(so2)) == 0) { 42937617Smckusick error = ECONNREFUSED; 43037617Smckusick goto bad; 43137617Smckusick } 43237617Smckusick unp2 = sotounpcb(so2); 43337617Smckusick unp3 = sotounpcb(so3); 43437617Smckusick if (unp2->unp_addr) 43537617Smckusick unp3->unp_addr = 43637617Smckusick m_copy(unp2->unp_addr, 0, (int)M_COPYALL); 43737617Smckusick so2 = so3; 43813115Ssam } 43926281Skarels error = unp_connect2(so, so2); 44012760Ssam bad: 44137728Smckusick vput(vp); 44212760Ssam return (error); 44312760Ssam } 44412760Ssam 44526281Skarels unp_connect2(so, so2) 44612760Ssam register struct socket *so; 44712760Ssam register struct socket *so2; 44812760Ssam { 44912760Ssam register struct unpcb *unp = sotounpcb(so); 45012760Ssam register struct unpcb *unp2; 45112760Ssam 45212760Ssam if (so2->so_type != so->so_type) 45312760Ssam return (EPROTOTYPE); 45414049Ssam unp2 = sotounpcb(so2); 45514049Ssam unp->unp_conn = unp2; 4568925Sroot switch (so->so_type) { 4578925Sroot 4588925Sroot case SOCK_DGRAM: 4598925Sroot unp->unp_nextref = unp2->unp_refs; 4608925Sroot unp2->unp_refs = unp; 46117543Skarels soisconnected(so); 4628925Sroot break; 4638925Sroot 4648925Sroot case SOCK_STREAM: 4659169Ssam unp2->unp_conn = unp; 46614049Ssam soisconnected(so2); 46714049Ssam soisconnected(so); 4688925Sroot break; 4698925Sroot 4708925Sroot default: 47112760Ssam panic("unp_connect2"); 4728925Sroot } 4738925Sroot return (0); 4748925Sroot } 4759169Ssam 4769169Ssam unp_disconnect(unp) 4779169Ssam struct unpcb *unp; 4789169Ssam { 4799169Ssam register struct unpcb *unp2 = unp->unp_conn; 4809169Ssam 4819169Ssam if (unp2 == 0) 4829169Ssam return; 4839169Ssam unp->unp_conn = 0; 4849169Ssam switch (unp->unp_socket->so_type) { 4859169Ssam 4869169Ssam case SOCK_DGRAM: 4879169Ssam if (unp2->unp_refs == unp) 4889169Ssam unp2->unp_refs = unp->unp_nextref; 4899169Ssam else { 4909169Ssam unp2 = unp2->unp_refs; 4919169Ssam for (;;) { 4929169Ssam if (unp2 == 0) 4939169Ssam panic("unp_disconnect"); 4949169Ssam if (unp2->unp_nextref == unp) 4959169Ssam break; 4969169Ssam unp2 = unp2->unp_nextref; 4979169Ssam } 4989169Ssam unp2->unp_nextref = unp->unp_nextref; 4999169Ssam } 5009169Ssam unp->unp_nextref = 0; 50121768Skarels unp->unp_socket->so_state &= ~SS_ISCONNECTED; 5029169Ssam break; 5039169Ssam 5049169Ssam case SOCK_STREAM: 50514049Ssam soisdisconnected(unp->unp_socket); 5069169Ssam unp2->unp_conn = 0; 5079169Ssam soisdisconnected(unp2->unp_socket); 5089169Ssam break; 5099169Ssam } 5109169Ssam } 5119169Ssam 51212760Ssam #ifdef notdef 5139169Ssam unp_abort(unp) 5149169Ssam struct unpcb *unp; 5159169Ssam { 5169169Ssam 5179169Ssam unp_detach(unp); 5189169Ssam } 51912760Ssam #endif 5209169Ssam 5219169Ssam /*ARGSUSED*/ 5229169Ssam unp_usrclosed(unp) 5239169Ssam struct unpcb *unp; 5249169Ssam { 5259169Ssam 5269169Ssam } 5279169Ssam 5289169Ssam unp_drop(unp, errno) 5299169Ssam struct unpcb *unp; 5309169Ssam int errno; 5319169Ssam { 53216054Skarels struct socket *so = unp->unp_socket; 5339169Ssam 53416054Skarels so->so_error = errno; 5359169Ssam unp_disconnect(unp); 53616054Skarels if (so->so_head) { 53716054Skarels so->so_pcb = (caddr_t) 0; 53825632Skarels m_freem(unp->unp_addr); 53916054Skarels (void) m_free(dtom(unp)); 54016054Skarels sofree(so); 54116054Skarels } 5429169Ssam } 5439169Ssam 54412760Ssam #ifdef notdef 5459169Ssam unp_drain() 5469169Ssam { 5479169Ssam 5489169Ssam } 54912760Ssam #endif 55012760Ssam 55112760Ssam unp_externalize(rights) 55212760Ssam struct mbuf *rights; 55312760Ssam { 55412760Ssam int newfds = rights->m_len / sizeof (int); 55512760Ssam register int i; 55612760Ssam register struct file **rp = mtod(rights, struct file **); 55712760Ssam register struct file *fp; 55812760Ssam int f; 55912760Ssam 56012760Ssam if (newfds > ufavail()) { 56112760Ssam for (i = 0; i < newfds; i++) { 56212760Ssam fp = *rp; 56312760Ssam unp_discard(fp); 56412760Ssam *rp++ = 0; 56512760Ssam } 56612760Ssam return (EMSGSIZE); 56712760Ssam } 56812760Ssam for (i = 0; i < newfds; i++) { 56937616Smckusick if (ufalloc(0, &f)) 57012760Ssam panic("unp_externalize"); 57112760Ssam fp = *rp; 57212760Ssam u.u_ofile[f] = fp; 57312760Ssam fp->f_msgcount--; 57425632Skarels unp_rights--; 57514927Smckusick *(int *)rp++ = f; 57612760Ssam } 57712760Ssam return (0); 57812760Ssam } 57912760Ssam 58012760Ssam unp_internalize(rights) 58112760Ssam struct mbuf *rights; 58212760Ssam { 58312760Ssam register struct file **rp; 58412760Ssam int oldfds = rights->m_len / sizeof (int); 58537728Smckusick register int i, fd; 58612760Ssam register struct file *fp; 58712760Ssam 58812760Ssam rp = mtod(rights, struct file **); 58937728Smckusick for (i = 0; i < oldfds; i++) { 59037728Smckusick fd = *(int *)rp++; 59137728Smckusick if ((unsigned)fd >= NOFILE || u.u_ofile[fd] == NULL) 59212760Ssam return (EBADF); 59337728Smckusick } 59412760Ssam rp = mtod(rights, struct file **); 59513084Ssam for (i = 0; i < oldfds; i++) { 59637728Smckusick fp = u.u_ofile[*(int *)rp]; 59712760Ssam *rp++ = fp; 59812760Ssam fp->f_count++; 59912760Ssam fp->f_msgcount++; 60025632Skarels unp_rights++; 60112760Ssam } 60212760Ssam return (0); 60312760Ssam } 60412760Ssam 60512760Ssam int unp_defer, unp_gcing; 60612760Ssam int unp_mark(); 60716995Skarels extern struct domain unixdomain; 60812760Ssam 60912760Ssam unp_gc() 61012760Ssam { 61112760Ssam register struct file *fp; 61212760Ssam register struct socket *so; 61312760Ssam 61412760Ssam if (unp_gcing) 61512760Ssam return; 61612760Ssam unp_gcing = 1; 61712760Ssam restart: 61812760Ssam unp_defer = 0; 61912760Ssam for (fp = file; fp < fileNFILE; fp++) 62012760Ssam fp->f_flag &= ~(FMARK|FDEFER); 62112760Ssam do { 62212760Ssam for (fp = file; fp < fileNFILE; fp++) { 62312760Ssam if (fp->f_count == 0) 62412760Ssam continue; 62512760Ssam if (fp->f_flag & FDEFER) { 62612760Ssam fp->f_flag &= ~FDEFER; 62712760Ssam unp_defer--; 62812760Ssam } else { 62912760Ssam if (fp->f_flag & FMARK) 63012760Ssam continue; 63112760Ssam if (fp->f_count == fp->f_msgcount) 63212760Ssam continue; 63312760Ssam fp->f_flag |= FMARK; 63412760Ssam } 63537617Smckusick if (fp->f_type != DTYPE_SOCKET || 63637617Smckusick (so = (struct socket *)fp->f_data) == 0) 63712760Ssam continue; 63816995Skarels if (so->so_proto->pr_domain != &unixdomain || 63921768Skarels (so->so_proto->pr_flags&PR_RIGHTS) == 0) 64012760Ssam continue; 64112760Ssam if (so->so_rcv.sb_flags & SB_LOCK) { 64212760Ssam sbwait(&so->so_rcv); 64312760Ssam goto restart; 64412760Ssam } 64512760Ssam unp_scan(so->so_rcv.sb_mb, unp_mark); 64612760Ssam } 64712760Ssam } while (unp_defer); 64812760Ssam for (fp = file; fp < fileNFILE; fp++) { 64912760Ssam if (fp->f_count == 0) 65012760Ssam continue; 65125632Skarels if (fp->f_count == fp->f_msgcount && (fp->f_flag & FMARK) == 0) 65225632Skarels while (fp->f_msgcount) 65325632Skarels unp_discard(fp); 65412760Ssam } 65512760Ssam unp_gcing = 0; 65612760Ssam } 65712760Ssam 65816995Skarels unp_dispose(m) 65916995Skarels struct mbuf *m; 66016995Skarels { 66116995Skarels int unp_discard(); 66216995Skarels 66317020Skarels if (m) 66417020Skarels unp_scan(m, unp_discard); 66516995Skarels } 66616995Skarels 66716995Skarels unp_scan(m0, op) 66816995Skarels register struct mbuf *m0; 66912760Ssam int (*op)(); 67012760Ssam { 67116995Skarels register struct mbuf *m; 67212760Ssam register struct file **rp; 67312760Ssam register int i; 67417020Skarels int qfds; 67512760Ssam 67616995Skarels while (m0) { 67716995Skarels for (m = m0; m; m = m->m_next) 67816995Skarels if (m->m_type == MT_RIGHTS && m->m_len) { 67916995Skarels qfds = m->m_len / sizeof (struct file *); 68016995Skarels rp = mtod(m, struct file **); 68116995Skarels for (i = 0; i < qfds; i++) 68216995Skarels (*op)(*rp++); 68316995Skarels break; /* XXX, but saves time */ 68416995Skarels } 68517020Skarels m0 = m0->m_act; 68612760Ssam } 68712760Ssam } 68812760Ssam 68912760Ssam unp_mark(fp) 69012760Ssam struct file *fp; 69112760Ssam { 69212760Ssam 69312760Ssam if (fp->f_flag & FMARK) 69412760Ssam return; 69512760Ssam unp_defer++; 69612760Ssam fp->f_flag |= (FMARK|FDEFER); 69712760Ssam } 69812760Ssam 69912760Ssam unp_discard(fp) 70012760Ssam struct file *fp; 70112760Ssam { 70212760Ssam 70312760Ssam fp->f_msgcount--; 70425632Skarels unp_rights--; 705*39353Smckusick (void) closef(fp); 70612760Ssam } 707