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 
7 struct task_struct;
8 
9 #ifdef CONFIG_KCOV
10 
11 enum 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 
27 void kcov_task_init(struct task_struct *t);
28 void kcov_task_exit(struct task_struct *t);
29 
30 #define kcov_prepare_switch(t)			\
31 do {						\
32 	(t)->kcov_mode |= KCOV_IN_CTXSW;	\
33 } while (0)
34 
35 #define kcov_finish_switch(t)			\
36 do {						\
37 	(t)->kcov_mode &= ~KCOV_IN_CTXSW;	\
38 } while (0)
39 
40 /* See Documentation/dev-tools/kcov.rst for usage details. */
41 void kcov_remote_start(u64 handle);
42 void kcov_remote_stop(void);
43 u64 kcov_common_handle(void);
44 
kcov_remote_start_common(u64 id)45 static inline void kcov_remote_start_common(u64 id)
46 {
47 	kcov_remote_start(kcov_remote_handle(KCOV_SUBSYSTEM_COMMON, id));
48 }
49 
kcov_remote_start_usb(u64 id)50 static inline void kcov_remote_start_usb(u64 id)
51 {
52 	kcov_remote_start(kcov_remote_handle(KCOV_SUBSYSTEM_USB, id));
53 }
54 
55 #else
56 
kcov_task_init(struct task_struct * t)57 static inline void kcov_task_init(struct task_struct *t) {}
kcov_task_exit(struct task_struct * t)58 static inline void kcov_task_exit(struct task_struct *t) {}
kcov_prepare_switch(struct task_struct * t)59 static inline void kcov_prepare_switch(struct task_struct *t) {}
kcov_finish_switch(struct task_struct * t)60 static inline void kcov_finish_switch(struct task_struct *t) {}
kcov_remote_start(u64 handle)61 static inline void kcov_remote_start(u64 handle) {}
kcov_remote_stop(void)62 static inline void kcov_remote_stop(void) {}
kcov_common_handle(void)63 static inline u64 kcov_common_handle(void)
64 {
65 	return 0;
66 }
kcov_remote_start_common(u64 id)67 static inline void kcov_remote_start_common(u64 id) {}
kcov_remote_start_usb(u64 id)68 static inline void kcov_remote_start_usb(u64 id) {}
69 
70 #endif /* CONFIG_KCOV */
71 #endif /* _LINUX_KCOV_H */
72