Home | History | Annotate | Line # | Download | only in kern
kern_lock.c revision 1.78
      1  1.78   hannken /*	$NetBSD: kern_lock.c,v 1.78 2004/05/25 14:54:57 hannken 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.78   hannken __KERNEL_RCSID(0, "$NetBSD: kern_lock.c,v 1.78 2004/05/25 14:54:57 hannken 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.43   thorpej 	INTERLOCK_ACQUIRE(lkp, lkp->lk_flags, s);
    560   1.1      fvdl 	if (flags & LK_INTERLOCK)
    561   1.1      fvdl 		simple_unlock(interlkp);
    562   1.1      fvdl 	extflags = (flags | lkp->lk_flags) & LK_EXTFLG_MASK;
    563  1.19   thorpej 
    564  1.21   thorpej #ifdef DIAGNOSTIC /* { */
    565  1.19   thorpej 	/*
    566  1.19   thorpej 	 * Don't allow spins on sleep locks and don't allow sleeps
    567  1.19   thorpej 	 * on spin locks.
    568  1.19   thorpej 	 */
    569  1.19   thorpej 	if ((flags ^ lkp->lk_flags) & LK_SPIN)
    570  1.64    provos 		panic("lockmgr: sleep/spin mismatch");
    571  1.21   thorpej #endif /* } */
    572  1.19   thorpej 
    573  1.69   thorpej 	if (extflags & LK_SPIN) {
    574  1.19   thorpej 		pid = LK_KERNPROC;
    575  1.69   thorpej 		lid = 0;
    576  1.69   thorpej 	} else {
    577  1.69   thorpej 		if (l == NULL) {
    578  1.32  sommerfe 			if (!doing_shutdown) {
    579  1.32  sommerfe 				panic("lockmgr: no context");
    580  1.32  sommerfe 			} else {
    581  1.69   thorpej 				l = &lwp0;
    582  1.32  sommerfe 				if (panicstr && (!(flags & LK_NOWAIT))) {
    583  1.32  sommerfe 					flags |= LK_NOWAIT;
    584  1.32  sommerfe 					lock_shutdown_noblock = 1;
    585  1.32  sommerfe 				}
    586  1.32  sommerfe 			}
    587  1.32  sommerfe 		}
    588  1.69   thorpej 		lid = l->l_lid;
    589  1.69   thorpej 		pid = l->l_proc->p_pid;
    590  1.19   thorpej 	}
    591  1.24   thorpej 	cpu_id = cpu_number();
    592  1.19   thorpej 
    593   1.1      fvdl 	/*
    594   1.1      fvdl 	 * Once a lock has drained, the LK_DRAINING flag is set and an
    595   1.1      fvdl 	 * exclusive lock is returned. The only valid operation thereafter
    596   1.1      fvdl 	 * is a single release of that exclusive lock. This final release
    597   1.1      fvdl 	 * clears the LK_DRAINING flag and sets the LK_DRAINED flag. Any
    598   1.1      fvdl 	 * further requests of any sort will result in a panic. The bits
    599   1.1      fvdl 	 * selected for these two flags are chosen so that they will be set
    600   1.1      fvdl 	 * in memory that is freed (freed memory is filled with 0xdeadbeef).
    601   1.1      fvdl 	 * The final release is permitted to give a new lease on life to
    602   1.1      fvdl 	 * the lock by specifying LK_REENABLE.
    603   1.1      fvdl 	 */
    604   1.1      fvdl 	if (lkp->lk_flags & (LK_DRAINING|LK_DRAINED)) {
    605  1.28   thorpej #ifdef DIAGNOSTIC /* { */
    606   1.1      fvdl 		if (lkp->lk_flags & LK_DRAINED)
    607   1.1      fvdl 			panic("lockmgr: using decommissioned lock");
    608   1.1      fvdl 		if ((flags & LK_TYPE_MASK) != LK_RELEASE ||
    609  1.69   thorpej 		    WEHOLDIT(lkp, pid, lid, cpu_id) == 0)
    610  1.64    provos 			panic("lockmgr: non-release on draining lock: %d",
    611   1.1      fvdl 			    flags & LK_TYPE_MASK);
    612  1.28   thorpej #endif /* DIAGNOSTIC */ /* } */
    613   1.1      fvdl 		lkp->lk_flags &= ~LK_DRAINING;
    614   1.1      fvdl 		if ((flags & LK_REENABLE) == 0)
    615   1.1      fvdl 			lkp->lk_flags |= LK_DRAINED;
    616   1.1      fvdl 	}
    617   1.1      fvdl 
    618   1.1      fvdl 	switch (flags & LK_TYPE_MASK) {
    619   1.1      fvdl 
    620   1.1      fvdl 	case LK_SHARED:
    621  1.69   thorpej 		if (WEHOLDIT(lkp, pid, lid, cpu_id) == 0) {
    622   1.1      fvdl 			/*
    623   1.1      fvdl 			 * If just polling, check to see if we will block.
    624   1.1      fvdl 			 */
    625   1.1      fvdl 			if ((extflags & LK_NOWAIT) && (lkp->lk_flags &
    626   1.1      fvdl 			    (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE))) {
    627   1.1      fvdl 				error = EBUSY;
    628   1.1      fvdl 				break;
    629   1.1      fvdl 			}
    630   1.1      fvdl 			/*
    631   1.1      fvdl 			 * Wait for exclusive locks and upgrades to clear.
    632   1.1      fvdl 			 */
    633  1.78   hannken 			error = acquire(&lkp, &s, extflags, 0,
    634  1.73      yamt 			    LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE);
    635   1.1      fvdl 			if (error)
    636   1.1      fvdl 				break;
    637   1.1      fvdl 			lkp->lk_sharecount++;
    638  1.73      yamt 			lkp->lk_flags |= LK_SHARE_NONZERO;
    639  1.69   thorpej 			COUNT(lkp, l, cpu_id, 1);
    640   1.1      fvdl 			break;
    641   1.1      fvdl 		}
    642   1.1      fvdl 		/*
    643   1.1      fvdl 		 * We hold an exclusive lock, so downgrade it to shared.
    644   1.1      fvdl 		 * An alternative would be to fail with EDEADLK.
    645   1.1      fvdl 		 */
    646   1.1      fvdl 		lkp->lk_sharecount++;
    647  1.73      yamt 		lkp->lk_flags |= LK_SHARE_NONZERO;
    648  1.69   thorpej 		COUNT(lkp, l, cpu_id, 1);
    649   1.1      fvdl 		/* fall into downgrade */
    650   1.1      fvdl 
    651   1.1      fvdl 	case LK_DOWNGRADE:
    652  1.69   thorpej 		if (WEHOLDIT(lkp, pid, lid, cpu_id) == 0 ||
    653  1.19   thorpej 		    lkp->lk_exclusivecount == 0)
    654   1.1      fvdl 			panic("lockmgr: not holding exclusive lock");
    655   1.1      fvdl 		lkp->lk_sharecount += lkp->lk_exclusivecount;
    656  1.73      yamt 		lkp->lk_flags |= LK_SHARE_NONZERO;
    657   1.1      fvdl 		lkp->lk_exclusivecount = 0;
    658  1.15      fvdl 		lkp->lk_recurselevel = 0;
    659   1.1      fvdl 		lkp->lk_flags &= ~LK_HAVE_EXCL;
    660  1.69   thorpej 		SETHOLDER(lkp, LK_NOPROC, 0, LK_NOCPU);
    661  1.50   thorpej #if defined(LOCKDEBUG)
    662  1.50   thorpej 		lkp->lk_unlock_file = file;
    663  1.50   thorpej 		lkp->lk_unlock_line = line;
    664  1.50   thorpej #endif
    665  1.21   thorpej 		DONTHAVEIT(lkp);
    666  1.23   thorpej 		WAKEUP_WAITER(lkp);
    667   1.1      fvdl 		break;
    668   1.1      fvdl 
    669   1.1      fvdl 	case LK_EXCLUPGRADE:
    670   1.1      fvdl 		/*
    671   1.1      fvdl 		 * If another process is ahead of us to get an upgrade,
    672   1.1      fvdl 		 * then we want to fail rather than have an intervening
    673   1.1      fvdl 		 * exclusive access.
    674   1.1      fvdl 		 */
    675   1.1      fvdl 		if (lkp->lk_flags & LK_WANT_UPGRADE) {
    676   1.1      fvdl 			lkp->lk_sharecount--;
    677  1.73      yamt 			if (lkp->lk_sharecount == 0)
    678  1.73      yamt 				lkp->lk_flags &= ~LK_SHARE_NONZERO;
    679  1.69   thorpej 			COUNT(lkp, l, cpu_id, -1);
    680   1.1      fvdl 			error = EBUSY;
    681   1.1      fvdl 			break;
    682   1.1      fvdl 		}
    683   1.1      fvdl 		/* fall into normal upgrade */
    684   1.1      fvdl 
    685   1.1      fvdl 	case LK_UPGRADE:
    686   1.1      fvdl 		/*
    687   1.1      fvdl 		 * Upgrade a shared lock to an exclusive one. If another
    688   1.1      fvdl 		 * shared lock has already requested an upgrade to an
    689   1.1      fvdl 		 * exclusive lock, our shared lock is released and an
    690   1.1      fvdl 		 * exclusive lock is requested (which will be granted
    691   1.1      fvdl 		 * after the upgrade). If we return an error, the file
    692   1.1      fvdl 		 * will always be unlocked.
    693   1.1      fvdl 		 */
    694  1.69   thorpej 		if (WEHOLDIT(lkp, pid, lid, cpu_id) || lkp->lk_sharecount <= 0)
    695   1.1      fvdl 			panic("lockmgr: upgrade exclusive lock");
    696   1.1      fvdl 		lkp->lk_sharecount--;
    697  1.73      yamt 		if (lkp->lk_sharecount == 0)
    698  1.73      yamt 			lkp->lk_flags &= ~LK_SHARE_NONZERO;
    699  1.69   thorpej 		COUNT(lkp, l, cpu_id, -1);
    700   1.1      fvdl 		/*
    701   1.1      fvdl 		 * If we are just polling, check to see if we will block.
    702   1.1      fvdl 		 */
    703   1.1      fvdl 		if ((extflags & LK_NOWAIT) &&
    704   1.1      fvdl 		    ((lkp->lk_flags & LK_WANT_UPGRADE) ||
    705   1.1      fvdl 		     lkp->lk_sharecount > 1)) {
    706   1.1      fvdl 			error = EBUSY;
    707   1.1      fvdl 			break;
    708   1.1      fvdl 		}
    709   1.1      fvdl 		if ((lkp->lk_flags & LK_WANT_UPGRADE) == 0) {
    710   1.1      fvdl 			/*
    711   1.1      fvdl 			 * We are first shared lock to request an upgrade, so
    712   1.1      fvdl 			 * request upgrade and wait for the shared count to
    713   1.1      fvdl 			 * drop to zero, then take exclusive lock.
    714   1.1      fvdl 			 */
    715   1.1      fvdl 			lkp->lk_flags |= LK_WANT_UPGRADE;
    716  1.78   hannken 			error = acquire(&lkp, &s, extflags, 0, LK_SHARE_NONZERO);
    717   1.1      fvdl 			lkp->lk_flags &= ~LK_WANT_UPGRADE;
    718   1.1      fvdl 			if (error)
    719   1.1      fvdl 				break;
    720   1.1      fvdl 			lkp->lk_flags |= LK_HAVE_EXCL;
    721  1.69   thorpej 			SETHOLDER(lkp, pid, lid, cpu_id);
    722  1.50   thorpej #if defined(LOCKDEBUG)
    723  1.50   thorpej 			lkp->lk_lock_file = file;
    724  1.50   thorpej 			lkp->lk_lock_line = line;
    725  1.50   thorpej #endif
    726  1.21   thorpej 			HAVEIT(lkp);
    727   1.1      fvdl 			if (lkp->lk_exclusivecount != 0)
    728   1.1      fvdl 				panic("lockmgr: non-zero exclusive count");
    729   1.1      fvdl 			lkp->lk_exclusivecount = 1;
    730  1.15      fvdl 			if (extflags & LK_SETRECURSE)
    731  1.15      fvdl 				lkp->lk_recurselevel = 1;
    732  1.69   thorpej 			COUNT(lkp, l, cpu_id, 1);
    733   1.1      fvdl 			break;
    734   1.1      fvdl 		}
    735   1.1      fvdl 		/*
    736   1.1      fvdl 		 * Someone else has requested upgrade. Release our shared
    737   1.1      fvdl 		 * lock, awaken upgrade requestor if we are the last shared
    738   1.1      fvdl 		 * lock, then request an exclusive lock.
    739   1.1      fvdl 		 */
    740  1.23   thorpej 		if (lkp->lk_sharecount == 0)
    741  1.23   thorpej 			WAKEUP_WAITER(lkp);
    742   1.1      fvdl 		/* fall into exclusive request */
    743   1.1      fvdl 
    744   1.1      fvdl 	case LK_EXCLUSIVE:
    745  1.69   thorpej 		if (WEHOLDIT(lkp, pid, lid, cpu_id)) {
    746   1.1      fvdl 			/*
    747  1.19   thorpej 			 * Recursive lock.
    748   1.1      fvdl 			 */
    749  1.15      fvdl 			if ((extflags & LK_CANRECURSE) == 0 &&
    750  1.16  sommerfe 			     lkp->lk_recurselevel == 0) {
    751  1.16  sommerfe 				if (extflags & LK_RECURSEFAIL) {
    752  1.16  sommerfe 					error = EDEADLK;
    753  1.16  sommerfe 					break;
    754  1.16  sommerfe 				} else
    755  1.16  sommerfe 					panic("lockmgr: locking against myself");
    756  1.16  sommerfe 			}
    757   1.1      fvdl 			lkp->lk_exclusivecount++;
    758  1.15      fvdl 			if (extflags & LK_SETRECURSE &&
    759  1.15      fvdl 			    lkp->lk_recurselevel == 0)
    760  1.15      fvdl 				lkp->lk_recurselevel = lkp->lk_exclusivecount;
    761  1.69   thorpej 			COUNT(lkp, l, cpu_id, 1);
    762   1.1      fvdl 			break;
    763   1.1      fvdl 		}
    764   1.1      fvdl 		/*
    765   1.1      fvdl 		 * If we are just polling, check to see if we will sleep.
    766   1.1      fvdl 		 */
    767  1.73      yamt 		if ((extflags & LK_NOWAIT) && (lkp->lk_flags &
    768  1.73      yamt 		     (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE |
    769  1.73      yamt 		     LK_SHARE_NONZERO))) {
    770   1.1      fvdl 			error = EBUSY;
    771   1.1      fvdl 			break;
    772   1.1      fvdl 		}
    773   1.1      fvdl 		/*
    774   1.1      fvdl 		 * Try to acquire the want_exclusive flag.
    775   1.1      fvdl 		 */
    776  1.78   hannken 		error = acquire(&lkp, &s, extflags, 0,
    777  1.74   hannken 		    LK_HAVE_EXCL | LK_WANT_EXCL);
    778   1.1      fvdl 		if (error)
    779   1.1      fvdl 			break;
    780   1.1      fvdl 		lkp->lk_flags |= LK_WANT_EXCL;
    781   1.1      fvdl 		/*
    782   1.1      fvdl 		 * Wait for shared locks and upgrades to finish.
    783   1.1      fvdl 		 */
    784  1.78   hannken 		error = acquire(&lkp, &s, extflags, 0,
    785  1.73      yamt 		    LK_WANT_UPGRADE | LK_SHARE_NONZERO);
    786   1.1      fvdl 		lkp->lk_flags &= ~LK_WANT_EXCL;
    787   1.1      fvdl 		if (error)
    788   1.1      fvdl 			break;
    789   1.1      fvdl 		lkp->lk_flags |= LK_HAVE_EXCL;
    790  1.69   thorpej 		SETHOLDER(lkp, pid, lid, cpu_id);
    791  1.50   thorpej #if defined(LOCKDEBUG)
    792  1.50   thorpej 		lkp->lk_lock_file = file;
    793  1.50   thorpej 		lkp->lk_lock_line = line;
    794  1.50   thorpej #endif
    795  1.21   thorpej 		HAVEIT(lkp);
    796   1.1      fvdl 		if (lkp->lk_exclusivecount != 0)
    797   1.1      fvdl 			panic("lockmgr: non-zero exclusive count");
    798   1.1      fvdl 		lkp->lk_exclusivecount = 1;
    799  1.15      fvdl 		if (extflags & LK_SETRECURSE)
    800  1.15      fvdl 			lkp->lk_recurselevel = 1;
    801  1.69   thorpej 		COUNT(lkp, l, cpu_id, 1);
    802   1.1      fvdl 		break;
    803   1.1      fvdl 
    804   1.1      fvdl 	case LK_RELEASE:
    805   1.1      fvdl 		if (lkp->lk_exclusivecount != 0) {
    806  1.69   thorpej 			if (WEHOLDIT(lkp, pid, lid, cpu_id) == 0) {
    807  1.19   thorpej 				if (lkp->lk_flags & LK_SPIN) {
    808  1.19   thorpej 					panic("lockmgr: processor %lu, not "
    809  1.19   thorpej 					    "exclusive lock holder %lu "
    810  1.19   thorpej 					    "unlocking", cpu_id, lkp->lk_cpu);
    811  1.19   thorpej 				} else {
    812  1.19   thorpej 					panic("lockmgr: pid %d, not "
    813  1.19   thorpej 					    "exclusive lock holder %d "
    814  1.19   thorpej 					    "unlocking", pid,
    815  1.19   thorpej 					    lkp->lk_lockholder);
    816  1.19   thorpej 				}
    817  1.19   thorpej 			}
    818  1.15      fvdl 			if (lkp->lk_exclusivecount == lkp->lk_recurselevel)
    819  1.15      fvdl 				lkp->lk_recurselevel = 0;
    820   1.1      fvdl 			lkp->lk_exclusivecount--;
    821  1.69   thorpej 			COUNT(lkp, l, cpu_id, -1);
    822   1.1      fvdl 			if (lkp->lk_exclusivecount == 0) {
    823   1.1      fvdl 				lkp->lk_flags &= ~LK_HAVE_EXCL;
    824  1.69   thorpej 				SETHOLDER(lkp, LK_NOPROC, 0, LK_NOCPU);
    825  1.50   thorpej #if defined(LOCKDEBUG)
    826  1.50   thorpej 				lkp->lk_unlock_file = file;
    827  1.50   thorpej 				lkp->lk_unlock_line = line;
    828  1.50   thorpej #endif
    829  1.21   thorpej 				DONTHAVEIT(lkp);
    830   1.1      fvdl 			}
    831   1.1      fvdl 		} else if (lkp->lk_sharecount != 0) {
    832   1.1      fvdl 			lkp->lk_sharecount--;
    833  1.73      yamt 			if (lkp->lk_sharecount == 0)
    834  1.73      yamt 				lkp->lk_flags &= ~LK_SHARE_NONZERO;
    835  1.69   thorpej 			COUNT(lkp, l, cpu_id, -1);
    836   1.1      fvdl 		}
    837  1.39   thorpej #ifdef DIAGNOSTIC
    838  1.39   thorpej 		else
    839  1.39   thorpej 			panic("lockmgr: release of unlocked lock!");
    840  1.39   thorpej #endif
    841  1.23   thorpej 		WAKEUP_WAITER(lkp);
    842   1.1      fvdl 		break;
    843   1.1      fvdl 
    844   1.1      fvdl 	case LK_DRAIN:
    845   1.1      fvdl 		/*
    846   1.1      fvdl 		 * Check that we do not already hold the lock, as it can
    847   1.1      fvdl 		 * never drain if we do. Unfortunately, we have no way to
    848   1.1      fvdl 		 * check for holding a shared lock, but at least we can
    849   1.1      fvdl 		 * check for an exclusive one.
    850   1.1      fvdl 		 */
    851  1.69   thorpej 		if (WEHOLDIT(lkp, pid, lid, cpu_id))
    852   1.1      fvdl 			panic("lockmgr: draining against myself");
    853   1.1      fvdl 		/*
    854   1.1      fvdl 		 * If we are just polling, check to see if we will sleep.
    855   1.1      fvdl 		 */
    856  1.73      yamt 		if ((extflags & LK_NOWAIT) && (lkp->lk_flags &
    857  1.73      yamt 		     (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE |
    858  1.73      yamt 		     LK_SHARE_NONZERO | LK_WAIT_NONZERO))) {
    859   1.1      fvdl 			error = EBUSY;
    860   1.1      fvdl 			break;
    861   1.1      fvdl 		}
    862  1.78   hannken 		error = acquire(&lkp, &s, extflags, 1,
    863  1.73      yamt 		    LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE |
    864  1.73      yamt 		    LK_SHARE_NONZERO | LK_WAIT_NONZERO);
    865  1.23   thorpej 		if (error)
    866  1.23   thorpej 			break;
    867   1.1      fvdl 		lkp->lk_flags |= LK_DRAINING | LK_HAVE_EXCL;
    868  1.69   thorpej 		SETHOLDER(lkp, pid, lid, cpu_id);
    869  1.50   thorpej #if defined(LOCKDEBUG)
    870  1.50   thorpej 		lkp->lk_lock_file = file;
    871  1.50   thorpej 		lkp->lk_lock_line = line;
    872  1.50   thorpej #endif
    873  1.21   thorpej 		HAVEIT(lkp);
    874   1.1      fvdl 		lkp->lk_exclusivecount = 1;
    875  1.15      fvdl 		/* XXX unlikely that we'd want this */
    876  1.15      fvdl 		if (extflags & LK_SETRECURSE)
    877  1.15      fvdl 			lkp->lk_recurselevel = 1;
    878  1.69   thorpej 		COUNT(lkp, l, cpu_id, 1);
    879   1.1      fvdl 		break;
    880   1.1      fvdl 
    881   1.1      fvdl 	default:
    882  1.43   thorpej 		INTERLOCK_RELEASE(lkp, lkp->lk_flags, s);
    883   1.1      fvdl 		panic("lockmgr: unknown locktype request %d",
    884   1.1      fvdl 		    flags & LK_TYPE_MASK);
    885   1.1      fvdl 		/* NOTREACHED */
    886   1.1      fvdl 	}
    887  1.23   thorpej 	if ((lkp->lk_flags & (LK_WAITDRAIN|LK_SPIN)) == LK_WAITDRAIN &&
    888  1.23   thorpej 	    ((lkp->lk_flags &
    889  1.73      yamt 	      (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE |
    890  1.73      yamt 	      LK_SHARE_NONZERO | LK_WAIT_NONZERO)) == 0)) {
    891   1.1      fvdl 		lkp->lk_flags &= ~LK_WAITDRAIN;
    892  1.59       chs 		wakeup((void *)&lkp->lk_flags);
    893   1.1      fvdl 	}
    894  1.32  sommerfe 	/*
    895  1.32  sommerfe 	 * Note that this panic will be a recursive panic, since
    896  1.32  sommerfe 	 * we only set lock_shutdown_noblock above if panicstr != NULL.
    897  1.32  sommerfe 	 */
    898  1.32  sommerfe 	if (error && lock_shutdown_noblock)
    899  1.32  sommerfe 		panic("lockmgr: deadlock (see previous panic)");
    900  1.32  sommerfe 
    901  1.43   thorpej 	INTERLOCK_RELEASE(lkp, lkp->lk_flags, s);
    902   1.1      fvdl 	return (error);
    903   1.1      fvdl }
    904   1.1      fvdl 
    905   1.1      fvdl /*
    906  1.47  sommerfe  * For a recursive spinlock held one or more times by the current CPU,
    907  1.47  sommerfe  * release all N locks, and return N.
    908  1.47  sommerfe  * Intended for use in mi_switch() shortly before context switching.
    909  1.47  sommerfe  */
    910  1.47  sommerfe 
    911  1.47  sommerfe int
    912  1.50   thorpej #if defined(LOCKDEBUG)
    913  1.50   thorpej _spinlock_release_all(__volatile struct lock *lkp, const char *file, int line)
    914  1.50   thorpej #else
    915  1.47  sommerfe spinlock_release_all(__volatile struct lock *lkp)
    916  1.50   thorpej #endif
    917  1.47  sommerfe {
    918  1.47  sommerfe 	int s, count;
    919  1.47  sommerfe 	cpuid_t cpu_id;
    920  1.47  sommerfe 
    921  1.47  sommerfe 	KASSERT(lkp->lk_flags & LK_SPIN);
    922  1.47  sommerfe 
    923  1.47  sommerfe 	INTERLOCK_ACQUIRE(lkp, LK_SPIN, s);
    924  1.47  sommerfe 
    925  1.47  sommerfe 	cpu_id = cpu_number();
    926  1.47  sommerfe 	count = lkp->lk_exclusivecount;
    927  1.47  sommerfe 
    928  1.47  sommerfe 	if (count != 0) {
    929  1.47  sommerfe #ifdef DIAGNOSTIC
    930  1.69   thorpej 		if (WEHOLDIT(lkp, 0, 0, cpu_id) == 0) {
    931  1.47  sommerfe 			panic("spinlock_release_all: processor %lu, not "
    932  1.47  sommerfe 			    "exclusive lock holder %lu "
    933  1.47  sommerfe 			    "unlocking", (long)cpu_id, lkp->lk_cpu);
    934  1.47  sommerfe 		}
    935  1.47  sommerfe #endif
    936  1.47  sommerfe 		lkp->lk_recurselevel = 0;
    937  1.47  sommerfe 		lkp->lk_exclusivecount = 0;
    938  1.47  sommerfe 		COUNT_CPU(cpu_id, -count);
    939  1.47  sommerfe 		lkp->lk_flags &= ~LK_HAVE_EXCL;
    940  1.69   thorpej 		SETHOLDER(lkp, LK_NOPROC, 0, LK_NOCPU);
    941  1.50   thorpej #if defined(LOCKDEBUG)
    942  1.50   thorpej 		lkp->lk_unlock_file = file;
    943  1.50   thorpej 		lkp->lk_unlock_line = line;
    944  1.50   thorpej #endif
    945  1.47  sommerfe 		DONTHAVEIT(lkp);
    946  1.47  sommerfe 	}
    947  1.47  sommerfe #ifdef DIAGNOSTIC
    948  1.47  sommerfe 	else if (lkp->lk_sharecount != 0)
    949  1.47  sommerfe 		panic("spinlock_release_all: release of shared lock!");
    950  1.47  sommerfe 	else
    951  1.47  sommerfe 		panic("spinlock_release_all: release of unlocked lock!");
    952  1.47  sommerfe #endif
    953  1.47  sommerfe 	INTERLOCK_RELEASE(lkp, LK_SPIN, s);
    954  1.47  sommerfe 
    955  1.47  sommerfe 	return (count);
    956  1.47  sommerfe }
    957  1.47  sommerfe 
    958  1.47  sommerfe /*
    959  1.47  sommerfe  * For a recursive spinlock held one or more times by the current CPU,
    960  1.47  sommerfe  * release all N locks, and return N.
    961  1.47  sommerfe  * Intended for use in mi_switch() right after resuming execution.
    962  1.47  sommerfe  */
    963  1.47  sommerfe 
    964  1.47  sommerfe void
    965  1.50   thorpej #if defined(LOCKDEBUG)
    966  1.50   thorpej _spinlock_acquire_count(__volatile struct lock *lkp, int count,
    967  1.50   thorpej     const char *file, int line)
    968  1.50   thorpej #else
    969  1.47  sommerfe spinlock_acquire_count(__volatile struct lock *lkp, int count)
    970  1.50   thorpej #endif
    971  1.47  sommerfe {
    972  1.47  sommerfe 	int s, error;
    973  1.47  sommerfe 	cpuid_t cpu_id;
    974  1.47  sommerfe 
    975  1.47  sommerfe 	KASSERT(lkp->lk_flags & LK_SPIN);
    976  1.47  sommerfe 
    977  1.47  sommerfe 	INTERLOCK_ACQUIRE(lkp, LK_SPIN, s);
    978  1.47  sommerfe 
    979  1.47  sommerfe 	cpu_id = cpu_number();
    980  1.47  sommerfe 
    981  1.47  sommerfe #ifdef DIAGNOSTIC
    982  1.69   thorpej 	if (WEHOLDIT(lkp, LK_NOPROC, 0, cpu_id))
    983  1.64    provos 		panic("spinlock_acquire_count: processor %lu already holds lock", (long)cpu_id);
    984  1.47  sommerfe #endif
    985  1.47  sommerfe 	/*
    986  1.47  sommerfe 	 * Try to acquire the want_exclusive flag.
    987  1.47  sommerfe 	 */
    988  1.78   hannken 	error = acquire(&lkp, &s, LK_SPIN, 0, LK_HAVE_EXCL | LK_WANT_EXCL);
    989  1.47  sommerfe 	lkp->lk_flags |= LK_WANT_EXCL;
    990  1.47  sommerfe 	/*
    991  1.47  sommerfe 	 * Wait for shared locks and upgrades to finish.
    992  1.47  sommerfe 	 */
    993  1.78   hannken 	error = acquire(&lkp, &s, LK_SPIN, 0,
    994  1.74   hannken 	    LK_SHARE_NONZERO | LK_WANT_UPGRADE);
    995  1.47  sommerfe 	lkp->lk_flags &= ~LK_WANT_EXCL;
    996  1.47  sommerfe 	lkp->lk_flags |= LK_HAVE_EXCL;
    997  1.69   thorpej 	SETHOLDER(lkp, LK_NOPROC, 0, cpu_id);
    998  1.50   thorpej #if defined(LOCKDEBUG)
    999  1.50   thorpej 	lkp->lk_lock_file = file;
   1000  1.50   thorpej 	lkp->lk_lock_line = line;
   1001  1.50   thorpej #endif
   1002  1.47  sommerfe 	HAVEIT(lkp);
   1003  1.47  sommerfe 	if (lkp->lk_exclusivecount != 0)
   1004  1.47  sommerfe 		panic("lockmgr: non-zero exclusive count");
   1005  1.47  sommerfe 	lkp->lk_exclusivecount = count;
   1006  1.47  sommerfe 	lkp->lk_recurselevel = 1;
   1007  1.47  sommerfe 	COUNT_CPU(cpu_id, count);
   1008  1.47  sommerfe 
   1009  1.47  sommerfe 	INTERLOCK_RELEASE(lkp, lkp->lk_flags, s);
   1010  1.47  sommerfe }
   1011  1.47  sommerfe 
   1012  1.47  sommerfe 
   1013  1.47  sommerfe 
   1014  1.47  sommerfe /*
   1015   1.1      fvdl  * Print out information about state of a lock. Used by VOP_PRINT
   1016   1.1      fvdl  * routines to display ststus about contained locks.
   1017   1.1      fvdl  */
   1018   1.2      fvdl void
   1019  1.33   thorpej lockmgr_printinfo(__volatile struct lock *lkp)
   1020   1.1      fvdl {
   1021   1.1      fvdl 
   1022   1.1      fvdl 	if (lkp->lk_sharecount)
   1023   1.1      fvdl 		printf(" lock type %s: SHARED (count %d)", lkp->lk_wmesg,
   1024   1.1      fvdl 		    lkp->lk_sharecount);
   1025  1.19   thorpej 	else if (lkp->lk_flags & LK_HAVE_EXCL) {
   1026  1.19   thorpej 		printf(" lock type %s: EXCL (count %d) by ",
   1027  1.19   thorpej 		    lkp->lk_wmesg, lkp->lk_exclusivecount);
   1028  1.19   thorpej 		if (lkp->lk_flags & LK_SPIN)
   1029  1.19   thorpej 			printf("processor %lu", lkp->lk_cpu);
   1030  1.19   thorpej 		else
   1031  1.69   thorpej 			printf("pid %d.%d", lkp->lk_lockholder,
   1032  1.69   thorpej 			    lkp->lk_locklwp);
   1033  1.19   thorpej 	} else
   1034  1.19   thorpej 		printf(" not locked");
   1035  1.19   thorpej 	if ((lkp->lk_flags & LK_SPIN) == 0 && lkp->lk_waitcount > 0)
   1036   1.1      fvdl 		printf(" with %d pending", lkp->lk_waitcount);
   1037   1.1      fvdl }
   1038   1.1      fvdl 
   1039  1.21   thorpej #if defined(LOCKDEBUG) /* { */
   1040  1.21   thorpej TAILQ_HEAD(, simplelock) simplelock_list =
   1041  1.21   thorpej     TAILQ_HEAD_INITIALIZER(simplelock_list);
   1042  1.21   thorpej 
   1043  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
   1044  1.21   thorpej struct simplelock simplelock_list_slock = SIMPLELOCK_INITIALIZER;
   1045  1.21   thorpej 
   1046  1.21   thorpej #define	SLOCK_LIST_LOCK()						\
   1047  1.29  sommerfe 	__cpu_simple_lock(&simplelock_list_slock.lock_data)
   1048  1.21   thorpej 
   1049  1.21   thorpej #define	SLOCK_LIST_UNLOCK()						\
   1050  1.29  sommerfe 	__cpu_simple_unlock(&simplelock_list_slock.lock_data)
   1051  1.21   thorpej 
   1052  1.21   thorpej #define	SLOCK_COUNT(x)							\
   1053  1.47  sommerfe 	curcpu()->ci_simple_locks += (x)
   1054  1.21   thorpej #else
   1055  1.21   thorpej u_long simple_locks;
   1056  1.21   thorpej 
   1057  1.21   thorpej #define	SLOCK_LIST_LOCK()	/* nothing */
   1058  1.21   thorpej 
   1059  1.21   thorpej #define	SLOCK_LIST_UNLOCK()	/* nothing */
   1060  1.21   thorpej 
   1061  1.21   thorpej #define	SLOCK_COUNT(x)		simple_locks += (x)
   1062  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
   1063  1.21   thorpej 
   1064  1.26  sommerfe #ifdef MULTIPROCESSOR
   1065  1.75       wiz #define SLOCK_MP()		lock_printf("on CPU %ld\n", 		\
   1066  1.46   thorpej 				    (u_long) cpu_number())
   1067  1.26  sommerfe #else
   1068  1.26  sommerfe #define SLOCK_MP()		/* nothing */
   1069  1.26  sommerfe #endif
   1070  1.26  sommerfe 
   1071  1.21   thorpej #define	SLOCK_WHERE(str, alp, id, l)					\
   1072  1.21   thorpej do {									\
   1073  1.58       chs 	lock_printf("\n");						\
   1074  1.25   thorpej 	lock_printf(str);						\
   1075  1.33   thorpej 	lock_printf("lock: %p, currently at: %s:%d\n", (alp), (id), (l)); \
   1076  1.26  sommerfe 	SLOCK_MP();							\
   1077  1.21   thorpej 	if ((alp)->lock_file != NULL)					\
   1078  1.25   thorpej 		lock_printf("last locked: %s:%d\n", (alp)->lock_file,	\
   1079  1.21   thorpej 		    (alp)->lock_line);					\
   1080  1.21   thorpej 	if ((alp)->unlock_file != NULL)					\
   1081  1.25   thorpej 		lock_printf("last unlocked: %s:%d\n", (alp)->unlock_file, \
   1082  1.21   thorpej 		    (alp)->unlock_line);				\
   1083  1.58       chs 	SLOCK_TRACE()							\
   1084  1.21   thorpej 	SLOCK_DEBUGGER();						\
   1085  1.30   thorpej } while (/*CONSTCOND*/0)
   1086  1.12       chs 
   1087   1.1      fvdl /*
   1088   1.1      fvdl  * Simple lock functions so that the debugger can see from whence
   1089   1.1      fvdl  * they are being called.
   1090   1.1      fvdl  */
   1091   1.1      fvdl void
   1092  1.33   thorpej simple_lock_init(struct simplelock *alp)
   1093   1.1      fvdl {
   1094  1.21   thorpej 
   1095  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
   1096  1.27   thorpej 	__cpu_simple_lock_init(&alp->lock_data);
   1097  1.21   thorpej #else
   1098  1.27   thorpej 	alp->lock_data = __SIMPLELOCK_UNLOCKED;
   1099  1.21   thorpej #endif /* } */
   1100   1.5       chs 	alp->lock_file = NULL;
   1101   1.5       chs 	alp->lock_line = 0;
   1102   1.5       chs 	alp->unlock_file = NULL;
   1103   1.5       chs 	alp->unlock_line = 0;
   1104  1.41   thorpej 	alp->lock_holder = LK_NOCPU;
   1105   1.1      fvdl }
   1106   1.1      fvdl 
   1107   1.1      fvdl void
   1108  1.33   thorpej _simple_lock(__volatile struct simplelock *alp, const char *id, int l)
   1109   1.1      fvdl {
   1110  1.24   thorpej 	cpuid_t cpu_id = cpu_number();
   1111  1.12       chs 	int s;
   1112  1.12       chs 
   1113  1.44   thorpej 	s = spllock();
   1114  1.21   thorpej 
   1115  1.21   thorpej 	/*
   1116  1.21   thorpej 	 * MULTIPROCESSOR case: This is `safe' since if it's not us, we
   1117  1.21   thorpej 	 * don't take any action, and just fall into the normal spin case.
   1118  1.21   thorpej 	 */
   1119  1.27   thorpej 	if (alp->lock_data == __SIMPLELOCK_LOCKED) {
   1120  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
   1121  1.21   thorpej 		if (alp->lock_holder == cpu_id) {
   1122  1.21   thorpej 			SLOCK_WHERE("simple_lock: locking against myself\n",
   1123  1.21   thorpej 			    alp, id, l);
   1124  1.21   thorpej 			goto out;
   1125   1.1      fvdl 		}
   1126  1.21   thorpej #else
   1127  1.21   thorpej 		SLOCK_WHERE("simple_lock: lock held\n", alp, id, l);
   1128  1.21   thorpej 		goto out;
   1129  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
   1130   1.1      fvdl 	}
   1131  1.21   thorpej 
   1132  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
   1133  1.21   thorpej 	/* Acquire the lock before modifying any fields. */
   1134  1.70        pk 	splx(s);
   1135  1.27   thorpej 	__cpu_simple_lock(&alp->lock_data);
   1136  1.70        pk 	s = spllock();
   1137  1.21   thorpej #else
   1138  1.27   thorpej 	alp->lock_data = __SIMPLELOCK_LOCKED;
   1139  1.21   thorpej #endif /* } */
   1140  1.21   thorpej 
   1141  1.45  sommerfe 	if (alp->lock_holder != LK_NOCPU) {
   1142  1.45  sommerfe 		SLOCK_WHERE("simple_lock: uninitialized lock\n",
   1143  1.45  sommerfe 		    alp, id, l);
   1144  1.45  sommerfe 	}
   1145   1.5       chs 	alp->lock_file = id;
   1146   1.5       chs 	alp->lock_line = l;
   1147  1.21   thorpej 	alp->lock_holder = cpu_id;
   1148  1.21   thorpej 
   1149  1.21   thorpej 	SLOCK_LIST_LOCK();
   1150  1.21   thorpej 	/* XXX Cast away volatile */
   1151  1.21   thorpej 	TAILQ_INSERT_TAIL(&simplelock_list, (struct simplelock *)alp, list);
   1152  1.21   thorpej 	SLOCK_LIST_UNLOCK();
   1153  1.21   thorpej 
   1154  1.21   thorpej 	SLOCK_COUNT(1);
   1155  1.21   thorpej 
   1156  1.21   thorpej  out:
   1157  1.18       chs 	splx(s);
   1158  1.38   thorpej }
   1159  1.38   thorpej 
   1160  1.38   thorpej int
   1161  1.38   thorpej _simple_lock_held(__volatile struct simplelock *alp)
   1162  1.38   thorpej {
   1163  1.54     enami #if defined(MULTIPROCESSOR) || defined(DIAGNOSTIC)
   1164  1.38   thorpej 	cpuid_t cpu_id = cpu_number();
   1165  1.54     enami #endif
   1166  1.38   thorpej 	int s, locked = 0;
   1167  1.38   thorpej 
   1168  1.44   thorpej 	s = spllock();
   1169  1.42   thorpej 
   1170  1.42   thorpej #if defined(MULTIPROCESSOR)
   1171  1.38   thorpej 	if (__cpu_simple_lock_try(&alp->lock_data) == 0)
   1172  1.38   thorpej 		locked = (alp->lock_holder == cpu_id);
   1173  1.38   thorpej 	else
   1174  1.38   thorpej 		__cpu_simple_unlock(&alp->lock_data);
   1175  1.38   thorpej #else
   1176  1.42   thorpej 	if (alp->lock_data == __SIMPLELOCK_LOCKED) {
   1177  1.42   thorpej 		locked = 1;
   1178  1.42   thorpej 		KASSERT(alp->lock_holder == cpu_id);
   1179  1.42   thorpej 	}
   1180  1.42   thorpej #endif
   1181  1.38   thorpej 
   1182  1.38   thorpej 	splx(s);
   1183  1.42   thorpej 
   1184  1.38   thorpej 	return (locked);
   1185   1.1      fvdl }
   1186   1.1      fvdl 
   1187   1.1      fvdl int
   1188  1.33   thorpej _simple_lock_try(__volatile struct simplelock *alp, const char *id, int l)
   1189   1.1      fvdl {
   1190  1.24   thorpej 	cpuid_t cpu_id = cpu_number();
   1191  1.21   thorpej 	int s, rv = 0;
   1192   1.1      fvdl 
   1193  1.44   thorpej 	s = spllock();
   1194  1.21   thorpej 
   1195  1.21   thorpej 	/*
   1196  1.21   thorpej 	 * MULTIPROCESSOR case: This is `safe' since if it's not us, we
   1197  1.21   thorpej 	 * don't take any action.
   1198  1.21   thorpej 	 */
   1199  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
   1200  1.27   thorpej 	if ((rv = __cpu_simple_lock_try(&alp->lock_data)) == 0) {
   1201  1.21   thorpej 		if (alp->lock_holder == cpu_id)
   1202  1.21   thorpej 			SLOCK_WHERE("simple_lock_try: locking against myself\n",
   1203  1.26  sommerfe 			    alp, id, l);
   1204  1.21   thorpej 		goto out;
   1205  1.21   thorpej 	}
   1206  1.21   thorpej #else
   1207  1.27   thorpej 	if (alp->lock_data == __SIMPLELOCK_LOCKED) {
   1208  1.21   thorpej 		SLOCK_WHERE("simple_lock_try: lock held\n", alp, id, l);
   1209  1.21   thorpej 		goto out;
   1210  1.18       chs 	}
   1211  1.27   thorpej 	alp->lock_data = __SIMPLELOCK_LOCKED;
   1212  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
   1213  1.21   thorpej 
   1214  1.21   thorpej 	/*
   1215  1.21   thorpej 	 * At this point, we have acquired the lock.
   1216  1.21   thorpej 	 */
   1217  1.21   thorpej 
   1218  1.21   thorpej 	rv = 1;
   1219  1.18       chs 
   1220   1.5       chs 	alp->lock_file = id;
   1221   1.5       chs 	alp->lock_line = l;
   1222  1.21   thorpej 	alp->lock_holder = cpu_id;
   1223  1.21   thorpej 
   1224  1.21   thorpej 	SLOCK_LIST_LOCK();
   1225  1.21   thorpej 	/* XXX Cast away volatile. */
   1226  1.21   thorpej 	TAILQ_INSERT_TAIL(&simplelock_list, (struct simplelock *)alp, list);
   1227  1.21   thorpej 	SLOCK_LIST_UNLOCK();
   1228  1.21   thorpej 
   1229  1.21   thorpej 	SLOCK_COUNT(1);
   1230  1.21   thorpej 
   1231  1.21   thorpej  out:
   1232  1.12       chs 	splx(s);
   1233  1.21   thorpej 	return (rv);
   1234   1.1      fvdl }
   1235   1.1      fvdl 
   1236   1.1      fvdl void
   1237  1.33   thorpej _simple_unlock(__volatile struct simplelock *alp, const char *id, int l)
   1238   1.1      fvdl {
   1239  1.12       chs 	int s;
   1240   1.1      fvdl 
   1241  1.44   thorpej 	s = spllock();
   1242  1.21   thorpej 
   1243  1.21   thorpej 	/*
   1244  1.21   thorpej 	 * MULTIPROCESSOR case: This is `safe' because we think we hold
   1245  1.21   thorpej 	 * the lock, and if we don't, we don't take any action.
   1246  1.21   thorpej 	 */
   1247  1.27   thorpej 	if (alp->lock_data == __SIMPLELOCK_UNLOCKED) {
   1248  1.21   thorpej 		SLOCK_WHERE("simple_unlock: lock not held\n",
   1249  1.21   thorpej 		    alp, id, l);
   1250  1.21   thorpej 		goto out;
   1251  1.21   thorpej 	}
   1252  1.21   thorpej 
   1253  1.21   thorpej 	SLOCK_LIST_LOCK();
   1254  1.21   thorpej 	TAILQ_REMOVE(&simplelock_list, alp, list);
   1255  1.21   thorpej 	SLOCK_LIST_UNLOCK();
   1256  1.21   thorpej 
   1257  1.21   thorpej 	SLOCK_COUNT(-1);
   1258  1.21   thorpej 
   1259  1.21   thorpej 	alp->list.tqe_next = NULL;	/* sanity */
   1260  1.21   thorpej 	alp->list.tqe_prev = NULL;	/* sanity */
   1261  1.21   thorpej 
   1262   1.5       chs 	alp->unlock_file = id;
   1263   1.5       chs 	alp->unlock_line = l;
   1264  1.21   thorpej 
   1265  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
   1266  1.26  sommerfe 	alp->lock_holder = LK_NOCPU;
   1267  1.21   thorpej 	/* Now that we've modified all fields, release the lock. */
   1268  1.27   thorpej 	__cpu_simple_unlock(&alp->lock_data);
   1269  1.21   thorpej #else
   1270  1.27   thorpej 	alp->lock_data = __SIMPLELOCK_UNLOCKED;
   1271  1.41   thorpej 	KASSERT(alp->lock_holder == cpu_number());
   1272  1.41   thorpej 	alp->lock_holder = LK_NOCPU;
   1273  1.21   thorpej #endif /* } */
   1274  1.21   thorpej 
   1275  1.21   thorpej  out:
   1276  1.18       chs 	splx(s);
   1277  1.12       chs }
   1278  1.12       chs 
   1279  1.12       chs void
   1280  1.33   thorpej simple_lock_dump(void)
   1281  1.12       chs {
   1282  1.12       chs 	struct simplelock *alp;
   1283  1.12       chs 	int s;
   1284  1.12       chs 
   1285  1.44   thorpej 	s = spllock();
   1286  1.21   thorpej 	SLOCK_LIST_LOCK();
   1287  1.25   thorpej 	lock_printf("all simple locks:\n");
   1288  1.58       chs 	TAILQ_FOREACH(alp, &simplelock_list, list) {
   1289  1.25   thorpej 		lock_printf("%p CPU %lu %s:%d\n", alp, alp->lock_holder,
   1290  1.21   thorpej 		    alp->lock_file, alp->lock_line);
   1291  1.12       chs 	}
   1292  1.21   thorpej 	SLOCK_LIST_UNLOCK();
   1293  1.12       chs 	splx(s);
   1294  1.12       chs }
   1295  1.12       chs 
   1296  1.12       chs void
   1297  1.33   thorpej simple_lock_freecheck(void *start, void *end)
   1298  1.12       chs {
   1299  1.12       chs 	struct simplelock *alp;
   1300  1.12       chs 	int s;
   1301  1.12       chs 
   1302  1.44   thorpej 	s = spllock();
   1303  1.21   thorpej 	SLOCK_LIST_LOCK();
   1304  1.58       chs 	TAILQ_FOREACH(alp, &simplelock_list, list) {
   1305  1.12       chs 		if ((void *)alp >= start && (void *)alp < end) {
   1306  1.25   thorpej 			lock_printf("freeing simple_lock %p CPU %lu %s:%d\n",
   1307  1.34   thorpej 			    alp, alp->lock_holder, alp->lock_file,
   1308  1.34   thorpej 			    alp->lock_line);
   1309  1.34   thorpej 			SLOCK_DEBUGGER();
   1310  1.34   thorpej 		}
   1311  1.34   thorpej 	}
   1312  1.34   thorpej 	SLOCK_LIST_UNLOCK();
   1313  1.34   thorpej 	splx(s);
   1314  1.34   thorpej }
   1315  1.34   thorpej 
   1316  1.55   thorpej /*
   1317  1.55   thorpej  * We must be holding exactly one lock: the sched_lock.
   1318  1.55   thorpej  */
   1319  1.55   thorpej 
   1320  1.34   thorpej void
   1321  1.34   thorpej simple_lock_switchcheck(void)
   1322  1.34   thorpej {
   1323  1.55   thorpej 
   1324  1.55   thorpej 	simple_lock_only_held(&sched_lock, "switching");
   1325  1.55   thorpej }
   1326  1.55   thorpej 
   1327  1.55   thorpej void
   1328  1.55   thorpej simple_lock_only_held(volatile struct simplelock *lp, const char *where)
   1329  1.55   thorpej {
   1330  1.34   thorpej 	struct simplelock *alp;
   1331  1.34   thorpej 	cpuid_t cpu_id = cpu_number();
   1332  1.34   thorpej 	int s;
   1333  1.34   thorpej 
   1334  1.55   thorpej 	if (lp) {
   1335  1.55   thorpej 		LOCK_ASSERT(simple_lock_held(lp));
   1336  1.55   thorpej 	}
   1337  1.44   thorpej 	s = spllock();
   1338  1.34   thorpej 	SLOCK_LIST_LOCK();
   1339  1.58       chs 	TAILQ_FOREACH(alp, &simplelock_list, list) {
   1340  1.55   thorpej 		if (alp == lp)
   1341  1.42   thorpej 			continue;
   1342  1.55   thorpej 		if (alp->lock_holder == cpu_id)
   1343  1.55   thorpej 			break;
   1344  1.12       chs 	}
   1345  1.21   thorpej 	SLOCK_LIST_UNLOCK();
   1346  1.12       chs 	splx(s);
   1347  1.55   thorpej 
   1348  1.55   thorpej 	if (alp != NULL) {
   1349  1.58       chs 		lock_printf("\n%s with held simple_lock %p "
   1350  1.55   thorpej 		    "CPU %lu %s:%d\n",
   1351  1.55   thorpej 		    where, alp, alp->lock_holder, alp->lock_file,
   1352  1.55   thorpej 		    alp->lock_line);
   1353  1.58       chs 		SLOCK_TRACE();
   1354  1.55   thorpej 		SLOCK_DEBUGGER();
   1355  1.55   thorpej 	}
   1356   1.1      fvdl }
   1357  1.21   thorpej #endif /* LOCKDEBUG */ /* } */
   1358  1.62   thorpej 
   1359  1.62   thorpej #if defined(MULTIPROCESSOR)
   1360  1.62   thorpej /*
   1361  1.62   thorpej  * Functions for manipulating the kernel_lock.  We put them here
   1362  1.62   thorpej  * so that they show up in profiles.
   1363  1.62   thorpej  */
   1364  1.62   thorpej 
   1365  1.62   thorpej struct lock kernel_lock;
   1366  1.62   thorpej 
   1367  1.62   thorpej void
   1368  1.62   thorpej _kernel_lock_init(void)
   1369  1.62   thorpej {
   1370  1.62   thorpej 
   1371  1.62   thorpej 	spinlockinit(&kernel_lock, "klock", 0);
   1372  1.62   thorpej }
   1373  1.62   thorpej 
   1374  1.62   thorpej /*
   1375  1.62   thorpej  * Acquire/release the kernel lock.  Intended for use in the scheduler
   1376  1.62   thorpej  * and the lower half of the kernel.
   1377  1.62   thorpej  */
   1378  1.62   thorpej void
   1379  1.62   thorpej _kernel_lock(int flag)
   1380  1.62   thorpej {
   1381  1.62   thorpej 
   1382  1.62   thorpej 	SCHED_ASSERT_UNLOCKED();
   1383  1.62   thorpej 	spinlockmgr(&kernel_lock, flag, 0);
   1384  1.62   thorpej }
   1385  1.62   thorpej 
   1386  1.62   thorpej void
   1387  1.62   thorpej _kernel_unlock(void)
   1388  1.62   thorpej {
   1389  1.62   thorpej 
   1390  1.62   thorpej 	spinlockmgr(&kernel_lock, LK_RELEASE, 0);
   1391  1.62   thorpej }
   1392  1.62   thorpej 
   1393  1.62   thorpej /*
   1394  1.62   thorpej  * Acquire/release the kernel_lock on behalf of a process.  Intended for
   1395  1.62   thorpej  * use in the top half of the kernel.
   1396  1.62   thorpej  */
   1397  1.62   thorpej void
   1398  1.69   thorpej _kernel_proc_lock(struct lwp *l)
   1399  1.62   thorpej {
   1400  1.62   thorpej 
   1401  1.62   thorpej 	SCHED_ASSERT_UNLOCKED();
   1402  1.62   thorpej 	spinlockmgr(&kernel_lock, LK_EXCLUSIVE, 0);
   1403  1.62   thorpej }
   1404  1.62   thorpej 
   1405  1.62   thorpej void
   1406  1.69   thorpej _kernel_proc_unlock(struct lwp *l)
   1407  1.62   thorpej {
   1408  1.62   thorpej 
   1409  1.62   thorpej 	spinlockmgr(&kernel_lock, LK_RELEASE, 0);
   1410  1.62   thorpej }
   1411  1.77      yamt 
   1412  1.77      yamt int
   1413  1.77      yamt _kernel_lock_release_all()
   1414  1.77      yamt {
   1415  1.77      yamt 	int hold_count;
   1416  1.77      yamt 
   1417  1.77      yamt 	if (lockstatus(&kernel_lock) == LK_EXCLUSIVE)
   1418  1.77      yamt 		hold_count = spinlock_release_all(&kernel_lock);
   1419  1.77      yamt 	else
   1420  1.77      yamt 		hold_count = 0;
   1421  1.77      yamt 
   1422  1.77      yamt 	return hold_count;
   1423  1.77      yamt }
   1424  1.77      yamt 
   1425  1.77      yamt void
   1426  1.77      yamt _kernel_lock_acquire_count(int hold_count)
   1427  1.77      yamt {
   1428  1.77      yamt 
   1429  1.77      yamt 	if (hold_count != 0)
   1430  1.77      yamt 		spinlock_acquire_count(&kernel_lock, hold_count);
   1431  1.77      yamt }
   1432  1.62   thorpej #endif /* MULTIPROCESSOR */
   1433