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*45515Skarels * @(#)uipc_socket.c 7.24 (Berkeley) 11/05/90 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; 37241908Skarels } 37335384Skarels #endif 37435384Skarels space -= MCLBYTES; 37516412Skarels } else { 37616412Skarels nopages: 37741908Skarels len = min(min(mlen, resid), space); 37821767Skarels space -= len; 37935384Skarels /* 38035384Skarels * For datagram protocols, leave room 38135384Skarels * for protocol headers in first mbuf. 38235384Skarels */ 38335391Skarels if (atomic && top == 0 && len < mlen) 38435384Skarels MH_ALIGN(m, len); 38516412Skarels } 386*45515Skarels error = uiomove(mtod(m, caddr_t), (int)len, uio); 38741908Skarels resid = uio->uio_resid; 38816412Skarels m->m_len = len; 38916412Skarels *mp = m; 39035384Skarels top->m_pkthdr.len += len; 39116412Skarels if (error) 39216412Skarels goto release; 39316412Skarels mp = &m->m_next; 39441908Skarels if (resid <= 0) { 39541908Skarels if (flags & MSG_EOR) 39635384Skarels top->m_flags |= M_EOR; 39721108Skarels break; 39835384Skarels } 39935384Skarels } while (space > 0 && atomic); 40035384Skarels if (dontroute) 40135384Skarels so->so_options |= SO_DONTROUTE; 40235384Skarels s = splnet(); /* XXX */ 40335384Skarels error = (*so->so_proto->pr_usrreq)(so, 40435384Skarels (flags & MSG_OOB) ? PRU_SENDOOB : PRU_SEND, 40541908Skarels top, addr, control); 40635384Skarels splx(s); 40735384Skarels if (dontroute) 40835384Skarels so->so_options &= ~SO_DONTROUTE; 40941908Skarels clen = 0; 41041908Skarels control = 0; 41135384Skarels top = 0; 41235384Skarels mp = ⊤ 41335384Skarels if (error) 41435384Skarels goto release; 41541908Skarels } while (resid && space > 0); 41641908Skarels } while (resid); 4174890Swnj 4184786Swnj release: 4194890Swnj sbunlock(&so->so_snd); 42041908Skarels out: 4216419Sroot if (top) 4226419Sroot m_freem(top); 42341908Skarels if (control) 42441908Skarels m_freem(control); 4254786Swnj return (error); 4264786Swnj } 4274786Swnj 42825629Skarels /* 42925629Skarels * Implement receive operations on a socket. 43025629Skarels * We depend on the way that records are added to the sockbuf 43125629Skarels * by sbappend*. In particular, each record (mbufs linked through m_next) 43225629Skarels * must begin with an address if the protocol so specifies, 43341908Skarels * followed by an optional mbuf or mbufs containing ancillary data, 43441908Skarels * and then zero or more mbufs of data. 43525629Skarels * In order to avoid blocking network interrupts for the entire time here, 43625629Skarels * we splx() while doing the actual copy to user space. 43725629Skarels * Although the sockbuf is locked, new data may still be appended, 43825629Skarels * and thus we must maintain consistency of the sockbuf during that time. 439*45515Skarels * 44041908Skarels * The caller may receive the data as a single mbuf chain by supplying 44143417Skarels * an mbuf **mp0 for use in returning the chain. The uio is then used 44241908Skarels * only for the count in uio_resid. 44325629Skarels */ 44443417Skarels soreceive(so, paddr, uio, mp0, controlp, flagsp) 4454786Swnj register struct socket *so; 44641908Skarels struct mbuf **paddr; 44741908Skarels struct uio *uio; 44843417Skarels struct mbuf **mp0; 44941908Skarels struct mbuf **controlp; 45035384Skarels int *flagsp; 4514786Swnj { 45243417Skarels register struct mbuf *m, **mp; 45343417Skarels register int flags, len, error, s, offset; 45412757Ssam struct protosw *pr = so->so_proto; 45541908Skarels struct mbuf *nextrecord; 45641908Skarels int moff, type; 4574786Swnj 45843417Skarels mp = mp0; 45941908Skarels if (paddr) 46041908Skarels *paddr = 0; 46135384Skarels if (controlp) 46235384Skarels *controlp = 0; 46335384Skarels if (flagsp) 46435384Skarels flags = *flagsp &~ MSG_EOR; 465*45515Skarels else 46635384Skarels flags = 0; 46712757Ssam if (flags & MSG_OOB) { 4689635Ssam m = m_get(M_WAIT, MT_DATA); 46912757Ssam error = (*pr->pr_usrreq)(so, PRU_RCVOOB, 47024768Skarels m, (struct mbuf *)(flags & MSG_PEEK), (struct mbuf *)0); 4718594Sroot if (error) 47210137Ssam goto bad; 4738319Sroot do { 47441908Skarels error = uiomove(mtod(m, caddr_t), 47541908Skarels (int) min(uio->uio_resid, m->m_len), uio); 4768319Sroot m = m_free(m); 4778594Sroot } while (uio->uio_resid && error == 0 && m); 47810137Ssam bad: 4798319Sroot if (m) 4808771Sroot m_freem(m); 4818594Sroot return (error); 4828319Sroot } 48341908Skarels if (mp) 48441908Skarels *mp = (struct mbuf *)0; 48543417Skarels if (so->so_state & SS_ISCONFIRMING && uio->uio_resid) 48635384Skarels (*pr->pr_usrreq)(so, PRU_RCVD, (struct mbuf *)0, 48735384Skarels (struct mbuf *)0, (struct mbuf *)0); 4888319Sroot 4894890Swnj restart: 49040706Skarels if (error = sblock(&so->so_rcv)) 49140706Skarels return (error); 4928835Sroot s = splnet(); 4934890Swnj 49437478Ssklower m = so->so_rcv.sb_mb; 49544383Skarels /* 49644383Skarels * If we have less data than requested, block awaiting more 49744383Skarels * (subject to any timeout) if: 49844383Skarels * 1. the current count is less than the low water mark, or 49944383Skarels * 2. MSG_WAITALL is set, and it is possible to do the entire 50044383Skarels * receive operation at once if we block (resid <= hiwat). 50144383Skarels * If MSG_WAITALL is set but resid is larger than the receive buffer, 50244383Skarels * we have to do the receive in sections, and thus risk returning 50344383Skarels * a short count if a timeout or signal occurs after we start. 50444383Skarels */ 505*45515Skarels if (m == 0 || so->so_rcv.sb_cc < uio->uio_resid && 50644383Skarels (so->so_rcv.sb_cc < so->so_rcv.sb_lowat || 50744383Skarels ((flags & MSG_WAITALL) && uio->uio_resid <= so->so_rcv.sb_hiwat))) { 50841908Skarels #ifdef DIAGNOSTIC 50941908Skarels if (m == 0 && so->so_rcv.sb_cc) 51037478Ssklower panic("receive 1"); 51141908Skarels #endif 5125168Swnj if (so->so_error) { 5135168Swnj error = so->so_error; 5145168Swnj so->so_error = 0; 5155168Swnj goto release; 5165168Swnj } 51732567Sbostic if (so->so_state & SS_CANTRCVMORE) 5184890Swnj goto release; 51938584Skarels if ((so->so_state & (SS_ISCONNECTED|SS_ISCONNECTING)) == 0 && 52032567Sbostic (so->so_proto->pr_flags & PR_CONNREQUIRED)) { 52132567Sbostic error = ENOTCONN; 52232567Sbostic goto release; 5234890Swnj } 52443417Skarels if (uio->uio_resid == 0) 52525629Skarels goto release; 52632567Sbostic if (so->so_state & SS_NBIO) { 52732567Sbostic error = EWOULDBLOCK; 52832567Sbostic goto release; 52932567Sbostic } 5304890Swnj sbunlock(&so->so_rcv); 53143417Skarels error = sbwait(&so->so_rcv); 5325012Swnj splx(s); 53341908Skarels if (error) 53441908Skarels return (error); 5354890Swnj goto restart; 5364786Swnj } 5378041Sroot u.u_ru.ru_msgrcv++; 53835384Skarels nextrecord = m->m_nextpkt; 53912757Ssam if (pr->pr_flags & PR_ADDR) { 54041908Skarels #ifdef DIAGNOSTIC 54125629Skarels if (m->m_type != MT_SONAME) 54216993Skarels panic("receive 1a"); 54341908Skarels #endif 54416993Skarels if (flags & MSG_PEEK) { 54541908Skarels if (paddr) 54641908Skarels *paddr = m_copy(m, 0, m->m_len); 54725629Skarels m = m->m_next; 54816993Skarels } else { 54925629Skarels sbfree(&so->so_rcv, m); 55041908Skarels if (paddr) { 55141908Skarels *paddr = m; 55235384Skarels so->so_rcv.sb_mb = m->m_next; 55335384Skarels m->m_next = 0; 55435384Skarels m = so->so_rcv.sb_mb; 55525629Skarels } else { 55626958Skarels MFREE(m, so->so_rcv.sb_mb); 55726958Skarels m = so->so_rcv.sb_mb; 55825629Skarels } 55916993Skarels } 56016993Skarels } 56141908Skarels while (m && m->m_type == MT_CONTROL && error == 0) { 56216993Skarels if (flags & MSG_PEEK) { 56335384Skarels if (controlp) 56435384Skarels *controlp = m_copy(m, 0, m->m_len); 56535384Skarels m = m->m_next; 56635384Skarels } else { 56735384Skarels sbfree(&so->so_rcv, m); 56835384Skarels if (controlp) { 56943097Ssklower if (pr->pr_domain->dom_externalize && 57043097Ssklower mtod(m, struct cmsghdr *)->cmsg_type == 57143097Ssklower SCM_RIGHTS) 57241908Skarels error = (*pr->pr_domain->dom_externalize)(m); 57335384Skarels *controlp = m; 57435384Skarels so->so_rcv.sb_mb = m->m_next; 57535384Skarels m->m_next = 0; 57635384Skarels m = so->so_rcv.sb_mb; 57735384Skarels } else { 57835384Skarels MFREE(m, so->so_rcv.sb_mb); 57935384Skarels m = so->so_rcv.sb_mb; 58035384Skarels } 58135384Skarels } 58241908Skarels if (controlp) 58341908Skarels controlp = &(*controlp)->m_next; 58435384Skarels } 58541908Skarels if (m) { 58644383Skarels if ((flags & MSG_PEEK) == 0) 58744383Skarels m->m_nextpkt = nextrecord; 58841908Skarels type = m->m_type; 58941908Skarels } 5908319Sroot moff = 0; 59132092Skarels offset = 0; 59243417Skarels while (m && m->m_type == type && uio->uio_resid > 0 && error == 0) { 59335384Skarels if (m->m_type == MT_OOBDATA) 59435384Skarels flags |= MSG_OOB; 59541908Skarels #ifdef DIAGNOSTIC 59635384Skarels else if (m->m_type != MT_DATA && m->m_type != MT_HEADER) 59725629Skarels panic("receive 3"); 59841908Skarels #endif 5997747Sroot so->so_state &= ~SS_RCVATMARK; 60043417Skarels len = uio->uio_resid; 60132092Skarels if (so->so_oobmark && len > so->so_oobmark - offset) 60232092Skarels len = so->so_oobmark - offset; 60321767Skarels if (len > m->m_len - moff) 6048319Sroot len = m->m_len - moff; 60541908Skarels /* 60641908Skarels * If mp is set, just pass back the mbufs. 60741908Skarels * Otherwise copy them out via the uio, then free. 60841908Skarels * Sockbuf must be consistent here (points to current mbuf, 60941908Skarels * it points to next record) when we drop priority; 61041908Skarels * we must note any additions to the sockbuf when we 61141908Skarels * block interrupts again. 61241908Skarels */ 61341908Skarels if (mp == 0) { 61441908Skarels splx(s); 61541908Skarels error = uiomove(mtod(m, caddr_t) + moff, (int)len, uio); 61641908Skarels s = splnet(); 61743417Skarels } else 61843417Skarels uio->uio_resid -= len; 61921767Skarels if (len == m->m_len - moff) { 62041908Skarels if (m->m_flags & M_EOR) 62141908Skarels flags |= MSG_EOR; 62225629Skarels if (flags & MSG_PEEK) { 62325629Skarels m = m->m_next; 62425629Skarels moff = 0; 62525629Skarels } else { 62635384Skarels nextrecord = m->m_nextpkt; 62725629Skarels sbfree(&so->so_rcv, m); 62841908Skarels if (mp) { 62941908Skarels *mp = m; 63041908Skarels mp = &m->m_next; 63143417Skarels so->so_rcv.sb_mb = m = m->m_next; 63243417Skarels *mp = (struct mbuf *)0; 63341908Skarels } else { 63441908Skarels MFREE(m, so->so_rcv.sb_mb); 63541908Skarels m = so->so_rcv.sb_mb; 63641908Skarels } 63726958Skarels if (m) 63835384Skarels m->m_nextpkt = nextrecord; 63925629Skarels } 6404786Swnj } else { 64112757Ssam if (flags & MSG_PEEK) 6428319Sroot moff += len; 6438319Sroot else { 64443417Skarels if (mp) 64543417Skarels *mp = m_copym(m, 0, len, M_WAIT); 64635384Skarels m->m_data += len; 6478319Sroot m->m_len -= len; 6488319Sroot so->so_rcv.sb_cc -= len; 6498319Sroot } 6504786Swnj } 65132092Skarels if (so->so_oobmark) { 65232092Skarels if ((flags & MSG_PEEK) == 0) { 65332092Skarels so->so_oobmark -= len; 65432092Skarels if (so->so_oobmark == 0) { 65532092Skarels so->so_state |= SS_RCVATMARK; 65632092Skarels break; 65732092Skarels } 65832092Skarels } else 65932092Skarels offset += len; 6607747Sroot } 66141908Skarels if (flags & MSG_EOR) 66240632Skarels break; 66341908Skarels /* 66441908Skarels * If the MSG_WAITALL flag is set (for non-atomic socket), 66543417Skarels * we must not quit until "uio->uio_resid == 0" or an error 66641908Skarels * termination. If a signal/timeout occurs, return 66743417Skarels * with a short count but without error. 66841908Skarels * Keep sockbuf locked against other readers. 66941908Skarels */ 67043417Skarels while (flags & MSG_WAITALL && m == 0 && uio->uio_resid > 0 && 67141908Skarels !sosendallatonce(so)) { 67241908Skarels error = sbwait(&so->so_rcv); 67341908Skarels if (error) { 67441908Skarels sbunlock(&so->so_rcv); 67541908Skarels splx(s); 67641908Skarels return (0); 67741908Skarels } 67841908Skarels if (m = so->so_rcv.sb_mb) 67941908Skarels nextrecord = m->m_nextpkt; 68041908Skarels if (so->so_error || so->so_state & SS_CANTRCVMORE) 68141908Skarels break; 68241908Skarels continue; 68341908Skarels } 68416993Skarels } 68516993Skarels if ((flags & MSG_PEEK) == 0) { 68626500Skarels if (m == 0) 68716993Skarels so->so_rcv.sb_mb = nextrecord; 68835384Skarels else if (pr->pr_flags & PR_ATOMIC) { 68935384Skarels flags |= MSG_TRUNC; 69026958Skarels (void) sbdroprecord(&so->so_rcv); 69135384Skarels } 69216993Skarels if (pr->pr_flags & PR_WANTRCVD && so->so_pcb) 69316993Skarels (*pr->pr_usrreq)(so, PRU_RCVD, (struct mbuf *)0, 69437478Ssklower (struct mbuf *)flags, (struct mbuf *)0, 69537478Ssklower (struct mbuf *)0); 69616993Skarels } 69735384Skarels if (flagsp) 69835384Skarels *flagsp |= flags; 6994890Swnj release: 7004916Swnj sbunlock(&so->so_rcv); 7014890Swnj splx(s); 7024916Swnj return (error); 7034786Swnj } 7044786Swnj 70510267Ssam soshutdown(so, how) 70612757Ssam register struct socket *so; 70712757Ssam register int how; 70810267Ssam { 70912757Ssam register struct protosw *pr = so->so_proto; 71010267Ssam 71110267Ssam how++; 71212757Ssam if (how & FREAD) 71312757Ssam sorflush(so); 71410267Ssam if (how & FWRITE) 71512757Ssam return ((*pr->pr_usrreq)(so, PRU_SHUTDOWN, 71612757Ssam (struct mbuf *)0, (struct mbuf *)0, (struct mbuf *)0)); 71710267Ssam return (0); 71810267Ssam } 71910267Ssam 72012757Ssam sorflush(so) 72112757Ssam register struct socket *so; 72212757Ssam { 72312757Ssam register struct sockbuf *sb = &so->so_rcv; 72412757Ssam register struct protosw *pr = so->so_proto; 72512757Ssam register int s; 72612757Ssam struct sockbuf asb; 72712757Ssam 72840706Skarels sb->sb_flags |= SB_NOINTR; 72940706Skarels (void) sblock(sb); 73012757Ssam s = splimp(); 73112757Ssam socantrcvmore(so); 73212757Ssam sbunlock(sb); 73312757Ssam asb = *sb; 73412757Ssam bzero((caddr_t)sb, sizeof (*sb)); 73512757Ssam splx(s); 73616993Skarels if (pr->pr_flags & PR_RIGHTS && pr->pr_domain->dom_dispose) 73716993Skarels (*pr->pr_domain->dom_dispose)(asb.sb_mb); 73812757Ssam sbrelease(&asb); 73912757Ssam } 74012757Ssam 74118553Skarels sosetopt(so, level, optname, m0) 74212757Ssam register struct socket *so; 74310267Ssam int level, optname; 74418553Skarels struct mbuf *m0; 74510267Ssam { 74617158Ssam int error = 0; 74718553Skarels register struct mbuf *m = m0; 74810267Ssam 74917158Ssam if (level != SOL_SOCKET) { 75018369Skarels if (so->so_proto && so->so_proto->pr_ctloutput) 75118369Skarels return ((*so->so_proto->pr_ctloutput) 75218553Skarels (PRCO_SETOPT, so, level, optname, &m0)); 75318369Skarels error = ENOPROTOOPT; 75418369Skarels } else { 75518369Skarels switch (optname) { 75610267Ssam 75718369Skarels case SO_LINGER: 75818369Skarels if (m == NULL || m->m_len != sizeof (struct linger)) { 75918369Skarels error = EINVAL; 76018369Skarels goto bad; 76118369Skarels } 76218369Skarels so->so_linger = mtod(m, struct linger *)->l_linger; 76318369Skarels /* fall thru... */ 76417158Ssam 76518369Skarels case SO_DEBUG: 76618369Skarels case SO_KEEPALIVE: 76718369Skarels case SO_DONTROUTE: 76818369Skarels case SO_USELOOPBACK: 76918369Skarels case SO_BROADCAST: 77018369Skarels case SO_REUSEADDR: 77127191Skarels case SO_OOBINLINE: 77218369Skarels if (m == NULL || m->m_len < sizeof (int)) { 77318369Skarels error = EINVAL; 77418369Skarels goto bad; 77518369Skarels } 77618369Skarels if (*mtod(m, int *)) 77718369Skarels so->so_options |= optname; 77818369Skarels else 77918369Skarels so->so_options &= ~optname; 78018369Skarels break; 78118369Skarels 78218369Skarels case SO_SNDBUF: 78341908Skarels case SO_RCVBUF: 78440706Skarels case SO_SNDLOWAT: 78541908Skarels case SO_RCVLOWAT: 78618369Skarels if (m == NULL || m->m_len < sizeof (int)) { 78718369Skarels error = EINVAL; 78818369Skarels goto bad; 78918369Skarels } 79018369Skarels switch (optname) { 79118369Skarels 79218369Skarels case SO_SNDBUF: 79318369Skarels case SO_RCVBUF: 79441908Skarels if (sbreserve(optname == SO_SNDBUF ? 79541908Skarels &so->so_snd : &so->so_rcv, 79641908Skarels (u_long) *mtod(m, int *)) == 0) { 79718369Skarels error = ENOBUFS; 79818369Skarels goto bad; 79918369Skarels } 80018369Skarels break; 80118369Skarels 80218369Skarels case SO_SNDLOWAT: 80341908Skarels so->so_snd.sb_lowat = *mtod(m, int *); 80441908Skarels break; 80518369Skarels case SO_RCVLOWAT: 80641908Skarels so->so_rcv.sb_lowat = *mtod(m, int *); 80718369Skarels break; 80844721Skarels } 80944721Skarels break; 81044721Skarels 81144721Skarels case SO_SNDTIMEO: 81244721Skarels case SO_RCVTIMEO: 81344721Skarels { 81444721Skarels struct timeval *tv; 81544721Skarels short val; 81644721Skarels 81744721Skarels if (m == NULL || m->m_len < sizeof (*tv)) { 81844721Skarels error = EINVAL; 81944721Skarels goto bad; 82044721Skarels } 82144721Skarels tv = mtod(m, struct timeval *); 82244721Skarels if (tv->tv_sec > SHRT_MAX / hz - hz) { 82344721Skarels error = EDOM; 82444721Skarels goto bad; 82544721Skarels } 82644721Skarels val = tv->tv_sec * hz + tv->tv_usec / tick; 82744721Skarels 82844721Skarels switch (optname) { 82944721Skarels 83018369Skarels case SO_SNDTIMEO: 83144721Skarels so->so_snd.sb_timeo = val; 83241908Skarels break; 83318369Skarels case SO_RCVTIMEO: 83444721Skarels so->so_rcv.sb_timeo = val; 83518369Skarels break; 83618369Skarels } 83718369Skarels break; 83844721Skarels } 83918369Skarels 84018369Skarels default: 84118369Skarels error = ENOPROTOOPT; 84218369Skarels break; 84317158Ssam } 84410267Ssam } 84517158Ssam bad: 84617158Ssam if (m) 84717158Ssam (void) m_free(m); 84817158Ssam return (error); 84910267Ssam } 85010267Ssam 85117158Ssam sogetopt(so, level, optname, mp) 85212757Ssam register struct socket *so; 85310267Ssam int level, optname; 85417158Ssam struct mbuf **mp; 85517158Ssam { 85612757Ssam register struct mbuf *m; 85710267Ssam 85818369Skarels if (level != SOL_SOCKET) { 85918369Skarels if (so->so_proto && so->so_proto->pr_ctloutput) { 86018369Skarels return ((*so->so_proto->pr_ctloutput) 86118369Skarels (PRCO_GETOPT, so, level, optname, mp)); 862*45515Skarels } else 86318369Skarels return (ENOPROTOOPT); 86418369Skarels } else { 86517158Ssam m = m_get(M_WAIT, MT_SOOPTS); 86625502Skarels m->m_len = sizeof (int); 86725502Skarels 86818369Skarels switch (optname) { 86917158Ssam 87018369Skarels case SO_LINGER: 87118369Skarels m->m_len = sizeof (struct linger); 87218369Skarels mtod(m, struct linger *)->l_onoff = 87318369Skarels so->so_options & SO_LINGER; 87418369Skarels mtod(m, struct linger *)->l_linger = so->so_linger; 87518369Skarels break; 87610267Ssam 87718369Skarels case SO_USELOOPBACK: 87818369Skarels case SO_DONTROUTE: 87918369Skarels case SO_DEBUG: 88018369Skarels case SO_KEEPALIVE: 88118369Skarels case SO_REUSEADDR: 88218369Skarels case SO_BROADCAST: 88327191Skarels case SO_OOBINLINE: 88418369Skarels *mtod(m, int *) = so->so_options & optname; 88518369Skarels break; 88618369Skarels 88725502Skarels case SO_TYPE: 88825502Skarels *mtod(m, int *) = so->so_type; 88925502Skarels break; 89025502Skarels 89124768Skarels case SO_ERROR: 89224768Skarels *mtod(m, int *) = so->so_error; 89324768Skarels so->so_error = 0; 89424768Skarels break; 89524768Skarels 89618369Skarels case SO_SNDBUF: 89718369Skarels *mtod(m, int *) = so->so_snd.sb_hiwat; 89818369Skarels break; 89918369Skarels 90018369Skarels case SO_RCVBUF: 90118369Skarels *mtod(m, int *) = so->so_rcv.sb_hiwat; 90218369Skarels break; 90318369Skarels 90418369Skarels case SO_SNDLOWAT: 90518369Skarels *mtod(m, int *) = so->so_snd.sb_lowat; 90618369Skarels break; 90718369Skarels 90818369Skarels case SO_RCVLOWAT: 90918369Skarels *mtod(m, int *) = so->so_rcv.sb_lowat; 91018369Skarels break; 91118369Skarels 91218369Skarels case SO_SNDTIMEO: 91344721Skarels case SO_RCVTIMEO: 91444721Skarels { 91544721Skarels int val = (optname == SO_SNDTIMEO ? 91644721Skarels so->so_snd.sb_timeo : so->so_rcv.sb_timeo); 91718369Skarels 91844721Skarels m->m_len = sizeof(struct timeval); 91944721Skarels mtod(m, struct timeval *)->tv_sec = val / hz; 92044721Skarels mtod(m, struct timeval *)->tv_usec = 92144721Skarels (val % hz) / tick; 92218369Skarels break; 92344721Skarels } 92418369Skarels 92518369Skarels default: 92626362Skarels (void)m_free(m); 92718369Skarels return (ENOPROTOOPT); 92818369Skarels } 92918369Skarels *mp = m; 93018369Skarels return (0); 93110267Ssam } 93210267Ssam } 93310267Ssam 9345423Swnj sohasoutofband(so) 93512757Ssam register struct socket *so; 9365423Swnj { 93723233Skarels struct proc *p; 9385423Swnj 93937478Ssklower if (so->so_pgid < 0) 94037478Ssklower gsignal(-so->so_pgid, SIGURG); 94137478Ssklower else if (so->so_pgid > 0 && (p = pfind(so->so_pgid)) != 0) 94223233Skarels psignal(p, SIGURG); 94324768Skarels if (so->so_rcv.sb_sel) { 94424768Skarels selwakeup(so->so_rcv.sb_sel, so->so_rcv.sb_flags & SB_COLL); 94524768Skarels so->so_rcv.sb_sel = 0; 94624768Skarels so->so_rcv.sb_flags &= ~SB_COLL; 94724768Skarels } 9485423Swnj } 949