1 files changed, 19 insertions, 0 deletions

diff --git a/sys/kern/kern_thread.c b/sys/kern/kern_thread.c
index 99b692b0e352..f16e2f61127f 100644
--- a/sys/kern/kern_thread.c
+++ b/sys/kern/kern_thread.c
@@ -1110,3 +1110,22 @@ thread_single_end(void)
 	mtx_unlock_spin(&sched_lock);
 }
 
+/*
+ * Called before going into an interruptible sleep to see if we have been
+ * interrupted or requested to exit.
+ */
+int
+thread_sleep_check(struct thread *td)
+{
+	struct proc *p;
+
+	p = td->td_proc;
+	mtx_assert(&sched_lock, MA_OWNED);
+	if (p->p_flag & P_SA || p->p_numthreads > 1) {
+		if ((p->p_flag & P_SINGLE_EXIT) && p->p_singlethread != td)
+			return (EINTR);
+		if (td->td_flags & TDF_INTERRUPT)
+			return (td->td_intrval);
+	}
+	return (0);
+}