• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * This test is based on source contained in the man pages for sendmmsg and
4  * recvmmsg in release 4.15 of the Linux man-pages project.
5  */
6 
7 #define _GNU_SOURCE
8 #include <netinet/ip.h>
9 #include <stdio.h>
10 #include <stdlib.h>
11 #include <string.h>
12 #include <sys/types.h>
13 
14 #include "tst_test.h"
15 #include "lapi/socket.h"
16 #include "tst_safe_macros.h"
17 
18 #include "sendmmsg_var.h"
19 
20 #define BUFSIZE 16
21 #define VLEN 2
22 
23 static int send_sockfd;
24 static int receive_sockfd;
25 static struct mmsghdr *snd_msg, *rcv_msg;
26 static struct iovec *snd1, *snd2, *rcv1, *rcv2;
27 
run(void)28 static void run(void)
29 {
30 	struct timespec timeout;
31 	int retval;
32 
33 	retval = do_sendmmsg(send_sockfd, snd_msg, VLEN, 0);
34 	if (retval < 0 || snd_msg[0].msg_len != 6 || snd_msg[1].msg_len != 6) {
35 		tst_res(TFAIL|TTERRNO, "sendmmsg failed");
36 		return;
37 	}
38 
39 	memset(rcv1->iov_base, 0, rcv1->iov_len);
40 	memset(rcv2->iov_base, 0, rcv2->iov_len);
41 
42 	timeout.tv_sec = 1;
43 	timeout.tv_nsec = 0;
44 
45 	retval = do_recvmmsg(receive_sockfd, rcv_msg, VLEN, 0, &timeout);
46 
47 	if (retval == -1) {
48 		tst_res(TFAIL | TTERRNO, "recvmmsg failed");
49 		return;
50 	}
51 	if (retval != 2) {
52 		tst_res(TFAIL, "Received unexpected number of messages (%d)",
53 			retval);
54 		return;
55 	}
56 
57 	if (memcmp(rcv1->iov_base, "onetwo", 6))
58 		tst_res(TFAIL, "Error in first received message");
59 	else
60 		tst_res(TPASS, "First message received successfully");
61 
62 	if (memcmp(rcv2->iov_base, "three", 5))
63 		tst_res(TFAIL, "Error in second received message");
64 	else
65 		tst_res(TPASS, "Second message received successfully");
66 }
67 
setup(void)68 static void setup(void)
69 {
70 	struct sockaddr_in addr;
71 	unsigned int port = TST_GET_UNUSED_PORT(AF_INET, SOCK_DGRAM);
72 
73 	send_sockfd = SAFE_SOCKET(AF_INET, SOCK_DGRAM, 0);
74 	receive_sockfd = SAFE_SOCKET(AF_INET, SOCK_DGRAM, 0);
75 
76 	addr.sin_family = AF_INET;
77 	addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
78 	addr.sin_port = port;
79 
80 	SAFE_BIND(receive_sockfd, (struct sockaddr *)&addr, sizeof(addr));
81 	SAFE_CONNECT(send_sockfd, (struct sockaddr *)&addr, sizeof(addr));
82 
83 	memcpy(snd1[0].iov_base, "one", snd1[0].iov_len);
84 	memcpy(snd1[1].iov_base, "two", snd1[1].iov_len);
85 	memcpy(snd2->iov_base, "three3", snd2->iov_len);
86 
87 	memset(snd_msg, 0, VLEN * sizeof(*snd_msg));
88 	snd_msg[0].msg_hdr.msg_iov = snd1;
89 	snd_msg[0].msg_hdr.msg_iovlen = 2;
90 	snd_msg[1].msg_hdr.msg_iov = snd2;
91 	snd_msg[1].msg_hdr.msg_iovlen = 1;
92 
93 	memset(rcv_msg, 0, VLEN * sizeof(*rcv_msg));
94 	rcv_msg[0].msg_hdr.msg_iov = rcv1;
95 	rcv_msg[0].msg_hdr.msg_iovlen = 1;
96 	rcv_msg[1].msg_hdr.msg_iov = rcv2;
97 	rcv_msg[1].msg_hdr.msg_iovlen = 1;
98 
99 	test_info();
100 }
101 
cleanup(void)102 static void cleanup(void)
103 {
104 	if (send_sockfd > 0)
105 		SAFE_CLOSE(send_sockfd);
106 	if (receive_sockfd > 0)
107 		SAFE_CLOSE(receive_sockfd);
108 }
109 
110 static struct tst_test test = {
111 	.test_all = run,
112 	.setup = setup,
113 	.cleanup = cleanup,
114 	.test_variants = TEST_VARIANTS,
115 	.bufs = (struct tst_buffers []) {
116 		{&snd1, .iov_sizes = (int[]){3, 3, -1}},
117 		{&snd2, .iov_sizes = (int[]){6, -1}},
118 		{&rcv1, .iov_sizes = (int[]){6, -1}},
119 		{&rcv2, .iov_sizes = (int[]){5, -1}},
120 		{&snd_msg, .size = VLEN * sizeof(*snd_msg)},
121 		{&rcv_msg, .size = VLEN * sizeof(*rcv_msg)},
122 		{},
123 	}
124 };
125