Fix one second delay when exiting

Signed-off-by: Torge Matthies <openglfreak@googlemail.com>

1 files changed, 15 insertions, 1 deletions

diff --git a/linux/shrinker.c b/linux/shrinker.c
index 0b5715b3..dfefad40 100644
--- a/linux/shrinker.c
+++ b/linux/shrinker.c
@@ -131,7 +131,21 @@ void run_shrinkers(gfp_t gfp_mask, bool allocation_failed)
 static int shrinker_thread(void *arg)
 {
 	while (!kthread_should_stop()) {
-		sleep(1);
+		struct timespec to;
+		int v;
+
+		clock_gettime(CLOCK_MONOTONIC, &to);
+		to.tv_sec += 1;
+		__set_current_state(TASK_INTERRUPTIBLE);
+		errno = 0;
+		while ((v = READ_ONCE(current->state)) != TASK_RUNNING &&
+		       errno != ETIMEDOUT)
+			futex(&current->state, FUTEX_WAIT_BITSET|FUTEX_PRIVATE_FLAG,
+			      v, &to, NULL, (uint32_t)~0);
+		if (kthread_should_stop())
+			break;
+		if (v != TASK_RUNNING)
+			__set_current_state(TASK_RUNNING);
 		run_shrinkers(GFP_KERNEL, false);
 	}