| version 1.66, 2002/05/07 08:06:35 |
version 1.66.2.3, 2002/08/29 05:23:14 |
|
|
| #endif /* SOSEND_COUNTERS */ |
#endif /* SOSEND_COUNTERS */ |
| } |
} |
| |
|
| #ifdef SOSEND_LOAN |
#ifdef SOSEND_NO_LOAN |
| int use_sosend_loan = 1; |
|
| #else |
|
| int use_sosend_loan = 0; |
int use_sosend_loan = 0; |
| |
#else |
| |
int use_sosend_loan = 1; |
| #endif |
#endif |
| |
|
| struct mbuf *so_pendfree; |
struct mbuf *so_pendfree; |
| Line 797 soreceive(struct socket *so, struct mbuf |
|
| Line 797 soreceive(struct socket *so, struct mbuf |
|
| int flags, len, error, s, offset, moff, type, orig_resid; |
int flags, len, error, s, offset, moff, type, orig_resid; |
| struct protosw *pr; |
struct protosw *pr; |
| struct mbuf *nextrecord; |
struct mbuf *nextrecord; |
| |
int mbuf_removed = 0; |
| |
|
| pr = so->so_proto; |
pr = so->so_proto; |
| mp = mp0; |
mp = mp0; |
| Line 893 soreceive(struct socket *so, struct mbuf |
|
| Line 894 soreceive(struct socket *so, struct mbuf |
|
| error = EWOULDBLOCK; |
error = EWOULDBLOCK; |
| goto release; |
goto release; |
| } |
} |
| |
SBLASTRECORDCHK(&so->so_rcv, "soreceive sbwait 1"); |
| |
SBLASTMBUFCHK(&so->so_rcv, "soreceive sbwait 1"); |
| sbunlock(&so->so_rcv); |
sbunlock(&so->so_rcv); |
| error = sbwait(&so->so_rcv); |
error = sbwait(&so->so_rcv); |
| splx(s); |
splx(s); |
| Line 901 soreceive(struct socket *so, struct mbuf |
|
| Line 904 soreceive(struct socket *so, struct mbuf |
|
| goto restart; |
goto restart; |
| } |
} |
| dontblock: |
dontblock: |
| |
/* |
| |
* On entry here, m points to the first record of the socket buffer. |
| |
* While we process the initial mbufs containing address and control |
| |
* info, we save a copy of m->m_nextpkt into nextrecord. |
| |
*/ |
| #ifdef notyet /* XXXX */ |
#ifdef notyet /* XXXX */ |
| if (uio->uio_procp) |
if (uio->uio_procp) |
| uio->uio_procp->p_stats->p_ru.ru_msgrcv++; |
uio->uio_procp->p_stats->p_ru.ru_msgrcv++; |
| #endif |
#endif |
| |
KASSERT(m == so->so_rcv.sb_mb); |
| |
SBLASTRECORDCHK(&so->so_rcv, "soreceive 1"); |
| |
SBLASTMBUFCHK(&so->so_rcv, "soreceive 1"); |
| nextrecord = m->m_nextpkt; |
nextrecord = m->m_nextpkt; |
| if (pr->pr_flags & PR_ADDR) { |
if (pr->pr_flags & PR_ADDR) { |
| #ifdef DIAGNOSTIC |
#ifdef DIAGNOSTIC |
| Line 918 soreceive(struct socket *so, struct mbuf |
|
| Line 929 soreceive(struct socket *so, struct mbuf |
|
| m = m->m_next; |
m = m->m_next; |
| } else { |
} else { |
| sbfree(&so->so_rcv, m); |
sbfree(&so->so_rcv, m); |
| |
mbuf_removed = 1; |
| if (paddr) { |
if (paddr) { |
| *paddr = m; |
*paddr = m; |
| so->so_rcv.sb_mb = m->m_next; |
so->so_rcv.sb_mb = m->m_next; |
| Line 936 soreceive(struct socket *so, struct mbuf |
|
| Line 948 soreceive(struct socket *so, struct mbuf |
|
| m = m->m_next; |
m = m->m_next; |
| } else { |
} else { |
| sbfree(&so->so_rcv, m); |
sbfree(&so->so_rcv, m); |
| |
mbuf_removed = 1; |
| if (controlp) { |
if (controlp) { |
| if (pr->pr_domain->dom_externalize && |
if (pr->pr_domain->dom_externalize && |
| mtod(m, struct cmsghdr *)->cmsg_type == |
mtod(m, struct cmsghdr *)->cmsg_type == |
| Line 955 soreceive(struct socket *so, struct mbuf |
|
| Line 968 soreceive(struct socket *so, struct mbuf |
|
| controlp = &(*controlp)->m_next; |
controlp = &(*controlp)->m_next; |
| } |
} |
| } |
} |
| |
|
| |
/* |
| |
* If m is non-NULL, we have some data to read. From now on, |
| |
* make sure to keep sb_lastrecord consistent when working on |
| |
* the last packet on the chain (nextrecord == NULL) and we |
| |
* change m->m_nextpkt. |
| |
*/ |
| if (m) { |
if (m) { |
| if ((flags & MSG_PEEK) == 0) |
if ((flags & MSG_PEEK) == 0) { |
| m->m_nextpkt = nextrecord; |
m->m_nextpkt = nextrecord; |
| |
/* |
| |
* If nextrecord == NULL (this is a single chain), |
| |
* then sb_lastrecord may not be valid here if m |
| |
* was changed earlier. |
| |
*/ |
| |
if (nextrecord == NULL) { |
| |
KASSERT(so->so_rcv.sb_mb == m); |
| |
so->so_rcv.sb_lastrecord = m; |
| |
} |
| |
} |
| type = m->m_type; |
type = m->m_type; |
| if (type == MT_OOBDATA) |
if (type == MT_OOBDATA) |
| flags |= MSG_OOB; |
flags |= MSG_OOB; |
| |
} else { |
| |
if ((flags & MSG_PEEK) == 0) { |
| |
KASSERT(so->so_rcv.sb_mb == m); |
| |
so->so_rcv.sb_mb = nextrecord; |
| |
SB_EMPTY_FIXUP(&so->so_rcv); |
| |
} |
| } |
} |
| |
SBLASTRECORDCHK(&so->so_rcv, "soreceive 2"); |
| |
SBLASTMBUFCHK(&so->so_rcv, "soreceive 2"); |
| |
|
| moff = 0; |
moff = 0; |
| offset = 0; |
offset = 0; |
| while (m && uio->uio_resid > 0 && error == 0) { |
while (m && uio->uio_resid > 0 && error == 0) { |
| Line 989 soreceive(struct socket *so, struct mbuf |
|
| Line 1028 soreceive(struct socket *so, struct mbuf |
|
| * block interrupts again. |
* block interrupts again. |
| */ |
*/ |
| if (mp == 0) { |
if (mp == 0) { |
| |
SBLASTRECORDCHK(&so->so_rcv, "soreceive uiomove"); |
| |
SBLASTMBUFCHK(&so->so_rcv, "soreceive uiomove"); |
| splx(s); |
splx(s); |
| error = uiomove(mtod(m, caddr_t) + moff, (int)len, uio); |
error = uiomove(mtod(m, caddr_t) + moff, (int)len, uio); |
| s = splsoftnet(); |
s = splsoftnet(); |
| if (error) |
if (error) { |
| |
/* |
| |
* If any part of the record has been removed |
| |
* (such as the MT_SONAME mbuf, which will |
| |
* happen when PR_ADDR, and thus also |
| |
* PR_ATOMIC, is set), then drop the entire |
| |
* record to maintain the atomicity of the |
| |
* receive operation. |
| |
* |
| |
* This avoids a later panic("receive 1a") |
| |
* when compiled with DIAGNOSTIC. |
| |
*/ |
| |
if (m && mbuf_removed |
| |
&& (pr->pr_flags & PR_ATOMIC)) |
| |
(void) sbdroprecord(&so->so_rcv); |
| |
|
| goto release; |
goto release; |
| |
} |
| } else |
} else |
| uio->uio_resid -= len; |
uio->uio_resid -= len; |
| if (len == m->m_len - moff) { |
if (len == m->m_len - moff) { |
| Line 1014 soreceive(struct socket *so, struct mbuf |
|
| Line 1071 soreceive(struct socket *so, struct mbuf |
|
| 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 (m) |
/* |
| |
* If m != NULL, we also know that |
| |
* so->so_rcv.sb_mb != NULL. |
| |
*/ |
| |
KASSERT(so->so_rcv.sb_mb == m); |
| |
if (m) { |
| m->m_nextpkt = nextrecord; |
m->m_nextpkt = nextrecord; |
| |
if (nextrecord == NULL) |
| |
so->so_rcv.sb_lastrecord = m; |
| |
} else { |
| |
so->so_rcv.sb_mb = nextrecord; |
| |
SB_EMPTY_FIXUP(&so->so_rcv); |
| |
} |
| |
SBLASTRECORDCHK(&so->so_rcv, "soreceive 3"); |
| |
SBLASTMBUFCHK(&so->so_rcv, "soreceive 3"); |
| } |
} |
| } else { |
} else { |
| if (flags & MSG_PEEK) |
if (flags & MSG_PEEK) |
| Line 1054 soreceive(struct socket *so, struct mbuf |
|
| Line 1124 soreceive(struct socket *so, struct mbuf |
|
| !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; |
| |
/* |
| |
* If we are peeking and the socket receive buffer is |
| |
* full, stop since we can't get more data to peek at. |
| |
*/ |
| |
if ((flags & MSG_PEEK) && sbspace(&so->so_rcv) <= 0) |
| |
break; |
| |
/* |
| |
* If we've drained the socket buffer, tell the |
| |
* protocol in case it needs to do something to |
| |
* get it filled again. |
| |
*/ |
| |
if ((pr->pr_flags & PR_WANTRCVD) && so->so_pcb) |
| |
(*pr->pr_usrreq)(so, PRU_RCVD, |
| |
(struct mbuf *)0, |
| |
(struct mbuf *)(long)flags, |
| |
(struct mbuf *)0, |
| |
(struct proc *)0); |
| |
SBLASTRECORDCHK(&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) { |
if (error) { |
| sbunlock(&so->so_rcv); |
sbunlock(&so->so_rcv); |
| Line 1071 soreceive(struct socket *so, struct mbuf |
|
| Line 1160 soreceive(struct socket *so, struct mbuf |
|
| (void) sbdroprecord(&so->so_rcv); |
(void) sbdroprecord(&so->so_rcv); |
| } |
} |
| if ((flags & MSG_PEEK) == 0) { |
if ((flags & MSG_PEEK) == 0) { |
| if (m == 0) |
if (m == 0) { |
| |
/* |
| |
* First part is an inline SB_EMPTY_FIXUP(). Second |
| |
* part makes sure sb_lastrecord is up-to-date if |
| |
* there is still data in the socket buffer. |
| |
*/ |
| so->so_rcv.sb_mb = nextrecord; |
so->so_rcv.sb_mb = nextrecord; |
| |
if (so->so_rcv.sb_mb == NULL) { |
| |
so->so_rcv.sb_mbtail = NULL; |
| |
so->so_rcv.sb_lastrecord = NULL; |
| |
} else if (nextrecord->m_nextpkt == NULL) |
| |
so->so_rcv.sb_lastrecord = nextrecord; |
| |
} |
| |
SBLASTRECORDCHK(&so->so_rcv, "soreceive 4"); |
| |
SBLASTMBUFCHK(&so->so_rcv, "soreceive 4"); |
| if (pr->pr_flags & PR_WANTRCVD && so->so_pcb) |
if (pr->pr_flags & PR_WANTRCVD && so->so_pcb) |
| (*pr->pr_usrreq)(so, PRU_RCVD, (struct mbuf *)0, |
(*pr->pr_usrreq)(so, PRU_RCVD, (struct mbuf *)0, |
| (struct mbuf *)(long)flags, (struct mbuf *)0, |
(struct mbuf *)(long)flags, (struct mbuf *)0, |