Home | History | Annotate | Line # | Download | only in kern
kern_lock.c revision 1.44
      1  1.44   thorpej /*	$NetBSD: kern_lock.c,v 1.44 2000/08/22 19:47:26 thorpej 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.1      fvdl  * 3. All advertising materials mentioning features or use of this software
     60   1.1      fvdl  *    must display the following acknowledgement:
     61   1.1      fvdl  *	This product includes software developed by the University of
     62   1.1      fvdl  *	California, Berkeley and its contributors.
     63   1.1      fvdl  * 4. Neither the name of the University nor the names of its contributors
     64   1.1      fvdl  *    may be used to endorse or promote products derived from this software
     65   1.1      fvdl  *    without specific prior written permission.
     66   1.1      fvdl  *
     67   1.1      fvdl  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     68   1.1      fvdl  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     69   1.1      fvdl  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     70   1.1      fvdl  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     71   1.1      fvdl  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     72   1.1      fvdl  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     73   1.1      fvdl  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     74   1.1      fvdl  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     75   1.1      fvdl  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     76   1.1      fvdl  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     77   1.1      fvdl  * SUCH DAMAGE.
     78   1.1      fvdl  *
     79   1.1      fvdl  *	@(#)kern_lock.c	8.18 (Berkeley) 5/21/95
     80   1.1      fvdl  */
     81   1.7   thorpej 
     82  1.21   thorpej #include "opt_multiprocessor.h"
     83   1.7   thorpej #include "opt_lockdebug.h"
     84  1.18       chs #include "opt_ddb.h"
     85   1.1      fvdl 
     86   1.1      fvdl #include <sys/param.h>
     87   1.1      fvdl #include <sys/proc.h>
     88   1.1      fvdl #include <sys/lock.h>
     89   1.2      fvdl #include <sys/systm.h>
     90   1.1      fvdl #include <machine/cpu.h>
     91   1.1      fvdl 
     92  1.28   thorpej #if defined(__HAVE_ATOMIC_OPERATIONS)
     93  1.28   thorpej #include <machine/atomic.h>
     94  1.28   thorpej #endif
     95  1.28   thorpej 
     96  1.25   thorpej #if defined(LOCKDEBUG)
     97  1.25   thorpej #include <sys/syslog.h>
     98  1.25   thorpej /*
     99  1.25   thorpej  * note that stdarg.h and the ansi style va_start macro is used for both
    100  1.25   thorpej  * ansi and traditional c compiles.
    101  1.25   thorpej  * XXX: this requires that stdarg.h define: va_alist and va_dcl
    102  1.25   thorpej  */
    103  1.25   thorpej #include <machine/stdarg.h>
    104  1.25   thorpej 
    105  1.36   thorpej void	lock_printf(const char *fmt, ...)
    106  1.37       eeh     __attribute__((__format__(__printf__,1,2)));
    107  1.25   thorpej 
    108  1.25   thorpej int	lock_debug_syslog = 0;	/* defaults to printf, but can be patched */
    109  1.25   thorpej #endif
    110  1.25   thorpej 
    111   1.1      fvdl /*
    112   1.1      fvdl  * Locking primitives implementation.
    113   1.1      fvdl  * Locks provide shared/exclusive sychronization.
    114   1.1      fvdl  */
    115   1.1      fvdl 
    116  1.21   thorpej #if defined(LOCKDEBUG) || defined(DIAGNOSTIC) /* { */
    117  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    118  1.28   thorpej #if defined(__HAVE_ATOMIC_OPERATIONS) /* { */
    119  1.21   thorpej #define	COUNT_CPU(cpu_id, x)						\
    120  1.30   thorpej 	atomic_add_ulong(&curcpu()->ci_spin_locks, (x))
    121  1.28   thorpej #else
    122  1.28   thorpej #define	COUNT_CPU(cpu_id, x)	/* not safe */
    123  1.28   thorpej #endif /* __HAVE_ATOMIC_OPERATIONS */ /* } */
    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.21   thorpej #define	COUNT(lkp, p, 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.21   thorpej 		(p)->p_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.21   thorpej #endif /* LOCKDEBUG || DIAGNOSTIC */ /* } */
    139   1.1      fvdl 
    140  1.43   thorpej #define	INTERLOCK_ACQUIRE(lkp, flags, s)				\
    141  1.40   thorpej do {									\
    142  1.43   thorpej 	if ((flags) & LK_SPIN)						\
    143  1.42   thorpej 		s = splsched();						\
    144  1.40   thorpej 	simple_lock(&(lkp)->lk_interlock);				\
    145  1.40   thorpej } while (0)
    146  1.40   thorpej 
    147  1.43   thorpej #define	INTERLOCK_RELEASE(lkp, flags, s)				\
    148  1.40   thorpej do {									\
    149  1.40   thorpej 	simple_unlock(&(lkp)->lk_interlock);				\
    150  1.43   thorpej 	if ((flags) & LK_SPIN)						\
    151  1.40   thorpej 		splx(s);						\
    152  1.40   thorpej } while (0)
    153  1.40   thorpej 
    154   1.1      fvdl /*
    155   1.1      fvdl  * Acquire a resource.
    156   1.1      fvdl  */
    157  1.23   thorpej #define ACQUIRE(lkp, error, extflags, drain, wanted)			\
    158  1.19   thorpej 	if ((extflags) & LK_SPIN) {					\
    159  1.19   thorpej 		int interlocked;					\
    160  1.19   thorpej 									\
    161  1.23   thorpej 		if ((drain) == 0)					\
    162  1.23   thorpej 			(lkp)->lk_waitcount++;				\
    163  1.19   thorpej 		for (interlocked = 1;;) {				\
    164  1.19   thorpej 			if (wanted) {					\
    165  1.19   thorpej 				if (interlocked) {			\
    166  1.43   thorpej 					INTERLOCK_RELEASE((lkp),	\
    167  1.43   thorpej 					    LK_SPIN, s);		\
    168  1.19   thorpej 					interlocked = 0;		\
    169  1.19   thorpej 				}					\
    170  1.19   thorpej 			} else if (interlocked) {			\
    171  1.19   thorpej 				break;					\
    172  1.19   thorpej 			} else {					\
    173  1.43   thorpej 				INTERLOCK_ACQUIRE((lkp), LK_SPIN, s);	\
    174  1.19   thorpej 				interlocked = 1;			\
    175  1.19   thorpej 			}						\
    176  1.19   thorpej 		}							\
    177  1.23   thorpej 		if ((drain) == 0)					\
    178  1.23   thorpej 			(lkp)->lk_waitcount--;				\
    179  1.19   thorpej 		KASSERT((wanted) == 0);					\
    180  1.19   thorpej 		error = 0;	/* sanity */				\
    181  1.19   thorpej 	} else {							\
    182  1.19   thorpej 		for (error = 0; wanted; ) {				\
    183  1.23   thorpej 			if ((drain))					\
    184  1.23   thorpej 				(lkp)->lk_flags |= LK_WAITDRAIN;	\
    185  1.23   thorpej 			else						\
    186  1.23   thorpej 				(lkp)->lk_waitcount++;			\
    187  1.23   thorpej 			/* XXX Cast away volatile. */			\
    188  1.31   thorpej 			error = ltsleep((drain) ? &(lkp)->lk_flags :	\
    189  1.23   thorpej 			    (void *)(lkp), (lkp)->lk_prio,		\
    190  1.31   thorpej 			    (lkp)->lk_wmesg, (lkp)->lk_timo,		\
    191  1.31   thorpej 			    &(lkp)->lk_interlock);			\
    192  1.23   thorpej 			if ((drain) == 0)				\
    193  1.23   thorpej 				(lkp)->lk_waitcount--;			\
    194  1.19   thorpej 			if (error)					\
    195  1.19   thorpej 				break;					\
    196  1.19   thorpej 			if ((extflags) & LK_SLEEPFAIL) {		\
    197  1.19   thorpej 				error = ENOLCK;				\
    198  1.19   thorpej 				break;					\
    199  1.19   thorpej 			}						\
    200   1.1      fvdl 		}							\
    201   1.1      fvdl 	}
    202   1.1      fvdl 
    203  1.19   thorpej #define	SETHOLDER(lkp, pid, cpu_id)					\
    204  1.19   thorpej do {									\
    205  1.19   thorpej 	if ((lkp)->lk_flags & LK_SPIN)					\
    206  1.19   thorpej 		(lkp)->lk_cpu = cpu_id;					\
    207  1.19   thorpej 	else								\
    208  1.19   thorpej 		(lkp)->lk_lockholder = pid;				\
    209  1.30   thorpej } while (/*CONSTCOND*/0)
    210  1.19   thorpej 
    211  1.19   thorpej #define	WEHOLDIT(lkp, pid, cpu_id)					\
    212  1.19   thorpej 	(((lkp)->lk_flags & LK_SPIN) != 0 ?				\
    213  1.19   thorpej 	 ((lkp)->lk_cpu == (cpu_id)) : ((lkp)->lk_lockholder == (pid)))
    214  1.19   thorpej 
    215  1.23   thorpej #define	WAKEUP_WAITER(lkp)						\
    216  1.23   thorpej do {									\
    217  1.23   thorpej 	if (((lkp)->lk_flags & LK_SPIN) == 0 && (lkp)->lk_waitcount) {	\
    218  1.23   thorpej 		/* XXX Cast away volatile. */				\
    219  1.23   thorpej 		wakeup_one((void *)(lkp));				\
    220  1.23   thorpej 	}								\
    221  1.30   thorpej } while (/*CONSTCOND*/0)
    222  1.23   thorpej 
    223  1.21   thorpej #if defined(LOCKDEBUG) /* { */
    224  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    225  1.21   thorpej struct simplelock spinlock_list_slock = SIMPLELOCK_INITIALIZER;
    226  1.21   thorpej 
    227  1.27   thorpej #define	SPINLOCK_LIST_LOCK()						\
    228  1.29  sommerfe 	__cpu_simple_lock(&spinlock_list_slock.lock_data)
    229  1.21   thorpej 
    230  1.27   thorpej #define	SPINLOCK_LIST_UNLOCK()						\
    231  1.29  sommerfe 	__cpu_simple_unlock(&spinlock_list_slock.lock_data)
    232  1.21   thorpej #else
    233  1.21   thorpej #define	SPINLOCK_LIST_LOCK()	/* nothing */
    234  1.21   thorpej 
    235  1.21   thorpej #define	SPINLOCK_LIST_UNLOCK()	/* nothing */
    236  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
    237  1.21   thorpej 
    238  1.21   thorpej TAILQ_HEAD(, lock) spinlock_list =
    239  1.21   thorpej     TAILQ_HEAD_INITIALIZER(spinlock_list);
    240  1.21   thorpej 
    241  1.21   thorpej #define	HAVEIT(lkp)							\
    242  1.21   thorpej do {									\
    243  1.21   thorpej 	if ((lkp)->lk_flags & LK_SPIN) {				\
    244  1.44   thorpej 		int s = spllock();					\
    245  1.21   thorpej 		SPINLOCK_LIST_LOCK();					\
    246  1.21   thorpej 		/* XXX Cast away volatile. */				\
    247  1.21   thorpej 		TAILQ_INSERT_TAIL(&spinlock_list, (struct lock *)(lkp),	\
    248  1.21   thorpej 		    lk_list);						\
    249  1.21   thorpej 		SPINLOCK_LIST_UNLOCK();					\
    250  1.21   thorpej 		splx(s);						\
    251  1.21   thorpej 	}								\
    252  1.30   thorpej } while (/*CONSTCOND*/0)
    253  1.21   thorpej 
    254  1.21   thorpej #define	DONTHAVEIT(lkp)							\
    255  1.21   thorpej do {									\
    256  1.21   thorpej 	if ((lkp)->lk_flags & LK_SPIN) {				\
    257  1.44   thorpej 		int s = spllock();					\
    258  1.21   thorpej 		SPINLOCK_LIST_LOCK();					\
    259  1.21   thorpej 		/* XXX Cast away volatile. */				\
    260  1.21   thorpej 		TAILQ_REMOVE(&spinlock_list, (struct lock *)(lkp),	\
    261  1.21   thorpej 		    lk_list);						\
    262  1.21   thorpej 		SPINLOCK_LIST_UNLOCK();					\
    263  1.21   thorpej 		splx(s);						\
    264  1.21   thorpej 	}								\
    265  1.30   thorpej } while (/*CONSTCOND*/0)
    266  1.21   thorpej #else
    267  1.21   thorpej #define	HAVEIT(lkp)		/* nothing */
    268  1.21   thorpej 
    269  1.21   thorpej #define	DONTHAVEIT(lkp)		/* nothing */
    270  1.21   thorpej #endif /* LOCKDEBUG */ /* } */
    271  1.21   thorpej 
    272  1.25   thorpej #if defined(LOCKDEBUG)
    273  1.25   thorpej /*
    274  1.25   thorpej  * Lock debug printing routine; can be configured to print to console
    275  1.25   thorpej  * or log to syslog.
    276  1.25   thorpej  */
    277  1.25   thorpej void
    278  1.25   thorpej lock_printf(const char *fmt, ...)
    279  1.25   thorpej {
    280  1.25   thorpej 	va_list ap;
    281  1.25   thorpej 
    282  1.25   thorpej 	va_start(ap, fmt);
    283  1.25   thorpej 	if (lock_debug_syslog)
    284  1.25   thorpej 		vlog(LOG_DEBUG, fmt, ap);
    285  1.25   thorpej 	else
    286  1.25   thorpej 		vprintf(fmt, ap);
    287  1.25   thorpej 	va_end(ap);
    288  1.25   thorpej }
    289  1.25   thorpej #endif /* LOCKDEBUG */
    290  1.25   thorpej 
    291   1.1      fvdl /*
    292   1.1      fvdl  * Initialize a lock; required before use.
    293   1.1      fvdl  */
    294   1.1      fvdl void
    295  1.33   thorpej lockinit(struct lock *lkp, int prio, const char *wmesg, int timo, int flags)
    296   1.1      fvdl {
    297   1.1      fvdl 
    298   1.8     perry 	memset(lkp, 0, sizeof(struct lock));
    299   1.1      fvdl 	simple_lock_init(&lkp->lk_interlock);
    300   1.1      fvdl 	lkp->lk_flags = flags & LK_EXTFLG_MASK;
    301  1.19   thorpej 	if (flags & LK_SPIN)
    302  1.19   thorpej 		lkp->lk_cpu = LK_NOCPU;
    303  1.19   thorpej 	else {
    304  1.19   thorpej 		lkp->lk_lockholder = LK_NOPROC;
    305  1.19   thorpej 		lkp->lk_prio = prio;
    306  1.19   thorpej 		lkp->lk_timo = timo;
    307  1.19   thorpej 	}
    308  1.19   thorpej 	lkp->lk_wmesg = wmesg;	/* just a name for spin locks */
    309   1.1      fvdl }
    310   1.1      fvdl 
    311   1.1      fvdl /*
    312   1.1      fvdl  * Determine the status of a lock.
    313   1.1      fvdl  */
    314   1.1      fvdl int
    315  1.33   thorpej lockstatus(struct lock *lkp)
    316   1.1      fvdl {
    317  1.40   thorpej 	int s, lock_type = 0;
    318   1.1      fvdl 
    319  1.43   thorpej 	INTERLOCK_ACQUIRE(lkp, lkp->lk_flags, s);
    320   1.1      fvdl 	if (lkp->lk_exclusivecount != 0)
    321   1.1      fvdl 		lock_type = LK_EXCLUSIVE;
    322   1.1      fvdl 	else if (lkp->lk_sharecount != 0)
    323   1.1      fvdl 		lock_type = LK_SHARED;
    324  1.43   thorpej 	INTERLOCK_RELEASE(lkp, lkp->lk_flags, s);
    325   1.1      fvdl 	return (lock_type);
    326   1.1      fvdl }
    327  1.35   thorpej 
    328  1.35   thorpej #if defined(LOCKDEBUG) || defined(DIAGNOSTIC)
    329  1.35   thorpej /*
    330  1.35   thorpej  * Make sure no spin locks are held by a CPU that is about
    331  1.35   thorpej  * to context switch.
    332  1.35   thorpej  */
    333  1.35   thorpej void
    334  1.35   thorpej spinlock_switchcheck(void)
    335  1.35   thorpej {
    336  1.35   thorpej 	u_long cnt;
    337  1.35   thorpej 	int s;
    338  1.35   thorpej 
    339  1.44   thorpej 	s = spllock();
    340  1.35   thorpej #if defined(MULTIPROCESSOR)
    341  1.35   thorpej 	cnt = curcpu()->ci_spin_locks;
    342  1.35   thorpej #else
    343  1.35   thorpej 	cnt = spin_locks;
    344  1.35   thorpej #endif
    345  1.35   thorpej 	splx(s);
    346  1.35   thorpej 
    347  1.35   thorpej 	if (cnt != 0)
    348  1.35   thorpej 		panic("spinlock_switchcheck: CPU %lu has %lu spin locks",
    349  1.35   thorpej 		    (u_long) cpu_number(), cnt);
    350  1.35   thorpej }
    351  1.35   thorpej #endif /* LOCKDEBUG || DIAGNOSTIC */
    352   1.1      fvdl 
    353   1.1      fvdl /*
    354  1.44   thorpej  * Locks and IPLs (interrupt priority levels):
    355  1.44   thorpej  *
    356  1.44   thorpej  * Locks which may be taken from interrupt context must be handled
    357  1.44   thorpej  * very carefully; you must spl to the highest IPL where the lock
    358  1.44   thorpej  * is needed before acquiring the lock.
    359  1.44   thorpej  *
    360  1.44   thorpej  * It is also important to avoid deadlock, since certain (very high
    361  1.44   thorpej  * priority) interrupts are often needed to keep the system as a whole
    362  1.44   thorpej  * from deadlocking, and must not be blocked while you are spinning
    363  1.44   thorpej  * waiting for a lower-priority lock.
    364  1.44   thorpej  *
    365  1.44   thorpej  * In addition, the lock-debugging hooks themselves need to use locks!
    366  1.44   thorpej  *
    367  1.44   thorpej  * A raw __cpu_simple_lock may be used from interrupts are long as it
    368  1.44   thorpej  * is acquired and held at a single IPL.
    369  1.44   thorpej  *
    370  1.44   thorpej  * A simple_lock (which is a __cpu_simple_lock wrapped with some
    371  1.44   thorpej  * debugging hooks) may be used at or below spllock(), which is
    372  1.44   thorpej  * typically at or just below splhigh() (i.e. blocks everything
    373  1.44   thorpej  * but certain machine-dependent extremely high priority interrupts).
    374  1.44   thorpej  *
    375  1.44   thorpej  * spinlockmgr spinlocks should be used at or below splsched().
    376  1.44   thorpej  *
    377  1.44   thorpej  * Some platforms may have interrupts of higher priority than splsched(),
    378  1.44   thorpej  * including hard serial interrupts, inter-processor interrupts, and
    379  1.44   thorpej  * kernel debugger traps.
    380  1.44   thorpej  */
    381  1.44   thorpej 
    382  1.44   thorpej /*
    383  1.32  sommerfe  * XXX XXX kludge around another kludge..
    384  1.32  sommerfe  *
    385  1.32  sommerfe  * vfs_shutdown() may be called from interrupt context, either as a result
    386  1.32  sommerfe  * of a panic, or from the debugger.   It proceeds to call
    387  1.32  sommerfe  * sys_sync(&proc0, ...), pretending its running on behalf of proc0
    388  1.32  sommerfe  *
    389  1.32  sommerfe  * We would like to make an attempt to sync the filesystems in this case, so
    390  1.32  sommerfe  * if this happens, we treat attempts to acquire locks specially.
    391  1.32  sommerfe  * All locks are acquired on behalf of proc0.
    392  1.32  sommerfe  *
    393  1.32  sommerfe  * If we've already paniced, we don't block waiting for locks, but
    394  1.32  sommerfe  * just barge right ahead since we're already going down in flames.
    395  1.32  sommerfe  */
    396  1.32  sommerfe 
    397  1.32  sommerfe /*
    398   1.1      fvdl  * Set, change, or release a lock.
    399   1.1      fvdl  *
    400   1.1      fvdl  * Shared requests increment the shared count. Exclusive requests set the
    401   1.1      fvdl  * LK_WANT_EXCL flag (preventing further shared locks), and wait for already
    402   1.1      fvdl  * accepted shared locks and shared-to-exclusive upgrades to go away.
    403   1.1      fvdl  */
    404   1.1      fvdl int
    405  1.33   thorpej lockmgr(__volatile struct lock *lkp, u_int flags,
    406  1.33   thorpej     struct simplelock *interlkp)
    407   1.1      fvdl {
    408   1.1      fvdl 	int error;
    409   1.1      fvdl 	pid_t pid;
    410   1.1      fvdl 	int extflags;
    411  1.24   thorpej 	cpuid_t cpu_id;
    412   1.6      fvdl 	struct proc *p = curproc;
    413  1.32  sommerfe 	int lock_shutdown_noblock = 0;
    414  1.40   thorpej 	int s;
    415   1.1      fvdl 
    416   1.1      fvdl 	error = 0;
    417  1.19   thorpej 
    418  1.43   thorpej 	INTERLOCK_ACQUIRE(lkp, lkp->lk_flags, s);
    419   1.1      fvdl 	if (flags & LK_INTERLOCK)
    420   1.1      fvdl 		simple_unlock(interlkp);
    421   1.1      fvdl 	extflags = (flags | lkp->lk_flags) & LK_EXTFLG_MASK;
    422  1.19   thorpej 
    423  1.21   thorpej #ifdef DIAGNOSTIC /* { */
    424  1.19   thorpej 	/*
    425  1.19   thorpej 	 * Don't allow spins on sleep locks and don't allow sleeps
    426  1.19   thorpej 	 * on spin locks.
    427  1.19   thorpej 	 */
    428  1.19   thorpej 	if ((flags ^ lkp->lk_flags) & LK_SPIN)
    429  1.19   thorpej 		panic("lockmgr: sleep/spin mismatch\n");
    430  1.21   thorpej #endif /* } */
    431  1.19   thorpej 
    432  1.19   thorpej 	if (extflags & LK_SPIN)
    433  1.19   thorpej 		pid = LK_KERNPROC;
    434  1.19   thorpej 	else {
    435  1.32  sommerfe 		if (p == NULL) {
    436  1.32  sommerfe 			if (!doing_shutdown) {
    437  1.32  sommerfe #ifdef DIAGNOSTIC
    438  1.32  sommerfe 				panic("lockmgr: no context");
    439  1.32  sommerfe #endif
    440  1.32  sommerfe 			} else {
    441  1.32  sommerfe 				p = &proc0;
    442  1.32  sommerfe 				if (panicstr && (!(flags & LK_NOWAIT))) {
    443  1.32  sommerfe 					flags |= LK_NOWAIT;
    444  1.32  sommerfe 					lock_shutdown_noblock = 1;
    445  1.32  sommerfe 				}
    446  1.32  sommerfe 			}
    447  1.32  sommerfe 		}
    448  1.19   thorpej 		pid = p->p_pid;
    449  1.19   thorpej 	}
    450  1.24   thorpej 	cpu_id = cpu_number();
    451  1.19   thorpej 
    452   1.1      fvdl 	/*
    453   1.1      fvdl 	 * Once a lock has drained, the LK_DRAINING flag is set and an
    454   1.1      fvdl 	 * exclusive lock is returned. The only valid operation thereafter
    455   1.1      fvdl 	 * is a single release of that exclusive lock. This final release
    456   1.1      fvdl 	 * clears the LK_DRAINING flag and sets the LK_DRAINED flag. Any
    457   1.1      fvdl 	 * further requests of any sort will result in a panic. The bits
    458   1.1      fvdl 	 * selected for these two flags are chosen so that they will be set
    459   1.1      fvdl 	 * in memory that is freed (freed memory is filled with 0xdeadbeef).
    460   1.1      fvdl 	 * The final release is permitted to give a new lease on life to
    461   1.1      fvdl 	 * the lock by specifying LK_REENABLE.
    462   1.1      fvdl 	 */
    463   1.1      fvdl 	if (lkp->lk_flags & (LK_DRAINING|LK_DRAINED)) {
    464  1.28   thorpej #ifdef DIAGNOSTIC /* { */
    465   1.1      fvdl 		if (lkp->lk_flags & LK_DRAINED)
    466   1.1      fvdl 			panic("lockmgr: using decommissioned lock");
    467   1.1      fvdl 		if ((flags & LK_TYPE_MASK) != LK_RELEASE ||
    468  1.19   thorpej 		    WEHOLDIT(lkp, pid, cpu_id) == 0)
    469   1.1      fvdl 			panic("lockmgr: non-release on draining lock: %d\n",
    470   1.1      fvdl 			    flags & LK_TYPE_MASK);
    471  1.28   thorpej #endif /* DIAGNOSTIC */ /* } */
    472   1.1      fvdl 		lkp->lk_flags &= ~LK_DRAINING;
    473   1.1      fvdl 		if ((flags & LK_REENABLE) == 0)
    474   1.1      fvdl 			lkp->lk_flags |= LK_DRAINED;
    475   1.1      fvdl 	}
    476   1.1      fvdl 
    477   1.1      fvdl 	switch (flags & LK_TYPE_MASK) {
    478   1.1      fvdl 
    479   1.1      fvdl 	case LK_SHARED:
    480  1.19   thorpej 		if (WEHOLDIT(lkp, pid, cpu_id) == 0) {
    481   1.1      fvdl 			/*
    482   1.1      fvdl 			 * If just polling, check to see if we will block.
    483   1.1      fvdl 			 */
    484   1.1      fvdl 			if ((extflags & LK_NOWAIT) && (lkp->lk_flags &
    485   1.1      fvdl 			    (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE))) {
    486   1.1      fvdl 				error = EBUSY;
    487   1.1      fvdl 				break;
    488   1.1      fvdl 			}
    489   1.1      fvdl 			/*
    490   1.1      fvdl 			 * Wait for exclusive locks and upgrades to clear.
    491   1.1      fvdl 			 */
    492  1.23   thorpej 			ACQUIRE(lkp, error, extflags, 0, lkp->lk_flags &
    493   1.1      fvdl 			    (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE));
    494   1.1      fvdl 			if (error)
    495   1.1      fvdl 				break;
    496   1.1      fvdl 			lkp->lk_sharecount++;
    497  1.21   thorpej 			COUNT(lkp, p, cpu_id, 1);
    498   1.1      fvdl 			break;
    499   1.1      fvdl 		}
    500   1.1      fvdl 		/*
    501   1.1      fvdl 		 * We hold an exclusive lock, so downgrade it to shared.
    502   1.1      fvdl 		 * An alternative would be to fail with EDEADLK.
    503   1.1      fvdl 		 */
    504   1.1      fvdl 		lkp->lk_sharecount++;
    505  1.21   thorpej 		COUNT(lkp, p, cpu_id, 1);
    506   1.1      fvdl 		/* fall into downgrade */
    507   1.1      fvdl 
    508   1.1      fvdl 	case LK_DOWNGRADE:
    509  1.19   thorpej 		if (WEHOLDIT(lkp, pid, cpu_id) == 0 ||
    510  1.19   thorpej 		    lkp->lk_exclusivecount == 0)
    511   1.1      fvdl 			panic("lockmgr: not holding exclusive lock");
    512   1.1      fvdl 		lkp->lk_sharecount += lkp->lk_exclusivecount;
    513   1.1      fvdl 		lkp->lk_exclusivecount = 0;
    514  1.15      fvdl 		lkp->lk_recurselevel = 0;
    515   1.1      fvdl 		lkp->lk_flags &= ~LK_HAVE_EXCL;
    516  1.19   thorpej 		SETHOLDER(lkp, LK_NOPROC, LK_NOCPU);
    517  1.21   thorpej 		DONTHAVEIT(lkp);
    518  1.23   thorpej 		WAKEUP_WAITER(lkp);
    519   1.1      fvdl 		break;
    520   1.1      fvdl 
    521   1.1      fvdl 	case LK_EXCLUPGRADE:
    522   1.1      fvdl 		/*
    523   1.1      fvdl 		 * If another process is ahead of us to get an upgrade,
    524   1.1      fvdl 		 * then we want to fail rather than have an intervening
    525   1.1      fvdl 		 * exclusive access.
    526   1.1      fvdl 		 */
    527   1.1      fvdl 		if (lkp->lk_flags & LK_WANT_UPGRADE) {
    528   1.1      fvdl 			lkp->lk_sharecount--;
    529  1.21   thorpej 			COUNT(lkp, p, cpu_id, -1);
    530   1.1      fvdl 			error = EBUSY;
    531   1.1      fvdl 			break;
    532   1.1      fvdl 		}
    533   1.1      fvdl 		/* fall into normal upgrade */
    534   1.1      fvdl 
    535   1.1      fvdl 	case LK_UPGRADE:
    536   1.1      fvdl 		/*
    537   1.1      fvdl 		 * Upgrade a shared lock to an exclusive one. If another
    538   1.1      fvdl 		 * shared lock has already requested an upgrade to an
    539   1.1      fvdl 		 * exclusive lock, our shared lock is released and an
    540   1.1      fvdl 		 * exclusive lock is requested (which will be granted
    541   1.1      fvdl 		 * after the upgrade). If we return an error, the file
    542   1.1      fvdl 		 * will always be unlocked.
    543   1.1      fvdl 		 */
    544  1.19   thorpej 		if (WEHOLDIT(lkp, pid, cpu_id) || lkp->lk_sharecount <= 0)
    545   1.1      fvdl 			panic("lockmgr: upgrade exclusive lock");
    546   1.1      fvdl 		lkp->lk_sharecount--;
    547  1.21   thorpej 		COUNT(lkp, p, cpu_id, -1);
    548   1.1      fvdl 		/*
    549   1.1      fvdl 		 * If we are just polling, check to see if we will block.
    550   1.1      fvdl 		 */
    551   1.1      fvdl 		if ((extflags & LK_NOWAIT) &&
    552   1.1      fvdl 		    ((lkp->lk_flags & LK_WANT_UPGRADE) ||
    553   1.1      fvdl 		     lkp->lk_sharecount > 1)) {
    554   1.1      fvdl 			error = EBUSY;
    555   1.1      fvdl 			break;
    556   1.1      fvdl 		}
    557   1.1      fvdl 		if ((lkp->lk_flags & LK_WANT_UPGRADE) == 0) {
    558   1.1      fvdl 			/*
    559   1.1      fvdl 			 * We are first shared lock to request an upgrade, so
    560   1.1      fvdl 			 * request upgrade and wait for the shared count to
    561   1.1      fvdl 			 * drop to zero, then take exclusive lock.
    562   1.1      fvdl 			 */
    563   1.1      fvdl 			lkp->lk_flags |= LK_WANT_UPGRADE;
    564  1.23   thorpej 			ACQUIRE(lkp, error, extflags, 0, lkp->lk_sharecount);
    565   1.1      fvdl 			lkp->lk_flags &= ~LK_WANT_UPGRADE;
    566   1.1      fvdl 			if (error)
    567   1.1      fvdl 				break;
    568   1.1      fvdl 			lkp->lk_flags |= LK_HAVE_EXCL;
    569  1.19   thorpej 			SETHOLDER(lkp, pid, cpu_id);
    570  1.21   thorpej 			HAVEIT(lkp);
    571   1.1      fvdl 			if (lkp->lk_exclusivecount != 0)
    572   1.1      fvdl 				panic("lockmgr: non-zero exclusive count");
    573   1.1      fvdl 			lkp->lk_exclusivecount = 1;
    574  1.15      fvdl 			if (extflags & LK_SETRECURSE)
    575  1.15      fvdl 				lkp->lk_recurselevel = 1;
    576  1.21   thorpej 			COUNT(lkp, p, cpu_id, 1);
    577   1.1      fvdl 			break;
    578   1.1      fvdl 		}
    579   1.1      fvdl 		/*
    580   1.1      fvdl 		 * Someone else has requested upgrade. Release our shared
    581   1.1      fvdl 		 * lock, awaken upgrade requestor if we are the last shared
    582   1.1      fvdl 		 * lock, then request an exclusive lock.
    583   1.1      fvdl 		 */
    584  1.23   thorpej 		if (lkp->lk_sharecount == 0)
    585  1.23   thorpej 			WAKEUP_WAITER(lkp);
    586   1.1      fvdl 		/* fall into exclusive request */
    587   1.1      fvdl 
    588   1.1      fvdl 	case LK_EXCLUSIVE:
    589  1.19   thorpej 		if (WEHOLDIT(lkp, pid, cpu_id)) {
    590   1.1      fvdl 			/*
    591  1.19   thorpej 			 * Recursive lock.
    592   1.1      fvdl 			 */
    593  1.15      fvdl 			if ((extflags & LK_CANRECURSE) == 0 &&
    594  1.16  sommerfe 			     lkp->lk_recurselevel == 0) {
    595  1.16  sommerfe 				if (extflags & LK_RECURSEFAIL) {
    596  1.16  sommerfe 					error = EDEADLK;
    597  1.16  sommerfe 					break;
    598  1.16  sommerfe 				} else
    599  1.16  sommerfe 					panic("lockmgr: locking against myself");
    600  1.16  sommerfe 			}
    601   1.1      fvdl 			lkp->lk_exclusivecount++;
    602  1.15      fvdl 			if (extflags & LK_SETRECURSE &&
    603  1.15      fvdl 			    lkp->lk_recurselevel == 0)
    604  1.15      fvdl 				lkp->lk_recurselevel = lkp->lk_exclusivecount;
    605  1.21   thorpej 			COUNT(lkp, p, cpu_id, 1);
    606   1.1      fvdl 			break;
    607   1.1      fvdl 		}
    608   1.1      fvdl 		/*
    609   1.1      fvdl 		 * If we are just polling, check to see if we will sleep.
    610   1.1      fvdl 		 */
    611   1.1      fvdl 		if ((extflags & LK_NOWAIT) && ((lkp->lk_flags &
    612   1.1      fvdl 		     (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE)) ||
    613   1.1      fvdl 		     lkp->lk_sharecount != 0)) {
    614   1.1      fvdl 			error = EBUSY;
    615   1.1      fvdl 			break;
    616   1.1      fvdl 		}
    617   1.1      fvdl 		/*
    618   1.1      fvdl 		 * Try to acquire the want_exclusive flag.
    619   1.1      fvdl 		 */
    620  1.23   thorpej 		ACQUIRE(lkp, error, extflags, 0, lkp->lk_flags &
    621   1.1      fvdl 		    (LK_HAVE_EXCL | LK_WANT_EXCL));
    622   1.1      fvdl 		if (error)
    623   1.1      fvdl 			break;
    624   1.1      fvdl 		lkp->lk_flags |= LK_WANT_EXCL;
    625   1.1      fvdl 		/*
    626   1.1      fvdl 		 * Wait for shared locks and upgrades to finish.
    627   1.1      fvdl 		 */
    628  1.23   thorpej 		ACQUIRE(lkp, error, extflags, 0, lkp->lk_sharecount != 0 ||
    629   1.1      fvdl 		       (lkp->lk_flags & LK_WANT_UPGRADE));
    630   1.1      fvdl 		lkp->lk_flags &= ~LK_WANT_EXCL;
    631   1.1      fvdl 		if (error)
    632   1.1      fvdl 			break;
    633   1.1      fvdl 		lkp->lk_flags |= LK_HAVE_EXCL;
    634  1.19   thorpej 		SETHOLDER(lkp, pid, cpu_id);
    635  1.21   thorpej 		HAVEIT(lkp);
    636   1.1      fvdl 		if (lkp->lk_exclusivecount != 0)
    637   1.1      fvdl 			panic("lockmgr: non-zero exclusive count");
    638   1.1      fvdl 		lkp->lk_exclusivecount = 1;
    639  1.15      fvdl 		if (extflags & LK_SETRECURSE)
    640  1.15      fvdl 			lkp->lk_recurselevel = 1;
    641  1.21   thorpej 		COUNT(lkp, p, cpu_id, 1);
    642   1.1      fvdl 		break;
    643   1.1      fvdl 
    644   1.1      fvdl 	case LK_RELEASE:
    645   1.1      fvdl 		if (lkp->lk_exclusivecount != 0) {
    646  1.19   thorpej 			if (WEHOLDIT(lkp, pid, cpu_id) == 0) {
    647  1.19   thorpej 				if (lkp->lk_flags & LK_SPIN) {
    648  1.19   thorpej 					panic("lockmgr: processor %lu, not "
    649  1.19   thorpej 					    "exclusive lock holder %lu "
    650  1.19   thorpej 					    "unlocking", cpu_id, lkp->lk_cpu);
    651  1.19   thorpej 				} else {
    652  1.19   thorpej 					panic("lockmgr: pid %d, not "
    653  1.19   thorpej 					    "exclusive lock holder %d "
    654  1.19   thorpej 					    "unlocking", pid,
    655  1.19   thorpej 					    lkp->lk_lockholder);
    656  1.19   thorpej 				}
    657  1.19   thorpej 			}
    658  1.15      fvdl 			if (lkp->lk_exclusivecount == lkp->lk_recurselevel)
    659  1.15      fvdl 				lkp->lk_recurselevel = 0;
    660   1.1      fvdl 			lkp->lk_exclusivecount--;
    661  1.21   thorpej 			COUNT(lkp, p, cpu_id, -1);
    662   1.1      fvdl 			if (lkp->lk_exclusivecount == 0) {
    663   1.1      fvdl 				lkp->lk_flags &= ~LK_HAVE_EXCL;
    664  1.19   thorpej 				SETHOLDER(lkp, LK_NOPROC, LK_NOCPU);
    665  1.21   thorpej 				DONTHAVEIT(lkp);
    666   1.1      fvdl 			}
    667   1.1      fvdl 		} else if (lkp->lk_sharecount != 0) {
    668   1.1      fvdl 			lkp->lk_sharecount--;
    669  1.21   thorpej 			COUNT(lkp, p, cpu_id, -1);
    670   1.1      fvdl 		}
    671  1.39   thorpej #ifdef DIAGNOSTIC
    672  1.39   thorpej 		else
    673  1.39   thorpej 			panic("lockmgr: release of unlocked lock!");
    674  1.39   thorpej #endif
    675  1.23   thorpej 		WAKEUP_WAITER(lkp);
    676   1.1      fvdl 		break;
    677   1.1      fvdl 
    678   1.1      fvdl 	case LK_DRAIN:
    679   1.1      fvdl 		/*
    680   1.1      fvdl 		 * Check that we do not already hold the lock, as it can
    681   1.1      fvdl 		 * never drain if we do. Unfortunately, we have no way to
    682   1.1      fvdl 		 * check for holding a shared lock, but at least we can
    683   1.1      fvdl 		 * check for an exclusive one.
    684   1.1      fvdl 		 */
    685  1.19   thorpej 		if (WEHOLDIT(lkp, pid, cpu_id))
    686   1.1      fvdl 			panic("lockmgr: draining against myself");
    687   1.1      fvdl 		/*
    688   1.1      fvdl 		 * If we are just polling, check to see if we will sleep.
    689   1.1      fvdl 		 */
    690   1.1      fvdl 		if ((extflags & LK_NOWAIT) && ((lkp->lk_flags &
    691   1.1      fvdl 		     (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE)) ||
    692   1.1      fvdl 		     lkp->lk_sharecount != 0 || lkp->lk_waitcount != 0)) {
    693   1.1      fvdl 			error = EBUSY;
    694   1.1      fvdl 			break;
    695   1.1      fvdl 		}
    696  1.23   thorpej 		ACQUIRE(lkp, error, extflags, 1,
    697  1.23   thorpej 		    ((lkp->lk_flags &
    698  1.23   thorpej 		     (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE)) ||
    699  1.23   thorpej 		     lkp->lk_sharecount != 0 ||
    700  1.23   thorpej 		     lkp->lk_waitcount != 0));
    701  1.23   thorpej 		if (error)
    702  1.23   thorpej 			break;
    703   1.1      fvdl 		lkp->lk_flags |= LK_DRAINING | LK_HAVE_EXCL;
    704  1.19   thorpej 		SETHOLDER(lkp, pid, cpu_id);
    705  1.21   thorpej 		HAVEIT(lkp);
    706   1.1      fvdl 		lkp->lk_exclusivecount = 1;
    707  1.15      fvdl 		/* XXX unlikely that we'd want this */
    708  1.15      fvdl 		if (extflags & LK_SETRECURSE)
    709  1.15      fvdl 			lkp->lk_recurselevel = 1;
    710  1.21   thorpej 		COUNT(lkp, p, cpu_id, 1);
    711   1.1      fvdl 		break;
    712   1.1      fvdl 
    713   1.1      fvdl 	default:
    714  1.43   thorpej 		INTERLOCK_RELEASE(lkp, lkp->lk_flags, s);
    715   1.1      fvdl 		panic("lockmgr: unknown locktype request %d",
    716   1.1      fvdl 		    flags & LK_TYPE_MASK);
    717   1.1      fvdl 		/* NOTREACHED */
    718   1.1      fvdl 	}
    719  1.23   thorpej 	if ((lkp->lk_flags & (LK_WAITDRAIN|LK_SPIN)) == LK_WAITDRAIN &&
    720  1.23   thorpej 	    ((lkp->lk_flags &
    721  1.23   thorpej 	      (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE)) == 0 &&
    722   1.1      fvdl 	     lkp->lk_sharecount == 0 && lkp->lk_waitcount == 0)) {
    723   1.1      fvdl 		lkp->lk_flags &= ~LK_WAITDRAIN;
    724  1.20   thorpej 		wakeup_one((void *)&lkp->lk_flags);
    725   1.1      fvdl 	}
    726  1.32  sommerfe 	/*
    727  1.32  sommerfe 	 * Note that this panic will be a recursive panic, since
    728  1.32  sommerfe 	 * we only set lock_shutdown_noblock above if panicstr != NULL.
    729  1.32  sommerfe 	 */
    730  1.32  sommerfe 	if (error && lock_shutdown_noblock)
    731  1.32  sommerfe 		panic("lockmgr: deadlock (see previous panic)");
    732  1.32  sommerfe 
    733  1.43   thorpej 	INTERLOCK_RELEASE(lkp, lkp->lk_flags, s);
    734   1.1      fvdl 	return (error);
    735   1.1      fvdl }
    736   1.1      fvdl 
    737   1.1      fvdl /*
    738   1.1      fvdl  * Print out information about state of a lock. Used by VOP_PRINT
    739   1.1      fvdl  * routines to display ststus about contained locks.
    740   1.1      fvdl  */
    741   1.2      fvdl void
    742  1.33   thorpej lockmgr_printinfo(__volatile struct lock *lkp)
    743   1.1      fvdl {
    744   1.1      fvdl 
    745   1.1      fvdl 	if (lkp->lk_sharecount)
    746   1.1      fvdl 		printf(" lock type %s: SHARED (count %d)", lkp->lk_wmesg,
    747   1.1      fvdl 		    lkp->lk_sharecount);
    748  1.19   thorpej 	else if (lkp->lk_flags & LK_HAVE_EXCL) {
    749  1.19   thorpej 		printf(" lock type %s: EXCL (count %d) by ",
    750  1.19   thorpej 		    lkp->lk_wmesg, lkp->lk_exclusivecount);
    751  1.19   thorpej 		if (lkp->lk_flags & LK_SPIN)
    752  1.19   thorpej 			printf("processor %lu", lkp->lk_cpu);
    753  1.19   thorpej 		else
    754  1.19   thorpej 			printf("pid %d", lkp->lk_lockholder);
    755  1.19   thorpej 	} else
    756  1.19   thorpej 		printf(" not locked");
    757  1.19   thorpej 	if ((lkp->lk_flags & LK_SPIN) == 0 && lkp->lk_waitcount > 0)
    758   1.1      fvdl 		printf(" with %d pending", lkp->lk_waitcount);
    759   1.1      fvdl }
    760   1.1      fvdl 
    761  1.21   thorpej #if defined(LOCKDEBUG) /* { */
    762  1.21   thorpej TAILQ_HEAD(, simplelock) simplelock_list =
    763  1.21   thorpej     TAILQ_HEAD_INITIALIZER(simplelock_list);
    764  1.21   thorpej 
    765  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    766  1.21   thorpej struct simplelock simplelock_list_slock = SIMPLELOCK_INITIALIZER;
    767  1.21   thorpej 
    768  1.21   thorpej #define	SLOCK_LIST_LOCK()						\
    769  1.29  sommerfe 	__cpu_simple_lock(&simplelock_list_slock.lock_data)
    770  1.21   thorpej 
    771  1.21   thorpej #define	SLOCK_LIST_UNLOCK()						\
    772  1.29  sommerfe 	__cpu_simple_unlock(&simplelock_list_slock.lock_data)
    773  1.21   thorpej 
    774  1.28   thorpej #if defined(__HAVE_ATOMIC_OPERATIONS) /* { */
    775  1.21   thorpej #define	SLOCK_COUNT(x)							\
    776  1.28   thorpej 	atomic_add_ulong(&curcpu()->ci_simple_locks, (x))
    777  1.28   thorpej #else
    778  1.28   thorpej #define	SLOCK_COUNT(x)		/* not safe */
    779  1.28   thorpej #endif /* __HAVE_ATOMIC_OPERATIONS */ /* } */
    780  1.21   thorpej #else
    781  1.21   thorpej u_long simple_locks;
    782  1.21   thorpej 
    783  1.21   thorpej #define	SLOCK_LIST_LOCK()	/* nothing */
    784  1.21   thorpej 
    785  1.21   thorpej #define	SLOCK_LIST_UNLOCK()	/* nothing */
    786  1.21   thorpej 
    787  1.21   thorpej #define	SLOCK_COUNT(x)		simple_locks += (x)
    788  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
    789  1.21   thorpej 
    790  1.21   thorpej #ifdef DDB /* { */
    791  1.18       chs int simple_lock_debugger = 0;
    792  1.21   thorpej #define	SLOCK_DEBUGGER()	if (simple_lock_debugger) Debugger()
    793  1.21   thorpej #else
    794  1.21   thorpej #define	SLOCK_DEBUGGER()	/* nothing */
    795  1.21   thorpej #endif /* } */
    796  1.21   thorpej 
    797  1.26  sommerfe #ifdef MULTIPROCESSOR
    798  1.26  sommerfe #define SLOCK_MP()		lock_printf("on cpu %d\n", cpu_number())
    799  1.26  sommerfe #else
    800  1.26  sommerfe #define SLOCK_MP()		/* nothing */
    801  1.26  sommerfe #endif
    802  1.26  sommerfe 
    803  1.21   thorpej #define	SLOCK_WHERE(str, alp, id, l)					\
    804  1.21   thorpej do {									\
    805  1.25   thorpej 	lock_printf(str);						\
    806  1.33   thorpej 	lock_printf("lock: %p, currently at: %s:%d\n", (alp), (id), (l)); \
    807  1.26  sommerfe 	SLOCK_MP();							\
    808  1.21   thorpej 	if ((alp)->lock_file != NULL)					\
    809  1.25   thorpej 		lock_printf("last locked: %s:%d\n", (alp)->lock_file,	\
    810  1.21   thorpej 		    (alp)->lock_line);					\
    811  1.21   thorpej 	if ((alp)->unlock_file != NULL)					\
    812  1.25   thorpej 		lock_printf("last unlocked: %s:%d\n", (alp)->unlock_file, \
    813  1.21   thorpej 		    (alp)->unlock_line);				\
    814  1.21   thorpej 	SLOCK_DEBUGGER();						\
    815  1.30   thorpej } while (/*CONSTCOND*/0)
    816  1.12       chs 
    817   1.1      fvdl /*
    818   1.1      fvdl  * Simple lock functions so that the debugger can see from whence
    819   1.1      fvdl  * they are being called.
    820   1.1      fvdl  */
    821   1.1      fvdl void
    822  1.33   thorpej simple_lock_init(struct simplelock *alp)
    823   1.1      fvdl {
    824  1.21   thorpej 
    825  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    826  1.27   thorpej 	__cpu_simple_lock_init(&alp->lock_data);
    827  1.21   thorpej #else
    828  1.27   thorpej 	alp->lock_data = __SIMPLELOCK_UNLOCKED;
    829  1.21   thorpej #endif /* } */
    830   1.5       chs 	alp->lock_file = NULL;
    831   1.5       chs 	alp->lock_line = 0;
    832   1.5       chs 	alp->unlock_file = NULL;
    833   1.5       chs 	alp->unlock_line = 0;
    834  1.41   thorpej 	alp->lock_holder = LK_NOCPU;
    835   1.1      fvdl }
    836   1.1      fvdl 
    837   1.1      fvdl void
    838  1.33   thorpej _simple_lock(__volatile struct simplelock *alp, const char *id, int l)
    839   1.1      fvdl {
    840  1.24   thorpej 	cpuid_t cpu_id = cpu_number();
    841  1.12       chs 	int s;
    842  1.12       chs 
    843  1.44   thorpej 	s = spllock();
    844  1.21   thorpej 
    845  1.21   thorpej 	/*
    846  1.21   thorpej 	 * MULTIPROCESSOR case: This is `safe' since if it's not us, we
    847  1.21   thorpej 	 * don't take any action, and just fall into the normal spin case.
    848  1.21   thorpej 	 */
    849  1.27   thorpej 	if (alp->lock_data == __SIMPLELOCK_LOCKED) {
    850  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    851  1.21   thorpej 		if (alp->lock_holder == cpu_id) {
    852  1.21   thorpej 			SLOCK_WHERE("simple_lock: locking against myself\n",
    853  1.21   thorpej 			    alp, id, l);
    854  1.21   thorpej 			goto out;
    855   1.1      fvdl 		}
    856  1.21   thorpej #else
    857  1.21   thorpej 		SLOCK_WHERE("simple_lock: lock held\n", alp, id, l);
    858  1.21   thorpej 		goto out;
    859  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
    860   1.1      fvdl 	}
    861  1.21   thorpej 
    862  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    863  1.21   thorpej 	/* Acquire the lock before modifying any fields. */
    864  1.27   thorpej 	__cpu_simple_lock(&alp->lock_data);
    865  1.21   thorpej #else
    866  1.27   thorpej 	alp->lock_data = __SIMPLELOCK_LOCKED;
    867  1.21   thorpej #endif /* } */
    868  1.21   thorpej 
    869  1.41   thorpej 	KASSERT(alp->lock_holder == LK_NOCPU);
    870  1.41   thorpej 
    871   1.5       chs 	alp->lock_file = id;
    872   1.5       chs 	alp->lock_line = l;
    873  1.21   thorpej 	alp->lock_holder = cpu_id;
    874  1.21   thorpej 
    875  1.21   thorpej 	SLOCK_LIST_LOCK();
    876  1.21   thorpej 	/* XXX Cast away volatile */
    877  1.21   thorpej 	TAILQ_INSERT_TAIL(&simplelock_list, (struct simplelock *)alp, list);
    878  1.21   thorpej 	SLOCK_LIST_UNLOCK();
    879  1.21   thorpej 
    880  1.21   thorpej 	SLOCK_COUNT(1);
    881  1.21   thorpej 
    882  1.21   thorpej  out:
    883  1.18       chs 	splx(s);
    884  1.38   thorpej }
    885  1.38   thorpej 
    886  1.38   thorpej int
    887  1.38   thorpej _simple_lock_held(__volatile struct simplelock *alp)
    888  1.38   thorpej {
    889  1.38   thorpej 	cpuid_t cpu_id = cpu_number();
    890  1.38   thorpej 	int s, locked = 0;
    891  1.38   thorpej 
    892  1.44   thorpej 	s = spllock();
    893  1.42   thorpej 
    894  1.42   thorpej #if defined(MULTIPROCESSOR)
    895  1.38   thorpej 	if (__cpu_simple_lock_try(&alp->lock_data) == 0)
    896  1.38   thorpej 		locked = (alp->lock_holder == cpu_id);
    897  1.38   thorpej 	else
    898  1.38   thorpej 		__cpu_simple_unlock(&alp->lock_data);
    899  1.38   thorpej #else
    900  1.42   thorpej 	if (alp->lock_data == __SIMPLELOCK_LOCKED) {
    901  1.42   thorpej 		locked = 1;
    902  1.42   thorpej 		KASSERT(alp->lock_holder == cpu_id);
    903  1.42   thorpej 	}
    904  1.42   thorpej #endif
    905  1.38   thorpej 
    906  1.38   thorpej 	splx(s);
    907  1.42   thorpej 
    908  1.38   thorpej 	return (locked);
    909   1.1      fvdl }
    910   1.1      fvdl 
    911   1.1      fvdl int
    912  1.33   thorpej _simple_lock_try(__volatile struct simplelock *alp, const char *id, int l)
    913   1.1      fvdl {
    914  1.24   thorpej 	cpuid_t cpu_id = cpu_number();
    915  1.21   thorpej 	int s, rv = 0;
    916   1.1      fvdl 
    917  1.44   thorpej 	s = spllock();
    918  1.21   thorpej 
    919  1.21   thorpej 	/*
    920  1.21   thorpej 	 * MULTIPROCESSOR case: This is `safe' since if it's not us, we
    921  1.21   thorpej 	 * don't take any action.
    922  1.21   thorpej 	 */
    923  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    924  1.27   thorpej 	if ((rv = __cpu_simple_lock_try(&alp->lock_data)) == 0) {
    925  1.21   thorpej 		if (alp->lock_holder == cpu_id)
    926  1.21   thorpej 			SLOCK_WHERE("simple_lock_try: locking against myself\n",
    927  1.26  sommerfe 			    alp, id, l);
    928  1.21   thorpej 		goto out;
    929  1.21   thorpej 	}
    930  1.21   thorpej #else
    931  1.27   thorpej 	if (alp->lock_data == __SIMPLELOCK_LOCKED) {
    932  1.21   thorpej 		SLOCK_WHERE("simple_lock_try: lock held\n", alp, id, l);
    933  1.21   thorpej 		goto out;
    934  1.18       chs 	}
    935  1.27   thorpej 	alp->lock_data = __SIMPLELOCK_LOCKED;
    936  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
    937  1.21   thorpej 
    938  1.21   thorpej 	/*
    939  1.21   thorpej 	 * At this point, we have acquired the lock.
    940  1.21   thorpej 	 */
    941  1.21   thorpej 
    942  1.21   thorpej 	rv = 1;
    943  1.18       chs 
    944   1.5       chs 	alp->lock_file = id;
    945   1.5       chs 	alp->lock_line = l;
    946  1.21   thorpej 	alp->lock_holder = cpu_id;
    947  1.21   thorpej 
    948  1.21   thorpej 	SLOCK_LIST_LOCK();
    949  1.21   thorpej 	/* XXX Cast away volatile. */
    950  1.21   thorpej 	TAILQ_INSERT_TAIL(&simplelock_list, (struct simplelock *)alp, list);
    951  1.21   thorpej 	SLOCK_LIST_UNLOCK();
    952  1.21   thorpej 
    953  1.21   thorpej 	SLOCK_COUNT(1);
    954  1.21   thorpej 
    955  1.21   thorpej  out:
    956  1.12       chs 	splx(s);
    957  1.21   thorpej 	return (rv);
    958   1.1      fvdl }
    959   1.1      fvdl 
    960   1.1      fvdl void
    961  1.33   thorpej _simple_unlock(__volatile struct simplelock *alp, const char *id, int l)
    962   1.1      fvdl {
    963  1.12       chs 	int s;
    964   1.1      fvdl 
    965  1.44   thorpej 	s = spllock();
    966  1.21   thorpej 
    967  1.21   thorpej 	/*
    968  1.21   thorpej 	 * MULTIPROCESSOR case: This is `safe' because we think we hold
    969  1.21   thorpej 	 * the lock, and if we don't, we don't take any action.
    970  1.21   thorpej 	 */
    971  1.27   thorpej 	if (alp->lock_data == __SIMPLELOCK_UNLOCKED) {
    972  1.21   thorpej 		SLOCK_WHERE("simple_unlock: lock not held\n",
    973  1.21   thorpej 		    alp, id, l);
    974  1.21   thorpej 		goto out;
    975  1.21   thorpej 	}
    976  1.21   thorpej 
    977  1.21   thorpej 	SLOCK_LIST_LOCK();
    978  1.21   thorpej 	TAILQ_REMOVE(&simplelock_list, alp, list);
    979  1.21   thorpej 	SLOCK_LIST_UNLOCK();
    980  1.21   thorpej 
    981  1.21   thorpej 	SLOCK_COUNT(-1);
    982  1.21   thorpej 
    983  1.21   thorpej 	alp->list.tqe_next = NULL;	/* sanity */
    984  1.21   thorpej 	alp->list.tqe_prev = NULL;	/* sanity */
    985  1.21   thorpej 
    986   1.5       chs 	alp->unlock_file = id;
    987   1.5       chs 	alp->unlock_line = l;
    988  1.21   thorpej 
    989  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    990  1.26  sommerfe 	alp->lock_holder = LK_NOCPU;
    991  1.21   thorpej 	/* Now that we've modified all fields, release the lock. */
    992  1.27   thorpej 	__cpu_simple_unlock(&alp->lock_data);
    993  1.21   thorpej #else
    994  1.27   thorpej 	alp->lock_data = __SIMPLELOCK_UNLOCKED;
    995  1.41   thorpej 	KASSERT(alp->lock_holder == cpu_number());
    996  1.41   thorpej 	alp->lock_holder = LK_NOCPU;
    997  1.21   thorpej #endif /* } */
    998  1.21   thorpej 
    999  1.21   thorpej  out:
   1000  1.18       chs 	splx(s);
   1001  1.12       chs }
   1002  1.12       chs 
   1003  1.12       chs void
   1004  1.33   thorpej simple_lock_dump(void)
   1005  1.12       chs {
   1006  1.12       chs 	struct simplelock *alp;
   1007  1.12       chs 	int s;
   1008  1.12       chs 
   1009  1.44   thorpej 	s = spllock();
   1010  1.21   thorpej 	SLOCK_LIST_LOCK();
   1011  1.25   thorpej 	lock_printf("all simple locks:\n");
   1012  1.21   thorpej 	for (alp = TAILQ_FIRST(&simplelock_list); alp != NULL;
   1013  1.21   thorpej 	     alp = TAILQ_NEXT(alp, list)) {
   1014  1.25   thorpej 		lock_printf("%p CPU %lu %s:%d\n", alp, alp->lock_holder,
   1015  1.21   thorpej 		    alp->lock_file, alp->lock_line);
   1016  1.12       chs 	}
   1017  1.21   thorpej 	SLOCK_LIST_UNLOCK();
   1018  1.12       chs 	splx(s);
   1019  1.12       chs }
   1020  1.12       chs 
   1021  1.12       chs void
   1022  1.33   thorpej simple_lock_freecheck(void *start, void *end)
   1023  1.12       chs {
   1024  1.12       chs 	struct simplelock *alp;
   1025  1.12       chs 	int s;
   1026  1.12       chs 
   1027  1.44   thorpej 	s = spllock();
   1028  1.21   thorpej 	SLOCK_LIST_LOCK();
   1029  1.21   thorpej 	for (alp = TAILQ_FIRST(&simplelock_list); alp != NULL;
   1030  1.21   thorpej 	     alp = TAILQ_NEXT(alp, list)) {
   1031  1.12       chs 		if ((void *)alp >= start && (void *)alp < end) {
   1032  1.25   thorpej 			lock_printf("freeing simple_lock %p CPU %lu %s:%d\n",
   1033  1.34   thorpej 			    alp, alp->lock_holder, alp->lock_file,
   1034  1.34   thorpej 			    alp->lock_line);
   1035  1.34   thorpej 			SLOCK_DEBUGGER();
   1036  1.34   thorpej 		}
   1037  1.34   thorpej 	}
   1038  1.34   thorpej 	SLOCK_LIST_UNLOCK();
   1039  1.34   thorpej 	splx(s);
   1040  1.34   thorpej }
   1041  1.34   thorpej 
   1042  1.34   thorpej void
   1043  1.34   thorpej simple_lock_switchcheck(void)
   1044  1.34   thorpej {
   1045  1.34   thorpej 	struct simplelock *alp;
   1046  1.34   thorpej 	cpuid_t cpu_id = cpu_number();
   1047  1.34   thorpej 	int s;
   1048  1.34   thorpej 
   1049  1.42   thorpej 	/*
   1050  1.42   thorpej 	 * We must be holding exactly one lock: the sched_lock.
   1051  1.42   thorpej 	 */
   1052  1.42   thorpej 
   1053  1.42   thorpej 	SCHED_ASSERT_LOCKED();
   1054  1.42   thorpej 
   1055  1.44   thorpej 	s = spllock();
   1056  1.34   thorpej 	SLOCK_LIST_LOCK();
   1057  1.34   thorpej 	for (alp = TAILQ_FIRST(&simplelock_list); alp != NULL;
   1058  1.34   thorpej 	     alp = TAILQ_NEXT(alp, list)) {
   1059  1.42   thorpej 		if (alp == &sched_lock)
   1060  1.42   thorpej 			continue;
   1061  1.34   thorpej 		if (alp->lock_holder == cpu_id) {
   1062  1.34   thorpej 			lock_printf("switching with held simple_lock %p "
   1063  1.36   thorpej 			    "CPU %lu %s:%d\n",
   1064  1.21   thorpej 			    alp, alp->lock_holder, alp->lock_file,
   1065  1.21   thorpej 			    alp->lock_line);
   1066  1.21   thorpej 			SLOCK_DEBUGGER();
   1067  1.12       chs 		}
   1068  1.12       chs 	}
   1069  1.21   thorpej 	SLOCK_LIST_UNLOCK();
   1070  1.12       chs 	splx(s);
   1071   1.1      fvdl }
   1072  1.21   thorpej #endif /* LOCKDEBUG */ /* } */
   1073