Re: [PATCH 3/6] tcp_splice: Clean up flow control path for splice forwarding

Ah, yes, it looks better now. Three remarks: On Wed, 20 May 2026 23:08:48 +1000 David Gibson wrote:

Splice forwarding can be blocked either waiting for data from one side or waiting for space on the other. For that reason, tcp_splice_sock_handler() on either socket can forward data in either or both directions, depending on whether we have EPOLLIN, EPOLLOUT or both events.

The flow control for this is quite hard to follow though, since we forward in one direction, then sometimes loop back with a goto to do it in the other direction. Simplify this by adding a tcp_splice_forward() function with the logic to forward in one direction and calling it either once or twice from tcp_splice_sock_handler().

Signed-off-by: David Gibson --- tcp_splice.c | 137 ++++++++++++++++++++++++++------------------------- 1 file changed, 71 insertions(+), 66 deletions(-)

diff --git a/tcp_splice.c b/tcp_splice.c index 34ffea73..18e8b303 100644 --- a/tcp_splice.c +++ b/tcp_splice.c @@ -474,67 +474,20 @@ void tcp_splice_conn_from_sock(const struct ctx *c, union flow *flow, int s0) }

/** - * tcp_splice_sock_handler() - Handler for socket mapped to spliced connection + * tcp_splice_forward() - Forward data in one direction using splice() * @c: Execution context - * @ref: epoll reference - * @events: epoll events bitmap + * @conn: Connection to forward data for + * @fromsidei: Side to forward data from * * #syscalls:pasta splice */ -void tcp_splice_sock_handler(struct ctx *c, union epoll_ref ref, - uint32_t events) +static int tcp_splice_forward(struct ctx *c, struct + tcp_splice_conn *conn, unsigned fromsidei)

I think the struct argument should all be on the same line.

{ - struct tcp_splice_conn *conn = conn_at_sidx(ref.flowside); - unsigned evsidei = ref.flowside.sidei, fromsidei; - uint8_t lowat_set_flag, lowat_act_flag; - int eof, never_read; - - assert(conn->f.type == FLOW_TCP_SPLICE); - - if (conn->events == SPLICE_CLOSED) - return; - - if (events & EPOLLERR) { - int err, rc; - socklen_t sl = sizeof(err); - - rc = getsockopt(ref.fd, SOL_SOCKET, SO_ERROR, &err, &sl); - if (rc) - flow_perror(conn, "Error retrieving SO_ERROR"); - else - flow_dbg(conn, "Error event on %s socket: %s", - pif_name(conn->f.pif[evsidei]), - strerror_(err)); - goto reset; - } - - if (conn->events == SPLICE_CONNECT) { - if (!(events & EPOLLOUT)) { - flow_err(conn, "Unexpected events 0x%x during connect", - events); - goto reset; - } - if (tcp_splice_connect_finish(c, conn)) - goto reset; - } - - if (events & EPOLLOUT) { - fromsidei = !evsidei; - conn_event(conn, ~OUT_WAIT(evsidei)); - } else { - fromsidei = evsidei; - } - - if (events & EPOLLRDHUP) - /* For side 0 this is fake, but implied */ - conn_event(conn, FIN_RCVD(evsidei)); - -swap: - eof = 0; - never_read = 1; - - lowat_set_flag = RCVLOWAT_SET(fromsidei); - lowat_act_flag = RCVLOWAT_ACT(fromsidei); + uint8_t lowat_set_flag = RCVLOWAT_SET(fromsidei); + uint8_t lowat_act_flag = RCVLOWAT_ACT(fromsidei); + int never_read = 1; + int eof = 0;

while (1) { ssize_t readlen, written, pending; @@ -551,7 +504,7 @@ retry: if (readlen < 0 && errno != EAGAIN) { flow_perror(conn, "Splicing from %s socket", pif_name(conn->f.pif[fromsidei])); - goto reset; + return -1; }

flow_trace(conn, "%zi from read-side call", readlen); @@ -578,7 +531,7 @@ retry: if (written < 0 && errno != EAGAIN) { flow_perror(conn, "Splicing to %s socket", pif_name(conn->f.pif[!fromsidei])); - goto reset; + return -1; }

flow_trace(conn, "%zi from write-side call (passed %zi)", @@ -639,24 +592,76 @@ retry: if (shutdown(conn->s[!sidei], SHUT_WR) < 0) { flow_perror(conn, "shutdown() on %s", pif_name(conn->f.pif[!sidei])); - goto reset; + return -1; } conn_event(conn, FIN_SENT(!sidei)); } } }

- if (CONN_HAS(conn, FIN_SENT(0) | FIN_SENT(1))) { - /* Clean close, no reset */ - conn_flag(conn, CLOSING); + return 0; +} + +/** + * tcp_splice_sock_handler() - Handler for socket mapped to spliced connection + * @c: Execution context + * @ref: epoll reference + * @events: epoll events bitmap + */ +void tcp_splice_sock_handler(struct ctx *c, union epoll_ref ref, + uint32_t events) +{ + struct tcp_splice_conn *conn = conn_at_sidx(ref.flowside); + unsigned evsidei = ref.flowside.sidei; + + assert(conn->f.type == FLOW_TCP_SPLICE); + + if (conn->events == SPLICE_CLOSED) return; + + if (events & EPOLLERR) { + int err, rc; + socklen_t sl = sizeof(err); + + rc = getsockopt(ref.fd, SOL_SOCKET, SO_ERROR, &err, &sl); + if (rc) + flow_perror(conn, "Error retrieving SO_ERROR"); + else + flow_dbg(conn, "Error event on %s socket: %s", + pif_name(conn->f.pif[evsidei]), + strerror_(err)); + goto reset; + } + + if (conn->events == SPLICE_CONNECT) { + if (!(events & EPOLLOUT)) { + flow_err(conn, "Unexpected events 0x%x during connect", + events); + goto reset; + } + if (tcp_splice_connect_finish(c, conn)) + goto reset; + } + + if (events & EPOLLRDHUP) + /* For side 0 this is fake, but implied */ + conn_event(conn, FIN_RCVD(evsidei));

I saw this all goes away in 5/6, so it wouldn't be relevant. But in case we decide to drop 5/6, here are my remarks on the this. EPOLLRDHUP is now handled before checking the other direction of the connection in case of EPOLLOUT. I think it actually makes more sense this way because we update flags with everything we know until that point, and it shouldn't have a functional effect (the check at the end of the new tcp_splice_forward() is on FIN_RCVD(fromsidei)), but I'm raising that in case the change wasn't intended.

+ + if (events & EPOLLOUT) { + if (tcp_splice_forward(c, conn, !evsidei)) + goto reset; + conn_event(conn, ~OUT_WAIT(evsidei)); }

- if ((events & (EPOLLIN | EPOLLOUT)) == (EPOLLIN | EPOLLOUT)) { - events = EPOLLIN; + if (events & EPOLLIN) { + if (tcp_splice_forward(c, conn, evsidei)) + goto reset;

This should be: goto reset; instead of: goto reset;

+ }

- fromsidei = !fromsidei; - goto swap; + if (CONN_HAS(conn, FIN_SENT(0) | FIN_SENT(1))) { + /* Clean close, no reset */ + conn_flag(conn, CLOSING); + return; }

if (events & EPOLLHUP) {

-- Stefano