system/linux
system/linux
1
0
Fork 0
mirror of https://github.com/torvalds/linux synced 2024-09-18 18:18:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

vsock/test: rework MSG_PEEK test for SOCK_STREAM

Browse source 
This new version makes test more complicated by adding empty read,
partial read and data comparisons between MSG_PEEK and normal reads.

Signed-off-by: Arseniy Krasnov <AVKrasnov@sberdevices.ru>
Reviewed-by: Stefano Garzarella <sgarzare@redhat.com>
Acked-by: Michael S. Tsirkin <mst@redhat.com>
Signed-off-by: Paolo Abeni <pabeni@redhat.com>
This commit is contained in:
Arseniy Krasnov 2023-07-25 20:29:11 +03:00 committed by Paolo Abeni
parent a75f501de8
commit 587ed79f62
1 changed files with 75 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

78
tools/testing/vsock/vsock_test.c
View file

@ -255,9 +255,14 @@ static void test_stream_multiconn_server(const struct test_opts *opts)
close(fds[i]);
}
#define MSG_PEEK_BUF_LEN 64
static void test_stream_msg_peek_client(const struct test_opts *opts)
{
unsigned char buf[MSG_PEEK_BUF_LEN];
ssize_t send_size;
int fd;
int i;
fd = vsock_stream_connect(opts->peer_cid, 1234);
if (fd < 0) {
@ -265,12 +270,32 @@ static void test_stream_msg_peek_client(const struct test_opts *opts)
exit(EXIT_FAILURE);
}
send_byte(fd, 1, 0);
for (i = 0; i < sizeof(buf); i++)
buf[i] = rand() & 0xFF;
control_expectln("SRVREADY");
send_size = send(fd, buf, sizeof(buf), 0);
if (send_size < 0) {
perror("send");
exit(EXIT_FAILURE);
}
if (send_size != sizeof(buf)) {
fprintf(stderr, "Invalid send size %zi\n", send_size);
exit(EXIT_FAILURE);
}
close(fd);
}
static void test_stream_msg_peek_server(const struct test_opts *opts)
{
unsigned char buf_half[MSG_PEEK_BUF_LEN / 2];
unsigned char buf_normal[MSG_PEEK_BUF_LEN];
unsigned char buf_peek[MSG_PEEK_BUF_LEN];
ssize_t res;
int fd;
fd = vsock_stream_accept(VMADDR_CID_ANY, 1234, NULL);
@ -279,8 +304,55 @@ static void test_stream_msg_peek_server(const struct test_opts *opts)
exit(EXIT_FAILURE);
}
recv_byte(fd, 1, MSG_PEEK);
recv_byte(fd, 1, 0);
/* Peek from empty socket. */
res = recv(fd, buf_peek, sizeof(buf_peek), MSG_PEEK | MSG_DONTWAIT);
if (res != -1) {
fprintf(stderr, "expected recv(2) failure, got %zi\n", res);
exit(EXIT_FAILURE);
}
if (errno != EAGAIN) {
perror("EAGAIN expected");
exit(EXIT_FAILURE);
}
control_writeln("SRVREADY");
/* Peek part of data. */
res = recv(fd, buf_half, sizeof(buf_half), MSG_PEEK);
if (res != sizeof(buf_half)) {
fprintf(stderr, "recv(2) + MSG_PEEK, expected %zu, got %zi\n",
sizeof(buf_half), res);
exit(EXIT_FAILURE);
}
/* Peek whole data. */
res = recv(fd, buf_peek, sizeof(buf_peek), MSG_PEEK);
if (res != sizeof(buf_peek)) {
fprintf(stderr, "recv(2) + MSG_PEEK, expected %zu, got %zi\n",
sizeof(buf_peek), res);
exit(EXIT_FAILURE);
}
/* Compare partial and full peek. */
if (memcmp(buf_half, buf_peek, sizeof(buf_half))) {
fprintf(stderr, "Partial peek data mismatch\n");
exit(EXIT_FAILURE);
}
res = recv(fd, buf_normal, sizeof(buf_normal), 0);
if (res != sizeof(buf_normal)) {
fprintf(stderr, "recv(2), expected %zu, got %zi\n",
sizeof(buf_normal), res);
exit(EXIT_FAILURE);
}
/* Compare full peek and normal read. */
if (memcmp(buf_peek, buf_normal, sizeof(buf_peek))) {
fprintf(stderr, "Full peek data mismatch\n");
exit(EXIT_FAILURE);
}
close(fd);
}