Home | History | Annotate | Line # | Download | only in kern
subr_lockdebug.c revision 1.27
      1  1.27       ad /*	$NetBSD: subr_lockdebug.c,v 1.27 2008/02/18 16:02:41 ad Exp $	*/
      2   1.2       ad 
      3   1.2       ad /*-
      4   1.2       ad  * Copyright (c) 2006, 2007 The NetBSD Foundation, Inc.
      5   1.2       ad  * All rights reserved.
      6   1.2       ad  *
      7   1.2       ad  * This code is derived from software contributed to The NetBSD Foundation
      8   1.2       ad  * by Andrew Doran.
      9   1.2       ad  *
     10   1.2       ad  * Redistribution and use in source and binary forms, with or without
     11   1.2       ad  * modification, are permitted provided that the following conditions
     12   1.2       ad  * are met:
     13   1.2       ad  * 1. Redistributions of source code must retain the above copyright
     14   1.2       ad  *    notice, this list of conditions and the following disclaimer.
     15   1.2       ad  * 2. Redistributions in binary form must reproduce the above copyright
     16   1.2       ad  *    notice, this list of conditions and the following disclaimer in the
     17   1.2       ad  *    documentation and/or other materials provided with the distribution.
     18   1.2       ad  * 3. All advertising materials mentioning features or use of this software
     19   1.2       ad  *    must display the following acknowledgement:
     20   1.2       ad  *	This product includes software developed by the NetBSD
     21   1.2       ad  *	Foundation, Inc. and its contributors.
     22   1.2       ad  * 4. Neither the name of The NetBSD Foundation nor the names of its
     23   1.2       ad  *    contributors may be used to endorse or promote products derived
     24   1.2       ad  *    from this software without specific prior written permission.
     25   1.2       ad  *
     26   1.2       ad  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     27   1.2       ad  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     28   1.2       ad  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     29   1.2       ad  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     30   1.2       ad  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     31   1.2       ad  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     32   1.2       ad  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     33   1.2       ad  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     34   1.2       ad  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     35   1.2       ad  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     36   1.2       ad  * POSSIBILITY OF SUCH DAMAGE.
     37   1.2       ad  */
     38   1.2       ad 
     39   1.2       ad /*
     40  1.11       ad  * Basic lock debugging code shared among lock primitives.
     41   1.2       ad  */
     42   1.2       ad 
     43   1.9      dsl #include <sys/cdefs.h>
     44  1.27       ad __KERNEL_RCSID(0, "$NetBSD: subr_lockdebug.c,v 1.27 2008/02/18 16:02:41 ad Exp $");
     45   1.9      dsl 
     46   1.2       ad #include "opt_ddb.h"
     47   1.2       ad 
     48   1.2       ad #include <sys/param.h>
     49   1.2       ad #include <sys/proc.h>
     50   1.2       ad #include <sys/systm.h>
     51  1.10       ad #include <sys/kernel.h>
     52   1.2       ad #include <sys/kmem.h>
     53   1.2       ad #include <sys/lockdebug.h>
     54   1.2       ad #include <sys/sleepq.h>
     55  1.10       ad #include <sys/cpu.h>
     56  1.22       ad #include <sys/atomic.h>
     57  1.26       ad #include <sys/lock.h>
     58   1.2       ad 
     59  1.16     yamt #include <lib/libkern/rb.h>
     60  1.16     yamt 
     61  1.25       ad #include <machine/lock.h>
     62  1.25       ad 
     63   1.2       ad #ifdef LOCKDEBUG
     64   1.2       ad 
     65   1.2       ad #define	LD_BATCH_SHIFT	9
     66   1.2       ad #define	LD_BATCH	(1 << LD_BATCH_SHIFT)
     67   1.2       ad #define	LD_BATCH_MASK	(LD_BATCH - 1)
     68   1.2       ad #define	LD_MAX_LOCKS	1048576
     69   1.2       ad #define	LD_SLOP		16
     70   1.2       ad 
     71   1.2       ad #define	LD_LOCKED	0x01
     72   1.2       ad #define	LD_SLEEPER	0x02
     73   1.2       ad 
     74  1.23       ad #define	LD_WRITE_LOCK	0x80000000
     75  1.23       ad 
     76   1.2       ad typedef union lockdebuglk {
     77   1.2       ad 	struct {
     78  1.23       ad 		u_int	lku_lock;
     79  1.23       ad 		int	lku_oldspl;
     80   1.2       ad 	} ul;
     81  1.23       ad 	uint8_t	lk_pad[CACHE_LINE_SIZE];
     82  1.23       ad } volatile __aligned(CACHE_LINE_SIZE) lockdebuglk_t;
     83   1.2       ad 
     84   1.2       ad #define	lk_lock		ul.lku_lock
     85   1.2       ad #define	lk_oldspl	ul.lku_oldspl
     86   1.2       ad 
     87   1.2       ad typedef struct lockdebug {
     88  1.16     yamt 	struct rb_node	ld_rb_node;	/* must be the first member */
     89   1.2       ad 	_TAILQ_ENTRY(struct lockdebug, volatile) ld_chain;
     90   1.2       ad 	_TAILQ_ENTRY(struct lockdebug, volatile) ld_achain;
     91   1.2       ad 	volatile void	*ld_lock;
     92   1.2       ad 	lockops_t	*ld_lockops;
     93   1.2       ad 	struct lwp	*ld_lwp;
     94   1.2       ad 	uintptr_t	ld_locked;
     95   1.2       ad 	uintptr_t	ld_unlocked;
     96  1.10       ad 	uintptr_t	ld_initaddr;
     97   1.2       ad 	uint16_t	ld_shares;
     98   1.2       ad 	uint16_t	ld_cpu;
     99   1.2       ad 	uint8_t		ld_flags;
    100   1.2       ad 	uint8_t		ld_shwant;	/* advisory */
    101   1.2       ad 	uint8_t		ld_exwant;	/* advisory */
    102   1.2       ad 	uint8_t		ld_unused;
    103   1.2       ad } volatile lockdebug_t;
    104   1.2       ad 
    105   1.2       ad typedef _TAILQ_HEAD(lockdebuglist, struct lockdebug, volatile) lockdebuglist_t;
    106   1.2       ad 
    107  1.16     yamt lockdebuglk_t		ld_tree_lk;
    108  1.15     matt lockdebuglk_t		ld_sleeper_lk;
    109  1.15     matt lockdebuglk_t		ld_spinner_lk;
    110  1.15     matt lockdebuglk_t		ld_free_lk;
    111   1.2       ad 
    112  1.13     matt lockdebuglist_t		ld_sleepers = TAILQ_HEAD_INITIALIZER(ld_sleepers);
    113  1.13     matt lockdebuglist_t		ld_spinners = TAILQ_HEAD_INITIALIZER(ld_spinners);
    114  1.13     matt lockdebuglist_t		ld_free = TAILQ_HEAD_INITIALIZER(ld_free);
    115  1.13     matt lockdebuglist_t		ld_all = TAILQ_HEAD_INITIALIZER(ld_all);
    116   1.2       ad int			ld_nfree;
    117   1.2       ad int			ld_freeptr;
    118   1.2       ad int			ld_recurse;
    119   1.5       ad bool			ld_nomore;
    120   1.2       ad lockdebug_t		*ld_table[LD_MAX_LOCKS / LD_BATCH];
    121  1.27       ad unsigned int		ld_panic;
    122   1.2       ad 
    123   1.2       ad lockdebug_t		ld_prime[LD_BATCH];
    124   1.2       ad 
    125   1.5       ad static void	lockdebug_abort1(lockdebug_t *, lockdebuglk_t *lk,
    126  1.10       ad 				 const char *, const char *, bool);
    127   1.5       ad static void	lockdebug_more(void);
    128   1.5       ad static void	lockdebug_init(void);
    129   1.2       ad 
    130  1.16     yamt static signed int
    131  1.16     yamt ld_rb_compare_nodes(const struct rb_node *n1, const struct rb_node *n2)
    132  1.16     yamt {
    133  1.16     yamt 	const lockdebug_t *ld1 = (const void *)n1;
    134  1.16     yamt 	const lockdebug_t *ld2 = (const void *)n2;
    135  1.20     yamt 	const uintptr_t a = (uintptr_t)ld1->ld_lock;
    136  1.20     yamt 	const uintptr_t b = (uintptr_t)ld2->ld_lock;
    137  1.20     yamt 
    138  1.20     yamt 	if (a < b)
    139  1.20     yamt 		return 1;
    140  1.20     yamt 	if (a > b)
    141  1.16     yamt 		return -1;
    142  1.16     yamt 	return 0;
    143  1.16     yamt }
    144  1.16     yamt 
    145  1.16     yamt static signed int
    146  1.16     yamt ld_rb_compare_key(const struct rb_node *n, const void *key)
    147  1.16     yamt {
    148  1.16     yamt 	const lockdebug_t *ld = (const void *)n;
    149  1.20     yamt 	const uintptr_t a = (uintptr_t)ld->ld_lock;
    150  1.20     yamt 	const uintptr_t b = (uintptr_t)key;
    151  1.20     yamt 
    152  1.20     yamt 	if (a < b)
    153  1.20     yamt 		return 1;
    154  1.20     yamt 	if (a > b)
    155  1.16     yamt 		return -1;
    156  1.16     yamt 	return 0;
    157  1.16     yamt }
    158  1.16     yamt 
    159  1.16     yamt static struct rb_tree ld_rb_tree;
    160  1.16     yamt 
    161  1.16     yamt static const struct rb_tree_ops ld_rb_tree_ops = {
    162  1.16     yamt 	.rb_compare_nodes = ld_rb_compare_nodes,
    163  1.16     yamt 	.rb_compare_key = ld_rb_compare_key,
    164  1.16     yamt };
    165  1.16     yamt 
    166  1.23       ad static void
    167  1.23       ad lockdebug_lock_init(lockdebuglk_t *lk)
    168  1.23       ad {
    169  1.23       ad 
    170  1.23       ad 	lk->lk_lock = 0;
    171  1.23       ad }
    172  1.23       ad 
    173  1.23       ad static void
    174   1.2       ad lockdebug_lock(lockdebuglk_t *lk)
    175   1.2       ad {
    176   1.2       ad 	int s;
    177   1.2       ad 
    178   1.8       ad 	s = splhigh();
    179  1.23       ad 	do {
    180  1.23       ad 		while (lk->lk_lock != 0) {
    181  1.23       ad 			SPINLOCK_SPIN_HOOK;
    182  1.23       ad 		}
    183  1.23       ad 	} while (atomic_cas_uint(&lk->lk_lock, 0, LD_WRITE_LOCK) != 0);
    184   1.2       ad 	lk->lk_oldspl = s;
    185  1.23       ad 	membar_enter();
    186   1.2       ad }
    187   1.2       ad 
    188  1.23       ad static void
    189   1.2       ad lockdebug_unlock(lockdebuglk_t *lk)
    190   1.2       ad {
    191   1.2       ad 	int s;
    192   1.2       ad 
    193   1.2       ad 	s = lk->lk_oldspl;
    194  1.23       ad 	membar_exit();
    195  1.23       ad 	lk->lk_lock = 0;
    196  1.23       ad 	splx(s);
    197  1.23       ad }
    198  1.23       ad 
    199  1.23       ad static int
    200  1.23       ad lockdebug_lock_rd(lockdebuglk_t *lk)
    201  1.23       ad {
    202  1.23       ad 	u_int val;
    203  1.23       ad 	int s;
    204  1.23       ad 
    205  1.23       ad 	s = splhigh();
    206  1.23       ad 	do {
    207  1.23       ad 		while ((val = lk->lk_lock) == LD_WRITE_LOCK){
    208  1.23       ad 			SPINLOCK_SPIN_HOOK;
    209  1.23       ad 		}
    210  1.23       ad 	} while (atomic_cas_uint(&lk->lk_lock, val, val + 1) != val);
    211  1.23       ad 	membar_enter();
    212  1.23       ad 	return s;
    213  1.23       ad }
    214  1.23       ad 
    215  1.23       ad static void
    216  1.23       ad lockdebug_unlock_rd(lockdebuglk_t *lk, int s)
    217  1.23       ad {
    218  1.23       ad 
    219  1.23       ad 	membar_exit();
    220  1.23       ad 	atomic_dec_uint(&lk->lk_lock);
    221   1.2       ad 	splx(s);
    222   1.2       ad }
    223   1.2       ad 
    224   1.2       ad static inline lockdebug_t *
    225  1.19     yamt lockdebug_lookup1(volatile void *lock, lockdebuglk_t **lk)
    226   1.2       ad {
    227  1.16     yamt 	lockdebug_t *ld;
    228  1.23       ad 	int s;
    229   1.2       ad 
    230  1.23       ad 	s = lockdebug_lock_rd(&ld_tree_lk);
    231  1.16     yamt 	ld = (lockdebug_t *)rb_tree_find_node(&ld_rb_tree, __UNVOLATILE(lock));
    232  1.23       ad 	lockdebug_unlock_rd(&ld_tree_lk, s);
    233  1.16     yamt 	if (ld == NULL)
    234  1.19     yamt 		return NULL;
    235   1.2       ad 
    236   1.2       ad 	if ((ld->ld_flags & LD_SLEEPER) != 0)
    237   1.2       ad 		*lk = &ld_sleeper_lk;
    238   1.2       ad 	else
    239   1.2       ad 		*lk = &ld_spinner_lk;
    240   1.2       ad 
    241   1.2       ad 	lockdebug_lock(*lk);
    242   1.2       ad 	return ld;
    243   1.2       ad }
    244   1.2       ad 
    245   1.2       ad /*
    246  1.19     yamt  * lockdebug_lookup:
    247  1.19     yamt  *
    248  1.19     yamt  *	Find a lockdebug structure by a pointer to a lock and return it locked.
    249  1.19     yamt  */
    250  1.19     yamt static inline lockdebug_t *
    251  1.19     yamt lockdebug_lookup(volatile void *lock, lockdebuglk_t **lk)
    252  1.19     yamt {
    253  1.19     yamt 	lockdebug_t *ld;
    254  1.19     yamt 
    255  1.19     yamt 	ld = lockdebug_lookup1(lock, lk);
    256  1.19     yamt 	if (ld == NULL)
    257  1.19     yamt 		panic("lockdebug_lookup: uninitialized lock (lock=%p)", lock);
    258  1.19     yamt 	return ld;
    259  1.19     yamt }
    260  1.19     yamt 
    261  1.19     yamt /*
    262   1.2       ad  * lockdebug_init:
    263   1.2       ad  *
    264   1.2       ad  *	Initialize the lockdebug system.  Allocate an initial pool of
    265   1.2       ad  *	lockdebug structures before the VM system is up and running.
    266   1.2       ad  */
    267   1.5       ad static void
    268   1.2       ad lockdebug_init(void)
    269   1.2       ad {
    270   1.2       ad 	lockdebug_t *ld;
    271   1.2       ad 	int i;
    272   1.2       ad 
    273  1.23       ad 	lockdebug_lock_init(&ld_tree_lk);
    274  1.23       ad 	lockdebug_lock_init(&ld_sleeper_lk);
    275  1.23       ad 	lockdebug_lock_init(&ld_spinner_lk);
    276  1.23       ad 	lockdebug_lock_init(&ld_free_lk);
    277  1.15     matt 
    278  1.16     yamt 	rb_tree_init(&ld_rb_tree, &ld_rb_tree_ops);
    279  1.16     yamt 
    280   1.2       ad 	ld = ld_prime;
    281   1.2       ad 	ld_table[0] = ld;
    282   1.2       ad 	for (i = 1, ld++; i < LD_BATCH; i++, ld++) {
    283   1.2       ad 		TAILQ_INSERT_TAIL(&ld_free, ld, ld_chain);
    284   1.2       ad 		TAILQ_INSERT_TAIL(&ld_all, ld, ld_achain);
    285   1.2       ad 	}
    286   1.2       ad 	ld_freeptr = 1;
    287   1.2       ad 	ld_nfree = LD_BATCH - 1;
    288   1.2       ad }
    289   1.2       ad 
    290   1.2       ad /*
    291   1.2       ad  * lockdebug_alloc:
    292   1.2       ad  *
    293   1.2       ad  *	A lock is being initialized, so allocate an associated debug
    294   1.2       ad  *	structure.
    295   1.2       ad  */
    296  1.16     yamt bool
    297  1.10       ad lockdebug_alloc(volatile void *lock, lockops_t *lo, uintptr_t initaddr)
    298   1.2       ad {
    299   1.2       ad 	struct cpu_info *ci;
    300   1.2       ad 	lockdebug_t *ld;
    301  1.19     yamt 	lockdebuglk_t *lk;
    302   1.2       ad 
    303  1.27       ad 	if (lo == NULL || panicstr != NULL || ld_panic)
    304  1.16     yamt 		return false;
    305   1.5       ad 	if (ld_freeptr == 0)
    306   1.5       ad 		lockdebug_init();
    307   1.2       ad 
    308  1.19     yamt 	if ((ld = lockdebug_lookup1(lock, &lk)) != NULL) {
    309  1.19     yamt 		lockdebug_abort1(ld, lk, __func__, "already initialized", true);
    310  1.27       ad 		return false;
    311  1.19     yamt 	}
    312  1.19     yamt 
    313   1.2       ad 	/*
    314   1.2       ad 	 * Pinch a new debug structure.  We may recurse because we call
    315   1.2       ad 	 * kmem_alloc(), which may need to initialize new locks somewhere
    316   1.7    skrll 	 * down the path.  If not recursing, we try to maintain at least
    317   1.2       ad 	 * LD_SLOP structures free, which should hopefully be enough to
    318   1.2       ad 	 * satisfy kmem_alloc().  If we can't provide a structure, not to
    319   1.2       ad 	 * worry: we'll just mark the lock as not having an ID.
    320   1.2       ad 	 */
    321   1.2       ad 	lockdebug_lock(&ld_free_lk);
    322  1.23       ad 	ci = curcpu();
    323   1.2       ad 	ci->ci_lkdebug_recurse++;
    324   1.2       ad 
    325   1.2       ad 	if (TAILQ_EMPTY(&ld_free)) {
    326   1.5       ad 		if (ci->ci_lkdebug_recurse > 1 || ld_nomore) {
    327   1.2       ad 			ci->ci_lkdebug_recurse--;
    328   1.2       ad 			lockdebug_unlock(&ld_free_lk);
    329  1.16     yamt 			return false;
    330   1.2       ad 		}
    331   1.2       ad 		lockdebug_more();
    332   1.2       ad 	} else if (ci->ci_lkdebug_recurse == 1 && ld_nfree < LD_SLOP)
    333   1.2       ad 		lockdebug_more();
    334   1.2       ad 
    335   1.2       ad 	if ((ld = TAILQ_FIRST(&ld_free)) == NULL) {
    336   1.2       ad 		lockdebug_unlock(&ld_free_lk);
    337  1.16     yamt 		return false;
    338   1.2       ad 	}
    339   1.2       ad 
    340   1.2       ad 	TAILQ_REMOVE(&ld_free, ld, ld_chain);
    341   1.2       ad 	ld_nfree--;
    342   1.2       ad 
    343   1.2       ad 	ci->ci_lkdebug_recurse--;
    344   1.2       ad 	lockdebug_unlock(&ld_free_lk);
    345   1.2       ad 
    346   1.2       ad 	if (ld->ld_lock != NULL)
    347   1.2       ad 		panic("lockdebug_alloc: corrupt table");
    348   1.2       ad 
    349   1.2       ad 	if (lo->lo_sleeplock)
    350   1.2       ad 		lockdebug_lock(&ld_sleeper_lk);
    351   1.2       ad 	else
    352   1.2       ad 		lockdebug_lock(&ld_spinner_lk);
    353   1.2       ad 
    354   1.2       ad 	/* Initialise the structure. */
    355   1.2       ad 	ld->ld_lock = lock;
    356   1.2       ad 	ld->ld_lockops = lo;
    357   1.2       ad 	ld->ld_locked = 0;
    358   1.2       ad 	ld->ld_unlocked = 0;
    359   1.2       ad 	ld->ld_lwp = NULL;
    360  1.10       ad 	ld->ld_initaddr = initaddr;
    361   1.2       ad 
    362  1.16     yamt 	lockdebug_lock(&ld_tree_lk);
    363  1.16     yamt 	rb_tree_insert_node(&ld_rb_tree, __UNVOLATILE(&ld->ld_rb_node));
    364  1.16     yamt 	lockdebug_unlock(&ld_tree_lk);
    365  1.16     yamt 
    366   1.2       ad 	if (lo->lo_sleeplock) {
    367   1.2       ad 		ld->ld_flags = LD_SLEEPER;
    368   1.2       ad 		lockdebug_unlock(&ld_sleeper_lk);
    369   1.2       ad 	} else {
    370   1.2       ad 		ld->ld_flags = 0;
    371   1.2       ad 		lockdebug_unlock(&ld_spinner_lk);
    372   1.2       ad 	}
    373   1.2       ad 
    374  1.16     yamt 	return true;
    375   1.2       ad }
    376   1.2       ad 
    377   1.2       ad /*
    378   1.2       ad  * lockdebug_free:
    379   1.2       ad  *
    380   1.2       ad  *	A lock is being destroyed, so release debugging resources.
    381   1.2       ad  */
    382   1.2       ad void
    383  1.16     yamt lockdebug_free(volatile void *lock)
    384   1.2       ad {
    385   1.2       ad 	lockdebug_t *ld;
    386   1.2       ad 	lockdebuglk_t *lk;
    387   1.2       ad 
    388  1.27       ad 	if (panicstr != NULL || ld_panic)
    389   1.2       ad 		return;
    390   1.2       ad 
    391  1.16     yamt 	ld = lockdebug_lookup(lock, &lk);
    392  1.16     yamt 	if (ld == NULL) {
    393   1.2       ad 		panic("lockdebug_free: destroying uninitialized lock %p"
    394  1.16     yamt 		    "(ld_lock=%p)", lock, ld->ld_lock);
    395  1.10       ad 		lockdebug_abort1(ld, lk, __func__, "lock record follows",
    396  1.10       ad 		    true);
    397  1.27       ad 		return;
    398   1.2       ad 	}
    399  1.27       ad 	if ((ld->ld_flags & LD_LOCKED) != 0 || ld->ld_shares != 0) {
    400  1.10       ad 		lockdebug_abort1(ld, lk, __func__, "is locked", true);
    401  1.27       ad 		return;
    402  1.27       ad 	}
    403  1.16     yamt 	lockdebug_lock(&ld_tree_lk);
    404  1.16     yamt 	rb_tree_remove_node(&ld_rb_tree, __UNVOLATILE(&ld->ld_rb_node));
    405  1.16     yamt 	lockdebug_unlock(&ld_tree_lk);
    406   1.2       ad 	ld->ld_lock = NULL;
    407   1.2       ad 	lockdebug_unlock(lk);
    408   1.2       ad 
    409   1.2       ad 	lockdebug_lock(&ld_free_lk);
    410   1.2       ad 	TAILQ_INSERT_TAIL(&ld_free, ld, ld_chain);
    411   1.2       ad 	ld_nfree++;
    412   1.2       ad 	lockdebug_unlock(&ld_free_lk);
    413   1.2       ad }
    414   1.2       ad 
    415   1.2       ad /*
    416   1.2       ad  * lockdebug_more:
    417   1.2       ad  *
    418   1.2       ad  *	Allocate a batch of debug structures and add to the free list.
    419   1.2       ad  *	Must be called with ld_free_lk held.
    420   1.2       ad  */
    421   1.5       ad static void
    422   1.2       ad lockdebug_more(void)
    423   1.2       ad {
    424   1.2       ad 	lockdebug_t *ld;
    425   1.2       ad 	void *block;
    426   1.5       ad 	int i, base, m;
    427   1.2       ad 
    428   1.2       ad 	while (ld_nfree < LD_SLOP) {
    429   1.2       ad 		lockdebug_unlock(&ld_free_lk);
    430   1.2       ad 		block = kmem_zalloc(LD_BATCH * sizeof(lockdebug_t), KM_SLEEP);
    431   1.2       ad 		lockdebug_lock(&ld_free_lk);
    432   1.2       ad 
    433   1.2       ad 		if (block == NULL)
    434   1.2       ad 			return;
    435   1.2       ad 
    436   1.2       ad 		if (ld_nfree > LD_SLOP) {
    437   1.2       ad 			/* Somebody beat us to it. */
    438   1.2       ad 			lockdebug_unlock(&ld_free_lk);
    439   1.2       ad 			kmem_free(block, LD_BATCH * sizeof(lockdebug_t));
    440   1.2       ad 			lockdebug_lock(&ld_free_lk);
    441   1.2       ad 			continue;
    442   1.2       ad 		}
    443   1.2       ad 
    444   1.2       ad 		base = ld_freeptr;
    445   1.2       ad 		ld_nfree += LD_BATCH;
    446   1.2       ad 		ld = block;
    447   1.2       ad 		base <<= LD_BATCH_SHIFT;
    448   1.5       ad 		m = min(LD_MAX_LOCKS, base + LD_BATCH);
    449   1.5       ad 
    450   1.5       ad 		if (m == LD_MAX_LOCKS)
    451   1.5       ad 			ld_nomore = true;
    452   1.2       ad 
    453   1.5       ad 		for (i = base; i < m; i++, ld++) {
    454   1.2       ad 			TAILQ_INSERT_TAIL(&ld_free, ld, ld_chain);
    455   1.2       ad 			TAILQ_INSERT_TAIL(&ld_all, ld, ld_achain);
    456   1.2       ad 		}
    457   1.2       ad 
    458  1.22       ad 		membar_producer();
    459   1.2       ad 		ld_table[ld_freeptr++] = block;
    460   1.2       ad 	}
    461   1.2       ad }
    462   1.2       ad 
    463   1.2       ad /*
    464   1.2       ad  * lockdebug_wantlock:
    465   1.2       ad  *
    466   1.2       ad  *	Process the preamble to a lock acquire.
    467   1.2       ad  */
    468   1.2       ad void
    469  1.16     yamt lockdebug_wantlock(volatile void *lock, uintptr_t where, int shared)
    470   1.2       ad {
    471   1.2       ad 	struct lwp *l = curlwp;
    472   1.2       ad 	lockdebuglk_t *lk;
    473   1.2       ad 	lockdebug_t *ld;
    474   1.3  thorpej 	bool recurse;
    475   1.2       ad 
    476   1.2       ad 	(void)shared;
    477   1.4  thorpej 	recurse = false;
    478   1.2       ad 
    479  1.27       ad 	if (panicstr != NULL || ld_panic)
    480   1.2       ad 		return;
    481   1.2       ad 
    482  1.16     yamt 	if ((ld = lockdebug_lookup(lock, &lk)) == NULL)
    483   1.2       ad 		return;
    484   1.2       ad 
    485   1.2       ad 	if ((ld->ld_flags & LD_LOCKED) != 0) {
    486   1.2       ad 		if ((ld->ld_flags & LD_SLEEPER) != 0) {
    487   1.2       ad 			if (ld->ld_lwp == l)
    488   1.4  thorpej 				recurse = true;
    489   1.2       ad 		} else if (ld->ld_cpu == (uint16_t)cpu_number())
    490   1.4  thorpej 			recurse = true;
    491   1.2       ad 	}
    492   1.2       ad 
    493  1.10       ad 	if (cpu_intr_p()) {
    494  1.27       ad 		if ((ld->ld_flags & LD_SLEEPER) != 0) {
    495  1.10       ad 			lockdebug_abort1(ld, lk, __func__,
    496  1.10       ad 			    "acquiring sleep lock from interrupt context",
    497  1.10       ad 			    true);
    498  1.27       ad 			return;
    499  1.27       ad 		}
    500  1.10       ad 	}
    501  1.10       ad 
    502   1.2       ad 	if (shared)
    503   1.2       ad 		ld->ld_shwant++;
    504   1.2       ad 	else
    505   1.2       ad 		ld->ld_exwant++;
    506   1.2       ad 
    507  1.27       ad 	if (recurse) {
    508  1.10       ad 		lockdebug_abort1(ld, lk, __func__, "locking against myself",
    509  1.10       ad 		    true);
    510  1.27       ad 		return;
    511  1.27       ad 	}
    512   1.2       ad 
    513   1.2       ad 	lockdebug_unlock(lk);
    514   1.2       ad }
    515   1.2       ad 
    516   1.2       ad /*
    517   1.2       ad  * lockdebug_locked:
    518   1.2       ad  *
    519   1.2       ad  *	Process a lock acquire operation.
    520   1.2       ad  */
    521   1.2       ad void
    522  1.16     yamt lockdebug_locked(volatile void *lock, uintptr_t where, int shared)
    523   1.2       ad {
    524   1.2       ad 	struct lwp *l = curlwp;
    525   1.2       ad 	lockdebuglk_t *lk;
    526   1.2       ad 	lockdebug_t *ld;
    527   1.2       ad 
    528  1.27       ad 	if (panicstr != NULL || ld_panic)
    529   1.2       ad 		return;
    530   1.2       ad 
    531  1.16     yamt 	if ((ld = lockdebug_lookup(lock, &lk)) == NULL)
    532   1.2       ad 		return;
    533   1.2       ad 
    534   1.2       ad 	if (shared) {
    535   1.2       ad 		l->l_shlocks++;
    536   1.2       ad 		ld->ld_shares++;
    537   1.2       ad 		ld->ld_shwant--;
    538   1.2       ad 	} else {
    539  1.27       ad 		if ((ld->ld_flags & LD_LOCKED) != 0) {
    540   1.2       ad 			lockdebug_abort1(ld, lk, __func__,
    541  1.10       ad 			    "already locked", true);
    542  1.27       ad 			return;
    543  1.27       ad 		}
    544   1.2       ad 
    545   1.2       ad 		ld->ld_flags |= LD_LOCKED;
    546   1.2       ad 		ld->ld_locked = where;
    547   1.2       ad 		ld->ld_cpu = (uint16_t)cpu_number();
    548   1.2       ad 		ld->ld_lwp = l;
    549   1.2       ad 		ld->ld_exwant--;
    550   1.2       ad 
    551   1.2       ad 		if ((ld->ld_flags & LD_SLEEPER) != 0) {
    552   1.2       ad 			l->l_exlocks++;
    553   1.2       ad 			TAILQ_INSERT_TAIL(&ld_sleepers, ld, ld_chain);
    554   1.2       ad 		} else {
    555   1.2       ad 			curcpu()->ci_spin_locks2++;
    556   1.2       ad 			TAILQ_INSERT_TAIL(&ld_spinners, ld, ld_chain);
    557   1.2       ad 		}
    558   1.2       ad 	}
    559   1.2       ad 
    560   1.2       ad 	lockdebug_unlock(lk);
    561   1.2       ad }
    562   1.2       ad 
    563   1.2       ad /*
    564   1.2       ad  * lockdebug_unlocked:
    565   1.2       ad  *
    566   1.2       ad  *	Process a lock release operation.
    567   1.2       ad  */
    568   1.2       ad void
    569  1.16     yamt lockdebug_unlocked(volatile void *lock, uintptr_t where, int shared)
    570   1.2       ad {
    571   1.2       ad 	struct lwp *l = curlwp;
    572   1.2       ad 	lockdebuglk_t *lk;
    573   1.2       ad 	lockdebug_t *ld;
    574   1.2       ad 
    575  1.27       ad 	if (panicstr != NULL || ld_panic)
    576   1.2       ad 		return;
    577   1.2       ad 
    578  1.16     yamt 	if ((ld = lockdebug_lookup(lock, &lk)) == NULL)
    579   1.2       ad 		return;
    580   1.2       ad 
    581   1.2       ad 	if (shared) {
    582  1.27       ad 		if (l->l_shlocks == 0) {
    583   1.2       ad 			lockdebug_abort1(ld, lk, __func__,
    584  1.10       ad 			    "no shared locks held by LWP", true);
    585  1.27       ad 			return;
    586  1.27       ad 		}
    587  1.27       ad 		if (ld->ld_shares == 0) {
    588   1.2       ad 			lockdebug_abort1(ld, lk, __func__,
    589  1.10       ad 			    "no shared holds on this lock", true);
    590  1.27       ad 			return;
    591  1.27       ad 		}
    592   1.2       ad 		l->l_shlocks--;
    593   1.2       ad 		ld->ld_shares--;
    594   1.2       ad 	} else {
    595  1.27       ad 		if ((ld->ld_flags & LD_LOCKED) == 0) {
    596  1.10       ad 			lockdebug_abort1(ld, lk, __func__, "not locked",
    597  1.10       ad 			    true);
    598  1.27       ad 			return;
    599  1.27       ad 		}
    600   1.2       ad 
    601   1.2       ad 		if ((ld->ld_flags & LD_SLEEPER) != 0) {
    602  1.27       ad 			if (ld->ld_lwp != curlwp) {
    603   1.2       ad 				lockdebug_abort1(ld, lk, __func__,
    604  1.10       ad 				    "not held by current LWP", true);
    605  1.27       ad 				return;
    606  1.27       ad 			}
    607   1.2       ad 			ld->ld_flags &= ~LD_LOCKED;
    608   1.2       ad 			ld->ld_unlocked = where;
    609   1.2       ad 			ld->ld_lwp = NULL;
    610   1.2       ad 			curlwp->l_exlocks--;
    611   1.2       ad 			TAILQ_REMOVE(&ld_sleepers, ld, ld_chain);
    612   1.2       ad 		} else {
    613  1.27       ad 			if (ld->ld_cpu != (uint16_t)cpu_number()) {
    614   1.2       ad 				lockdebug_abort1(ld, lk, __func__,
    615  1.10       ad 				    "not held by current CPU", true);
    616  1.27       ad 				return;
    617  1.27       ad 			}
    618   1.2       ad 			ld->ld_flags &= ~LD_LOCKED;
    619   1.2       ad 			ld->ld_unlocked = where;
    620   1.2       ad 			ld->ld_lwp = NULL;
    621   1.2       ad 			curcpu()->ci_spin_locks2--;
    622   1.2       ad 			TAILQ_REMOVE(&ld_spinners, ld, ld_chain);
    623   1.2       ad 		}
    624   1.2       ad 	}
    625   1.2       ad 
    626   1.2       ad 	lockdebug_unlock(lk);
    627   1.2       ad }
    628   1.2       ad 
    629   1.2       ad /*
    630   1.2       ad  * lockdebug_barrier:
    631   1.2       ad  *
    632   1.2       ad  *	Panic if we hold more than one specified spin lock, and optionally,
    633   1.2       ad  *	if we hold sleep locks.
    634   1.2       ad  */
    635   1.2       ad void
    636   1.2       ad lockdebug_barrier(volatile void *spinlock, int slplocks)
    637   1.2       ad {
    638   1.2       ad 	struct lwp *l = curlwp;
    639   1.2       ad 	lockdebug_t *ld;
    640   1.2       ad 	uint16_t cpuno;
    641  1.23       ad 	int s;
    642   1.2       ad 
    643  1.27       ad 	if (panicstr != NULL || ld_panic)
    644   1.2       ad 		return;
    645   1.2       ad 
    646  1.23       ad 	crit_enter();
    647  1.23       ad 
    648   1.2       ad 	if (curcpu()->ci_spin_locks2 != 0) {
    649   1.2       ad 		cpuno = (uint16_t)cpu_number();
    650   1.2       ad 
    651  1.23       ad 		s = lockdebug_lock_rd(&ld_spinner_lk);
    652   1.2       ad 		TAILQ_FOREACH(ld, &ld_spinners, ld_chain) {
    653   1.2       ad 			if (ld->ld_lock == spinlock) {
    654  1.27       ad 				if (ld->ld_cpu != cpuno) {
    655   1.2       ad 					lockdebug_abort1(ld, &ld_spinner_lk,
    656   1.2       ad 					    __func__,
    657  1.10       ad 					    "not held by current CPU", true);
    658  1.27       ad 					return;
    659  1.27       ad 				}
    660   1.2       ad 				continue;
    661   1.2       ad 			}
    662  1.27       ad 			if (ld->ld_cpu == cpuno && (l->l_pflag & LP_INTR) == 0) {
    663   1.2       ad 				lockdebug_abort1(ld, &ld_spinner_lk,
    664  1.10       ad 				    __func__, "spin lock held", true);
    665  1.27       ad 				return;
    666  1.27       ad 			}
    667   1.2       ad 		}
    668  1.23       ad 		lockdebug_unlock_rd(&ld_spinner_lk, s);
    669   1.2       ad 	}
    670   1.2       ad 
    671   1.2       ad 	if (!slplocks) {
    672   1.2       ad 		if (l->l_exlocks != 0) {
    673  1.23       ad 			s = lockdebug_lock_rd(&ld_sleeper_lk);
    674   1.2       ad 			TAILQ_FOREACH(ld, &ld_sleepers, ld_chain) {
    675  1.27       ad 				if (ld->ld_lwp == l) {
    676   1.2       ad 					lockdebug_abort1(ld, &ld_sleeper_lk,
    677  1.10       ad 					    __func__, "sleep lock held", true);
    678  1.27       ad 					return;
    679  1.27       ad 				}
    680   1.2       ad 			}
    681  1.23       ad 			lockdebug_unlock_rd(&ld_sleeper_lk, s);
    682   1.2       ad 		}
    683   1.2       ad 		if (l->l_shlocks != 0)
    684   1.2       ad 			panic("lockdebug_barrier: holding %d shared locks",
    685   1.2       ad 			    l->l_shlocks);
    686   1.2       ad 	}
    687  1.23       ad 
    688  1.23       ad 	crit_exit();
    689   1.2       ad }
    690   1.2       ad 
    691   1.2       ad /*
    692  1.10       ad  * lockdebug_mem_check:
    693  1.10       ad  *
    694  1.10       ad  *	Check for in-use locks within a memory region that is
    695  1.16     yamt  *	being freed.
    696  1.10       ad  */
    697  1.10       ad void
    698  1.10       ad lockdebug_mem_check(const char *func, void *base, size_t sz)
    699  1.10       ad {
    700  1.16     yamt 	lockdebug_t *ld;
    701  1.10       ad 	lockdebuglk_t *lk;
    702  1.23       ad 	int s;
    703  1.10       ad 
    704  1.27       ad 	if (panicstr != NULL || ld_panic)
    705  1.24       ad 		return;
    706  1.24       ad 
    707  1.23       ad 	s = lockdebug_lock_rd(&ld_tree_lk);
    708  1.16     yamt 	ld = (lockdebug_t *)rb_tree_find_node_geq(&ld_rb_tree, base);
    709  1.23       ad 	if (ld != NULL) {
    710  1.23       ad 		const uintptr_t lock = (uintptr_t)ld->ld_lock;
    711  1.23       ad 
    712  1.23       ad 		if ((uintptr_t)base > lock)
    713  1.23       ad 			panic("%s: corrupt tree ld=%p, base=%p, sz=%zu",
    714  1.23       ad 			    __func__, ld, base, sz);
    715  1.23       ad 		if (lock >= (uintptr_t)base + sz)
    716  1.23       ad 			ld = NULL;
    717  1.23       ad 	}
    718  1.23       ad 	lockdebug_unlock_rd(&ld_tree_lk, s);
    719  1.16     yamt 	if (ld == NULL)
    720  1.16     yamt 		return;
    721  1.23       ad 
    722  1.16     yamt 	if ((ld->ld_flags & LD_SLEEPER) != 0)
    723  1.16     yamt 		lk = &ld_sleeper_lk;
    724  1.16     yamt 	else
    725  1.16     yamt 		lk = &ld_spinner_lk;
    726  1.10       ad 
    727  1.16     yamt 	lockdebug_lock(lk);
    728  1.23       ad 	lockdebug_abort1(ld, lk, func,
    729  1.23       ad 	    "allocation contains active lock", !cold);
    730  1.10       ad }
    731  1.10       ad 
    732  1.10       ad /*
    733   1.2       ad  * lockdebug_dump:
    734   1.2       ad  *
    735   1.2       ad  *	Dump information about a lock on panic, or for DDB.
    736   1.2       ad  */
    737   1.2       ad static void
    738   1.2       ad lockdebug_dump(lockdebug_t *ld, void (*pr)(const char *, ...))
    739   1.2       ad {
    740   1.2       ad 	int sleeper = (ld->ld_flags & LD_SLEEPER);
    741   1.2       ad 
    742   1.2       ad 	(*pr)(
    743   1.2       ad 	    "lock address : %#018lx type     : %18s\n"
    744   1.2       ad 	    "shared holds : %18u exclusive: %18u\n"
    745   1.2       ad 	    "shares wanted: %18u exclusive: %18u\n"
    746   1.2       ad 	    "current cpu  : %18u last held: %18u\n"
    747   1.2       ad 	    "current lwp  : %#018lx last held: %#018lx\n"
    748  1.10       ad 	    "last locked  : %#018lx unlocked : %#018lx\n"
    749  1.10       ad 	    "initialized  : %#018lx\n",
    750   1.2       ad 	    (long)ld->ld_lock, (sleeper ? "sleep/adaptive" : "spin"),
    751   1.2       ad 	    (unsigned)ld->ld_shares, ((ld->ld_flags & LD_LOCKED) != 0),
    752   1.2       ad 	    (unsigned)ld->ld_shwant, (unsigned)ld->ld_exwant,
    753   1.2       ad 	    (unsigned)cpu_number(), (unsigned)ld->ld_cpu,
    754   1.2       ad 	    (long)curlwp, (long)ld->ld_lwp,
    755  1.10       ad 	    (long)ld->ld_locked, (long)ld->ld_unlocked,
    756  1.10       ad 	    (long)ld->ld_initaddr);
    757   1.2       ad 
    758   1.2       ad 	if (ld->ld_lockops->lo_dump != NULL)
    759   1.2       ad 		(*ld->ld_lockops->lo_dump)(ld->ld_lock);
    760   1.2       ad 
    761   1.2       ad 	if (sleeper) {
    762   1.2       ad 		(*pr)("\n");
    763   1.2       ad 		turnstile_print(ld->ld_lock, pr);
    764   1.2       ad 	}
    765   1.2       ad }
    766   1.2       ad 
    767   1.2       ad /*
    768  1.27       ad  * lockdebug_abort1:
    769   1.2       ad  *
    770  1.27       ad  *	An error has been trapped - dump lock info and panic.
    771   1.2       ad  */
    772   1.5       ad static void
    773   1.2       ad lockdebug_abort1(lockdebug_t *ld, lockdebuglk_t *lk, const char *func,
    774  1.10       ad 		 const char *msg, bool dopanic)
    775   1.2       ad {
    776   1.2       ad 
    777  1.27       ad 	/*
    778  1.27       ad 	 * Don't make the situation wose if the system is already going
    779  1.27       ad 	 * down in flames.  Once a panic is triggered, lockdebug state
    780  1.27       ad 	 * becomes stale and cannot be trusted.
    781  1.27       ad 	 */
    782  1.27       ad 	if (atomic_inc_uint_nv(&ld_panic) != 1) {
    783  1.27       ad 		lockdebug_unlock(lk);
    784  1.27       ad 		return;
    785  1.27       ad 	}
    786  1.27       ad 
    787   1.2       ad 	printf_nolog("%s error: %s: %s\n\n", ld->ld_lockops->lo_name,
    788   1.2       ad 	    func, msg);
    789   1.2       ad 	lockdebug_dump(ld, printf_nolog);
    790   1.2       ad 	lockdebug_unlock(lk);
    791   1.2       ad 	printf_nolog("\n");
    792  1.10       ad 	if (dopanic)
    793  1.10       ad 		panic("LOCKDEBUG");
    794   1.2       ad }
    795   1.2       ad 
    796   1.2       ad #endif	/* LOCKDEBUG */
    797   1.2       ad 
    798   1.2       ad /*
    799   1.2       ad  * lockdebug_lock_print:
    800   1.2       ad  *
    801   1.2       ad  *	Handle the DDB 'show lock' command.
    802   1.2       ad  */
    803   1.2       ad #ifdef DDB
    804   1.2       ad void
    805   1.2       ad lockdebug_lock_print(void *addr, void (*pr)(const char *, ...))
    806   1.2       ad {
    807   1.2       ad #ifdef LOCKDEBUG
    808   1.2       ad 	lockdebug_t *ld;
    809   1.2       ad 
    810   1.2       ad 	TAILQ_FOREACH(ld, &ld_all, ld_achain) {
    811   1.2       ad 		if (ld->ld_lock == addr) {
    812   1.2       ad 			lockdebug_dump(ld, pr);
    813   1.2       ad 			return;
    814   1.2       ad 		}
    815   1.2       ad 	}
    816   1.2       ad 	(*pr)("Sorry, no record of a lock with address %p found.\n", addr);
    817   1.2       ad #else
    818   1.2       ad 	(*pr)("Sorry, kernel not built with the LOCKDEBUG option.\n");
    819   1.2       ad #endif	/* LOCKDEBUG */
    820   1.2       ad }
    821   1.2       ad #endif	/* DDB */
    822   1.2       ad 
    823   1.2       ad /*
    824   1.2       ad  * lockdebug_abort:
    825   1.2       ad  *
    826   1.2       ad  *	An error has been trapped - dump lock info and call panic().
    827   1.2       ad  */
    828   1.2       ad void
    829  1.16     yamt lockdebug_abort(volatile void *lock, lockops_t *ops, const char *func,
    830  1.16     yamt 		const char *msg)
    831   1.2       ad {
    832   1.2       ad #ifdef LOCKDEBUG
    833   1.2       ad 	lockdebug_t *ld;
    834   1.2       ad 	lockdebuglk_t *lk;
    835   1.2       ad 
    836  1.16     yamt 	if ((ld = lockdebug_lookup(lock, &lk)) != NULL) {
    837  1.10       ad 		lockdebug_abort1(ld, lk, func, msg, true);
    838   1.2       ad 		/* NOTREACHED */
    839   1.2       ad 	}
    840   1.2       ad #endif	/* LOCKDEBUG */
    841   1.2       ad 
    842  1.27       ad 	/*
    843  1.27       ad 	 * Complain first on the occurrance only.  Otherwise proceeed to
    844  1.27       ad 	 * panic where we will `rendezvous' with other CPUs if the machine
    845  1.27       ad 	 * is going down in flames.
    846  1.27       ad 	 */
    847  1.27       ad 	if (atomic_inc_uint_nv(&ld_panic) == 1) {
    848  1.27       ad 		printf_nolog("%s error: %s: %s\n\n"
    849  1.27       ad 		    "lock address : %#018lx\n"
    850  1.27       ad 		    "current cpu  : %18d\n"
    851  1.27       ad 		    "current lwp  : %#018lx\n",
    852  1.27       ad 		    ops->lo_name, func, msg, (long)lock, (int)cpu_number(),
    853  1.27       ad 		    (long)curlwp);
    854  1.27       ad 		(*ops->lo_dump)(lock);
    855  1.27       ad 		printf_nolog("\n");
    856  1.27       ad 	}
    857   1.2       ad 
    858   1.2       ad 	panic("lock error");
    859   1.2       ad }
    860