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*46479Skarels * @(#)uipc_socket.c 7.26 (Berkeley) 02/19/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; 37246452Ssklower } else 37346452Ssklower 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 */ 506*46479Skarels while (m == 0 || so->so_rcv.sb_cc < uio->uio_resid && 50744383Skarels (so->so_rcv.sb_cc < so->so_rcv.sb_lowat || 508*46479Skarels ((flags & MSG_WAITALL) && uio->uio_resid <= so->so_rcv.sb_hiwat)) && 509*46479Skarels m->m_nextpkt == 0) { 51041908Skarels #ifdef DIAGNOSTIC 51141908Skarels if (m == 0 && so->so_rcv.sb_cc) 51237478Ssklower panic("receive 1"); 51341908Skarels #endif 5145168Swnj if (so->so_error) { 515*46479Skarels if (m) 516*46479Skarels break; 5175168Swnj error = so->so_error; 518*46479Skarels if ((flags & MSG_PEEK) == 0) 519*46479Skarels so->so_error = 0; 5205168Swnj goto release; 5215168Swnj } 522*46479Skarels if (so->so_state & SS_CANTRCVMORE) { 523*46479Skarels if (m) 524*46479Skarels break; 525*46479Skarels else 526*46479Skarels goto release; 527*46479Skarels } 528*46479Skarels for (; m; m = m->m_next) 529*46479Skarels if (m->m_type == MT_OOBDATA || (m->m_flags & M_EOR)) { 530*46479Skarels m = so->so_rcv.sb_mb; 531*46479Skarels goto dontblock; 532*46479Skarels } 53338584Skarels if ((so->so_state & (SS_ISCONNECTED|SS_ISCONNECTING)) == 0 && 53432567Sbostic (so->so_proto->pr_flags & PR_CONNREQUIRED)) { 53532567Sbostic error = ENOTCONN; 53632567Sbostic goto release; 5374890Swnj } 53843417Skarels if (uio->uio_resid == 0) 53925629Skarels goto release; 54032567Sbostic if (so->so_state & SS_NBIO) { 54132567Sbostic error = EWOULDBLOCK; 54232567Sbostic goto release; 54332567Sbostic } 5444890Swnj sbunlock(&so->so_rcv); 54543417Skarels error = sbwait(&so->so_rcv); 5465012Swnj splx(s); 54741908Skarels if (error) 54841908Skarels return (error); 5494890Swnj goto restart; 5504786Swnj } 551*46479Skarels dontblock: 5528041Sroot u.u_ru.ru_msgrcv++; 55335384Skarels nextrecord = m->m_nextpkt; 55412757Ssam if (pr->pr_flags & PR_ADDR) { 55541908Skarels #ifdef DIAGNOSTIC 55625629Skarels if (m->m_type != MT_SONAME) 55716993Skarels panic("receive 1a"); 55841908Skarels #endif 55916993Skarels if (flags & MSG_PEEK) { 56041908Skarels if (paddr) 56141908Skarels *paddr = m_copy(m, 0, m->m_len); 56225629Skarels m = m->m_next; 56316993Skarels } else { 56425629Skarels sbfree(&so->so_rcv, m); 56541908Skarels if (paddr) { 56641908Skarels *paddr = m; 56735384Skarels so->so_rcv.sb_mb = m->m_next; 56835384Skarels m->m_next = 0; 56935384Skarels m = so->so_rcv.sb_mb; 57025629Skarels } else { 57126958Skarels MFREE(m, so->so_rcv.sb_mb); 57226958Skarels m = so->so_rcv.sb_mb; 57325629Skarels } 57416993Skarels } 57516993Skarels } 57641908Skarels while (m && m->m_type == MT_CONTROL && error == 0) { 57716993Skarels if (flags & MSG_PEEK) { 57835384Skarels if (controlp) 57935384Skarels *controlp = m_copy(m, 0, m->m_len); 58035384Skarels m = m->m_next; 58135384Skarels } else { 58235384Skarels sbfree(&so->so_rcv, m); 58335384Skarels if (controlp) { 58443097Ssklower if (pr->pr_domain->dom_externalize && 58543097Ssklower mtod(m, struct cmsghdr *)->cmsg_type == 58643097Ssklower SCM_RIGHTS) 58741908Skarels error = (*pr->pr_domain->dom_externalize)(m); 58835384Skarels *controlp = m; 58935384Skarels so->so_rcv.sb_mb = m->m_next; 59035384Skarels m->m_next = 0; 59135384Skarels m = so->so_rcv.sb_mb; 59235384Skarels } else { 59335384Skarels MFREE(m, so->so_rcv.sb_mb); 59435384Skarels m = so->so_rcv.sb_mb; 59535384Skarels } 59635384Skarels } 59741908Skarels if (controlp) 59841908Skarels controlp = &(*controlp)->m_next; 59935384Skarels } 60041908Skarels if (m) { 60144383Skarels if ((flags & MSG_PEEK) == 0) 60244383Skarels m->m_nextpkt = nextrecord; 60341908Skarels type = m->m_type; 60446452Ssklower if (type == MT_OOBDATA) 60546452Ssklower flags |= MSG_OOB; 60641908Skarels } 6078319Sroot moff = 0; 60832092Skarels offset = 0; 60946452Ssklower while (m && uio->uio_resid > 0 && error == 0) { 61046452Ssklower if (m->m_type == MT_OOBDATA) { 61146452Ssklower if (type != MT_OOBDATA) 61246452Ssklower break; 61346452Ssklower } else if (type == MT_OOBDATA) 61446452Ssklower break; 61541908Skarels #ifdef DIAGNOSTIC 61635384Skarels else if (m->m_type != MT_DATA && m->m_type != MT_HEADER) 61725629Skarels panic("receive 3"); 61841908Skarels #endif 6197747Sroot so->so_state &= ~SS_RCVATMARK; 62043417Skarels len = uio->uio_resid; 62132092Skarels if (so->so_oobmark && len > so->so_oobmark - offset) 62232092Skarels len = so->so_oobmark - offset; 62321767Skarels if (len > m->m_len - moff) 6248319Sroot len = m->m_len - moff; 62541908Skarels /* 62641908Skarels * If mp is set, just pass back the mbufs. 62741908Skarels * Otherwise copy them out via the uio, then free. 62841908Skarels * Sockbuf must be consistent here (points to current mbuf, 62941908Skarels * it points to next record) when we drop priority; 63041908Skarels * we must note any additions to the sockbuf when we 63141908Skarels * block interrupts again. 63241908Skarels */ 63341908Skarels if (mp == 0) { 63441908Skarels splx(s); 63541908Skarels error = uiomove(mtod(m, caddr_t) + moff, (int)len, uio); 63641908Skarels s = splnet(); 63743417Skarels } else 63843417Skarels uio->uio_resid -= len; 63921767Skarels if (len == m->m_len - moff) { 64041908Skarels if (m->m_flags & M_EOR) 64141908Skarels flags |= MSG_EOR; 64225629Skarels if (flags & MSG_PEEK) { 64325629Skarels m = m->m_next; 64425629Skarels moff = 0; 64525629Skarels } else { 64635384Skarels nextrecord = m->m_nextpkt; 64725629Skarels sbfree(&so->so_rcv, m); 64841908Skarels if (mp) { 64941908Skarels *mp = m; 65041908Skarels mp = &m->m_next; 65143417Skarels so->so_rcv.sb_mb = m = m->m_next; 65243417Skarels *mp = (struct mbuf *)0; 65341908Skarels } else { 65441908Skarels MFREE(m, so->so_rcv.sb_mb); 65541908Skarels m = so->so_rcv.sb_mb; 65641908Skarels } 65726958Skarels if (m) 65835384Skarels m->m_nextpkt = nextrecord; 65925629Skarels } 6604786Swnj } else { 66112757Ssam if (flags & MSG_PEEK) 6628319Sroot moff += len; 6638319Sroot else { 66443417Skarels if (mp) 66543417Skarels *mp = m_copym(m, 0, len, M_WAIT); 66635384Skarels m->m_data += len; 6678319Sroot m->m_len -= len; 6688319Sroot so->so_rcv.sb_cc -= len; 6698319Sroot } 6704786Swnj } 67132092Skarels if (so->so_oobmark) { 67232092Skarels if ((flags & MSG_PEEK) == 0) { 67332092Skarels so->so_oobmark -= len; 67432092Skarels if (so->so_oobmark == 0) { 67532092Skarels so->so_state |= SS_RCVATMARK; 67632092Skarels break; 67732092Skarels } 67832092Skarels } else 67932092Skarels offset += len; 6807747Sroot } 68141908Skarels if (flags & MSG_EOR) 68240632Skarels break; 68341908Skarels /* 68441908Skarels * If the MSG_WAITALL flag is set (for non-atomic socket), 68543417Skarels * we must not quit until "uio->uio_resid == 0" or an error 68641908Skarels * termination. If a signal/timeout occurs, return 68743417Skarels * with a short count but without error. 68841908Skarels * Keep sockbuf locked against other readers. 68941908Skarels */ 69043417Skarels while (flags & MSG_WAITALL && m == 0 && uio->uio_resid > 0 && 69141908Skarels !sosendallatonce(so)) { 692*46479Skarels if (so->so_error || so->so_state & SS_CANTRCVMORE) 693*46479Skarels break; 69441908Skarels error = sbwait(&so->so_rcv); 69541908Skarels if (error) { 69641908Skarels sbunlock(&so->so_rcv); 69741908Skarels splx(s); 69841908Skarels return (0); 69941908Skarels } 70041908Skarels if (m = so->so_rcv.sb_mb) 70141908Skarels nextrecord = m->m_nextpkt; 70241908Skarels } 70316993Skarels } 70416993Skarels if ((flags & MSG_PEEK) == 0) { 70526500Skarels if (m == 0) 70616993Skarels so->so_rcv.sb_mb = nextrecord; 70735384Skarels else if (pr->pr_flags & PR_ATOMIC) { 70835384Skarels flags |= MSG_TRUNC; 70926958Skarels (void) sbdroprecord(&so->so_rcv); 71035384Skarels } 71116993Skarels if (pr->pr_flags & PR_WANTRCVD && so->so_pcb) 71216993Skarels (*pr->pr_usrreq)(so, PRU_RCVD, (struct mbuf *)0, 71337478Ssklower (struct mbuf *)flags, (struct mbuf *)0, 71437478Ssklower (struct mbuf *)0); 71516993Skarels } 71635384Skarels if (flagsp) 71735384Skarels *flagsp |= flags; 7184890Swnj release: 7194916Swnj sbunlock(&so->so_rcv); 7204890Swnj splx(s); 7214916Swnj return (error); 7224786Swnj } 7234786Swnj 72410267Ssam soshutdown(so, how) 72512757Ssam register struct socket *so; 72612757Ssam register int how; 72710267Ssam { 72812757Ssam register struct protosw *pr = so->so_proto; 72910267Ssam 73010267Ssam how++; 73112757Ssam if (how & FREAD) 73212757Ssam sorflush(so); 73310267Ssam if (how & FWRITE) 73412757Ssam return ((*pr->pr_usrreq)(so, PRU_SHUTDOWN, 73512757Ssam (struct mbuf *)0, (struct mbuf *)0, (struct mbuf *)0)); 73610267Ssam return (0); 73710267Ssam } 73810267Ssam 73912757Ssam sorflush(so) 74012757Ssam register struct socket *so; 74112757Ssam { 74212757Ssam register struct sockbuf *sb = &so->so_rcv; 74312757Ssam register struct protosw *pr = so->so_proto; 74412757Ssam register int s; 74512757Ssam struct sockbuf asb; 74612757Ssam 74740706Skarels sb->sb_flags |= SB_NOINTR; 74840706Skarels (void) sblock(sb); 74912757Ssam s = splimp(); 75012757Ssam socantrcvmore(so); 75112757Ssam sbunlock(sb); 75212757Ssam asb = *sb; 75312757Ssam bzero((caddr_t)sb, sizeof (*sb)); 75412757Ssam splx(s); 75516993Skarels if (pr->pr_flags & PR_RIGHTS && pr->pr_domain->dom_dispose) 75616993Skarels (*pr->pr_domain->dom_dispose)(asb.sb_mb); 75712757Ssam sbrelease(&asb); 75812757Ssam } 75912757Ssam 76018553Skarels sosetopt(so, level, optname, m0) 76112757Ssam register struct socket *so; 76210267Ssam int level, optname; 76318553Skarels struct mbuf *m0; 76410267Ssam { 76517158Ssam int error = 0; 76618553Skarels register struct mbuf *m = m0; 76710267Ssam 76817158Ssam if (level != SOL_SOCKET) { 76918369Skarels if (so->so_proto && so->so_proto->pr_ctloutput) 77018369Skarels return ((*so->so_proto->pr_ctloutput) 77118553Skarels (PRCO_SETOPT, so, level, optname, &m0)); 77218369Skarels error = ENOPROTOOPT; 77318369Skarels } else { 77418369Skarels switch (optname) { 77510267Ssam 77618369Skarels case SO_LINGER: 77718369Skarels if (m == NULL || m->m_len != sizeof (struct linger)) { 77818369Skarels error = EINVAL; 77918369Skarels goto bad; 78018369Skarels } 78118369Skarels so->so_linger = mtod(m, struct linger *)->l_linger; 78218369Skarels /* fall thru... */ 78317158Ssam 78418369Skarels case SO_DEBUG: 78518369Skarels case SO_KEEPALIVE: 78618369Skarels case SO_DONTROUTE: 78718369Skarels case SO_USELOOPBACK: 78818369Skarels case SO_BROADCAST: 78918369Skarels case SO_REUSEADDR: 79027191Skarels case SO_OOBINLINE: 79118369Skarels if (m == NULL || m->m_len < sizeof (int)) { 79218369Skarels error = EINVAL; 79318369Skarels goto bad; 79418369Skarels } 79518369Skarels if (*mtod(m, int *)) 79618369Skarels so->so_options |= optname; 79718369Skarels else 79818369Skarels so->so_options &= ~optname; 79918369Skarels break; 80018369Skarels 80118369Skarels case SO_SNDBUF: 80241908Skarels case SO_RCVBUF: 80340706Skarels case SO_SNDLOWAT: 80441908Skarels case SO_RCVLOWAT: 80518369Skarels if (m == NULL || m->m_len < sizeof (int)) { 80618369Skarels error = EINVAL; 80718369Skarels goto bad; 80818369Skarels } 80918369Skarels switch (optname) { 81018369Skarels 81118369Skarels case SO_SNDBUF: 81218369Skarels case SO_RCVBUF: 81341908Skarels if (sbreserve(optname == SO_SNDBUF ? 81441908Skarels &so->so_snd : &so->so_rcv, 81541908Skarels (u_long) *mtod(m, int *)) == 0) { 81618369Skarels error = ENOBUFS; 81718369Skarels goto bad; 81818369Skarels } 81918369Skarels break; 82018369Skarels 82118369Skarels case SO_SNDLOWAT: 82241908Skarels so->so_snd.sb_lowat = *mtod(m, int *); 82341908Skarels break; 82418369Skarels case SO_RCVLOWAT: 82541908Skarels so->so_rcv.sb_lowat = *mtod(m, int *); 82618369Skarels break; 82744721Skarels } 82844721Skarels break; 82944721Skarels 83044721Skarels case SO_SNDTIMEO: 83144721Skarels case SO_RCVTIMEO: 83244721Skarels { 83344721Skarels struct timeval *tv; 83444721Skarels short val; 83544721Skarels 83644721Skarels if (m == NULL || m->m_len < sizeof (*tv)) { 83744721Skarels error = EINVAL; 83844721Skarels goto bad; 83944721Skarels } 84044721Skarels tv = mtod(m, struct timeval *); 84144721Skarels if (tv->tv_sec > SHRT_MAX / hz - hz) { 84244721Skarels error = EDOM; 84344721Skarels goto bad; 84444721Skarels } 84544721Skarels val = tv->tv_sec * hz + tv->tv_usec / tick; 84644721Skarels 84744721Skarels switch (optname) { 84844721Skarels 84918369Skarels case SO_SNDTIMEO: 85044721Skarels so->so_snd.sb_timeo = val; 85141908Skarels break; 85218369Skarels case SO_RCVTIMEO: 85344721Skarels so->so_rcv.sb_timeo = val; 85418369Skarels break; 85518369Skarels } 85618369Skarels break; 85744721Skarels } 85818369Skarels 85918369Skarels default: 86018369Skarels error = ENOPROTOOPT; 86118369Skarels break; 86217158Ssam } 86310267Ssam } 86417158Ssam bad: 86517158Ssam if (m) 86617158Ssam (void) m_free(m); 86717158Ssam return (error); 86810267Ssam } 86910267Ssam 87017158Ssam sogetopt(so, level, optname, mp) 87112757Ssam register struct socket *so; 87210267Ssam int level, optname; 87317158Ssam struct mbuf **mp; 87417158Ssam { 87512757Ssam register struct mbuf *m; 87610267Ssam 87718369Skarels if (level != SOL_SOCKET) { 87818369Skarels if (so->so_proto && so->so_proto->pr_ctloutput) { 87918369Skarels return ((*so->so_proto->pr_ctloutput) 88018369Skarels (PRCO_GETOPT, so, level, optname, mp)); 88145515Skarels } else 88218369Skarels return (ENOPROTOOPT); 88318369Skarels } else { 88417158Ssam m = m_get(M_WAIT, MT_SOOPTS); 88525502Skarels m->m_len = sizeof (int); 88625502Skarels 88718369Skarels switch (optname) { 88817158Ssam 88918369Skarels case SO_LINGER: 89018369Skarels m->m_len = sizeof (struct linger); 89118369Skarels mtod(m, struct linger *)->l_onoff = 89218369Skarels so->so_options & SO_LINGER; 89318369Skarels mtod(m, struct linger *)->l_linger = so->so_linger; 89418369Skarels break; 89510267Ssam 89618369Skarels case SO_USELOOPBACK: 89718369Skarels case SO_DONTROUTE: 89818369Skarels case SO_DEBUG: 89918369Skarels case SO_KEEPALIVE: 90018369Skarels case SO_REUSEADDR: 90118369Skarels case SO_BROADCAST: 90227191Skarels case SO_OOBINLINE: 90318369Skarels *mtod(m, int *) = so->so_options & optname; 90418369Skarels break; 90518369Skarels 90625502Skarels case SO_TYPE: 90725502Skarels *mtod(m, int *) = so->so_type; 90825502Skarels break; 90925502Skarels 91024768Skarels case SO_ERROR: 91124768Skarels *mtod(m, int *) = so->so_error; 91224768Skarels so->so_error = 0; 91324768Skarels break; 91424768Skarels 91518369Skarels case SO_SNDBUF: 91618369Skarels *mtod(m, int *) = so->so_snd.sb_hiwat; 91718369Skarels break; 91818369Skarels 91918369Skarels case SO_RCVBUF: 92018369Skarels *mtod(m, int *) = so->so_rcv.sb_hiwat; 92118369Skarels break; 92218369Skarels 92318369Skarels case SO_SNDLOWAT: 92418369Skarels *mtod(m, int *) = so->so_snd.sb_lowat; 92518369Skarels break; 92618369Skarels 92718369Skarels case SO_RCVLOWAT: 92818369Skarels *mtod(m, int *) = so->so_rcv.sb_lowat; 92918369Skarels break; 93018369Skarels 93118369Skarels case SO_SNDTIMEO: 93244721Skarels case SO_RCVTIMEO: 93344721Skarels { 93444721Skarels int val = (optname == SO_SNDTIMEO ? 93544721Skarels so->so_snd.sb_timeo : so->so_rcv.sb_timeo); 93618369Skarels 93744721Skarels m->m_len = sizeof(struct timeval); 93844721Skarels mtod(m, struct timeval *)->tv_sec = val / hz; 93944721Skarels mtod(m, struct timeval *)->tv_usec = 94044721Skarels (val % hz) / tick; 94118369Skarels break; 94244721Skarels } 94318369Skarels 94418369Skarels default: 94526362Skarels (void)m_free(m); 94618369Skarels return (ENOPROTOOPT); 94718369Skarels } 94818369Skarels *mp = m; 94918369Skarels return (0); 95010267Ssam } 95110267Ssam } 95210267Ssam 9535423Swnj sohasoutofband(so) 95412757Ssam register struct socket *so; 9555423Swnj { 95623233Skarels struct proc *p; 9575423Swnj 95837478Ssklower if (so->so_pgid < 0) 95937478Ssklower gsignal(-so->so_pgid, SIGURG); 96037478Ssklower else if (so->so_pgid > 0 && (p = pfind(so->so_pgid)) != 0) 96123233Skarels psignal(p, SIGURG); 96224768Skarels if (so->so_rcv.sb_sel) { 96324768Skarels selwakeup(so->so_rcv.sb_sel, so->so_rcv.sb_flags & SB_COLL); 96424768Skarels so->so_rcv.sb_sel = 0; 96524768Skarels so->so_rcv.sb_flags &= ~SB_COLL; 96624768Skarels } 9675423Swnj } 968