xref: /dragonfly/sys/kern/kern_lockf.c (revision dc71b7ab)
1 /*
2  * Copyright (c) 2004 Joerg Sonnenberger <joerg@bec.de>.  All rights reserved.
3  * Copyright (c) 2006 Matthew Dillon <dillon@backplane.com>.  All rights reserved.
4  *
5  * Copyright (c) 1982, 1986, 1989, 1993
6  *	The Regents of the University of California.  All rights reserved.
7  *
8  * This code is derived from software contributed to Berkeley by
9  * Scooter Morris at Genentech Inc.
10  *
11  * Redistribution and use in source and binary forms, with or without
12  * modification, are permitted provided that the following conditions
13  * are met:
14  * 1. Redistributions of source code must retain the above copyright
15  *    notice, this list of conditions and the following disclaimer.
16  * 2. Redistributions in binary form must reproduce the above copyright
17  *    notice, this list of conditions and the following disclaimer in the
18  *    documentation and/or other materials provided with the distribution.
19  * 3. Neither the name of the University nor the names of its contributors
20  *    may be used to endorse or promote products derived from this software
21  *    without specific prior written permission.
22  *
23  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33  * SUCH DAMAGE.
34  *
35  *	@(#)ufs_lockf.c	8.3 (Berkeley) 1/6/94
36  * $FreeBSD: src/sys/kern/kern_lockf.c,v 1.25 1999/11/16 16:28:56 phk Exp $
37  * $DragonFly: src/sys/kern/kern_lockf.c,v 1.37 2007/11/01 22:48:16 dillon Exp $
38  */
39 
40 #include <sys/param.h>
41 #include <sys/systm.h>
42 #include <sys/kernel.h>
43 #include <sys/lock.h>
44 #include <sys/proc.h>
45 #include <sys/unistd.h>
46 #include <sys/vnode.h>
47 #include <sys/malloc.h>
48 #include <sys/fcntl.h>
49 #include <sys/resourcevar.h>
50 
51 #include <sys/lockf.h>
52 #include <machine/limits.h>	/* for LLONG_MAX */
53 #include <machine/stdarg.h>
54 
55 #include <sys/spinlock2.h>
56 
57 #ifdef INVARIANTS
58 int lf_global_counter = 0;
59 #endif
60 
61 #ifdef LOCKF_DEBUG
62 int lf_print_ranges = 0;
63 
64 static void _lf_print_lock(const struct lockf *);
65 static void _lf_printf(const char *, ...);
66 
67 #define lf_print_lock(lock) if (lf_print_ranges) _lf_print_lock(lock)
68 #define lf_printf(ctl, args...)	if (lf_print_ranges) _lf_printf(ctl, args)
69 #else
70 #define lf_print_lock(lock)
71 #define lf_printf(ctl, args...)
72 #endif
73 
74 static MALLOC_DEFINE(M_LOCKF, "lockf", "Byte-range locking structures");
75 
76 static void	lf_wakeup(struct lockf *, off_t, off_t);
77 static struct lockf_range *lf_alloc_range(void);
78 static void	lf_create_range(struct lockf_range *, struct proc *, int, int,
79 				off_t, off_t);
80 static void	lf_insert(struct lockf_range_list *list,
81 				struct lockf_range *elm,
82 				struct lockf_range *insert_point);
83 static void	lf_destroy_range(struct lockf_range *);
84 
85 static int	lf_setlock(struct lockf *, struct proc *, int, int,
86 			   off_t, off_t);
87 static int	lf_getlock(struct flock *, struct lockf *, struct proc *,
88 			   int, int, off_t, off_t);
89 
90 static int	lf_count_change(struct proc *, int);
91 
92 /*
93  * Return TRUE (non-zero) if the type and posix flags match.
94  */
95 static __inline
96 int
97 lf_match(struct lockf_range *range, int type, int flags)
98 {
99 	if (range->lf_type != type)
100 		return(0);
101 	if ((range->lf_flags ^ flags) & F_POSIX)
102 		return(0);
103 	return(1);
104 }
105 
106 /*
107  * Check whether range and [start, end] overlap.
108  */
109 static __inline
110 int
111 lf_overlap(const struct lockf_range *range, off_t start, off_t end)
112 {
113 	if (range->lf_start >= start && range->lf_start <= end)
114 		return(1);
115 	else if (start >= range->lf_start && start <= range->lf_end)
116 		return(1);
117 	else
118 		return(0);
119 }
120 
121 
122 /*
123  * Change the POSIX lock accounting for the given process.
124  */
125 void
126 lf_count_adjust(struct proc *p, int increase)
127 {
128 	struct uidinfo *uip;
129 
130 	KKASSERT(p != NULL);
131 
132 	uip = p->p_ucred->cr_uidinfo;
133 	spin_lock(&uip->ui_lock);
134 
135 	if (increase)
136 		uip->ui_posixlocks += p->p_numposixlocks;
137 	else
138 		uip->ui_posixlocks -= p->p_numposixlocks;
139 
140 	KASSERT(uip->ui_posixlocks >= 0,
141 		("Negative number of POSIX locks held by %s user: %d.",
142 		 increase ? "new" : "old", uip->ui_posixlocks));
143 	spin_unlock(&uip->ui_lock);
144 }
145 
146 static int
147 lf_count_change(struct proc *owner, int diff)
148 {
149 	struct uidinfo *uip;
150 	int max, ret;
151 
152 	/* we might actually not have a process context */
153 	if (owner == NULL)
154 		return(0);
155 
156 	uip = owner->p_ucred->cr_uidinfo;
157 
158 	max = MIN(owner->p_rlimit[RLIMIT_POSIXLOCKS].rlim_cur,
159 		  maxposixlocksperuid);
160 
161 	spin_lock(&uip->ui_lock);
162 	if (diff > 0 && owner->p_ucred->cr_uid != 0 && max != -1 &&
163 	    uip->ui_posixlocks >= max ) {
164 		ret = 1;
165 	} else {
166 		uip->ui_posixlocks += diff;
167 		owner->p_numposixlocks += diff;
168 		KASSERT(uip->ui_posixlocks >= 0,
169 			("Negative number of POSIX locks held by user: %d.",
170 			 uip->ui_posixlocks));
171 		KASSERT(owner->p_numposixlocks >= 0,
172 			("Negative number of POSIX locks held by proc: %d.",
173 			 uip->ui_posixlocks));
174 		ret = 0;
175 	}
176 	spin_unlock(&uip->ui_lock);
177 	return ret;
178 }
179 
180 /*
181  * Advisory record locking support
182  */
183 int
184 lf_advlock(struct vop_advlock_args *ap, struct lockf *lock, u_quad_t size)
185 {
186 	struct flock *fl = ap->a_fl;
187 	struct proc *owner;
188 	off_t start, end;
189 	int type, flags, error;
190 	lwkt_token_t token;
191 
192 	/*
193 	 * Convert the flock structure into a start and end.
194 	 */
195 	switch (fl->l_whence) {
196 	case SEEK_SET:
197 	case SEEK_CUR:
198 		/*
199 		 * Caller is responsible for adding any necessary offset
200 		 * when SEEK_CUR is used.
201 		 */
202 		start = fl->l_start;
203 		break;
204 
205 	case SEEK_END:
206 		start = size + fl->l_start;
207 		break;
208 
209 	default:
210 		return(EINVAL);
211 	}
212 
213 	flags = ap->a_flags;
214 	if (start < 0)
215 		return(EINVAL);
216 	if (fl->l_len == 0) {
217 		flags |= F_NOEND;
218 		end = LLONG_MAX;
219 	} else if (fl->l_len < 0) {
220 		return(EINVAL);
221 	} else {
222 		end = start + fl->l_len - 1;
223 		if (end < start)
224 			return(EINVAL);
225 	}
226 
227 	type = fl->l_type;
228 	/*
229 	 * This isn't really correct for flock-style locks,
230 	 * but the current handling is somewhat broken anyway.
231 	 */
232 	owner = (struct proc *)ap->a_id;
233 
234 	/*
235 	 * Do the requested operation.
236 	 */
237 	token = lwkt_getpooltoken(lock);
238 
239 	if (lock->init_done == 0) {
240 		TAILQ_INIT(&lock->lf_range);
241 		TAILQ_INIT(&lock->lf_blocked);
242 		lock->init_done = 1;
243 	}
244 
245 	switch(ap->a_op) {
246 	case F_SETLK:
247 		/*
248 		 * NOTE: It is possible for both lf_range and lf_blocked to
249 		 * be empty if we block and get woken up, but another process
250 		 * then gets in and issues an unlock.  So VMAYHAVELOCKS must
251 		 * be set after the lf_setlock() operation completes rather
252 		 * then before.
253 		 */
254 		error = lf_setlock(lock, owner, type, flags, start, end);
255 		vsetflags(ap->a_vp, VMAYHAVELOCKS);
256 		break;
257 
258 	case F_UNLCK:
259 		error = lf_setlock(lock, owner, type, flags, start, end);
260 		if (TAILQ_EMPTY(&lock->lf_range) &&
261 		    TAILQ_EMPTY(&lock->lf_blocked)) {
262 			vclrflags(ap->a_vp, VMAYHAVELOCKS);
263 		}
264 		break;
265 
266 	case F_GETLK:
267 		error = lf_getlock(fl, lock, owner, type, flags, start, end);
268 		break;
269 
270 	default:
271 		error = EINVAL;
272 		break;
273 	}
274 	lwkt_reltoken(token);
275 	return(error);
276 }
277 
278 static int
279 lf_setlock(struct lockf *lock, struct proc *owner, int type, int flags,
280 	   off_t start, off_t end)
281 {
282 	struct lockf_range *range;
283 	struct lockf_range *brange;
284 	struct lockf_range *next;
285 	struct lockf_range *first_match;
286 	struct lockf_range *last_match;
287 	struct lockf_range *insert_point;
288 	struct lockf_range *new_range1;
289 	struct lockf_range *new_range2;
290 	int wakeup_needed;
291 	int double_clip;
292 	int unlock_override;
293 	int error = 0;
294 	int count;
295 	struct lockf_range_list deadlist;
296 
297 	new_range1 = NULL;
298 	new_range2 = NULL;
299 	count = 0;
300 
301 restart:
302 	/*
303 	 * Preallocate two ranges so we don't have to worry about blocking
304 	 * in the middle of the lock code.
305 	 */
306 	if (new_range1 == NULL)
307 		new_range1 = lf_alloc_range();
308 	if (new_range2 == NULL)
309 		new_range2 = lf_alloc_range();
310 	first_match = NULL;
311 	last_match = NULL;
312 	insert_point = NULL;
313 	wakeup_needed = 0;
314 
315 	lf_print_lock(lock);
316 
317 	/*
318 	 * Locate the insertion point for the new lock (the first range
319 	 * with an lf_start >= start).
320 	 *
321 	 * Locate the first and latch ranges owned by us that overlap
322 	 * the requested range.
323 	 */
324 	TAILQ_FOREACH(range, &lock->lf_range, lf_link) {
325 		if (insert_point == NULL && range->lf_start >= start)
326 			insert_point = range;
327 
328 		/*
329 		 * Skip non-overlapping locks.  Locks are sorted by lf_start
330 		 * So we can terminate the search when lf_start exceeds the
331 		 * requested range (insert_point is still guarenteed to be
332 		 * set properly).
333 		 */
334 		if (range->lf_end < start)
335 			continue;
336 		if (range->lf_start > end) {
337 			range = NULL;
338 			break;
339 		}
340 
341 		/*
342 		 * Overlapping lock.  Set first_match and last_match if we
343 		 * are the owner.
344 		 */
345 		if (range->lf_owner == owner) {
346 			if (first_match == NULL)
347 				first_match = range;
348 			last_match = range;
349 			continue;
350 		}
351 
352 		/*
353 		 * If we aren't the owner check for a conflicting lock.  Only
354 		 * if not unlocking.
355 		 */
356 		if (type != F_UNLCK) {
357 			if (type == F_WRLCK || range->lf_type == F_WRLCK)
358 				break;
359 		}
360 	}
361 
362 	/*
363 	 * If a conflicting lock was observed, block or fail as appropriate.
364 	 * (this code is skipped when unlocking)
365 	 */
366 	if (range != NULL) {
367 		if ((flags & F_WAIT) == 0) {
368 			error = EAGAIN;
369 			goto do_cleanup;
370 		}
371 
372 		/*
373 		 * We are blocked. For POSIX locks we have to check
374 		 * for deadlocks and return with EDEADLK. This is done
375 		 * by checking whether range->lf_owner is already
376 		 * blocked.
377 		 *
378 		 * Since flock-style locks cover the whole file, a
379 		 * deadlock between those is nearly impossible.
380 		 * This can only occur if a process tries to lock the
381 		 * same inode exclusively while holding a shared lock
382 		 * with another descriptor.
383 		 * XXX How can we cleanly detect this?
384 		 * XXX The current mixing of flock & fcntl/lockf is evil.
385 		 *
386 		 * Handle existing locks of flock-style like POSIX locks.
387 		 */
388 		if (flags & F_POSIX) {
389 			TAILQ_FOREACH(brange, &lock->lf_blocked, lf_link) {
390 				if (brange->lf_owner == range->lf_owner) {
391 					error = EDEADLK;
392 					goto do_cleanup;
393 				}
394 			}
395 		}
396 
397 		/*
398 		 * For flock-style locks, we must first remove
399 		 * any shared locks that we hold before we sleep
400 		 * waiting for an exclusive lock.
401 		 */
402 		if ((flags & F_POSIX) == 0 && type == F_WRLCK)
403 			lf_setlock(lock, owner, F_UNLCK, 0, start, end);
404 
405 		brange = new_range1;
406 		new_range1 = NULL;
407 		lf_create_range(brange, owner, type, 0, start, end);
408 		TAILQ_INSERT_TAIL(&lock->lf_blocked, brange, lf_link);
409 		error = tsleep(brange, PCATCH, "lockf", 0);
410 
411 		/*
412 		 * We may have been awaked by a signal and/or by a
413 		 * debugger continuing us (in which case we must remove
414 		 * ourselves from the blocked list) and/or by another
415 		 * process releasing/downgrading a lock (in which case
416 		 * we have already been removed from the blocked list
417 		 * and our lf_flags field is 1).
418 		 *
419 		 * Sleep if it looks like we might be livelocking.
420 		 */
421 		if (brange->lf_flags == 0)
422 			TAILQ_REMOVE(&lock->lf_blocked, brange, lf_link);
423 		if (count == 2)
424 			tsleep(brange, 0, "lockfz", 2);
425 		else
426 			++count;
427 		lf_destroy_range(brange);
428 
429 		if (error)
430 			goto do_cleanup;
431 		goto restart;
432 	}
433 
434 	/*
435 	 * If there are no overlapping locks owned by us then creating
436 	 * the new lock is easy.  This is the most common case.
437 	 */
438 	if (first_match == NULL) {
439 		if (type == F_UNLCK)
440 			goto do_wakeup;
441 		if (flags & F_POSIX) {
442 			if (lf_count_change(owner, 1)) {
443 				error = ENOLCK;
444 				goto do_cleanup;
445 			}
446 		}
447 		range = new_range1;
448 		new_range1 = NULL;
449 		lf_create_range(range, owner, type, flags, start, end);
450 		lf_insert(&lock->lf_range, range, insert_point);
451 		goto do_wakeup;
452 	}
453 
454 	/*
455 	 * double_clip - Calculate a special case where TWO locks may have
456 	 *		 to be added due to the new lock breaking up an
457 	 *		 existing incompatible lock in the middle.
458 	 *
459 	 * unlock_override - Calculate a special case where NO locks
460 	 *		 need to be created.  This occurs when an unlock
461 	 *		 does not clip any locks at the front and rear.
462 	 *
463 	 * WARNING!  closef() and fdrop() assume that an F_UNLCK of the
464 	 *	     entire range will always succeed so the unlock_override
465 	 *	     case is mandatory.
466 	 */
467 	double_clip = 0;
468 	unlock_override = 0;
469 	if (first_match->lf_start < start) {
470 		if (first_match == last_match && last_match->lf_end > end)
471 			double_clip = 1;
472 	} else if (type == F_UNLCK && last_match->lf_end <= end) {
473 		unlock_override = 1;
474 	}
475 
476 	/*
477 	 * Figure out the worst case net increase in POSIX locks and account
478 	 * for it now before we start modifying things.  If neither the
479 	 * first or last locks match we have an issue.  If there is only
480 	 * one overlapping range which needs to be clipped on both ends
481 	 * we wind up having to create up to two new locks, else only one.
482 	 *
483 	 * When unlocking the worst case is always 1 new lock if our
484 	 * unlock request cuts the middle out of an existing lock range.
485 	 *
486 	 * count represents the 'cleanup' adjustment needed.  It starts
487 	 * negative, is incremented whenever we create a new POSIX lock,
488 	 * and decremented whenever we delete an existing one.  At the
489 	 * end of the day it had better be <= 0 or we didn't calculate the
490 	 * worse case properly here.
491 	 */
492 	count = 0;
493 	if ((flags & F_POSIX) && !unlock_override) {
494 		if (!lf_match(first_match, type, flags) &&
495 		    !lf_match(last_match, type, flags)
496 		) {
497 			if (double_clip && type != F_UNLCK)
498 				count = -2;
499 			else
500 				count = -1;
501 		}
502 		if (count && lf_count_change(owner, -count)) {
503 			error = ENOLCK;
504 			goto do_cleanup;
505 		}
506 	}
507 	/* else flock style lock which encompasses entire range */
508 
509 	/*
510 	 * Create and insert the lock represented the requested range.
511 	 * Adjust the net POSIX lock count.  We have to move our insertion
512 	 * point since brange now represents the first record >= start.
513 	 *
514 	 * When unlocking, no new lock is inserted but we still clip.
515 	 */
516 	if (type != F_UNLCK) {
517 		brange = new_range1;
518 		new_range1 = NULL;
519 		lf_create_range(brange, owner, type, flags, start, end);
520 		lf_insert(&lock->lf_range, brange, insert_point);
521 		insert_point = brange;
522 		if (flags & F_POSIX)
523 			++count;
524 	} else {
525 		brange = NULL;
526 	}
527 
528 	/*
529 	 * Handle the double_clip case.  This is the only case where
530 	 * we wind up having to add TWO locks.
531 	 */
532 	if (double_clip) {
533 		KKASSERT(first_match == last_match);
534 		last_match = new_range2;
535 		new_range2 = NULL;
536 		lf_create_range(last_match, first_match->lf_owner,
537 				first_match->lf_type, first_match->lf_flags,
538 				end + 1, first_match->lf_end);
539 		first_match->lf_end = start - 1;
540 		first_match->lf_flags &= ~F_NOEND;
541 
542 		/*
543 		 * Figure out where to insert the right side clip.
544 		 */
545 		lf_insert(&lock->lf_range, last_match, first_match);
546 		if (last_match->lf_flags & F_POSIX)
547 			++count;
548 	}
549 
550 	/*
551 	 * Clip or destroy the locks between first_match and last_match,
552 	 * inclusive.  Ignore the primary lock we created (brange).  Note
553 	 * that if double-clipped, first_match and last_match will be
554 	 * outside our clipping range.  Otherwise first_match and last_match
555 	 * will be deleted.
556 	 *
557 	 * We have already taken care of any double clipping.
558 	 *
559 	 * The insert_point may become invalid as we delete records, do not
560 	 * use that pointer any more.  Also, when removing something other
561 	 * then 'range' we have to check to see if the item we are removing
562 	 * is 'next' and adjust 'next' properly.
563 	 *
564 	 * NOTE: brange will be NULL if F_UNLCKing.
565 	 */
566 	TAILQ_INIT(&deadlist);
567 	next = first_match;
568 
569 	while ((range = next) != NULL) {
570 		next = TAILQ_NEXT(range, lf_link);
571 
572 		/*
573 		 * Ignore elements that we do not own and ignore the
574 		 * primary request range which we just created.
575 		 */
576 		if (range->lf_owner != owner || range == brange)
577 			continue;
578 
579 		/*
580 		 * We may have to wakeup a waiter when downgrading a lock.
581 		 */
582 		if (type == F_UNLCK)
583 			wakeup_needed = 1;
584 		if (type == F_RDLCK && range->lf_type == F_WRLCK)
585 			wakeup_needed = 1;
586 
587 		/*
588 		 * Clip left.  This can only occur on first_match.
589 		 *
590 		 * Merge the left clip with brange if possible.  This must
591 		 * be done specifically, not in the optimized merge heuristic
592 		 * below, since we may have counted on it in our 'count'
593 		 * calculation above.
594 		 */
595 		if (range->lf_start < start) {
596 			KKASSERT(range == first_match);
597 			if (brange &&
598 			    range->lf_end >= start - 1 &&
599 			    lf_match(range, type, flags)) {
600 				range->lf_end = brange->lf_end;
601 				range->lf_flags |= brange->lf_flags & F_NOEND;
602 				/*
603 				 * Removing something other then 'range',
604 				 * adjust 'next' if necessary.
605 				 */
606 				if (next == brange)
607 					next = TAILQ_NEXT(next, lf_link);
608 				TAILQ_REMOVE(&lock->lf_range, brange, lf_link);
609 				if (brange->lf_flags & F_POSIX)
610 					--count;
611 				TAILQ_INSERT_TAIL(&deadlist, brange, lf_link);
612 				brange = range;
613 			} else if (range->lf_end >= start) {
614 				range->lf_end = start - 1;
615 				if (type != F_UNLCK)
616 					range->lf_flags &= ~F_NOEND;
617 			}
618 			if (range == last_match)
619 				break;
620 			continue;
621 		}
622 
623 		/*
624 		 * Clip right.  This can only occur on last_match.
625 		 *
626 		 * Merge the right clip if possible.  This must be done
627 		 * specifically, not in the optimized merge heuristic
628 		 * below, since we may have counted on it in our 'count'
629 		 * calculation.
630 		 *
631 		 * Since we are adjusting lf_start, we have to move the
632 		 * record to maintain the sorted list.  Since lf_start is
633 		 * only getting larger we can use the next element as the
634 		 * insert point (we don't have to backtrack).
635 		 */
636 		if (range->lf_end > end) {
637 			KKASSERT(range == last_match);
638 			if (brange &&
639 			    range->lf_start <= end + 1 &&
640 			    lf_match(range, type, flags)) {
641 				brange->lf_end = range->lf_end;
642 				brange->lf_flags |= range->lf_flags & F_NOEND;
643 				TAILQ_REMOVE(&lock->lf_range, range, lf_link);
644 				if (range->lf_flags & F_POSIX)
645 					--count;
646 				TAILQ_INSERT_TAIL(&deadlist, range, lf_link);
647 			} else if (range->lf_start <= end) {
648 				range->lf_start = end + 1;
649 				TAILQ_REMOVE(&lock->lf_range, range, lf_link);
650 				lf_insert(&lock->lf_range, range, next);
651 			}
652 			/* range == last_match, we are done */
653 			break;
654 		}
655 
656 		/*
657 		 * The record must be entirely enclosed.  Note that the
658 		 * record could be first_match or last_match, and will be
659 		 * deleted.
660 		 */
661 		KKASSERT(range->lf_start >= start && range->lf_end <= end);
662 		TAILQ_REMOVE(&lock->lf_range, range, lf_link);
663 		if (range->lf_flags & F_POSIX)
664 			--count;
665 		TAILQ_INSERT_TAIL(&deadlist, range, lf_link);
666 		if (range == last_match)
667 			break;
668 	}
669 
670 	/*
671 	 * Attempt to merge locks adjacent to brange.  For example, we may
672 	 * have had to clip first_match and/or last_match, and they might
673 	 * be adjacent.  Or there might simply have been an adjacent lock
674 	 * already there.
675 	 *
676 	 * Don't get fancy, just check adjacent elements in the list if they
677 	 * happen to be owned by us.
678 	 *
679 	 * This case only gets hit if we have a situation where a shared
680 	 * and exclusive lock are adjacent, and the exclusive lock is
681 	 * downgraded to shared or the shared lock is upgraded to exclusive.
682 	 */
683 	if (brange) {
684 		range = TAILQ_PREV(brange, lockf_range_list, lf_link);
685 		if (range &&
686 		    range->lf_owner == owner &&
687 		    range->lf_end == brange->lf_start - 1 &&
688 		    lf_match(range, type, flags)
689 		) {
690 			/*
691 			 * Extend range to cover brange and scrap brange.
692 			 */
693 			range->lf_end = brange->lf_end;
694 			range->lf_flags |= brange->lf_flags & F_NOEND;
695 			TAILQ_REMOVE(&lock->lf_range, brange, lf_link);
696 			if (brange->lf_flags & F_POSIX)
697 				--count;
698 			TAILQ_INSERT_TAIL(&deadlist, brange, lf_link);
699 			brange = range;
700 		}
701 		range = TAILQ_NEXT(brange, lf_link);
702 		if (range &&
703 		    range->lf_owner == owner &&
704 		    range->lf_start == brange->lf_end + 1 &&
705 		    lf_match(range, type, flags)
706 		) {
707 			/*
708 			 * Extend brange to cover range and scrap range.
709 			 */
710 			brange->lf_end = range->lf_end;
711 			brange->lf_flags |= range->lf_flags & F_NOEND;
712 			TAILQ_REMOVE(&lock->lf_range, range, lf_link);
713 			if (range->lf_flags & F_POSIX)
714 				--count;
715 			TAILQ_INSERT_TAIL(&deadlist, range, lf_link);
716 		}
717 	}
718 
719 	/*
720 	 * Destroy deleted elements.  We didn't want to do it in the loop
721 	 * because the free() might have blocked.
722 	 *
723 	 * Adjust the count for any posix locks we thought we might create
724 	 * but didn't.
725 	 */
726 	while ((range = TAILQ_FIRST(&deadlist)) != NULL) {
727 		TAILQ_REMOVE(&deadlist, range, lf_link);
728 		lf_destroy_range(range);
729 	}
730 
731 	KKASSERT(count <= 0);
732 	if (count < 0)
733 		lf_count_change(owner, count);
734 do_wakeup:
735 	lf_print_lock(lock);
736 	if (wakeup_needed)
737 		lf_wakeup(lock, start, end);
738 	error = 0;
739 do_cleanup:
740 	if (new_range1 != NULL)
741 		lf_destroy_range(new_range1);
742 	if (new_range2 != NULL)
743 		lf_destroy_range(new_range2);
744 	return(error);
745 }
746 
747 /*
748  * Check whether there is a blocking lock,
749  * and if so return its process identifier.
750  */
751 static int
752 lf_getlock(struct flock *fl, struct lockf *lock, struct proc *owner,
753 	   int type, int flags, off_t start, off_t end)
754 {
755 	struct lockf_range *range;
756 
757 	TAILQ_FOREACH(range, &lock->lf_range, lf_link)
758 		if (range->lf_owner != owner &&
759 		    lf_overlap(range, start, end) &&
760 		    (type == F_WRLCK || range->lf_type == F_WRLCK))
761 			break;
762 	if (range == NULL) {
763 		fl->l_type = F_UNLCK;
764 		return(0);
765 	}
766 	fl->l_type = range->lf_type;
767 	fl->l_whence = SEEK_SET;
768 	fl->l_start = range->lf_start;
769 	if (range->lf_flags & F_NOEND)
770 		fl->l_len = 0;
771 	else
772 		fl->l_len = range->lf_end - range->lf_start + 1;
773 	if (range->lf_owner != NULL && (range->lf_flags & F_POSIX))
774 		fl->l_pid = range->lf_owner->p_pid;
775 	else
776 		fl->l_pid = -1;
777 	return(0);
778 }
779 
780 /*
781  * Wakeup pending lock attempts.  Theoretically we can stop as soon as
782  * we encounter an exclusive request that covers the whole range (at least
783  * insofar as the sleep code above calls lf_wakeup() if it would otherwise
784  * exit instead of loop), but for now just wakeup all overlapping
785  * requests.  XXX
786  */
787 static void
788 lf_wakeup(struct lockf *lock, off_t start, off_t end)
789 {
790 	struct lockf_range *range, *nrange;
791 
792 	TAILQ_FOREACH_MUTABLE(range, &lock->lf_blocked, lf_link, nrange) {
793 		if (lf_overlap(range, start, end) == 0)
794 			continue;
795 		TAILQ_REMOVE(&lock->lf_blocked, range, lf_link);
796 		range->lf_flags = 1;
797 		wakeup(range);
798 	}
799 }
800 
801 /*
802  * Allocate a range structure and initialize it sufficiently such that
803  * lf_destroy_range() does not barf.
804  */
805 static struct lockf_range *
806 lf_alloc_range(void)
807 {
808 	struct lockf_range *range;
809 
810 #ifdef INVARIANTS
811 	atomic_add_int(&lf_global_counter, 1);
812 #endif
813 	range = kmalloc(sizeof(struct lockf_range), M_LOCKF, M_WAITOK);
814 	range->lf_owner = NULL;
815 	return(range);
816 }
817 
818 static void
819 lf_insert(struct lockf_range_list *list, struct lockf_range *elm,
820 	  struct lockf_range *insert_point)
821 {
822 	while (insert_point && insert_point->lf_start < elm->lf_start)
823 		insert_point = TAILQ_NEXT(insert_point, lf_link);
824 	if (insert_point != NULL)
825 		TAILQ_INSERT_BEFORE(insert_point, elm, lf_link);
826 	else
827 		TAILQ_INSERT_TAIL(list, elm, lf_link);
828 }
829 
830 static void
831 lf_create_range(struct lockf_range *range, struct proc *owner, int type,
832 		int flags, off_t start, off_t end)
833 {
834 	KKASSERT(start <= end);
835 	range->lf_type = type;
836 	range->lf_flags = flags;
837 	range->lf_start = start;
838 	range->lf_end = end;
839 	range->lf_owner = owner;
840 
841 	lf_printf("lf_create_range: %lld..%lld\n",
842 			range->lf_start, range->lf_end);
843 }
844 
845 static void
846 lf_destroy_range(struct lockf_range *range)
847 {
848 	lf_printf("lf_destroy_range: %lld..%lld\n",
849 		  range->lf_start, range->lf_end);
850 	kfree(range, M_LOCKF);
851 #ifdef INVARIANTS
852 	atomic_add_int(&lf_global_counter, -1);
853 	KKASSERT(lf_global_counter >= 0);
854 #endif
855 }
856 
857 #ifdef LOCKF_DEBUG
858 
859 static void
860 _lf_printf(const char *ctl, ...)
861 {
862 	struct proc *p;
863 	__va_list va;
864 
865 	if (lf_print_ranges) {
866 	    if ((p = curproc) != NULL)
867 		kprintf("pid %d (%s): ", p->p_pid, p->p_comm);
868 	}
869 	__va_start(va, ctl);
870 	kvprintf(ctl, va);
871 	__va_end(va);
872 }
873 
874 static void
875 _lf_print_lock(const struct lockf *lock)
876 {
877 	struct lockf_range *range;
878 
879 	if (lf_print_ranges == 0)
880 		return;
881 
882 	if (TAILQ_EMPTY(&lock->lf_range)) {
883 		lf_printf("lockf %p: no ranges locked\n", lock);
884 	} else {
885 		lf_printf("lockf %p:\n", lock);
886 	}
887 	TAILQ_FOREACH(range, &lock->lf_range, lf_link)
888 		kprintf("\t%lld..%lld type %s owned by %d\n",
889 		       range->lf_start, range->lf_end,
890 		       range->lf_type == F_RDLCK ? "shared" : "exclusive",
891 		       range->lf_flags & F_POSIX ? range->lf_owner->p_pid : -1);
892 	if (TAILQ_EMPTY(&lock->lf_blocked))
893 		kprintf("no process waiting for range\n");
894 	else
895 		kprintf("blocked locks:");
896 	TAILQ_FOREACH(range, &lock->lf_blocked, lf_link)
897 		kprintf("\t%lld..%lld type %s waiting on %p\n",
898 		       range->lf_start, range->lf_end,
899 		       range->lf_type == F_RDLCK ? "shared" : "exclusive",
900 		       range);
901 }
902 #endif /* LOCKF_DEBUG */
903