mirror of
https://git.kernel.org/pub/scm/linux/kernel/git/stable/linux.git
synced 2025-01-07 13:43:51 +00:00
kunit: Fix kthread reference
There is a race condition when a kthread finishes after the deadline and
before the call to kthread_stop(), which may lead to use after free.
Cc: Brendan Higgins <brendanhiggins@google.com>
Cc: Shuah Khan <skhan@linuxfoundation.org>
Reviewed-by: Kees Cook <keescook@chromium.org>
Fixes: adf5054570
("kunit: fix UAF when run kfence test case test_gfpzero")
Reviewed-by: David Gow <davidgow@google.com>
Reviewed-by: Rae Moar <rmoar@google.com>
Signed-off-by: Mickaël Salaün <mic@digikod.net>
Link: https://lore.kernel.org/r/20240408074625.65017-3-mic@digikod.net
Signed-off-by: Shuah Khan <skhan@linuxfoundation.org>
This commit is contained in:
parent
cde5e1b4a9
commit
f8aa1b98ce
@ -11,6 +11,7 @@
|
||||
#include <linux/completion.h>
|
||||
#include <linux/kernel.h>
|
||||
#include <linux/kthread.h>
|
||||
#include <linux/sched/task.h>
|
||||
|
||||
#include "try-catch-impl.h"
|
||||
|
||||
@ -65,14 +66,15 @@ void kunit_try_catch_run(struct kunit_try_catch *try_catch, void *context)
|
||||
try_catch->context = context;
|
||||
try_catch->try_completion = &try_completion;
|
||||
try_catch->try_result = 0;
|
||||
task_struct = kthread_run(kunit_generic_run_threadfn_adapter,
|
||||
try_catch,
|
||||
"kunit_try_catch_thread");
|
||||
task_struct = kthread_create(kunit_generic_run_threadfn_adapter,
|
||||
try_catch, "kunit_try_catch_thread");
|
||||
if (IS_ERR(task_struct)) {
|
||||
try_catch->try_result = PTR_ERR(task_struct);
|
||||
try_catch->catch(try_catch->context);
|
||||
return;
|
||||
}
|
||||
get_task_struct(task_struct);
|
||||
wake_up_process(task_struct);
|
||||
|
||||
time_remaining = wait_for_completion_timeout(&try_completion,
|
||||
kunit_test_timeout());
|
||||
@ -82,6 +84,7 @@ void kunit_try_catch_run(struct kunit_try_catch *try_catch, void *context)
|
||||
kthread_stop(task_struct);
|
||||
}
|
||||
|
||||
put_task_struct(task_struct);
|
||||
exit_code = try_catch->try_result;
|
||||
|
||||
if (!exit_code)
|
||||
|
Loading…
Reference in New Issue
Block a user