BUG/MEDIUM: mux-h1: Fix freeze when the kernel splicing is used
First of all, we need to be sure to keep the flag H1S_F_BUF_FLUSH on the H1S
reading data until all data was flushed from the buffer. Then we need to know
when the kernel splicing is in use and when it ends. This is handled with the
new flag H1S_F_SPLICED_DATA.
Then, we must subscribe to send when some data remain in the pipe after a
snd_pipe(). It is mandatory to wakeup the stream and avoid a freeze.
Finally, we must be sure to update the message state when we restart to use the
channel's buffer. Among other things, it is mandatory to swith the message from
DATA to DONE state when all data were sent using the kernel splicing.
diff --git a/src/mux_h1.c b/src/mux_h1.c
index 66d12cc..6c0afad 100644
--- a/src/mux_h1.c
+++ b/src/mux_h1.c
@@ -60,6 +60,7 @@
#define H1S_F_WANT_MSK 0x00000070
#define H1S_F_NOT_FIRST 0x00000080 /* The H1 stream is not the first one */
#define H1S_F_BUF_FLUSH 0x00000100 /* Flush input buffer and don't read more data */
+#define H1S_F_SPLICED_DATA 0x00000200 /* Set when the kernel splicing is in used */
/* H1 connection descriptor */
@@ -728,6 +729,9 @@
union h1_sl sl;
int ret = 0;
+ if (!max)
+ goto end;
+
/* Realing input buffer if necessary */
if (b_head(buf) + b_data(buf) > b_wrap(buf))
b_slow_realign(buf, trash.area, 0);
@@ -1064,7 +1068,7 @@
errflag = H1S_F_RES_ERROR;
}
- while (!(h1s->flags & errflag) && count) {
+ do {
if (h1m->state <= H1_MSG_LAST_LF) {
ret = h1_process_headers(h1s, h1m, htx, &h1c->ibuf, &total, count);
if (!ret)
@@ -1088,7 +1092,7 @@
}
count -= ret;
- }
+ } while (!(h1s->flags & errflag) && count);
if (h1s->flags & errflag)
goto parsing_err;
@@ -1323,7 +1327,7 @@
if (!h1_recv_allowed(h1c))
goto end;
- if (h1c->h1s && (h1c->h1s->flags & H1S_F_BUF_FLUSH)) {
+ if (h1c->h1s && (h1c->h1s->flags & (H1S_F_BUF_FLUSH|H1S_F_SPLICED_DATA))) {
rcvd = 1;
goto end;
}
@@ -1702,8 +1706,8 @@
if (flags & CO_RFL_BUF_FLUSH)
h1s->flags |= H1S_F_BUF_FLUSH;
- else if (ret > 0 || (h1s->flags & H1S_F_BUF_FLUSH)) {
- h1s->flags &= ~H1S_F_BUF_FLUSH;
+ else if (ret > 0 || (h1s->flags & H1S_F_SPLICED_DATA)) {
+ h1s->flags &= ~H1S_F_SPLICED_DATA;
if (!(h1c->wait_event.wait_reason & SUB_CAN_RECV))
tasklet_wakeup(h1c->wait_event.task);
}
@@ -1750,8 +1754,13 @@
struct h1m *h1m = (!conn_is_back(cs->conn) ? &h1s->req : &h1s->res);
int ret = 0;
- if (b_data(&h1s->h1c->ibuf))
+ if (b_data(&h1s->h1c->ibuf)) {
+ h1s->flags |= H1S_F_BUF_FLUSH;
goto end;
+ }
+
+ h1s->flags &= ~H1S_F_BUF_FLUSH;
+ h1s->flags |= H1S_F_SPLICED_DATA;
if (h1m->state == H1_MSG_DATA && count > h1m->curr_len)
count = h1m->curr_len;
ret = cs->conn->xprt->rcv_pipe(cs->conn, pipe, count);
@@ -1765,16 +1774,17 @@
static int h1_snd_pipe(struct conn_stream *cs, struct pipe *pipe)
{
struct h1s *h1s = cs->ctx;
- struct h1m *h1m = (!conn_is_back(cs->conn) ? &h1s->res : &h1s->req);
int ret = 0;
if (b_data(&h1s->h1c->obuf))
goto end;
ret = cs->conn->xprt->snd_pipe(cs->conn, pipe);
- if (h1m->state == H1_MSG_DATA && ret > 0)
- h1m->curr_len -= ret;
end:
+ if (pipe->data) {
+ if (!(h1s->h1c->wait_event.wait_reason & SUB_CAN_SEND))
+ cs->conn->xprt->subscribe(cs->conn, SUB_CAN_SEND, &h1s->h1c->wait_event);
+ }
return ret;
}
#endif