As it was done for SS_CANTSENDMORE bit. The SS_CANTRCVMORE and
SS_RCVATMARK definition kept as is, but now these bits belongs to the
`sb_state' of receive buffer. `sb_state' ored with `so_state' when
socket data exporting to the userland.
Index: sys/kern/kern_sysctl.c
===================================================================
RCS file: /cvs/src/sys/kern/kern_sysctl.c,v
retrieving revision 1.410
diff -u -p -r1.410 kern_sysctl.c
--- sys/kern/kern_sysctl.c 21 Jan 2023 11:23:23 -0000 1.410
+++ sys/kern/kern_sysctl.c 21 Jan 2023 12:55:38 -0000
@@ -1184,7 +1184,8 @@ fill_file(struct kinfo_file *kf, struct
}
kf->so_type = so->so_type;
- kf->so_state = so->so_state | so->so_snd.sb_state;
+ kf->so_state = so->so_state | so->so_snd.sb_state |
+ so->so_rcv.sb_state;
if (show_pointers)
kf->so_pcb = PTRTOINT64(so->so_pcb);
else
Index: sys/kern/sys_socket.c
===================================================================
RCS file: /cvs/src/sys/kern/sys_socket.c,v
retrieving revision 1.59
diff -u -p -r1.59 sys_socket.c
--- sys/kern/sys_socket.c 21 Jan 2023 11:23:23 -0000 1.59
+++ sys/kern/sys_socket.c 21 Jan 2023 12:55:38 -0000
@@ -119,7 +119,7 @@ soo_ioctl(struct file *fp, u_long cmd, c
break;
case SIOCATMARK:
- *(int *)data = (so->so_state&SS_RCVATMARK) != 0;
+ *(int *)data = (so->so_rcv.sb_state & SS_RCVATMARK) != 0;
break;
default:
@@ -149,7 +149,8 @@ soo_stat(struct file *fp, struct stat *u
memset(ub, 0, sizeof (*ub));
ub->st_mode = S_IFSOCK;
solock(so);
- if ((so->so_state & SS_CANTRCVMORE) == 0 || so->so_rcv.sb_cc != 0)
+ if ((so->so_rcv.sb_state & SS_CANTRCVMORE) == 0 ||
+ so->so_rcv.sb_cc != 0)
ub->st_mode |= S_IRUSR | S_IRGRP | S_IROTH;
if ((so->so_snd.sb_state & SS_CANTSENDMORE) == 0)
ub->st_mode |= S_IWUSR | S_IWGRP | S_IWOTH;
Index: sys/kern/uipc_socket.c
===================================================================
RCS file: /cvs/src/sys/kern/uipc_socket.c,v
retrieving revision 1.294
diff -u -p -r1.294 uipc_socket.c
--- sys/kern/uipc_socket.c 21 Jan 2023 11:23:23 -0000 1.294
+++ sys/kern/uipc_socket.c 21 Jan 2023 12:55:38 -0000
@@ -867,7 +867,7 @@ restart:
so->so_error = 0;
goto release;
}
- if (so->so_state & SS_CANTRCVMORE) {
+ if (so->so_rcv.sb_state & SS_CANTRCVMORE) {
if (m)
goto dontblock;
else if (so->so_rcv.sb_cc == 0)
@@ -1023,7 +1023,7 @@ dontblock:
so, so->so_type, m, m->m_type);
#endif
}
- so->so_state &= ~SS_RCVATMARK;
+ so->so_rcv.sb_state &= ~SS_RCVATMARK;
len = uio->uio_resid;
if (so->so_oobmark && len > so->so_oobmark - offset)
len = so->so_oobmark - offset;
@@ -1100,7 +1100,7 @@ dontblock:
if ((flags & MSG_PEEK) == 0) {
so->so_oobmark -= len;
if (so->so_oobmark == 0) {
- so->so_state |= SS_RCVATMARK;
+ so->so_rcv.sb_state |= SS_RCVATMARK;
break;
}
} else {
@@ -1120,7 +1120,8 @@ dontblock:
*/
while (flags & MSG_WAITALL && m == NULL && uio->uio_resid > 0 &&
!sosendallatonce(so) && !nextrecord) {
- if (so->so_error || so->so_state & SS_CANTRCVMORE)
+ if (so->so_rcv.sb_state & SS_CANTRCVMORE ||
+ so->so_error)
break;
SBLASTRECORDCHK(&so->so_rcv, "soreceive sbwait 2");
SBLASTMBUFCHK(&so->so_rcv, "soreceive sbwait 2");
@@ -1160,7 +1161,8 @@ dontblock:
pru_rcvd(so);
}
if (orig_resid == uio->uio_resid && orig_resid &&
- (flags & MSG_EOR) == 0 && (so->so_state & SS_CANTRCVMORE) == 0) {
+ (flags & MSG_EOR) == 0 &&
+ (so->so_rcv.sb_state & SS_CANTRCVMORE) == 0) {
sbunlock(so, &so->so_rcv);
goto restart;
}
@@ -1632,13 +1634,13 @@ somove(struct socket *so, int wait)
pru_rcvd(so);
/* Receive buffer did shrink by len bytes, adjust oob. */
- state = so->so_state;
- so->so_state &= ~SS_RCVATMARK;
+ state = so->so_rcv.sb_state;
+ so->so_rcv.sb_state &= ~SS_RCVATMARK;
oobmark = so->so_oobmark;
so->so_oobmark = oobmark > len ? oobmark - len : 0;
if (oobmark) {
if (oobmark == len)
- so->so_state |= SS_RCVATMARK;
+ so->so_rcv.sb_state |= SS_RCVATMARK;
if (oobmark >= len)
oobmark = 0;
}
@@ -1714,7 +1716,8 @@ somove(struct socket *so, int wait)
error = EFBIG;
if (error)
so->so_error = error;
- if (((so->so_state & SS_CANTRCVMORE) && so->so_rcv.sb_cc == 0) ||
+ if (((so->so_rcv.sb_state & SS_CANTRCVMORE) &&
+ so->so_rcv.sb_cc == 0) ||
(sosp->so_snd.sb_state & SS_CANTSENDMORE) ||
maxreached || error) {
sounsplice(so, sosp, 0);
@@ -1850,7 +1853,7 @@ sosetopt(struct socket *so, int level, i
break;
case SO_RCVBUF:
- if (so->so_state & SS_CANTRCVMORE)
+ if (so->so_rcv.sb_state & SS_CANTRCVMORE)
return (EINVAL);
if (sbcheckreserve(cnt, so->so_rcv.sb_wat) ||
sbreserve(so, &so->so_rcv, cnt))
@@ -2151,7 +2154,7 @@ filt_soread(struct knote *kn, long hint)
rv = 0;
} else
#endif /* SOCKET_SPLICE */
- if (so->so_state & SS_CANTRCVMORE) {
+ if (so->so_rcv.sb_state & SS_CANTRCVMORE) {
kn->kn_flags |= EV_EOF;
if (kn->kn_flags & __EV_POLL) {
if (so->so_state & SS_ISDISCONNECTED)
@@ -2223,7 +2226,7 @@ filt_soexcept(struct knote *kn, long hin
} else
#endif /* SOCKET_SPLICE */
if (kn->kn_sfflags & NOTE_OOB) {
- if (so->so_oobmark || (so->so_state & SS_RCVATMARK)) {
+ if (so->so_oobmark || (so->so_rcv.sb_state & SS_RCVATMARK)) {
kn->kn_fflags |= NOTE_OOB;
kn->kn_data -= so->so_oobmark;
rv = 1;
Index: sys/kern/uipc_socket2.c
===================================================================
RCS file: /cvs/src/sys/kern/uipc_socket2.c,v
retrieving revision 1.132
diff -u -p -r1.132 uipc_socket2.c
--- sys/kern/uipc_socket2.c 21 Jan 2023 11:23:23 -0000 1.132
+++ sys/kern/uipc_socket2.c 21 Jan 2023 12:55:38 -0000
@@ -142,7 +142,8 @@ soisdisconnecting(struct socket *so)
{
soassertlocked(so);
so->so_state &= ~SS_ISCONNECTING;
- so->so_state |= (SS_ISDISCONNECTING|SS_CANTRCVMORE);
+ so->so_state |= SS_ISDISCONNECTING;
+ so->so_rcv.sb_state |= SS_CANTRCVMORE;
so->so_snd.sb_state |= SS_CANTSENDMORE;
wakeup(&so->so_timeo);
sowwakeup(so);
@@ -154,7 +155,8 @@ soisdisconnected(struct socket *so)
{
soassertlocked(so);
so->so_state &= ~(SS_ISCONNECTING|SS_ISCONNECTED|SS_ISDISCONNECTING);
- so->so_state |= (SS_CANTRCVMORE|SS_ISDISCONNECTED);
+ so->so_state |= SS_ISDISCONNECTED;
+ so->so_rcv.sb_state |= SS_CANTRCVMORE;
so->so_snd.sb_state |= SS_CANTSENDMORE;
wakeup(&so->so_timeo);
sowwakeup(so);
@@ -344,7 +346,7 @@ void
socantrcvmore(struct socket *so)
{
soassertlocked(so);
- so->so_state |= SS_CANTRCVMORE;
+ so->so_rcv.sb_state |= SS_CANTRCVMORE;
sorwakeup(so);
}
Index: sys/kern/uipc_syscalls.c
===================================================================
RCS file: /cvs/src/sys/kern/uipc_syscalls.c,v
retrieving revision 1.208
diff -u -p -r1.208 uipc_syscalls.c
--- sys/kern/uipc_syscalls.c 12 Jan 2023 10:59:36 -0000 1.208
+++ sys/kern/uipc_syscalls.c 21 Jan 2023 12:55:39 -0000
@@ -288,14 +288,14 @@ doaccept(struct proc *p, int sock, struc
goto out_unlock;
}
if ((headfp->f_flag & FNONBLOCK) && head->so_qlen == 0) {
- if (head->so_state & SS_CANTRCVMORE)
+ if (head->so_rcv.sb_state & SS_CANTRCVMORE)
error = ECONNABORTED;
else
error = EWOULDBLOCK;
goto out_unlock;
}
while (head->so_qlen == 0 && head->so_error == 0) {
- if (head->so_state & SS_CANTRCVMORE) {
+ if (head->so_rcv.sb_state & SS_CANTRCVMORE) {
head->so_error = ECONNABORTED;
break;
}
Index: sys/miscfs/fifofs/fifo_vnops.c
===================================================================
RCS file: /cvs/src/sys/miscfs/fifofs/fifo_vnops.c,v
retrieving revision 1.99
diff -u -p -r1.99 fifo_vnops.c
--- sys/miscfs/fifofs/fifo_vnops.c 21 Jan 2023 11:23:23 -0000 1.99
+++ sys/miscfs/fifofs/fifo_vnops.c 21 Jan 2023 12:55:39 -0000
@@ -199,7 +199,8 @@ fifo_open(void *v)
}
if (fip->fi_writers == 1) {
solock(rso);
- rso->so_state &= ~(SS_CANTRCVMORE|SS_ISDISCONNECTED);
+ rso->so_state &= ~SS_ISDISCONNECTED;
+ rso->so_rcv.sb_state &= ~SS_CANTRCVMORE;
sounlock(rso);
if (fip->fi_readers > 0)
wakeup(&fip->fi_readers);
@@ -525,7 +526,7 @@ filt_fiforead(struct knote *kn, long hin
soassertlocked(so);
kn->kn_data = so->so_rcv.sb_cc;
- if (so->so_state & SS_CANTRCVMORE) {
+ if (so->so_rcv.sb_state & SS_CANTRCVMORE) {
kn->kn_flags |= EV_EOF;
if (kn->kn_flags & __EV_POLL) {
if (so->so_state & SS_ISDISCONNECTED)
Index: sys/net/rtsock.c
===================================================================
RCS file: /cvs/src/sys/net/rtsock.c,v
retrieving revision 1.358
diff -u -p -r1.358 rtsock.c
--- sys/net/rtsock.c 17 Oct 2022 14:49:02 -0000 1.358
+++ sys/net/rtsock.c 21 Jan 2023 12:55:39 -0000
@@ -464,7 +464,7 @@ rtm_senddesync(struct socket *so)
* timeout(9), otherwise timeout_del_barrier(9) can't help us.
*/
if ((so->so_state & SS_ISCONNECTED) == 0 ||
- (so->so_state & SS_CANTRCVMORE))
+ (so->so_rcv.sb_state & SS_CANTRCVMORE))
return;
/* If we are in a DESYNC state, try to send a RTM_DESYNC packet */
@@ -524,7 +524,7 @@ route_input(struct mbuf *m0, struct sock
*/
if ((so0 == so && !(so0->so_options & SO_USELOOPBACK)) ||
!(so->so_state & SS_ISCONNECTED) ||
- (so->so_state & SS_CANTRCVMORE))
+ (so->so_rcv.sb_state & SS_CANTRCVMORE))
goto next;
/* filter messages that the process does not want */
Index: sys/netinet/raw_ip.c
===================================================================
RCS file: /cvs/src/sys/netinet/raw_ip.c,v
retrieving revision 1.150
diff -u -p -r1.150 raw_ip.c
--- sys/netinet/raw_ip.c 17 Oct 2022 14:49:02 -0000 1.150
+++ sys/netinet/raw_ip.c 21 Jan 2023 12:55:39 -0000
@@ -171,7 +171,7 @@ rip_input(struct mbuf **mp, int *offp, i
rw_enter_write(&rawcbtable.inpt_notify);
mtx_enter(&rawcbtable.inpt_mtx);
TAILQ_FOREACH(inp, &rawcbtable.inpt_queue, inp_queue) {
- if (inp->inp_socket->so_state & SS_CANTRCVMORE)
+ if (inp->inp_socket->so_rcv.sb_state & SS_CANTRCVMORE)
continue;
#ifdef INET6
if (inp->inp_flags & INP_IPV6)
Index: sys/netinet/tcp_input.c
===================================================================
RCS file: /cvs/src/sys/netinet/tcp_input.c,v
retrieving revision 1.385
diff -u -p -r1.385 tcp_input.c
--- sys/netinet/tcp_input.c 12 Jan 2023 13:09:47 -0000 1.385
+++ sys/netinet/tcp_input.c 21 Jan 2023 12:55:39 -0000
@@ -340,7 +340,7 @@ tcp_flush_queue(struct tcpcb *tp)
nq = TAILQ_NEXT(q, tcpqe_q);
TAILQ_REMOVE(&tp->t_segq, q, tcpqe_q);
ND6_HINT(tp);
- if (so->so_state & SS_CANTRCVMORE)
+ if (so->so_rcv.sb_state & SS_CANTRCVMORE)
m_freem(q->tcpqe_m);
else
sbappendstream(so, &so->so_rcv, q->tcpqe_m);
@@ -1041,7 +1041,7 @@ findpcb:
* Drop TCP, IP headers and TCP options then add data
* to socket buffer.
*/
- if (so->so_state & SS_CANTRCVMORE)
+ if (so->so_rcv.sb_state & SS_CANTRCVMORE)
m_freem(m);
else {
if (tp->t_srtt != 0 && tp->rfbuf_ts != 0 &&
@@ -1794,7 +1794,7 @@ trimthenstep6:
* specification, but if we don't get a FIN
* we'll hang forever.
*/
- if (so->so_state & SS_CANTRCVMORE) {
+ if (so->so_rcv.sb_state & SS_CANTRCVMORE) {
tp->t_flags |= TF_BLOCKOUTPUT;
soisdisconnected(so);
tp->t_flags &= ~TF_BLOCKOUTPUT;
@@ -1903,7 +1903,7 @@ step6:
so->so_oobmark = so->so_rcv.sb_cc +
(tp->rcv_up - tp->rcv_nxt) - 1;
if (so->so_oobmark == 0)
- so->so_state |= SS_RCVATMARK;
+ so->so_rcv.sb_state |= SS_RCVATMARK;
sohasoutofband(so);
tp->t_oobflags &= ~(TCPOOB_HAVEDATA | TCPOOB_HADDATA);
}
@@ -1946,7 +1946,7 @@ dodata:
/* XXX */
tiflags = th->th_flags & TH_FIN;
tcpstat_pkt(tcps_rcvpack, tcps_rcvbyte, tlen);
ND6_HINT(tp);
- if (so->so_state & SS_CANTRCVMORE)
+ if (so->so_rcv.sb_state & SS_CANTRCVMORE)
m_freem(m);
else {
m_adj(m, hdroptlen);
Index: sys/netinet/tcp_usrreq.c
===================================================================
RCS file: /cvs/src/sys/netinet/tcp_usrreq.c,v
retrieving revision 1.215
diff -u -p -r1.215 tcp_usrreq.c
--- sys/netinet/tcp_usrreq.c 21 Jan 2023 11:23:24 -0000 1.215
+++ sys/netinet/tcp_usrreq.c 21 Jan 2023 12:55:39 -0000
@@ -918,7 +918,7 @@ tcp_rcvoob(struct socket *so, struct mbu
return (error);
if ((so->so_oobmark == 0 &&
- (so->so_state & SS_RCVATMARK) == 0) ||
+ (so->so_rcv.sb_state & SS_RCVATMARK) == 0) ||
so->so_options & SO_OOBINLINE ||
tp->t_oobflags & TCPOOB_HADDATA) {
error = EINVAL;
Index: sys/netinet/udp_usrreq.c
===================================================================
RCS file: /cvs/src/sys/netinet/udp_usrreq.c,v
retrieving revision 1.304
diff -u -p -r1.304 udp_usrreq.c
--- sys/netinet/udp_usrreq.c 17 Oct 2022 14:49:02 -0000 1.304
+++ sys/netinet/udp_usrreq.c 21 Jan 2023 12:55:39 -0000
@@ -400,7 +400,7 @@ udp_input(struct mbuf **mp, int *offp, i
rw_enter_write(&udbtable.inpt_notify);
mtx_enter(&udbtable.inpt_mtx);
TAILQ_FOREACH(inp, &udbtable.inpt_queue, inp_queue) {
- if (inp->inp_socket->so_state & SS_CANTRCVMORE)
+ if (inp->inp_socket->so_rcv.sb_state & SS_CANTRCVMORE)
continue;
#ifdef INET6
/* don't accept it if AF does not match */
Index: sys/netinet6/raw_ip6.c
===================================================================
RCS file: /cvs/src/sys/netinet6/raw_ip6.c,v
retrieving revision 1.171
diff -u -p -r1.171 raw_ip6.c
--- sys/netinet6/raw_ip6.c 17 Oct 2022 14:49:02 -0000 1.171
+++ sys/netinet6/raw_ip6.c 21 Jan 2023 12:55:39 -0000
@@ -183,7 +183,7 @@ rip6_input(struct mbuf **mp, int *offp,
rw_enter_write(&rawin6pcbtable.inpt_notify);
mtx_enter(&rawin6pcbtable.inpt_mtx);
TAILQ_FOREACH(in6p, &rawin6pcbtable.inpt_queue, inp_queue) {
- if (in6p->inp_socket->so_state & SS_CANTRCVMORE)
+ if (in6p->inp_socket->so_rcv.sb_state & SS_CANTRCVMORE)
continue;
if (rtable_l2(in6p->inp_rtableid) !=
rtable_l2(m->m_pkthdr.ph_rtableid))
Index: sys/sys/socketvar.h
===================================================================
RCS file: /cvs/src/sys/sys/socketvar.h,v
retrieving revision 1.115
diff -u -p -r1.115 socketvar.h
--- sys/sys/socketvar.h 21 Jan 2023 11:23:24 -0000 1.115
+++ sys/sys/socketvar.h 21 Jan 2023 12:55:39 -0000
@@ -147,6 +147,8 @@ struct socket {
* buffer `sb_state' only:
*
* SS_CANTSENDMORE with `so_snd'
+ * SS_CANTRCVMORE with `so_rcv'
+ * SS_RCVATMARK with `so_rcv'
*/
#define SS_NOFDREF 0x001 /* no file table ref any more */
@@ -232,8 +234,8 @@ soreadable(struct socket *so)
soassertlocked(so);
if (isspliced(so))
return 0;
- return (so->so_state & SS_CANTRCVMORE) || so->so_qlen || so->so_error ||
- so->so_rcv.sb_cc >= so->so_rcv.sb_lowat;
+ return (so->so_rcv.sb_state & SS_CANTRCVMORE) || so->so_qlen ||
+ so->so_error || so->so_rcv.sb_cc >= so->so_rcv.sb_lowat;
}
/* can we write something to so? */
