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*40937Skarels * @(#)uipc_usrreq.c 7.15 (Berkeley) 04/16/90 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 }; 4340800Ssklower ino_t unp_ino; /* prototype for fake inode numbers */ 448925Sroot 458925Sroot /*ARGSUSED*/ 4640800Ssklower uipc_usrreq(so, req, m, nam, control) 478925Sroot struct socket *so; 488925Sroot int req; 4940800Ssklower struct mbuf *m, *nam, *control; 508925Sroot { 518925Sroot struct unpcb *unp = sotounpcb(so); 528925Sroot register struct socket *so2; 53*40937Skarels register int error = 0; 548925Sroot 5525555Skarels if (req == PRU_CONTROL) 5625555Skarels return (EOPNOTSUPP); 5740800Ssklower if (req != PRU_SEND && control && control->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: 153*40937Skarels if (control && (error = unp_internalize(control))) 154*40937Skarels break; 1558925Sroot switch (so->so_type) { 1568925Sroot 15725632Skarels case SOCK_DGRAM: { 15825632Skarels struct sockaddr *from; 15925632Skarels 1609028Sroot if (nam) { 1618925Sroot if (unp->unp_conn) { 1628925Sroot error = EISCONN; 1638925Sroot break; 1648925Sroot } 1659028Sroot error = unp_connect(so, nam); 1668925Sroot if (error) 1678925Sroot break; 1688925Sroot } else { 1698925Sroot if (unp->unp_conn == 0) { 1708925Sroot error = ENOTCONN; 1718925Sroot break; 1728925Sroot } 1738925Sroot } 1748925Sroot so2 = unp->unp_conn->unp_socket; 17525632Skarels if (unp->unp_addr) 17625632Skarels from = mtod(unp->unp_addr, struct sockaddr *); 17725632Skarels else 17825632Skarels from = &sun_noname; 179*40937Skarels if (sbappendaddr(&so2->so_rcv, from, m, control)) { 18025632Skarels sorwakeup(so2); 18125632Skarels m = 0; 182*40937Skarels control = 0; 18325632Skarels } else 18425632Skarels error = ENOBUFS; 1859028Sroot if (nam) 1869169Ssam unp_disconnect(unp); 1878925Sroot break; 18825632Skarels } 1898925Sroot 1908925Sroot case SOCK_STREAM: 1918925Sroot #define rcv (&so2->so_rcv) 1928925Sroot #define snd (&so->so_snd) 19323524Skarels if (so->so_state & SS_CANTSENDMORE) { 19423524Skarels error = EPIPE; 19523524Skarels break; 19623524Skarels } 1978925Sroot if (unp->unp_conn == 0) 1988925Sroot panic("uipc 3"); 1998925Sroot so2 = unp->unp_conn->unp_socket; 2008925Sroot /* 20125632Skarels * Send to paired receive port, and then reduce 20225632Skarels * send buffer hiwater marks to maintain backpressure. 2038925Sroot * Wake up readers. 2048925Sroot */ 205*40937Skarels if (control) { 206*40937Skarels (void)sbappendcontrol(rcv, m, control); 207*40937Skarels control = 0; 208*40937Skarels } else 20925632Skarels sbappend(rcv, m); 21025632Skarels snd->sb_mbmax -= 21125632Skarels rcv->sb_mbcnt - unp->unp_conn->unp_mbcnt; 21225632Skarels unp->unp_conn->unp_mbcnt = rcv->sb_mbcnt; 21325632Skarels snd->sb_hiwat -= rcv->sb_cc - unp->unp_conn->unp_cc; 21425632Skarels unp->unp_conn->unp_cc = rcv->sb_cc; 21517543Skarels sorwakeup(so2); 21617543Skarels m = 0; 2178925Sroot #undef snd 2188925Sroot #undef rcv 2198925Sroot break; 2208925Sroot 2218925Sroot default: 2228925Sroot panic("uipc 4"); 2238925Sroot } 2248925Sroot break; 2258925Sroot 2268925Sroot case PRU_ABORT: 2278925Sroot unp_drop(unp, ECONNABORTED); 2288925Sroot break; 2298925Sroot 2308925Sroot case PRU_SENSE: 23116973Skarels ((struct stat *) m)->st_blksize = so->so_snd.sb_hiwat; 23216973Skarels if (so->so_type == SOCK_STREAM && unp->unp_conn != 0) { 23316973Skarels so2 = unp->unp_conn->unp_socket; 23416973Skarels ((struct stat *) m)->st_blksize += so2->so_rcv.sb_cc; 23516973Skarels } 23621110Skarels ((struct stat *) m)->st_dev = NODEV; 23740800Ssklower if (unp->unp_ino == 0) 23840800Ssklower unp->unp_ino = unp_ino++; 23940800Ssklower ((struct stat *) m)->st_ino = unp->unp_ino; 24016973Skarels return (0); 2418925Sroot 2428925Sroot case PRU_RCVOOB: 24316774Sbloom return (EOPNOTSUPP); 2448925Sroot 2458925Sroot case PRU_SENDOOB: 24617543Skarels error = EOPNOTSUPP; 2478925Sroot break; 2488925Sroot 2498925Sroot case PRU_SOCKADDR: 25037617Smckusick if (unp->unp_addr) { 25137617Smckusick nam->m_len = unp->unp_addr->m_len; 25237617Smckusick bcopy(mtod(unp->unp_addr, caddr_t), 25337617Smckusick mtod(nam, caddr_t), (unsigned)nam->m_len); 25437617Smckusick } else 25537617Smckusick nam->m_len = 0; 2568925Sroot break; 2578925Sroot 25814121Ssam case PRU_PEERADDR: 25928292Skarels if (unp->unp_conn && unp->unp_conn->unp_addr) { 26028292Skarels nam->m_len = unp->unp_conn->unp_addr->m_len; 26128292Skarels bcopy(mtod(unp->unp_conn->unp_addr, caddr_t), 26233287Sbostic mtod(nam, caddr_t), (unsigned)nam->m_len); 26337617Smckusick } else 26437617Smckusick nam->m_len = 0; 26514121Ssam break; 26614121Ssam 2678925Sroot case PRU_SLOWTIMO: 2688925Sroot break; 2698925Sroot 2708925Sroot default: 2718925Sroot panic("piusrreq"); 2728925Sroot } 27312760Ssam release: 274*40937Skarels if (control) 275*40937Skarels m_freem(control); 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 || 42840800Ssklower (so3 = sonewconn(so2, 0)) == 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; 46640800Ssklower soisconnected(so); 46714049Ssam soisconnected(so2); 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 register int i; 55540800Ssklower register struct cmsghdr *cm = mtod(rights, struct cmsghdr *); 55640800Ssklower register struct file **rp = (struct file **)(cm + 1); 55712760Ssam register struct file *fp; 55840800Ssklower int newfds = (cm->cmsg_len - sizeof(*cm)) / sizeof (int); 55912760Ssam int f; 56012760Ssam 56112760Ssam if (newfds > ufavail()) { 56212760Ssam for (i = 0; i < newfds; i++) { 56312760Ssam fp = *rp; 56412760Ssam unp_discard(fp); 56512760Ssam *rp++ = 0; 56612760Ssam } 56712760Ssam return (EMSGSIZE); 56812760Ssam } 56912760Ssam for (i = 0; i < newfds; i++) { 57037616Smckusick if (ufalloc(0, &f)) 57112760Ssam panic("unp_externalize"); 57212760Ssam fp = *rp; 57312760Ssam u.u_ofile[f] = fp; 57412760Ssam fp->f_msgcount--; 57525632Skarels unp_rights--; 57614927Smckusick *(int *)rp++ = f; 57712760Ssam } 57812760Ssam return (0); 57912760Ssam } 58012760Ssam 581*40937Skarels unp_internalize(control) 582*40937Skarels struct mbuf *control; 58312760Ssam { 584*40937Skarels register struct cmsghdr *cm = mtod(control, struct cmsghdr *); 58512760Ssam register struct file **rp; 586*40937Skarels register struct file *fp; 58737728Smckusick register int i, fd; 588*40937Skarels int oldfds; 58912760Ssam 590*40937Skarels if (cm->cmsg_type != SCM_RIGHTS || cm->cmsg_level != SOL_SOCKET || 591*40937Skarels cm->cmsg_len != control->m_len) 59240800Ssklower return (EINVAL); 59340800Ssklower oldfds = (cm->cmsg_len - sizeof (*cm)) / sizeof (int); 59440800Ssklower rp = (struct file **)(cm + 1); 59537728Smckusick for (i = 0; i < oldfds; i++) { 59637728Smckusick fd = *(int *)rp++; 59737728Smckusick if ((unsigned)fd >= NOFILE || u.u_ofile[fd] == NULL) 59812760Ssam return (EBADF); 59937728Smckusick } 60040800Ssklower rp = (struct file **)(cm + 1); 60113084Ssam for (i = 0; i < oldfds; i++) { 60237728Smckusick fp = u.u_ofile[*(int *)rp]; 60312760Ssam *rp++ = fp; 60412760Ssam fp->f_count++; 60512760Ssam fp->f_msgcount++; 60625632Skarels unp_rights++; 60712760Ssam } 60812760Ssam return (0); 60912760Ssam } 61012760Ssam 61112760Ssam int unp_defer, unp_gcing; 61212760Ssam int unp_mark(); 61316995Skarels extern struct domain unixdomain; 61412760Ssam 61512760Ssam unp_gc() 61612760Ssam { 61712760Ssam register struct file *fp; 61812760Ssam register struct socket *so; 61912760Ssam 62012760Ssam if (unp_gcing) 62112760Ssam return; 62212760Ssam unp_gcing = 1; 62312760Ssam restart: 62412760Ssam unp_defer = 0; 62512760Ssam for (fp = file; fp < fileNFILE; fp++) 62612760Ssam fp->f_flag &= ~(FMARK|FDEFER); 62712760Ssam do { 62812760Ssam for (fp = file; fp < fileNFILE; fp++) { 62912760Ssam if (fp->f_count == 0) 63012760Ssam continue; 63112760Ssam if (fp->f_flag & FDEFER) { 63212760Ssam fp->f_flag &= ~FDEFER; 63312760Ssam unp_defer--; 63412760Ssam } else { 63512760Ssam if (fp->f_flag & FMARK) 63612760Ssam continue; 63712760Ssam if (fp->f_count == fp->f_msgcount) 63812760Ssam continue; 63912760Ssam fp->f_flag |= FMARK; 64012760Ssam } 64137617Smckusick if (fp->f_type != DTYPE_SOCKET || 64237617Smckusick (so = (struct socket *)fp->f_data) == 0) 64312760Ssam continue; 64416995Skarels if (so->so_proto->pr_domain != &unixdomain || 64521768Skarels (so->so_proto->pr_flags&PR_RIGHTS) == 0) 64612760Ssam continue; 64712760Ssam if (so->so_rcv.sb_flags & SB_LOCK) { 64812760Ssam sbwait(&so->so_rcv); 64912760Ssam goto restart; 65012760Ssam } 65112760Ssam unp_scan(so->so_rcv.sb_mb, unp_mark); 65212760Ssam } 65312760Ssam } while (unp_defer); 65412760Ssam for (fp = file; fp < fileNFILE; fp++) { 65512760Ssam if (fp->f_count == 0) 65612760Ssam continue; 65725632Skarels if (fp->f_count == fp->f_msgcount && (fp->f_flag & FMARK) == 0) 65825632Skarels while (fp->f_msgcount) 65925632Skarels unp_discard(fp); 66012760Ssam } 66112760Ssam unp_gcing = 0; 66212760Ssam } 66312760Ssam 66416995Skarels unp_dispose(m) 66516995Skarels struct mbuf *m; 66616995Skarels { 66716995Skarels int unp_discard(); 66816995Skarels 66917020Skarels if (m) 67017020Skarels unp_scan(m, unp_discard); 67116995Skarels } 67216995Skarels 67316995Skarels unp_scan(m0, op) 67416995Skarels register struct mbuf *m0; 67512760Ssam int (*op)(); 67612760Ssam { 67716995Skarels register struct mbuf *m; 67812760Ssam register struct file **rp; 679*40937Skarels register struct cmsghdr *cm; 68012760Ssam register int i; 68117020Skarels int qfds; 68212760Ssam 68316995Skarels while (m0) { 68416995Skarels for (m = m0; m; m = m->m_next) 685*40937Skarels if (m->m_type == MT_CONTROL && 686*40937Skarels m->m_len >= sizeof(*cm)) { 68740800Ssklower cm = mtod(m, struct cmsghdr *); 688*40937Skarels if (cm->cmsg_level != SOL_SOCKET || 689*40937Skarels cm->cmsg_type != SCM_RIGHTS) 690*40937Skarels continue; 69140800Ssklower qfds = (cm->cmsg_len - sizeof *cm) 69240800Ssklower / sizeof (struct file *); 69340800Ssklower rp = (struct file **)(cm + 1); 69416995Skarels for (i = 0; i < qfds; i++) 69516995Skarels (*op)(*rp++); 69616995Skarels break; /* XXX, but saves time */ 69716995Skarels } 69817020Skarels m0 = m0->m_act; 69912760Ssam } 70012760Ssam } 70112760Ssam 70212760Ssam unp_mark(fp) 70312760Ssam struct file *fp; 70412760Ssam { 70512760Ssam 70612760Ssam if (fp->f_flag & FMARK) 70712760Ssam return; 70812760Ssam unp_defer++; 70912760Ssam fp->f_flag |= (FMARK|FDEFER); 71012760Ssam } 71112760Ssam 71212760Ssam unp_discard(fp) 71312760Ssam struct file *fp; 71412760Ssam { 71512760Ssam 71612760Ssam fp->f_msgcount--; 71725632Skarels unp_rights--; 71839353Smckusick (void) closef(fp); 71912760Ssam } 720