mirror of
https://mirrors.bfsu.edu.cn/git/linux.git
synced 2024-12-16 23:45:31 +08:00
df3cb7ac0e
The header uses `NULL` in both `CONFIG_KUNIT=y` and `=n` cases, but does not include it explicitly. When `CONFIG_KUNIT=y`, the header is already getting included via the other headers, so it is not a problem for users. However, when `CONFIG_KUNIT=n`, it is not, and thus a user could hit a build error when including `kunit/test-bug.h`, like we are doing later in this series [1]. Thus include `linux/stddef.h`, and do so outside the `#if`, since it is used in both cases. Reported-by: Boqun Feng <boqun.feng@gmail.com> Closes: https://lore.kernel.org/rust-for-linux/ZJ8cNUW3oR2p+gL1@boqun-archlinux/ [1] Signed-off-by: Miguel Ojeda <ojeda@kernel.org> Reviewed-by: David Gow <davidgow@google.com> Signed-off-by: Shuah Khan <skhan@linuxfoundation.org>
72 lines
2.0 KiB
C
72 lines
2.0 KiB
C
/* SPDX-License-Identifier: GPL-2.0 */
|
|
/*
|
|
* KUnit API providing hooks for non-test code to interact with tests.
|
|
*
|
|
* Copyright (C) 2020, Google LLC.
|
|
* Author: Uriel Guajardo <urielguajardo@google.com>
|
|
*/
|
|
|
|
#ifndef _KUNIT_TEST_BUG_H
|
|
#define _KUNIT_TEST_BUG_H
|
|
|
|
#include <linux/stddef.h> /* for NULL */
|
|
|
|
#if IS_ENABLED(CONFIG_KUNIT)
|
|
|
|
#include <linux/jump_label.h> /* For static branch */
|
|
#include <linux/sched.h>
|
|
|
|
/* Static key if KUnit is running any tests. */
|
|
DECLARE_STATIC_KEY_FALSE(kunit_running);
|
|
|
|
/* Hooks table: a table of function pointers filled in when kunit loads */
|
|
extern struct kunit_hooks_table {
|
|
__printf(3, 4) void (*fail_current_test)(const char*, int, const char*, ...);
|
|
void *(*get_static_stub_address)(struct kunit *test, void *real_fn_addr);
|
|
} kunit_hooks;
|
|
|
|
/**
|
|
* kunit_get_current_test() - Return a pointer to the currently running
|
|
* KUnit test.
|
|
*
|
|
* If a KUnit test is running in the current task, returns a pointer to its
|
|
* associated struct kunit. This pointer can then be passed to any KUnit
|
|
* function or assertion. If no test is running (or a test is running in a
|
|
* different task), returns NULL.
|
|
*
|
|
* This function is safe to call even when KUnit is disabled. If CONFIG_KUNIT
|
|
* is not enabled, it will compile down to nothing and will return quickly no
|
|
* test is running.
|
|
*/
|
|
static inline struct kunit *kunit_get_current_test(void)
|
|
{
|
|
if (!static_branch_unlikely(&kunit_running))
|
|
return NULL;
|
|
|
|
return current->kunit_test;
|
|
}
|
|
|
|
|
|
/**
|
|
* kunit_fail_current_test() - If a KUnit test is running, fail it.
|
|
*
|
|
* If a KUnit test is running in the current task, mark that test as failed.
|
|
*/
|
|
#define kunit_fail_current_test(fmt, ...) do { \
|
|
if (static_branch_unlikely(&kunit_running)) { \
|
|
/* Guaranteed to be non-NULL when kunit_running true*/ \
|
|
kunit_hooks.fail_current_test(__FILE__, __LINE__, \
|
|
fmt, ##__VA_ARGS__); \
|
|
} \
|
|
} while (0)
|
|
|
|
#else
|
|
|
|
static inline struct kunit *kunit_get_current_test(void) { return NULL; }
|
|
|
|
#define kunit_fail_current_test(fmt, ...) do {} while (0)
|
|
|
|
#endif
|
|
|
|
#endif /* _KUNIT_TEST_BUG_H */
|