| Message ID | 20210413124701.3407363-1-arseny.krasnov@kaspersky.com (mailing list archive) |
|---|---|
| State | New, archived |
| Headers | show |
| Series | virtio/vsock: introduce SOCK_SEQPACKET support | expand |
On Tue, Apr 13, 2021 at 03:46:58PM +0300, Arseny Krasnov wrote: >This adds test of SOCK_SEQPACKET socket: it transfer data and >then tests MSG_TRUNC flag. Cases for connect(), bind(), etc. are >not tested, because it is same as for stream socket. > >Signed-off-by: Arseny Krasnov <arseny.krasnov@kaspersky.com> >--- >v7 -> v8: > - Test for MSG_EOR flags now removed. Why did we remove it? Thanks, Stefano > > tools/testing/vsock/util.c | 32 +++++++++++++--- > tools/testing/vsock/util.h | 3 ++ > tools/testing/vsock/vsock_test.c | 63 ++++++++++++++++++++++++++++++++ > 3 files changed, 93 insertions(+), 5 deletions(-) > >diff --git a/tools/testing/vsock/util.c b/tools/testing/vsock/util.c >index 93cbd6f603f9..2acbb7703c6a 100644 >--- a/tools/testing/vsock/util.c >+++ b/tools/testing/vsock/util.c >@@ -84,7 +84,7 @@ void vsock_wait_remote_close(int fd) > } > > /* Connect to <cid, port> and return the file descriptor. */ >-int vsock_stream_connect(unsigned int cid, unsigned int port) >+static int vsock_connect(unsigned int cid, unsigned int port, int type) > { > union { > struct sockaddr sa; >@@ -101,7 +101,7 @@ int vsock_stream_connect(unsigned int cid, unsigned int port) > > control_expectln("LISTENING"); > >- fd = socket(AF_VSOCK, SOCK_STREAM, 0); >+ fd = socket(AF_VSOCK, type, 0); > > timeout_begin(TIMEOUT); > do { >@@ -120,11 +120,21 @@ int vsock_stream_connect(unsigned int cid, unsigned int port) > return fd; > } > >+int vsock_stream_connect(unsigned int cid, unsigned int port) >+{ >+ return vsock_connect(cid, port, SOCK_STREAM); >+} >+ >+int vsock_seqpacket_connect(unsigned int cid, unsigned int port) >+{ >+ return vsock_connect(cid, port, SOCK_SEQPACKET); >+} >+ > /* Listen on <cid, port> and return the first incoming connection. The remote > * address is stored to clientaddrp. clientaddrp may be NULL. > */ >-int vsock_stream_accept(unsigned int cid, unsigned int port, >- struct sockaddr_vm *clientaddrp) >+static int vsock_accept(unsigned int cid, unsigned int port, >+ struct sockaddr_vm *clientaddrp, int type) > { > union { > struct sockaddr sa; >@@ -145,7 +155,7 @@ int vsock_stream_accept(unsigned int cid, unsigned int port, > int client_fd; > int old_errno; > >- fd = socket(AF_VSOCK, SOCK_STREAM, 0); >+ fd = socket(AF_VSOCK, type, 0); > > if (bind(fd, &addr.sa, sizeof(addr.svm)) < 0) { > perror("bind"); >@@ -189,6 +199,18 @@ int vsock_stream_accept(unsigned int cid, unsigned int port, > return client_fd; > } > >+int vsock_stream_accept(unsigned int cid, unsigned int port, >+ struct sockaddr_vm *clientaddrp) >+{ >+ return vsock_accept(cid, port, clientaddrp, SOCK_STREAM); >+} >+ >+int vsock_seqpacket_accept(unsigned int cid, unsigned int port, >+ struct sockaddr_vm *clientaddrp) >+{ >+ return vsock_accept(cid, port, clientaddrp, SOCK_SEQPACKET); >+} >+ > /* Transmit one byte and check the return value. > * > * expected_ret: >diff --git a/tools/testing/vsock/util.h b/tools/testing/vsock/util.h >index e53dd09d26d9..a3375ad2fb7f 100644 >--- a/tools/testing/vsock/util.h >+++ b/tools/testing/vsock/util.h >@@ -36,8 +36,11 @@ struct test_case { > void init_signals(void); > unsigned int parse_cid(const char *str); > int vsock_stream_connect(unsigned int cid, unsigned int port); >+int vsock_seqpacket_connect(unsigned int cid, unsigned int port); > int vsock_stream_accept(unsigned int cid, unsigned int port, > struct sockaddr_vm *clientaddrp); >+int vsock_seqpacket_accept(unsigned int cid, unsigned int port, >+ struct sockaddr_vm *clientaddrp); > void vsock_wait_remote_close(int fd); > void send_byte(int fd, int expected_ret, int flags); > void recv_byte(int fd, int expected_ret, int flags); >diff --git a/tools/testing/vsock/vsock_test.c b/tools/testing/vsock/vsock_test.c >index 5a4fb80fa832..ffec985fd36f 100644 >--- a/tools/testing/vsock/vsock_test.c >+++ b/tools/testing/vsock/vsock_test.c >@@ -14,6 +14,8 @@ > #include <errno.h> > #include <unistd.h> > #include <linux/kernel.h> >+#include <sys/types.h> >+#include <sys/socket.h> > > #include "timeout.h" > #include "control.h" >@@ -279,6 +281,62 @@ static void test_stream_msg_peek_server(const struct test_opts *opts) > close(fd); > } > >+#define MESSAGE_TRUNC_SZ 32 >+static void test_seqpacket_msg_trunc_client(const struct test_opts *opts) >+{ >+ int fd; >+ char buf[MESSAGE_TRUNC_SZ]; >+ >+ fd = vsock_seqpacket_connect(opts->peer_cid, 1234); >+ if (fd < 0) { >+ perror("connect"); >+ exit(EXIT_FAILURE); >+ } >+ >+ if (send(fd, buf, sizeof(buf), 0) != sizeof(buf)) { >+ perror("send failed"); >+ exit(EXIT_FAILURE); >+ } >+ >+ control_writeln("SENDDONE"); >+ close(fd); >+} >+ >+static void test_seqpacket_msg_trunc_server(const struct test_opts *opts) >+{ >+ int fd; >+ char buf[MESSAGE_TRUNC_SZ / 2]; >+ struct msghdr msg = {0}; >+ struct iovec iov = {0}; >+ >+ fd = vsock_seqpacket_accept(VMADDR_CID_ANY, 1234, NULL); >+ if (fd < 0) { >+ perror("accept"); >+ exit(EXIT_FAILURE); >+ } >+ >+ control_expectln("SENDDONE"); >+ iov.iov_base = buf; >+ iov.iov_len = sizeof(buf); >+ msg.msg_iov = &iov; >+ msg.msg_iovlen = 1; >+ >+ ssize_t ret = recvmsg(fd, &msg, MSG_TRUNC); >+ >+ if (ret != MESSAGE_TRUNC_SZ) { >+ printf("%zi\n", ret); >+ perror("MSG_TRUNC doesn't work"); >+ exit(EXIT_FAILURE); >+ } >+ >+ if (!(msg.msg_flags & MSG_TRUNC)) { >+ fprintf(stderr, "MSG_TRUNC expected\n"); >+ exit(EXIT_FAILURE); >+ } >+ >+ close(fd); >+} >+ > static struct test_case test_cases[] = { > { > .name = "SOCK_STREAM connection reset", >@@ -309,6 +367,11 @@ static struct test_case test_cases[] = { > .run_client = test_stream_msg_peek_client, > .run_server = test_stream_msg_peek_server, > }, >+ { >+ .name = "SOCK_SEQPACKET send data MSG_TRUNC", >+ .run_client = test_seqpacket_msg_trunc_client, >+ .run_server = test_seqpacket_msg_trunc_server, >+ }, > {}, > }; > >-- >2.25.1 >
On 21.04.2021 12:35, Stefano Garzarella wrote: > On Tue, Apr 13, 2021 at 03:46:58PM +0300, Arseny Krasnov wrote: >> This adds test of SOCK_SEQPACKET socket: it transfer data and >> then tests MSG_TRUNC flag. Cases for connect(), bind(), etc. are >> not tested, because it is same as for stream socket. >> >> Signed-off-by: Arseny Krasnov <arseny.krasnov@kaspersky.com> >> --- >> v7 -> v8: >> - Test for MSG_EOR flags now removed. > Why did we remove it? Because MSG_EOR flag support was removed. > > Thanks, > Stefano > >> tools/testing/vsock/util.c | 32 +++++++++++++--- >> tools/testing/vsock/util.h | 3 ++ >> tools/testing/vsock/vsock_test.c | 63 ++++++++++++++++++++++++++++++++ >> 3 files changed, 93 insertions(+), 5 deletions(-) >> >> diff --git a/tools/testing/vsock/util.c b/tools/testing/vsock/util.c >> index 93cbd6f603f9..2acbb7703c6a 100644 >> --- a/tools/testing/vsock/util.c >> +++ b/tools/testing/vsock/util.c >> @@ -84,7 +84,7 @@ void vsock_wait_remote_close(int fd) >> } >> >> /* Connect to <cid, port> and return the file descriptor. */ >> -int vsock_stream_connect(unsigned int cid, unsigned int port) >> +static int vsock_connect(unsigned int cid, unsigned int port, int type) >> { >> union { >> struct sockaddr sa; >> @@ -101,7 +101,7 @@ int vsock_stream_connect(unsigned int cid, unsigned int port) >> >> control_expectln("LISTENING"); >> >> - fd = socket(AF_VSOCK, SOCK_STREAM, 0); >> + fd = socket(AF_VSOCK, type, 0); >> >> timeout_begin(TIMEOUT); >> do { >> @@ -120,11 +120,21 @@ int vsock_stream_connect(unsigned int cid, unsigned int port) >> return fd; >> } >> >> +int vsock_stream_connect(unsigned int cid, unsigned int port) >> +{ >> + return vsock_connect(cid, port, SOCK_STREAM); >> +} >> + >> +int vsock_seqpacket_connect(unsigned int cid, unsigned int port) >> +{ >> + return vsock_connect(cid, port, SOCK_SEQPACKET); >> +} >> + >> /* Listen on <cid, port> and return the first incoming connection. The remote >> * address is stored to clientaddrp. clientaddrp may be NULL. >> */ >> -int vsock_stream_accept(unsigned int cid, unsigned int port, >> - struct sockaddr_vm *clientaddrp) >> +static int vsock_accept(unsigned int cid, unsigned int port, >> + struct sockaddr_vm *clientaddrp, int type) >> { >> union { >> struct sockaddr sa; >> @@ -145,7 +155,7 @@ int vsock_stream_accept(unsigned int cid, unsigned int port, >> int client_fd; >> int old_errno; >> >> - fd = socket(AF_VSOCK, SOCK_STREAM, 0); >> + fd = socket(AF_VSOCK, type, 0); >> >> if (bind(fd, &addr.sa, sizeof(addr.svm)) < 0) { >> perror("bind"); >> @@ -189,6 +199,18 @@ int vsock_stream_accept(unsigned int cid, unsigned int port, >> return client_fd; >> } >> >> +int vsock_stream_accept(unsigned int cid, unsigned int port, >> + struct sockaddr_vm *clientaddrp) >> +{ >> + return vsock_accept(cid, port, clientaddrp, SOCK_STREAM); >> +} >> + >> +int vsock_seqpacket_accept(unsigned int cid, unsigned int port, >> + struct sockaddr_vm *clientaddrp) >> +{ >> + return vsock_accept(cid, port, clientaddrp, SOCK_SEQPACKET); >> +} >> + >> /* Transmit one byte and check the return value. >> * >> * expected_ret: >> diff --git a/tools/testing/vsock/util.h b/tools/testing/vsock/util.h >> index e53dd09d26d9..a3375ad2fb7f 100644 >> --- a/tools/testing/vsock/util.h >> +++ b/tools/testing/vsock/util.h >> @@ -36,8 +36,11 @@ struct test_case { >> void init_signals(void); >> unsigned int parse_cid(const char *str); >> int vsock_stream_connect(unsigned int cid, unsigned int port); >> +int vsock_seqpacket_connect(unsigned int cid, unsigned int port); >> int vsock_stream_accept(unsigned int cid, unsigned int port, >> struct sockaddr_vm *clientaddrp); >> +int vsock_seqpacket_accept(unsigned int cid, unsigned int port, >> + struct sockaddr_vm *clientaddrp); >> void vsock_wait_remote_close(int fd); >> void send_byte(int fd, int expected_ret, int flags); >> void recv_byte(int fd, int expected_ret, int flags); >> diff --git a/tools/testing/vsock/vsock_test.c b/tools/testing/vsock/vsock_test.c >> index 5a4fb80fa832..ffec985fd36f 100644 >> --- a/tools/testing/vsock/vsock_test.c >> +++ b/tools/testing/vsock/vsock_test.c >> @@ -14,6 +14,8 @@ >> #include <errno.h> >> #include <unistd.h> >> #include <linux/kernel.h> >> +#include <sys/types.h> >> +#include <sys/socket.h> >> >> #include "timeout.h" >> #include "control.h" >> @@ -279,6 +281,62 @@ static void test_stream_msg_peek_server(const struct test_opts *opts) >> close(fd); >> } >> >> +#define MESSAGE_TRUNC_SZ 32 >> +static void test_seqpacket_msg_trunc_client(const struct test_opts *opts) >> +{ >> + int fd; >> + char buf[MESSAGE_TRUNC_SZ]; >> + >> + fd = vsock_seqpacket_connect(opts->peer_cid, 1234); >> + if (fd < 0) { >> + perror("connect"); >> + exit(EXIT_FAILURE); >> + } >> + >> + if (send(fd, buf, sizeof(buf), 0) != sizeof(buf)) { >> + perror("send failed"); >> + exit(EXIT_FAILURE); >> + } >> + >> + control_writeln("SENDDONE"); >> + close(fd); >> +} >> + >> +static void test_seqpacket_msg_trunc_server(const struct test_opts *opts) >> +{ >> + int fd; >> + char buf[MESSAGE_TRUNC_SZ / 2]; >> + struct msghdr msg = {0}; >> + struct iovec iov = {0}; >> + >> + fd = vsock_seqpacket_accept(VMADDR_CID_ANY, 1234, NULL); >> + if (fd < 0) { >> + perror("accept"); >> + exit(EXIT_FAILURE); >> + } >> + >> + control_expectln("SENDDONE"); >> + iov.iov_base = buf; >> + iov.iov_len = sizeof(buf); >> + msg.msg_iov = &iov; >> + msg.msg_iovlen = 1; >> + >> + ssize_t ret = recvmsg(fd, &msg, MSG_TRUNC); >> + >> + if (ret != MESSAGE_TRUNC_SZ) { >> + printf("%zi\n", ret); >> + perror("MSG_TRUNC doesn't work"); >> + exit(EXIT_FAILURE); >> + } >> + >> + if (!(msg.msg_flags & MSG_TRUNC)) { >> + fprintf(stderr, "MSG_TRUNC expected\n"); >> + exit(EXIT_FAILURE); >> + } >> + >> + close(fd); >> +} >> + >> static struct test_case test_cases[] = { >> { >> .name = "SOCK_STREAM connection reset", >> @@ -309,6 +367,11 @@ static struct test_case test_cases[] = { >> .run_client = test_stream_msg_peek_client, >> .run_server = test_stream_msg_peek_server, >> }, >> + { >> + .name = "SOCK_SEQPACKET send data MSG_TRUNC", >> + .run_client = test_seqpacket_msg_trunc_client, >> + .run_server = test_seqpacket_msg_trunc_server, >> + }, >> {}, >> }; >> >> -- >> 2.25.1 >> >
diff --git a/tools/testing/vsock/util.c b/tools/testing/vsock/util.c index 93cbd6f603f9..2acbb7703c6a 100644 --- a/tools/testing/vsock/util.c +++ b/tools/testing/vsock/util.c @@ -84,7 +84,7 @@ void vsock_wait_remote_close(int fd) } /* Connect to <cid, port> and return the file descriptor. */ -int vsock_stream_connect(unsigned int cid, unsigned int port) +static int vsock_connect(unsigned int cid, unsigned int port, int type) { union { struct sockaddr sa; @@ -101,7 +101,7 @@ int vsock_stream_connect(unsigned int cid, unsigned int port) control_expectln("LISTENING"); - fd = socket(AF_VSOCK, SOCK_STREAM, 0); + fd = socket(AF_VSOCK, type, 0); timeout_begin(TIMEOUT); do { @@ -120,11 +120,21 @@ int vsock_stream_connect(unsigned int cid, unsigned int port) return fd; } +int vsock_stream_connect(unsigned int cid, unsigned int port) +{ + return vsock_connect(cid, port, SOCK_STREAM); +} + +int vsock_seqpacket_connect(unsigned int cid, unsigned int port) +{ + return vsock_connect(cid, port, SOCK_SEQPACKET); +} + /* Listen on <cid, port> and return the first incoming connection. The remote * address is stored to clientaddrp. clientaddrp may be NULL. */ -int vsock_stream_accept(unsigned int cid, unsigned int port, - struct sockaddr_vm *clientaddrp) +static int vsock_accept(unsigned int cid, unsigned int port, + struct sockaddr_vm *clientaddrp, int type) { union { struct sockaddr sa; @@ -145,7 +155,7 @@ int vsock_stream_accept(unsigned int cid, unsigned int port, int client_fd; int old_errno; - fd = socket(AF_VSOCK, SOCK_STREAM, 0); + fd = socket(AF_VSOCK, type, 0); if (bind(fd, &addr.sa, sizeof(addr.svm)) < 0) { perror("bind"); @@ -189,6 +199,18 @@ int vsock_stream_accept(unsigned int cid, unsigned int port, return client_fd; } +int vsock_stream_accept(unsigned int cid, unsigned int port, + struct sockaddr_vm *clientaddrp) +{ + return vsock_accept(cid, port, clientaddrp, SOCK_STREAM); +} + +int vsock_seqpacket_accept(unsigned int cid, unsigned int port, + struct sockaddr_vm *clientaddrp) +{ + return vsock_accept(cid, port, clientaddrp, SOCK_SEQPACKET); +} + /* Transmit one byte and check the return value. * * expected_ret: diff --git a/tools/testing/vsock/util.h b/tools/testing/vsock/util.h index e53dd09d26d9..a3375ad2fb7f 100644 --- a/tools/testing/vsock/util.h +++ b/tools/testing/vsock/util.h @@ -36,8 +36,11 @@ struct test_case { void init_signals(void); unsigned int parse_cid(const char *str); int vsock_stream_connect(unsigned int cid, unsigned int port); +int vsock_seqpacket_connect(unsigned int cid, unsigned int port); int vsock_stream_accept(unsigned int cid, unsigned int port, struct sockaddr_vm *clientaddrp); +int vsock_seqpacket_accept(unsigned int cid, unsigned int port, + struct sockaddr_vm *clientaddrp); void vsock_wait_remote_close(int fd); void send_byte(int fd, int expected_ret, int flags); void recv_byte(int fd, int expected_ret, int flags); diff --git a/tools/testing/vsock/vsock_test.c b/tools/testing/vsock/vsock_test.c index 5a4fb80fa832..ffec985fd36f 100644 --- a/tools/testing/vsock/vsock_test.c +++ b/tools/testing/vsock/vsock_test.c @@ -14,6 +14,8 @@ #include <errno.h> #include <unistd.h> #include <linux/kernel.h> +#include <sys/types.h> +#include <sys/socket.h> #include "timeout.h" #include "control.h" @@ -279,6 +281,62 @@ static void test_stream_msg_peek_server(const struct test_opts *opts) close(fd); } +#define MESSAGE_TRUNC_SZ 32 +static void test_seqpacket_msg_trunc_client(const struct test_opts *opts) +{ + int fd; + char buf[MESSAGE_TRUNC_SZ]; + + fd = vsock_seqpacket_connect(opts->peer_cid, 1234); + if (fd < 0) { + perror("connect"); + exit(EXIT_FAILURE); + } + + if (send(fd, buf, sizeof(buf), 0) != sizeof(buf)) { + perror("send failed"); + exit(EXIT_FAILURE); + } + + control_writeln("SENDDONE"); + close(fd); +} + +static void test_seqpacket_msg_trunc_server(const struct test_opts *opts) +{ + int fd; + char buf[MESSAGE_TRUNC_SZ / 2]; + struct msghdr msg = {0}; + struct iovec iov = {0}; + + fd = vsock_seqpacket_accept(VMADDR_CID_ANY, 1234, NULL); + if (fd < 0) { + perror("accept"); + exit(EXIT_FAILURE); + } + + control_expectln("SENDDONE"); + iov.iov_base = buf; + iov.iov_len = sizeof(buf); + msg.msg_iov = &iov; + msg.msg_iovlen = 1; + + ssize_t ret = recvmsg(fd, &msg, MSG_TRUNC); + + if (ret != MESSAGE_TRUNC_SZ) { + printf("%zi\n", ret); + perror("MSG_TRUNC doesn't work"); + exit(EXIT_FAILURE); + } + + if (!(msg.msg_flags & MSG_TRUNC)) { + fprintf(stderr, "MSG_TRUNC expected\n"); + exit(EXIT_FAILURE); + } + + close(fd); +} + static struct test_case test_cases[] = { { .name = "SOCK_STREAM connection reset", @@ -309,6 +367,11 @@ static struct test_case test_cases[] = { .run_client = test_stream_msg_peek_client, .run_server = test_stream_msg_peek_server, }, + { + .name = "SOCK_SEQPACKET send data MSG_TRUNC", + .run_client = test_seqpacket_msg_trunc_client, + .run_server = test_seqpacket_msg_trunc_server, + }, {}, };
This adds test of SOCK_SEQPACKET socket: it transfer data and then tests MSG_TRUNC flag. Cases for connect(), bind(), etc. are not tested, because it is same as for stream socket. Signed-off-by: Arseny Krasnov <arseny.krasnov@kaspersky.com> --- v7 -> v8: - Test for MSG_EOR flags now removed. tools/testing/vsock/util.c | 32 +++++++++++++--- tools/testing/vsock/util.h | 3 ++ tools/testing/vsock/vsock_test.c | 63 ++++++++++++++++++++++++++++++++ 3 files changed, 93 insertions(+), 5 deletions(-)