Index: sys/task.h
===================================================================
RCS file: /cvs/src/sys/sys/task.h,v
retrieving revision 1.14
diff -u -p -r1.14 task.h
--- sys/task.h	1 Apr 2019 03:23:45 -0000	1.14
+++ sys/task.h	26 Apr 2019 02:44:23 -0000
@@ -46,6 +46,8 @@ struct taskq	*taskq_create(const char *,
 void		 taskq_destroy(struct taskq *);
 void		 taskq_barrier(struct taskq *);
 
+void		 taskq_del_barrier(struct taskq *, struct task *);
+
 void		 task_set(struct task *, void (*)(void *), void *);
 int		 task_add(struct taskq *, struct task *);
 int		 task_del(struct taskq *, struct task *);
Index: kern/kern_task.c
===================================================================
RCS file: /cvs/src/sys/kern/kern_task.c,v
retrieving revision 1.24
diff -u -p -r1.24 kern_task.c
--- kern/kern_task.c	1 Apr 2019 03:23:45 -0000	1.24
+++ kern/kern_task.c	26 Apr 2019 02:44:23 -0000
@@ -23,6 +23,18 @@
 #include <sys/kthread.h>
 #include <sys/task.h>
 #include <sys/proc.h>
+#include <sys/witness.h>
+
+#ifdef WITNESS
+
+static struct lock_type taskq_lock_type = {
+	.lt_name = "taskq"
+};
+
+#define TASKQ_LOCK_FLAGS LO_WITNESS | LO_INITIALIZED | LO_SLEEPABLE | \
+    (LO_CLASS_RWLOCK << LO_CLASSSHIFT)
+
+#endif /* WITNESS */
 
 struct taskq {
 	enum {
@@ -37,26 +49,45 @@ struct taskq {
 
 	struct mutex		 tq_mtx;
 	struct task_list	 tq_worklist;
+#ifdef WITNESS
+	struct lock_object	 tq_lock_object;
+#endif
 };
 
+static const char taskq_sys_name[] = "systq";
+
 struct taskq taskq_sys = {
 	TQ_S_CREATED,
 	0,
 	1,
 	0,
-	"systq",
+	taskq_sys_name,
 	MUTEX_INITIALIZER(IPL_HIGH),
-	TAILQ_HEAD_INITIALIZER(taskq_sys.tq_worklist)
+	TAILQ_HEAD_INITIALIZER(taskq_sys.tq_worklist),
+#ifdef WITNESS
+	{
+		.lo_name = taskq_sys_name,
+		.lo_flags = TASKQ_LOCK_FLAGS,
+	},
+#endif
 };
 
+static const char taskq_sys_mp_name[] = "systqmp";
+
 struct taskq taskq_sys_mp = {
 	TQ_S_CREATED,
 	0,
 	1,
 	TASKQ_MPSAFE,
-	"systqmp",
+	taskq_sys_mp_name,
 	MUTEX_INITIALIZER(IPL_HIGH),
-	TAILQ_HEAD_INITIALIZER(taskq_sys_mp.tq_worklist)
+	TAILQ_HEAD_INITIALIZER(taskq_sys_mp.tq_worklist),
+#ifdef WITNESS
+	{
+		.lo_name = taskq_sys_mp_name,
+		.lo_flags = TASKQ_LOCK_FLAGS,
+	},
+#endif
 };
 
 struct taskq *const systq = &taskq_sys;
@@ -73,7 +104,9 @@ void	taskq_thread(void *);
 void
 taskq_init(void)
 {
+	WITNESS_INIT(&systq->tq_lock_object, &taskq_lock_type);
 	kthread_create_deferred(taskq_create_thread, systq);
+	WITNESS_INIT(&systqmp->tq_lock_object, &taskq_lock_type);
 	kthread_create_deferred(taskq_create_thread, systqmp);
 }
 
@@ -96,6 +129,13 @@ taskq_create(const char *name, unsigned 
 	mtx_init_flags(&tq->tq_mtx, ipl, name, 0);
 	TAILQ_INIT(&tq->tq_worklist);
 
+#ifdef WITNESS
+	memset(&tq->tq_lock_object, 0, sizeof(tq->tq_lock_object));
+	tq->tq_lock_object.lo_name = name;
+	tq->tq_lock_object.lo_flags = TASKQ_LOCK_FLAGS;
+	witness_init(&tq->tq_lock_object, &taskq_lock_type);
+#endif
+
 	/* try to create a thread to guarantee that tasks will be serviced */
 	kthread_create_deferred(taskq_create_thread, tq);
 
@@ -181,8 +221,24 @@ taskq_barrier(struct taskq *tq)
 	struct cond c = COND_INITIALIZER();
 	struct task t = TASK_INITIALIZER(taskq_barrier_task, &c);
 
+	WITNESS_CHECKORDER(&tq->tq_lock_object, LOP_NEWORDER, NULL);
+
 	task_add(tq, &t);
+	cond_wait(&c, "tqbar");
+}
 
+void
+taskq_del_barrier(struct taskq *tq, struct task *del)
+{
+	struct cond c = COND_INITIALIZER();
+	struct task t = TASK_INITIALIZER(taskq_barrier_task, &c);
+
+	WITNESS_CHECKORDER(&tq->tq_lock_object, LOP_NEWORDER, NULL);
+
+	if (task_del(tq, del))
+		return;
+
+	task_add(tq, &t);
 	cond_wait(&c, "tqbar");
 }
 
@@ -281,8 +337,12 @@ taskq_thread(void *xtq)
 	if (ISSET(tq->tq_flags, TASKQ_MPSAFE))
 		KERNEL_UNLOCK();
 
+	WITNESS_CHECKORDER(&tq->tq_lock_object, LOP_NEWORDER, NULL);
+
 	while (taskq_next_work(tq, &work)) {
+		WITNESS_LOCK(&tq->tq_lock_object, 0);
 		(*work.t_func)(work.t_arg);
+		WITNESS_UNLOCK(&tq->tq_lock_object, 0);
 		sched_pause(yield);
 	}