diff mbox series

[RFC,v8,17/19] vsock_test: add SOCK_SEQPACKET tests

Message ID 20210413124701.3407363-1-arseny.krasnov@kaspersky.com
State Superseded
Headers show
Series None | expand

Commit Message

Arseny Krasnov April 13, 2021, 12:46 p.m. UTC 
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(-)

Comments

Stefano Garzarella April 21, 2021, 9:35 a.m. UTC | #1 
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

>
Arseny Krasnov April 21, 2021, 3 p.m. UTC | #2 
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 mbox series

Patch

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,
+	},
 	{},
 };