version 1.140.6.6, 2007/12/09 19:38:27 |
version 1.141, 2007/08/06 11:41:52 |
Line 79 __KERNEL_RCSID(0, "$NetBSD$"); |
|
Line 79 __KERNEL_RCSID(0, "$NetBSD$"); |
|
#include <sys/systm.h> |
#include <sys/systm.h> |
#include <sys/proc.h> |
#include <sys/proc.h> |
#include <sys/file.h> |
#include <sys/file.h> |
#include <sys/filedesc.h> |
|
#include <sys/malloc.h> |
#include <sys/malloc.h> |
#include <sys/mbuf.h> |
#include <sys/mbuf.h> |
#include <sys/domain.h> |
#include <sys/domain.h> |
Line 104 POOL_INIT(socket_pool, sizeof(struct soc |
|
Line 103 POOL_INIT(socket_pool, sizeof(struct soc |
|
MALLOC_DEFINE(M_SOOPTS, "soopts", "socket options"); |
MALLOC_DEFINE(M_SOOPTS, "soopts", "socket options"); |
MALLOC_DEFINE(M_SONAME, "soname", "socket name"); |
MALLOC_DEFINE(M_SONAME, "soname", "socket name"); |
|
|
extern const struct fileops socketops; |
|
|
|
extern int somaxconn; /* patchable (XXX sysctl) */ |
extern int somaxconn; /* patchable (XXX sysctl) */ |
int somaxconn = SOMAXCONN; |
int somaxconn = SOMAXCONN; |
|
|
|
|
{ |
{ |
struct mbuf *m, *next; |
struct mbuf *m, *next; |
size_t rv = 0; |
size_t rv = 0; |
|
int s; |
|
|
KASSERT(mutex_owned(&so_pendfree_lock)); |
KASSERT(mutex_owned(&so_pendfree_lock)); |
|
|
|
|
sodoloanfree((m->m_flags & M_EXT_PAGES) ? |
sodoloanfree((m->m_flags & M_EXT_PAGES) ? |
m->m_ext.ext_pgs : NULL, m->m_ext.ext_buf, |
m->m_ext.ext_pgs : NULL, m->m_ext.ext_buf, |
m->m_ext.ext_size); |
m->m_ext.ext_size); |
pool_cache_put(mb_cache, m); |
s = splvm(); |
|
pool_cache_put(&mbpool_cache, m); |
|
splx(s); |
} |
} |
|
|
mutex_enter(&so_pendfree_lock); |
mutex_enter(&so_pendfree_lock); |
Line 427 sokva_reclaim_callback(struct callback_e |
|
Line 427 sokva_reclaim_callback(struct callback_e |
|
return CALLBACK_CHAIN_CONTINUE; |
return CALLBACK_CHAIN_CONTINUE; |
} |
} |
|
|
struct mbuf * |
|
getsombuf(struct socket *so, int type) |
|
{ |
|
struct mbuf *m; |
|
|
|
m = m_get(M_WAIT, type); |
|
MCLAIM(m, so->so_mowner); |
|
return m; |
|
} |
|
|
|
struct mbuf * |
|
m_intopt(struct socket *so, int val) |
|
{ |
|
struct mbuf *m; |
|
|
|
m = getsombuf(so, MT_SOOPTS); |
|
m->m_len = sizeof(int); |
|
*mtod(m, int *) = val; |
|
return m; |
|
} |
|
|
|
void |
void |
soinit(void) |
soinit(void) |
{ |
{ |
|
|
mutex_init(&so_pendfree_lock, MUTEX_DEFAULT, IPL_VM); |
mutex_init(&so_pendfree_lock, MUTEX_DRIVER, IPL_VM); |
cv_init(&socurkva_cv, "sokva"); |
cv_init(&socurkva_cv, "sokva"); |
|
|
/* Set the initial adjusted socket buffer size. */ |
/* Set the initial adjusted socket buffer size. */ |
Line 516 socreate(int dom, struct socket **aso, i |
|
Line 495 socreate(int dom, struct socket **aso, i |
|
so->so_snd.sb_mowner = &prp->pr_domain->dom_mowner; |
so->so_snd.sb_mowner = &prp->pr_domain->dom_mowner; |
so->so_mowner = &prp->pr_domain->dom_mowner; |
so->so_mowner = &prp->pr_domain->dom_mowner; |
#endif |
#endif |
selinit(&so->so_rcv.sb_sel); |
|
selinit(&so->so_snd.sb_sel); |
|
uid = kauth_cred_geteuid(l->l_cred); |
uid = kauth_cred_geteuid(l->l_cred); |
so->so_uidinfo = uid_find(uid); |
so->so_uidinfo = uid_find(uid); |
error = (*prp->pr_usrreq)(so, PRU_ATTACH, NULL, |
error = (*prp->pr_usrreq)(so, PRU_ATTACH, NULL, |
Line 533 socreate(int dom, struct socket **aso, i |
|
Line 510 socreate(int dom, struct socket **aso, i |
|
return 0; |
return 0; |
} |
} |
|
|
/* On success, write file descriptor to fdout and return zero. On |
|
* failure, return non-zero; *fdout will be undefined. |
|
*/ |
|
int |
|
fsocreate(int domain, struct socket **sop, int type, int protocol, |
|
struct lwp *l, int *fdout) |
|
{ |
|
struct filedesc *fdp; |
|
struct socket *so; |
|
struct file *fp; |
|
int fd, error; |
|
|
|
fdp = l->l_proc->p_fd; |
|
/* falloc() will use the desciptor for us */ |
|
if ((error = falloc(l, &fp, &fd)) != 0) |
|
return (error); |
|
fp->f_flag = FREAD|FWRITE; |
|
fp->f_type = DTYPE_SOCKET; |
|
fp->f_ops = &socketops; |
|
error = socreate(domain, &so, type, protocol, l); |
|
if (error != 0) { |
|
FILE_UNUSE(fp, l); |
|
fdremove(fdp, fd); |
|
ffree(fp); |
|
} else { |
|
if (sop != NULL) |
|
*sop = so; |
|
fp->f_data = so; |
|
FILE_SET_MATURE(fp); |
|
FILE_UNUSE(fp, l); |
|
*fdout = fd; |
|
} |
|
return error; |
|
} |
|
|
|
int |
int |
sobind(struct socket *so, struct mbuf *nam, struct lwp *l) |
sobind(struct socket *so, struct mbuf *nam, struct lwp *l) |
{ |
{ |
Line 623 sofree(struct socket *so) |
|
Line 565 sofree(struct socket *so) |
|
RLIM_INFINITY); |
RLIM_INFINITY); |
sbrelease(&so->so_snd, so); |
sbrelease(&so->so_snd, so); |
sorflush(so); |
sorflush(so); |
seldestroy(&so->so_rcv.sb_sel); |
|
seldestroy(&so->so_snd.sb_sel); |
|
pool_put(&socket_pool, so); |
pool_put(&socket_pool, so); |
} |
} |
|
|
Line 891 sosend(struct socket *so, struct mbuf *a |
|
Line 831 sosend(struct socket *so, struct mbuf *a |
|
if (flags & MSG_EOR) |
if (flags & MSG_EOR) |
top->m_flags |= M_EOR; |
top->m_flags |= M_EOR; |
} else do { |
} else do { |
if (top == NULL) { |
if (top == 0) { |
m = m_gethdr(M_WAIT, MT_DATA); |
m = m_gethdr(M_WAIT, MT_DATA); |
mlen = MHLEN; |
mlen = MHLEN; |
m->m_pkthdr.len = 0; |
m->m_pkthdr.len = 0; |
Line 935 sosend(struct socket *so, struct mbuf *a |
|
Line 875 sosend(struct socket *so, struct mbuf *a |
|
if (atomic && top == 0 && len < mlen) |
if (atomic && top == 0 && len < mlen) |
MH_ALIGN(m, len); |
MH_ALIGN(m, len); |
} |
} |
error = uiomove(mtod(m, void *), (int)len, uio); |
error = uiomove(mtod(m, void *), (int)len, |
|
uio); |
have_data: |
have_data: |
resid = uio->uio_resid; |
resid = uio->uio_resid; |
m->m_len = len; |
m->m_len = len; |
*mp = m; |
*mp = m; |
top->m_pkthdr.len += len; |
top->m_pkthdr.len += len; |
if (error != 0) |
if (error) |
goto release; |
goto release; |
mp = &m->m_next; |
mp = &m->m_next; |
if (resid <= 0) { |
if (resid <= 0) { |
Line 970 sosend(struct socket *so, struct mbuf *a |
|
Line 911 sosend(struct socket *so, struct mbuf *a |
|
splx(s); |
splx(s); |
|
|
clen = 0; |
clen = 0; |
control = NULL; |
control = 0; |
top = NULL; |
top = 0; |
mp = ⊤ |
mp = ⊤ |
if (error != 0) |
if (error) |
goto release; |
goto release; |
} while (resid && space > 0); |
} while (resid && space > 0); |
} while (resid); |
} while (resid); |
Line 1010 soreceive(struct socket *so, struct mbuf |
|
Line 951 soreceive(struct socket *so, struct mbuf |
|
{ |
{ |
struct lwp *l = curlwp; |
struct lwp *l = curlwp; |
struct mbuf *m, **mp; |
struct mbuf *m, **mp; |
int atomic, flags, len, error, s, offset, moff, type, orig_resid; |
int flags, len, error, s, offset, moff, type, orig_resid; |
const struct protosw *pr; |
const struct protosw *pr; |
struct mbuf *nextrecord; |
struct mbuf *nextrecord; |
int mbuf_removed = 0; |
int mbuf_removed = 0; |
const struct domain *dom; |
|
|
|
pr = so->so_proto; |
pr = so->so_proto; |
atomic = pr->pr_flags & PR_ATOMIC; |
|
dom = pr->pr_domain; |
|
mp = mp0; |
mp = mp0; |
type = 0; |
type = 0; |
orig_resid = uio->uio_resid; |
orig_resid = uio->uio_resid; |
|
|
if (paddr != NULL) |
if (paddr) |
*paddr = NULL; |
*paddr = 0; |
if (controlp != NULL) |
if (controlp) |
*controlp = NULL; |
*controlp = 0; |
if (flagsp != NULL) |
if (flagsp) |
flags = *flagsp &~ MSG_EOR; |
flags = *flagsp &~ MSG_EOR; |
else |
else |
flags = 0; |
flags = 0; |
Line 1045 soreceive(struct socket *so, struct mbuf |
|
Line 983 soreceive(struct socket *so, struct mbuf |
|
error = uiomove(mtod(m, void *), |
error = uiomove(mtod(m, void *), |
(int) min(uio->uio_resid, m->m_len), uio); |
(int) min(uio->uio_resid, m->m_len), uio); |
m = m_free(m); |
m = m_free(m); |
} while (uio->uio_resid > 0 && error == 0 && m); |
} while (uio->uio_resid && error == 0 && m); |
bad: |
bad: |
if (m != NULL) |
if (m) |
m_freem(m); |
m_freem(m); |
return error; |
return (error); |
} |
} |
if (mp != NULL) |
if (mp) |
*mp = NULL; |
*mp = NULL; |
if (so->so_state & SS_ISCONFIRMING && uio->uio_resid) |
if (so->so_state & SS_ISCONFIRMING && uio->uio_resid) |
(*pr->pr_usrreq)(so, PRU_RCVD, NULL, NULL, NULL, l); |
(*pr->pr_usrreq)(so, PRU_RCVD, NULL, NULL, NULL, l); |
|
|
restart: |
restart: |
if ((error = sblock(&so->so_rcv, SBLOCKWAIT(flags))) != 0) |
if ((error = sblock(&so->so_rcv, SBLOCKWAIT(flags))) != 0) |
return error; |
return (error); |
s = splsoftnet(); |
s = splsoftnet(); |
|
|
m = so->so_rcv.sb_mb; |
m = so->so_rcv.sb_mb; |
Line 1073 soreceive(struct socket *so, struct mbuf |
|
Line 1011 soreceive(struct socket *so, struct mbuf |
|
* we have to do the receive in sections, and thus risk returning |
* we have to do the receive in sections, and thus risk returning |
* a short count if a timeout or signal occurs after we start. |
* a short count if a timeout or signal occurs after we start. |
*/ |
*/ |
if (m == NULL || |
if (m == 0 || (((flags & MSG_DONTWAIT) == 0 && |
((flags & MSG_DONTWAIT) == 0 && |
so->so_rcv.sb_cc < uio->uio_resid) && |
so->so_rcv.sb_cc < uio->uio_resid && |
(so->so_rcv.sb_cc < so->so_rcv.sb_lowat || |
(so->so_rcv.sb_cc < so->so_rcv.sb_lowat || |
((flags & MSG_WAITALL) && uio->uio_resid <= so->so_rcv.sb_hiwat)) && |
((flags & MSG_WAITALL) && |
m->m_nextpkt == 0 && (pr->pr_flags & PR_ATOMIC) == 0)) { |
uio->uio_resid <= so->so_rcv.sb_hiwat)) && |
|
m->m_nextpkt == NULL && !atomic)) { |
|
#ifdef DIAGNOSTIC |
#ifdef DIAGNOSTIC |
if (m == NULL && so->so_rcv.sb_cc) |
if (m == 0 && so->so_rcv.sb_cc) |
panic("receive 1"); |
panic("receive 1"); |
#endif |
#endif |
if (so->so_error) { |
if (so->so_error) { |
if (m != NULL) |
if (m) |
goto dontblock; |
goto dontblock; |
error = so->so_error; |
error = so->so_error; |
if ((flags & MSG_PEEK) == 0) |
if ((flags & MSG_PEEK) == 0) |
Line 1093 soreceive(struct socket *so, struct mbuf |
|
Line 1029 soreceive(struct socket *so, struct mbuf |
|
goto release; |
goto release; |
} |
} |
if (so->so_state & SS_CANTRCVMORE) { |
if (so->so_state & SS_CANTRCVMORE) { |
if (m != NULL) |
if (m) |
goto dontblock; |
goto dontblock; |
else |
else |
goto release; |
goto release; |
} |
} |
for (; m != NULL; m = m->m_next) |
for (; m; m = m->m_next) |
if (m->m_type == MT_OOBDATA || (m->m_flags & M_EOR)) { |
if (m->m_type == MT_OOBDATA || (m->m_flags & M_EOR)) { |
m = so->so_rcv.sb_mb; |
m = so->so_rcv.sb_mb; |
goto dontblock; |
goto dontblock; |
Line 1119 soreceive(struct socket *so, struct mbuf |
|
Line 1055 soreceive(struct socket *so, struct mbuf |
|
sbunlock(&so->so_rcv); |
sbunlock(&so->so_rcv); |
error = sbwait(&so->so_rcv); |
error = sbwait(&so->so_rcv); |
splx(s); |
splx(s); |
if (error != 0) |
if (error) |
return error; |
return (error); |
goto restart; |
goto restart; |
} |
} |
dontblock: |
dontblock: |
Line 1129 soreceive(struct socket *so, struct mbuf |
|
Line 1065 soreceive(struct socket *so, struct mbuf |
|
* While we process the initial mbufs containing address and control |
* While we process the initial mbufs containing address and control |
* info, we save a copy of m->m_nextpkt into nextrecord. |
* info, we save a copy of m->m_nextpkt into nextrecord. |
*/ |
*/ |
if (l != NULL) |
if (l) |
l->l_proc->p_stats->p_ru.ru_msgrcv++; |
l->l_proc->p_stats->p_ru.ru_msgrcv++; |
KASSERT(m == so->so_rcv.sb_mb); |
KASSERT(m == so->so_rcv.sb_mb); |
SBLASTRECORDCHK(&so->so_rcv, "soreceive 1"); |
SBLASTRECORDCHK(&so->so_rcv, "soreceive 1"); |
Line 1148 soreceive(struct socket *so, struct mbuf |
|
Line 1084 soreceive(struct socket *so, struct mbuf |
|
} else { |
} else { |
sbfree(&so->so_rcv, m); |
sbfree(&so->so_rcv, m); |
mbuf_removed = 1; |
mbuf_removed = 1; |
if (paddr != NULL) { |
if (paddr) { |
*paddr = m; |
*paddr = m; |
so->so_rcv.sb_mb = m->m_next; |
so->so_rcv.sb_mb = m->m_next; |
m->m_next = NULL; |
m->m_next = 0; |
m = so->so_rcv.sb_mb; |
m = so->so_rcv.sb_mb; |
} else { |
} else { |
MFREE(m, so->so_rcv.sb_mb); |
MFREE(m, so->so_rcv.sb_mb); |
Line 1159 soreceive(struct socket *so, struct mbuf |
|
Line 1095 soreceive(struct socket *so, struct mbuf |
|
} |
} |
} |
} |
} |
} |
while (m != NULL && m->m_type == MT_CONTROL && error == 0) { |
while (m && m->m_type == MT_CONTROL && error == 0) { |
if (flags & MSG_PEEK) { |
if (flags & MSG_PEEK) { |
if (controlp != NULL) |
if (controlp) |
*controlp = m_copy(m, 0, m->m_len); |
*controlp = m_copy(m, 0, m->m_len); |
m = m->m_next; |
m = m->m_next; |
} else { |
} else { |
sbfree(&so->so_rcv, m); |
sbfree(&so->so_rcv, m); |
mbuf_removed = 1; |
mbuf_removed = 1; |
if (controlp != NULL) { |
if (controlp) { |
|
struct domain *dom = pr->pr_domain; |
if (dom->dom_externalize && l && |
if (dom->dom_externalize && l && |
mtod(m, struct cmsghdr *)->cmsg_type == |
mtod(m, struct cmsghdr *)->cmsg_type == |
SCM_RIGHTS) |
SCM_RIGHTS) |
error = (*dom->dom_externalize)(m, l); |
error = (*dom->dom_externalize)(m, l); |
*controlp = m; |
*controlp = m; |
so->so_rcv.sb_mb = m->m_next; |
so->so_rcv.sb_mb = m->m_next; |
m->m_next = NULL; |
m->m_next = 0; |
m = so->so_rcv.sb_mb; |
m = so->so_rcv.sb_mb; |
} else { |
} else { |
/* |
/* |
* Dispose of any SCM_RIGHTS message that went |
* Dispose of any SCM_RIGHTS message that went |
* through the read path rather than recv. |
* through the read path rather than recv. |
*/ |
*/ |
if (dom->dom_dispose && |
if (pr->pr_domain->dom_dispose && |
mtod(m, struct cmsghdr *)->cmsg_type == SCM_RIGHTS) |
mtod(m, struct cmsghdr *)->cmsg_type == SCM_RIGHTS) |
(*dom->dom_dispose)(m); |
(*pr->pr_domain->dom_dispose)(m); |
MFREE(m, so->so_rcv.sb_mb); |
MFREE(m, so->so_rcv.sb_mb); |
m = so->so_rcv.sb_mb; |
m = so->so_rcv.sb_mb; |
} |
} |
} |
} |
if (controlp != NULL) { |
if (controlp) { |
orig_resid = 0; |
orig_resid = 0; |
controlp = &(*controlp)->m_next; |
controlp = &(*controlp)->m_next; |
} |
} |
Line 1200 soreceive(struct socket *so, struct mbuf |
|
Line 1137 soreceive(struct socket *so, struct mbuf |
|
* the last packet on the chain (nextrecord == NULL) and we |
* the last packet on the chain (nextrecord == NULL) and we |
* change m->m_nextpkt. |
* change m->m_nextpkt. |
*/ |
*/ |
if (m != NULL) { |
if (m) { |
if ((flags & MSG_PEEK) == 0) { |
if ((flags & MSG_PEEK) == 0) { |
m->m_nextpkt = nextrecord; |
m->m_nextpkt = nextrecord; |
/* |
/* |
Line 1228 soreceive(struct socket *so, struct mbuf |
|
Line 1165 soreceive(struct socket *so, struct mbuf |
|
|
|
moff = 0; |
moff = 0; |
offset = 0; |
offset = 0; |
while (m != NULL && uio->uio_resid > 0 && error == 0) { |
while (m && uio->uio_resid > 0 && error == 0) { |
if (m->m_type == MT_OOBDATA) { |
if (m->m_type == MT_OOBDATA) { |
if (type != MT_OOBDATA) |
if (type != MT_OOBDATA) |
break; |
break; |
Line 1252 soreceive(struct socket *so, struct mbuf |
|
Line 1189 soreceive(struct socket *so, struct mbuf |
|
* we must note any additions to the sockbuf when we |
* we must note any additions to the sockbuf when we |
* block interrupts again. |
* block interrupts again. |
*/ |
*/ |
if (mp == NULL) { |
if (mp == 0) { |
SBLASTRECORDCHK(&so->so_rcv, "soreceive uiomove"); |
SBLASTRECORDCHK(&so->so_rcv, "soreceive uiomove"); |
SBLASTMBUFCHK(&so->so_rcv, "soreceive uiomove"); |
SBLASTMBUFCHK(&so->so_rcv, "soreceive uiomove"); |
splx(s); |
splx(s); |
error = uiomove(mtod(m, char *) + moff, (int)len, uio); |
error = uiomove(mtod(m, char *) + moff, (int)len, uio); |
s = splsoftnet(); |
s = splsoftnet(); |
if (error != 0) { |
if (error) { |
/* |
/* |
* If any part of the record has been removed |
* If any part of the record has been removed |
* (such as the MT_SONAME mbuf, which will |
* (such as the MT_SONAME mbuf, which will |
Line 1270 soreceive(struct socket *so, struct mbuf |
|
Line 1207 soreceive(struct socket *so, struct mbuf |
|
* This avoids a later panic("receive 1a") |
* This avoids a later panic("receive 1a") |
* when compiled with DIAGNOSTIC. |
* when compiled with DIAGNOSTIC. |
*/ |
*/ |
if (m && mbuf_removed && atomic) |
if (m && mbuf_removed |
|
&& (pr->pr_flags & PR_ATOMIC)) |
(void) sbdroprecord(&so->so_rcv); |
(void) sbdroprecord(&so->so_rcv); |
|
|
goto release; |
goto release; |
Line 1311 soreceive(struct socket *so, struct mbuf |
|
Line 1249 soreceive(struct socket *so, struct mbuf |
|
SBLASTRECORDCHK(&so->so_rcv, "soreceive 3"); |
SBLASTRECORDCHK(&so->so_rcv, "soreceive 3"); |
SBLASTMBUFCHK(&so->so_rcv, "soreceive 3"); |
SBLASTMBUFCHK(&so->so_rcv, "soreceive 3"); |
} |
} |
} else if (flags & MSG_PEEK) |
} else { |
moff += len; |
if (flags & MSG_PEEK) |
else { |
moff += len; |
if (mp != NULL) |
else { |
*mp = m_copym(m, 0, len, M_WAIT); |
if (mp) |
m->m_data += len; |
*mp = m_copym(m, 0, len, M_WAIT); |
m->m_len -= len; |
m->m_data += len; |
so->so_rcv.sb_cc -= len; |
m->m_len -= len; |
|
so->so_rcv.sb_cc -= len; |
|
} |
} |
} |
if (so->so_oobmark) { |
if (so->so_oobmark) { |
if ((flags & MSG_PEEK) == 0) { |
if ((flags & MSG_PEEK) == 0) { |
Line 1342 soreceive(struct socket *so, struct mbuf |
|
Line 1282 soreceive(struct socket *so, struct mbuf |
|
* with a short count but without error. |
* with a short count but without error. |
* Keep sockbuf locked against other readers. |
* Keep sockbuf locked against other readers. |
*/ |
*/ |
while (flags & MSG_WAITALL && m == NULL && uio->uio_resid > 0 && |
while (flags & MSG_WAITALL && m == 0 && uio->uio_resid > 0 && |
!sosendallatonce(so) && !nextrecord) { |
!sosendallatonce(so) && !nextrecord) { |
if (so->so_error || so->so_state & SS_CANTRCVMORE) |
if (so->so_error || so->so_state & SS_CANTRCVMORE) |
break; |
break; |
Line 1363 soreceive(struct socket *so, struct mbuf |
|
Line 1303 soreceive(struct socket *so, struct mbuf |
|
SBLASTRECORDCHK(&so->so_rcv, "soreceive sbwait 2"); |
SBLASTRECORDCHK(&so->so_rcv, "soreceive sbwait 2"); |
SBLASTMBUFCHK(&so->so_rcv, "soreceive sbwait 2"); |
SBLASTMBUFCHK(&so->so_rcv, "soreceive sbwait 2"); |
error = sbwait(&so->so_rcv); |
error = sbwait(&so->so_rcv); |
if (error != 0) { |
if (error) { |
sbunlock(&so->so_rcv); |
sbunlock(&so->so_rcv); |
splx(s); |
splx(s); |
return 0; |
return (0); |
} |
} |
if ((m = so->so_rcv.sb_mb) != NULL) |
if ((m = so->so_rcv.sb_mb) != NULL) |
nextrecord = m->m_nextpkt; |
nextrecord = m->m_nextpkt; |
} |
} |
} |
} |
|
|
if (m && atomic) { |
if (m && pr->pr_flags & PR_ATOMIC) { |
flags |= MSG_TRUNC; |
flags |= MSG_TRUNC; |
if ((flags & MSG_PEEK) == 0) |
if ((flags & MSG_PEEK) == 0) |
(void) sbdroprecord(&so->so_rcv); |
(void) sbdroprecord(&so->so_rcv); |
} |
} |
if ((flags & MSG_PEEK) == 0) { |
if ((flags & MSG_PEEK) == 0) { |
if (m == NULL) { |
if (m == 0) { |
/* |
/* |
* First part is an inline SB_EMPTY_FIXUP(). Second |
* First part is an inline SB_EMPTY_FIXUP(). Second |
* part makes sure sb_lastrecord is up-to-date if |
* part makes sure sb_lastrecord is up-to-date if |
Line 1405 soreceive(struct socket *so, struct mbuf |
|
Line 1345 soreceive(struct socket *so, struct mbuf |
|
goto restart; |
goto restart; |
} |
} |
|
|
if (flagsp != NULL) |
if (flagsp) |
*flagsp |= flags; |
*flagsp |= flags; |
release: |
release: |
sbunlock(&so->so_rcv); |
sbunlock(&so->so_rcv); |
splx(s); |
splx(s); |
return error; |
return (error); |
} |
} |
|
|
int |
int |
Line 1427 soshutdown(struct socket *so, int how) |
|
Line 1367 soshutdown(struct socket *so, int how) |
|
if (how == SHUT_WR || how == SHUT_RDWR) |
if (how == SHUT_WR || how == SHUT_RDWR) |
return (*pr->pr_usrreq)(so, PRU_SHUTDOWN, NULL, |
return (*pr->pr_usrreq)(so, PRU_SHUTDOWN, NULL, |
NULL, NULL, NULL); |
NULL, NULL, NULL); |
return 0; |
return (0); |
} |
} |
|
|
void |
void |
Line 1457 sorflush(struct socket *so) |
|
Line 1397 sorflush(struct socket *so) |
|
sbrelease(&asb, so); |
sbrelease(&asb, so); |
} |
} |
|
|
static int |
int |
sosetopt1(struct socket *so, int level, int optname, struct mbuf *m) |
sosetopt(struct socket *so, int level, int optname, struct mbuf *m0) |
{ |
{ |
int optval, val; |
int error; |
|
struct mbuf *m; |
struct linger *l; |
struct linger *l; |
struct sockbuf *sb; |
struct sockbuf *sb; |
struct timeval *tv; |
|
|
|
switch (optname) { |
|
|
|
case SO_LINGER: |
|
if (m == NULL || m->m_len != sizeof(struct linger)) |
|
return EINVAL; |
|
l = mtod(m, struct linger *); |
|
if (l->l_linger < 0 || l->l_linger > USHRT_MAX || |
|
l->l_linger > (INT_MAX / hz)) |
|
return EDOM; |
|
so->so_linger = l->l_linger; |
|
if (l->l_onoff) |
|
so->so_options |= SO_LINGER; |
|
else |
|
so->so_options &= ~SO_LINGER; |
|
break; |
|
|
|
case SO_DEBUG: |
|
case SO_KEEPALIVE: |
|
case SO_DONTROUTE: |
|
case SO_USELOOPBACK: |
|
case SO_BROADCAST: |
|
case SO_REUSEADDR: |
|
case SO_REUSEPORT: |
|
case SO_OOBINLINE: |
|
case SO_TIMESTAMP: |
|
if (m == NULL || m->m_len < sizeof(int)) |
|
return EINVAL; |
|
if (*mtod(m, int *)) |
|
so->so_options |= optname; |
|
else |
|
so->so_options &= ~optname; |
|
break; |
|
|
|
case SO_SNDBUF: |
error = 0; |
case SO_RCVBUF: |
m = m0; |
case SO_SNDLOWAT: |
if (level != SOL_SOCKET) { |
case SO_RCVLOWAT: |
if (so->so_proto && so->so_proto->pr_ctloutput) |
if (m == NULL || m->m_len < sizeof(int)) |
return ((*so->so_proto->pr_ctloutput) |
return EINVAL; |
(PRCO_SETOPT, so, level, optname, &m0)); |
|
error = ENOPROTOOPT; |
|
} else { |
|
switch (optname) { |
|
|
/* |
case SO_LINGER: |
* Values < 1 make no sense for any of these |
if (m == NULL || m->m_len != sizeof(struct linger)) { |
* options, so disallow them. |
error = EINVAL; |
*/ |
goto bad; |
optval = *mtod(m, int *); |
} |
if (optval < 1) |
l = mtod(m, struct linger *); |
return EINVAL; |
if (l->l_linger < 0 || l->l_linger > USHRT_MAX || |
|
l->l_linger > (INT_MAX / hz)) { |
|
error = EDOM; |
|
goto bad; |
|
} |
|
so->so_linger = l->l_linger; |
|
if (l->l_onoff) |
|
so->so_options |= SO_LINGER; |
|
else |
|
so->so_options &= ~SO_LINGER; |
|
break; |
|
|
switch (optname) { |
case SO_DEBUG: |
|
case SO_KEEPALIVE: |
|
case SO_DONTROUTE: |
|
case SO_USELOOPBACK: |
|
case SO_BROADCAST: |
|
case SO_REUSEADDR: |
|
case SO_REUSEPORT: |
|
case SO_OOBINLINE: |
|
case SO_TIMESTAMP: |
|
if (m == NULL || m->m_len < sizeof(int)) { |
|
error = EINVAL; |
|
goto bad; |
|
} |
|
if (*mtod(m, int *)) |
|
so->so_options |= optname; |
|
else |
|
so->so_options &= ~optname; |
|
break; |
|
|
case SO_SNDBUF: |
case SO_SNDBUF: |
case SO_RCVBUF: |
case SO_RCVBUF: |
sb = (optname == SO_SNDBUF) ? |
|
&so->so_snd : &so->so_rcv; |
|
if (sbreserve(sb, (u_long)optval, so) == 0) |
|
return ENOBUFS; |
|
sb->sb_flags &= ~SB_AUTOSIZE; |
|
break; |
|
|
|
/* |
|
* Make sure the low-water is never greater than |
|
* the high-water. |
|
*/ |
|
case SO_SNDLOWAT: |
case SO_SNDLOWAT: |
so->so_snd.sb_lowat = |
|
(optval > so->so_snd.sb_hiwat) ? |
|
so->so_snd.sb_hiwat : optval; |
|
break; |
|
case SO_RCVLOWAT: |
case SO_RCVLOWAT: |
so->so_rcv.sb_lowat = |
{ |
(optval > so->so_rcv.sb_hiwat) ? |
int optval; |
so->so_rcv.sb_hiwat : optval; |
|
break; |
|
} |
|
break; |
|
|
|
case SO_SNDTIMEO: |
if (m == NULL || m->m_len < sizeof(int)) { |
case SO_RCVTIMEO: |
error = EINVAL; |
if (m == NULL || m->m_len < sizeof(*tv)) |
goto bad; |
return EINVAL; |
} |
tv = mtod(m, struct timeval *); |
|
if (tv->tv_sec > (INT_MAX - tv->tv_usec / tick) / hz) |
|
return EDOM; |
|
val = tv->tv_sec * hz + tv->tv_usec / tick; |
|
if (val == 0 && tv->tv_usec != 0) |
|
val = 1; |
|
|
|
switch (optname) { |
/* |
|
* Values < 1 make no sense for any of these |
|
* options, so disallow them. |
|
*/ |
|
optval = *mtod(m, int *); |
|
if (optval < 1) { |
|
error = EINVAL; |
|
goto bad; |
|
} |
|
|
case SO_SNDTIMEO: |
switch (optname) { |
so->so_snd.sb_timeo = val; |
|
|
case SO_SNDBUF: |
|
case SO_RCVBUF: |
|
sb = (optname == SO_SNDBUF) ? |
|
&so->so_snd : &so->so_rcv; |
|
if (sbreserve(sb, (u_long)optval, so) == 0) { |
|
error = ENOBUFS; |
|
goto bad; |
|
} |
|
sb->sb_flags &= ~SB_AUTOSIZE; |
|
break; |
|
|
|
/* |
|
* Make sure the low-water is never greater than |
|
* the high-water. |
|
*/ |
|
case SO_SNDLOWAT: |
|
so->so_snd.sb_lowat = |
|
(optval > so->so_snd.sb_hiwat) ? |
|
so->so_snd.sb_hiwat : optval; |
|
break; |
|
case SO_RCVLOWAT: |
|
so->so_rcv.sb_lowat = |
|
(optval > so->so_rcv.sb_hiwat) ? |
|
so->so_rcv.sb_hiwat : optval; |
|
break; |
|
} |
break; |
break; |
|
} |
|
|
|
case SO_SNDTIMEO: |
case SO_RCVTIMEO: |
case SO_RCVTIMEO: |
so->so_rcv.sb_timeo = val; |
{ |
break; |
struct timeval *tv; |
} |
int val; |
break; |
|
|
|
default: |
if (m == NULL || m->m_len < sizeof(*tv)) { |
return ENOPROTOOPT; |
error = EINVAL; |
} |
goto bad; |
return 0; |
} |
} |
tv = mtod(m, struct timeval *); |
|
if (tv->tv_sec > (INT_MAX - tv->tv_usec / tick) / hz) { |
|
error = EDOM; |
|
goto bad; |
|
} |
|
val = tv->tv_sec * hz + tv->tv_usec / tick; |
|
if (val == 0 && tv->tv_usec != 0) |
|
val = 1; |
|
|
int |
switch (optname) { |
sosetopt(struct socket *so, int level, int optname, struct mbuf *m) |
|
{ |
|
int error, prerr; |
|
|
|
if (level == SOL_SOCKET) |
case SO_SNDTIMEO: |
error = sosetopt1(so, level, optname, m); |
so->so_snd.sb_timeo = val; |
else |
break; |
error = ENOPROTOOPT; |
case SO_RCVTIMEO: |
|
so->so_rcv.sb_timeo = val; |
|
break; |
|
} |
|
break; |
|
} |
|
|
if ((error == 0 || error == ENOPROTOOPT) && |
default: |
so->so_proto != NULL && so->so_proto->pr_ctloutput != NULL) { |
error = ENOPROTOOPT; |
/* give the protocol stack a shot */ |
break; |
prerr = (*so->so_proto->pr_ctloutput)(PRCO_SETOPT, so, level, |
} |
optname, &m); |
if (error == 0 && so->so_proto && so->so_proto->pr_ctloutput) { |
if (prerr == 0) |
(void) ((*so->so_proto->pr_ctloutput) |
error = 0; |
(PRCO_SETOPT, so, level, optname, &m0)); |
else if (prerr != ENOPROTOOPT) |
m = NULL; /* freed by protocol */ |
error = prerr; |
} |
} else if (m != NULL) |
} |
(void)m_free(m); |
bad: |
return error; |
if (m) |
|
(void) m_free(m); |
|
return (error); |
} |
} |
|
|
int |
int |
Line 1795 soo_kqfilter(struct file *fp, struct kno |
|
Line 1754 soo_kqfilter(struct file *fp, struct kno |
|
sb = &so->so_snd; |
sb = &so->so_snd; |
break; |
break; |
default: |
default: |
return (EINVAL); |
return (1); |
} |
} |
SLIST_INSERT_HEAD(&sb->sb_sel.sel_klist, kn, kn_selnext); |
SLIST_INSERT_HEAD(&sb->sb_sel.sel_klist, kn, kn_selnext); |
sb->sb_flags |= SB_KNOTE; |
sb->sb_flags |= SB_KNOTE; |