--- //depot/vendor/freebsd/src/sys/kern/kern_timeout.c	2007/11/22 12:16:40
+++ //depot/user/attilio/attilio_schedlock/kern/kern_timeout.c	2008/02/05 17:08:42
@@ -639,8 +639,8 @@
 	    ("callout_init_lock: bad flags %d", flags));
 	KASSERT(lock != NULL || (flags & CALLOUT_RETURNUNLOCKED) == 0,
 	    ("callout_init_lock: CALLOUT_RETURNUNLOCKED with no lock"));
-	KASSERT(lock == NULL || LOCK_CLASS(lock) == &lock_class_mtx_sleep ||
-	    LOCK_CLASS(lock) == &lock_class_rw, ("%s: invalid lock class",
+	KASSERT(lock == NULL || !(LOCK_CLASS(lock)->lc_flags &
+	    (LC_SPINLOCK | LC_SLEEPABLE)), ("%s: invalid lock class",
 	    __func__));
 	c->c_flags = flags & (CALLOUT_RETURNUNLOCKED | CALLOUT_SHAREDLOCK);
 }
--- //depot/vendor/freebsd/src/sys/kern/subr_lock.c	2008/01/08 01:12:41
+++ //depot/user/attilio/attilio_schedlock/kern/subr_lock.c	2008/01/11 21:47:00
@@ -494,7 +494,7 @@
 	if (lock_prof_skipcount &&
 	    (++lock_prof_count % lock_prof_skipcount) != 0)
 		return;
-	spin = LOCK_CLASS(lo) == &lock_class_mtx_spin;
+	spin = (LOCK_CLASS(lo)->lc_flags & LC_SPINLOCK) ? 1 : 0;
 	if (spin && lock_prof_skipspin == 1)
 		return;
 	l = lock_profile_object_lookup(lo, spin, file, line);
@@ -523,7 +523,7 @@
 
 	if (!lock_prof_enable || (lo->lo_flags & LO_NOPROFILE))
 		return;
-	spin = LOCK_CLASS(lo) == &lock_class_mtx_spin;
+	spin = (LOCK_CLASS(lo)->lc_flags & LC_SPINLOCK) ? 1 : 0;
 	head = &curthread->td_lprof[spin];
 	critical_enter();
 	LIST_FOREACH(l, head, lpo_link)