1*e3723e1fSApple OSS Distributions #include <stdlib.h>
2*e3723e1fSApple OSS Distributions
3*e3723e1fSApple OSS Distributions #include <pthread.h>
4*e3723e1fSApple OSS Distributions #include <sys/kern_control.h>
5*e3723e1fSApple OSS Distributions #include <sys/socket.h>
6*e3723e1fSApple OSS Distributions #include <sys/sys_domain.h>
7*e3723e1fSApple OSS Distributions #include <unistd.h>
8*e3723e1fSApple OSS Distributions
9*e3723e1fSApple OSS Distributions #include <darwintest.h>
10*e3723e1fSApple OSS Distributions
11*e3723e1fSApple OSS Distributions /* we should win the race in this window: */
12*e3723e1fSApple OSS Distributions #define NTRIES 200000
13*e3723e1fSApple OSS Distributions
14*e3723e1fSApple OSS Distributions static void *
connect_race(void * data)15*e3723e1fSApple OSS Distributions connect_race(void *data)
16*e3723e1fSApple OSS Distributions {
17*e3723e1fSApple OSS Distributions int *ps = data;
18*e3723e1fSApple OSS Distributions struct sockaddr_ctl sc = {
19*e3723e1fSApple OSS Distributions .sc_id = 1 /* com.apple.flow-divert */
20*e3723e1fSApple OSS Distributions };
21*e3723e1fSApple OSS Distributions int n;
22*e3723e1fSApple OSS Distributions
23*e3723e1fSApple OSS Distributions for (n = 0; n < NTRIES; ++n) {
24*e3723e1fSApple OSS Distributions connect(*ps, (const struct sockaddr *)&sc, sizeof(sc));
25*e3723e1fSApple OSS Distributions }
26*e3723e1fSApple OSS Distributions
27*e3723e1fSApple OSS Distributions return NULL;
28*e3723e1fSApple OSS Distributions }
29*e3723e1fSApple OSS Distributions
30*e3723e1fSApple OSS Distributions T_DECL(flow_div_doubleconnect_55917185, "Bad error path in double-connect for flow_divert_kctl_connect", T_META_TAG_VM_PREFERRED)
31*e3723e1fSApple OSS Distributions {
32*e3723e1fSApple OSS Distributions int s = -1;
33*e3723e1fSApple OSS Distributions int tmp_s;
34*e3723e1fSApple OSS Distributions struct sockaddr_ctl sc = {
35*e3723e1fSApple OSS Distributions .sc_id = 1 /* com.apple.flow-divert */
36*e3723e1fSApple OSS Distributions };
37*e3723e1fSApple OSS Distributions pthread_t t;
38*e3723e1fSApple OSS Distributions int n;
39*e3723e1fSApple OSS Distributions
40*e3723e1fSApple OSS Distributions T_SETUPBEGIN;
41*e3723e1fSApple OSS Distributions T_ASSERT_POSIX_ZERO(pthread_create(&t, NULL, connect_race, &s), NULL);
42*e3723e1fSApple OSS Distributions T_SETUPEND;
43*e3723e1fSApple OSS Distributions
44*e3723e1fSApple OSS Distributions for (n = 0; n < NTRIES; ++n) {
45*e3723e1fSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(tmp_s = socket(AF_SYSTEM, SOCK_DGRAM, SYSPROTO_CONTROL), NULL);
46*e3723e1fSApple OSS Distributions
47*e3723e1fSApple OSS Distributions /*
48*e3723e1fSApple OSS Distributions * this bind will fail, but that's ok because it initialises
49*e3723e1fSApple OSS Distributions * kctl:
50*e3723e1fSApple OSS Distributions */
51*e3723e1fSApple OSS Distributions bind(tmp_s, (const struct sockaddr *)&sc, sizeof(sc));
52*e3723e1fSApple OSS Distributions
53*e3723e1fSApple OSS Distributions /* this is what we're racing the other thread for: */
54*e3723e1fSApple OSS Distributions s = tmp_s;
55*e3723e1fSApple OSS Distributions connect(s, (const struct sockaddr *)&sc, sizeof(sc));
56*e3723e1fSApple OSS Distributions
57*e3723e1fSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(close(s), NULL);
58*e3723e1fSApple OSS Distributions s = -1;
59*e3723e1fSApple OSS Distributions }
60*e3723e1fSApple OSS Distributions
61*e3723e1fSApple OSS Distributions T_ASSERT_POSIX_ZERO(pthread_join(t, NULL), NULL);
62*e3723e1fSApple OSS Distributions T_PASS("flow_divert_kctl_connect race didn't trigger panic");
63*e3723e1fSApple OSS Distributions }
64