mirror of
https://passt.top/passt
synced 2024-12-22 21:55:22 +00:00
tcp: Full batched processing for tap messages
Similar to UDP, but using a simple sendmsg() on iovec-style buffers from tap instead, as we don't need to preserve message boundaries. A quick test in PASTA mode, from namespace to init via tap: # ip link set dev pasta0 mtu 16384 # iperf3 -c 192.168.1.222 -t 60 [...] [ ID] Interval Transfer Bitrate [ 5] 0.00-60.00 sec 80.4 GBytes 11.5 Gbits/sec receiver # iperf3 -c 2a02:6d40:3cfc:3a01:2b20:4a6a:c25a:3056 -t 60 [...] [ ID] Interval Transfer Bitrate [ 5] 0.00-60.01 sec 39.9 GBytes 5.71 Gbits/sec receiver # ip link set dev pasta0 mtu 65520 # iperf3 -c 192.168.1.222 -t 60 [...] [ ID] Interval Transfer Bitrate [ 5] 0.00-60.01 sec 88.7 GBytes 12.7 Gbits/sec receiver # iperf3 -c 2a02:6d40:3cfc:3a01:2b20:4a6a:c25a:3056 -t 60 [...] [ ID] Interval Transfer Bitrate [ 5] 0.00-60.00 sec 79.5 GBytes 11.4 Gbits/sec receiver Signed-off-by: Stefano Brivio <sbrivio@redhat.com>
This commit is contained in:
Stefano Brivio
parent
fd5050ccba
commit
dc169643a4
277
tcp.c
277
tcp.c
@ -603,6 +603,9 @@ static struct msghdr tcp6_l2_mh_sock;
|
|||||||
|
|
||||||
static struct mmsghdr tcp_l2_mh_tap [TCP_TAP_FRAMES];
|
static struct mmsghdr tcp_l2_mh_tap [TCP_TAP_FRAMES];
|
||||||
|
|
||||||
|
/* sendmsg() to socket */
|
||||||
|
static struct iovec tcp_tap_iov [TAP_MSGS];
|
||||||
|
|
||||||
/* Bitmap, activity monitoring needed for connection via tap */
|
/* Bitmap, activity monitoring needed for connection via tap */
|
||||||
static uint8_t tcp_act[MAX_TAP_CONNS / 8] = { 0 };
|
static uint8_t tcp_act[MAX_TAP_CONNS / 8] = { 0 };
|
||||||
|
|
||||||
@ -1479,38 +1482,6 @@ static void tcp_splice_destroy(struct ctx *c, struct tcp_splice_conn *conn)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* tcp_send_to_sock() - Send buffer to socket, update timestamp and sequence
|
|
||||||
* @c: Execution context
|
|
||||||
* @conn: Connection pointer
|
|
||||||
* @data: Data buffer
|
|
||||||
* @len: Length at L4
|
|
||||||
* @extra_flags: Additional flags for send(), if any
|
|
||||||
*
|
|
||||||
* Return: negative on socket error with connection reset, 0 otherwise
|
|
||||||
*/
|
|
||||||
static int tcp_send_to_sock(struct ctx *c, struct tcp_tap_conn *conn,
|
|
||||||
char *data, int len, int extra_flags)
|
|
||||||
{
|
|
||||||
int err = send(conn->sock, data, len,
|
|
||||||
MSG_DONTWAIT | MSG_NOSIGNAL | extra_flags);
|
|
||||||
|
|
||||||
if (err < 0) {
|
|
||||||
if (errno == EAGAIN || errno == EWOULDBLOCK) {
|
|
||||||
tcp_send_to_tap(c, conn, ZERO_WINDOW, NULL, 0);
|
|
||||||
return err;
|
|
||||||
}
|
|
||||||
|
|
||||||
err = errno;
|
|
||||||
tcp_rst(c, conn);
|
|
||||||
return -err;
|
|
||||||
}
|
|
||||||
|
|
||||||
conn->seq_from_tap += err;
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* tcp_sock_consume() - Consume (discard) data from buffer, update ACK sequence
|
* tcp_sock_consume() - Consume (discard) data from buffer, update ACK sequence
|
||||||
* @conn: Connection pointer
|
* @conn: Connection pointer
|
||||||
@ -1811,6 +1782,154 @@ out_restore_iov:
|
|||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* tcp_data_from_tap() - tap data in ESTABLISHED{,SOCK_FIN}, CLOSE_WAIT states
|
||||||
|
* @c: Execution context
|
||||||
|
* @conn: Connection pointer
|
||||||
|
* @msg: Array of messages from tap
|
||||||
|
* @count: Count of messages
|
||||||
|
* @now: Current timestamp
|
||||||
|
*/
|
||||||
|
static void tcp_data_from_tap(struct ctx *c, struct tcp_tap_conn *conn,
|
||||||
|
struct tap_msg *msg, int count,
|
||||||
|
struct timespec *now)
|
||||||
|
{
|
||||||
|
struct msghdr mh = { .msg_iov = tcp_tap_iov };
|
||||||
|
uint32_t max_ack_seq = conn->seq_ack_from_tap;
|
||||||
|
uint32_t seq_from_tap = conn->seq_from_tap;
|
||||||
|
int i, iov_i, keep = -1, ack = 0, fin = 0;
|
||||||
|
ssize_t len;
|
||||||
|
|
||||||
|
for (i = 0, iov_i = 0; i < count; i++) {
|
||||||
|
struct tcphdr *th = (struct tcphdr *)msg[i].l4h;
|
||||||
|
uint32_t seq, seq_offset, ack_seq;
|
||||||
|
size_t len = msg[i].l4_len, off;
|
||||||
|
char *data;
|
||||||
|
|
||||||
|
if (len < sizeof(*th)) {
|
||||||
|
tcp_rst(c, conn);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
off = th->doff * 4;
|
||||||
|
if (off < sizeof(*th) || off > len) {
|
||||||
|
tcp_rst(c, conn);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (th->rst) {
|
||||||
|
tcp_tap_destroy(c, conn);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
len -= off;
|
||||||
|
data = (char *)th + off;
|
||||||
|
|
||||||
|
seq = ntohl(th->seq);
|
||||||
|
ack_seq = ntohl(th->ack_seq);
|
||||||
|
|
||||||
|
if (th->ack) {
|
||||||
|
ack = 1;
|
||||||
|
if (ack_seq - conn->seq_ack_from_tap < MAX_WINDOW &&
|
||||||
|
ack_seq - max_ack_seq < MAX_WINDOW)
|
||||||
|
max_ack_seq = ack_seq;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (th->fin)
|
||||||
|
fin = 1;
|
||||||
|
|
||||||
|
seq_offset = seq_from_tap - seq;
|
||||||
|
/* Use data from this buffer only in these two cases:
|
||||||
|
*
|
||||||
|
* , seq_from_tap , seq_from_tap
|
||||||
|
* |--------| <-- len |--------| <-- len
|
||||||
|
* '----' <-- offset ' <-- offset
|
||||||
|
* ^ seq ^ seq
|
||||||
|
*
|
||||||
|
* (offset >= 0, seq + len > seq_from_tap)
|
||||||
|
*
|
||||||
|
* discard in these two cases:
|
||||||
|
* , seq_from_tap , seq_from_tap
|
||||||
|
* |--------| <-- len |--------| <-- len
|
||||||
|
* '--------' <-- offset '-----| <- offset
|
||||||
|
* ^ seq ^ seq
|
||||||
|
* (offset >= 0, seq + len <= seq_from_tap)
|
||||||
|
*
|
||||||
|
* keep, look for another buffer, then go back, in this case:
|
||||||
|
* , seq_from_tap
|
||||||
|
* |--------| <-- len
|
||||||
|
* '===' <-- offset
|
||||||
|
* ^ seq
|
||||||
|
* (offset < 0 i.e. > MAX_WINDOW)
|
||||||
|
*/
|
||||||
|
if (seq_offset < MAX_WINDOW && seq + len <= seq_from_tap)
|
||||||
|
continue;
|
||||||
|
|
||||||
|
if (seq_offset > MAX_WINDOW) {
|
||||||
|
if (keep != -1)
|
||||||
|
keep = i;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
tcp_tap_iov[iov_i].iov_base = data + seq_offset;
|
||||||
|
tcp_tap_iov[iov_i].iov_len = len - seq_offset;
|
||||||
|
seq_from_tap += tcp_tap_iov[iov_i].iov_len;
|
||||||
|
iov_i++;
|
||||||
|
|
||||||
|
if (keep == i) {
|
||||||
|
i = keep + 1;
|
||||||
|
keep = -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (ack) {
|
||||||
|
conn->ts_ack_tap = *now;
|
||||||
|
tcp_sock_consume(conn, max_ack_seq);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!iov_i) {
|
||||||
|
if (keep != -1) {
|
||||||
|
tcp_send_to_tap(c, conn, ACK, NULL, 0);
|
||||||
|
tcp_send_to_tap(c, conn, ACK, NULL, 0);
|
||||||
|
}
|
||||||
|
goto fin;
|
||||||
|
}
|
||||||
|
|
||||||
|
mh.msg_iovlen = iov_i;
|
||||||
|
len = sendmsg(conn->sock, &mh, MSG_DONTWAIT | MSG_NOSIGNAL);
|
||||||
|
if (len < 0) {
|
||||||
|
if (errno == EAGAIN || errno == EWOULDBLOCK) {
|
||||||
|
tcp_send_to_tap(c, conn, ZERO_WINDOW, NULL, 0);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
tcp_rst(c, conn);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (len < (seq_from_tap - conn->seq_from_tap)) {
|
||||||
|
conn->seq_from_tap += len;
|
||||||
|
tcp_send_to_tap(c, conn, ZERO_WINDOW, NULL, 0);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
conn->seq_from_tap += len;
|
||||||
|
tcp_send_to_tap(c, conn, 0, NULL, 0);
|
||||||
|
|
||||||
|
fin:
|
||||||
|
if (conn->state == ESTABLISHED_SOCK_FIN && ack &&
|
||||||
|
!tcp_data_from_sock(c, conn, now))
|
||||||
|
tcp_tap_state(conn, CLOSE_WAIT);
|
||||||
|
|
||||||
|
if (fin) {
|
||||||
|
shutdown(conn->sock, SHUT_WR);
|
||||||
|
if (conn->state == ESTABLISHED)
|
||||||
|
tcp_tap_state(conn, FIN_WAIT_1);
|
||||||
|
else
|
||||||
|
tcp_tap_state(conn, LAST_ACK);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* tcp_tap_handler() - Handle packets from tap and state transitions
|
* tcp_tap_handler() - Handle packets from tap and state transitions
|
||||||
* @c: Execution context
|
* @c: Execution context
|
||||||
@ -1827,16 +1946,11 @@ int tcp_tap_handler(struct ctx *c, int af, void *addr,
|
|||||||
{
|
{
|
||||||
union epoll_ref ref = { .proto = IPPROTO_TCP,
|
union epoll_ref ref = { .proto = IPPROTO_TCP,
|
||||||
.tcp.v6 = ( af == AF_INET6 ) };
|
.tcp.v6 = ( af == AF_INET6 ) };
|
||||||
|
|
||||||
/* TODO: Implement message batching for TCP */
|
|
||||||
struct tcphdr *th = (struct tcphdr *)msg[0].l4h;
|
struct tcphdr *th = (struct tcphdr *)msg[0].l4h;
|
||||||
size_t len = msg[0].l4_len;
|
size_t len = msg[0].l4_len, off;
|
||||||
uint32_t ack_seq;
|
|
||||||
|
|
||||||
struct tcp_tap_conn *conn;
|
struct tcp_tap_conn *conn;
|
||||||
struct epoll_event ev;
|
struct epoll_event ev;
|
||||||
size_t off, skip = 0;
|
int ws;
|
||||||
int ws, i;
|
|
||||||
|
|
||||||
if (len < sizeof(*th))
|
if (len < sizeof(*th))
|
||||||
return 1;
|
return 1;
|
||||||
@ -1852,41 +1966,15 @@ int tcp_tap_handler(struct ctx *c, int af, void *addr,
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* TODO: Partial ACK coalescing, merge with message coalescing */
|
|
||||||
ack_seq = ntohl(th->ack_seq);
|
|
||||||
for (i = 0; conn->state == ESTABLISHED && i < count; i++) {
|
|
||||||
struct tcphdr *__th = (struct tcphdr *)msg[i].l4h;
|
|
||||||
size_t __len = msg[i].l4_len;
|
|
||||||
|
|
||||||
if (__len < sizeof(*th))
|
|
||||||
break;
|
|
||||||
|
|
||||||
off = __th->doff * 4;
|
|
||||||
if (off < sizeof(*th) || off > __len)
|
|
||||||
break;
|
|
||||||
|
|
||||||
if (!th->ack)
|
|
||||||
continue;
|
|
||||||
|
|
||||||
if (ntohl(th->ack_seq) - ack_seq < MAX_WINDOW)
|
|
||||||
ack_seq = ntohl(th->ack_seq);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (th->rst) {
|
if (th->rst) {
|
||||||
tcp_tap_destroy(c, conn);
|
tcp_tap_destroy(c, conn);
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (count == 1)
|
tcp_clamp_window(conn, th, len, th->syn && th->ack);
|
||||||
tcp_clamp_window(conn, th, len, th->syn && th->ack);
|
|
||||||
|
|
||||||
conn->ts_tap = *now;
|
conn->ts_tap = *now;
|
||||||
|
|
||||||
if (ntohl(th->seq) < conn->seq_from_tap &&
|
|
||||||
conn->seq_from_tap - ntohl(th->seq) < MAX_WINDOW) {
|
|
||||||
skip = conn->seq_from_tap - ntohl(th->seq);
|
|
||||||
}
|
|
||||||
|
|
||||||
switch (conn->state) {
|
switch (conn->state) {
|
||||||
case SOCK_SYN_SENT:
|
case SOCK_SYN_SENT:
|
||||||
if (!th->syn || !th->ack) {
|
if (!th->syn || !th->ack) {
|
||||||
@ -1951,62 +2039,9 @@ int tcp_tap_handler(struct ctx *c, int af, void *addr,
|
|||||||
break;
|
break;
|
||||||
case ESTABLISHED:
|
case ESTABLISHED:
|
||||||
case ESTABLISHED_SOCK_FIN:
|
case ESTABLISHED_SOCK_FIN:
|
||||||
conn->ts_ack_tap = *now;
|
|
||||||
|
|
||||||
if (ntohl(th->ack_seq) > conn->seq_to_tap &&
|
|
||||||
(conn->seq_to_tap - ntohl(th->ack_seq)) > MAX_WINDOW) {
|
|
||||||
return count;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (th->ack) {
|
|
||||||
if (count == 1)
|
|
||||||
tcp_sock_consume(conn, ack_seq);
|
|
||||||
|
|
||||||
if (conn->state == ESTABLISHED_SOCK_FIN) {
|
|
||||||
if (!tcp_data_from_sock(c, conn, now))
|
|
||||||
tcp_tap_state(conn, CLOSE_WAIT);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (ntohl(th->seq) > conn->seq_from_tap) {
|
|
||||||
tcp_send_to_tap(c, conn, ACK, NULL, 0);
|
|
||||||
tcp_send_to_tap(c, conn, ACK, NULL, 0);
|
|
||||||
return count;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (skip < len - off &&
|
|
||||||
tcp_send_to_sock(c, conn,
|
|
||||||
msg[0].l4h + off + skip, len - off - skip,
|
|
||||||
(count > 1) ? MSG_MORE : 0))
|
|
||||||
return 1;
|
|
||||||
|
|
||||||
if (count == 1)
|
|
||||||
tcp_send_to_tap(c, conn, ACK, NULL, 0);
|
|
||||||
|
|
||||||
if (th->fin) {
|
|
||||||
shutdown(conn->sock, SHUT_WR);
|
|
||||||
if (conn->state == ESTABLISHED)
|
|
||||||
tcp_tap_state(conn, FIN_WAIT_1);
|
|
||||||
else
|
|
||||||
tcp_tap_state(conn, LAST_ACK);
|
|
||||||
}
|
|
||||||
|
|
||||||
break;
|
|
||||||
case CLOSE_WAIT:
|
case CLOSE_WAIT:
|
||||||
tcp_sock_consume(conn, ntohl(th->ack_seq));
|
tcp_data_from_tap(c, conn, msg, count, now);
|
||||||
|
return count;
|
||||||
if (skip < (len - off) &&
|
|
||||||
tcp_send_to_sock(c, conn,
|
|
||||||
msg[0].l4h + off + skip, len - off - skip,
|
|
||||||
th->psh ? 0 : MSG_MORE))
|
|
||||||
break;
|
|
||||||
|
|
||||||
if (th->fin) {
|
|
||||||
shutdown(conn->sock, SHUT_WR);
|
|
||||||
tcp_tap_state(conn, LAST_ACK);
|
|
||||||
}
|
|
||||||
|
|
||||||
break;
|
|
||||||
case FIN_WAIT_1_SOCK_FIN:
|
case FIN_WAIT_1_SOCK_FIN:
|
||||||
if (th->ack)
|
if (th->ack)
|
||||||
tcp_tap_destroy(c, conn);
|
tcp_tap_destroy(c, conn);
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user