Home | History | Annotate | Line # | Download | only in kern
kern_lock.c revision 1.29
      1  1.29  sommerfe /*	$NetBSD: kern_lock.c,v 1.29 2000/05/03 13:53:59 sommerfeld 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.25   thorpej void	lock_printf __P((const char *fmt, ...));
    106  1.25   thorpej 
    107  1.25   thorpej int	lock_debug_syslog = 0;	/* defaults to printf, but can be patched */
    108  1.25   thorpej #endif
    109  1.25   thorpej 
    110   1.1      fvdl /*
    111   1.1      fvdl  * Locking primitives implementation.
    112   1.1      fvdl  * Locks provide shared/exclusive sychronization.
    113   1.1      fvdl  */
    114   1.1      fvdl 
    115  1.21   thorpej #if defined(LOCKDEBUG) || defined(DIAGNOSTIC) /* { */
    116  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    117  1.28   thorpej #if defined(__HAVE_ATOMIC_OPERATIONS) /* { */
    118  1.21   thorpej #define	COUNT_CPU(cpu_id, x)						\
    119  1.28   thorpej 	atomic_add_ulong(&curcpu().ci_spin_locks, (x))
    120  1.28   thorpej #else
    121  1.28   thorpej #define	COUNT_CPU(cpu_id, x)	/* not safe */
    122  1.28   thorpej #endif /* __HAVE_ATOMIC_OPERATIONS */ /* } */
    123  1.21   thorpej #else
    124  1.21   thorpej u_long	spin_locks;
    125  1.21   thorpej #define	COUNT_CPU(cpu_id, x)	spin_locks += (x)
    126  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
    127  1.21   thorpej 
    128  1.21   thorpej #define	COUNT(lkp, p, cpu_id, x)					\
    129  1.21   thorpej do {									\
    130  1.21   thorpej 	if ((lkp)->lk_flags & LK_SPIN)					\
    131  1.21   thorpej 		COUNT_CPU((cpu_id), (x));				\
    132  1.21   thorpej 	else								\
    133  1.21   thorpej 		(p)->p_locks += (x);					\
    134  1.21   thorpej } while (0)
    135   1.1      fvdl #else
    136  1.22    mellon #define COUNT(lkp, p, cpu_id, x)
    137  1.21   thorpej #endif /* LOCKDEBUG || DIAGNOSTIC */ /* } */
    138   1.1      fvdl 
    139   1.1      fvdl /*
    140   1.1      fvdl  * Acquire a resource.
    141   1.1      fvdl  */
    142  1.23   thorpej #define ACQUIRE(lkp, error, extflags, drain, wanted)			\
    143  1.19   thorpej 	if ((extflags) & LK_SPIN) {					\
    144  1.19   thorpej 		int interlocked;					\
    145  1.19   thorpej 									\
    146  1.23   thorpej 		if ((drain) == 0)					\
    147  1.23   thorpej 			(lkp)->lk_waitcount++;				\
    148  1.19   thorpej 		for (interlocked = 1;;) {				\
    149  1.19   thorpej 			if (wanted) {					\
    150  1.19   thorpej 				if (interlocked) {			\
    151  1.19   thorpej 					simple_unlock(&(lkp)->lk_interlock); \
    152  1.19   thorpej 					interlocked = 0;		\
    153  1.19   thorpej 				}					\
    154  1.19   thorpej 			} else if (interlocked) {			\
    155  1.19   thorpej 				break;					\
    156  1.19   thorpej 			} else {					\
    157  1.19   thorpej 				simple_lock(&(lkp)->lk_interlock);	\
    158  1.19   thorpej 				interlocked = 1;			\
    159  1.19   thorpej 			}						\
    160  1.19   thorpej 		}							\
    161  1.23   thorpej 		if ((drain) == 0)					\
    162  1.23   thorpej 			(lkp)->lk_waitcount--;				\
    163  1.19   thorpej 		KASSERT((wanted) == 0);					\
    164  1.19   thorpej 		error = 0;	/* sanity */				\
    165  1.19   thorpej 	} else {							\
    166  1.19   thorpej 		for (error = 0; wanted; ) {				\
    167  1.23   thorpej 			if ((drain))					\
    168  1.23   thorpej 				(lkp)->lk_flags |= LK_WAITDRAIN;	\
    169  1.23   thorpej 			else						\
    170  1.23   thorpej 				(lkp)->lk_waitcount++;			\
    171  1.19   thorpej 			simple_unlock(&(lkp)->lk_interlock);		\
    172  1.23   thorpej 			/* XXX Cast away volatile. */			\
    173  1.23   thorpej 			error = tsleep((drain) ? &(lkp)->lk_flags :	\
    174  1.23   thorpej 			    (void *)(lkp), (lkp)->lk_prio,		\
    175  1.19   thorpej 			    (lkp)->lk_wmesg, (lkp)->lk_timo);		\
    176  1.19   thorpej 			simple_lock(&(lkp)->lk_interlock);		\
    177  1.23   thorpej 			if ((drain) == 0)				\
    178  1.23   thorpej 				(lkp)->lk_waitcount--;			\
    179  1.19   thorpej 			if (error)					\
    180  1.19   thorpej 				break;					\
    181  1.19   thorpej 			if ((extflags) & LK_SLEEPFAIL) {		\
    182  1.19   thorpej 				error = ENOLCK;				\
    183  1.19   thorpej 				break;					\
    184  1.19   thorpej 			}						\
    185   1.1      fvdl 		}							\
    186   1.1      fvdl 	}
    187   1.1      fvdl 
    188  1.19   thorpej #define	SETHOLDER(lkp, pid, cpu_id)					\
    189  1.19   thorpej do {									\
    190  1.19   thorpej 	if ((lkp)->lk_flags & LK_SPIN)					\
    191  1.19   thorpej 		(lkp)->lk_cpu = cpu_id;					\
    192  1.19   thorpej 	else								\
    193  1.19   thorpej 		(lkp)->lk_lockholder = pid;				\
    194  1.19   thorpej } while (0)
    195  1.19   thorpej 
    196  1.19   thorpej #define	WEHOLDIT(lkp, pid, cpu_id)					\
    197  1.19   thorpej 	(((lkp)->lk_flags & LK_SPIN) != 0 ?				\
    198  1.19   thorpej 	 ((lkp)->lk_cpu == (cpu_id)) : ((lkp)->lk_lockholder == (pid)))
    199  1.19   thorpej 
    200  1.23   thorpej #define	WAKEUP_WAITER(lkp)						\
    201  1.23   thorpej do {									\
    202  1.23   thorpej 	if (((lkp)->lk_flags & LK_SPIN) == 0 && (lkp)->lk_waitcount) {	\
    203  1.23   thorpej 		/* XXX Cast away volatile. */				\
    204  1.23   thorpej 		wakeup_one((void *)(lkp));				\
    205  1.23   thorpej 	}								\
    206  1.23   thorpej } while (0)
    207  1.23   thorpej 
    208  1.21   thorpej #if defined(LOCKDEBUG) /* { */
    209  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    210  1.21   thorpej struct simplelock spinlock_list_slock = SIMPLELOCK_INITIALIZER;
    211  1.21   thorpej 
    212  1.27   thorpej #define	SPINLOCK_LIST_LOCK()						\
    213  1.29  sommerfe 	__cpu_simple_lock(&spinlock_list_slock.lock_data)
    214  1.21   thorpej 
    215  1.27   thorpej #define	SPINLOCK_LIST_UNLOCK()						\
    216  1.29  sommerfe 	__cpu_simple_unlock(&spinlock_list_slock.lock_data)
    217  1.21   thorpej #else
    218  1.21   thorpej #define	SPINLOCK_LIST_LOCK()	/* nothing */
    219  1.21   thorpej 
    220  1.21   thorpej #define	SPINLOCK_LIST_UNLOCK()	/* nothing */
    221  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
    222  1.21   thorpej 
    223  1.21   thorpej TAILQ_HEAD(, lock) spinlock_list =
    224  1.21   thorpej     TAILQ_HEAD_INITIALIZER(spinlock_list);
    225  1.21   thorpej 
    226  1.21   thorpej #define	HAVEIT(lkp)							\
    227  1.21   thorpej do {									\
    228  1.21   thorpej 	if ((lkp)->lk_flags & LK_SPIN) {				\
    229  1.21   thorpej 		int s = splhigh();					\
    230  1.21   thorpej 		SPINLOCK_LIST_LOCK();					\
    231  1.21   thorpej 		/* XXX Cast away volatile. */				\
    232  1.21   thorpej 		TAILQ_INSERT_TAIL(&spinlock_list, (struct lock *)(lkp),	\
    233  1.21   thorpej 		    lk_list);						\
    234  1.21   thorpej 		SPINLOCK_LIST_UNLOCK();					\
    235  1.21   thorpej 		splx(s);						\
    236  1.21   thorpej 	}								\
    237  1.21   thorpej } while (0)
    238  1.21   thorpej 
    239  1.21   thorpej #define	DONTHAVEIT(lkp)							\
    240  1.21   thorpej do {									\
    241  1.21   thorpej 	if ((lkp)->lk_flags & LK_SPIN) {				\
    242  1.21   thorpej 		int s = splhigh();					\
    243  1.21   thorpej 		SPINLOCK_LIST_LOCK();					\
    244  1.21   thorpej 		/* XXX Cast away volatile. */				\
    245  1.21   thorpej 		TAILQ_REMOVE(&spinlock_list, (struct lock *)(lkp),	\
    246  1.21   thorpej 		    lk_list);						\
    247  1.21   thorpej 		SPINLOCK_LIST_UNLOCK();					\
    248  1.21   thorpej 		splx(s);						\
    249  1.21   thorpej 	}								\
    250  1.21   thorpej } while (0)
    251  1.21   thorpej #else
    252  1.21   thorpej #define	HAVEIT(lkp)		/* nothing */
    253  1.21   thorpej 
    254  1.21   thorpej #define	DONTHAVEIT(lkp)		/* nothing */
    255  1.21   thorpej #endif /* LOCKDEBUG */ /* } */
    256  1.21   thorpej 
    257  1.25   thorpej #if defined(LOCKDEBUG)
    258  1.25   thorpej /*
    259  1.25   thorpej  * Lock debug printing routine; can be configured to print to console
    260  1.25   thorpej  * or log to syslog.
    261  1.25   thorpej  */
    262  1.25   thorpej void
    263  1.25   thorpej #ifdef __STDC__
    264  1.25   thorpej lock_printf(const char *fmt, ...)
    265  1.25   thorpej #else
    266  1.25   thorpej lock_printf(fmt, va_alist)
    267  1.25   thorpej 	char *fmt;
    268  1.25   thorpej 	va_dcl
    269  1.25   thorpej #endif
    270  1.25   thorpej {
    271  1.25   thorpej 	va_list ap;
    272  1.25   thorpej 
    273  1.25   thorpej 	va_start(ap, fmt);
    274  1.25   thorpej 	if (lock_debug_syslog)
    275  1.25   thorpej 		vlog(LOG_DEBUG, fmt, ap);
    276  1.25   thorpej 	else
    277  1.25   thorpej 		vprintf(fmt, ap);
    278  1.25   thorpej 	va_end(ap);
    279  1.25   thorpej }
    280  1.25   thorpej #endif /* LOCKDEBUG */
    281  1.25   thorpej 
    282   1.1      fvdl /*
    283   1.1      fvdl  * Initialize a lock; required before use.
    284   1.1      fvdl  */
    285   1.1      fvdl void
    286   1.1      fvdl lockinit(lkp, prio, wmesg, timo, flags)
    287   1.1      fvdl 	struct lock *lkp;
    288   1.1      fvdl 	int prio;
    289   1.4   mycroft 	const char *wmesg;
    290   1.1      fvdl 	int timo;
    291   1.1      fvdl 	int flags;
    292   1.1      fvdl {
    293   1.1      fvdl 
    294   1.8     perry 	memset(lkp, 0, sizeof(struct lock));
    295   1.1      fvdl 	simple_lock_init(&lkp->lk_interlock);
    296   1.1      fvdl 	lkp->lk_flags = flags & LK_EXTFLG_MASK;
    297  1.19   thorpej 	if (flags & LK_SPIN)
    298  1.19   thorpej 		lkp->lk_cpu = LK_NOCPU;
    299  1.19   thorpej 	else {
    300  1.19   thorpej 		lkp->lk_lockholder = LK_NOPROC;
    301  1.19   thorpej 		lkp->lk_prio = prio;
    302  1.19   thorpej 		lkp->lk_timo = timo;
    303  1.19   thorpej 	}
    304  1.19   thorpej 	lkp->lk_wmesg = wmesg;	/* just a name for spin locks */
    305   1.1      fvdl }
    306   1.1      fvdl 
    307   1.1      fvdl /*
    308   1.1      fvdl  * Determine the status of a lock.
    309   1.1      fvdl  */
    310   1.1      fvdl int
    311   1.1      fvdl lockstatus(lkp)
    312   1.1      fvdl 	struct lock *lkp;
    313   1.1      fvdl {
    314   1.1      fvdl 	int lock_type = 0;
    315   1.1      fvdl 
    316   1.1      fvdl 	simple_lock(&lkp->lk_interlock);
    317   1.1      fvdl 	if (lkp->lk_exclusivecount != 0)
    318   1.1      fvdl 		lock_type = LK_EXCLUSIVE;
    319   1.1      fvdl 	else if (lkp->lk_sharecount != 0)
    320   1.1      fvdl 		lock_type = LK_SHARED;
    321   1.1      fvdl 	simple_unlock(&lkp->lk_interlock);
    322   1.1      fvdl 	return (lock_type);
    323   1.1      fvdl }
    324   1.1      fvdl 
    325   1.1      fvdl /*
    326   1.1      fvdl  * Set, change, or release a lock.
    327   1.1      fvdl  *
    328   1.1      fvdl  * Shared requests increment the shared count. Exclusive requests set the
    329   1.1      fvdl  * LK_WANT_EXCL flag (preventing further shared locks), and wait for already
    330   1.1      fvdl  * accepted shared locks and shared-to-exclusive upgrades to go away.
    331   1.1      fvdl  */
    332   1.1      fvdl int
    333   1.6      fvdl lockmgr(lkp, flags, interlkp)
    334   1.1      fvdl 	__volatile struct lock *lkp;
    335   1.1      fvdl 	u_int flags;
    336   1.1      fvdl 	struct simplelock *interlkp;
    337   1.1      fvdl {
    338   1.1      fvdl 	int error;
    339   1.1      fvdl 	pid_t pid;
    340   1.1      fvdl 	int extflags;
    341  1.24   thorpej 	cpuid_t cpu_id;
    342   1.6      fvdl 	struct proc *p = curproc;
    343   1.1      fvdl 
    344   1.1      fvdl 	error = 0;
    345  1.19   thorpej 
    346   1.1      fvdl 	simple_lock(&lkp->lk_interlock);
    347   1.1      fvdl 	if (flags & LK_INTERLOCK)
    348   1.1      fvdl 		simple_unlock(interlkp);
    349   1.1      fvdl 	extflags = (flags | lkp->lk_flags) & LK_EXTFLG_MASK;
    350  1.19   thorpej 
    351  1.21   thorpej #ifdef DIAGNOSTIC /* { */
    352  1.19   thorpej 	/*
    353  1.19   thorpej 	 * Don't allow spins on sleep locks and don't allow sleeps
    354  1.19   thorpej 	 * on spin locks.
    355  1.19   thorpej 	 */
    356  1.19   thorpej 	if ((flags ^ lkp->lk_flags) & LK_SPIN)
    357  1.19   thorpej 		panic("lockmgr: sleep/spin mismatch\n");
    358  1.21   thorpej #endif /* } */
    359  1.19   thorpej 
    360  1.19   thorpej 	if (extflags & LK_SPIN)
    361  1.19   thorpej 		pid = LK_KERNPROC;
    362  1.19   thorpej 	else {
    363  1.21   thorpej #ifdef DIAGNOSTIC /* { */
    364  1.19   thorpej 		if (p == NULL)
    365  1.19   thorpej 			panic("lockmgr: no context");
    366  1.21   thorpej #endif /* } */
    367  1.19   thorpej 		pid = p->p_pid;
    368  1.19   thorpej 	}
    369  1.24   thorpej 	cpu_id = cpu_number();
    370  1.19   thorpej 
    371   1.1      fvdl 	/*
    372   1.1      fvdl 	 * Once a lock has drained, the LK_DRAINING flag is set and an
    373   1.1      fvdl 	 * exclusive lock is returned. The only valid operation thereafter
    374   1.1      fvdl 	 * is a single release of that exclusive lock. This final release
    375   1.1      fvdl 	 * clears the LK_DRAINING flag and sets the LK_DRAINED flag. Any
    376   1.1      fvdl 	 * further requests of any sort will result in a panic. The bits
    377   1.1      fvdl 	 * selected for these two flags are chosen so that they will be set
    378   1.1      fvdl 	 * in memory that is freed (freed memory is filled with 0xdeadbeef).
    379   1.1      fvdl 	 * The final release is permitted to give a new lease on life to
    380   1.1      fvdl 	 * the lock by specifying LK_REENABLE.
    381   1.1      fvdl 	 */
    382   1.1      fvdl 	if (lkp->lk_flags & (LK_DRAINING|LK_DRAINED)) {
    383  1.28   thorpej #ifdef DIAGNOSTIC /* { */
    384   1.1      fvdl 		if (lkp->lk_flags & LK_DRAINED)
    385   1.1      fvdl 			panic("lockmgr: using decommissioned lock");
    386   1.1      fvdl 		if ((flags & LK_TYPE_MASK) != LK_RELEASE ||
    387  1.19   thorpej 		    WEHOLDIT(lkp, pid, cpu_id) == 0)
    388   1.1      fvdl 			panic("lockmgr: non-release on draining lock: %d\n",
    389   1.1      fvdl 			    flags & LK_TYPE_MASK);
    390  1.28   thorpej #endif /* DIAGNOSTIC */ /* } */
    391   1.1      fvdl 		lkp->lk_flags &= ~LK_DRAINING;
    392   1.1      fvdl 		if ((flags & LK_REENABLE) == 0)
    393   1.1      fvdl 			lkp->lk_flags |= LK_DRAINED;
    394   1.1      fvdl 	}
    395   1.1      fvdl 
    396   1.1      fvdl 	switch (flags & LK_TYPE_MASK) {
    397   1.1      fvdl 
    398   1.1      fvdl 	case LK_SHARED:
    399  1.19   thorpej 		if (WEHOLDIT(lkp, pid, cpu_id) == 0) {
    400   1.1      fvdl 			/*
    401   1.1      fvdl 			 * If just polling, check to see if we will block.
    402   1.1      fvdl 			 */
    403   1.1      fvdl 			if ((extflags & LK_NOWAIT) && (lkp->lk_flags &
    404   1.1      fvdl 			    (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE))) {
    405   1.1      fvdl 				error = EBUSY;
    406   1.1      fvdl 				break;
    407   1.1      fvdl 			}
    408   1.1      fvdl 			/*
    409   1.1      fvdl 			 * Wait for exclusive locks and upgrades to clear.
    410   1.1      fvdl 			 */
    411  1.23   thorpej 			ACQUIRE(lkp, error, extflags, 0, lkp->lk_flags &
    412   1.1      fvdl 			    (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE));
    413   1.1      fvdl 			if (error)
    414   1.1      fvdl 				break;
    415   1.1      fvdl 			lkp->lk_sharecount++;
    416  1.21   thorpej 			COUNT(lkp, p, cpu_id, 1);
    417   1.1      fvdl 			break;
    418   1.1      fvdl 		}
    419   1.1      fvdl 		/*
    420   1.1      fvdl 		 * We hold an exclusive lock, so downgrade it to shared.
    421   1.1      fvdl 		 * An alternative would be to fail with EDEADLK.
    422   1.1      fvdl 		 */
    423   1.1      fvdl 		lkp->lk_sharecount++;
    424  1.21   thorpej 		COUNT(lkp, p, cpu_id, 1);
    425   1.1      fvdl 		/* fall into downgrade */
    426   1.1      fvdl 
    427   1.1      fvdl 	case LK_DOWNGRADE:
    428  1.19   thorpej 		if (WEHOLDIT(lkp, pid, cpu_id) == 0 ||
    429  1.19   thorpej 		    lkp->lk_exclusivecount == 0)
    430   1.1      fvdl 			panic("lockmgr: not holding exclusive lock");
    431   1.1      fvdl 		lkp->lk_sharecount += lkp->lk_exclusivecount;
    432   1.1      fvdl 		lkp->lk_exclusivecount = 0;
    433  1.15      fvdl 		lkp->lk_recurselevel = 0;
    434   1.1      fvdl 		lkp->lk_flags &= ~LK_HAVE_EXCL;
    435  1.19   thorpej 		SETHOLDER(lkp, LK_NOPROC, LK_NOCPU);
    436  1.21   thorpej 		DONTHAVEIT(lkp);
    437  1.23   thorpej 		WAKEUP_WAITER(lkp);
    438   1.1      fvdl 		break;
    439   1.1      fvdl 
    440   1.1      fvdl 	case LK_EXCLUPGRADE:
    441   1.1      fvdl 		/*
    442   1.1      fvdl 		 * If another process is ahead of us to get an upgrade,
    443   1.1      fvdl 		 * then we want to fail rather than have an intervening
    444   1.1      fvdl 		 * exclusive access.
    445   1.1      fvdl 		 */
    446   1.1      fvdl 		if (lkp->lk_flags & LK_WANT_UPGRADE) {
    447   1.1      fvdl 			lkp->lk_sharecount--;
    448  1.21   thorpej 			COUNT(lkp, p, cpu_id, -1);
    449   1.1      fvdl 			error = EBUSY;
    450   1.1      fvdl 			break;
    451   1.1      fvdl 		}
    452   1.1      fvdl 		/* fall into normal upgrade */
    453   1.1      fvdl 
    454   1.1      fvdl 	case LK_UPGRADE:
    455   1.1      fvdl 		/*
    456   1.1      fvdl 		 * Upgrade a shared lock to an exclusive one. If another
    457   1.1      fvdl 		 * shared lock has already requested an upgrade to an
    458   1.1      fvdl 		 * exclusive lock, our shared lock is released and an
    459   1.1      fvdl 		 * exclusive lock is requested (which will be granted
    460   1.1      fvdl 		 * after the upgrade). If we return an error, the file
    461   1.1      fvdl 		 * will always be unlocked.
    462   1.1      fvdl 		 */
    463  1.19   thorpej 		if (WEHOLDIT(lkp, pid, cpu_id) || lkp->lk_sharecount <= 0)
    464   1.1      fvdl 			panic("lockmgr: upgrade exclusive lock");
    465   1.1      fvdl 		lkp->lk_sharecount--;
    466  1.21   thorpej 		COUNT(lkp, p, cpu_id, -1);
    467   1.1      fvdl 		/*
    468   1.1      fvdl 		 * If we are just polling, check to see if we will block.
    469   1.1      fvdl 		 */
    470   1.1      fvdl 		if ((extflags & LK_NOWAIT) &&
    471   1.1      fvdl 		    ((lkp->lk_flags & LK_WANT_UPGRADE) ||
    472   1.1      fvdl 		     lkp->lk_sharecount > 1)) {
    473   1.1      fvdl 			error = EBUSY;
    474   1.1      fvdl 			break;
    475   1.1      fvdl 		}
    476   1.1      fvdl 		if ((lkp->lk_flags & LK_WANT_UPGRADE) == 0) {
    477   1.1      fvdl 			/*
    478   1.1      fvdl 			 * We are first shared lock to request an upgrade, so
    479   1.1      fvdl 			 * request upgrade and wait for the shared count to
    480   1.1      fvdl 			 * drop to zero, then take exclusive lock.
    481   1.1      fvdl 			 */
    482   1.1      fvdl 			lkp->lk_flags |= LK_WANT_UPGRADE;
    483  1.23   thorpej 			ACQUIRE(lkp, error, extflags, 0, lkp->lk_sharecount);
    484   1.1      fvdl 			lkp->lk_flags &= ~LK_WANT_UPGRADE;
    485   1.1      fvdl 			if (error)
    486   1.1      fvdl 				break;
    487   1.1      fvdl 			lkp->lk_flags |= LK_HAVE_EXCL;
    488  1.19   thorpej 			SETHOLDER(lkp, pid, cpu_id);
    489  1.21   thorpej 			HAVEIT(lkp);
    490   1.1      fvdl 			if (lkp->lk_exclusivecount != 0)
    491   1.1      fvdl 				panic("lockmgr: non-zero exclusive count");
    492   1.1      fvdl 			lkp->lk_exclusivecount = 1;
    493  1.15      fvdl 			if (extflags & LK_SETRECURSE)
    494  1.15      fvdl 				lkp->lk_recurselevel = 1;
    495  1.21   thorpej 			COUNT(lkp, p, cpu_id, 1);
    496   1.1      fvdl 			break;
    497   1.1      fvdl 		}
    498   1.1      fvdl 		/*
    499   1.1      fvdl 		 * Someone else has requested upgrade. Release our shared
    500   1.1      fvdl 		 * lock, awaken upgrade requestor if we are the last shared
    501   1.1      fvdl 		 * lock, then request an exclusive lock.
    502   1.1      fvdl 		 */
    503  1.23   thorpej 		if (lkp->lk_sharecount == 0)
    504  1.23   thorpej 			WAKEUP_WAITER(lkp);
    505   1.1      fvdl 		/* fall into exclusive request */
    506   1.1      fvdl 
    507   1.1      fvdl 	case LK_EXCLUSIVE:
    508  1.19   thorpej 		if (WEHOLDIT(lkp, pid, cpu_id)) {
    509   1.1      fvdl 			/*
    510  1.19   thorpej 			 * Recursive lock.
    511   1.1      fvdl 			 */
    512  1.15      fvdl 			if ((extflags & LK_CANRECURSE) == 0 &&
    513  1.16  sommerfe 			     lkp->lk_recurselevel == 0) {
    514  1.16  sommerfe 				if (extflags & LK_RECURSEFAIL) {
    515  1.16  sommerfe 					error = EDEADLK;
    516  1.16  sommerfe 					break;
    517  1.16  sommerfe 				} else
    518  1.16  sommerfe 					panic("lockmgr: locking against myself");
    519  1.16  sommerfe 			}
    520   1.1      fvdl 			lkp->lk_exclusivecount++;
    521  1.15      fvdl 			if (extflags & LK_SETRECURSE &&
    522  1.15      fvdl 			    lkp->lk_recurselevel == 0)
    523  1.15      fvdl 				lkp->lk_recurselevel = lkp->lk_exclusivecount;
    524  1.21   thorpej 			COUNT(lkp, p, cpu_id, 1);
    525   1.1      fvdl 			break;
    526   1.1      fvdl 		}
    527   1.1      fvdl 		/*
    528   1.1      fvdl 		 * If we are just polling, check to see if we will sleep.
    529   1.1      fvdl 		 */
    530   1.1      fvdl 		if ((extflags & LK_NOWAIT) && ((lkp->lk_flags &
    531   1.1      fvdl 		     (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE)) ||
    532   1.1      fvdl 		     lkp->lk_sharecount != 0)) {
    533   1.1      fvdl 			error = EBUSY;
    534   1.1      fvdl 			break;
    535   1.1      fvdl 		}
    536   1.1      fvdl 		/*
    537   1.1      fvdl 		 * Try to acquire the want_exclusive flag.
    538   1.1      fvdl 		 */
    539  1.23   thorpej 		ACQUIRE(lkp, error, extflags, 0, lkp->lk_flags &
    540   1.1      fvdl 		    (LK_HAVE_EXCL | LK_WANT_EXCL));
    541   1.1      fvdl 		if (error)
    542   1.1      fvdl 			break;
    543   1.1      fvdl 		lkp->lk_flags |= LK_WANT_EXCL;
    544   1.1      fvdl 		/*
    545   1.1      fvdl 		 * Wait for shared locks and upgrades to finish.
    546   1.1      fvdl 		 */
    547  1.23   thorpej 		ACQUIRE(lkp, error, extflags, 0, lkp->lk_sharecount != 0 ||
    548   1.1      fvdl 		       (lkp->lk_flags & LK_WANT_UPGRADE));
    549   1.1      fvdl 		lkp->lk_flags &= ~LK_WANT_EXCL;
    550   1.1      fvdl 		if (error)
    551   1.1      fvdl 			break;
    552   1.1      fvdl 		lkp->lk_flags |= LK_HAVE_EXCL;
    553  1.19   thorpej 		SETHOLDER(lkp, pid, cpu_id);
    554  1.21   thorpej 		HAVEIT(lkp);
    555   1.1      fvdl 		if (lkp->lk_exclusivecount != 0)
    556   1.1      fvdl 			panic("lockmgr: non-zero exclusive count");
    557   1.1      fvdl 		lkp->lk_exclusivecount = 1;
    558  1.15      fvdl 		if (extflags & LK_SETRECURSE)
    559  1.15      fvdl 			lkp->lk_recurselevel = 1;
    560  1.21   thorpej 		COUNT(lkp, p, cpu_id, 1);
    561   1.1      fvdl 		break;
    562   1.1      fvdl 
    563   1.1      fvdl 	case LK_RELEASE:
    564   1.1      fvdl 		if (lkp->lk_exclusivecount != 0) {
    565  1.19   thorpej 			if (WEHOLDIT(lkp, pid, cpu_id) == 0) {
    566  1.19   thorpej 				if (lkp->lk_flags & LK_SPIN) {
    567  1.19   thorpej 					panic("lockmgr: processor %lu, not "
    568  1.19   thorpej 					    "exclusive lock holder %lu "
    569  1.19   thorpej 					    "unlocking", cpu_id, lkp->lk_cpu);
    570  1.19   thorpej 				} else {
    571  1.19   thorpej 					panic("lockmgr: pid %d, not "
    572  1.19   thorpej 					    "exclusive lock holder %d "
    573  1.19   thorpej 					    "unlocking", pid,
    574  1.19   thorpej 					    lkp->lk_lockholder);
    575  1.19   thorpej 				}
    576  1.19   thorpej 			}
    577  1.15      fvdl 			if (lkp->lk_exclusivecount == lkp->lk_recurselevel)
    578  1.15      fvdl 				lkp->lk_recurselevel = 0;
    579   1.1      fvdl 			lkp->lk_exclusivecount--;
    580  1.21   thorpej 			COUNT(lkp, p, cpu_id, -1);
    581   1.1      fvdl 			if (lkp->lk_exclusivecount == 0) {
    582   1.1      fvdl 				lkp->lk_flags &= ~LK_HAVE_EXCL;
    583  1.19   thorpej 				SETHOLDER(lkp, LK_NOPROC, LK_NOCPU);
    584  1.21   thorpej 				DONTHAVEIT(lkp);
    585   1.1      fvdl 			}
    586   1.1      fvdl 		} else if (lkp->lk_sharecount != 0) {
    587   1.1      fvdl 			lkp->lk_sharecount--;
    588  1.21   thorpej 			COUNT(lkp, p, cpu_id, -1);
    589   1.1      fvdl 		}
    590  1.23   thorpej 		WAKEUP_WAITER(lkp);
    591   1.1      fvdl 		break;
    592   1.1      fvdl 
    593   1.1      fvdl 	case LK_DRAIN:
    594   1.1      fvdl 		/*
    595   1.1      fvdl 		 * Check that we do not already hold the lock, as it can
    596   1.1      fvdl 		 * never drain if we do. Unfortunately, we have no way to
    597   1.1      fvdl 		 * check for holding a shared lock, but at least we can
    598   1.1      fvdl 		 * check for an exclusive one.
    599   1.1      fvdl 		 */
    600  1.19   thorpej 		if (WEHOLDIT(lkp, pid, cpu_id))
    601   1.1      fvdl 			panic("lockmgr: draining against myself");
    602   1.1      fvdl 		/*
    603   1.1      fvdl 		 * If we are just polling, check to see if we will sleep.
    604   1.1      fvdl 		 */
    605   1.1      fvdl 		if ((extflags & LK_NOWAIT) && ((lkp->lk_flags &
    606   1.1      fvdl 		     (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE)) ||
    607   1.1      fvdl 		     lkp->lk_sharecount != 0 || lkp->lk_waitcount != 0)) {
    608   1.1      fvdl 			error = EBUSY;
    609   1.1      fvdl 			break;
    610   1.1      fvdl 		}
    611  1.23   thorpej 		ACQUIRE(lkp, error, extflags, 1,
    612  1.23   thorpej 		    ((lkp->lk_flags &
    613  1.23   thorpej 		     (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE)) ||
    614  1.23   thorpej 		     lkp->lk_sharecount != 0 ||
    615  1.23   thorpej 		     lkp->lk_waitcount != 0));
    616  1.23   thorpej 		if (error)
    617  1.23   thorpej 			break;
    618   1.1      fvdl 		lkp->lk_flags |= LK_DRAINING | LK_HAVE_EXCL;
    619  1.19   thorpej 		SETHOLDER(lkp, pid, cpu_id);
    620  1.21   thorpej 		HAVEIT(lkp);
    621   1.1      fvdl 		lkp->lk_exclusivecount = 1;
    622  1.15      fvdl 		/* XXX unlikely that we'd want this */
    623  1.15      fvdl 		if (extflags & LK_SETRECURSE)
    624  1.15      fvdl 			lkp->lk_recurselevel = 1;
    625  1.21   thorpej 		COUNT(lkp, p, cpu_id, 1);
    626   1.1      fvdl 		break;
    627   1.1      fvdl 
    628   1.1      fvdl 	default:
    629   1.1      fvdl 		simple_unlock(&lkp->lk_interlock);
    630   1.1      fvdl 		panic("lockmgr: unknown locktype request %d",
    631   1.1      fvdl 		    flags & LK_TYPE_MASK);
    632   1.1      fvdl 		/* NOTREACHED */
    633   1.1      fvdl 	}
    634  1.23   thorpej 	if ((lkp->lk_flags & (LK_WAITDRAIN|LK_SPIN)) == LK_WAITDRAIN &&
    635  1.23   thorpej 	    ((lkp->lk_flags &
    636  1.23   thorpej 	      (LK_HAVE_EXCL | LK_WANT_EXCL | LK_WANT_UPGRADE)) == 0 &&
    637   1.1      fvdl 	     lkp->lk_sharecount == 0 && lkp->lk_waitcount == 0)) {
    638   1.1      fvdl 		lkp->lk_flags &= ~LK_WAITDRAIN;
    639  1.20   thorpej 		wakeup_one((void *)&lkp->lk_flags);
    640   1.1      fvdl 	}
    641   1.1      fvdl 	simple_unlock(&lkp->lk_interlock);
    642   1.1      fvdl 	return (error);
    643   1.1      fvdl }
    644   1.1      fvdl 
    645   1.1      fvdl /*
    646   1.1      fvdl  * Print out information about state of a lock. Used by VOP_PRINT
    647   1.1      fvdl  * routines to display ststus about contained locks.
    648   1.1      fvdl  */
    649   1.2      fvdl void
    650   1.1      fvdl lockmgr_printinfo(lkp)
    651  1.19   thorpej 	__volatile struct lock *lkp;
    652   1.1      fvdl {
    653   1.1      fvdl 
    654   1.1      fvdl 	if (lkp->lk_sharecount)
    655   1.1      fvdl 		printf(" lock type %s: SHARED (count %d)", lkp->lk_wmesg,
    656   1.1      fvdl 		    lkp->lk_sharecount);
    657  1.19   thorpej 	else if (lkp->lk_flags & LK_HAVE_EXCL) {
    658  1.19   thorpej 		printf(" lock type %s: EXCL (count %d) by ",
    659  1.19   thorpej 		    lkp->lk_wmesg, lkp->lk_exclusivecount);
    660  1.19   thorpej 		if (lkp->lk_flags & LK_SPIN)
    661  1.19   thorpej 			printf("processor %lu", lkp->lk_cpu);
    662  1.19   thorpej 		else
    663  1.19   thorpej 			printf("pid %d", lkp->lk_lockholder);
    664  1.19   thorpej 	} else
    665  1.19   thorpej 		printf(" not locked");
    666  1.19   thorpej 	if ((lkp->lk_flags & LK_SPIN) == 0 && lkp->lk_waitcount > 0)
    667   1.1      fvdl 		printf(" with %d pending", lkp->lk_waitcount);
    668   1.1      fvdl }
    669   1.1      fvdl 
    670  1.21   thorpej #if defined(LOCKDEBUG) /* { */
    671  1.21   thorpej TAILQ_HEAD(, simplelock) simplelock_list =
    672  1.21   thorpej     TAILQ_HEAD_INITIALIZER(simplelock_list);
    673  1.21   thorpej 
    674  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    675  1.21   thorpej struct simplelock simplelock_list_slock = SIMPLELOCK_INITIALIZER;
    676  1.21   thorpej 
    677  1.21   thorpej #define	SLOCK_LIST_LOCK()						\
    678  1.29  sommerfe 	__cpu_simple_lock(&simplelock_list_slock.lock_data)
    679  1.21   thorpej 
    680  1.21   thorpej #define	SLOCK_LIST_UNLOCK()						\
    681  1.29  sommerfe 	__cpu_simple_unlock(&simplelock_list_slock.lock_data)
    682  1.21   thorpej 
    683  1.28   thorpej #if defined(__HAVE_ATOMIC_OPERATIONS) /* { */
    684  1.21   thorpej #define	SLOCK_COUNT(x)							\
    685  1.28   thorpej 	atomic_add_ulong(&curcpu()->ci_simple_locks, (x))
    686  1.28   thorpej #else
    687  1.28   thorpej #define	SLOCK_COUNT(x)		/* not safe */
    688  1.28   thorpej #endif /* __HAVE_ATOMIC_OPERATIONS */ /* } */
    689  1.21   thorpej #else
    690  1.21   thorpej u_long simple_locks;
    691  1.21   thorpej 
    692  1.21   thorpej #define	SLOCK_LIST_LOCK()	/* nothing */
    693  1.21   thorpej 
    694  1.21   thorpej #define	SLOCK_LIST_UNLOCK()	/* nothing */
    695  1.21   thorpej 
    696  1.21   thorpej #define	SLOCK_COUNT(x)		simple_locks += (x)
    697  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
    698  1.21   thorpej 
    699  1.21   thorpej #ifdef DDB /* { */
    700  1.18       chs int simple_lock_debugger = 0;
    701  1.21   thorpej #define	SLOCK_DEBUGGER()	if (simple_lock_debugger) Debugger()
    702  1.21   thorpej #else
    703  1.21   thorpej #define	SLOCK_DEBUGGER()	/* nothing */
    704  1.21   thorpej #endif /* } */
    705  1.21   thorpej 
    706  1.26  sommerfe #ifdef MULTIPROCESSOR
    707  1.26  sommerfe #define SLOCK_MP()		lock_printf("on cpu %d\n", cpu_number())
    708  1.26  sommerfe #else
    709  1.26  sommerfe #define SLOCK_MP()		/* nothing */
    710  1.26  sommerfe #endif
    711  1.26  sommerfe 
    712  1.21   thorpej #define	SLOCK_WHERE(str, alp, id, l)					\
    713  1.21   thorpej do {									\
    714  1.25   thorpej 	lock_printf(str);						\
    715  1.26  sommerfe 	lock_printf("lock: %p, currently at: %s:%d\n", (alp), (id), (l));		\
    716  1.26  sommerfe 	SLOCK_MP();							\
    717  1.21   thorpej 	if ((alp)->lock_file != NULL)					\
    718  1.25   thorpej 		lock_printf("last locked: %s:%d\n", (alp)->lock_file,	\
    719  1.21   thorpej 		    (alp)->lock_line);					\
    720  1.21   thorpej 	if ((alp)->unlock_file != NULL)					\
    721  1.25   thorpej 		lock_printf("last unlocked: %s:%d\n", (alp)->unlock_file, \
    722  1.21   thorpej 		    (alp)->unlock_line);				\
    723  1.21   thorpej 	SLOCK_DEBUGGER();						\
    724  1.21   thorpej } while (0)
    725  1.12       chs 
    726   1.1      fvdl /*
    727   1.1      fvdl  * Simple lock functions so that the debugger can see from whence
    728   1.1      fvdl  * they are being called.
    729   1.1      fvdl  */
    730   1.1      fvdl void
    731   1.1      fvdl simple_lock_init(alp)
    732   1.1      fvdl 	struct simplelock *alp;
    733   1.1      fvdl {
    734  1.21   thorpej 
    735  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    736  1.27   thorpej 	__cpu_simple_lock_init(&alp->lock_data);
    737  1.21   thorpej #else
    738  1.27   thorpej 	alp->lock_data = __SIMPLELOCK_UNLOCKED;
    739  1.21   thorpej #endif /* } */
    740   1.5       chs 	alp->lock_file = NULL;
    741   1.5       chs 	alp->lock_line = 0;
    742   1.5       chs 	alp->unlock_file = NULL;
    743   1.5       chs 	alp->unlock_line = 0;
    744  1.10   thorpej 	alp->lock_holder = 0;
    745   1.1      fvdl }
    746   1.1      fvdl 
    747   1.1      fvdl void
    748   1.1      fvdl _simple_lock(alp, id, l)
    749   1.1      fvdl 	__volatile struct simplelock *alp;
    750   1.1      fvdl 	const char *id;
    751   1.1      fvdl 	int l;
    752   1.1      fvdl {
    753  1.24   thorpej 	cpuid_t cpu_id = cpu_number();
    754  1.12       chs 	int s;
    755  1.12       chs 
    756  1.18       chs 	s = splhigh();
    757  1.21   thorpej 
    758  1.21   thorpej 	/*
    759  1.21   thorpej 	 * MULTIPROCESSOR case: This is `safe' since if it's not us, we
    760  1.21   thorpej 	 * don't take any action, and just fall into the normal spin case.
    761  1.21   thorpej 	 */
    762  1.27   thorpej 	if (alp->lock_data == __SIMPLELOCK_LOCKED) {
    763  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    764  1.21   thorpej 		if (alp->lock_holder == cpu_id) {
    765  1.21   thorpej 			SLOCK_WHERE("simple_lock: locking against myself\n",
    766  1.21   thorpej 			    alp, id, l);
    767  1.21   thorpej 			goto out;
    768   1.1      fvdl 		}
    769  1.21   thorpej #else
    770  1.21   thorpej 		SLOCK_WHERE("simple_lock: lock held\n", alp, id, l);
    771  1.21   thorpej 		goto out;
    772  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
    773   1.1      fvdl 	}
    774  1.21   thorpej 
    775  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    776  1.21   thorpej 	/* Acquire the lock before modifying any fields. */
    777  1.27   thorpej 	__cpu_simple_lock(&alp->lock_data);
    778  1.21   thorpej #else
    779  1.27   thorpej 	alp->lock_data = __SIMPLELOCK_LOCKED;
    780  1.21   thorpej #endif /* } */
    781  1.21   thorpej 
    782   1.5       chs 	alp->lock_file = id;
    783   1.5       chs 	alp->lock_line = l;
    784  1.21   thorpej 	alp->lock_holder = cpu_id;
    785  1.21   thorpej 
    786  1.21   thorpej 	SLOCK_LIST_LOCK();
    787  1.21   thorpej 	/* XXX Cast away volatile */
    788  1.21   thorpej 	TAILQ_INSERT_TAIL(&simplelock_list, (struct simplelock *)alp, list);
    789  1.21   thorpej 	SLOCK_LIST_UNLOCK();
    790  1.21   thorpej 
    791  1.21   thorpej 	SLOCK_COUNT(1);
    792  1.21   thorpej 
    793  1.21   thorpej  out:
    794  1.18       chs 	splx(s);
    795   1.1      fvdl }
    796   1.1      fvdl 
    797   1.1      fvdl int
    798   1.1      fvdl _simple_lock_try(alp, id, l)
    799   1.1      fvdl 	__volatile struct simplelock *alp;
    800   1.1      fvdl 	const char *id;
    801   1.1      fvdl 	int l;
    802   1.1      fvdl {
    803  1.24   thorpej 	cpuid_t cpu_id = cpu_number();
    804  1.21   thorpej 	int s, rv = 0;
    805   1.1      fvdl 
    806  1.18       chs 	s = splhigh();
    807  1.21   thorpej 
    808  1.21   thorpej 	/*
    809  1.21   thorpej 	 * MULTIPROCESSOR case: This is `safe' since if it's not us, we
    810  1.21   thorpej 	 * don't take any action.
    811  1.21   thorpej 	 */
    812  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    813  1.27   thorpej 	if ((rv = __cpu_simple_lock_try(&alp->lock_data)) == 0) {
    814  1.21   thorpej 		if (alp->lock_holder == cpu_id)
    815  1.21   thorpej 			SLOCK_WHERE("simple_lock_try: locking against myself\n",
    816  1.26  sommerfe 			    alp, id, l);
    817  1.21   thorpej 		goto out;
    818  1.21   thorpej 	}
    819  1.21   thorpej #else
    820  1.27   thorpej 	if (alp->lock_data == __SIMPLELOCK_LOCKED) {
    821  1.21   thorpej 		SLOCK_WHERE("simple_lock_try: lock held\n", alp, id, l);
    822  1.21   thorpej 		goto out;
    823  1.18       chs 	}
    824  1.27   thorpej 	alp->lock_data = __SIMPLELOCK_LOCKED;
    825  1.21   thorpej #endif /* MULTIPROCESSOR */ /* } */
    826  1.21   thorpej 
    827  1.21   thorpej 	/*
    828  1.21   thorpej 	 * At this point, we have acquired the lock.
    829  1.21   thorpej 	 */
    830  1.21   thorpej 
    831  1.21   thorpej 	rv = 1;
    832  1.18       chs 
    833   1.5       chs 	alp->lock_file = id;
    834   1.5       chs 	alp->lock_line = l;
    835  1.21   thorpej 	alp->lock_holder = cpu_id;
    836  1.21   thorpej 
    837  1.21   thorpej 	SLOCK_LIST_LOCK();
    838  1.21   thorpej 	/* XXX Cast away volatile. */
    839  1.21   thorpej 	TAILQ_INSERT_TAIL(&simplelock_list, (struct simplelock *)alp, list);
    840  1.21   thorpej 	SLOCK_LIST_UNLOCK();
    841  1.21   thorpej 
    842  1.21   thorpej 	SLOCK_COUNT(1);
    843  1.21   thorpej 
    844  1.21   thorpej  out:
    845  1.12       chs 	splx(s);
    846  1.21   thorpej 	return (rv);
    847   1.1      fvdl }
    848   1.1      fvdl 
    849   1.1      fvdl void
    850   1.1      fvdl _simple_unlock(alp, id, l)
    851   1.1      fvdl 	__volatile struct simplelock *alp;
    852   1.1      fvdl 	const char *id;
    853   1.1      fvdl 	int l;
    854   1.1      fvdl {
    855  1.12       chs 	int s;
    856   1.1      fvdl 
    857  1.18       chs 	s = splhigh();
    858  1.21   thorpej 
    859  1.21   thorpej 	/*
    860  1.21   thorpej 	 * MULTIPROCESSOR case: This is `safe' because we think we hold
    861  1.21   thorpej 	 * the lock, and if we don't, we don't take any action.
    862  1.21   thorpej 	 */
    863  1.27   thorpej 	if (alp->lock_data == __SIMPLELOCK_UNLOCKED) {
    864  1.21   thorpej 		SLOCK_WHERE("simple_unlock: lock not held\n",
    865  1.21   thorpej 		    alp, id, l);
    866  1.21   thorpej 		goto out;
    867  1.21   thorpej 	}
    868  1.21   thorpej 
    869  1.21   thorpej 	SLOCK_LIST_LOCK();
    870  1.21   thorpej 	TAILQ_REMOVE(&simplelock_list, alp, list);
    871  1.21   thorpej 	SLOCK_LIST_UNLOCK();
    872  1.21   thorpej 
    873  1.21   thorpej 	SLOCK_COUNT(-1);
    874  1.21   thorpej 
    875  1.21   thorpej 	alp->list.tqe_next = NULL;	/* sanity */
    876  1.21   thorpej 	alp->list.tqe_prev = NULL;	/* sanity */
    877  1.21   thorpej 
    878   1.5       chs 	alp->unlock_file = id;
    879   1.5       chs 	alp->unlock_line = l;
    880  1.21   thorpej 
    881  1.21   thorpej #if defined(MULTIPROCESSOR) /* { */
    882  1.26  sommerfe 	alp->lock_holder = LK_NOCPU;
    883  1.21   thorpej 	/* Now that we've modified all fields, release the lock. */
    884  1.27   thorpej 	__cpu_simple_unlock(&alp->lock_data);
    885  1.21   thorpej #else
    886  1.27   thorpej 	alp->lock_data = __SIMPLELOCK_UNLOCKED;
    887  1.21   thorpej #endif /* } */
    888  1.21   thorpej 
    889  1.21   thorpej  out:
    890  1.18       chs 	splx(s);
    891  1.12       chs }
    892  1.12       chs 
    893  1.12       chs void
    894  1.12       chs simple_lock_dump()
    895  1.12       chs {
    896  1.12       chs 	struct simplelock *alp;
    897  1.12       chs 	int s;
    898  1.12       chs 
    899  1.12       chs 	s = splhigh();
    900  1.21   thorpej 	SLOCK_LIST_LOCK();
    901  1.25   thorpej 	lock_printf("all simple locks:\n");
    902  1.21   thorpej 	for (alp = TAILQ_FIRST(&simplelock_list); alp != NULL;
    903  1.21   thorpej 	     alp = TAILQ_NEXT(alp, list)) {
    904  1.25   thorpej 		lock_printf("%p CPU %lu %s:%d\n", alp, alp->lock_holder,
    905  1.21   thorpej 		    alp->lock_file, alp->lock_line);
    906  1.12       chs 	}
    907  1.21   thorpej 	SLOCK_LIST_UNLOCK();
    908  1.12       chs 	splx(s);
    909  1.12       chs }
    910  1.12       chs 
    911  1.12       chs void
    912  1.12       chs simple_lock_freecheck(start, end)
    913  1.21   thorpej 	void *start, *end;
    914  1.12       chs {
    915  1.12       chs 	struct simplelock *alp;
    916  1.12       chs 	int s;
    917  1.12       chs 
    918  1.12       chs 	s = splhigh();
    919  1.21   thorpej 	SLOCK_LIST_LOCK();
    920  1.21   thorpej 	for (alp = TAILQ_FIRST(&simplelock_list); alp != NULL;
    921  1.21   thorpej 	     alp = TAILQ_NEXT(alp, list)) {
    922  1.12       chs 		if ((void *)alp >= start && (void *)alp < end) {
    923  1.25   thorpej 			lock_printf("freeing simple_lock %p CPU %lu %s:%d\n",
    924  1.21   thorpej 			    alp, alp->lock_holder, alp->lock_file,
    925  1.21   thorpej 			    alp->lock_line);
    926  1.21   thorpej 			SLOCK_DEBUGGER();
    927  1.12       chs 		}
    928  1.12       chs 	}
    929  1.21   thorpej 	SLOCK_LIST_UNLOCK();
    930  1.12       chs 	splx(s);
    931   1.1      fvdl }
    932  1.21   thorpej #endif /* LOCKDEBUG */ /* } */
    933