123421Smckusick /* 241908Skarels * Copyright (c) 1982, 1986, 1988, 1990 Regents of the University of California. 333185Sbostic * All rights reserved. 423421Smckusick * 544450Sbostic * %sccs.include.redist.c% 633185Sbostic * 7*46452Ssklower * @(#)uipc_socket.c 7.25 (Berkeley) 02/18/91 823421Smckusick */ 94786Swnj 1017102Sbloom #include "param.h" 1117102Sbloom #include "user.h" 1217102Sbloom #include "proc.h" 1317102Sbloom #include "file.h" 1435384Skarels #include "malloc.h" 1517102Sbloom #include "mbuf.h" 1617102Sbloom #include "domain.h" 1744721Skarels #include "kernel.h" 1817102Sbloom #include "protosw.h" 1917102Sbloom #include "socket.h" 2017102Sbloom #include "socketvar.h" 2144721Skarels #include "time.h" 224786Swnj 234786Swnj /* 248300Sroot * Socket operation routines. 258300Sroot * These routines are called by the routines in 268300Sroot * sys_socket.c or from a system process, and 278300Sroot * implement the semantics of socket operations by 288300Sroot * switching out to the protocol specific routines. 2912757Ssam * 3012757Ssam * TODO: 3112757Ssam * test socketpair 3221767Skarels * clean up async 3312757Ssam * out-of-band is a kludge 344786Swnj */ 358594Sroot /*ARGSUSED*/ 3610267Ssam socreate(dom, aso, type, proto) 374786Swnj struct socket **aso; 3812757Ssam register int type; 3912757Ssam int proto; 404786Swnj { 414786Swnj register struct protosw *prp; 424786Swnj register struct socket *so; 4312757Ssam register int error; 444786Swnj 454890Swnj if (proto) 4621767Skarels prp = pffindproto(dom, proto, type); 474890Swnj else 489168Ssam prp = pffindtype(dom, type); 494890Swnj if (prp == 0) 504890Swnj return (EPROTONOSUPPORT); 518300Sroot if (prp->pr_type != type) 528300Sroot return (EPROTOTYPE); 5337478Ssklower MALLOC(so, struct socket *, sizeof(*so), M_SOCKET, M_WAIT); 5437478Ssklower bzero((caddr_t)so, sizeof(*so)); 559168Ssam so->so_type = type; 566214Swnj if (u.u_uid == 0) 576214Swnj so->so_state = SS_PRIV; 584786Swnj so->so_proto = prp; 5912757Ssam error = 6012757Ssam (*prp->pr_usrreq)(so, PRU_ATTACH, 6121767Skarels (struct mbuf *)0, (struct mbuf *)proto, (struct mbuf *)0); 624979Swnj if (error) { 637507Sroot so->so_state |= SS_NOFDREF; 647180Swnj sofree(so); 654890Swnj return (error); 664786Swnj } 674786Swnj *aso = so; 684786Swnj return (0); 694786Swnj } 704786Swnj 7110267Ssam sobind(so, nam) 728300Sroot struct socket *so; 738300Sroot struct mbuf *nam; 748300Sroot { 758300Sroot int s = splnet(); 768300Sroot int error; 778300Sroot 7841908Skarels error = 7941908Skarels (*so->so_proto->pr_usrreq)(so, PRU_BIND, 8012757Ssam (struct mbuf *)0, nam, (struct mbuf *)0); 818300Sroot splx(s); 828300Sroot return (error); 838300Sroot } 848300Sroot 858300Sroot solisten(so, backlog) 8612757Ssam register struct socket *so; 878300Sroot int backlog; 888300Sroot { 8912757Ssam int s = splnet(), error; 908300Sroot 9112757Ssam error = 9212757Ssam (*so->so_proto->pr_usrreq)(so, PRU_LISTEN, 9312757Ssam (struct mbuf *)0, (struct mbuf *)0, (struct mbuf *)0); 948300Sroot if (error) { 958300Sroot splx(s); 968300Sroot return (error); 978300Sroot } 9838584Skarels if (so->so_q == 0) 998300Sroot so->so_options |= SO_ACCEPTCONN; 1008300Sroot if (backlog < 0) 1018300Sroot backlog = 0; 10235384Skarels so->so_qlimit = min(backlog, SOMAXCONN); 10312493Ssam splx(s); 1048300Sroot return (0); 1058300Sroot } 1068300Sroot 1074916Swnj sofree(so) 10812757Ssam register struct socket *so; 1094916Swnj { 1104916Swnj 11131810Skarels if (so->so_pcb || (so->so_state & SS_NOFDREF) == 0) 11231810Skarels return; 1137507Sroot if (so->so_head) { 1147507Sroot if (!soqremque(so, 0) && !soqremque(so, 1)) 1157507Sroot panic("sofree dq"); 1167507Sroot so->so_head = 0; 1177507Sroot } 1184950Swnj sbrelease(&so->so_snd); 11912757Ssam sorflush(so); 12037478Ssklower FREE(so, M_SOCKET); 1214916Swnj } 1224916Swnj 1234786Swnj /* 1244890Swnj * Close a socket on last file table reference removal. 1254890Swnj * Initiate disconnect if connected. 1264890Swnj * Free socket when disconnect complete. 1274829Swnj */ 12812757Ssam soclose(so) 1294829Swnj register struct socket *so; 1304829Swnj { 1314890Swnj int s = splnet(); /* conservative */ 13233372Sbostic int error = 0; 1334829Swnj 1347507Sroot if (so->so_options & SO_ACCEPTCONN) { 13538584Skarels while (so->so_q0) 13610399Ssam (void) soabort(so->so_q0); 13738584Skarels while (so->so_q) 13810399Ssam (void) soabort(so->so_q); 1397507Sroot } 1404890Swnj if (so->so_pcb == 0) 1414890Swnj goto discard; 1424890Swnj if (so->so_state & SS_ISCONNECTED) { 1434890Swnj if ((so->so_state & SS_ISDISCONNECTING) == 0) { 14426245Skarels error = sodisconnect(so); 14512757Ssam if (error) 14612757Ssam goto drop; 1474890Swnj } 14810267Ssam if (so->so_options & SO_LINGER) { 1495281Sroot if ((so->so_state & SS_ISDISCONNECTING) && 15012757Ssam (so->so_state & SS_NBIO)) 15112757Ssam goto drop; 1525281Sroot while (so->so_state & SS_ISCONNECTED) 15340706Skarels if (error = tsleep((caddr_t)&so->so_timeo, 15440706Skarels PSOCK | PCATCH, netcls, so->so_linger)) 15540706Skarels break; 1564890Swnj } 1574890Swnj } 1585580Sroot drop: 1596880Ssam if (so->so_pcb) { 16012757Ssam int error2 = 16112757Ssam (*so->so_proto->pr_usrreq)(so, PRU_DETACH, 16212757Ssam (struct mbuf *)0, (struct mbuf *)0, (struct mbuf *)0); 16312757Ssam if (error == 0) 16412757Ssam error = error2; 1656880Ssam } 1664890Swnj discard: 16710399Ssam if (so->so_state & SS_NOFDREF) 16810399Ssam panic("soclose: NOFDREF"); 1697507Sroot so->so_state |= SS_NOFDREF; 1704950Swnj sofree(so); 1714890Swnj splx(s); 17212757Ssam return (error); 1734829Swnj } 1744829Swnj 17510399Ssam /* 17610399Ssam * Must be called at splnet... 17710399Ssam */ 17810399Ssam soabort(so) 17910399Ssam struct socket *so; 18010399Ssam { 18110399Ssam 18212757Ssam return ( 18312757Ssam (*so->so_proto->pr_usrreq)(so, PRU_ABORT, 18412757Ssam (struct mbuf *)0, (struct mbuf *)0, (struct mbuf *)0)); 18510399Ssam } 18610399Ssam 18710267Ssam soaccept(so, nam) 18812757Ssam register struct socket *so; 1898300Sroot struct mbuf *nam; 1904927Swnj { 1914927Swnj int s = splnet(); 1924927Swnj int error; 1934927Swnj 19410399Ssam if ((so->so_state & SS_NOFDREF) == 0) 19510399Ssam panic("soaccept: !NOFDREF"); 19610267Ssam so->so_state &= ~SS_NOFDREF; 1978300Sroot error = (*so->so_proto->pr_usrreq)(so, PRU_ACCEPT, 19812757Ssam (struct mbuf *)0, nam, (struct mbuf *)0); 1994927Swnj splx(s); 2004927Swnj return (error); 2014927Swnj } 2024927Swnj 20310267Ssam soconnect(so, nam) 20412757Ssam register struct socket *so; 2058300Sroot struct mbuf *nam; 2064786Swnj { 20730414Skarels int s; 2084890Swnj int error; 2094786Swnj 21030414Skarels if (so->so_options & SO_ACCEPTCONN) 21130414Skarels return (EOPNOTSUPP); 21230414Skarels s = splnet(); 21324768Skarels /* 21424768Skarels * If protocol is connection-based, can only connect once. 21524768Skarels * Otherwise, if connected, try to disconnect first. 21624768Skarels * This allows user to disconnect by connecting to, e.g., 21724768Skarels * a null address. 21824768Skarels */ 21924768Skarels if (so->so_state & (SS_ISCONNECTED|SS_ISCONNECTING) && 22024768Skarels ((so->so_proto->pr_flags & PR_CONNREQUIRED) || 22124768Skarels (error = sodisconnect(so)))) 2224890Swnj error = EISCONN; 22324768Skarels else 22424768Skarels error = (*so->so_proto->pr_usrreq)(so, PRU_CONNECT, 22524768Skarels (struct mbuf *)0, nam, (struct mbuf *)0); 2264890Swnj splx(s); 2274890Swnj return (error); 2284786Swnj } 2294786Swnj 23012757Ssam soconnect2(so1, so2) 23112757Ssam register struct socket *so1; 23212757Ssam struct socket *so2; 23312757Ssam { 23412757Ssam int s = splnet(); 23512757Ssam int error; 23612757Ssam 23713113Ssam error = (*so1->so_proto->pr_usrreq)(so1, PRU_CONNECT2, 23813113Ssam (struct mbuf *)0, (struct mbuf *)so2, (struct mbuf *)0); 23912757Ssam splx(s); 24012757Ssam return (error); 24112757Ssam } 24212757Ssam 24326245Skarels sodisconnect(so) 24412757Ssam register struct socket *so; 2454786Swnj { 2464890Swnj int s = splnet(); 2474890Swnj int error; 2484786Swnj 2494890Swnj if ((so->so_state & SS_ISCONNECTED) == 0) { 2504890Swnj error = ENOTCONN; 2514890Swnj goto bad; 2524890Swnj } 2534890Swnj if (so->so_state & SS_ISDISCONNECTING) { 2544890Swnj error = EALREADY; 2554890Swnj goto bad; 2564890Swnj } 2578300Sroot error = (*so->so_proto->pr_usrreq)(so, PRU_DISCONNECT, 25826245Skarels (struct mbuf *)0, (struct mbuf *)0, (struct mbuf *)0); 2594890Swnj bad: 2604890Swnj splx(s); 2614890Swnj return (error); 2624786Swnj } 2634786Swnj 2644786Swnj /* 2654890Swnj * Send on a socket. 2664890Swnj * If send must go all at once and message is larger than 2674890Swnj * send buffering, then hard error. 2684890Swnj * Lock against other senders. 2694890Swnj * If must go all at once and not enough room now, then 2704890Swnj * inform user that this would block and do nothing. 27116412Skarels * Otherwise, if nonblocking, send as much as possible. 27241908Skarels * The data to be sent is described by "uio" if nonzero, 27341908Skarels * otherwise by the mbuf chain "top" (which must be null 27441908Skarels * if uio is not). Data provided in mbuf chain must be small 27541908Skarels * enough to send all at once. 27641908Skarels * 27741908Skarels * Returns nonzero on error, timeout or signal; callers 27841908Skarels * must check for short counts if EINTR/ERESTART are returned. 27941908Skarels * Data and control buffers are freed on return. 2804786Swnj */ 28143417Skarels sosend(so, addr, uio, top, control, flags) 2824786Swnj register struct socket *so; 28341908Skarels struct mbuf *addr; 28441908Skarels struct uio *uio; 28541908Skarels struct mbuf *top; 28641908Skarels struct mbuf *control; 2878319Sroot int flags; 2884786Swnj { 28941908Skarels struct mbuf **mp; 29035384Skarels register struct mbuf *m; 29141908Skarels register long space, len, resid; 29241908Skarels int clen = 0, error, s, dontroute, mlen; 29341908Skarels int atomic = sosendallatonce(so) || top; 2944786Swnj 29541908Skarels if (uio) 29641908Skarels resid = uio->uio_resid; 29741908Skarels else 29841908Skarels resid = top->m_pkthdr.len; 29912757Ssam dontroute = 30012757Ssam (flags & MSG_DONTROUTE) && (so->so_options & SO_DONTROUTE) == 0 && 30112757Ssam (so->so_proto->pr_flags & PR_ATOMIC); 30216412Skarels u.u_ru.ru_msgsnd++; 30340632Skarels if (control) 30441908Skarels clen = control->m_len; 30516412Skarels #define snderr(errno) { error = errno; splx(s); goto release; } 30616412Skarels 3076419Sroot restart: 30840706Skarels if (error = sblock(&so->so_snd)) 30941908Skarels goto out; 31016412Skarels do { 31116412Skarels s = splnet(); 31221108Skarels if (so->so_state & SS_CANTSENDMORE) 31316412Skarels snderr(EPIPE); 31437478Ssklower if (so->so_error) 31537478Ssklower snderr(so->so_error); 31616412Skarels if ((so->so_state & SS_ISCONNECTED) == 0) { 31737478Ssklower if (so->so_proto->pr_flags & PR_CONNREQUIRED) { 31838584Skarels if ((so->so_state & SS_ISCONFIRMING) == 0) 31937478Ssklower snderr(ENOTCONN); 32041908Skarels } else if (addr == 0) 32116412Skarels snderr(EDESTADDRREQ); 32216412Skarels } 32341908Skarels space = sbspace(&so->so_snd); 32416412Skarels if (flags & MSG_OOB) 32541908Skarels space += 1024; 32641908Skarels if (space < resid + clen && 32741908Skarels (atomic || space < so->so_snd.sb_lowat || space < clen)) { 32841908Skarels if (atomic && resid > so->so_snd.sb_hiwat || 32941908Skarels clen > so->so_snd.sb_hiwat) 33041908Skarels snderr(EMSGSIZE); 33141908Skarels if (so->so_state & SS_NBIO) 33241908Skarels snderr(EWOULDBLOCK); 33341908Skarels sbunlock(&so->so_snd); 33443417Skarels error = sbwait(&so->so_snd); 33541908Skarels splx(s); 33641908Skarels if (error) 33741908Skarels goto out; 33841908Skarels goto restart; 33916412Skarels } 34016412Skarels splx(s); 34116412Skarels mp = ⊤ 34241908Skarels space -= clen; 34343417Skarels do { 34443417Skarels if (uio == NULL) { 34541908Skarels /* 34641908Skarels * Data is prepackaged in "top". 34741908Skarels */ 34841908Skarels resid = 0; 34941908Skarels if (flags & MSG_EOR) 35041908Skarels top->m_flags |= M_EOR; 35143417Skarels } else do { 35235384Skarels if (top == 0) { 35335384Skarels MGETHDR(m, M_WAIT, MT_DATA); 35435384Skarels mlen = MHLEN; 35535384Skarels m->m_pkthdr.len = 0; 35635384Skarels m->m_pkthdr.rcvif = (struct ifnet *)0; 35735384Skarels } else { 35835384Skarels MGET(m, M_WAIT, MT_DATA); 35935384Skarels mlen = MLEN; 36035384Skarels } 36141908Skarels if (resid >= MINCLSIZE && space >= MCLBYTES) { 36235384Skarels MCLGET(m, M_WAIT); 36335384Skarels if ((m->m_flags & M_EXT) == 0) 36416412Skarels goto nopages; 36535384Skarels mlen = MCLBYTES; 36635384Skarels #ifdef MAPPED_MBUFS 36741908Skarels len = min(MCLBYTES, resid); 36841908Skarels #else 36941908Skarels if (top == 0) { 37041908Skarels len = min(MCLBYTES - max_hdr, resid); 37135384Skarels m->m_data += max_hdr; 372*46452Ssklower } else 373*46452Ssklower len = min(MCLBYTES, resid); 37435384Skarels #endif 37535384Skarels space -= MCLBYTES; 37616412Skarels } else { 37716412Skarels nopages: 37841908Skarels len = min(min(mlen, resid), space); 37921767Skarels space -= len; 38035384Skarels /* 38135384Skarels * For datagram protocols, leave room 38235384Skarels * for protocol headers in first mbuf. 38335384Skarels */ 38435391Skarels if (atomic && top == 0 && len < mlen) 38535384Skarels MH_ALIGN(m, len); 38616412Skarels } 38745515Skarels error = uiomove(mtod(m, caddr_t), (int)len, uio); 38841908Skarels resid = uio->uio_resid; 38916412Skarels m->m_len = len; 39016412Skarels *mp = m; 39135384Skarels top->m_pkthdr.len += len; 39216412Skarels if (error) 39316412Skarels goto release; 39416412Skarels mp = &m->m_next; 39541908Skarels if (resid <= 0) { 39641908Skarels if (flags & MSG_EOR) 39735384Skarels top->m_flags |= M_EOR; 39821108Skarels break; 39935384Skarels } 40035384Skarels } while (space > 0 && atomic); 40135384Skarels if (dontroute) 40235384Skarels so->so_options |= SO_DONTROUTE; 40335384Skarels s = splnet(); /* XXX */ 40435384Skarels error = (*so->so_proto->pr_usrreq)(so, 40535384Skarels (flags & MSG_OOB) ? PRU_SENDOOB : PRU_SEND, 40641908Skarels top, addr, control); 40735384Skarels splx(s); 40835384Skarels if (dontroute) 40935384Skarels so->so_options &= ~SO_DONTROUTE; 41041908Skarels clen = 0; 41141908Skarels control = 0; 41235384Skarels top = 0; 41335384Skarels mp = ⊤ 41435384Skarels if (error) 41535384Skarels goto release; 41641908Skarels } while (resid && space > 0); 41741908Skarels } while (resid); 4184890Swnj 4194786Swnj release: 4204890Swnj sbunlock(&so->so_snd); 42141908Skarels out: 4226419Sroot if (top) 4236419Sroot m_freem(top); 42441908Skarels if (control) 42541908Skarels m_freem(control); 4264786Swnj return (error); 4274786Swnj } 4284786Swnj 42925629Skarels /* 43025629Skarels * Implement receive operations on a socket. 43125629Skarels * We depend on the way that records are added to the sockbuf 43225629Skarels * by sbappend*. In particular, each record (mbufs linked through m_next) 43325629Skarels * must begin with an address if the protocol so specifies, 43441908Skarels * followed by an optional mbuf or mbufs containing ancillary data, 43541908Skarels * and then zero or more mbufs of data. 43625629Skarels * In order to avoid blocking network interrupts for the entire time here, 43725629Skarels * we splx() while doing the actual copy to user space. 43825629Skarels * Although the sockbuf is locked, new data may still be appended, 43925629Skarels * and thus we must maintain consistency of the sockbuf during that time. 44045515Skarels * 44141908Skarels * The caller may receive the data as a single mbuf chain by supplying 44243417Skarels * an mbuf **mp0 for use in returning the chain. The uio is then used 44341908Skarels * only for the count in uio_resid. 44425629Skarels */ 44543417Skarels soreceive(so, paddr, uio, mp0, controlp, flagsp) 4464786Swnj register struct socket *so; 44741908Skarels struct mbuf **paddr; 44841908Skarels struct uio *uio; 44943417Skarels struct mbuf **mp0; 45041908Skarels struct mbuf **controlp; 45135384Skarels int *flagsp; 4524786Swnj { 45343417Skarels register struct mbuf *m, **mp; 45443417Skarels register int flags, len, error, s, offset; 45512757Ssam struct protosw *pr = so->so_proto; 45641908Skarels struct mbuf *nextrecord; 45741908Skarels int moff, type; 4584786Swnj 45943417Skarels mp = mp0; 46041908Skarels if (paddr) 46141908Skarels *paddr = 0; 46235384Skarels if (controlp) 46335384Skarels *controlp = 0; 46435384Skarels if (flagsp) 46535384Skarels flags = *flagsp &~ MSG_EOR; 46645515Skarels else 46735384Skarels flags = 0; 46812757Ssam if (flags & MSG_OOB) { 4699635Ssam m = m_get(M_WAIT, MT_DATA); 47012757Ssam error = (*pr->pr_usrreq)(so, PRU_RCVOOB, 47124768Skarels m, (struct mbuf *)(flags & MSG_PEEK), (struct mbuf *)0); 4728594Sroot if (error) 47310137Ssam goto bad; 4748319Sroot do { 47541908Skarels error = uiomove(mtod(m, caddr_t), 47641908Skarels (int) min(uio->uio_resid, m->m_len), uio); 4778319Sroot m = m_free(m); 4788594Sroot } while (uio->uio_resid && error == 0 && m); 47910137Ssam bad: 4808319Sroot if (m) 4818771Sroot m_freem(m); 4828594Sroot return (error); 4838319Sroot } 48441908Skarels if (mp) 48541908Skarels *mp = (struct mbuf *)0; 48643417Skarels if (so->so_state & SS_ISCONFIRMING && uio->uio_resid) 48735384Skarels (*pr->pr_usrreq)(so, PRU_RCVD, (struct mbuf *)0, 48835384Skarels (struct mbuf *)0, (struct mbuf *)0); 4898319Sroot 4904890Swnj restart: 49140706Skarels if (error = sblock(&so->so_rcv)) 49240706Skarels return (error); 4938835Sroot s = splnet(); 4944890Swnj 49537478Ssklower m = so->so_rcv.sb_mb; 49644383Skarels /* 49744383Skarels * If we have less data than requested, block awaiting more 49844383Skarels * (subject to any timeout) if: 49944383Skarels * 1. the current count is less than the low water mark, or 50044383Skarels * 2. MSG_WAITALL is set, and it is possible to do the entire 50144383Skarels * receive operation at once if we block (resid <= hiwat). 50244383Skarels * If MSG_WAITALL is set but resid is larger than the receive buffer, 50344383Skarels * we have to do the receive in sections, and thus risk returning 50444383Skarels * a short count if a timeout or signal occurs after we start. 50544383Skarels */ 50645515Skarels if (m == 0 || so->so_rcv.sb_cc < uio->uio_resid && 50744383Skarels (so->so_rcv.sb_cc < so->so_rcv.sb_lowat || 50844383Skarels ((flags & MSG_WAITALL) && uio->uio_resid <= so->so_rcv.sb_hiwat))) { 50941908Skarels #ifdef DIAGNOSTIC 51041908Skarels if (m == 0 && so->so_rcv.sb_cc) 51137478Ssklower panic("receive 1"); 51241908Skarels #endif 5135168Swnj if (so->so_error) { 5145168Swnj error = so->so_error; 5155168Swnj so->so_error = 0; 5165168Swnj goto release; 5175168Swnj } 51832567Sbostic if (so->so_state & SS_CANTRCVMORE) 5194890Swnj goto release; 52038584Skarels if ((so->so_state & (SS_ISCONNECTED|SS_ISCONNECTING)) == 0 && 52132567Sbostic (so->so_proto->pr_flags & PR_CONNREQUIRED)) { 52232567Sbostic error = ENOTCONN; 52332567Sbostic goto release; 5244890Swnj } 52543417Skarels if (uio->uio_resid == 0) 52625629Skarels goto release; 52732567Sbostic if (so->so_state & SS_NBIO) { 52832567Sbostic error = EWOULDBLOCK; 52932567Sbostic goto release; 53032567Sbostic } 5314890Swnj sbunlock(&so->so_rcv); 53243417Skarels error = sbwait(&so->so_rcv); 5335012Swnj splx(s); 53441908Skarels if (error) 53541908Skarels return (error); 5364890Swnj goto restart; 5374786Swnj } 5388041Sroot u.u_ru.ru_msgrcv++; 53935384Skarels nextrecord = m->m_nextpkt; 54012757Ssam if (pr->pr_flags & PR_ADDR) { 54141908Skarels #ifdef DIAGNOSTIC 54225629Skarels if (m->m_type != MT_SONAME) 54316993Skarels panic("receive 1a"); 54441908Skarels #endif 54516993Skarels if (flags & MSG_PEEK) { 54641908Skarels if (paddr) 54741908Skarels *paddr = m_copy(m, 0, m->m_len); 54825629Skarels m = m->m_next; 54916993Skarels } else { 55025629Skarels sbfree(&so->so_rcv, m); 55141908Skarels if (paddr) { 55241908Skarels *paddr = m; 55335384Skarels so->so_rcv.sb_mb = m->m_next; 55435384Skarels m->m_next = 0; 55535384Skarels m = so->so_rcv.sb_mb; 55625629Skarels } else { 55726958Skarels MFREE(m, so->so_rcv.sb_mb); 55826958Skarels m = so->so_rcv.sb_mb; 55925629Skarels } 56016993Skarels } 56116993Skarels } 56241908Skarels while (m && m->m_type == MT_CONTROL && error == 0) { 56316993Skarels if (flags & MSG_PEEK) { 56435384Skarels if (controlp) 56535384Skarels *controlp = m_copy(m, 0, m->m_len); 56635384Skarels m = m->m_next; 56735384Skarels } else { 56835384Skarels sbfree(&so->so_rcv, m); 56935384Skarels if (controlp) { 57043097Ssklower if (pr->pr_domain->dom_externalize && 57143097Ssklower mtod(m, struct cmsghdr *)->cmsg_type == 57243097Ssklower SCM_RIGHTS) 57341908Skarels error = (*pr->pr_domain->dom_externalize)(m); 57435384Skarels *controlp = m; 57535384Skarels so->so_rcv.sb_mb = m->m_next; 57635384Skarels m->m_next = 0; 57735384Skarels m = so->so_rcv.sb_mb; 57835384Skarels } else { 57935384Skarels MFREE(m, so->so_rcv.sb_mb); 58035384Skarels m = so->so_rcv.sb_mb; 58135384Skarels } 58235384Skarels } 58341908Skarels if (controlp) 58441908Skarels controlp = &(*controlp)->m_next; 58535384Skarels } 58641908Skarels if (m) { 58744383Skarels if ((flags & MSG_PEEK) == 0) 58844383Skarels m->m_nextpkt = nextrecord; 58941908Skarels type = m->m_type; 590*46452Ssklower if (type == MT_OOBDATA) 591*46452Ssklower flags |= MSG_OOB; 59241908Skarels } 5938319Sroot moff = 0; 59432092Skarels offset = 0; 595*46452Ssklower while (m && uio->uio_resid > 0 && error == 0) { 596*46452Ssklower if (m->m_type == MT_OOBDATA) { 597*46452Ssklower if (type != MT_OOBDATA) 598*46452Ssklower break; 599*46452Ssklower } else if (type == MT_OOBDATA) 600*46452Ssklower break; 60141908Skarels #ifdef DIAGNOSTIC 60235384Skarels else if (m->m_type != MT_DATA && m->m_type != MT_HEADER) 60325629Skarels panic("receive 3"); 60441908Skarels #endif 6057747Sroot so->so_state &= ~SS_RCVATMARK; 60643417Skarels len = uio->uio_resid; 60732092Skarels if (so->so_oobmark && len > so->so_oobmark - offset) 60832092Skarels len = so->so_oobmark - offset; 60921767Skarels if (len > m->m_len - moff) 6108319Sroot len = m->m_len - moff; 61141908Skarels /* 61241908Skarels * If mp is set, just pass back the mbufs. 61341908Skarels * Otherwise copy them out via the uio, then free. 61441908Skarels * Sockbuf must be consistent here (points to current mbuf, 61541908Skarels * it points to next record) when we drop priority; 61641908Skarels * we must note any additions to the sockbuf when we 61741908Skarels * block interrupts again. 61841908Skarels */ 61941908Skarels if (mp == 0) { 62041908Skarels splx(s); 62141908Skarels error = uiomove(mtod(m, caddr_t) + moff, (int)len, uio); 62241908Skarels s = splnet(); 62343417Skarels } else 62443417Skarels uio->uio_resid -= len; 62521767Skarels if (len == m->m_len - moff) { 62641908Skarels if (m->m_flags & M_EOR) 62741908Skarels flags |= MSG_EOR; 62825629Skarels if (flags & MSG_PEEK) { 62925629Skarels m = m->m_next; 63025629Skarels moff = 0; 63125629Skarels } else { 63235384Skarels nextrecord = m->m_nextpkt; 63325629Skarels sbfree(&so->so_rcv, m); 63441908Skarels if (mp) { 63541908Skarels *mp = m; 63641908Skarels mp = &m->m_next; 63743417Skarels so->so_rcv.sb_mb = m = m->m_next; 63843417Skarels *mp = (struct mbuf *)0; 63941908Skarels } else { 64041908Skarels MFREE(m, so->so_rcv.sb_mb); 64141908Skarels m = so->so_rcv.sb_mb; 64241908Skarels } 64326958Skarels if (m) 64435384Skarels m->m_nextpkt = nextrecord; 64525629Skarels } 6464786Swnj } else { 64712757Ssam if (flags & MSG_PEEK) 6488319Sroot moff += len; 6498319Sroot else { 65043417Skarels if (mp) 65143417Skarels *mp = m_copym(m, 0, len, M_WAIT); 65235384Skarels m->m_data += len; 6538319Sroot m->m_len -= len; 6548319Sroot so->so_rcv.sb_cc -= len; 6558319Sroot } 6564786Swnj } 65732092Skarels if (so->so_oobmark) { 65832092Skarels if ((flags & MSG_PEEK) == 0) { 65932092Skarels so->so_oobmark -= len; 66032092Skarels if (so->so_oobmark == 0) { 66132092Skarels so->so_state |= SS_RCVATMARK; 66232092Skarels break; 66332092Skarels } 66432092Skarels } else 66532092Skarels offset += len; 6667747Sroot } 66741908Skarels if (flags & MSG_EOR) 66840632Skarels break; 66941908Skarels /* 67041908Skarels * If the MSG_WAITALL flag is set (for non-atomic socket), 67143417Skarels * we must not quit until "uio->uio_resid == 0" or an error 67241908Skarels * termination. If a signal/timeout occurs, return 67343417Skarels * with a short count but without error. 67441908Skarels * Keep sockbuf locked against other readers. 67541908Skarels */ 67643417Skarels while (flags & MSG_WAITALL && m == 0 && uio->uio_resid > 0 && 67741908Skarels !sosendallatonce(so)) { 67841908Skarels error = sbwait(&so->so_rcv); 67941908Skarels if (error) { 68041908Skarels sbunlock(&so->so_rcv); 68141908Skarels splx(s); 68241908Skarels return (0); 68341908Skarels } 68441908Skarels if (m = so->so_rcv.sb_mb) 68541908Skarels nextrecord = m->m_nextpkt; 68641908Skarels if (so->so_error || so->so_state & SS_CANTRCVMORE) 68741908Skarels break; 68841908Skarels continue; 68941908Skarels } 69016993Skarels } 69116993Skarels if ((flags & MSG_PEEK) == 0) { 69226500Skarels if (m == 0) 69316993Skarels so->so_rcv.sb_mb = nextrecord; 69435384Skarels else if (pr->pr_flags & PR_ATOMIC) { 69535384Skarels flags |= MSG_TRUNC; 69626958Skarels (void) sbdroprecord(&so->so_rcv); 69735384Skarels } 69816993Skarels if (pr->pr_flags & PR_WANTRCVD && so->so_pcb) 69916993Skarels (*pr->pr_usrreq)(so, PRU_RCVD, (struct mbuf *)0, 70037478Ssklower (struct mbuf *)flags, (struct mbuf *)0, 70137478Ssklower (struct mbuf *)0); 70216993Skarels } 70335384Skarels if (flagsp) 70435384Skarels *flagsp |= flags; 7054890Swnj release: 7064916Swnj sbunlock(&so->so_rcv); 7074890Swnj splx(s); 7084916Swnj return (error); 7094786Swnj } 7104786Swnj 71110267Ssam soshutdown(so, how) 71212757Ssam register struct socket *so; 71312757Ssam register int how; 71410267Ssam { 71512757Ssam register struct protosw *pr = so->so_proto; 71610267Ssam 71710267Ssam how++; 71812757Ssam if (how & FREAD) 71912757Ssam sorflush(so); 72010267Ssam if (how & FWRITE) 72112757Ssam return ((*pr->pr_usrreq)(so, PRU_SHUTDOWN, 72212757Ssam (struct mbuf *)0, (struct mbuf *)0, (struct mbuf *)0)); 72310267Ssam return (0); 72410267Ssam } 72510267Ssam 72612757Ssam sorflush(so) 72712757Ssam register struct socket *so; 72812757Ssam { 72912757Ssam register struct sockbuf *sb = &so->so_rcv; 73012757Ssam register struct protosw *pr = so->so_proto; 73112757Ssam register int s; 73212757Ssam struct sockbuf asb; 73312757Ssam 73440706Skarels sb->sb_flags |= SB_NOINTR; 73540706Skarels (void) sblock(sb); 73612757Ssam s = splimp(); 73712757Ssam socantrcvmore(so); 73812757Ssam sbunlock(sb); 73912757Ssam asb = *sb; 74012757Ssam bzero((caddr_t)sb, sizeof (*sb)); 74112757Ssam splx(s); 74216993Skarels if (pr->pr_flags & PR_RIGHTS && pr->pr_domain->dom_dispose) 74316993Skarels (*pr->pr_domain->dom_dispose)(asb.sb_mb); 74412757Ssam sbrelease(&asb); 74512757Ssam } 74612757Ssam 74718553Skarels sosetopt(so, level, optname, m0) 74812757Ssam register struct socket *so; 74910267Ssam int level, optname; 75018553Skarels struct mbuf *m0; 75110267Ssam { 75217158Ssam int error = 0; 75318553Skarels register struct mbuf *m = m0; 75410267Ssam 75517158Ssam if (level != SOL_SOCKET) { 75618369Skarels if (so->so_proto && so->so_proto->pr_ctloutput) 75718369Skarels return ((*so->so_proto->pr_ctloutput) 75818553Skarels (PRCO_SETOPT, so, level, optname, &m0)); 75918369Skarels error = ENOPROTOOPT; 76018369Skarels } else { 76118369Skarels switch (optname) { 76210267Ssam 76318369Skarels case SO_LINGER: 76418369Skarels if (m == NULL || m->m_len != sizeof (struct linger)) { 76518369Skarels error = EINVAL; 76618369Skarels goto bad; 76718369Skarels } 76818369Skarels so->so_linger = mtod(m, struct linger *)->l_linger; 76918369Skarels /* fall thru... */ 77017158Ssam 77118369Skarels case SO_DEBUG: 77218369Skarels case SO_KEEPALIVE: 77318369Skarels case SO_DONTROUTE: 77418369Skarels case SO_USELOOPBACK: 77518369Skarels case SO_BROADCAST: 77618369Skarels case SO_REUSEADDR: 77727191Skarels case SO_OOBINLINE: 77818369Skarels if (m == NULL || m->m_len < sizeof (int)) { 77918369Skarels error = EINVAL; 78018369Skarels goto bad; 78118369Skarels } 78218369Skarels if (*mtod(m, int *)) 78318369Skarels so->so_options |= optname; 78418369Skarels else 78518369Skarels so->so_options &= ~optname; 78618369Skarels break; 78718369Skarels 78818369Skarels case SO_SNDBUF: 78941908Skarels case SO_RCVBUF: 79040706Skarels case SO_SNDLOWAT: 79141908Skarels case SO_RCVLOWAT: 79218369Skarels if (m == NULL || m->m_len < sizeof (int)) { 79318369Skarels error = EINVAL; 79418369Skarels goto bad; 79518369Skarels } 79618369Skarels switch (optname) { 79718369Skarels 79818369Skarels case SO_SNDBUF: 79918369Skarels case SO_RCVBUF: 80041908Skarels if (sbreserve(optname == SO_SNDBUF ? 80141908Skarels &so->so_snd : &so->so_rcv, 80241908Skarels (u_long) *mtod(m, int *)) == 0) { 80318369Skarels error = ENOBUFS; 80418369Skarels goto bad; 80518369Skarels } 80618369Skarels break; 80718369Skarels 80818369Skarels case SO_SNDLOWAT: 80941908Skarels so->so_snd.sb_lowat = *mtod(m, int *); 81041908Skarels break; 81118369Skarels case SO_RCVLOWAT: 81241908Skarels so->so_rcv.sb_lowat = *mtod(m, int *); 81318369Skarels break; 81444721Skarels } 81544721Skarels break; 81644721Skarels 81744721Skarels case SO_SNDTIMEO: 81844721Skarels case SO_RCVTIMEO: 81944721Skarels { 82044721Skarels struct timeval *tv; 82144721Skarels short val; 82244721Skarels 82344721Skarels if (m == NULL || m->m_len < sizeof (*tv)) { 82444721Skarels error = EINVAL; 82544721Skarels goto bad; 82644721Skarels } 82744721Skarels tv = mtod(m, struct timeval *); 82844721Skarels if (tv->tv_sec > SHRT_MAX / hz - hz) { 82944721Skarels error = EDOM; 83044721Skarels goto bad; 83144721Skarels } 83244721Skarels val = tv->tv_sec * hz + tv->tv_usec / tick; 83344721Skarels 83444721Skarels switch (optname) { 83544721Skarels 83618369Skarels case SO_SNDTIMEO: 83744721Skarels so->so_snd.sb_timeo = val; 83841908Skarels break; 83918369Skarels case SO_RCVTIMEO: 84044721Skarels so->so_rcv.sb_timeo = val; 84118369Skarels break; 84218369Skarels } 84318369Skarels break; 84444721Skarels } 84518369Skarels 84618369Skarels default: 84718369Skarels error = ENOPROTOOPT; 84818369Skarels break; 84917158Ssam } 85010267Ssam } 85117158Ssam bad: 85217158Ssam if (m) 85317158Ssam (void) m_free(m); 85417158Ssam return (error); 85510267Ssam } 85610267Ssam 85717158Ssam sogetopt(so, level, optname, mp) 85812757Ssam register struct socket *so; 85910267Ssam int level, optname; 86017158Ssam struct mbuf **mp; 86117158Ssam { 86212757Ssam register struct mbuf *m; 86310267Ssam 86418369Skarels if (level != SOL_SOCKET) { 86518369Skarels if (so->so_proto && so->so_proto->pr_ctloutput) { 86618369Skarels return ((*so->so_proto->pr_ctloutput) 86718369Skarels (PRCO_GETOPT, so, level, optname, mp)); 86845515Skarels } else 86918369Skarels return (ENOPROTOOPT); 87018369Skarels } else { 87117158Ssam m = m_get(M_WAIT, MT_SOOPTS); 87225502Skarels m->m_len = sizeof (int); 87325502Skarels 87418369Skarels switch (optname) { 87517158Ssam 87618369Skarels case SO_LINGER: 87718369Skarels m->m_len = sizeof (struct linger); 87818369Skarels mtod(m, struct linger *)->l_onoff = 87918369Skarels so->so_options & SO_LINGER; 88018369Skarels mtod(m, struct linger *)->l_linger = so->so_linger; 88118369Skarels break; 88210267Ssam 88318369Skarels case SO_USELOOPBACK: 88418369Skarels case SO_DONTROUTE: 88518369Skarels case SO_DEBUG: 88618369Skarels case SO_KEEPALIVE: 88718369Skarels case SO_REUSEADDR: 88818369Skarels case SO_BROADCAST: 88927191Skarels case SO_OOBINLINE: 89018369Skarels *mtod(m, int *) = so->so_options & optname; 89118369Skarels break; 89218369Skarels 89325502Skarels case SO_TYPE: 89425502Skarels *mtod(m, int *) = so->so_type; 89525502Skarels break; 89625502Skarels 89724768Skarels case SO_ERROR: 89824768Skarels *mtod(m, int *) = so->so_error; 89924768Skarels so->so_error = 0; 90024768Skarels break; 90124768Skarels 90218369Skarels case SO_SNDBUF: 90318369Skarels *mtod(m, int *) = so->so_snd.sb_hiwat; 90418369Skarels break; 90518369Skarels 90618369Skarels case SO_RCVBUF: 90718369Skarels *mtod(m, int *) = so->so_rcv.sb_hiwat; 90818369Skarels break; 90918369Skarels 91018369Skarels case SO_SNDLOWAT: 91118369Skarels *mtod(m, int *) = so->so_snd.sb_lowat; 91218369Skarels break; 91318369Skarels 91418369Skarels case SO_RCVLOWAT: 91518369Skarels *mtod(m, int *) = so->so_rcv.sb_lowat; 91618369Skarels break; 91718369Skarels 91818369Skarels case SO_SNDTIMEO: 91944721Skarels case SO_RCVTIMEO: 92044721Skarels { 92144721Skarels int val = (optname == SO_SNDTIMEO ? 92244721Skarels so->so_snd.sb_timeo : so->so_rcv.sb_timeo); 92318369Skarels 92444721Skarels m->m_len = sizeof(struct timeval); 92544721Skarels mtod(m, struct timeval *)->tv_sec = val / hz; 92644721Skarels mtod(m, struct timeval *)->tv_usec = 92744721Skarels (val % hz) / tick; 92818369Skarels break; 92944721Skarels } 93018369Skarels 93118369Skarels default: 93226362Skarels (void)m_free(m); 93318369Skarels return (ENOPROTOOPT); 93418369Skarels } 93518369Skarels *mp = m; 93618369Skarels return (0); 93710267Ssam } 93810267Ssam } 93910267Ssam 9405423Swnj sohasoutofband(so) 94112757Ssam register struct socket *so; 9425423Swnj { 94323233Skarels struct proc *p; 9445423Swnj 94537478Ssklower if (so->so_pgid < 0) 94637478Ssklower gsignal(-so->so_pgid, SIGURG); 94737478Ssklower else if (so->so_pgid > 0 && (p = pfind(so->so_pgid)) != 0) 94823233Skarels psignal(p, SIGURG); 94924768Skarels if (so->so_rcv.sb_sel) { 95024768Skarels selwakeup(so->so_rcv.sb_sel, so->so_rcv.sb_flags & SB_COLL); 95124768Skarels so->so_rcv.sb_sel = 0; 95224768Skarels so->so_rcv.sb_flags &= ~SB_COLL; 95324768Skarels } 9545423Swnj } 955