1*d4514f0bSApple OSS Distributions /*
2*d4514f0bSApple OSS Distributions * Copyright (c) 2022 Apple Inc. All rights reserved.
3*d4514f0bSApple OSS Distributions */
4*d4514f0bSApple OSS Distributions
5*d4514f0bSApple OSS Distributions #define PRIVATE 1 /* Needed for some F_OFD_* definitions */
6*d4514f0bSApple OSS Distributions #include <fcntl.h>
7*d4514f0bSApple OSS Distributions #include <sys/fcntl.h>
8*d4514f0bSApple OSS Distributions #include <errno.h>
9*d4514f0bSApple OSS Distributions #include <darwintest.h>
10*d4514f0bSApple OSS Distributions #include <darwintest_utils.h>
11*d4514f0bSApple OSS Distributions #include <stdatomic.h>
12*d4514f0bSApple OSS Distributions
13*d4514f0bSApple OSS Distributions #ifndef O_CLOFORK
14*d4514f0bSApple OSS Distributions #define O_CLOFORK 0x08000000
15*d4514f0bSApple OSS Distributions #endif
16*d4514f0bSApple OSS Distributions
17*d4514f0bSApple OSS Distributions T_GLOBAL_META(
18*d4514f0bSApple OSS Distributions T_META_NAMESPACE("xnu.locks"),
19*d4514f0bSApple OSS Distributions T_META_RADAR_COMPONENT_NAME("xnu"),
20*d4514f0bSApple OSS Distributions T_META_RADAR_COMPONENT_VERSION("locks"),
21*d4514f0bSApple OSS Distributions T_META_OWNER("tim_marsland"), // with thanks to Peter Rutenbar
22*d4514f0bSApple OSS Distributions T_META_RUN_CONCURRENTLY(TRUE));
23*d4514f0bSApple OSS Distributions
24*d4514f0bSApple OSS Distributions enum lock_flags {
25*d4514f0bSApple OSS Distributions EXCL = 1,
26*d4514f0bSApple OSS Distributions WAIT = 2,
27*d4514f0bSApple OSS Distributions UNLOCK = 4,
28*d4514f0bSApple OSS Distributions };
29*d4514f0bSApple OSS Distributions
30*d4514f0bSApple OSS Distributions enum lock_type {
31*d4514f0bSApple OSS Distributions TYPE_FLOCK = 0,
32*d4514f0bSApple OSS Distributions TYPE_POSIX = 1,
33*d4514f0bSApple OSS Distributions TYPE_OFD = 2,
34*d4514f0bSApple OSS Distributions };
35*d4514f0bSApple OSS Distributions
36*d4514f0bSApple OSS Distributions static int
ofd_get(struct flock * fl,int fd,off_t start,off_t len,pid_t pid,uint32_t flags)37*d4514f0bSApple OSS Distributions ofd_get(struct flock *fl,
38*d4514f0bSApple OSS Distributions int fd, off_t start, off_t len, pid_t pid, uint32_t flags)
39*d4514f0bSApple OSS Distributions {
40*d4514f0bSApple OSS Distributions fl->l_start = start;
41*d4514f0bSApple OSS Distributions fl->l_len = len;
42*d4514f0bSApple OSS Distributions fl->l_pid = pid;
43*d4514f0bSApple OSS Distributions fl->l_type = (flags & EXCL) ? F_WRLCK : F_RDLCK;
44*d4514f0bSApple OSS Distributions fl->l_whence = SEEK_SET;
45*d4514f0bSApple OSS Distributions return fcntl(fd, (pid == -1) ? F_OFD_GETLK : F_OFD_GETLKPID, fl);
46*d4514f0bSApple OSS Distributions }
47*d4514f0bSApple OSS Distributions
48*d4514f0bSApple OSS Distributions static int
posix_get(struct flock * fl,int fd,off_t start,off_t len,pid_t pid,uint32_t flags)49*d4514f0bSApple OSS Distributions posix_get(struct flock *fl,
50*d4514f0bSApple OSS Distributions int fd, off_t start, off_t len, pid_t pid, uint32_t flags)
51*d4514f0bSApple OSS Distributions {
52*d4514f0bSApple OSS Distributions fl->l_start = start;
53*d4514f0bSApple OSS Distributions fl->l_len = len;
54*d4514f0bSApple OSS Distributions fl->l_pid = pid;
55*d4514f0bSApple OSS Distributions fl->l_type = (flags & EXCL) ? F_WRLCK : F_RDLCK;
56*d4514f0bSApple OSS Distributions fl->l_whence = SEEK_SET;
57*d4514f0bSApple OSS Distributions return fcntl(fd, (pid == -1) ? F_GETLK : F_GETLKPID, fl);
58*d4514f0bSApple OSS Distributions }
59*d4514f0bSApple OSS Distributions
60*d4514f0bSApple OSS Distributions static int
posix_lock(int fd,off_t start,off_t len,uint32_t flags)61*d4514f0bSApple OSS Distributions posix_lock(int fd, off_t start, off_t len, uint32_t flags)
62*d4514f0bSApple OSS Distributions {
63*d4514f0bSApple OSS Distributions struct flock fl = {
64*d4514f0bSApple OSS Distributions .l_start = start,
65*d4514f0bSApple OSS Distributions .l_len = len,
66*d4514f0bSApple OSS Distributions .l_pid = -1,
67*d4514f0bSApple OSS Distributions .l_type = (flags & EXCL) ? F_WRLCK : F_RDLCK,
68*d4514f0bSApple OSS Distributions .l_whence = SEEK_SET,
69*d4514f0bSApple OSS Distributions };
70*d4514f0bSApple OSS Distributions return fcntl(fd, (flags & WAIT) ? F_SETLKW : F_SETLK, &fl);
71*d4514f0bSApple OSS Distributions }
72*d4514f0bSApple OSS Distributions
73*d4514f0bSApple OSS Distributions static int
ofd_lock(int fd,off_t start,off_t len,uint32_t flags)74*d4514f0bSApple OSS Distributions ofd_lock(int fd, off_t start, off_t len, uint32_t flags)
75*d4514f0bSApple OSS Distributions {
76*d4514f0bSApple OSS Distributions struct flock fl = {
77*d4514f0bSApple OSS Distributions .l_start = start,
78*d4514f0bSApple OSS Distributions .l_len = len,
79*d4514f0bSApple OSS Distributions .l_pid = -1,
80*d4514f0bSApple OSS Distributions .l_type = (flags & EXCL) ? F_WRLCK : F_RDLCK,
81*d4514f0bSApple OSS Distributions .l_whence = SEEK_SET,
82*d4514f0bSApple OSS Distributions };
83*d4514f0bSApple OSS Distributions return fcntl(fd, (flags & WAIT) ? F_OFD_SETLKW : F_OFD_SETLK, &fl);
84*d4514f0bSApple OSS Distributions }
85*d4514f0bSApple OSS Distributions
86*d4514f0bSApple OSS Distributions static int
posix_unlock(int fd,off_t start,off_t len)87*d4514f0bSApple OSS Distributions posix_unlock(int fd, off_t start, off_t len)
88*d4514f0bSApple OSS Distributions {
89*d4514f0bSApple OSS Distributions struct flock fl = {
90*d4514f0bSApple OSS Distributions .l_start = start,
91*d4514f0bSApple OSS Distributions .l_len = len,
92*d4514f0bSApple OSS Distributions .l_pid = -1,
93*d4514f0bSApple OSS Distributions .l_type = F_UNLCK,
94*d4514f0bSApple OSS Distributions .l_whence = SEEK_SET,
95*d4514f0bSApple OSS Distributions };
96*d4514f0bSApple OSS Distributions return fcntl(fd, F_SETLK, &fl);
97*d4514f0bSApple OSS Distributions }
98*d4514f0bSApple OSS Distributions
99*d4514f0bSApple OSS Distributions static int
ofd_unlock(int fd,off_t start,off_t len)100*d4514f0bSApple OSS Distributions ofd_unlock(int fd, off_t start, off_t len)
101*d4514f0bSApple OSS Distributions {
102*d4514f0bSApple OSS Distributions struct flock fl = {
103*d4514f0bSApple OSS Distributions .l_start = start,
104*d4514f0bSApple OSS Distributions .l_len = len,
105*d4514f0bSApple OSS Distributions .l_pid = -1,
106*d4514f0bSApple OSS Distributions .l_type = F_UNLCK,
107*d4514f0bSApple OSS Distributions .l_whence = SEEK_SET,
108*d4514f0bSApple OSS Distributions };
109*d4514f0bSApple OSS Distributions return fcntl(fd, F_OFD_SETLK, &fl);
110*d4514f0bSApple OSS Distributions }
111*d4514f0bSApple OSS Distributions
112*d4514f0bSApple OSS Distributions /* Is the given flock equal to the given arguments */
113*d4514f0bSApple OSS Distributions static bool
flequal(struct flock * fl,off_t start,off_t len,pid_t pid,int flags)114*d4514f0bSApple OSS Distributions flequal(struct flock *fl, off_t start, off_t len, pid_t pid, int flags)
115*d4514f0bSApple OSS Distributions {
116*d4514f0bSApple OSS Distributions if (start == fl->l_start && len == fl->l_len && pid == fl->l_pid) {
117*d4514f0bSApple OSS Distributions if (flags == EXCL && fl->l_type == F_WRLCK) {
118*d4514f0bSApple OSS Distributions return true;
119*d4514f0bSApple OSS Distributions } else if (flags == UNLOCK && fl->l_type == F_UNLCK) {
120*d4514f0bSApple OSS Distributions return true;
121*d4514f0bSApple OSS Distributions } else if (flags == 0 && fl->l_type == F_RDLCK) {
122*d4514f0bSApple OSS Distributions return true;
123*d4514f0bSApple OSS Distributions }
124*d4514f0bSApple OSS Distributions }
125*d4514f0bSApple OSS Distributions T_LOG("flequal: %lld %lld %d %x %x\n",
126*d4514f0bSApple OSS Distributions fl->l_start, fl->l_len, fl->l_pid, fl->l_type, fl->l_whence);
127*d4514f0bSApple OSS Distributions return false;
128*d4514f0bSApple OSS Distributions }
129*d4514f0bSApple OSS Distributions
130*d4514f0bSApple OSS Distributions typedef struct {
131*d4514f0bSApple OSS Distributions pthread_t thread;
132*d4514f0bSApple OSS Distributions int fd, err;
133*d4514f0bSApple OSS Distributions bool complete;
134*d4514f0bSApple OSS Distributions enum lock_type lock_type;
135*d4514f0bSApple OSS Distributions uint32_t flags;
136*d4514f0bSApple OSS Distributions off_t start, end;
137*d4514f0bSApple OSS Distributions } lock_thread_state_t;
138*d4514f0bSApple OSS Distributions
139*d4514f0bSApple OSS Distributions static void *
lock_thread(void * arg)140*d4514f0bSApple OSS Distributions lock_thread(
141*d4514f0bSApple OSS Distributions void *arg)
142*d4514f0bSApple OSS Distributions {
143*d4514f0bSApple OSS Distributions lock_thread_state_t *lts = (lock_thread_state_t *)arg;
144*d4514f0bSApple OSS Distributions
145*d4514f0bSApple OSS Distributions switch (lts->lock_type) {
146*d4514f0bSApple OSS Distributions case TYPE_FLOCK: {
147*d4514f0bSApple OSS Distributions int op = (lts->flags & EXCL) ? LOCK_EX : LOCK_SH;
148*d4514f0bSApple OSS Distributions op |= (lts->flags & WAIT) ? 0 : LOCK_NB;
149*d4514f0bSApple OSS Distributions lts->err = flock(lts->fd, op) ? errno : 0;
150*d4514f0bSApple OSS Distributions break;
151*d4514f0bSApple OSS Distributions }
152*d4514f0bSApple OSS Distributions case TYPE_POSIX:
153*d4514f0bSApple OSS Distributions lts->err = posix_lock(lts->fd,
154*d4514f0bSApple OSS Distributions lts->start, lts->end, lts->flags) ? errno : 0;
155*d4514f0bSApple OSS Distributions break;
156*d4514f0bSApple OSS Distributions case TYPE_OFD:
157*d4514f0bSApple OSS Distributions lts->err = ofd_lock(lts->fd,
158*d4514f0bSApple OSS Distributions lts->start, lts->end, lts->flags) ? errno : 0;
159*d4514f0bSApple OSS Distributions break;
160*d4514f0bSApple OSS Distributions }
161*d4514f0bSApple OSS Distributions
162*d4514f0bSApple OSS Distributions atomic_thread_fence(memory_order_acquire);
163*d4514f0bSApple OSS Distributions lts->complete = true;
164*d4514f0bSApple OSS Distributions atomic_thread_fence(memory_order_release);
165*d4514f0bSApple OSS Distributions
166*d4514f0bSApple OSS Distributions return NULL;
167*d4514f0bSApple OSS Distributions }
168*d4514f0bSApple OSS Distributions
169*d4514f0bSApple OSS Distributions static bool
has_completed(lock_thread_state_t * lts)170*d4514f0bSApple OSS Distributions has_completed(lock_thread_state_t *lts)
171*d4514f0bSApple OSS Distributions {
172*d4514f0bSApple OSS Distributions atomic_thread_fence(memory_order_acquire);
173*d4514f0bSApple OSS Distributions const bool r = lts->complete;
174*d4514f0bSApple OSS Distributions atomic_thread_fence(memory_order_release);
175*d4514f0bSApple OSS Distributions return r;
176*d4514f0bSApple OSS Distributions }
177*d4514f0bSApple OSS Distributions
178*d4514f0bSApple OSS Distributions static void
start_lock_thread(enum lock_type type,lock_thread_state_t * lts,int fd,off_t start,off_t end,uint32_t flags)179*d4514f0bSApple OSS Distributions start_lock_thread(enum lock_type type, lock_thread_state_t *lts,
180*d4514f0bSApple OSS Distributions int fd, off_t start, off_t end, uint32_t flags)
181*d4514f0bSApple OSS Distributions {
182*d4514f0bSApple OSS Distributions lts->fd = fd;
183*d4514f0bSApple OSS Distributions lts->err = 0;
184*d4514f0bSApple OSS Distributions lts->complete = false;
185*d4514f0bSApple OSS Distributions lts->lock_type = type;
186*d4514f0bSApple OSS Distributions lts->flags = flags;
187*d4514f0bSApple OSS Distributions lts->start = start;
188*d4514f0bSApple OSS Distributions lts->end = end;
189*d4514f0bSApple OSS Distributions
190*d4514f0bSApple OSS Distributions pthread_create(<s->thread, NULL, lock_thread, lts);
191*d4514f0bSApple OSS Distributions }
192*d4514f0bSApple OSS Distributions
193*d4514f0bSApple OSS Distributions static void
random_pause(void)194*d4514f0bSApple OSS Distributions random_pause(void)
195*d4514f0bSApple OSS Distributions {
196*d4514f0bSApple OSS Distributions const useconds_t usec = rand() & (16384 - 1);
197*d4514f0bSApple OSS Distributions usleep(usec + 1);
198*d4514f0bSApple OSS Distributions }
199*d4514f0bSApple OSS Distributions
200*d4514f0bSApple OSS Distributions #define GET_CHECK( \
201*d4514f0bSApple OSS Distributions str, fd, \
202*d4514f0bSApple OSS Distributions get_start, get_len, get_flags, get_type, \
203*d4514f0bSApple OSS Distributions chk_start, chk_len, chk_flags, chk_pid) \
204*d4514f0bSApple OSS Distributions do { \
205*d4514f0bSApple OSS Distributions struct flock _fl; \
206*d4514f0bSApple OSS Distributions if (get_type == TYPE_OFD) { \
207*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(ofd_get(&_fl, fd, get_start, get_len, -1, get_flags), str " (ofd_get)"); \
208*d4514f0bSApple OSS Distributions } else { \
209*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(posix_get(&_fl, fd, get_start, get_len, -1, get_flags), str " (posix_get)"); \
210*d4514f0bSApple OSS Distributions } \
211*d4514f0bSApple OSS Distributions T_ASSERT_TRUE(flequal(&_fl, chk_start, chk_len, chk_pid, chk_flags), str " (flequal)"); \
212*d4514f0bSApple OSS Distributions } while (0)
213*d4514f0bSApple OSS Distributions
214*d4514f0bSApple OSS Distributions #define LOCK_AND_CHECK( \
215*d4514f0bSApple OSS Distributions str, fd, \
216*d4514f0bSApple OSS Distributions lck_start, lck_len, lck_flags, lock_type, \
217*d4514f0bSApple OSS Distributions get_start, get_len, get_flags, get_type, \
218*d4514f0bSApple OSS Distributions chk_start, chk_len, chk_flags, chk_pid) \
219*d4514f0bSApple OSS Distributions do { \
220*d4514f0bSApple OSS Distributions if (lock_type == TYPE_OFD) { \
221*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(ofd_lock(fd, lck_start, lck_len, lck_flags), str " (ofd_lock)"); \
222*d4514f0bSApple OSS Distributions } else { \
223*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(posix_lock(fd, lck_start, lck_len, lck_flags), str " (posix_lock)"); \
224*d4514f0bSApple OSS Distributions } \
225*d4514f0bSApple OSS Distributions GET_CHECK(str, fd, get_start, get_len, get_flags, get_type, chk_start, chk_len, chk_flags, chk_pid); \
226*d4514f0bSApple OSS Distributions } while (0)
227*d4514f0bSApple OSS Distributions
228*d4514f0bSApple OSS Distributions #define UNLOCK_AND_CHECK( \
229*d4514f0bSApple OSS Distributions str, fd, \
230*d4514f0bSApple OSS Distributions lck_start, lck_len, unlock_type, \
231*d4514f0bSApple OSS Distributions get_start, get_len, get_flags, get_type, \
232*d4514f0bSApple OSS Distributions chk_start, chk_len, chk_flags, chk_pid) \
233*d4514f0bSApple OSS Distributions do { \
234*d4514f0bSApple OSS Distributions if (unlock_type == TYPE_OFD) { \
235*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(ofd_unlock(fd, lck_start, lck_len), str " (ofd_unlock)"); \
236*d4514f0bSApple OSS Distributions } else { \
237*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(posix_unlock(fd, lck_start, lck_len), str " (posix_unlock)"); \
238*d4514f0bSApple OSS Distributions } \
239*d4514f0bSApple OSS Distributions GET_CHECK(str, fd, get_start, get_len, get_flags, get_type, chk_start, chk_len, chk_flags, chk_pid); \
240*d4514f0bSApple OSS Distributions } while (0)
241*d4514f0bSApple OSS Distributions
242*d4514f0bSApple OSS Distributions #define A_PATH "basic_lockf_a"
243*d4514f0bSApple OSS Distributions #define B_PATH "basic_lockf_b"
244*d4514f0bSApple OSS Distributions
245*d4514f0bSApple OSS Distributions T_DECL(lockf_basic,
246*d4514f0bSApple OSS Distributions "Basic test of flock/POSIX/OFD advisory file locks",
247*d4514f0bSApple OSS Distributions T_META_CHECK_LEAKS(false), T_META_TAG_VM_PREFERRED)
248*d4514f0bSApple OSS Distributions {
249*d4514f0bSApple OSS Distributions const char *tmpdir = dt_tmpdir();
250*d4514f0bSApple OSS Distributions lock_thread_state_t lts[4];
251*d4514f0bSApple OSS Distributions pid_t pid = getpid();
252*d4514f0bSApple OSS Distributions int a, a_confined, b, b_confined;
253*d4514f0bSApple OSS Distributions const off_t file_len = 0x10000;
254*d4514f0bSApple OSS Distributions T_SETUPBEGIN;
255*d4514f0bSApple OSS Distributions
256*d4514f0bSApple OSS Distributions /* random sleeping to hunt for races */
257*d4514f0bSApple OSS Distributions
258*d4514f0bSApple OSS Distributions unsigned seed = (unsigned)pid;
259*d4514f0bSApple OSS Distributions const char *p = getenv("LOCKF_BASIC_SRAND_SEED");
260*d4514f0bSApple OSS Distributions if (p) {
261*d4514f0bSApple OSS Distributions seed = (unsigned)atol(p);
262*d4514f0bSApple OSS Distributions }
263*d4514f0bSApple OSS Distributions srand(seed);
264*d4514f0bSApple OSS Distributions
265*d4514f0bSApple OSS Distributions /* Create two test files, a and b */
266*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(chdir(tmpdir), "chdir(%s)", tmpdir);
267*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS((a = open(A_PATH, O_CREAT | O_RDWR, 0666)), "open(a)");
268*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS((b = open(B_PATH, O_CREAT | O_RDWR, 0666)), "open(b)");
269*d4514f0bSApple OSS Distributions
270*d4514f0bSApple OSS Distributions /* Give both files 64KiB */
271*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(ftruncate(a, file_len), "truncate a");
272*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(ftruncate(b, 0x10000), "truncate b");
273*d4514f0bSApple OSS Distributions
274*d4514f0bSApple OSS Distributions /* Open a/b again, but CONFINED this time */
275*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS((a_confined = open(A_PATH, O_CLOFORK | O_RDWR)), "open(a, O_CLOFORK)");
276*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(fcntl(a_confined, F_SETCONFINED, 1), "F_SETCONFINED");
277*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS((b_confined = open(B_PATH, O_CLOFORK | O_RDWR)), "open(b, O_CLOFORK)");
278*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(fcntl(b_confined, F_SETCONFINED, 1), "F_SETCONFINED");
279*d4514f0bSApple OSS Distributions
280*d4514f0bSApple OSS Distributions T_SETUPEND;
281*d4514f0bSApple OSS Distributions
282*d4514f0bSApple OSS Distributions /* Test all coalescence cases (non-upgrade/downgrade) */
283*d4514f0bSApple OSS Distributions
284*d4514f0bSApple OSS Distributions /*
285*d4514f0bSApple OSS Distributions * [ ]
286*d4514f0bSApple OSS Distributions * + [ ]
287*d4514f0bSApple OSS Distributions * = [ ]
288*d4514f0bSApple OSS Distributions */
289*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(posix_lock(a, 130, 20, 0), "Coalesce: initial posix lock");
290*d4514f0bSApple OSS Distributions LOCK_AND_CHECK("Coalesce: equal", a,
291*d4514f0bSApple OSS Distributions 130, 20, 0, TYPE_POSIX, /* POSIX-lock a, shared, from [100..199] */
292*d4514f0bSApple OSS Distributions 0, 0, EXCL, TYPE_OFD, /* OFD-get the entire file, exclusively */
293*d4514f0bSApple OSS Distributions 130, 20, 0, pid); /* The result should be: [100..199] is locked shared by our PID */
294*d4514f0bSApple OSS Distributions
295*d4514f0bSApple OSS Distributions /*
296*d4514f0bSApple OSS Distributions * [ ]
297*d4514f0bSApple OSS Distributions * + [ ]
298*d4514f0bSApple OSS Distributions * = [ ]
299*d4514f0bSApple OSS Distributions */
300*d4514f0bSApple OSS Distributions LOCK_AND_CHECK("Coalesce: adjacent high", a,
301*d4514f0bSApple OSS Distributions 150, 25, 0, TYPE_POSIX,
302*d4514f0bSApple OSS Distributions 0, 0, EXCL, TYPE_OFD,
303*d4514f0bSApple OSS Distributions 130, 45, 0, pid);
304*d4514f0bSApple OSS Distributions
305*d4514f0bSApple OSS Distributions /*
306*d4514f0bSApple OSS Distributions * [ ]
307*d4514f0bSApple OSS Distributions * + [ ]
308*d4514f0bSApple OSS Distributions * = [ ]
309*d4514f0bSApple OSS Distributions */
310*d4514f0bSApple OSS Distributions LOCK_AND_CHECK("Coalesce: adjacent low", a,
311*d4514f0bSApple OSS Distributions 125, 5, 0, TYPE_POSIX,
312*d4514f0bSApple OSS Distributions 0, 0, EXCL, TYPE_OFD,
313*d4514f0bSApple OSS Distributions 125, 50, 0, pid);
314*d4514f0bSApple OSS Distributions
315*d4514f0bSApple OSS Distributions /*
316*d4514f0bSApple OSS Distributions * [ ]
317*d4514f0bSApple OSS Distributions * + [ ]
318*d4514f0bSApple OSS Distributions * = [ ]
319*d4514f0bSApple OSS Distributions */
320*d4514f0bSApple OSS Distributions LOCK_AND_CHECK("Coalesce: subsume smaller", a,
321*d4514f0bSApple OSS Distributions 150, 10, 0, TYPE_POSIX,
322*d4514f0bSApple OSS Distributions 0, 0, EXCL, TYPE_OFD,
323*d4514f0bSApple OSS Distributions 125, 50, 0, pid);
324*d4514f0bSApple OSS Distributions
325*d4514f0bSApple OSS Distributions /*
326*d4514f0bSApple OSS Distributions * [ ]
327*d4514f0bSApple OSS Distributions * + [ ]
328*d4514f0bSApple OSS Distributions * = [ ]
329*d4514f0bSApple OSS Distributions */
330*d4514f0bSApple OSS Distributions LOCK_AND_CHECK("Coalesce: subsume larger", a,
331*d4514f0bSApple OSS Distributions 100, 100, 0, TYPE_POSIX,
332*d4514f0bSApple OSS Distributions 0, 0, EXCL, TYPE_OFD,
333*d4514f0bSApple OSS Distributions 100, 100, 0, pid);
334*d4514f0bSApple OSS Distributions
335*d4514f0bSApple OSS Distributions /*
336*d4514f0bSApple OSS Distributions * [ ]
337*d4514f0bSApple OSS Distributions * + [ ]
338*d4514f0bSApple OSS Distributions * = [ ]
339*d4514f0bSApple OSS Distributions */
340*d4514f0bSApple OSS Distributions LOCK_AND_CHECK("Coalesce: extend high", a,
341*d4514f0bSApple OSS Distributions 100, 125, 0, TYPE_POSIX,
342*d4514f0bSApple OSS Distributions 0, 0, EXCL, TYPE_OFD,
343*d4514f0bSApple OSS Distributions 100, 125, 0, pid);
344*d4514f0bSApple OSS Distributions
345*d4514f0bSApple OSS Distributions /*
346*d4514f0bSApple OSS Distributions * [ ]
347*d4514f0bSApple OSS Distributions * + [ ]
348*d4514f0bSApple OSS Distributions * = [ ]
349*d4514f0bSApple OSS Distributions */
350*d4514f0bSApple OSS Distributions LOCK_AND_CHECK("Coalesce: extend low", a,
351*d4514f0bSApple OSS Distributions 75, 150, 0, TYPE_POSIX,
352*d4514f0bSApple OSS Distributions 0, 0, EXCL, TYPE_OFD,
353*d4514f0bSApple OSS Distributions 75, 150, 0, pid);
354*d4514f0bSApple OSS Distributions
355*d4514f0bSApple OSS Distributions /*
356*d4514f0bSApple OSS Distributions * [ ]
357*d4514f0bSApple OSS Distributions * + [ ]
358*d4514f0bSApple OSS Distributions * = [ ]
359*d4514f0bSApple OSS Distributions */
360*d4514f0bSApple OSS Distributions LOCK_AND_CHECK("Coalesce: overlap start", a,
361*d4514f0bSApple OSS Distributions 50, 100, 0, TYPE_POSIX,
362*d4514f0bSApple OSS Distributions 0, 0, EXCL, TYPE_OFD,
363*d4514f0bSApple OSS Distributions 50, 175, 0, pid);
364*d4514f0bSApple OSS Distributions
365*d4514f0bSApple OSS Distributions /*
366*d4514f0bSApple OSS Distributions * [ ]
367*d4514f0bSApple OSS Distributions * + [ ]
368*d4514f0bSApple OSS Distributions * = [ ]
369*d4514f0bSApple OSS Distributions */
370*d4514f0bSApple OSS Distributions LOCK_AND_CHECK("Coalesce: overlap end", a,
371*d4514f0bSApple OSS Distributions 150, 100, 0, TYPE_POSIX,
372*d4514f0bSApple OSS Distributions 0, 0, EXCL, TYPE_OFD,
373*d4514f0bSApple OSS Distributions 50, 200, 0, pid);
374*d4514f0bSApple OSS Distributions
375*d4514f0bSApple OSS Distributions /* Test all upgrade cases */
376*d4514f0bSApple OSS Distributions
377*d4514f0bSApple OSS Distributions /*
378*d4514f0bSApple OSS Distributions * [ R ]
379*d4514f0bSApple OSS Distributions * + [ W ]
380*d4514f0bSApple OSS Distributions * = [ R | W ]
381*d4514f0bSApple OSS Distributions */
382*d4514f0bSApple OSS Distributions LOCK_AND_CHECK("Upgrade: adjacent high not-coalesced", a,
383*d4514f0bSApple OSS Distributions 250, 50, EXCL, TYPE_POSIX, /* Take the posix lock exclusively */
384*d4514f0bSApple OSS Distributions 50, 250, 0, TYPE_OFD, /* and OFD-get shared */
385*d4514f0bSApple OSS Distributions 250, 50, EXCL, pid);
386*d4514f0bSApple OSS Distributions
387*d4514f0bSApple OSS Distributions /*
388*d4514f0bSApple OSS Distributions * [ R | W ]
389*d4514f0bSApple OSS Distributions * + [ W ]
390*d4514f0bSApple OSS Distributions * = [ W | R | W ]
391*d4514f0bSApple OSS Distributions */
392*d4514f0bSApple OSS Distributions LOCK_AND_CHECK("Upgrade: adjacent low not-coalesced", a,
393*d4514f0bSApple OSS Distributions 25, 25, EXCL, TYPE_POSIX,
394*d4514f0bSApple OSS Distributions 25, 225, 0, TYPE_OFD,
395*d4514f0bSApple OSS Distributions 25, 25, EXCL, pid);
396*d4514f0bSApple OSS Distributions
397*d4514f0bSApple OSS Distributions /*
398*d4514f0bSApple OSS Distributions * 25 50 250 300
399*d4514f0bSApple OSS Distributions * [ W | R | W ]
400*d4514f0bSApple OSS Distributions * + [ W ]
401*d4514f0bSApple OSS Distributions * [ W | R | W ]
402*d4514f0bSApple OSS Distributions * 25 50 225 300
403*d4514f0bSApple OSS Distributions */
404*d4514f0bSApple OSS Distributions LOCK_AND_CHECK("Upgrade: truncate shared-end, grow excl-start", a,
405*d4514f0bSApple OSS Distributions 225, 50, EXCL, TYPE_POSIX,
406*d4514f0bSApple OSS Distributions 50, 250, 0, TYPE_OFD,
407*d4514f0bSApple OSS Distributions 225, 75, EXCL, pid);
408*d4514f0bSApple OSS Distributions
409*d4514f0bSApple OSS Distributions /*
410*d4514f0bSApple OSS Distributions * 25 50 225 300
411*d4514f0bSApple OSS Distributions * [ W | R | W ]
412*d4514f0bSApple OSS Distributions * + [ W ]
413*d4514f0bSApple OSS Distributions * = [ W | R | W ]
414*d4514f0bSApple OSS Distributions * 25 60 225 300
415*d4514f0bSApple OSS Distributions */
416*d4514f0bSApple OSS Distributions LOCK_AND_CHECK("Upgrade: truncate shared-start, grow excl-end", a,
417*d4514f0bSApple OSS Distributions 40, 20, EXCL, TYPE_POSIX,
418*d4514f0bSApple OSS Distributions 0, 225, 0, TYPE_OFD,
419*d4514f0bSApple OSS Distributions 25, 35, EXCL, pid);
420*d4514f0bSApple OSS Distributions
421*d4514f0bSApple OSS Distributions /*
422*d4514f0bSApple OSS Distributions * 25 60 225 300
423*d4514f0bSApple OSS Distributions * [ W | R | W ]
424*d4514f0bSApple OSS Distributions * + [ W ]
425*d4514f0bSApple OSS Distributions * = [ W | | W | | W ]
426*d4514f0bSApple OSS Distributions * 25 60 100 150 225 300
427*d4514f0bSApple OSS Distributions */
428*d4514f0bSApple OSS Distributions LOCK_AND_CHECK("Upgrade: 3-way split", a,
429*d4514f0bSApple OSS Distributions 100, 50, EXCL, TYPE_POSIX,
430*d4514f0bSApple OSS Distributions 60, 165, 0, TYPE_OFD,
431*d4514f0bSApple OSS Distributions 100, 50, EXCL, pid);
432*d4514f0bSApple OSS Distributions
433*d4514f0bSApple OSS Distributions /*
434*d4514f0bSApple OSS Distributions * 25 60 100 150 225 300
435*d4514f0bSApple OSS Distributions * [ W | | W | | W ]
436*d4514f0bSApple OSS Distributions * + [ W ]
437*d4514f0bSApple OSS Distributions * = [ W ]
438*d4514f0bSApple OSS Distributions * 25 300
439*d4514f0bSApple OSS Distributions */
440*d4514f0bSApple OSS Distributions LOCK_AND_CHECK("Upgrade: subsume multiple locks", a,
441*d4514f0bSApple OSS Distributions 25, 275, EXCL, TYPE_POSIX,
442*d4514f0bSApple OSS Distributions 0, 0, 0, TYPE_OFD,
443*d4514f0bSApple OSS Distributions 25, 275, EXCL, pid);
444*d4514f0bSApple OSS Distributions
445*d4514f0bSApple OSS Distributions /* Unlock / waiter-wakeup cases */
446*d4514f0bSApple OSS Distributions
447*d4514f0bSApple OSS Distributions /*
448*d4514f0bSApple OSS Distributions * 25 300
449*d4514f0bSApple OSS Distributions * [ W ]
450*d4514f0bSApple OSS Distributions * + [W (wait)]
451*d4514f0bSApple OSS Distributions * 290 310
452*d4514f0bSApple OSS Distributions */
453*d4514f0bSApple OSS Distributions start_lock_thread(TYPE_OFD, <s[0], a, 290, 20, EXCL | WAIT); /* Wait on this lock in another thread */
454*d4514f0bSApple OSS Distributions random_pause();
455*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[0]), "Unlock: created waiting lock");
456*d4514f0bSApple OSS Distributions
457*d4514f0bSApple OSS Distributions /*
458*d4514f0bSApple OSS Distributions * 25 300
459*d4514f0bSApple OSS Distributions * [ W ]
460*d4514f0bSApple OSS Distributions * - [ - ]
461*d4514f0bSApple OSS Distributions * = [ W ]
462*d4514f0bSApple OSS Distributions * 25 300
463*d4514f0bSApple OSS Distributions */
464*d4514f0bSApple OSS Distributions UNLOCK_AND_CHECK("Unlock: region with no locks", a,
465*d4514f0bSApple OSS Distributions 300, 100, TYPE_POSIX,
466*d4514f0bSApple OSS Distributions 0, 0, 0, TYPE_OFD,
467*d4514f0bSApple OSS Distributions 25, 275, EXCL, pid);
468*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[0]), "Unlock: waiter still waiting");
469*d4514f0bSApple OSS Distributions
470*d4514f0bSApple OSS Distributions /*
471*d4514f0bSApple OSS Distributions * 25 300
472*d4514f0bSApple OSS Distributions * [ W ]
473*d4514f0bSApple OSS Distributions * - [ - ]
474*d4514f0bSApple OSS Distributions * = [ W ]
475*d4514f0bSApple OSS Distributions * 25 250
476*d4514f0bSApple OSS Distributions */
477*d4514f0bSApple OSS Distributions UNLOCK_AND_CHECK("Unlock: overlap end", a,
478*d4514f0bSApple OSS Distributions 250, 100, TYPE_POSIX,
479*d4514f0bSApple OSS Distributions 25, 1, 0, TYPE_OFD,
480*d4514f0bSApple OSS Distributions 25, 225, EXCL, pid);
481*d4514f0bSApple OSS Distributions
482*d4514f0bSApple OSS Distributions /*
483*d4514f0bSApple OSS Distributions * 25 250
484*d4514f0bSApple OSS Distributions * [ W ]
485*d4514f0bSApple OSS Distributions * [ W ]
486*d4514f0bSApple OSS Distributions * 290 310
487*d4514f0bSApple OSS Distributions */
488*d4514f0bSApple OSS Distributions
489*d4514f0bSApple OSS Distributions while (!has_completed(<s[0])) {
490*d4514f0bSApple OSS Distributions random_pause();
491*d4514f0bSApple OSS Distributions }
492*d4514f0bSApple OSS Distributions T_ASSERT_TRUE(has_completed(<s[0]), "Unlock: waiter woke up");
493*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_ZERO(lts[0].err, "Unlock: no err granting waiter");
494*d4514f0bSApple OSS Distributions GET_CHECK("Unlock: waiter granted confirmation", a,
495*d4514f0bSApple OSS Distributions 250, 100, 0, TYPE_POSIX,
496*d4514f0bSApple OSS Distributions 290, 20, EXCL, -1); /* -1 because it's a non-CONFINED OFD lock (with no PID) */
497*d4514f0bSApple OSS Distributions
498*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(ofd_unlock(a, 290, 20), "Unlock: unlock that now-granted waiter");
499*d4514f0bSApple OSS Distributions
500*d4514f0bSApple OSS Distributions /*
501*d4514f0bSApple OSS Distributions * 25 250
502*d4514f0bSApple OSS Distributions * [ W ] 290 310
503*d4514f0bSApple OSS Distributions * [ W ]
504*d4514f0bSApple OSS Distributions * - [ - ]
505*d4514f0bSApple OSS Distributions * = [ W ]
506*d4514f0bSApple OSS Distributions * 25 250
507*d4514f0bSApple OSS Distributions */
508*d4514f0bSApple OSS Distributions UNLOCK_AND_CHECK("Unlock: equal range", a,
509*d4514f0bSApple OSS Distributions 290, 20, TYPE_POSIX,
510*d4514f0bSApple OSS Distributions 0, 0, 0, TYPE_OFD,
511*d4514f0bSApple OSS Distributions 25, 225, EXCL, pid);
512*d4514f0bSApple OSS Distributions
513*d4514f0bSApple OSS Distributions /*
514*d4514f0bSApple OSS Distributions * 25 250
515*d4514f0bSApple OSS Distributions * [ W ]
516*d4514f0bSApple OSS Distributions * - [ - ]
517*d4514f0bSApple OSS Distributions * = [ W ]
518*d4514f0bSApple OSS Distributions * 50 250
519*d4514f0bSApple OSS Distributions */
520*d4514f0bSApple OSS Distributions UNLOCK_AND_CHECK("Unlock: overlap start", a,
521*d4514f0bSApple OSS Distributions 0, 50, TYPE_POSIX,
522*d4514f0bSApple OSS Distributions 0, 0, 0, TYPE_OFD,
523*d4514f0bSApple OSS Distributions 50, 200, EXCL, pid);
524*d4514f0bSApple OSS Distributions
525*d4514f0bSApple OSS Distributions /*
526*d4514f0bSApple OSS Distributions * 50 250
527*d4514f0bSApple OSS Distributions * [ W ]
528*d4514f0bSApple OSS Distributions * - [ ]
529*d4514f0bSApple OSS Distributions * = [ W ]
530*d4514f0bSApple OSS Distributions * 100 250
531*d4514f0bSApple OSS Distributions */
532*d4514f0bSApple OSS Distributions UNLOCK_AND_CHECK("Unlock: start-aligned", a,
533*d4514f0bSApple OSS Distributions 50, 50, TYPE_POSIX,
534*d4514f0bSApple OSS Distributions 0, 0, 0, TYPE_OFD,
535*d4514f0bSApple OSS Distributions 100, 150, EXCL, pid);
536*d4514f0bSApple OSS Distributions
537*d4514f0bSApple OSS Distributions /*
538*d4514f0bSApple OSS Distributions * 100 250
539*d4514f0bSApple OSS Distributions * [ W ]
540*d4514f0bSApple OSS Distributions * - [ ]
541*d4514f0bSApple OSS Distributions * = [ W ]
542*d4514f0bSApple OSS Distributions * 100 200
543*d4514f0bSApple OSS Distributions */
544*d4514f0bSApple OSS Distributions UNLOCK_AND_CHECK("Unlock: end-aligned", a,
545*d4514f0bSApple OSS Distributions 200, 50, TYPE_POSIX,
546*d4514f0bSApple OSS Distributions 0, 0, 0, TYPE_OFD,
547*d4514f0bSApple OSS Distributions 100, 100, EXCL, pid);
548*d4514f0bSApple OSS Distributions
549*d4514f0bSApple OSS Distributions /*
550*d4514f0bSApple OSS Distributions * 100 200
551*d4514f0bSApple OSS Distributions * [ W ]
552*d4514f0bSApple OSS Distributions * - [ ]
553*d4514f0bSApple OSS Distributions * = [ W ] [ W ]
554*d4514f0bSApple OSS Distributions * 100 125 175 200
555*d4514f0bSApple OSS Distributions */
556*d4514f0bSApple OSS Distributions UNLOCK_AND_CHECK("Unlock: split", a,
557*d4514f0bSApple OSS Distributions 125, 50, TYPE_POSIX,
558*d4514f0bSApple OSS Distributions 0, 150, 0, TYPE_OFD,
559*d4514f0bSApple OSS Distributions 100, 25, EXCL, pid);
560*d4514f0bSApple OSS Distributions
561*d4514f0bSApple OSS Distributions /* Check the tail-fragment too */
562*d4514f0bSApple OSS Distributions GET_CHECK("Unlock: split (tail-check)", a,
563*d4514f0bSApple OSS Distributions 125, 200, 0, TYPE_OFD,
564*d4514f0bSApple OSS Distributions 175, 25, EXCL, pid);
565*d4514f0bSApple OSS Distributions
566*d4514f0bSApple OSS Distributions /*
567*d4514f0bSApple OSS Distributions * 100 125 175 200
568*d4514f0bSApple OSS Distributions * [ W ] [ W ]
569*d4514f0bSApple OSS Distributions * - [ ]
570*d4514f0bSApple OSS Distributions * = (no locks)
571*d4514f0bSApple OSS Distributions */
572*d4514f0bSApple OSS Distributions UNLOCK_AND_CHECK("Unlock: multiple locks", a,
573*d4514f0bSApple OSS Distributions 0, 0, TYPE_POSIX,
574*d4514f0bSApple OSS Distributions 0, 0, 0, TYPE_OFD,
575*d4514f0bSApple OSS Distributions 0, 0, UNLOCK, -1);
576*d4514f0bSApple OSS Distributions
577*d4514f0bSApple OSS Distributions
578*d4514f0bSApple OSS Distributions /*
579*d4514f0bSApple OSS Distributions * Downgrade / waiter-wakeup test:
580*d4514f0bSApple OSS Distributions * The only interesting different between this and upgrade is that waiting locks
581*d4514f0bSApple OSS Distributions * may now be granted. So let's test that...
582*d4514f0bSApple OSS Distributions */
583*d4514f0bSApple OSS Distributions
584*d4514f0bSApple OSS Distributions /*
585*d4514f0bSApple OSS Distributions * Create this lock layout
586*d4514f0bSApple OSS Distributions * 0 500
587*d4514f0bSApple OSS Distributions * [ W ]
588*d4514f0bSApple OSS Distributions *
589*d4514f0bSApple OSS Distributions * 0 100 200 300 400 600
590*d4514f0bSApple OSS Distributions * [R(wait)] [R(wait)] [ R(wait) ]
591*d4514f0bSApple OSS Distributions *
592*d4514f0bSApple OSS Distributions * 50 450
593*d4514f0bSApple OSS Distributions * [ R(wait) ]
594*d4514f0bSApple OSS Distributions */
595*d4514f0bSApple OSS Distributions
596*d4514f0bSApple OSS Distributions LOCK_AND_CHECK("Downgrade: first lock", a,
597*d4514f0bSApple OSS Distributions 0, 500, EXCL, TYPE_POSIX,
598*d4514f0bSApple OSS Distributions 0, 0, 0, TYPE_OFD,
599*d4514f0bSApple OSS Distributions 0, 500, EXCL, pid);
600*d4514f0bSApple OSS Distributions
601*d4514f0bSApple OSS Distributions start_lock_thread(TYPE_OFD, <s[0], a, 0, 100, WAIT);
602*d4514f0bSApple OSS Distributions start_lock_thread(TYPE_OFD, <s[1], a, 200, 100, WAIT);
603*d4514f0bSApple OSS Distributions start_lock_thread(TYPE_OFD, <s[2], a, 400, 200, WAIT);
604*d4514f0bSApple OSS Distributions start_lock_thread(TYPE_OFD, <s[3], a, 50, 400, WAIT);
605*d4514f0bSApple OSS Distributions
606*d4514f0bSApple OSS Distributions random_pause(); /* wait a bit to allow the lock threads to run */
607*d4514f0bSApple OSS Distributions
608*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[0]), "Downgrade: waiter 0 waiting");;
609*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[1]), "Downgrade: waiter 1 waiting");
610*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[2]), "Downgrade: waiter 2 waiting");
611*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[3]), "Downgrade: waiter 3 waiting");
612*d4514f0bSApple OSS Distributions
613*d4514f0bSApple OSS Distributions /* Open a gap just wide enough for the [200..300] lock to be granted */
614*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(posix_lock(a, 200, 100, 0), "Downgrade [200..300]");
615*d4514f0bSApple OSS Distributions
616*d4514f0bSApple OSS Distributions /*
617*d4514f0bSApple OSS Distributions * (We can't use LOCK_AND_CHECK here, because there may be two shared locks
618*d4514f0bSApple OSS Distributions * with that range (our downgrade, plus waiter #1) - so ofd_get() could return
619*d4514f0bSApple OSS Distributions * either one non-deterministically.)
620*d4514f0bSApple OSS Distributions */
621*d4514f0bSApple OSS Distributions
622*d4514f0bSApple OSS Distributions while (!has_completed(<s[1])) {
623*d4514f0bSApple OSS Distributions random_pause(); /* wait for waiter #1 to complete */
624*d4514f0bSApple OSS Distributions }
625*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[0]), "Downgrade: waiter 0 waiting");
626*d4514f0bSApple OSS Distributions T_ASSERT_TRUE(has_completed(<s[1]), "Downgrade: waiter 1 awoken");
627*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[2]), "Downgrade: waiter 2 waiting");
628*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[3]), "Downgrade: waiter 3 waiting");
629*d4514f0bSApple OSS Distributions
630*d4514f0bSApple OSS Distributions /* Open a gap just wide enough for the [400..600] lock to be granted */
631*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(posix_lock(a, 400, 100, 0), "Downgrade [400..500]");
632*d4514f0bSApple OSS Distributions
633*d4514f0bSApple OSS Distributions while (!has_completed(<s[2])) {
634*d4514f0bSApple OSS Distributions random_pause(); /* wait for waiter #2 to complete */
635*d4514f0bSApple OSS Distributions }
636*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[0]), "Downgrade: waiter 0 waiting");
637*d4514f0bSApple OSS Distributions T_ASSERT_TRUE(has_completed(<s[2]), "Downgrade: waiter 2 awoken");
638*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[3]), "Downgrade: waiter 3 waiting");
639*d4514f0bSApple OSS Distributions
640*d4514f0bSApple OSS Distributions /* Downgrade the remaining chunks */
641*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(posix_lock(a, 0, 500, 0), "Downgrade [0..500]");
642*d4514f0bSApple OSS Distributions
643*d4514f0bSApple OSS Distributions while (!has_completed(<s[0]) || !has_completed(<s[3])) {
644*d4514f0bSApple OSS Distributions random_pause(); /* wait for waiters #0 and #3 to complete */
645*d4514f0bSApple OSS Distributions }
646*d4514f0bSApple OSS Distributions T_ASSERT_TRUE(has_completed(<s[0]), "Downgrade: waiter 0 awoken");
647*d4514f0bSApple OSS Distributions T_ASSERT_TRUE(has_completed(<s[3]), "Downgrade: waiter 3 awoken");
648*d4514f0bSApple OSS Distributions
649*d4514f0bSApple OSS Distributions /* Unlock the remaining OFD shared locks */
650*d4514f0bSApple OSS Distributions UNLOCK_AND_CHECK("Downgrade: cleanup (unlock all shared OFD locks)", a,
651*d4514f0bSApple OSS Distributions 0, 0, TYPE_OFD,
652*d4514f0bSApple OSS Distributions 0, 0, 0, TYPE_POSIX,
653*d4514f0bSApple OSS Distributions 0, 0, UNLOCK, -1);
654*d4514f0bSApple OSS Distributions
655*d4514f0bSApple OSS Distributions /* Unlock the remaining POSIX lock [0..500] */
656*d4514f0bSApple OSS Distributions UNLOCK_AND_CHECK("Downgrade: cleanup (unlock all shared posix locks)", a,
657*d4514f0bSApple OSS Distributions 0, 500, TYPE_POSIX,
658*d4514f0bSApple OSS Distributions 0, 0, 0, TYPE_OFD,
659*d4514f0bSApple OSS Distributions 0, 0, UNLOCK, -1);
660*d4514f0bSApple OSS Distributions
661*d4514f0bSApple OSS Distributions
662*d4514f0bSApple OSS Distributions /* Test SEEK_END flock range decoding */
663*d4514f0bSApple OSS Distributions
664*d4514f0bSApple OSS Distributions {
665*d4514f0bSApple OSS Distributions /* SEEK_END start=-10 len=10 */
666*d4514f0bSApple OSS Distributions struct flock fl = {.l_start = -10, .l_len = 10, .l_type = F_WRLCK, .l_whence = SEEK_END};
667*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(fcntl(a, F_SETLK, &fl), "SEEK_END: -10, 10");
668*d4514f0bSApple OSS Distributions GET_CHECK("SEEK_END: -10, 10 ", a,
669*d4514f0bSApple OSS Distributions 0, 0, EXCL, TYPE_OFD,
670*d4514f0bSApple OSS Distributions file_len - 10, 10, EXCL, pid);
671*d4514f0bSApple OSS Distributions }
672*d4514f0bSApple OSS Distributions
673*d4514f0bSApple OSS Distributions {
674*d4514f0bSApple OSS Distributions /* SEEK_END start=-10 len=10 */
675*d4514f0bSApple OSS Distributions struct flock fl = {.l_start = -file_len, .l_len = file_len, .l_type = F_WRLCK, .l_whence = SEEK_END};
676*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(fcntl(a, F_SETLK, &fl), "SEEK_END: -file_len, file_len");
677*d4514f0bSApple OSS Distributions GET_CHECK("SEEK_END: -file_len, file_len", a,
678*d4514f0bSApple OSS Distributions 0, 0, EXCL, TYPE_OFD,
679*d4514f0bSApple OSS Distributions 0, file_len, EXCL, pid);
680*d4514f0bSApple OSS Distributions }
681*d4514f0bSApple OSS Distributions
682*d4514f0bSApple OSS Distributions {
683*d4514f0bSApple OSS Distributions /* Negative case: SEEK_END start=-(file_len + 10) len=20 */
684*d4514f0bSApple OSS Distributions struct flock fl = {.l_start = -(file_len + 10), .l_len = 20, .l_type = F_WRLCK, .l_whence = SEEK_END};
685*d4514f0bSApple OSS Distributions T_EXPECT_TRUE(fcntl(a, F_SETLK, &fl) && errno == EINVAL, "SEEK_END: -(file_len + 10), 20 => EINVAL");
686*d4514f0bSApple OSS Distributions }
687*d4514f0bSApple OSS Distributions
688*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(posix_unlock(a, 0, 0), "SEEK_END: cleanup (release locks)");
689*d4514f0bSApple OSS Distributions
690*d4514f0bSApple OSS Distributions /* Test interactions between all 3 lock types */
691*d4514f0bSApple OSS Distributions
692*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(flock(a, LOCK_SH | LOCK_NB), "Interaction: flock(a, shared)");
693*d4514f0bSApple OSS Distributions
694*d4514f0bSApple OSS Distributions /* Take (waiting) exclusive locks in all 3 types on a_confined */
695*d4514f0bSApple OSS Distributions start_lock_thread(TYPE_FLOCK, <s[0], a_confined, 0, 0, EXCL | WAIT);
696*d4514f0bSApple OSS Distributions start_lock_thread(TYPE_POSIX, <s[1], a_confined, 0, 0, EXCL | WAIT);
697*d4514f0bSApple OSS Distributions start_lock_thread(TYPE_OFD, <s[2], a_confined, 0, 0, EXCL | WAIT);
698*d4514f0bSApple OSS Distributions
699*d4514f0bSApple OSS Distributions /* Yuck. Allow time for these threads to run -and- block on the flock lock */
700*d4514f0bSApple OSS Distributions sleep(1);
701*d4514f0bSApple OSS Distributions
702*d4514f0bSApple OSS Distributions /* Take shared locks in the remaining 2 types (posix/ofd) on a */
703*d4514f0bSApple OSS Distributions
704*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(posix_lock(a, 0, 0, 0), "Interaction: posix_lock(a, 0, 0, 0)");
705*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(ofd_lock(a, 0, 0, 0), "Interaction: ofd_lock(a, 0, 0, 0)");
706*d4514f0bSApple OSS Distributions
707*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[0]), "Interaction: flock-waiter is starting or waiting");
708*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[1]), "Interaction: posix-waiter is starting or waiting");
709*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[2]), "Interaction: ofd-waiter is starting or waiting");
710*d4514f0bSApple OSS Distributions
711*d4514f0bSApple OSS Distributions T_EXPECT_POSIX_FAILURE(flock(a, LOCK_EX | LOCK_NB), EAGAIN, "Interaction: can't flock-upgrade");
712*d4514f0bSApple OSS Distributions T_EXPECT_POSIX_FAILURE(posix_lock(a, 0, 0, EXCL), EAGAIN, "Interaction: can't posix-upgrade");
713*d4514f0bSApple OSS Distributions T_EXPECT_POSIX_FAILURE(ofd_lock(a, 0, 0, EXCL), EAGAIN, "Interaction: can't ofd-upgrade");
714*d4514f0bSApple OSS Distributions
715*d4514f0bSApple OSS Distributions /*
716*d4514f0bSApple OSS Distributions * At this point:
717*d4514f0bSApple OSS Distributions * - 'a' owns a shared flock && a shared OFD lock
718*d4514f0bSApple OSS Distributions * - this process owns a shared POSIX lock
719*d4514f0bSApple OSS Distributions * - three threads are (hopefully) blocked waiting to take exclusive locks
720*d4514f0bSApple OSS Distributions * on 'a_confined.'
721*d4514f0bSApple OSS Distributions *
722*d4514f0bSApple OSS Distributions * Drop the POSIX lock ...
723*d4514f0bSApple OSS Distributions */
724*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(posix_unlock(a, 0, 0), "Interaction: Unlock posix");
725*d4514f0bSApple OSS Distributions
726*d4514f0bSApple OSS Distributions random_pause(); /* wait a bit to see if there are consequences for the waiting locks */
727*d4514f0bSApple OSS Distributions
728*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[0]), "Interaction: flock-waiter is still starting or waiting");
729*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[1]), "Interaction: posix-waiter is still starting or waiting");
730*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[2]), "Interaction: ofd-waiter is still starting or waiting");
731*d4514f0bSApple OSS Distributions
732*d4514f0bSApple OSS Distributions /*
733*d4514f0bSApple OSS Distributions * and drop the flock lock ...
734*d4514f0bSApple OSS Distributions */
735*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(flock(a, LOCK_UN), "Interaction: Unlock flock");
736*d4514f0bSApple OSS Distributions
737*d4514f0bSApple OSS Distributions random_pause(); /* wait a bit to see if there are consequences for the waiting locks */
738*d4514f0bSApple OSS Distributions
739*d4514f0bSApple OSS Distributions // Check that rdar://102160410 remains fixed.
740*d4514f0bSApple OSS Distributions // Before that, the LOCK_UN above would release the OFD lock too
741*d4514f0bSApple OSS Distributions
742*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[0]), "Interaction: flock-waiter is still starting or waiting");
743*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[1]), "Interaction: posix-waiter is still starting or waiting");
744*d4514f0bSApple OSS Distributions T_ASSERT_FALSE(has_completed(<s[2]), "Interaction: ofd-waiter is still starting or waiting");
745*d4514f0bSApple OSS Distributions
746*d4514f0bSApple OSS Distributions /*
747*d4514f0bSApple OSS Distributions * and finally drop the OFD lock, which should let one of the blocked threads
748*d4514f0bSApple OSS Distributions * acquire an exclusive lock. Work through them turn by turn.
749*d4514f0bSApple OSS Distributions */
750*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(ofd_unlock(a, 0, 0), "Interaction: Unlock ofd");
751*d4514f0bSApple OSS Distributions
752*d4514f0bSApple OSS Distributions bool unlocked[3] = {
753*d4514f0bSApple OSS Distributions false, false, false
754*d4514f0bSApple OSS Distributions };
755*d4514f0bSApple OSS Distributions
756*d4514f0bSApple OSS Distributions for (uint32_t waiters = 3; waiters > 0; waiters--) {
757*d4514f0bSApple OSS Distributions uint32_t num_waiting = 0;
758*d4514f0bSApple OSS Distributions do {
759*d4514f0bSApple OSS Distributions random_pause(); /* wait for consequences for the waiting locks */
760*d4514f0bSApple OSS Distributions num_waiting = !has_completed(<s[0]) + !has_completed(<s[1]) + !has_completed(<s[2]);
761*d4514f0bSApple OSS Distributions } while (num_waiting == waiters);
762*d4514f0bSApple OSS Distributions
763*d4514f0bSApple OSS Distributions T_ASSERT_EQ(num_waiting, waiters - 1, "Interaction: 1 waiter awoke");
764*d4514f0bSApple OSS Distributions
765*d4514f0bSApple OSS Distributions if (has_completed(<s[0]) && !unlocked[0]) {
766*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(flock(a_confined, LOCK_UN), "Interaction: Flock awoke, unlocking");
767*d4514f0bSApple OSS Distributions unlocked[0] = true;
768*d4514f0bSApple OSS Distributions } else if (has_completed(<s[1]) && !unlocked[1]) {
769*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(posix_unlock(a_confined, 0, 0), "Interaction: posix awoke, unlocking");
770*d4514f0bSApple OSS Distributions unlocked[1] = true;
771*d4514f0bSApple OSS Distributions } else if (has_completed(<s[2]) && !unlocked[2]) {
772*d4514f0bSApple OSS Distributions T_ASSERT_POSIX_SUCCESS(ofd_unlock(a_confined, 0, 0), "Interaction: ofd awoke, unlocking");
773*d4514f0bSApple OSS Distributions unlocked[2] = true;
774*d4514f0bSApple OSS Distributions }
775*d4514f0bSApple OSS Distributions }
776*d4514f0bSApple OSS Distributions
777*d4514f0bSApple OSS Distributions T_ASSERT_TRUE(has_completed(<s[0]), "Interaction: flock-waiter has completed");
778*d4514f0bSApple OSS Distributions T_ASSERT_TRUE(unlocked[0], "Interaction: flock-waiter was unlocked");
779*d4514f0bSApple OSS Distributions
780*d4514f0bSApple OSS Distributions T_ASSERT_TRUE(has_completed(<s[1]), "Interaction: posix-waiter has completed");
781*d4514f0bSApple OSS Distributions T_ASSERT_TRUE(unlocked[1], "Interaction: posix-waiter was unlocked");
782*d4514f0bSApple OSS Distributions
783*d4514f0bSApple OSS Distributions T_ASSERT_TRUE(has_completed(<s[2]), "Interaction: ofd-waiter has completed");
784*d4514f0bSApple OSS Distributions T_ASSERT_TRUE(unlocked[2], "Interaction: ofd-waiter was unlocked");
785*d4514f0bSApple OSS Distributions }
786