1 #include <arpa/inet.h>
2 #include <pthread.h>
3 #include <unistd.h>
4 #include <darwintest.h>
5 #include <TargetConditionals.h>
6
7 volatile static int lock_a;
8 volatile static int lock_b;
9
10 static int fd;
11 static struct sockaddr_in saddr;
12
13 static struct ip_mreq filler_group;
14 static struct ip_mreq group_a;
15 static struct ip_mreq group_b;
16
17 #define ITERATIONS_LIMIT 1000
18
19 static void *
thread_func(__unused void * arg)20 thread_func(__unused void* arg)
21 {
22 lock_a = 1;
23 while (lock_b == 0) {
24 }
25
26 setsockopt(fd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &group_a, sizeof(group_a));
27
28 return NULL;
29 }
30
31 T_DECL(mcast_group_race_82820812, "Race between multicast group join operations.",
32 T_META_ASROOT(true),
33 T_META_ENABLED(!TARGET_OS_BRIDGE && !TARGET_OS_SIMULATOR))
34 {
35 pthread_t th;
36 uint32_t i = 0;
37 uint32_t j = 0;
38
39 saddr.sin_family = AF_INET;
40
41 group_a.imr_multiaddr.s_addr = inet_addr("224.0.0.1");
42 group_b.imr_multiaddr.s_addr = inet_addr("224.0.0.2");
43
44 for (i = 0; i < ITERATIONS_LIMIT; ++i) {
45 T_ASSERT_POSIX_SUCCESS(fd = socket(AF_INET, SOCK_DGRAM, 0), "socket");
46 T_ASSERT_POSIX_SUCCESS(bind(fd, (struct sockaddr *) &saddr, sizeof(saddr)), "bind");
47
48 for (j = 0; j < IP_MIN_MEMBERSHIPS - 1; ++j) {
49 filler_group.imr_multiaddr.s_addr = htonl(ntohl(inet_addr("224.0.0.3")) + j);
50 setsockopt(fd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &filler_group, sizeof(filler_group));
51 }
52
53 T_ASSERT_POSIX_ZERO(pthread_create(&th, NULL, thread_func, NULL), "pthread_create");
54
55 while (lock_a == 0) {
56 }
57 lock_b = 1;
58
59 setsockopt(fd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &group_b, sizeof(group_b));
60
61 T_ASSERT_POSIX_ZERO(pthread_join(th, NULL), "pthread_join");
62 T_ASSERT_POSIX_SUCCESS(close(fd), "close");
63 }
64 }
65