vsock_test: add SOCK_STREAM MSG_PEEK test

Test if the MSG_PEEK flags of recv(2) works as expected.

Signed-off-by: Stefano Garzarella <sgarzare@redhat.com>
Signed-off-by: David S. Miller <davem@davemloft.net>
This commit is contained in:
Stefano Garzarella 2019-12-18 19:07:08 +01:00 committed by David S. Miller
parent 8d00b93fec
commit d6269a93ff

View file

@ -178,6 +178,35 @@ static void test_stream_multiconn_server(const struct test_opts *opts)
close(fds[i]);
}
static void test_stream_msg_peek_client(const struct test_opts *opts)
{
int fd;
fd = vsock_stream_connect(opts->peer_cid, 1234);
if (fd < 0) {
perror("connect");
exit(EXIT_FAILURE);
}
send_byte(fd, 1, 0);
close(fd);
}
static void test_stream_msg_peek_server(const struct test_opts *opts)
{
int fd;
fd = vsock_stream_accept(VMADDR_CID_ANY, 1234, NULL);
if (fd < 0) {
perror("accept");
exit(EXIT_FAILURE);
}
recv_byte(fd, 1, MSG_PEEK);
recv_byte(fd, 1, 0);
close(fd);
}
static struct test_case test_cases[] = {
{
.name = "SOCK_STREAM connection reset",
@ -198,6 +227,11 @@ static struct test_case test_cases[] = {
.run_client = test_stream_multiconn_client,
.run_server = test_stream_multiconn_server,
},
{
.name = "SOCK_STREAM MSG_PEEK",
.run_client = test_stream_msg_peek_client,
.run_server = test_stream_msg_peek_server,
},
{},
};