Work queue thread safety checks for DEBUG build.

1 files changed, 37 insertions, 0 deletions

diff --git a/src/nxt_work_queue.c b/src/nxt_work_queue.c
index ecdc65ff..85f5cc3e 100644
--- a/src/nxt_work_queue.c
+++ b/src/nxt_work_queue.c
@@ -32,6 +32,39 @@ static void nxt_work_queue_allocate(nxt_work_queue_cache_t *cache);
 static nxt_uint_t  nxt_work_queue_bucket_items = 409;
 
 
+#if (NXT_DEBUG)
+
+nxt_inline void
+nxt_work_queue_thread_assert(nxt_work_queue_t *wq)
+{
+    nxt_tid_t     tid;
+    nxt_thread_t  *thread;
+
+    thread = nxt_thread();
+    tid = nxt_thread_tid(thread);
+
+    if (nxt_fast_path(wq->tid == tid)) {
+        return;
+    }
+
+    if (nxt_slow_path(nxt_pid != wq->pid)) {
+        wq->pid = nxt_pid;
+        wq->tid = tid;
+
+        return;
+    }
+
+    nxt_log_alert(thread->log, "work queue locked by thread %PT", wq->tid);
+    nxt_abort();
+}
+
+#else
+
+#define nxt_work_queue_thread_assert(wq)
+
+#endif
+
+
 void
 nxt_work_queue_cache_create(nxt_work_queue_cache_t *cache, size_t chunk_size)
 {
@@ -109,6 +142,8 @@ nxt_work_queue_add(nxt_work_queue_t *wq, nxt_work_handler_t handler,
 {
     nxt_work_t  *work;
 
+    nxt_work_queue_thread_assert(wq);
+
     for ( ;; ) {
         work = wq->cache->next;
 
@@ -144,6 +179,8 @@ nxt_work_queue_pop(nxt_work_queue_t *wq, nxt_task_t **task, void **obj,
 {
     nxt_work_t  *work;
 
+    nxt_work_queue_thread_assert(wq);
+
     work = wq->head;
 
     wq->head = work->next;