Home | History | Annotate | Line # | Download | only in kern
kern_lock.c revision 1.82
      1  1.82      yamt /*	$NetBSD: kern_lock.c,v 1.82 2004/08/04 01:16:06 yamt Exp $	*/
      2  1.19   thorpej 
      3  1.19   thorpej /*-
      4  1.28   thorpej  * Copyright (c) 1999, 2000 The NetBSD Foundation, Inc.
      5  1.19   thorpej  * All rights reserved.
      6  1.19   thorpej  *
      7  1.19   thorpej  * This code is derived from software contributed to The NetBSD Foundation
      8  1.19   thorpej  * by Jason R. Thorpe of the Numerical Aerospace Simulation Facility,
      9  1.19   thorpej  * NASA Ames Research Center.
     10  1.19   thorpej  *
     11  1.19   thorpej  * This code is derived from software contributed to The NetBSD Foundation
     12  1.19   thorpej  * by Ross Harvey.
     13  1.19   thorpej  *
     14  1.19   thorpej  * Redistribution and use in source and binary forms, with or without
     15  1.19   thorpej  * modification, are permitted provided that the following conditions
     16  1.19   thorpej  * are met:
     17  1.19   thorpej  * 1. Redistributions of source code must retain the above copyright
     18  1.19   thorpej  *    notice, this list of conditions and the following disclaimer.
     19  1.19   thorpej  * 2. Redistributions in binary form must reproduce the above copyright
     20  1.19   thorpej  *    notice, this list of conditions and the following disclaimer in the
     21  1.19   thorpej  *    documentation and/or other materials provided with the distribution.
     22  1.19   thorpej  * 3. All advertising materials mentioning features or use of this software
     23  1.19   thorpej  *    must display the following acknowledgement:
     24  1.19   thorpej  *	This product includes software developed by the NetBSD
     25  1.19   thorpej  *	Foundation, Inc. and its contributors.
     26  1.19   thorpej  * 4. Neither the name of The NetBSD Foundation nor the names of its
     27  1.19   thorpej  *    contributors may be used to endorse or promote products derived
     28  1.19   thorpej  *    from this software without specific prior written permission.
     29  1.19   thorpej  *
     30  1.19   thorpej  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     31  1.19   thorpej  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     32  1.19   thorpej  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     33  1.19   thorpej  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     34  1.19   thorpej  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     35  1.19   thorpej  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     36  1.19   thorpej  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     37  1.19   thorpej  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     38  1.19   thorpej  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     39  1.19   thorpej  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     40  1.19   thorpej  * POSSIBILITY OF SUCH DAMAGE.
     41  1.19   thorpej  */
     42   1.2      fvdl 
     43   1.1      fvdl /*
     44   1.1      fvdl  * Copyright (c) 1995
     45   1.1      fvdl  *	The Regents of the University of California.  All rights reserved.
     46   1.1      fvdl  *
     47   1.1      fvdl  * This code contains ideas from software contributed to Berkeley by
     48   1.1      fvdl  * Avadis Tevanian, Jr., Michael Wayne Young, and the Mach Operating
     49   1.1      fvdl  * System project at Carnegie-Mellon University.
     50   1.1      fvdl  *
     51   1.1      fvdl  * Redistribution and use in source and binary forms, with or without
     52   1.1      fvdl  * modification, are permitted provided that the following conditions
     53   1.1      fvdl  * are met:
     54   1.1      fvdl  * 1. Redistributions of source code must retain the above copyright
     55   1.1      fvdl  *    notice, this list of conditions and the following disclaimer.
     56   1.1      fvdl  * 2. Redistributions in binary form must reproduce the above copyright
     57   1.1      fvdl  *    notice, this list of conditions and the following disclaimer in the
     58   1.1      fvdl  *    documentation and/or other materials provided with the distribution.
     59  1.72       agc  * 3. Neither the name of the University nor the names of its contributors
     60   1.1      fvdl  *    may be used to endorse or promote products derived from this software
     61   1.1      fvdl  *    without specific prior written permission.
     62   1.1      fvdl  *
     63   1.1      fvdl  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     64   1.1      fvdl  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     65   1.1      fvdl  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     66   1.1      fvdl  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     67   1.1      fvdl  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     68   1.1      fvdl  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     69   1.1      fvdl  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     70   1.1      fvdl  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     71   1.1      fvdl  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     72   1.1      fvdl  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     73   1.1      fvdl  * SUCH DAMAGE.
     74   1.1      fvdl  *
     75   1.1      fvdl  *	@(#)kern_lock.c	8.18 (Berkeley) 5/21/95
     76   1.1      fvdl  */
     77  1.60     lukem 
     78  1.60     lukem #include <sys/cdefs.h>
     79  1.82      yamt __KERNEL_RCSID(0, "$NetBSD: kern_lock.c,v 1.82 2004/08/04 01:16:06 yamt Exp $");
     80   1.7   thorpej 
     81  1.21   thorpej #include "opt_multiprocessor.h"
     82   1.7   thorpej #include "opt_lockdebug.h"
     83  1.18       chs #include "opt_ddb.h"
     84   1.1      fvdl 
     85   1.1      fvdl #include <sys/param.h>
     86   1.1      fvdl #include <sys/proc.h>
     87   1.1      fvdl #include <sys/lock.h>
     88   1.2      fvdl #include <sys/systm.h>
     89   1.1      fvdl #include <machine/cpu.h>
     90   1.1      fvdl 
     91  1.25   thorpej #if defined(LOCKDEBUG)
     92  1.25   thorpej #include <sys/syslog.h>
     93  1.25   thorpej /*
     94  1.25   thorpej  * note that stdarg.h and the ansi style va_start macro is used for both
     95  1.25   thorpej  * ansi and traditional c compiles.
     96  1.25   thorpej  * XXX: this requires that stdarg.h define: va_alist and va_dcl
     97  1.25   thorpej  */
     98  1.25   thorpej #include <machine/stdarg.h>
     99  1.25   thorpej 
    100  1.36   thorpej void	lock_printf(const char *fmt, ...)
    101  1.37       eeh     __attribute__((__format__(__printf__,1,2)));
    102  1.25   thorpej 
    103  1.78   hannken static int acquire(__volatile struct lock **, int *, int, int, int);
    104  1.73      yamt 
    105  1.57  sommerfe int	lock_debug_syslog = 0;	/* defaults to printf, but can be patched */
    106  1.55   thorpej 
    107  1.55   thorpej #ifdef DDB
    108  1.55   thorpej #include <ddb/ddbvar.h>
    109  1.55   thorpej #include <machine/db_machdep.h>
    110  1.55   thorpej #include <ddb/db_command.h>
    111  1.55   thorpej #include <ddb/db_interface.h>
    112  1.55   thorpej #endif
    113  1.25   thorpej #endif
    114  1.25   thorpej 
    115   1.1      fvdl /*
    116   1.1      fvdl  * Locking primitives implementation.
    117  1.56       wiz  * Locks provide shared/exclusive synchronization.
    118   1.1      fvdl  */
    119   1.1      fvdl 
    120  1.21   thorpej #if defined(LOCKDEBUG) || defined(DIAGNOSTIC) /* { */
    121  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    122  1.21   thorpej #define	COUNT_CPU(cpu_id, x)						\
    123  1.47  sommerfe 	curcpu()->ci_spin_locks += (x)
    124  1.21   thorpej #else
    125  1.21   thorpej u_long	spin_locks;
    126  1.21   thorpej #define	COUNT_CPU(cpu_id, x)	spin_locks += (x)
    127  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
    128  1.21   thorpej 
    129  1.69   thorpej #define	COUNT(lkp, l, cpu_id, x)					\
    130  1.21   thorpej do {									\
    131  1.21   thorpej 	if ((lkp)->lk_flags & LK_SPIN)					\
    132  1.21   thorpej 		COUNT_CPU((cpu_id), (x));				\
    133  1.21   thorpej 	else								\
    134  1.69   thorpej 		(l)->l_locks += (x);					\
    135  1.30   thorpej } while (/*CONSTCOND*/0)
    136   1.1      fvdl #else
    137  1.22    mellon #define COUNT(lkp, p, cpu_id, x)
    138  1.48  sommerfe #define COUNT_CPU(cpu_id, x)
    139  1.21   thorpej #endif /* LOCKDEBUG || DIAGNOSTIC */ /* } */
    140   1.1      fvdl 
    141  1.52   thorpej #ifndef SPINLOCK_SPIN_HOOK		/* from <machine/lock.h> */
    142  1.52   thorpej #define	SPINLOCK_SPIN_HOOK		/* nothing */
    143  1.49   thorpej #endif
    144  1.49   thorpej 
    145  1.43   thorpej #define	INTERLOCK_ACQUIRE(lkp, flags, s)				\
    146  1.40   thorpej do {									\
    147  1.43   thorpej 	if ((flags) & LK_SPIN)						\
    148  1.65      fvdl 		s = spllock();						\
    149  1.40   thorpej 	simple_lock(&(lkp)->lk_interlock);				\
    150  1.66     perry } while (/*CONSTCOND*/ 0)
    151  1.40   thorpej 
    152  1.43   thorpej #define	INTERLOCK_RELEASE(lkp, flags, s)				\
    153  1.40   thorpej do {									\
    154  1.40   thorpej 	simple_unlock(&(lkp)->lk_interlock);				\
    155  1.52   thorpej 	if ((flags) & LK_SPIN)						\
    156  1.40   thorpej 		splx(s);						\
    157  1.66     perry } while (/*CONSTCOND*/ 0)
    158  1.40   thorpej 
    159  1.63       chs #ifdef DDB /* { */
    160  1.63       chs #ifdef MULTIPROCESSOR
    161  1.63       chs int simple_lock_debugger = 1;	/* more serious on MP */
    162  1.63       chs #else
    163  1.63       chs int simple_lock_debugger = 0;
    164  1.63       chs #endif
    165  1.63       chs #define	SLOCK_DEBUGGER()	if (simple_lock_debugger) Debugger()
    166  1.63       chs #define	SLOCK_TRACE()							\
    167  1.63       chs 	db_stack_trace_print((db_expr_t)__builtin_frame_address(0),	\
    168  1.71        pk 	    TRUE, 65535, "", lock_printf);
    169  1.63       chs #else
    170  1.63       chs #define	SLOCK_DEBUGGER()	/* nothing */
    171  1.63       chs #define	SLOCK_TRACE()		/* nothing */
    172  1.63       chs #endif /* } */
    173  1.63       chs 
    174  1.50   thorpej #if defined(LOCKDEBUG)
    175  1.50   thorpej #if defined(DDB)
    176  1.50   thorpej #define	SPINLOCK_SPINCHECK_DEBUGGER	Debugger()
    177  1.50   thorpej #else
    178  1.50   thorpej #define	SPINLOCK_SPINCHECK_DEBUGGER	/* nothing */
    179  1.50   thorpej #endif
    180  1.50   thorpej 
    181  1.50   thorpej #define	SPINLOCK_SPINCHECK_DECL						\
    182  1.50   thorpej 	/* 32-bits of count -- wrap constitutes a "spinout" */		\
    183  1.50   thorpej 	uint32_t __spinc = 0
    184  1.50   thorpej 
    185  1.50   thorpej #define	SPINLOCK_SPINCHECK						\
    186  1.50   thorpej do {									\
    187  1.50   thorpej 	if (++__spinc == 0) {						\
    188  1.71        pk 		lock_printf("LK_SPIN spinout, excl %d, share %d\n",	\
    189  1.50   thorpej 		    lkp->lk_exclusivecount, lkp->lk_sharecount);	\
    190  1.50   thorpej 		if (lkp->lk_exclusivecount)				\
    191  1.71        pk 			lock_printf("held by CPU %lu\n",		\
    192  1.50   thorpej 			    (u_long) lkp->lk_cpu);			\
    193  1.50   thorpej 		if (lkp->lk_lock_file)					\
    194  1.71        pk 			lock_printf("last locked at %s:%d\n",		\
    195  1.50   thorpej 			    lkp->lk_lock_file, lkp->lk_lock_line);	\
    196  1.50   thorpej 		if (lkp->lk_unlock_file)				\
    197  1.71        pk 			lock_printf("last unlocked at %s:%d\n",		\
    198  1.50   thorpej 			    lkp->lk_unlock_file, lkp->lk_unlock_line);	\
    199  1.63       chs 		SLOCK_TRACE();						\
    200  1.50   thorpej 		SPINLOCK_SPINCHECK_DEBUGGER;				\
    201  1.50   thorpej 	}								\
    202  1.66     perry } while (/*CONSTCOND*/ 0)
    203  1.50   thorpej #else
    204  1.50   thorpej #define	SPINLOCK_SPINCHECK_DECL			/* nothing */
    205  1.50   thorpej #define	SPINLOCK_SPINCHECK			/* nothing */
    206  1.50   thorpej #endif /* LOCKDEBUG && DDB */
    207  1.50   thorpej 
    208   1.1      fvdl /*
    209   1.1      fvdl  * Acquire a resource.
    210   1.1      fvdl  */
    211  1.73      yamt static int
    212  1.78   hannken acquire(__volatile struct lock **lkpp, int *s, int extflags,
    213  1.74   hannken     int drain, int wanted)
    214  1.73      yamt {
    215  1.73      yamt 	int error;
    216  1.78   hannken 	__volatile struct lock *lkp = *lkpp;
    217  1.73      yamt 
    218  1.73      yamt 	KASSERT(drain || (wanted & LK_WAIT_NONZERO) == 0);
    219  1.73      yamt 
    220  1.73      yamt 	if (extflags & LK_SPIN) {
    221  1.73      yamt 		int interlocked;
    222  1.73      yamt 
    223  1.73      yamt 		SPINLOCK_SPINCHECK_DECL;
    224  1.73      yamt 
    225  1.73      yamt 		if (!drain) {
    226  1.73      yamt 			lkp->lk_waitcount++;
    227  1.73      yamt 			lkp->lk_flags |= LK_WAIT_NONZERO;
    228  1.73      yamt 		}
    229  1.73      yamt 		for (interlocked = 1;;) {
    230  1.73      yamt 			SPINLOCK_SPINCHECK;
    231  1.73      yamt 			if ((lkp->lk_flags & wanted) != 0) {
    232  1.73      yamt 				if (interlocked) {
    233  1.74   hannken 					INTERLOCK_RELEASE(lkp, LK_SPIN, *s);
    234  1.73      yamt 					interlocked = 0;
    235  1.73      yamt 				}
    236  1.73      yamt 				SPINLOCK_SPIN_HOOK;
    237  1.73      yamt 			} else if (interlocked) {
    238  1.73      yamt 				break;
    239  1.73      yamt 			} else {
    240  1.74   hannken 				INTERLOCK_ACQUIRE(lkp, LK_SPIN, *s);
    241  1.73      yamt 				interlocked = 1;
    242  1.73      yamt 			}
    243  1.73      yamt 		}
    244  1.73      yamt 		if (!drain) {
    245  1.73      yamt 			lkp->lk_waitcount--;
    246  1.73      yamt 			if (lkp->lk_waitcount == 0)
    247  1.73      yamt 				lkp->lk_flags &= ~LK_WAIT_NONZERO;
    248  1.73      yamt 		}
    249  1.73      yamt 		KASSERT((lkp->lk_flags & wanted) == 0);
    250  1.73      yamt 		error = 0;	/* sanity */
    251  1.73      yamt 	} else {
    252  1.73      yamt 		for (error = 0; (lkp->lk_flags & wanted) != 0; ) {
    253  1.73      yamt 			if (drain)
    254  1.73      yamt 				lkp->lk_flags |= LK_WAITDRAIN;
    255  1.73      yamt 			else {
    256  1.73      yamt 				lkp->lk_waitcount++;
    257  1.73      yamt 				lkp->lk_flags |= LK_WAIT_NONZERO;
    258  1.73      yamt 			}
    259  1.73      yamt 			/* XXX Cast away volatile. */
    260  1.73      yamt 			error = ltsleep(drain ?
    261  1.73      yamt 			    (void *)&lkp->lk_flags :
    262  1.73      yamt 			    (void *)lkp, lkp->lk_prio,
    263  1.73      yamt 			    lkp->lk_wmesg, lkp->lk_timo, &lkp->lk_interlock);
    264  1.73      yamt 			if (!drain) {
    265  1.73      yamt 				lkp->lk_waitcount--;
    266  1.73      yamt 				if (lkp->lk_waitcount == 0)
    267  1.73      yamt 					lkp->lk_flags &= ~LK_WAIT_NONZERO;
    268  1.73      yamt 			}
    269  1.73      yamt 			if (error)
    270  1.73      yamt 				break;
    271  1.73      yamt 			if (extflags & LK_SLEEPFAIL) {
    272  1.73      yamt 				error = ENOLCK;
    273  1.73      yamt 				break;
    274  1.73      yamt 			}
    275  1.78   hannken 			if (lkp->lk_newlock != NULL) {
    276  1.78   hannken 				simple_lock(&lkp->lk_newlock->lk_interlock);
    277  1.78   hannken 				simple_unlock(&lkp->lk_interlock);
    278  1.78   hannken 				if (lkp->lk_waitcount == 0)
    279  1.78   hannken 					wakeup((void *)&lkp->lk_newlock);
    280  1.78   hannken 				*lkpp = lkp = lkp->lk_newlock;
    281  1.78   hannken 			}
    282  1.73      yamt 		}
    283   1.1      fvdl 	}
    284   1.1      fvdl 
    285  1.73      yamt 	return error;
    286  1.73      yamt }
    287  1.73      yamt 
    288  1.69   thorpej #define	SETHOLDER(lkp, pid, lid, cpu_id)				\
    289  1.19   thorpej do {									\
    290  1.19   thorpej 	if ((lkp)->lk_flags & LK_SPIN)					\
    291  1.19   thorpej 		(lkp)->lk_cpu = cpu_id;					\
    292  1.69   thorpej 	else {								\
    293  1.19   thorpej 		(lkp)->lk_lockholder = pid;				\
    294  1.69   thorpej 		(lkp)->lk_locklwp = lid;				\
    295  1.69   thorpej 	}								\
    296  1.30   thorpej } while (/*CONSTCOND*/0)
    297  1.19   thorpej 
    298  1.69   thorpej #define	WEHOLDIT(lkp, pid, lid, cpu_id)					\
    299  1.19   thorpej 	(((lkp)->lk_flags & LK_SPIN) != 0 ?				\
    300  1.69   thorpej 	 ((lkp)->lk_cpu == (cpu_id)) :					\
    301  1.69   thorpej 	 ((lkp)->lk_lockholder == (pid) && (lkp)->lk_locklwp == (lid)))
    302  1.19   thorpej 
    303  1.23   thorpej #define	WAKEUP_WAITER(lkp)						\
    304  1.23   thorpej do {									\
    305  1.73      yamt 	if (((lkp)->lk_flags & (LK_SPIN | LK_WAIT_NONZERO)) ==		\
    306  1.73      yamt 	    LK_WAIT_NONZERO) {						\
    307  1.23   thorpej 		/* XXX Cast away volatile. */				\
    308  1.59       chs 		wakeup((void *)(lkp));					\
    309  1.23   thorpej 	}								\
    310  1.30   thorpej } while (/*CONSTCOND*/0)
    311  1.23   thorpej 
    312  1.21   thorpej #if defined(LOCKDEBUG) /* { */
    313  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    314  1.21   thorpej struct simplelock spinlock_list_slock = SIMPLELOCK_INITIALIZER;
    315  1.21   thorpej 
    316  1.27   thorpej #define	SPINLOCK_LIST_LOCK()						\
    317  1.29  sommerfe 	__cpu_simple_lock(&spinlock_list_slock.lock_data)
    318  1.21   thorpej 
    319  1.27   thorpej #define	SPINLOCK_LIST_UNLOCK()						\
    320  1.29  sommerfe 	__cpu_simple_unlock(&spinlock_list_slock.lock_data)
    321  1.21   thorpej #else
    322  1.21   thorpej #define	SPINLOCK_LIST_LOCK()	/* nothing */
    323  1.21   thorpej 
    324  1.21   thorpej #define	SPINLOCK_LIST_UNLOCK()	/* nothing */
    325  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
    326  1.21   thorpej 
    327  1.21   thorpej TAILQ_HEAD(, lock) spinlock_list =
    328  1.21   thorpej     TAILQ_HEAD_INITIALIZER(spinlock_list);
    329  1.21   thorpej 
    330  1.21   thorpej #define	HAVEIT(lkp)							\
    331  1.21   thorpej do {									\
    332  1.21   thorpej 	if ((lkp)->lk_flags & LK_SPIN) {				\
    333  1.44   thorpej 		int s = spllock();					\
    334  1.21   thorpej 		SPINLOCK_LIST_LOCK();					\
    335  1.21   thorpej 		/* XXX Cast away volatile. */				\
    336  1.21   thorpej 		TAILQ_INSERT_TAIL(&spinlock_list, (struct lock *)(lkp),	\
    337  1.21   thorpej 		    lk_list);						\
    338  1.21   thorpej 		SPINLOCK_LIST_UNLOCK();					\
    339  1.21   thorpej 		splx(s);						\
    340  1.21   thorpej 	}								\
    341  1.30   thorpej } while (/*CONSTCOND*/0)
    342  1.21   thorpej 
    343  1.21   thorpej #define	DONTHAVEIT(lkp)							\
    344  1.21   thorpej do {									\
    345  1.21   thorpej 	if ((lkp)->lk_flags & LK_SPIN) {				\
    346  1.44   thorpej 		int s = spllock();					\
    347  1.21   thorpej 		SPINLOCK_LIST_LOCK();					\
    348  1.21   thorpej 		/* XXX Cast away volatile. */				\
    349  1.21   thorpej 		TAILQ_REMOVE(&spinlock_list, (struct lock *)(lkp),	\
    350  1.21   thorpej 		    lk_list);						\
    351  1.21   thorpej 		SPINLOCK_LIST_UNLOCK();					\
    352  1.21   thorpej 		splx(s);						\
    353  1.21   thorpej 	}								\
    354  1.30   thorpej } while (/*CONSTCOND*/0)
    355  1.21   thorpej #else
    356  1.21   thorpej #define	HAVEIT(lkp)		/* nothing */
    357  1.21   thorpej 
    358  1.21   thorpej #define	DONTHAVEIT(lkp)		/* nothing */
    359  1.21   thorpej #endif /* LOCKDEBUG */ /* } */
    360  1.21   thorpej 
    361  1.25   thorpej #if defined(LOCKDEBUG)
    362  1.25   thorpej /*
    363  1.25   thorpej  * Lock debug printing routine; can be configured to print to console
    364  1.25   thorpej  * or log to syslog.
    365  1.25   thorpej  */
    366  1.25   thorpej void
    367  1.25   thorpej lock_printf(const char *fmt, ...)
    368  1.25   thorpej {
    369  1.68        pk 	char b[150];
    370  1.25   thorpej 	va_list ap;
    371  1.25   thorpej 
    372  1.25   thorpej 	va_start(ap, fmt);
    373  1.25   thorpej 	if (lock_debug_syslog)
    374  1.25   thorpej 		vlog(LOG_DEBUG, fmt, ap);
    375  1.68        pk 	else {
    376  1.68        pk 		vsnprintf(b, sizeof(b), fmt, ap);
    377  1.68        pk 		printf_nolog("%s", b);
    378  1.68        pk 	}
    379  1.25   thorpej 	va_end(ap);
    380  1.25   thorpej }
    381  1.25   thorpej #endif /* LOCKDEBUG */
    382  1.25   thorpej 
    383   1.1      fvdl /*
    384  1.78   hannken  * Transfer any waiting processes from one lock to another.
    385  1.78   hannken  */
    386  1.78   hannken void
    387  1.78   hannken transferlockers(struct lock *from, struct lock *to)
    388  1.78   hannken {
    389  1.78   hannken 
    390  1.78   hannken 	KASSERT(from != to);
    391  1.78   hannken 	KASSERT((from->lk_flags & LK_WAITDRAIN) == 0);
    392  1.78   hannken 	if (from->lk_waitcount == 0)
    393  1.78   hannken 		return;
    394  1.78   hannken 	from->lk_newlock = to;
    395  1.78   hannken 	wakeup((void *)from);
    396  1.78   hannken 	tsleep((void *)&from->lk_newlock, from->lk_prio, "lkxfer", 0);
    397  1.78   hannken 	from->lk_newlock = NULL;
    398  1.78   hannken 	from->lk_flags &= ~(LK_WANT_EXCL | LK_WANT_UPGRADE);
    399  1.78   hannken 	KASSERT(from->lk_waitcount == 0);
    400  1.78   hannken }
    401  1.78   hannken 
    402  1.78   hannken 
    403  1.78   hannken /*
    404   1.1      fvdl  * Initialize a lock; required before use.
    405   1.1      fvdl  */
    406   1.1      fvdl void
    407  1.33   thorpej lockinit(struct lock *lkp, int prio, const char *wmesg, int timo, int flags)
    408   1.1      fvdl {
    409   1.1      fvdl 
    410   1.8     perry 	memset(lkp, 0, sizeof(struct lock));
    411   1.1      fvdl 	simple_lock_init(&lkp->lk_interlock);
    412   1.1      fvdl 	lkp->lk_flags = flags & LK_EXTFLG_MASK;
    413  1.19   thorpej 	if (flags & LK_SPIN)
    414  1.19   thorpej 		lkp->lk_cpu = LK_NOCPU;
    415  1.19   thorpej 	else {
    416  1.19   thorpej 		lkp->lk_lockholder = LK_NOPROC;
    417  1.78   hannken 		lkp->lk_newlock = NULL;
    418  1.19   thorpej 		lkp->lk_prio = prio;
    419  1.19   thorpej 		lkp->lk_timo = timo;
    420  1.19   thorpej 	}
    421  1.19   thorpej 	lkp->lk_wmesg = wmesg;	/* just a name for spin locks */
    422  1.50   thorpej #if defined(LOCKDEBUG)
    423  1.50   thorpej 	lkp->lk_lock_file = NULL;
    424  1.50   thorpej 	lkp->lk_unlock_file = NULL;
    425  1.50   thorpej #endif
    426   1.1      fvdl }
    427   1.1      fvdl 
    428   1.1      fvdl /*
    429   1.1      fvdl  * Determine the status of a lock.
    430   1.1      fvdl  */
    431   1.1      fvdl int
    432  1.33   thorpej lockstatus(struct lock *lkp)
    433   1.1      fvdl {
    434  1.76      yamt 	int s = 0; /* XXX: gcc */
    435  1.76      yamt 	int lock_type = 0;
    436  1.76      yamt 	struct lwp *l = curlwp; /* XXX */
    437  1.76      yamt 	pid_t pid;
    438  1.76      yamt 	lwpid_t lid;
    439  1.76      yamt 	cpuid_t cpu_id;
    440  1.76      yamt 
    441  1.76      yamt 	if ((lkp->lk_flags & LK_SPIN) || l == NULL) {
    442  1.76      yamt 		cpu_id = cpu_number();
    443  1.76      yamt 		pid = LK_KERNPROC;
    444  1.76      yamt 		lid = 0;
    445  1.76      yamt 	} else {
    446  1.76      yamt 		cpu_id = LK_NOCPU;
    447  1.76      yamt 		pid = l->l_proc->p_pid;
    448  1.76      yamt 		lid = l->l_lid;
    449  1.76      yamt 	}
    450   1.1      fvdl 
    451  1.43   thorpej 	INTERLOCK_ACQUIRE(lkp, lkp->lk_flags, s);
    452  1.76      yamt 	if (lkp->lk_exclusivecount != 0) {
    453  1.76      yamt 		if (WEHOLDIT(lkp, pid, lid, cpu_id))
    454  1.76      yamt 			lock_type = LK_EXCLUSIVE;
    455  1.76      yamt 		else
    456  1.76      yamt 			lock_type = LK_EXCLOTHER;
    457  1.76      yamt 	} else if (lkp->lk_sharecount != 0)
    458   1.1      fvdl 		lock_type = LK_SHARED;
    459  1.43   thorpej 	INTERLOCK_RELEASE(lkp, lkp->lk_flags, s);
    460   1.1      fvdl 	return (lock_type);
    461   1.1      fvdl }
    462  1.35   thorpej 
    463  1.35   thorpej #if defined(LOCKDEBUG) || defined(DIAGNOSTIC)
    464  1.35   thorpej /*
    465  1.35   thorpej  * Make sure no spin locks are held by a CPU that is about
    466  1.35   thorpej  * to context switch.
    467  1.35   thorpej  */
    468  1.35   thorpej void
    469  1.35   thorpej spinlock_switchcheck(void)
    470  1.35   thorpej {
    471  1.35   thorpej 	u_long cnt;
    472  1.35   thorpej 	int s;
    473  1.35   thorpej 
    474  1.44   thorpej 	s = spllock();
    475  1.35   thorpej #if defined(MULTIPROCESSOR)
    476  1.35   thorpej 	cnt = curcpu()->ci_spin_locks;
    477  1.35   thorpej #else
    478  1.35   thorpej 	cnt = spin_locks;
    479  1.35   thorpej #endif
    480  1.35   thorpej 	splx(s);
    481  1.35   thorpej 
    482  1.35   thorpej 	if (cnt != 0)
    483  1.35   thorpej 		panic("spinlock_switchcheck: CPU %lu has %lu spin locks",
    484  1.35   thorpej 		    (u_long) cpu_number(), cnt);
    485  1.35   thorpej }
    486  1.35   thorpej #endif /* LOCKDEBUG || DIAGNOSTIC */
    487   1.1      fvdl 
    488   1.1      fvdl /*
    489  1.44   thorpej  * Locks and IPLs (interrupt priority levels):
    490  1.44   thorpej  *
    491  1.44   thorpej  * Locks which may be taken from interrupt context must be handled
    492  1.44   thorpej  * very carefully; you must spl to the highest IPL where the lock
    493  1.44   thorpej  * is needed before acquiring the lock.
    494  1.44   thorpej  *
    495  1.44   thorpej  * It is also important to avoid deadlock, since certain (very high
    496  1.44   thorpej  * priority) interrupts are often needed to keep the system as a whole
    497  1.44   thorpej  * from deadlocking, and must not be blocked while you are spinning
    498  1.44   thorpej  * waiting for a lower-priority lock.
    499  1.44   thorpej  *
    500  1.44   thorpej  * In addition, the lock-debugging hooks themselves need to use locks!
    501  1.44   thorpej  *
    502  1.44   thorpej  * A raw __cpu_simple_lock may be used from interrupts are long as it
    503  1.44   thorpej  * is acquired and held at a single IPL.
    504  1.44   thorpej  *
    505  1.44   thorpej  * A simple_lock (which is a __cpu_simple_lock wrapped with some
    506  1.44   thorpej  * debugging hooks) may be used at or below spllock(), which is
    507  1.44   thorpej  * typically at or just below splhigh() (i.e. blocks everything
    508  1.44   thorpej  * but certain machine-dependent extremely high priority interrupts).
    509  1.44   thorpej  *
    510  1.44   thorpej  * spinlockmgr spinlocks should be used at or below splsched().
    511  1.44   thorpej  *
    512  1.44   thorpej  * Some platforms may have interrupts of higher priority than splsched(),
    513  1.44   thorpej  * including hard serial interrupts, inter-processor interrupts, and
    514  1.44   thorpej  * kernel debugger traps.
    515  1.44   thorpej  */
    516  1.44   thorpej 
    517  1.44   thorpej /*
    518  1.32  sommerfe  * XXX XXX kludge around another kludge..
    519  1.32  sommerfe  *
    520  1.32  sommerfe  * vfs_shutdown() may be called from interrupt context, either as a result
    521  1.32  sommerfe  * of a panic, or from the debugger.   It proceeds to call
    522  1.32  sommerfe  * sys_sync(&proc0, ...), pretending its running on behalf of proc0
    523  1.32  sommerfe  *
    524  1.32  sommerfe  * We would like to make an attempt to sync the filesystems in this case, so
    525  1.32  sommerfe  * if this happens, we treat attempts to acquire locks specially.
    526  1.32  sommerfe  * All locks are acquired on behalf of proc0.
    527  1.32  sommerfe  *
    528  1.32  sommerfe  * If we've already paniced, we don't block waiting for locks, but
    529  1.32  sommerfe  * just barge right ahead since we're already going down in flames.
    530  1.32  sommerfe  */
    531  1.32  sommerfe 
    532  1.32  sommerfe /*
    533   1.1      fvdl  * Set, change, or release a lock.
    534   1.1      fvdl  *
    535   1.1      fvdl  * Shared requests increment the shared count. Exclusive requests set the
    536   1.1      fvdl  * LK_WANT_EXCL flag (preventing further shared locks), and wait for already
    537   1.1      fvdl  * accepted shared locks and shared-to-exclusive upgrades to go away.
    538   1.1      fvdl  */
    539   1.1      fvdl int
    540  1.50   thorpej #if defined(LOCKDEBUG)
    541  1.50   thorpej _lockmgr(__volatile struct lock *lkp, u_int flags,
    542  1.50   thorpej     struct simplelock *interlkp, const char *file, int line)
    543  1.50   thorpej #else
    544  1.33   thorpej lockmgr(__volatile struct lock *lkp, u_int flags,
    545  1.33   thorpej     struct simplelock *interlkp)
    546  1.50   thorpej #endif
    547   1.1      fvdl {
    548   1.1      fvdl 	int error;
    549   1.1      fvdl 	pid_t pid;
    550  1.69   thorpej 	lwpid_t lid;
    551   1.1      fvdl 	int extflags;
    552  1.24   thorpej 	cpuid_t cpu_id;
    553  1.69   thorpej 	struct lwp *l = curlwp;
    554  1.32  sommerfe 	int lock_shutdown_noblock = 0;
    555  1.67       scw 	int s = 0;
    556   1.1      fvdl 
    557   1.1      fvdl 	error = 0;
    558  1.19   thorpej 
    559  1.80      yamt 	/* LK_RETRY is for vn_lock, not for lockmgr. */
    560  1.79      yamt 	KASSERT((flags & LK_RETRY) == 0);
    561  1.79      yamt 
    562  1.43   thorpej 	INTERLOCK_ACQUIRE(lkp, lkp->lk_flags, s);
    563   1.1      fvdl 	if (flags & LK_INTERLOCK)
    564   1.1      fvdl 		simple_unlock(interlkp);
    565   1.1      fvdl 	extflags = (flags | lkp->lk_flags) & LK_EXTFLG_MASK;
    566  1.19   thorpej 
    567  1.21   thorpej #ifdef DIAGNOSTIC /* { */
    568  1.19   thorpej 	/*
    569  1.19   thorpej 	 * Don't allow spins on sleep locks and don't allow sleeps
    570  1.19   thorpej 	 * on spin locks.
    571  1.19   thorpej 	 */
    572  1.19   thorpej 	if ((flags ^ lkp->lk_flags) & LK_SPIN)
    573  1.64    provos 		panic("lockmgr: sleep/spin mismatch");
    574  1.21   thorpej #endif /* } */
    575  1.19   thorpej 
    576  1.69   thorpej 	if (extflags & LK_SPIN) {
    577  1.19   thorpej 		pid = LK_KERNPROC;
    578  1.69   thorpej 		lid = 0;
    579  1.69   thorpej 	} else {
    580  1.69   thorpej 		if (l == NULL) {
    581  1.32  sommerfe 			if (!doing_shutdown) {
    582  1.32  sommerfe 				panic("lockmgr: no context");
    583  1.32  sommerfe 			} else {
    584  1.69   thorpej 				l = &lwp0;
    585  1.32  sommerfe 				if (panicstr && (!(flags & LK_NOWAIT))) {
    586  1.32  sommerfe 					flags |= LK_NOWAIT;
    587  1.32  sommerfe 					lock_shutdown_noblock = 1;
    588  1.32  sommerfe 				}
    589  1.32  sommerfe 			}
    590  1.32  sommerfe 		}
    591  1.69   thorpej 		lid = l->l_lid;
    592  1.69   thorpej 		pid = l->l_proc->p_pid;
    593  1.19   thorpej 	}
    594  1.24   thorpej 	cpu_id = cpu_number();
    595  1.19   thorpej 
    596   1.1      fvdl 	/*
    597   1.1      fvdl 	 * Once a lock has drained, the LK_DRAINING flag is set and an
    598   1.1      fvdl 	 * exclusive lock is returned. The only valid operation thereafter
    599   1.1      fvdl 	 * is a single release of that exclusive lock. This final release
    600   1.1      fvdl 	 * clears the LK_DRAINING flag and sets the LK_DRAINED flag. Any
    601   1.1      fvdl 	 * further requests of any sort will result in a panic. The bits
    602   1.1      fvdl 	 * selected for these two flags are chosen so that they will be set
    603   1.1      fvdl 	 * in memory that is freed (freed memory is filled with 0xdeadbeef).
    604   1.1      fvdl 	 * The final release is permitted to give a new lease on life to
    605   1.1      fvdl 	 * the lock by specifying LK_REENABLE.
    606   1.1      fvdl 	 */
    607   1.1      fvdl 	if (lkp->lk_flags & (LK_DRAINING|LK_DRAINED)) {
    608  1.28   thorpej #ifdef DIAGNOSTIC /* { */
    609   1.1      fvdl 		if (lkp->lk_flags & LK_DRAINED)
    610   1.1      fvdl 			panic("lockmgr: using decommissioned lock");
    611   1.1      fvdl 		if ((flags & LK_TYPE_MASK) != LK_RELEASE ||
    612  1.69   thorpej 		    WEHOLDIT(lkp, pid, lid, cpu_id) == 0)
    613  1.64    provos 			panic("lockmgr: non-release on draining lock: %d",
    614   1.1      fvdl 			    flags & LK_TYPE_MASK);
    615  1.28   thorpej #endif /* DIAGNOSTIC */ /* } */
    616   1.1      fvdl 		lkp->lk_flags &= ~LK_DRAINING;
    617   1.1      fvdl 		if ((flags & LK_REENABLE) == 0)
    618   1.1      fvdl 			lkp->lk_flags |= LK_DRAINED;
    619   1.1      fvdl 	}
    620   1.1      fvdl 
    621   1.1      fvdl 	switch (flags & LK_TYPE_MASK) {
    622   1.1      fvdl 
    623   1.1      fvdl 	case LK_SHARED:
    624  1.69   thorpej 		if (WEHOLDIT(lkp, pid, lid, cpu_id) == 0) {
    625   1.1      fvdl 			/*
    626   1.1      fvdl 			 * If just polling, check to see if we will block.
    627   1.1      fvdl 			 */
    628   1.1      fvdl 			if ((extflags & LK_NOWAIT) && (lkp->lk_flags &
    629   1.1      fvdl 			    (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE))) {
    630   1.1      fvdl 				error = EBUSY;
    631   1.1      fvdl 				break;
    632   1.1      fvdl 			}
    633   1.1      fvdl 			/*
    634   1.1      fvdl 			 * Wait for exclusive locks and upgrades to clear.
    635   1.1      fvdl 			 */
    636  1.78   hannken 			error = acquire(&lkp, &s, extflags, 0,
    637  1.73      yamt 			    LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE);
    638   1.1      fvdl 			if (error)
    639   1.1      fvdl 				break;
    640   1.1      fvdl 			lkp->lk_sharecount++;
    641  1.73      yamt 			lkp->lk_flags |= LK_SHARE_NONZERO;
    642  1.69   thorpej 			COUNT(lkp, l, cpu_id, 1);
    643   1.1      fvdl 			break;
    644   1.1      fvdl 		}
    645   1.1      fvdl 		/*
    646   1.1      fvdl 		 * We hold an exclusive lock, so downgrade it to shared.
    647   1.1      fvdl 		 * An alternative would be to fail with EDEADLK.
    648   1.1      fvdl 		 */
    649   1.1      fvdl 		lkp->lk_sharecount++;
    650  1.73      yamt 		lkp->lk_flags |= LK_SHARE_NONZERO;
    651  1.69   thorpej 		COUNT(lkp, l, cpu_id, 1);
    652   1.1      fvdl 		/* fall into downgrade */
    653   1.1      fvdl 
    654   1.1      fvdl 	case LK_DOWNGRADE:
    655  1.69   thorpej 		if (WEHOLDIT(lkp, pid, lid, cpu_id) == 0 ||
    656  1.19   thorpej 		    lkp->lk_exclusivecount == 0)
    657   1.1      fvdl 			panic("lockmgr: not holding exclusive lock");
    658   1.1      fvdl 		lkp->lk_sharecount += lkp->lk_exclusivecount;
    659  1.73      yamt 		lkp->lk_flags |= LK_SHARE_NONZERO;
    660   1.1      fvdl 		lkp->lk_exclusivecount = 0;
    661  1.15      fvdl 		lkp->lk_recurselevel = 0;
    662   1.1      fvdl 		lkp->lk_flags &= ~LK_HAVE_EXCL;
    663  1.69   thorpej 		SETHOLDER(lkp, LK_NOPROC, 0, LK_NOCPU);
    664  1.50   thorpej #if defined(LOCKDEBUG)
    665  1.50   thorpej 		lkp->lk_unlock_file = file;
    666  1.50   thorpej 		lkp->lk_unlock_line = line;
    667  1.50   thorpej #endif
    668  1.21   thorpej 		DONTHAVEIT(lkp);
    669  1.23   thorpej 		WAKEUP_WAITER(lkp);
    670   1.1      fvdl 		break;
    671   1.1      fvdl 
    672   1.1      fvdl 	case LK_EXCLUPGRADE:
    673   1.1      fvdl 		/*
    674   1.1      fvdl 		 * If another process is ahead of us to get an upgrade,
    675   1.1      fvdl 		 * then we want to fail rather than have an intervening
    676   1.1      fvdl 		 * exclusive access.
    677   1.1      fvdl 		 */
    678   1.1      fvdl 		if (lkp->lk_flags & LK_WANT_UPGRADE) {
    679   1.1      fvdl 			lkp->lk_sharecount--;
    680  1.73      yamt 			if (lkp->lk_sharecount == 0)
    681  1.73      yamt 				lkp->lk_flags &= ~LK_SHARE_NONZERO;
    682  1.69   thorpej 			COUNT(lkp, l, cpu_id, -1);
    683   1.1      fvdl 			error = EBUSY;
    684   1.1      fvdl 			break;
    685   1.1      fvdl 		}
    686   1.1      fvdl 		/* fall into normal upgrade */
    687   1.1      fvdl 
    688   1.1      fvdl 	case LK_UPGRADE:
    689   1.1      fvdl 		/*
    690   1.1      fvdl 		 * Upgrade a shared lock to an exclusive one. If another
    691   1.1      fvdl 		 * shared lock has already requested an upgrade to an
    692   1.1      fvdl 		 * exclusive lock, our shared lock is released and an
    693   1.1      fvdl 		 * exclusive lock is requested (which will be granted
    694   1.1      fvdl 		 * after the upgrade). If we return an error, the file
    695   1.1      fvdl 		 * will always be unlocked.
    696   1.1      fvdl 		 */
    697  1.69   thorpej 		if (WEHOLDIT(lkp, pid, lid, cpu_id) || lkp->lk_sharecount <= 0)
    698   1.1      fvdl 			panic("lockmgr: upgrade exclusive lock");
    699   1.1      fvdl 		lkp->lk_sharecount--;
    700  1.73      yamt 		if (lkp->lk_sharecount == 0)
    701  1.73      yamt 			lkp->lk_flags &= ~LK_SHARE_NONZERO;
    702  1.69   thorpej 		COUNT(lkp, l, cpu_id, -1);
    703   1.1      fvdl 		/*
    704   1.1      fvdl 		 * If we are just polling, check to see if we will block.
    705   1.1      fvdl 		 */
    706   1.1      fvdl 		if ((extflags & LK_NOWAIT) &&
    707   1.1      fvdl 		    ((lkp->lk_flags & LK_WANT_UPGRADE) ||
    708   1.1      fvdl 		     lkp->lk_sharecount > 1)) {
    709   1.1      fvdl 			error = EBUSY;
    710   1.1      fvdl 			break;
    711   1.1      fvdl 		}
    712   1.1      fvdl 		if ((lkp->lk_flags & LK_WANT_UPGRADE) == 0) {
    713   1.1      fvdl 			/*
    714   1.1      fvdl 			 * We are first shared lock to request an upgrade, so
    715   1.1      fvdl 			 * request upgrade and wait for the shared count to
    716   1.1      fvdl 			 * drop to zero, then take exclusive lock.
    717   1.1      fvdl 			 */
    718   1.1      fvdl 			lkp->lk_flags |= LK_WANT_UPGRADE;
    719  1.78   hannken 			error = acquire(&lkp, &s, extflags, 0, LK_SHARE_NONZERO);
    720   1.1      fvdl 			lkp->lk_flags &= ~LK_WANT_UPGRADE;
    721   1.1      fvdl 			if (error)
    722   1.1      fvdl 				break;
    723   1.1      fvdl 			lkp->lk_flags |= LK_HAVE_EXCL;
    724  1.69   thorpej 			SETHOLDER(lkp, pid, lid, cpu_id);
    725  1.50   thorpej #if defined(LOCKDEBUG)
    726  1.50   thorpej 			lkp->lk_lock_file = file;
    727  1.50   thorpej 			lkp->lk_lock_line = line;
    728  1.50   thorpej #endif
    729  1.21   thorpej 			HAVEIT(lkp);
    730   1.1      fvdl 			if (lkp->lk_exclusivecount != 0)
    731   1.1      fvdl 				panic("lockmgr: non-zero exclusive count");
    732   1.1      fvdl 			lkp->lk_exclusivecount = 1;
    733  1.15      fvdl 			if (extflags & LK_SETRECURSE)
    734  1.15      fvdl 				lkp->lk_recurselevel = 1;
    735  1.69   thorpej 			COUNT(lkp, l, cpu_id, 1);
    736   1.1      fvdl 			break;
    737   1.1      fvdl 		}
    738   1.1      fvdl 		/*
    739   1.1      fvdl 		 * Someone else has requested upgrade. Release our shared
    740   1.1      fvdl 		 * lock, awaken upgrade requestor if we are the last shared
    741   1.1      fvdl 		 * lock, then request an exclusive lock.
    742   1.1      fvdl 		 */
    743  1.23   thorpej 		if (lkp->lk_sharecount == 0)
    744  1.23   thorpej 			WAKEUP_WAITER(lkp);
    745   1.1      fvdl 		/* fall into exclusive request */
    746   1.1      fvdl 
    747   1.1      fvdl 	case LK_EXCLUSIVE:
    748  1.69   thorpej 		if (WEHOLDIT(lkp, pid, lid, cpu_id)) {
    749   1.1      fvdl 			/*
    750  1.19   thorpej 			 * Recursive lock.
    751   1.1      fvdl 			 */
    752  1.15      fvdl 			if ((extflags & LK_CANRECURSE) == 0 &&
    753  1.16  sommerfe 			     lkp->lk_recurselevel == 0) {
    754  1.16  sommerfe 				if (extflags & LK_RECURSEFAIL) {
    755  1.16  sommerfe 					error = EDEADLK;
    756  1.16  sommerfe 					break;
    757  1.16  sommerfe 				} else
    758  1.16  sommerfe 					panic("lockmgr: locking against myself");
    759  1.16  sommerfe 			}
    760   1.1      fvdl 			lkp->lk_exclusivecount++;
    761  1.15      fvdl 			if (extflags & LK_SETRECURSE &&
    762  1.15      fvdl 			    lkp->lk_recurselevel == 0)
    763  1.15      fvdl 				lkp->lk_recurselevel = lkp->lk_exclusivecount;
    764  1.69   thorpej 			COUNT(lkp, l, cpu_id, 1);
    765   1.1      fvdl 			break;
    766   1.1      fvdl 		}
    767   1.1      fvdl 		/*
    768   1.1      fvdl 		 * If we are just polling, check to see if we will sleep.
    769   1.1      fvdl 		 */
    770  1.73      yamt 		if ((extflags & LK_NOWAIT) && (lkp->lk_flags &
    771  1.73      yamt 		     (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE |
    772  1.73      yamt 		     LK_SHARE_NONZERO))) {
    773   1.1      fvdl 			error = EBUSY;
    774   1.1      fvdl 			break;
    775   1.1      fvdl 		}
    776   1.1      fvdl 		/*
    777   1.1      fvdl 		 * Try to acquire the want_exclusive flag.
    778   1.1      fvdl 		 */
    779  1.82      yamt 		error = acquire(&lkp, &s, extflags, 0,
    780  1.82      yamt 		    LK_HAVE_EXCL | LK_WANT_EXCL);
    781   1.1      fvdl 		if (error)
    782   1.1      fvdl 			break;
    783   1.1      fvdl 		lkp->lk_flags |= LK_WANT_EXCL;
    784   1.1      fvdl 		/*
    785   1.1      fvdl 		 * Wait for shared locks and upgrades to finish.
    786   1.1      fvdl 		 */
    787  1.78   hannken 		error = acquire(&lkp, &s, extflags, 0,
    788  1.81      yamt 		    LK_HAVE_EXCL | LK_WANT_UPGRADE | LK_SHARE_NONZERO);
    789   1.1      fvdl 		lkp->lk_flags &= ~LK_WANT_EXCL;
    790   1.1      fvdl 		if (error)
    791   1.1      fvdl 			break;
    792   1.1      fvdl 		lkp->lk_flags |= LK_HAVE_EXCL;
    793  1.69   thorpej 		SETHOLDER(lkp, pid, lid, cpu_id);
    794  1.50   thorpej #if defined(LOCKDEBUG)
    795  1.50   thorpej 		lkp->lk_lock_file = file;
    796  1.50   thorpej 		lkp->lk_lock_line = line;
    797  1.50   thorpej #endif
    798  1.21   thorpej 		HAVEIT(lkp);
    799   1.1      fvdl 		if (lkp->lk_exclusivecount != 0)
    800   1.1      fvdl 			panic("lockmgr: non-zero exclusive count");
    801   1.1      fvdl 		lkp->lk_exclusivecount = 1;
    802  1.15      fvdl 		if (extflags & LK_SETRECURSE)
    803  1.15      fvdl 			lkp->lk_recurselevel = 1;
    804  1.69   thorpej 		COUNT(lkp, l, cpu_id, 1);
    805   1.1      fvdl 		break;
    806   1.1      fvdl 
    807   1.1      fvdl 	case LK_RELEASE:
    808   1.1      fvdl 		if (lkp->lk_exclusivecount != 0) {
    809  1.69   thorpej 			if (WEHOLDIT(lkp, pid, lid, cpu_id) == 0) {
    810  1.19   thorpej 				if (lkp->lk_flags & LK_SPIN) {
    811  1.19   thorpej 					panic("lockmgr: processor %lu, not "
    812  1.19   thorpej 					    "exclusive lock holder %lu "
    813  1.19   thorpej 					    "unlocking", cpu_id, lkp->lk_cpu);
    814  1.19   thorpej 				} else {
    815  1.19   thorpej 					panic("lockmgr: pid %d, not "
    816  1.19   thorpej 					    "exclusive lock holder %d "
    817  1.19   thorpej 					    "unlocking", pid,
    818  1.19   thorpej 					    lkp->lk_lockholder);
    819  1.19   thorpej 				}
    820  1.19   thorpej 			}
    821  1.15      fvdl 			if (lkp->lk_exclusivecount == lkp->lk_recurselevel)
    822  1.15      fvdl 				lkp->lk_recurselevel = 0;
    823   1.1      fvdl 			lkp->lk_exclusivecount--;
    824  1.69   thorpej 			COUNT(lkp, l, cpu_id, -1);
    825   1.1      fvdl 			if (lkp->lk_exclusivecount == 0) {
    826   1.1      fvdl 				lkp->lk_flags &= ~LK_HAVE_EXCL;
    827  1.69   thorpej 				SETHOLDER(lkp, LK_NOPROC, 0, LK_NOCPU);
    828  1.50   thorpej #if defined(LOCKDEBUG)
    829  1.50   thorpej 				lkp->lk_unlock_file = file;
    830  1.50   thorpej 				lkp->lk_unlock_line = line;
    831  1.50   thorpej #endif
    832  1.21   thorpej 				DONTHAVEIT(lkp);
    833   1.1      fvdl 			}
    834   1.1      fvdl 		} else if (lkp->lk_sharecount != 0) {
    835   1.1      fvdl 			lkp->lk_sharecount--;
    836  1.73      yamt 			if (lkp->lk_sharecount == 0)
    837  1.73      yamt 				lkp->lk_flags &= ~LK_SHARE_NONZERO;
    838  1.69   thorpej 			COUNT(lkp, l, cpu_id, -1);
    839   1.1      fvdl 		}
    840  1.39   thorpej #ifdef DIAGNOSTIC
    841  1.39   thorpej 		else
    842  1.39   thorpej 			panic("lockmgr: release of unlocked lock!");
    843  1.39   thorpej #endif
    844  1.23   thorpej 		WAKEUP_WAITER(lkp);
    845   1.1      fvdl 		break;
    846   1.1      fvdl 
    847   1.1      fvdl 	case LK_DRAIN:
    848   1.1      fvdl 		/*
    849   1.1      fvdl 		 * Check that we do not already hold the lock, as it can
    850   1.1      fvdl 		 * never drain if we do. Unfortunately, we have no way to
    851   1.1      fvdl 		 * check for holding a shared lock, but at least we can
    852   1.1      fvdl 		 * check for an exclusive one.
    853   1.1      fvdl 		 */
    854  1.69   thorpej 		if (WEHOLDIT(lkp, pid, lid, cpu_id))
    855   1.1      fvdl 			panic("lockmgr: draining against myself");
    856   1.1      fvdl 		/*
    857   1.1      fvdl 		 * If we are just polling, check to see if we will sleep.
    858   1.1      fvdl 		 */
    859  1.73      yamt 		if ((extflags & LK_NOWAIT) && (lkp->lk_flags &
    860  1.73      yamt 		     (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE |
    861  1.73      yamt 		     LK_SHARE_NONZERO | LK_WAIT_NONZERO))) {
    862   1.1      fvdl 			error = EBUSY;
    863   1.1      fvdl 			break;
    864   1.1      fvdl 		}
    865  1.78   hannken 		error = acquire(&lkp, &s, extflags, 1,
    866  1.73      yamt 		    LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE |
    867  1.73      yamt 		    LK_SHARE_NONZERO | LK_WAIT_NONZERO);
    868  1.23   thorpej 		if (error)
    869  1.23   thorpej 			break;
    870   1.1      fvdl 		lkp->lk_flags |= LK_DRAINING | LK_HAVE_EXCL;
    871  1.69   thorpej 		SETHOLDER(lkp, pid, lid, cpu_id);
    872  1.50   thorpej #if defined(LOCKDEBUG)
    873  1.50   thorpej 		lkp->lk_lock_file = file;
    874  1.50   thorpej 		lkp->lk_lock_line = line;
    875  1.50   thorpej #endif
    876  1.21   thorpej 		HAVEIT(lkp);
    877   1.1      fvdl 		lkp->lk_exclusivecount = 1;
    878  1.15      fvdl 		/* XXX unlikely that we'd want this */
    879  1.15      fvdl 		if (extflags & LK_SETRECURSE)
    880  1.15      fvdl 			lkp->lk_recurselevel = 1;
    881  1.69   thorpej 		COUNT(lkp, l, cpu_id, 1);
    882   1.1      fvdl 		break;
    883   1.1      fvdl 
    884   1.1      fvdl 	default:
    885  1.43   thorpej 		INTERLOCK_RELEASE(lkp, lkp->lk_flags, s);
    886   1.1      fvdl 		panic("lockmgr: unknown locktype request %d",
    887   1.1      fvdl 		    flags & LK_TYPE_MASK);
    888   1.1      fvdl 		/* NOTREACHED */
    889   1.1      fvdl 	}
    890  1.23   thorpej 	if ((lkp->lk_flags & (LK_WAITDRAIN|LK_SPIN)) == LK_WAITDRAIN &&
    891  1.23   thorpej 	    ((lkp->lk_flags &
    892  1.73      yamt 	      (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE |
    893  1.73      yamt 	      LK_SHARE_NONZERO | LK_WAIT_NONZERO)) == 0)) {
    894   1.1      fvdl 		lkp->lk_flags &= ~LK_WAITDRAIN;
    895  1.59       chs 		wakeup((void *)&lkp->lk_flags);
    896   1.1      fvdl 	}
    897  1.32  sommerfe 	/*
    898  1.32  sommerfe 	 * Note that this panic will be a recursive panic, since
    899  1.32  sommerfe 	 * we only set lock_shutdown_noblock above if panicstr != NULL.
    900  1.32  sommerfe 	 */
    901  1.32  sommerfe 	if (error && lock_shutdown_noblock)
    902  1.32  sommerfe 		panic("lockmgr: deadlock (see previous panic)");
    903  1.32  sommerfe 
    904  1.43   thorpej 	INTERLOCK_RELEASE(lkp, lkp->lk_flags, s);
    905   1.1      fvdl 	return (error);
    906   1.1      fvdl }
    907   1.1      fvdl 
    908   1.1      fvdl /*
    909  1.47  sommerfe  * For a recursive spinlock held one or more times by the current CPU,
    910  1.47  sommerfe  * release all N locks, and return N.
    911  1.47  sommerfe  * Intended for use in mi_switch() shortly before context switching.
    912  1.47  sommerfe  */
    913  1.47  sommerfe 
    914  1.47  sommerfe int
    915  1.50   thorpej #if defined(LOCKDEBUG)
    916  1.50   thorpej _spinlock_release_all(__volatile struct lock *lkp, const char *file, int line)
    917  1.50   thorpej #else
    918  1.47  sommerfe spinlock_release_all(__volatile struct lock *lkp)
    919  1.50   thorpej #endif
    920  1.47  sommerfe {
    921  1.47  sommerfe 	int s, count;
    922  1.47  sommerfe 	cpuid_t cpu_id;
    923  1.47  sommerfe 
    924  1.47  sommerfe 	KASSERT(lkp->lk_flags & LK_SPIN);
    925  1.47  sommerfe 
    926  1.47  sommerfe 	INTERLOCK_ACQUIRE(lkp, LK_SPIN, s);
    927  1.47  sommerfe 
    928  1.47  sommerfe 	cpu_id = cpu_number();
    929  1.47  sommerfe 	count = lkp->lk_exclusivecount;
    930  1.47  sommerfe 
    931  1.47  sommerfe 	if (count != 0) {
    932  1.47  sommerfe #ifdef DIAGNOSTIC
    933  1.69   thorpej 		if (WEHOLDIT(lkp, 0, 0, cpu_id) == 0) {
    934  1.47  sommerfe 			panic("spinlock_release_all: processor %lu, not "
    935  1.47  sommerfe 			    "exclusive lock holder %lu "
    936  1.47  sommerfe 			    "unlocking", (long)cpu_id, lkp->lk_cpu);
    937  1.47  sommerfe 		}
    938  1.47  sommerfe #endif
    939  1.47  sommerfe 		lkp->lk_recurselevel = 0;
    940  1.47  sommerfe 		lkp->lk_exclusivecount = 0;
    941  1.47  sommerfe 		COUNT_CPU(cpu_id, -count);
    942  1.47  sommerfe 		lkp->lk_flags &= ~LK_HAVE_EXCL;
    943  1.69   thorpej 		SETHOLDER(lkp, LK_NOPROC, 0, LK_NOCPU);
    944  1.50   thorpej #if defined(LOCKDEBUG)
    945  1.50   thorpej 		lkp->lk_unlock_file = file;
    946  1.50   thorpej 		lkp->lk_unlock_line = line;
    947  1.50   thorpej #endif
    948  1.47  sommerfe 		DONTHAVEIT(lkp);
    949  1.47  sommerfe 	}
    950  1.47  sommerfe #ifdef DIAGNOSTIC
    951  1.47  sommerfe 	else if (lkp->lk_sharecount != 0)
    952  1.47  sommerfe 		panic("spinlock_release_all: release of shared lock!");
    953  1.47  sommerfe 	else
    954  1.47  sommerfe 		panic("spinlock_release_all: release of unlocked lock!");
    955  1.47  sommerfe #endif
    956  1.47  sommerfe 	INTERLOCK_RELEASE(lkp, LK_SPIN, s);
    957  1.47  sommerfe 
    958  1.47  sommerfe 	return (count);
    959  1.47  sommerfe }
    960  1.47  sommerfe 
    961  1.47  sommerfe /*
    962  1.47  sommerfe  * For a recursive spinlock held one or more times by the current CPU,
    963  1.47  sommerfe  * release all N locks, and return N.
    964  1.47  sommerfe  * Intended for use in mi_switch() right after resuming execution.
    965  1.47  sommerfe  */
    966  1.47  sommerfe 
    967  1.47  sommerfe void
    968  1.50   thorpej #if defined(LOCKDEBUG)
    969  1.50   thorpej _spinlock_acquire_count(__volatile struct lock *lkp, int count,
    970  1.50   thorpej     const char *file, int line)
    971  1.50   thorpej #else
    972  1.47  sommerfe spinlock_acquire_count(__volatile struct lock *lkp, int count)
    973  1.50   thorpej #endif
    974  1.47  sommerfe {
    975  1.47  sommerfe 	int s, error;
    976  1.47  sommerfe 	cpuid_t cpu_id;
    977  1.47  sommerfe 
    978  1.47  sommerfe 	KASSERT(lkp->lk_flags & LK_SPIN);
    979  1.47  sommerfe 
    980  1.47  sommerfe 	INTERLOCK_ACQUIRE(lkp, LK_SPIN, s);
    981  1.47  sommerfe 
    982  1.47  sommerfe 	cpu_id = cpu_number();
    983  1.47  sommerfe 
    984  1.47  sommerfe #ifdef DIAGNOSTIC
    985  1.69   thorpej 	if (WEHOLDIT(lkp, LK_NOPROC, 0, cpu_id))
    986  1.64    provos 		panic("spinlock_acquire_count: processor %lu already holds lock", (long)cpu_id);
    987  1.47  sommerfe #endif
    988  1.47  sommerfe 	/*
    989  1.47  sommerfe 	 * Try to acquire the want_exclusive flag.
    990  1.47  sommerfe 	 */
    991  1.78   hannken 	error = acquire(&lkp, &s, LK_SPIN, 0, LK_HAVE_EXCL | LK_WANT_EXCL);
    992  1.47  sommerfe 	lkp->lk_flags |= LK_WANT_EXCL;
    993  1.47  sommerfe 	/*
    994  1.47  sommerfe 	 * Wait for shared locks and upgrades to finish.
    995  1.47  sommerfe 	 */
    996  1.78   hannken 	error = acquire(&lkp, &s, LK_SPIN, 0,
    997  1.82      yamt 	    LK_HAVE_EXCL | LK_SHARE_NONZERO | LK_WANT_UPGRADE);
    998  1.47  sommerfe 	lkp->lk_flags &= ~LK_WANT_EXCL;
    999  1.47  sommerfe 	lkp->lk_flags |= LK_HAVE_EXCL;
   1000  1.69   thorpej 	SETHOLDER(lkp, LK_NOPROC, 0, cpu_id);
   1001  1.50   thorpej #if defined(LOCKDEBUG)
   1002  1.50   thorpej 	lkp->lk_lock_file = file;
   1003  1.50   thorpej 	lkp->lk_lock_line = line;
   1004  1.50   thorpej #endif
   1005  1.47  sommerfe 	HAVEIT(lkp);
   1006  1.47  sommerfe 	if (lkp->lk_exclusivecount != 0)
   1007  1.47  sommerfe 		panic("lockmgr: non-zero exclusive count");
   1008  1.47  sommerfe 	lkp->lk_exclusivecount = count;
   1009  1.47  sommerfe 	lkp->lk_recurselevel = 1;
   1010  1.47  sommerfe 	COUNT_CPU(cpu_id, count);
   1011  1.47  sommerfe 
   1012  1.47  sommerfe 	INTERLOCK_RELEASE(lkp, lkp->lk_flags, s);
   1013  1.47  sommerfe }
   1014  1.47  sommerfe 
   1015  1.47  sommerfe 
   1016  1.47  sommerfe 
   1017  1.47  sommerfe /*
   1018   1.1      fvdl  * Print out information about state of a lock. Used by VOP_PRINT
   1019   1.1      fvdl  * routines to display ststus about contained locks.
   1020   1.1      fvdl  */
   1021   1.2      fvdl void
   1022  1.33   thorpej lockmgr_printinfo(__volatile struct lock *lkp)
   1023   1.1      fvdl {
   1024   1.1      fvdl 
   1025   1.1      fvdl 	if (lkp->lk_sharecount)
   1026   1.1      fvdl 		printf(" lock type %s: SHARED (count %d)", lkp->lk_wmesg,
   1027   1.1      fvdl 		    lkp->lk_sharecount);
   1028  1.19   thorpej 	else if (lkp->lk_flags & LK_HAVE_EXCL) {
   1029  1.19   thorpej 		printf(" lock type %s: EXCL (count %d) by ",
   1030  1.19   thorpej 		    lkp->lk_wmesg, lkp->lk_exclusivecount);
   1031  1.19   thorpej 		if (lkp->lk_flags & LK_SPIN)
   1032  1.19   thorpej 			printf("processor %lu", lkp->lk_cpu);
   1033  1.19   thorpej 		else
   1034  1.69   thorpej 			printf("pid %d.%d", lkp->lk_lockholder,
   1035  1.69   thorpej 			    lkp->lk_locklwp);
   1036  1.19   thorpej 	} else
   1037  1.19   thorpej 		printf(" not locked");
   1038  1.19   thorpej 	if ((lkp->lk_flags & LK_SPIN) == 0 && lkp->lk_waitcount > 0)
   1039   1.1      fvdl 		printf(" with %d pending", lkp->lk_waitcount);
   1040   1.1      fvdl }
   1041   1.1      fvdl 
   1042  1.21   thorpej #if defined(LOCKDEBUG) /* { */
   1043  1.21   thorpej TAILQ_HEAD(, simplelock) simplelock_list =
   1044  1.21   thorpej     TAILQ_HEAD_INITIALIZER(simplelock_list);
   1045  1.21   thorpej 
   1046  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
   1047  1.21   thorpej struct simplelock simplelock_list_slock = SIMPLELOCK_INITIALIZER;
   1048  1.21   thorpej 
   1049  1.21   thorpej #define	SLOCK_LIST_LOCK()						\
   1050  1.29  sommerfe 	__cpu_simple_lock(&simplelock_list_slock.lock_data)
   1051  1.21   thorpej 
   1052  1.21   thorpej #define	SLOCK_LIST_UNLOCK()						\
   1053  1.29  sommerfe 	__cpu_simple_unlock(&simplelock_list_slock.lock_data)
   1054  1.21   thorpej 
   1055  1.21   thorpej #define	SLOCK_COUNT(x)							\
   1056  1.47  sommerfe 	curcpu()->ci_simple_locks += (x)
   1057  1.21   thorpej #else
   1058  1.21   thorpej u_long simple_locks;
   1059  1.21   thorpej 
   1060  1.21   thorpej #define	SLOCK_LIST_LOCK()	/* nothing */
   1061  1.21   thorpej 
   1062  1.21   thorpej #define	SLOCK_LIST_UNLOCK()	/* nothing */
   1063  1.21   thorpej 
   1064  1.21   thorpej #define	SLOCK_COUNT(x)		simple_locks += (x)
   1065  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
   1066  1.21   thorpej 
   1067  1.26  sommerfe #ifdef MULTIPROCESSOR
   1068  1.75       wiz #define SLOCK_MP()		lock_printf("on CPU %ld\n", 		\
   1069  1.46   thorpej 				    (u_long) cpu_number())
   1070  1.26  sommerfe #else
   1071  1.26  sommerfe #define SLOCK_MP()		/* nothing */
   1072  1.26  sommerfe #endif
   1073  1.26  sommerfe 
   1074  1.21   thorpej #define	SLOCK_WHERE(str, alp, id, l)					\
   1075  1.21   thorpej do {									\
   1076  1.58       chs 	lock_printf("\n");						\
   1077  1.25   thorpej 	lock_printf(str);						\
   1078  1.33   thorpej 	lock_printf("lock: %p, currently at: %s:%d\n", (alp), (id), (l)); \
   1079  1.26  sommerfe 	SLOCK_MP();							\
   1080  1.21   thorpej 	if ((alp)->lock_file != NULL)					\
   1081  1.25   thorpej 		lock_printf("last locked: %s:%d\n", (alp)->lock_file,	\
   1082  1.21   thorpej 		    (alp)->lock_line);					\
   1083  1.21   thorpej 	if ((alp)->unlock_file != NULL)					\
   1084  1.25   thorpej 		lock_printf("last unlocked: %s:%d\n", (alp)->unlock_file, \
   1085  1.21   thorpej 		    (alp)->unlock_line);				\
   1086  1.58       chs 	SLOCK_TRACE()							\
   1087  1.21   thorpej 	SLOCK_DEBUGGER();						\
   1088  1.30   thorpej } while (/*CONSTCOND*/0)
   1089  1.12       chs 
   1090   1.1      fvdl /*
   1091   1.1      fvdl  * Simple lock functions so that the debugger can see from whence
   1092   1.1      fvdl  * they are being called.
   1093   1.1      fvdl  */
   1094   1.1      fvdl void
   1095  1.33   thorpej simple_lock_init(struct simplelock *alp)
   1096   1.1      fvdl {
   1097  1.21   thorpej 
   1098  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
   1099  1.27   thorpej 	__cpu_simple_lock_init(&alp->lock_data);
   1100  1.21   thorpej #else
   1101  1.27   thorpej 	alp->lock_data = __SIMPLELOCK_UNLOCKED;
   1102  1.21   thorpej #endif /* } */
   1103   1.5       chs 	alp->lock_file = NULL;
   1104   1.5       chs 	alp->lock_line = 0;
   1105   1.5       chs 	alp->unlock_file = NULL;
   1106   1.5       chs 	alp->unlock_line = 0;
   1107  1.41   thorpej 	alp->lock_holder = LK_NOCPU;
   1108   1.1      fvdl }
   1109   1.1      fvdl 
   1110   1.1      fvdl void
   1111  1.33   thorpej _simple_lock(__volatile struct simplelock *alp, const char *id, int l)
   1112   1.1      fvdl {
   1113  1.24   thorpej 	cpuid_t cpu_id = cpu_number();
   1114  1.12       chs 	int s;
   1115  1.12       chs 
   1116  1.44   thorpej 	s = spllock();
   1117  1.21   thorpej 
   1118  1.21   thorpej 	/*
   1119  1.21   thorpej 	 * MULTIPROCESSOR case: This is `safe' since if it's not us, we
   1120  1.21   thorpej 	 * don't take any action, and just fall into the normal spin case.
   1121  1.21   thorpej 	 */
   1122  1.27   thorpej 	if (alp->lock_data == __SIMPLELOCK_LOCKED) {
   1123  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
   1124  1.21   thorpej 		if (alp->lock_holder == cpu_id) {
   1125  1.21   thorpej 			SLOCK_WHERE("simple_lock: locking against myself\n",
   1126  1.21   thorpej 			    alp, id, l);
   1127  1.21   thorpej 			goto out;
   1128   1.1      fvdl 		}
   1129  1.21   thorpej #else
   1130  1.21   thorpej 		SLOCK_WHERE("simple_lock: lock held\n", alp, id, l);
   1131  1.21   thorpej 		goto out;
   1132  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
   1133   1.1      fvdl 	}
   1134  1.21   thorpej 
   1135  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
   1136  1.21   thorpej 	/* Acquire the lock before modifying any fields. */
   1137  1.70        pk 	splx(s);
   1138  1.27   thorpej 	__cpu_simple_lock(&alp->lock_data);
   1139  1.70        pk 	s = spllock();
   1140  1.21   thorpej #else
   1141  1.27   thorpej 	alp->lock_data = __SIMPLELOCK_LOCKED;
   1142  1.21   thorpej #endif /* } */
   1143  1.21   thorpej 
   1144  1.45  sommerfe 	if (alp->lock_holder != LK_NOCPU) {
   1145  1.45  sommerfe 		SLOCK_WHERE("simple_lock: uninitialized lock\n",
   1146  1.45  sommerfe 		    alp, id, l);
   1147  1.45  sommerfe 	}
   1148   1.5       chs 	alp->lock_file = id;
   1149   1.5       chs 	alp->lock_line = l;
   1150  1.21   thorpej 	alp->lock_holder = cpu_id;
   1151  1.21   thorpej 
   1152  1.21   thorpej 	SLOCK_LIST_LOCK();
   1153  1.21   thorpej 	/* XXX Cast away volatile */
   1154  1.21   thorpej 	TAILQ_INSERT_TAIL(&simplelock_list, (struct simplelock *)alp, list);
   1155  1.21   thorpej 	SLOCK_LIST_UNLOCK();
   1156  1.21   thorpej 
   1157  1.21   thorpej 	SLOCK_COUNT(1);
   1158  1.21   thorpej 
   1159  1.21   thorpej  out:
   1160  1.18       chs 	splx(s);
   1161  1.38   thorpej }
   1162  1.38   thorpej 
   1163  1.38   thorpej int
   1164  1.38   thorpej _simple_lock_held(__volatile struct simplelock *alp)
   1165  1.38   thorpej {
   1166  1.54     enami #if defined(MULTIPROCESSOR) || defined(DIAGNOSTIC)
   1167  1.38   thorpej 	cpuid_t cpu_id = cpu_number();
   1168  1.54     enami #endif
   1169  1.38   thorpej 	int s, locked = 0;
   1170  1.38   thorpej 
   1171  1.44   thorpej 	s = spllock();
   1172  1.42   thorpej 
   1173  1.42   thorpej #if defined(MULTIPROCESSOR)
   1174  1.38   thorpej 	if (__cpu_simple_lock_try(&alp->lock_data) == 0)
   1175  1.38   thorpej 		locked = (alp->lock_holder == cpu_id);
   1176  1.38   thorpej 	else
   1177  1.38   thorpej 		__cpu_simple_unlock(&alp->lock_data);
   1178  1.38   thorpej #else
   1179  1.42   thorpej 	if (alp->lock_data == __SIMPLELOCK_LOCKED) {
   1180  1.42   thorpej 		locked = 1;
   1181  1.42   thorpej 		KASSERT(alp->lock_holder == cpu_id);
   1182  1.42   thorpej 	}
   1183  1.42   thorpej #endif
   1184  1.38   thorpej 
   1185  1.38   thorpej 	splx(s);
   1186  1.42   thorpej 
   1187  1.38   thorpej 	return (locked);
   1188   1.1      fvdl }
   1189   1.1      fvdl 
   1190   1.1      fvdl int
   1191  1.33   thorpej _simple_lock_try(__volatile struct simplelock *alp, const char *id, int l)
   1192   1.1      fvdl {
   1193  1.24   thorpej 	cpuid_t cpu_id = cpu_number();
   1194  1.21   thorpej 	int s, rv = 0;
   1195   1.1      fvdl 
   1196  1.44   thorpej 	s = spllock();
   1197  1.21   thorpej 
   1198  1.21   thorpej 	/*
   1199  1.21   thorpej 	 * MULTIPROCESSOR case: This is `safe' since if it's not us, we
   1200  1.21   thorpej 	 * don't take any action.
   1201  1.21   thorpej 	 */
   1202  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
   1203  1.27   thorpej 	if ((rv = __cpu_simple_lock_try(&alp->lock_data)) == 0) {
   1204  1.21   thorpej 		if (alp->lock_holder == cpu_id)
   1205  1.21   thorpej 			SLOCK_WHERE("simple_lock_try: locking against myself\n",
   1206  1.26  sommerfe 			    alp, id, l);
   1207  1.21   thorpej 		goto out;
   1208  1.21   thorpej 	}
   1209  1.21   thorpej #else
   1210  1.27   thorpej 	if (alp->lock_data == __SIMPLELOCK_LOCKED) {
   1211  1.21   thorpej 		SLOCK_WHERE("simple_lock_try: lock held\n", alp, id, l);
   1212  1.21   thorpej 		goto out;
   1213  1.18       chs 	}
   1214  1.27   thorpej 	alp->lock_data = __SIMPLELOCK_LOCKED;
   1215  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
   1216  1.21   thorpej 
   1217  1.21   thorpej 	/*
   1218  1.21   thorpej 	 * At this point, we have acquired the lock.
   1219  1.21   thorpej 	 */
   1220  1.21   thorpej 
   1221  1.21   thorpej 	rv = 1;
   1222  1.18       chs 
   1223   1.5       chs 	alp->lock_file = id;
   1224   1.5       chs 	alp->lock_line = l;
   1225  1.21   thorpej 	alp->lock_holder = cpu_id;
   1226  1.21   thorpej 
   1227  1.21   thorpej 	SLOCK_LIST_LOCK();
   1228  1.21   thorpej 	/* XXX Cast away volatile. */
   1229  1.21   thorpej 	TAILQ_INSERT_TAIL(&simplelock_list, (struct simplelock *)alp, list);
   1230  1.21   thorpej 	SLOCK_LIST_UNLOCK();
   1231  1.21   thorpej 
   1232  1.21   thorpej 	SLOCK_COUNT(1);
   1233  1.21   thorpej 
   1234  1.21   thorpej  out:
   1235  1.12       chs 	splx(s);
   1236  1.21   thorpej 	return (rv);
   1237   1.1      fvdl }
   1238   1.1      fvdl 
   1239   1.1      fvdl void
   1240  1.33   thorpej _simple_unlock(__volatile struct simplelock *alp, const char *id, int l)
   1241   1.1      fvdl {
   1242  1.12       chs 	int s;
   1243   1.1      fvdl 
   1244  1.44   thorpej 	s = spllock();
   1245  1.21   thorpej 
   1246  1.21   thorpej 	/*
   1247  1.21   thorpej 	 * MULTIPROCESSOR case: This is `safe' because we think we hold
   1248  1.21   thorpej 	 * the lock, and if we don't, we don't take any action.
   1249  1.21   thorpej 	 */
   1250  1.27   thorpej 	if (alp->lock_data == __SIMPLELOCK_UNLOCKED) {
   1251  1.21   thorpej 		SLOCK_WHERE("simple_unlock: lock not held\n",
   1252  1.21   thorpej 		    alp, id, l);
   1253  1.21   thorpej 		goto out;
   1254  1.21   thorpej 	}
   1255  1.21   thorpej 
   1256  1.21   thorpej 	SLOCK_LIST_LOCK();
   1257  1.21   thorpej 	TAILQ_REMOVE(&simplelock_list, alp, list);
   1258  1.21   thorpej 	SLOCK_LIST_UNLOCK();
   1259  1.21   thorpej 
   1260  1.21   thorpej 	SLOCK_COUNT(-1);
   1261  1.21   thorpej 
   1262  1.21   thorpej 	alp->list.tqe_next = NULL;	/* sanity */
   1263  1.21   thorpej 	alp->list.tqe_prev = NULL;	/* sanity */
   1264  1.21   thorpej 
   1265   1.5       chs 	alp->unlock_file = id;
   1266   1.5       chs 	alp->unlock_line = l;
   1267  1.21   thorpej 
   1268  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
   1269  1.26  sommerfe 	alp->lock_holder = LK_NOCPU;
   1270  1.21   thorpej 	/* Now that we've modified all fields, release the lock. */
   1271  1.27   thorpej 	__cpu_simple_unlock(&alp->lock_data);
   1272  1.21   thorpej #else
   1273  1.27   thorpej 	alp->lock_data = __SIMPLELOCK_UNLOCKED;
   1274  1.41   thorpej 	KASSERT(alp->lock_holder == cpu_number());
   1275  1.41   thorpej 	alp->lock_holder = LK_NOCPU;
   1276  1.21   thorpej #endif /* } */
   1277  1.21   thorpej 
   1278  1.21   thorpej  out:
   1279  1.18       chs 	splx(s);
   1280  1.12       chs }
   1281  1.12       chs 
   1282  1.12       chs void
   1283  1.33   thorpej simple_lock_dump(void)
   1284  1.12       chs {
   1285  1.12       chs 	struct simplelock *alp;
   1286  1.12       chs 	int s;
   1287  1.12       chs 
   1288  1.44   thorpej 	s = spllock();
   1289  1.21   thorpej 	SLOCK_LIST_LOCK();
   1290  1.25   thorpej 	lock_printf("all simple locks:\n");
   1291  1.58       chs 	TAILQ_FOREACH(alp, &simplelock_list, list) {
   1292  1.25   thorpej 		lock_printf("%p CPU %lu %s:%d\n", alp, alp->lock_holder,
   1293  1.21   thorpej 		    alp->lock_file, alp->lock_line);
   1294  1.12       chs 	}
   1295  1.21   thorpej 	SLOCK_LIST_UNLOCK();
   1296  1.12       chs 	splx(s);
   1297  1.12       chs }
   1298  1.12       chs 
   1299  1.12       chs void
   1300  1.33   thorpej simple_lock_freecheck(void *start, void *end)
   1301  1.12       chs {
   1302  1.12       chs 	struct simplelock *alp;
   1303  1.12       chs 	int s;
   1304  1.12       chs 
   1305  1.44   thorpej 	s = spllock();
   1306  1.21   thorpej 	SLOCK_LIST_LOCK();
   1307  1.58       chs 	TAILQ_FOREACH(alp, &simplelock_list, list) {
   1308  1.12       chs 		if ((void *)alp >= start && (void *)alp < end) {
   1309  1.25   thorpej 			lock_printf("freeing simple_lock %p CPU %lu %s:%d\n",
   1310  1.34   thorpej 			    alp, alp->lock_holder, alp->lock_file,
   1311  1.34   thorpej 			    alp->lock_line);
   1312  1.34   thorpej 			SLOCK_DEBUGGER();
   1313  1.34   thorpej 		}
   1314  1.34   thorpej 	}
   1315  1.34   thorpej 	SLOCK_LIST_UNLOCK();
   1316  1.34   thorpej 	splx(s);
   1317  1.34   thorpej }
   1318  1.34   thorpej 
   1319  1.55   thorpej /*
   1320  1.55   thorpej  * We must be holding exactly one lock: the sched_lock.
   1321  1.55   thorpej  */
   1322  1.55   thorpej 
   1323  1.34   thorpej void
   1324  1.34   thorpej simple_lock_switchcheck(void)
   1325  1.34   thorpej {
   1326  1.55   thorpej 
   1327  1.55   thorpej 	simple_lock_only_held(&sched_lock, "switching");
   1328  1.55   thorpej }
   1329  1.55   thorpej 
   1330  1.55   thorpej void
   1331  1.55   thorpej simple_lock_only_held(volatile struct simplelock *lp, const char *where)
   1332  1.55   thorpej {
   1333  1.34   thorpej 	struct simplelock *alp;
   1334  1.34   thorpej 	cpuid_t cpu_id = cpu_number();
   1335  1.34   thorpej 	int s;
   1336  1.34   thorpej 
   1337  1.55   thorpej 	if (lp) {
   1338  1.55   thorpej 		LOCK_ASSERT(simple_lock_held(lp));
   1339  1.55   thorpej 	}
   1340  1.44   thorpej 	s = spllock();
   1341  1.34   thorpej 	SLOCK_LIST_LOCK();
   1342  1.58       chs 	TAILQ_FOREACH(alp, &simplelock_list, list) {
   1343  1.55   thorpej 		if (alp == lp)
   1344  1.42   thorpej 			continue;
   1345  1.55   thorpej 		if (alp->lock_holder == cpu_id)
   1346  1.55   thorpej 			break;
   1347  1.12       chs 	}
   1348  1.21   thorpej 	SLOCK_LIST_UNLOCK();
   1349  1.12       chs 	splx(s);
   1350  1.55   thorpej 
   1351  1.55   thorpej 	if (alp != NULL) {
   1352  1.58       chs 		lock_printf("\n%s with held simple_lock %p "
   1353  1.55   thorpej 		    "CPU %lu %s:%d\n",
   1354  1.55   thorpej 		    where, alp, alp->lock_holder, alp->lock_file,
   1355  1.55   thorpej 		    alp->lock_line);
   1356  1.58       chs 		SLOCK_TRACE();
   1357  1.55   thorpej 		SLOCK_DEBUGGER();
   1358  1.55   thorpej 	}
   1359   1.1      fvdl }
   1360  1.21   thorpej #endif /* LOCKDEBUG */ /* } */
   1361  1.62   thorpej 
   1362  1.62   thorpej #if defined(MULTIPROCESSOR)
   1363  1.62   thorpej /*
   1364  1.62   thorpej  * Functions for manipulating the kernel_lock.  We put them here
   1365  1.62   thorpej  * so that they show up in profiles.
   1366  1.62   thorpej  */
   1367  1.62   thorpej 
   1368  1.62   thorpej struct lock kernel_lock;
   1369  1.62   thorpej 
   1370  1.62   thorpej void
   1371  1.62   thorpej _kernel_lock_init(void)
   1372  1.62   thorpej {
   1373  1.62   thorpej 
   1374  1.62   thorpej 	spinlockinit(&kernel_lock, "klock", 0);
   1375  1.62   thorpej }
   1376  1.62   thorpej 
   1377  1.62   thorpej /*
   1378  1.62   thorpej  * Acquire/release the kernel lock.  Intended for use in the scheduler
   1379  1.62   thorpej  * and the lower half of the kernel.
   1380  1.62   thorpej  */
   1381  1.62   thorpej void
   1382  1.62   thorpej _kernel_lock(int flag)
   1383  1.62   thorpej {
   1384  1.62   thorpej 
   1385  1.62   thorpej 	SCHED_ASSERT_UNLOCKED();
   1386  1.62   thorpej 	spinlockmgr(&kernel_lock, flag, 0);
   1387  1.62   thorpej }
   1388  1.62   thorpej 
   1389  1.62   thorpej void
   1390  1.62   thorpej _kernel_unlock(void)
   1391  1.62   thorpej {
   1392  1.62   thorpej 
   1393  1.62   thorpej 	spinlockmgr(&kernel_lock, LK_RELEASE, 0);
   1394  1.62   thorpej }
   1395  1.62   thorpej 
   1396  1.62   thorpej /*
   1397  1.62   thorpej  * Acquire/release the kernel_lock on behalf of a process.  Intended for
   1398  1.62   thorpej  * use in the top half of the kernel.
   1399  1.62   thorpej  */
   1400  1.62   thorpej void
   1401  1.69   thorpej _kernel_proc_lock(struct lwp *l)
   1402  1.62   thorpej {
   1403  1.62   thorpej 
   1404  1.62   thorpej 	SCHED_ASSERT_UNLOCKED();
   1405  1.62   thorpej 	spinlockmgr(&kernel_lock, LK_EXCLUSIVE, 0);
   1406  1.62   thorpej }
   1407  1.62   thorpej 
   1408  1.62   thorpej void
   1409  1.69   thorpej _kernel_proc_unlock(struct lwp *l)
   1410  1.62   thorpej {
   1411  1.62   thorpej 
   1412  1.62   thorpej 	spinlockmgr(&kernel_lock, LK_RELEASE, 0);
   1413  1.62   thorpej }
   1414  1.77      yamt 
   1415  1.77      yamt int
   1416  1.77      yamt _kernel_lock_release_all()
   1417  1.77      yamt {
   1418  1.77      yamt 	int hold_count;
   1419  1.77      yamt 
   1420  1.77      yamt 	if (lockstatus(&kernel_lock) == LK_EXCLUSIVE)
   1421  1.77      yamt 		hold_count = spinlock_release_all(&kernel_lock);
   1422  1.77      yamt 	else
   1423  1.77      yamt 		hold_count = 0;
   1424  1.77      yamt 
   1425  1.77      yamt 	return hold_count;
   1426  1.77      yamt }
   1427  1.77      yamt 
   1428  1.77      yamt void
   1429  1.77      yamt _kernel_lock_acquire_count(int hold_count)
   1430  1.77      yamt {
   1431  1.77      yamt 
   1432  1.77      yamt 	if (hold_count != 0)
   1433  1.77      yamt 		spinlock_acquire_count(&kernel_lock, hold_count);
   1434  1.77      yamt }
   1435  1.62   thorpej #endif /* MULTIPROCESSOR */
   1436