1/* SPDX-License-Identifier: GPL-2.0 */
2#ifndef _LINUX_KCOV_H
3#define _LINUX_KCOV_H
4
5#include <uapi/linux/kcov.h>
6
7struct task_struct;
8
9#ifdef CONFIG_KCOV
10
11enum kcov_mode {
12 /* Coverage collection is not enabled yet. */
13 KCOV_MODE_DISABLED = 0,
14 /* KCOV was initialized, but tracing mode hasn't been chosen yet. */
15 KCOV_MODE_INIT = 1,
16 /*
17 * Tracing coverage collection mode.
18 * Covered PCs are collected in a per-task buffer.
19 */
20 KCOV_MODE_TRACE_PC = 2,
21 /* Collecting comparison operands mode. */
22 KCOV_MODE_TRACE_CMP = 3,
23};
24
25#define KCOV_IN_CTXSW (1 << 30)
26
27void kcov_task_init(struct task_struct *t);
28void kcov_task_exit(struct task_struct *t);
29
30#define kcov_prepare_switch(t) \
31do { \
32 (t)->kcov_mode |= KCOV_IN_CTXSW; \
33} while (0)
34
35#define kcov_finish_switch(t) \
36do { \
37 (t)->kcov_mode &= ~KCOV_IN_CTXSW; \
38} while (0)
39
40#else
41
42static inline void kcov_task_init(struct task_struct *t) {}
43static inline void kcov_task_exit(struct task_struct *t) {}
44static inline void kcov_prepare_switch(struct task_struct *t) {}
45static inline void kcov_finish_switch(struct task_struct *t) {}
46
47#endif /* CONFIG_KCOV */
48#endif /* _LINUX_KCOV_H */
49