kern_lock.c revision 1.26 1 /* $NetBSD: kern_lock.c,v 1.26 2000/02/09 16:46:09 sommerfeld Exp $ */
2
3 /*-
4 * Copyright (c) 1999 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by Jason R. Thorpe of the Numerical Aerospace Simulation Facility,
9 * NASA Ames Research Center.
10 *
11 * This code is derived from software contributed to The NetBSD Foundation
12 * by Ross Harvey.
13 *
14 * Redistribution and use in source and binary forms, with or without
15 * modification, are permitted provided that the following conditions
16 * are met:
17 * 1. Redistributions of source code must retain the above copyright
18 * notice, this list of conditions and the following disclaimer.
19 * 2. Redistributions in binary form must reproduce the above copyright
20 * notice, this list of conditions and the following disclaimer in the
21 * documentation and/or other materials provided with the distribution.
22 * 3. All advertising materials mentioning features or use of this software
23 * must display the following acknowledgement:
24 * This product includes software developed by the NetBSD
25 * Foundation, Inc. and its contributors.
26 * 4. Neither the name of The NetBSD Foundation nor the names of its
27 * contributors may be used to endorse or promote products derived
28 * from this software without specific prior written permission.
29 *
30 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
31 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
32 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
33 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
34 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
35 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
36 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
37 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
38 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
39 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
40 * POSSIBILITY OF SUCH DAMAGE.
41 */
42
43 /*
44 * Copyright (c) 1995
45 * The Regents of the University of California. All rights reserved.
46 *
47 * This code contains ideas from software contributed to Berkeley by
48 * Avadis Tevanian, Jr., Michael Wayne Young, and the Mach Operating
49 * System project at Carnegie-Mellon University.
50 *
51 * Redistribution and use in source and binary forms, with or without
52 * modification, are permitted provided that the following conditions
53 * are met:
54 * 1. Redistributions of source code must retain the above copyright
55 * notice, this list of conditions and the following disclaimer.
56 * 2. Redistributions in binary form must reproduce the above copyright
57 * notice, this list of conditions and the following disclaimer in the
58 * documentation and/or other materials provided with the distribution.
59 * 3. All advertising materials mentioning features or use of this software
60 * must display the following acknowledgement:
61 * This product includes software developed by the University of
62 * California, Berkeley and its contributors.
63 * 4. Neither the name of the University nor the names of its contributors
64 * may be used to endorse or promote products derived from this software
65 * without specific prior written permission.
66 *
67 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
68 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
69 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
70 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
71 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
72 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
73 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
74 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
75 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
76 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
77 * SUCH DAMAGE.
78 *
79 * @(#)kern_lock.c 8.18 (Berkeley) 5/21/95
80 */
81
82 #include "opt_multiprocessor.h"
83 #include "opt_lockdebug.h"
84 #include "opt_ddb.h"
85
86 #include <sys/param.h>
87 #include <sys/proc.h>
88 #include <sys/lock.h>
89 #include <sys/systm.h>
90 #include <machine/cpu.h>
91
92 #if defined(LOCKDEBUG)
93 #include <sys/syslog.h>
94 /*
95 * note that stdarg.h and the ansi style va_start macro is used for both
96 * ansi and traditional c compiles.
97 * XXX: this requires that stdarg.h define: va_alist and va_dcl
98 */
99 #include <machine/stdarg.h>
100
101 void lock_printf __P((const char *fmt, ...));
102
103 int lock_debug_syslog = 0; /* defaults to printf, but can be patched */
104 #endif
105
106 /*
107 * Locking primitives implementation.
108 * Locks provide shared/exclusive sychronization.
109 */
110
111 #if defined(LOCKDEBUG) || defined(DIAGNOSTIC) /* { */
112 #if defined(MULTIPROCESSOR) /* { */
113 #define COUNT_CPU(cpu_id, x) \
114 /* atomic_add_ulong(&curcpu().ci_spin_locks, (x)) */
115 #else
116 u_long spin_locks;
117 #define COUNT_CPU(cpu_id, x) spin_locks += (x)
118 #endif /* MULTIPROCESSOR */ /* } */
119
120 #define COUNT(lkp, p, cpu_id, x) \
121 do { \
122 if ((lkp)->lk_flags & LK_SPIN) \
123 COUNT_CPU((cpu_id), (x)); \
124 else \
125 (p)->p_locks += (x); \
126 } while (0)
127 #else
128 #define COUNT(lkp, p, cpu_id, x)
129 #endif /* LOCKDEBUG || DIAGNOSTIC */ /* } */
130
131 /*
132 * Acquire a resource.
133 */
134 #define ACQUIRE(lkp, error, extflags, drain, wanted) \
135 if ((extflags) & LK_SPIN) { \
136 int interlocked; \
137 \
138 if ((drain) == 0) \
139 (lkp)->lk_waitcount++; \
140 for (interlocked = 1;;) { \
141 if (wanted) { \
142 if (interlocked) { \
143 simple_unlock(&(lkp)->lk_interlock); \
144 interlocked = 0; \
145 } \
146 } else if (interlocked) { \
147 break; \
148 } else { \
149 simple_lock(&(lkp)->lk_interlock); \
150 interlocked = 1; \
151 } \
152 } \
153 if ((drain) == 0) \
154 (lkp)->lk_waitcount--; \
155 KASSERT((wanted) == 0); \
156 error = 0; /* sanity */ \
157 } else { \
158 for (error = 0; wanted; ) { \
159 if ((drain)) \
160 (lkp)->lk_flags |= LK_WAITDRAIN; \
161 else \
162 (lkp)->lk_waitcount++; \
163 simple_unlock(&(lkp)->lk_interlock); \
164 /* XXX Cast away volatile. */ \
165 error = tsleep((drain) ? &(lkp)->lk_flags : \
166 (void *)(lkp), (lkp)->lk_prio, \
167 (lkp)->lk_wmesg, (lkp)->lk_timo); \
168 simple_lock(&(lkp)->lk_interlock); \
169 if ((drain) == 0) \
170 (lkp)->lk_waitcount--; \
171 if (error) \
172 break; \
173 if ((extflags) & LK_SLEEPFAIL) { \
174 error = ENOLCK; \
175 break; \
176 } \
177 } \
178 }
179
180 #define SETHOLDER(lkp, pid, cpu_id) \
181 do { \
182 if ((lkp)->lk_flags & LK_SPIN) \
183 (lkp)->lk_cpu = cpu_id; \
184 else \
185 (lkp)->lk_lockholder = pid; \
186 } while (0)
187
188 #define WEHOLDIT(lkp, pid, cpu_id) \
189 (((lkp)->lk_flags & LK_SPIN) != 0 ? \
190 ((lkp)->lk_cpu == (cpu_id)) : ((lkp)->lk_lockholder == (pid)))
191
192 #define WAKEUP_WAITER(lkp) \
193 do { \
194 if (((lkp)->lk_flags & LK_SPIN) == 0 && (lkp)->lk_waitcount) { \
195 /* XXX Cast away volatile. */ \
196 wakeup_one((void *)(lkp)); \
197 } \
198 } while (0)
199
200 #if defined(LOCKDEBUG) /* { */
201 #if defined(MULTIPROCESSOR) /* { */
202 struct simplelock spinlock_list_slock = SIMPLELOCK_INITIALIZER;
203
204 #define SPINLOCK_LIST_LOCK() cpu_simple_lock(&spinlock_list_slock)
205
206 #define SPINLOCK_LIST_UNLOCK() cpu_simple_unlock(&spinlock_list_slock)
207 #else
208 #define SPINLOCK_LIST_LOCK() /* nothing */
209
210 #define SPINLOCK_LIST_UNLOCK() /* nothing */
211 #endif /* MULTIPROCESSOR */ /* } */
212
213 TAILQ_HEAD(, lock) spinlock_list =
214 TAILQ_HEAD_INITIALIZER(spinlock_list);
215
216 #define HAVEIT(lkp) \
217 do { \
218 if ((lkp)->lk_flags & LK_SPIN) { \
219 int s = splhigh(); \
220 SPINLOCK_LIST_LOCK(); \
221 /* XXX Cast away volatile. */ \
222 TAILQ_INSERT_TAIL(&spinlock_list, (struct lock *)(lkp), \
223 lk_list); \
224 SPINLOCK_LIST_UNLOCK(); \
225 splx(s); \
226 } \
227 } while (0)
228
229 #define DONTHAVEIT(lkp) \
230 do { \
231 if ((lkp)->lk_flags & LK_SPIN) { \
232 int s = splhigh(); \
233 SPINLOCK_LIST_LOCK(); \
234 /* XXX Cast away volatile. */ \
235 TAILQ_REMOVE(&spinlock_list, (struct lock *)(lkp), \
236 lk_list); \
237 SPINLOCK_LIST_UNLOCK(); \
238 splx(s); \
239 } \
240 } while (0)
241 #else
242 #define HAVEIT(lkp) /* nothing */
243
244 #define DONTHAVEIT(lkp) /* nothing */
245 #endif /* LOCKDEBUG */ /* } */
246
247 #if defined(LOCKDEBUG)
248 /*
249 * Lock debug printing routine; can be configured to print to console
250 * or log to syslog.
251 */
252 void
253 #ifdef __STDC__
254 lock_printf(const char *fmt, ...)
255 #else
256 lock_printf(fmt, va_alist)
257 char *fmt;
258 va_dcl
259 #endif
260 {
261 va_list ap;
262
263 va_start(ap, fmt);
264 if (lock_debug_syslog)
265 vlog(LOG_DEBUG, fmt, ap);
266 else
267 vprintf(fmt, ap);
268 va_end(ap);
269 }
270 #endif /* LOCKDEBUG */
271
272 /*
273 * Initialize a lock; required before use.
274 */
275 void
276 lockinit(lkp, prio, wmesg, timo, flags)
277 struct lock *lkp;
278 int prio;
279 const char *wmesg;
280 int timo;
281 int flags;
282 {
283
284 memset(lkp, 0, sizeof(struct lock));
285 simple_lock_init(&lkp->lk_interlock);
286 lkp->lk_flags = flags & LK_EXTFLG_MASK;
287 if (flags & LK_SPIN)
288 lkp->lk_cpu = LK_NOCPU;
289 else {
290 lkp->lk_lockholder = LK_NOPROC;
291 lkp->lk_prio = prio;
292 lkp->lk_timo = timo;
293 }
294 lkp->lk_wmesg = wmesg; /* just a name for spin locks */
295 }
296
297 /*
298 * Determine the status of a lock.
299 */
300 int
301 lockstatus(lkp)
302 struct lock *lkp;
303 {
304 int lock_type = 0;
305
306 simple_lock(&lkp->lk_interlock);
307 if (lkp->lk_exclusivecount != 0)
308 lock_type = LK_EXCLUSIVE;
309 else if (lkp->lk_sharecount != 0)
310 lock_type = LK_SHARED;
311 simple_unlock(&lkp->lk_interlock);
312 return (lock_type);
313 }
314
315 /*
316 * Set, change, or release a lock.
317 *
318 * Shared requests increment the shared count. Exclusive requests set the
319 * LK_WANT_EXCL flag (preventing further shared locks), and wait for already
320 * accepted shared locks and shared-to-exclusive upgrades to go away.
321 */
322 int
323 lockmgr(lkp, flags, interlkp)
324 __volatile struct lock *lkp;
325 u_int flags;
326 struct simplelock *interlkp;
327 {
328 int error;
329 pid_t pid;
330 int extflags;
331 cpuid_t cpu_id;
332 struct proc *p = curproc;
333
334 error = 0;
335
336 simple_lock(&lkp->lk_interlock);
337 if (flags & LK_INTERLOCK)
338 simple_unlock(interlkp);
339 extflags = (flags | lkp->lk_flags) & LK_EXTFLG_MASK;
340
341 #ifdef DIAGNOSTIC /* { */
342 /*
343 * Don't allow spins on sleep locks and don't allow sleeps
344 * on spin locks.
345 */
346 if ((flags ^ lkp->lk_flags) & LK_SPIN)
347 panic("lockmgr: sleep/spin mismatch\n");
348 #endif /* } */
349
350 if (extflags & LK_SPIN)
351 pid = LK_KERNPROC;
352 else {
353 #ifdef DIAGNOSTIC /* { */
354 if (p == NULL)
355 panic("lockmgr: no context");
356 #endif /* } */
357 pid = p->p_pid;
358 }
359 cpu_id = cpu_number();
360
361 #ifdef DIAGNOSTIC /* { */
362 /*
363 * Once a lock has drained, the LK_DRAINING flag is set and an
364 * exclusive lock is returned. The only valid operation thereafter
365 * is a single release of that exclusive lock. This final release
366 * clears the LK_DRAINING flag and sets the LK_DRAINED flag. Any
367 * further requests of any sort will result in a panic. The bits
368 * selected for these two flags are chosen so that they will be set
369 * in memory that is freed (freed memory is filled with 0xdeadbeef).
370 * The final release is permitted to give a new lease on life to
371 * the lock by specifying LK_REENABLE.
372 */
373 if (lkp->lk_flags & (LK_DRAINING|LK_DRAINED)) {
374 if (lkp->lk_flags & LK_DRAINED)
375 panic("lockmgr: using decommissioned lock");
376 if ((flags & LK_TYPE_MASK) != LK_RELEASE ||
377 WEHOLDIT(lkp, pid, cpu_id) == 0)
378 panic("lockmgr: non-release on draining lock: %d\n",
379 flags & LK_TYPE_MASK);
380 lkp->lk_flags &= ~LK_DRAINING;
381 if ((flags & LK_REENABLE) == 0)
382 lkp->lk_flags |= LK_DRAINED;
383 }
384 #endif /* DIAGNOSTIC */ /* } */
385
386 switch (flags & LK_TYPE_MASK) {
387
388 case LK_SHARED:
389 if (WEHOLDIT(lkp, pid, cpu_id) == 0) {
390 /*
391 * If just polling, check to see if we will block.
392 */
393 if ((extflags & LK_NOWAIT) && (lkp->lk_flags &
394 (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE))) {
395 error = EBUSY;
396 break;
397 }
398 /*
399 * Wait for exclusive locks and upgrades to clear.
400 */
401 ACQUIRE(lkp, error, extflags, 0, lkp->lk_flags &
402 (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE));
403 if (error)
404 break;
405 lkp->lk_sharecount++;
406 COUNT(lkp, p, cpu_id, 1);
407 break;
408 }
409 /*
410 * We hold an exclusive lock, so downgrade it to shared.
411 * An alternative would be to fail with EDEADLK.
412 */
413 lkp->lk_sharecount++;
414 COUNT(lkp, p, cpu_id, 1);
415 /* fall into downgrade */
416
417 case LK_DOWNGRADE:
418 if (WEHOLDIT(lkp, pid, cpu_id) == 0 ||
419 lkp->lk_exclusivecount == 0)
420 panic("lockmgr: not holding exclusive lock");
421 lkp->lk_sharecount += lkp->lk_exclusivecount;
422 lkp->lk_exclusivecount = 0;
423 lkp->lk_recurselevel = 0;
424 lkp->lk_flags &= ~LK_HAVE_EXCL;
425 SETHOLDER(lkp, LK_NOPROC, LK_NOCPU);
426 DONTHAVEIT(lkp);
427 WAKEUP_WAITER(lkp);
428 break;
429
430 case LK_EXCLUPGRADE:
431 /*
432 * If another process is ahead of us to get an upgrade,
433 * then we want to fail rather than have an intervening
434 * exclusive access.
435 */
436 if (lkp->lk_flags & LK_WANT_UPGRADE) {
437 lkp->lk_sharecount--;
438 COUNT(lkp, p, cpu_id, -1);
439 error = EBUSY;
440 break;
441 }
442 /* fall into normal upgrade */
443
444 case LK_UPGRADE:
445 /*
446 * Upgrade a shared lock to an exclusive one. If another
447 * shared lock has already requested an upgrade to an
448 * exclusive lock, our shared lock is released and an
449 * exclusive lock is requested (which will be granted
450 * after the upgrade). If we return an error, the file
451 * will always be unlocked.
452 */
453 if (WEHOLDIT(lkp, pid, cpu_id) || lkp->lk_sharecount <= 0)
454 panic("lockmgr: upgrade exclusive lock");
455 lkp->lk_sharecount--;
456 COUNT(lkp, p, cpu_id, -1);
457 /*
458 * If we are just polling, check to see if we will block.
459 */
460 if ((extflags & LK_NOWAIT) &&
461 ((lkp->lk_flags & LK_WANT_UPGRADE) ||
462 lkp->lk_sharecount > 1)) {
463 error = EBUSY;
464 break;
465 }
466 if ((lkp->lk_flags & LK_WANT_UPGRADE) == 0) {
467 /*
468 * We are first shared lock to request an upgrade, so
469 * request upgrade and wait for the shared count to
470 * drop to zero, then take exclusive lock.
471 */
472 lkp->lk_flags |= LK_WANT_UPGRADE;
473 ACQUIRE(lkp, error, extflags, 0, lkp->lk_sharecount);
474 lkp->lk_flags &= ~LK_WANT_UPGRADE;
475 if (error)
476 break;
477 lkp->lk_flags |= LK_HAVE_EXCL;
478 SETHOLDER(lkp, pid, cpu_id);
479 HAVEIT(lkp);
480 if (lkp->lk_exclusivecount != 0)
481 panic("lockmgr: non-zero exclusive count");
482 lkp->lk_exclusivecount = 1;
483 if (extflags & LK_SETRECURSE)
484 lkp->lk_recurselevel = 1;
485 COUNT(lkp, p, cpu_id, 1);
486 break;
487 }
488 /*
489 * Someone else has requested upgrade. Release our shared
490 * lock, awaken upgrade requestor if we are the last shared
491 * lock, then request an exclusive lock.
492 */
493 if (lkp->lk_sharecount == 0)
494 WAKEUP_WAITER(lkp);
495 /* fall into exclusive request */
496
497 case LK_EXCLUSIVE:
498 if (WEHOLDIT(lkp, pid, cpu_id)) {
499 /*
500 * Recursive lock.
501 */
502 if ((extflags & LK_CANRECURSE) == 0 &&
503 lkp->lk_recurselevel == 0) {
504 if (extflags & LK_RECURSEFAIL) {
505 error = EDEADLK;
506 break;
507 } else
508 panic("lockmgr: locking against myself");
509 }
510 lkp->lk_exclusivecount++;
511 if (extflags & LK_SETRECURSE &&
512 lkp->lk_recurselevel == 0)
513 lkp->lk_recurselevel = lkp->lk_exclusivecount;
514 COUNT(lkp, p, cpu_id, 1);
515 break;
516 }
517 /*
518 * If we are just polling, check to see if we will sleep.
519 */
520 if ((extflags & LK_NOWAIT) && ((lkp->lk_flags &
521 (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE)) ||
522 lkp->lk_sharecount != 0)) {
523 error = EBUSY;
524 break;
525 }
526 /*
527 * Try to acquire the want_exclusive flag.
528 */
529 ACQUIRE(lkp, error, extflags, 0, lkp->lk_flags &
530 (LK_HAVE_EXCL | LK_WANT_EXCL));
531 if (error)
532 break;
533 lkp->lk_flags |= LK_WANT_EXCL;
534 /*
535 * Wait for shared locks and upgrades to finish.
536 */
537 ACQUIRE(lkp, error, extflags, 0, lkp->lk_sharecount != 0 ||
538 (lkp->lk_flags & LK_WANT_UPGRADE));
539 lkp->lk_flags &= ~LK_WANT_EXCL;
540 if (error)
541 break;
542 lkp->lk_flags |= LK_HAVE_EXCL;
543 SETHOLDER(lkp, pid, cpu_id);
544 HAVEIT(lkp);
545 if (lkp->lk_exclusivecount != 0)
546 panic("lockmgr: non-zero exclusive count");
547 lkp->lk_exclusivecount = 1;
548 if (extflags & LK_SETRECURSE)
549 lkp->lk_recurselevel = 1;
550 COUNT(lkp, p, cpu_id, 1);
551 break;
552
553 case LK_RELEASE:
554 if (lkp->lk_exclusivecount != 0) {
555 if (WEHOLDIT(lkp, pid, cpu_id) == 0) {
556 if (lkp->lk_flags & LK_SPIN) {
557 panic("lockmgr: processor %lu, not "
558 "exclusive lock holder %lu "
559 "unlocking", cpu_id, lkp->lk_cpu);
560 } else {
561 panic("lockmgr: pid %d, not "
562 "exclusive lock holder %d "
563 "unlocking", pid,
564 lkp->lk_lockholder);
565 }
566 }
567 if (lkp->lk_exclusivecount == lkp->lk_recurselevel)
568 lkp->lk_recurselevel = 0;
569 lkp->lk_exclusivecount--;
570 COUNT(lkp, p, cpu_id, -1);
571 if (lkp->lk_exclusivecount == 0) {
572 lkp->lk_flags &= ~LK_HAVE_EXCL;
573 SETHOLDER(lkp, LK_NOPROC, LK_NOCPU);
574 DONTHAVEIT(lkp);
575 }
576 } else if (lkp->lk_sharecount != 0) {
577 lkp->lk_sharecount--;
578 COUNT(lkp, p, cpu_id, -1);
579 }
580 WAKEUP_WAITER(lkp);
581 break;
582
583 case LK_DRAIN:
584 /*
585 * Check that we do not already hold the lock, as it can
586 * never drain if we do. Unfortunately, we have no way to
587 * check for holding a shared lock, but at least we can
588 * check for an exclusive one.
589 */
590 if (WEHOLDIT(lkp, pid, cpu_id))
591 panic("lockmgr: draining against myself");
592 /*
593 * If we are just polling, check to see if we will sleep.
594 */
595 if ((extflags & LK_NOWAIT) && ((lkp->lk_flags &
596 (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE)) ||
597 lkp->lk_sharecount != 0 || lkp->lk_waitcount != 0)) {
598 error = EBUSY;
599 break;
600 }
601 ACQUIRE(lkp, error, extflags, 1,
602 ((lkp->lk_flags &
603 (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE)) ||
604 lkp->lk_sharecount != 0 ||
605 lkp->lk_waitcount != 0));
606 if (error)
607 break;
608 lkp->lk_flags |= LK_DRAINING | LK_HAVE_EXCL;
609 SETHOLDER(lkp, pid, cpu_id);
610 HAVEIT(lkp);
611 lkp->lk_exclusivecount = 1;
612 /* XXX unlikely that we'd want this */
613 if (extflags & LK_SETRECURSE)
614 lkp->lk_recurselevel = 1;
615 COUNT(lkp, p, cpu_id, 1);
616 break;
617
618 default:
619 simple_unlock(&lkp->lk_interlock);
620 panic("lockmgr: unknown locktype request %d",
621 flags & LK_TYPE_MASK);
622 /* NOTREACHED */
623 }
624 if ((lkp->lk_flags & (LK_WAITDRAIN|LK_SPIN)) == LK_WAITDRAIN &&
625 ((lkp->lk_flags &
626 (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE)) == 0 &&
627 lkp->lk_sharecount == 0 && lkp->lk_waitcount == 0)) {
628 lkp->lk_flags &= ~LK_WAITDRAIN;
629 wakeup_one((void *)&lkp->lk_flags);
630 }
631 simple_unlock(&lkp->lk_interlock);
632 return (error);
633 }
634
635 /*
636 * Print out information about state of a lock. Used by VOP_PRINT
637 * routines to display ststus about contained locks.
638 */
639 void
640 lockmgr_printinfo(lkp)
641 __volatile struct lock *lkp;
642 {
643
644 if (lkp->lk_sharecount)
645 printf(" lock type %s: SHARED (count %d)", lkp->lk_wmesg,
646 lkp->lk_sharecount);
647 else if (lkp->lk_flags & LK_HAVE_EXCL) {
648 printf(" lock type %s: EXCL (count %d) by ",
649 lkp->lk_wmesg, lkp->lk_exclusivecount);
650 if (lkp->lk_flags & LK_SPIN)
651 printf("processor %lu", lkp->lk_cpu);
652 else
653 printf("pid %d", lkp->lk_lockholder);
654 } else
655 printf(" not locked");
656 if ((lkp->lk_flags & LK_SPIN) == 0 && lkp->lk_waitcount > 0)
657 printf(" with %d pending", lkp->lk_waitcount);
658 }
659
660 #if defined(LOCKDEBUG) /* { */
661 TAILQ_HEAD(, simplelock) simplelock_list =
662 TAILQ_HEAD_INITIALIZER(simplelock_list);
663
664 #if defined(MULTIPROCESSOR) /* { */
665 struct simplelock simplelock_list_slock = SIMPLELOCK_INITIALIZER;
666
667 #define SLOCK_LIST_LOCK() \
668 cpu_simple_lock(&simplelock_list_slock)
669
670 #define SLOCK_LIST_UNLOCK() \
671 cpu_simple_unlock(&simplelock_list_slock)
672
673 #define SLOCK_COUNT(x) \
674 /* atomic_add_ulong(&curcpu()->ci_simple_locks, (x)) */
675 #else
676 u_long simple_locks;
677
678 #define SLOCK_LIST_LOCK() /* nothing */
679
680 #define SLOCK_LIST_UNLOCK() /* nothing */
681
682 #define SLOCK_COUNT(x) simple_locks += (x)
683 #endif /* MULTIPROCESSOR */ /* } */
684
685 #ifdef DDB /* { */
686 int simple_lock_debugger = 0;
687 #define SLOCK_DEBUGGER() if (simple_lock_debugger) Debugger()
688 #else
689 #define SLOCK_DEBUGGER() /* nothing */
690 #endif /* } */
691
692 #ifdef MULTIPROCESSOR
693 #define SLOCK_MP() lock_printf("on cpu %d\n", cpu_number())
694 #else
695 #define SLOCK_MP() /* nothing */
696 #endif
697
698 #define SLOCK_WHERE(str, alp, id, l) \
699 do { \
700 lock_printf(str); \
701 lock_printf("lock: %p, currently at: %s:%d\n", (alp), (id), (l)); \
702 SLOCK_MP(); \
703 if ((alp)->lock_file != NULL) \
704 lock_printf("last locked: %s:%d\n", (alp)->lock_file, \
705 (alp)->lock_line); \
706 if ((alp)->unlock_file != NULL) \
707 lock_printf("last unlocked: %s:%d\n", (alp)->unlock_file, \
708 (alp)->unlock_line); \
709 SLOCK_DEBUGGER(); \
710 } while (0)
711
712 /*
713 * Simple lock functions so that the debugger can see from whence
714 * they are being called.
715 */
716 void
717 simple_lock_init(alp)
718 struct simplelock *alp;
719 {
720
721 #if defined(MULTIPROCESSOR) /* { */
722 cpu_simple_lock_init(alp);
723 #else
724 alp->lock_data = SIMPLELOCK_UNLOCKED;
725 #endif /* } */
726 alp->lock_file = NULL;
727 alp->lock_line = 0;
728 alp->unlock_file = NULL;
729 alp->unlock_line = 0;
730 alp->lock_holder = 0;
731 }
732
733 void
734 _simple_lock(alp, id, l)
735 __volatile struct simplelock *alp;
736 const char *id;
737 int l;
738 {
739 cpuid_t cpu_id = cpu_number();
740 int s;
741
742 s = splhigh();
743
744 /*
745 * MULTIPROCESSOR case: This is `safe' since if it's not us, we
746 * don't take any action, and just fall into the normal spin case.
747 */
748 if (alp->lock_data == SIMPLELOCK_LOCKED) {
749 #if defined(MULTIPROCESSOR) /* { */
750 if (alp->lock_holder == cpu_id) {
751 SLOCK_WHERE("simple_lock: locking against myself\n",
752 alp, id, l);
753 goto out;
754 }
755 #else
756 SLOCK_WHERE("simple_lock: lock held\n", alp, id, l);
757 goto out;
758 #endif /* MULTIPROCESSOR */ /* } */
759 }
760
761 #if defined(MULTIPROCESSOR) /* { */
762 /* Acquire the lock before modifying any fields. */
763 cpu_simple_lock(alp);
764 #else
765 alp->lock_data = SIMPLELOCK_LOCKED;
766 #endif /* } */
767
768 alp->lock_file = id;
769 alp->lock_line = l;
770 alp->lock_holder = cpu_id;
771
772 SLOCK_LIST_LOCK();
773 /* XXX Cast away volatile */
774 TAILQ_INSERT_TAIL(&simplelock_list, (struct simplelock *)alp, list);
775 SLOCK_LIST_UNLOCK();
776
777 SLOCK_COUNT(1);
778
779 out:
780 splx(s);
781 }
782
783 int
784 _simple_lock_try(alp, id, l)
785 __volatile struct simplelock *alp;
786 const char *id;
787 int l;
788 {
789 cpuid_t cpu_id = cpu_number();
790 int s, rv = 0;
791
792 s = splhigh();
793
794 /*
795 * MULTIPROCESSOR case: This is `safe' since if it's not us, we
796 * don't take any action.
797 */
798 #if defined(MULTIPROCESSOR) /* { */
799 if ((rv = cpu_simple_lock_try(alp)) == 0) {
800 if (alp->lock_holder == cpu_id)
801 SLOCK_WHERE("simple_lock_try: locking against myself\n",
802 alp, id, l);
803 goto out;
804 }
805 #else
806 if (alp->lock_data == SIMPLELOCK_LOCKED) {
807 SLOCK_WHERE("simple_lock_try: lock held\n", alp, id, l);
808 goto out;
809 }
810 alp->lock_data = SIMPLELOCK_LOCKED;
811 #endif /* MULTIPROCESSOR */ /* } */
812
813 /*
814 * At this point, we have acquired the lock.
815 */
816
817 rv = 1;
818
819 alp->lock_file = id;
820 alp->lock_line = l;
821 alp->lock_holder = cpu_id;
822
823 SLOCK_LIST_LOCK();
824 /* XXX Cast away volatile. */
825 TAILQ_INSERT_TAIL(&simplelock_list, (struct simplelock *)alp, list);
826 SLOCK_LIST_UNLOCK();
827
828 SLOCK_COUNT(1);
829
830 out:
831 splx(s);
832 return (rv);
833 }
834
835 void
836 _simple_unlock(alp, id, l)
837 __volatile struct simplelock *alp;
838 const char *id;
839 int l;
840 {
841 int s;
842
843 s = splhigh();
844
845 /*
846 * MULTIPROCESSOR case: This is `safe' because we think we hold
847 * the lock, and if we don't, we don't take any action.
848 */
849 if (alp->lock_data == SIMPLELOCK_UNLOCKED) {
850 SLOCK_WHERE("simple_unlock: lock not held\n",
851 alp, id, l);
852 goto out;
853 }
854
855 SLOCK_LIST_LOCK();
856 TAILQ_REMOVE(&simplelock_list, alp, list);
857 SLOCK_LIST_UNLOCK();
858
859 SLOCK_COUNT(-1);
860
861 alp->list.tqe_next = NULL; /* sanity */
862 alp->list.tqe_prev = NULL; /* sanity */
863
864 alp->unlock_file = id;
865 alp->unlock_line = l;
866
867 #if defined(MULTIPROCESSOR) /* { */
868 alp->lock_holder = LK_NOCPU;
869 /* Now that we've modified all fields, release the lock. */
870 cpu_simple_unlock(alp);
871 #else
872 alp->lock_data = SIMPLELOCK_UNLOCKED;
873 #endif /* } */
874
875 out:
876 splx(s);
877 }
878
879 void
880 simple_lock_dump()
881 {
882 struct simplelock *alp;
883 int s;
884
885 s = splhigh();
886 SLOCK_LIST_LOCK();
887 lock_printf("all simple locks:\n");
888 for (alp = TAILQ_FIRST(&simplelock_list); alp != NULL;
889 alp = TAILQ_NEXT(alp, list)) {
890 lock_printf("%p CPU %lu %s:%d\n", alp, alp->lock_holder,
891 alp->lock_file, alp->lock_line);
892 }
893 SLOCK_LIST_UNLOCK();
894 splx(s);
895 }
896
897 void
898 simple_lock_freecheck(start, end)
899 void *start, *end;
900 {
901 struct simplelock *alp;
902 int s;
903
904 s = splhigh();
905 SLOCK_LIST_LOCK();
906 for (alp = TAILQ_FIRST(&simplelock_list); alp != NULL;
907 alp = TAILQ_NEXT(alp, list)) {
908 if ((void *)alp >= start && (void *)alp < end) {
909 lock_printf("freeing simple_lock %p CPU %lu %s:%d\n",
910 alp, alp->lock_holder, alp->lock_file,
911 alp->lock_line);
912 SLOCK_DEBUGGER();
913 }
914 }
915 SLOCK_LIST_UNLOCK();
916 splx(s);
917 }
918 #endif /* LOCKDEBUG */ /* } */
919