selftests/tls: test for lowat overshoot with multiple records

Set SO_RCVLOWAT and test it gets respected when gathering
data from multiple records.

Signed-off-by: Jakub Kicinski <jakub.kicinski@netronome.com>
Reviewed-by: Dirk van der Merwe <dirk.vandermerwe@netronome.com>
Signed-off-by: David S. Miller <davem@davemloft.net>

diff --git a/tools/testing/selftests/net/tls.c b/tools/testing/selftests/net/tls.c
index 47ddfc154036e358d01f32ee9f46250202c2d703..01efbcd2258c9368ea0d96b4e780fc08c44a067a 100644 (file)
--- a/tools/testing/selftests/net/tls.c
+++ b/tools/testing/selftests/net/tls.c
@@ -575,6 +575,25 @@ TEST_F(tls, recv_peek_large_buf_mult_recs)
        EXPECT_EQ(memcmp(test_str, buf, len), 0);
 }
 
+TEST_F(tls, recv_lowat)
+{
+       char send_mem[10] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };
+       char recv_mem[20];
+       int lowat = 8;
+
+       EXPECT_EQ(send(self->fd, send_mem, 10, 0), 10);
+       EXPECT_EQ(send(self->fd, send_mem, 5, 0), 5);
+
+       memset(recv_mem, 0, 20);
+       EXPECT_EQ(setsockopt(self->cfd, SOL_SOCKET, SO_RCVLOWAT,
+                            &lowat, sizeof(lowat)), 0);
+       EXPECT_EQ(recv(self->cfd, recv_mem, 1, MSG_WAITALL), 1);
+       EXPECT_EQ(recv(self->cfd, recv_mem + 1, 6, MSG_WAITALL), 6);
+       EXPECT_EQ(recv(self->cfd, recv_mem + 7, 10, 0), 8);
+
+       EXPECT_EQ(memcmp(send_mem, recv_mem, 10), 0);
+       EXPECT_EQ(memcmp(send_mem, recv_mem + 10, 5), 0);
+}
 
 TEST_F(tls, pollin)
 {