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