core: fix context unregister bug

2 files changed, 8 insertions, 1 deletions

diff --git a/accel-pppd/log.c b/accel-pppd/log.c
index e1dd6d0..43e98a9 100644
--- a/accel-pppd/log.c
+++ b/accel-pppd/log.c
@@ -163,6 +163,8 @@ void __export log_debug(const char *fmt,...)
 void __export log_debug2(const char *fmt,...)
 {
 	va_list ap;
+	if (!debug_file)
+		return;
 	va_start(ap, fmt);
 	vfprintf(debug_file, fmt, ap);
 	va_end(ap);
diff --git a/accel-pppd/triton/triton.c b/accel-pppd/triton/triton.c
index 66c6cef..d806231 100644
--- a/accel-pppd/triton/triton.c
+++ b/accel-pppd/triton/triton.c
@@ -168,11 +168,11 @@ cont:
 			goto cont;
 		}
 		thread->ctx->thread = NULL;
+
 		spin_unlock(&thread->ctx->lock);
 
 		if (thread->ctx->need_free) {
 			log_debug2("- context %p removed\n", thread->ctx);
-			thread->ctx->ud->tpd = NULL;
 			mempool_free(thread->ctx);
 		}
 
@@ -349,20 +349,25 @@ void __export triton_context_unregister(struct triton_context_t *ud)
 		}
 		abort();
 	}
+
 	if (!list_empty(&ctx->pending_handlers)) {
 		triton_log_error("BUG:ctx:triton_unregister_ctx: pending_handlers is not empty");
 		abort();
 	}
+
 	if (!list_empty(&ctx->timers)) {
 		triton_log_error("BUG:ctx:triton_unregister_ctx: timers is not empty");
 		abort();
 	}
+
 	if (!list_empty(&ctx->pending_timers)) {
 		triton_log_error("BUG:ctx:triton_unregister_ctx: pending_timers is not empty");
 		abort();
 	}
 
 	ctx->need_free = 1;
+	ud->tpd = NULL;
+
 	spin_lock(&ctx_list_lock);
 	list_del(&ctx->entry);
 	if (__sync_sub_and_fetch(&triton_stat.context_count, 1) == 1) {