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