vfs_lockf.c revision 1.52 1 /* $NetBSD: vfs_lockf.c,v 1.52 2006/05/20 12:06:20 yamt Exp $ */
2
3 /*
4 * Copyright (c) 1982, 1986, 1989, 1993
5 * The Regents of the University of California. All rights reserved.
6 *
7 * This code is derived from software contributed to Berkeley by
8 * Scooter Morris at Genentech Inc.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 * 3. Neither the name of the University nor the names of its contributors
19 * may be used to endorse or promote products derived from this software
20 * without specific prior written permission.
21 *
22 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
23 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
26 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 * SUCH DAMAGE.
33 *
34 * @(#)ufs_lockf.c 8.4 (Berkeley) 10/26/94
35 */
36
37 #include <sys/cdefs.h>
38 __KERNEL_RCSID(0, "$NetBSD: vfs_lockf.c,v 1.52 2006/05/20 12:06:20 yamt Exp $");
39
40 #include <sys/param.h>
41 #include <sys/systm.h>
42 #include <sys/kernel.h>
43 #include <sys/file.h>
44 #include <sys/proc.h>
45 #include <sys/vnode.h>
46 #include <sys/pool.h>
47 #include <sys/fcntl.h>
48 #include <sys/lockf.h>
49 #include <sys/kauth.h>
50
51 /*
52 * The lockf structure is a kernel structure which contains the information
53 * associated with a byte range lock. The lockf structures are linked into
54 * the inode structure. Locks are sorted by the starting byte of the lock for
55 * efficiency.
56 *
57 * lf_next is used for two purposes, depending on whether the lock is
58 * being held, or is in conflict with an existing lock. If this lock
59 * is held, it indicates the next lock on the same vnode.
60 * For pending locks, if lock->lf_next is non-NULL, then lock->lf_block
61 * must be queued on the lf_blkhd TAILQ of lock->lf_next.
62 */
63
64 TAILQ_HEAD(locklist, lockf);
65
66 struct lockf {
67 short lf_flags; /* Lock semantics: F_POSIX, F_FLOCK, F_WAIT */
68 short lf_type; /* Lock type: F_RDLCK, F_WRLCK */
69 off_t lf_start; /* The byte # of the start of the lock */
70 off_t lf_end; /* The byte # of the end of the lock (-1=EOF)*/
71 void *lf_id; /* process or file description holding lock */
72 struct lockf **lf_head; /* Back pointer to the head of lockf list */
73 struct lockf *lf_next; /* Next lock on this vnode, or blocking lock */
74 struct locklist lf_blkhd; /* List of requests blocked on this lock */
75 TAILQ_ENTRY(lockf) lf_block;/* A request waiting for a lock */
76 uid_t lf_uid; /* User ID responsible */
77 };
78
79 /* Maximum length of sleep chains to traverse to try and detect deadlock. */
80 #define MAXDEPTH 50
81
82 static POOL_INIT(lockfpool, sizeof(struct lockf), 0, 0, 0, "lockfpl",
83 &pool_allocator_nointr);
84
85 /*
86 * This variable controls the maximum number of processes that will
87 * be checked in doing deadlock detection.
88 */
89 int maxlockdepth = MAXDEPTH;
90
91 #ifdef LOCKF_DEBUG
92 int lockf_debug = 0;
93 #endif
94
95 #define NOLOCKF (struct lockf *)0
96 #define SELF 0x1
97 #define OTHERS 0x2
98
99 /*
100 * XXX TODO
101 * Misc cleanups: "caddr_t id" should be visible in the API as a
102 * "struct proc *".
103 * (This requires rototilling all VFS's which support advisory locking).
104 */
105
106 /*
107 * If there's a lot of lock contention on a single vnode, locking
108 * schemes which allow for more paralleism would be needed. Given how
109 * infrequently byte-range locks are actually used in typical BSD
110 * code, a more complex approach probably isn't worth it.
111 */
112
113 /*
114 * We enforce a limit on locks by uid, so that a single user cannot
115 * run the kernel out of memory. For now, the limit is pretty coarse.
116 * There is no limit on root.
117 *
118 * Splitting a lock will always succeed, regardless of current allocations.
119 * If you're slightly above the limit, we still have to permit an allocation
120 * so that the unlock can succeed. If the unlocking causes too many splits,
121 * however, you're totally cutoff.
122 */
123 int maxlocksperuid = 1024;
124
125 #ifdef LOCKF_DEBUG
126 /*
127 * Print out a lock.
128 */
129 static void
130 lf_print(char *tag, struct lockf *lock)
131 {
132
133 printf("%s: lock %p for ", tag, lock);
134 if (lock->lf_flags & F_POSIX)
135 printf("proc %d", ((struct proc *)lock->lf_id)->p_pid);
136 else
137 printf("file %p", (struct file *)lock->lf_id);
138 printf(" %s, start %qx, end %qx",
139 lock->lf_type == F_RDLCK ? "shared" :
140 lock->lf_type == F_WRLCK ? "exclusive" :
141 lock->lf_type == F_UNLCK ? "unlock" :
142 "unknown", lock->lf_start, lock->lf_end);
143 if (TAILQ_FIRST(&lock->lf_blkhd))
144 printf(" block %p\n", TAILQ_FIRST(&lock->lf_blkhd));
145 else
146 printf("\n");
147 }
148
149 static void
150 lf_printlist(char *tag, struct lockf *lock)
151 {
152 struct lockf *lf, *blk;
153
154 printf("%s: Lock list:\n", tag);
155 for (lf = *lock->lf_head; lf; lf = lf->lf_next) {
156 printf("\tlock %p for ", lf);
157 if (lf->lf_flags & F_POSIX)
158 printf("proc %d", ((struct proc *)lf->lf_id)->p_pid);
159 else
160 printf("file %p", (struct file *)lf->lf_id);
161 printf(", %s, start %qx, end %qx",
162 lf->lf_type == F_RDLCK ? "shared" :
163 lf->lf_type == F_WRLCK ? "exclusive" :
164 lf->lf_type == F_UNLCK ? "unlock" :
165 "unknown", lf->lf_start, lf->lf_end);
166 TAILQ_FOREACH(blk, &lf->lf_blkhd, lf_block) {
167 if (blk->lf_flags & F_POSIX)
168 printf("proc %d",
169 ((struct proc *)blk->lf_id)->p_pid);
170 else
171 printf("file %p", (struct file *)blk->lf_id);
172 printf(", %s, start %qx, end %qx",
173 blk->lf_type == F_RDLCK ? "shared" :
174 blk->lf_type == F_WRLCK ? "exclusive" :
175 blk->lf_type == F_UNLCK ? "unlock" :
176 "unknown", blk->lf_start, blk->lf_end);
177 if (TAILQ_FIRST(&blk->lf_blkhd))
178 panic("lf_printlist: bad list");
179 }
180 printf("\n");
181 }
182 }
183 #endif /* LOCKF_DEBUG */
184
185 /*
186 * 3 options for allowfail.
187 * 0 - always allocate. 1 - cutoff at limit. 2 - cutoff at double limit.
188 */
189 static struct lockf *
190 lf_alloc(uid_t uid, int allowfail)
191 {
192 struct uidinfo *uip;
193 struct lockf *lock;
194 int s;
195
196 uip = uid_find(uid);
197 UILOCK(uip, s);
198 if (uid && allowfail && uip->ui_lockcnt >
199 (allowfail == 1 ? maxlocksperuid : (maxlocksperuid * 2))) {
200 UIUNLOCK(uip, s);
201 return NULL;
202 }
203 uip->ui_lockcnt++;
204 UIUNLOCK(uip, s);
205 lock = pool_get(&lockfpool, PR_WAITOK);
206 lock->lf_uid = uid;
207 return lock;
208 }
209
210 static void
211 lf_free(struct lockf *lock)
212 {
213 struct uidinfo *uip;
214 int s;
215
216 uip = uid_find(lock->lf_uid);
217 UILOCK(uip, s);
218 uip->ui_lockcnt--;
219 UIUNLOCK(uip, s);
220 pool_put(&lockfpool, lock);
221 }
222
223 /*
224 * Walk the list of locks for an inode to
225 * find an overlapping lock (if any).
226 *
227 * NOTE: this returns only the FIRST overlapping lock. There
228 * may be more than one.
229 */
230 static int
231 lf_findoverlap(struct lockf *lf, struct lockf *lock, int type,
232 struct lockf ***prev, struct lockf **overlap)
233 {
234 off_t start, end;
235
236 *overlap = lf;
237 if (lf == NOLOCKF)
238 return 0;
239 #ifdef LOCKF_DEBUG
240 if (lockf_debug & 2)
241 lf_print("lf_findoverlap: looking for overlap in", lock);
242 #endif /* LOCKF_DEBUG */
243 start = lock->lf_start;
244 end = lock->lf_end;
245 while (lf != NOLOCKF) {
246 if (((type == SELF) && lf->lf_id != lock->lf_id) ||
247 ((type == OTHERS) && lf->lf_id == lock->lf_id)) {
248 *prev = &lf->lf_next;
249 *overlap = lf = lf->lf_next;
250 continue;
251 }
252 #ifdef LOCKF_DEBUG
253 if (lockf_debug & 2)
254 lf_print("\tchecking", lf);
255 #endif /* LOCKF_DEBUG */
256 /*
257 * OK, check for overlap
258 *
259 * Six cases:
260 * 0) no overlap
261 * 1) overlap == lock
262 * 2) overlap contains lock
263 * 3) lock contains overlap
264 * 4) overlap starts before lock
265 * 5) overlap ends after lock
266 */
267 if ((lf->lf_end != -1 && start > lf->lf_end) ||
268 (end != -1 && lf->lf_start > end)) {
269 /* Case 0 */
270 #ifdef LOCKF_DEBUG
271 if (lockf_debug & 2)
272 printf("no overlap\n");
273 #endif /* LOCKF_DEBUG */
274 if ((type & SELF) && end != -1 && lf->lf_start > end)
275 return 0;
276 *prev = &lf->lf_next;
277 *overlap = lf = lf->lf_next;
278 continue;
279 }
280 if ((lf->lf_start == start) && (lf->lf_end == end)) {
281 /* Case 1 */
282 #ifdef LOCKF_DEBUG
283 if (lockf_debug & 2)
284 printf("overlap == lock\n");
285 #endif /* LOCKF_DEBUG */
286 return 1;
287 }
288 if ((lf->lf_start <= start) &&
289 (end != -1) &&
290 ((lf->lf_end >= end) || (lf->lf_end == -1))) {
291 /* Case 2 */
292 #ifdef LOCKF_DEBUG
293 if (lockf_debug & 2)
294 printf("overlap contains lock\n");
295 #endif /* LOCKF_DEBUG */
296 return 2;
297 }
298 if (start <= lf->lf_start &&
299 (end == -1 ||
300 (lf->lf_end != -1 && end >= lf->lf_end))) {
301 /* Case 3 */
302 #ifdef LOCKF_DEBUG
303 if (lockf_debug & 2)
304 printf("lock contains overlap\n");
305 #endif /* LOCKF_DEBUG */
306 return 3;
307 }
308 if ((lf->lf_start < start) &&
309 ((lf->lf_end >= start) || (lf->lf_end == -1))) {
310 /* Case 4 */
311 #ifdef LOCKF_DEBUG
312 if (lockf_debug & 2)
313 printf("overlap starts before lock\n");
314 #endif /* LOCKF_DEBUG */
315 return 4;
316 }
317 if ((lf->lf_start > start) &&
318 (end != -1) &&
319 ((lf->lf_end > end) || (lf->lf_end == -1))) {
320 /* Case 5 */
321 #ifdef LOCKF_DEBUG
322 if (lockf_debug & 2)
323 printf("overlap ends after lock\n");
324 #endif /* LOCKF_DEBUG */
325 return 5;
326 }
327 panic("lf_findoverlap: default");
328 }
329 return 0;
330 }
331
332 /*
333 * Split a lock and a contained region into
334 * two or three locks as necessary.
335 */
336 static void
337 lf_split(struct lockf *lock1, struct lockf *lock2, struct lockf **sparelock)
338 {
339 struct lockf *splitlock;
340
341 #ifdef LOCKF_DEBUG
342 if (lockf_debug & 2) {
343 lf_print("lf_split", lock1);
344 lf_print("splitting from", lock2);
345 }
346 #endif /* LOCKF_DEBUG */
347 /*
348 * Check to see if spliting into only two pieces.
349 */
350 if (lock1->lf_start == lock2->lf_start) {
351 lock1->lf_start = lock2->lf_end + 1;
352 lock2->lf_next = lock1;
353 return;
354 }
355 if (lock1->lf_end == lock2->lf_end) {
356 lock1->lf_end = lock2->lf_start - 1;
357 lock2->lf_next = lock1->lf_next;
358 lock1->lf_next = lock2;
359 return;
360 }
361 /*
362 * Make a new lock consisting of the last part of
363 * the encompassing lock
364 */
365 splitlock = *sparelock;
366 *sparelock = NULL;
367 memcpy(splitlock, lock1, sizeof(*splitlock));
368 splitlock->lf_start = lock2->lf_end + 1;
369 TAILQ_INIT(&splitlock->lf_blkhd);
370 lock1->lf_end = lock2->lf_start - 1;
371 /*
372 * OK, now link it in
373 */
374 splitlock->lf_next = lock1->lf_next;
375 lock2->lf_next = splitlock;
376 lock1->lf_next = lock2;
377 }
378
379 /*
380 * Wakeup a blocklist
381 */
382 static void
383 lf_wakelock(struct lockf *listhead)
384 {
385 struct lockf *wakelock;
386
387 while ((wakelock = TAILQ_FIRST(&listhead->lf_blkhd))) {
388 KASSERT(wakelock->lf_next == listhead);
389 TAILQ_REMOVE(&listhead->lf_blkhd, wakelock, lf_block);
390 wakelock->lf_next = NOLOCKF;
391 #ifdef LOCKF_DEBUG
392 if (lockf_debug & 2)
393 lf_print("lf_wakelock: awakening", wakelock);
394 #endif
395 wakeup(wakelock);
396 }
397 }
398
399 /*
400 * Remove a byte-range lock on an inode.
401 *
402 * Generally, find the lock (or an overlap to that lock)
403 * and remove it (or shrink it), then wakeup anyone we can.
404 */
405 static int
406 lf_clearlock(struct lockf *unlock, struct lockf **sparelock)
407 {
408 struct lockf **head = unlock->lf_head;
409 struct lockf *lf = *head;
410 struct lockf *overlap, **prev;
411 int ovcase;
412
413 if (lf == NOLOCKF)
414 return 0;
415 #ifdef LOCKF_DEBUG
416 if (unlock->lf_type != F_UNLCK)
417 panic("lf_clearlock: bad type");
418 if (lockf_debug & 1)
419 lf_print("lf_clearlock", unlock);
420 #endif /* LOCKF_DEBUG */
421 prev = head;
422 while ((ovcase = lf_findoverlap(lf, unlock, SELF,
423 &prev, &overlap)) != 0) {
424 /*
425 * Wakeup the list of locks to be retried.
426 */
427 lf_wakelock(overlap);
428
429 switch (ovcase) {
430
431 case 1: /* overlap == lock */
432 *prev = overlap->lf_next;
433 lf_free(overlap);
434 break;
435
436 case 2: /* overlap contains lock: split it */
437 if (overlap->lf_start == unlock->lf_start) {
438 overlap->lf_start = unlock->lf_end + 1;
439 break;
440 }
441 lf_split(overlap, unlock, sparelock);
442 overlap->lf_next = unlock->lf_next;
443 break;
444
445 case 3: /* lock contains overlap */
446 *prev = overlap->lf_next;
447 lf = overlap->lf_next;
448 lf_free(overlap);
449 continue;
450
451 case 4: /* overlap starts before lock */
452 overlap->lf_end = unlock->lf_start - 1;
453 prev = &overlap->lf_next;
454 lf = overlap->lf_next;
455 continue;
456
457 case 5: /* overlap ends after lock */
458 overlap->lf_start = unlock->lf_end + 1;
459 break;
460 }
461 break;
462 }
463 #ifdef LOCKF_DEBUG
464 if (lockf_debug & 1)
465 lf_printlist("lf_clearlock", unlock);
466 #endif /* LOCKF_DEBUG */
467 return 0;
468 }
469
470 /*
471 * Walk the list of locks for an inode and
472 * return the first blocking lock.
473 */
474 static struct lockf *
475 lf_getblock(struct lockf *lock)
476 {
477 struct lockf **prev, *overlap, *lf = *(lock->lf_head);
478
479 prev = lock->lf_head;
480 while (lf_findoverlap(lf, lock, OTHERS, &prev, &overlap) != 0) {
481 /*
482 * We've found an overlap, see if it blocks us
483 */
484 if ((lock->lf_type == F_WRLCK || overlap->lf_type == F_WRLCK))
485 return overlap;
486 /*
487 * Nope, point to the next one on the list and
488 * see if it blocks us
489 */
490 lf = overlap->lf_next;
491 }
492 return NOLOCKF;
493 }
494
495 /*
496 * Set a byte-range lock.
497 */
498 static int
499 lf_setlock(struct lockf *lock, struct lockf **sparelock,
500 struct simplelock *interlock)
501 {
502 struct lockf *block;
503 struct lockf **head = lock->lf_head;
504 struct lockf **prev, *overlap, *ltmp;
505 static char lockstr[] = "lockf";
506 int ovcase, priority, needtolink, error;
507
508 #ifdef LOCKF_DEBUG
509 if (lockf_debug & 1)
510 lf_print("lf_setlock", lock);
511 #endif /* LOCKF_DEBUG */
512
513 /*
514 * Set the priority
515 */
516 priority = PLOCK;
517 if (lock->lf_type == F_WRLCK)
518 priority += 4;
519 priority |= PCATCH;
520 /*
521 * Scan lock list for this file looking for locks that would block us.
522 */
523 while ((block = lf_getblock(lock)) != NULL) {
524 /*
525 * Free the structure and return if nonblocking.
526 */
527 if ((lock->lf_flags & F_WAIT) == 0) {
528 lf_free(lock);
529 return EAGAIN;
530 }
531 /*
532 * We are blocked. Since flock style locks cover
533 * the whole file, there is no chance for deadlock.
534 * For byte-range locks we must check for deadlock.
535 *
536 * Deadlock detection is done by looking through the
537 * wait channels to see if there are any cycles that
538 * involve us. MAXDEPTH is set just to make sure we
539 * do not go off into neverneverland.
540 */
541 if ((lock->lf_flags & F_POSIX) &&
542 (block->lf_flags & F_POSIX)) {
543 struct lwp *wlwp;
544 volatile const struct lockf *waitblock;
545 int i = 0;
546 struct proc *p;
547
548 p = (struct proc *)block->lf_id;
549 KASSERT(p != NULL);
550 while (i++ < maxlockdepth) {
551 simple_lock(&p->p_lock);
552 if (p->p_nlwps > 1) {
553 simple_unlock(&p->p_lock);
554 break;
555 }
556 wlwp = LIST_FIRST(&p->p_lwps);
557 if (wlwp->l_wmesg != lockstr) {
558 simple_unlock(&p->p_lock);
559 break;
560 }
561 simple_unlock(&p->p_lock);
562 waitblock = wlwp->l_wchan;
563 if (waitblock == NULL) {
564 /*
565 * this lwp just got up but
566 * not returned from ltsleep yet.
567 */
568 break;
569 }
570 /* Get the owner of the blocking lock */
571 waitblock = waitblock->lf_next;
572 if ((waitblock->lf_flags & F_POSIX) == 0)
573 break;
574 p = (struct proc *)waitblock->lf_id;
575 if (p == curproc) {
576 lf_free(lock);
577 printf("deadlock detected pid=%d\n",
578 (int)curproc->p_pid);
579 return EDEADLK;
580 }
581 }
582 /*
583 * If we're still following a dependency chain
584 * after maxlockdepth iterations, assume we're in
585 * a cycle to be safe.
586 */
587 if (i >= maxlockdepth) {
588 lf_free(lock);
589 return EDEADLK;
590 }
591 }
592 /*
593 * For flock type locks, we must first remove
594 * any shared locks that we hold before we sleep
595 * waiting for an exclusive lock.
596 */
597 if ((lock->lf_flags & F_FLOCK) &&
598 lock->lf_type == F_WRLCK) {
599 lock->lf_type = F_UNLCK;
600 (void) lf_clearlock(lock, NULL);
601 lock->lf_type = F_WRLCK;
602 }
603 /*
604 * Add our lock to the blocked list and sleep until we're free.
605 * Remember who blocked us (for deadlock detection).
606 */
607 lock->lf_next = block;
608 TAILQ_INSERT_TAIL(&block->lf_blkhd, lock, lf_block);
609 #ifdef LOCKF_DEBUG
610 if (lockf_debug & 1) {
611 lf_print("lf_setlock: blocking on", block);
612 lf_printlist("lf_setlock", block);
613 }
614 #endif /* LOCKF_DEBUG */
615 error = ltsleep(lock, priority, lockstr, 0, interlock);
616
617 /*
618 * We may have been awakened by a signal (in
619 * which case we must remove ourselves from the
620 * blocked list) and/or by another process
621 * releasing a lock (in which case we have already
622 * been removed from the blocked list and our
623 * lf_next field set to NOLOCKF).
624 */
625 if (lock->lf_next != NOLOCKF) {
626 TAILQ_REMOVE(&lock->lf_next->lf_blkhd, lock, lf_block);
627 lock->lf_next = NOLOCKF;
628 }
629 if (error) {
630 lf_free(lock);
631 return error;
632 }
633 }
634 /*
635 * No blocks!! Add the lock. Note that we will
636 * downgrade or upgrade any overlapping locks this
637 * process already owns.
638 *
639 * Skip over locks owned by other processes.
640 * Handle any locks that overlap and are owned by ourselves.
641 */
642 prev = head;
643 block = *head;
644 needtolink = 1;
645 for (;;) {
646 ovcase = lf_findoverlap(block, lock, SELF, &prev, &overlap);
647 if (ovcase)
648 block = overlap->lf_next;
649 /*
650 * Six cases:
651 * 0) no overlap
652 * 1) overlap == lock
653 * 2) overlap contains lock
654 * 3) lock contains overlap
655 * 4) overlap starts before lock
656 * 5) overlap ends after lock
657 */
658 switch (ovcase) {
659 case 0: /* no overlap */
660 if (needtolink) {
661 *prev = lock;
662 lock->lf_next = overlap;
663 }
664 break;
665
666 case 1: /* overlap == lock */
667 /*
668 * If downgrading lock, others may be
669 * able to acquire it.
670 */
671 if (lock->lf_type == F_RDLCK &&
672 overlap->lf_type == F_WRLCK)
673 lf_wakelock(overlap);
674 overlap->lf_type = lock->lf_type;
675 lf_free(lock);
676 lock = overlap; /* for debug output below */
677 break;
678
679 case 2: /* overlap contains lock */
680 /*
681 * Check for common starting point and different types.
682 */
683 if (overlap->lf_type == lock->lf_type) {
684 lf_free(lock);
685 lock = overlap; /* for debug output below */
686 break;
687 }
688 if (overlap->lf_start == lock->lf_start) {
689 *prev = lock;
690 lock->lf_next = overlap;
691 overlap->lf_start = lock->lf_end + 1;
692 } else
693 lf_split(overlap, lock, sparelock);
694 lf_wakelock(overlap);
695 break;
696
697 case 3: /* lock contains overlap */
698 /*
699 * If downgrading lock, others may be able to
700 * acquire it, otherwise take the list.
701 */
702 if (lock->lf_type == F_RDLCK &&
703 overlap->lf_type == F_WRLCK) {
704 lf_wakelock(overlap);
705 } else {
706 while ((ltmp = TAILQ_FIRST(&overlap->lf_blkhd))) {
707 KASSERT(ltmp->lf_next == overlap);
708 TAILQ_REMOVE(&overlap->lf_blkhd, ltmp,
709 lf_block);
710 ltmp->lf_next = lock;
711 TAILQ_INSERT_TAIL(&lock->lf_blkhd,
712 ltmp, lf_block);
713 }
714 }
715 /*
716 * Add the new lock if necessary and delete the overlap.
717 */
718 if (needtolink) {
719 *prev = lock;
720 lock->lf_next = overlap->lf_next;
721 prev = &lock->lf_next;
722 needtolink = 0;
723 } else
724 *prev = overlap->lf_next;
725 lf_free(overlap);
726 continue;
727
728 case 4: /* overlap starts before lock */
729 /*
730 * Add lock after overlap on the list.
731 */
732 lock->lf_next = overlap->lf_next;
733 overlap->lf_next = lock;
734 overlap->lf_end = lock->lf_start - 1;
735 prev = &lock->lf_next;
736 lf_wakelock(overlap);
737 needtolink = 0;
738 continue;
739
740 case 5: /* overlap ends after lock */
741 /*
742 * Add the new lock before overlap.
743 */
744 if (needtolink) {
745 *prev = lock;
746 lock->lf_next = overlap;
747 }
748 overlap->lf_start = lock->lf_end + 1;
749 lf_wakelock(overlap);
750 break;
751 }
752 break;
753 }
754 #ifdef LOCKF_DEBUG
755 if (lockf_debug & 1) {
756 lf_print("lf_setlock: got the lock", lock);
757 lf_printlist("lf_setlock", lock);
758 }
759 #endif /* LOCKF_DEBUG */
760 return 0;
761 }
762
763 /*
764 * Check whether there is a blocking lock,
765 * and if so return its process identifier.
766 */
767 static int
768 lf_getlock(struct lockf *lock, struct flock *fl)
769 {
770 struct lockf *block;
771
772 #ifdef LOCKF_DEBUG
773 if (lockf_debug & 1)
774 lf_print("lf_getlock", lock);
775 #endif /* LOCKF_DEBUG */
776
777 if ((block = lf_getblock(lock)) != NULL) {
778 fl->l_type = block->lf_type;
779 fl->l_whence = SEEK_SET;
780 fl->l_start = block->lf_start;
781 if (block->lf_end == -1)
782 fl->l_len = 0;
783 else
784 fl->l_len = block->lf_end - block->lf_start + 1;
785 if (block->lf_flags & F_POSIX)
786 fl->l_pid = ((struct proc *)block->lf_id)->p_pid;
787 else
788 fl->l_pid = -1;
789 } else {
790 fl->l_type = F_UNLCK;
791 }
792 return 0;
793 }
794
795 /*
796 * Do an advisory lock operation.
797 */
798 int
799 lf_advlock(struct vop_advlock_args *ap, struct lockf **head, off_t size)
800 {
801 struct proc *p = curproc;
802 struct flock *fl = ap->a_fl;
803 struct lockf *lock = NULL;
804 struct lockf *sparelock;
805 struct simplelock *interlock = &ap->a_vp->v_interlock;
806 off_t start, end;
807 int error = 0;
808
809 /*
810 * Convert the flock structure into a start and end.
811 */
812 switch (fl->l_whence) {
813 case SEEK_SET:
814 case SEEK_CUR:
815 /*
816 * Caller is responsible for adding any necessary offset
817 * when SEEK_CUR is used.
818 */
819 start = fl->l_start;
820 break;
821
822 case SEEK_END:
823 start = size + fl->l_start;
824 break;
825
826 default:
827 return EINVAL;
828 }
829 if (start < 0)
830 return EINVAL;
831
832 /*
833 * allocate locks before acquire simple lock.
834 * we need two locks in the worst case.
835 */
836 switch (ap->a_op) {
837 case F_SETLK:
838 case F_UNLCK:
839 /*
840 * XXX for F_UNLCK case, we can re-use lock.
841 */
842 if ((ap->a_flags & F_FLOCK) == 0) {
843 /*
844 * byte-range lock might need one more lock.
845 */
846 sparelock = lf_alloc(kauth_cred_geteuid(p->p_cred), 0);
847 if (sparelock == NULL) {
848 error = ENOMEM;
849 goto quit;
850 }
851 break;
852 }
853 /* FALLTHROUGH */
854
855 case F_GETLK:
856 sparelock = NULL;
857 break;
858
859 default:
860 return EINVAL;
861 }
862
863 lock = lf_alloc(kauth_cred_geteuid(p->p_cred), ap->a_op != F_UNLCK ? 1 : 2);
864 if (lock == NULL) {
865 error = ENOMEM;
866 goto quit;
867 }
868
869 simple_lock(interlock);
870
871 /*
872 * Avoid the common case of unlocking when inode has no locks.
873 */
874 if (*head == (struct lockf *)0) {
875 if (ap->a_op != F_SETLK) {
876 fl->l_type = F_UNLCK;
877 error = 0;
878 goto quit_unlock;
879 }
880 }
881
882 if (fl->l_len == 0)
883 end = -1;
884 else
885 end = start + fl->l_len - 1;
886 /*
887 * Create the lockf structure.
888 */
889 lock->lf_start = start;
890 lock->lf_end = end;
891 /* XXX NJWLWP
892 * I don't want to make the entire VFS universe use LWPs, because
893 * they don't need them, for the most part. This is an exception,
894 * and a kluge.
895 */
896
897 lock->lf_head = head;
898 lock->lf_type = fl->l_type;
899 lock->lf_next = (struct lockf *)0;
900 TAILQ_INIT(&lock->lf_blkhd);
901 lock->lf_flags = ap->a_flags;
902 if (lock->lf_flags & F_POSIX) {
903 KASSERT(curproc == (struct proc *)ap->a_id);
904 }
905 lock->lf_id = (struct proc *)ap->a_id;
906
907 /*
908 * Do the requested operation.
909 */
910 switch (ap->a_op) {
911
912 case F_SETLK:
913 error = lf_setlock(lock, &sparelock, interlock);
914 lock = NULL; /* lf_setlock freed it */
915 break;
916
917 case F_UNLCK:
918 error = lf_clearlock(lock, &sparelock);
919 break;
920
921 case F_GETLK:
922 error = lf_getlock(lock, fl);
923 break;
924
925 default:
926 break;
927 /* NOTREACHED */
928 }
929
930 quit_unlock:
931 simple_unlock(interlock);
932 quit:
933 if (lock)
934 lf_free(lock);
935 if (sparelock)
936 lf_free(sparelock);
937
938 return error;
939 }
940