1 /* Check recvmmsg cancellation.
2
3 Copyright (C) 2016-2022 Free Software Foundation, Inc.
4 This file is part of the GNU C Library.
5
6 The GNU C Library is free software; you can redistribute it and/or
7 modify it under the terms of the GNU Lesser General Public
8 License as published by the Free Software Foundation; either
9 version 2.1 of the License, or (at your option) any later version.
10
11 The GNU C Library is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 Lesser General Public License for more details.
15
16 You should have received a copy of the GNU Lesser General Public
17 License along with the GNU C Library; if not, see
18 <https://www.gnu.org/licenses/>. */
19
20 #include <stdio.h>
21 #include <stddef.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <sys/types.h>
25 #include <sys/socket.h>
26 #include <sys/un.h>
27 #include <sys/ipc.h>
28 #include <sys/msg.h>
29 #include <unistd.h>
30 #include <errno.h>
31 #include <pthread.h>
32
33 #include "tst-cancel4-common.h"
34
35 static void *
tf_recvmmsg(void * arg)36 tf_recvmmsg (void *arg)
37 {
38 struct sockaddr_un sun;
39
40 tempfd = socket (AF_UNIX, SOCK_DGRAM, 0);
41 if (tempfd == -1)
42 FAIL_EXIT1 ("socket (AF_UNIX, SOCK_DGRAM, 0): %m");
43
44 int tries = 0;
45 do
46 {
47 if (++tries > 10)
48 FAIL_EXIT1 ("too many unsuccessful bind calls");
49
50 strcpy (sun.sun_path, "/tmp/tst-cancel4-socket-5-XXXXXX");
51 if (mktemp (sun.sun_path) == NULL)
52 FAIL_EXIT1 ("cannot generate temp file name");
53
54 sun.sun_family = AF_UNIX;
55 }
56 while (bind (tempfd, (struct sockaddr *) &sun,
57 offsetof (struct sockaddr_un, sun_path)
58 + strlen (sun.sun_path) + 1) != 0);
59
60 tempfname = strdup (sun.sun_path);
61
62 tempfd2 = socket (AF_UNIX, SOCK_DGRAM, 0);
63 if (tempfd2 == -1)
64 FAIL_EXIT1 ("socket (AF_UNIX, SOCK_DGRAM, 0): %m");
65
66 int r = pthread_barrier_wait (&b2);
67 if (r != 0 && r != PTHREAD_BARRIER_SERIAL_THREAD)
68 FAIL_EXIT1 ("pthread_barrier_wait");
69
70 if (arg != NULL)
71 {
72 r = pthread_barrier_wait (&b2);
73 if (r != 0 && r != PTHREAD_BARRIER_SERIAL_THREAD)
74 FAIL_EXIT1 ("pthread_barrier_wait");
75 }
76
77 pthread_cleanup_push (cl, NULL);
78
79 char mem[70];
80 struct iovec iov[1];
81 iov[0].iov_base = mem;
82 iov[0].iov_len = arg == NULL ? sizeof (mem) : 0;
83
84 struct mmsghdr mm;
85 mm.msg_hdr.msg_name = &sun;
86 mm.msg_hdr.msg_namelen = sizeof (sun);
87 mm.msg_hdr.msg_iov = iov;
88 mm.msg_hdr.msg_iovlen = 1;
89 mm.msg_hdr.msg_control = NULL;
90 mm.msg_hdr.msg_controllen = 0;
91
92 ssize_t ret = recvmmsg (tempfd2, &mm, 1, 0, NULL);
93 if (ret == -1 && errno == ENOSYS)
94 exit (77);
95
96 pthread_cleanup_pop (0);
97
98 FAIL_EXIT1 ("recvmmsg returned");
99 }
100
101 struct cancel_tests tests[] =
102 {
103 ADD_TEST (recvmmsg, 2, 1),
104 };
105 #define ntest_tf (sizeof (tests) / sizeof (tests[0]))
106
107 #include "tst-cancel4-common.c"
108