1 /* SPDX-License-Identifier: GPL-2.0 */
2 #ifndef _LINUX_KCOV_H
3 #define _LINUX_KCOV_H
4
5 #include <linux/sched.h>
6 #include <uapi/linux/kcov.h>
7
8 struct task_struct;
9
10 #ifdef CONFIG_KCOV
11
12 enum kcov_mode {
13 /* Coverage collection is not enabled yet. */
14 KCOV_MODE_DISABLED = 0,
15 /* KCOV was initialized, but tracing mode hasn't been chosen yet. */
16 KCOV_MODE_INIT = 1,
17 /*
18 * Tracing coverage collection mode.
19 * Covered PCs are collected in a per-task buffer.
20 */
21 KCOV_MODE_TRACE_PC = 2,
22 /* Collecting comparison operands mode. */
23 KCOV_MODE_TRACE_CMP = 3,
24 /* The process owns a KCOV remote reference. */
25 KCOV_MODE_REMOTE = 4,
26 };
27
28 #define KCOV_IN_CTXSW (1 << 30)
29
30 void kcov_task_init(struct task_struct *t);
31 void kcov_task_exit(struct task_struct *t);
32
33 #define kcov_prepare_switch(t) \
34 do { \
35 (t)->kcov_mode |= KCOV_IN_CTXSW; \
36 } while (0)
37
38 #define kcov_finish_switch(t) \
39 do { \
40 (t)->kcov_mode &= ~KCOV_IN_CTXSW; \
41 } while (0)
42
43 /* See Documentation/dev-tools/kcov.rst for usage details. */
44 void kcov_remote_start(u64 handle);
45 void kcov_remote_stop(void);
46 u64 kcov_common_handle(void);
47
kcov_remote_start_common(u64 id)48 static inline void kcov_remote_start_common(u64 id)
49 {
50 kcov_remote_start(kcov_remote_handle(KCOV_SUBSYSTEM_COMMON, id));
51 }
52
kcov_remote_start_usb(u64 id)53 static inline void kcov_remote_start_usb(u64 id)
54 {
55 kcov_remote_start(kcov_remote_handle(KCOV_SUBSYSTEM_USB, id));
56 }
57
58 /*
59 * The softirq flavor of kcov_remote_*() functions is introduced as a temporary
60 * workaround for KCOV's lack of nested remote coverage sections support.
61 *
62 * Adding support is tracked in https://bugzilla.kernel.org/show_bug.cgi?id=210337.
63 *
64 * kcov_remote_start_usb_softirq():
65 *
66 * 1. Only collects coverage when called in the softirq context. This allows
67 * avoiding nested remote coverage collection sections in the task context.
68 * For example, USB/IP calls usb_hcd_giveback_urb() in the task context
69 * within an existing remote coverage collection section. Thus, KCOV should
70 * not attempt to start collecting coverage within the coverage collection
71 * section in __usb_hcd_giveback_urb() in this case.
72 *
73 * 2. Disables interrupts for the duration of the coverage collection section.
74 * This allows avoiding nested remote coverage collection sections in the
75 * softirq context (a softirq might occur during the execution of a work in
76 * the BH workqueue, which runs with in_serving_softirq() > 0).
77 * For example, usb_giveback_urb_bh() runs in the BH workqueue with
78 * interrupts enabled, so __usb_hcd_giveback_urb() might be interrupted in
79 * the middle of its remote coverage collection section, and the interrupt
80 * handler might invoke __usb_hcd_giveback_urb() again.
81 */
82
kcov_remote_start_usb_softirq(u64 id)83 static inline unsigned long kcov_remote_start_usb_softirq(u64 id)
84 {
85 unsigned long flags = 0;
86
87 if (in_serving_softirq()) {
88 local_irq_save(flags);
89 kcov_remote_start_usb(id);
90 }
91
92 return flags;
93 }
94
kcov_remote_stop_softirq(unsigned long flags)95 static inline void kcov_remote_stop_softirq(unsigned long flags)
96 {
97 if (in_serving_softirq()) {
98 kcov_remote_stop();
99 local_irq_restore(flags);
100 }
101 }
102
103 #ifdef CONFIG_64BIT
104 typedef unsigned long kcov_u64;
105 #else
106 typedef unsigned long long kcov_u64;
107 #endif
108
109 void __sanitizer_cov_trace_pc(void);
110 void __sanitizer_cov_trace_cmp1(u8 arg1, u8 arg2);
111 void __sanitizer_cov_trace_cmp2(u16 arg1, u16 arg2);
112 void __sanitizer_cov_trace_cmp4(u32 arg1, u32 arg2);
113 void __sanitizer_cov_trace_cmp8(kcov_u64 arg1, kcov_u64 arg2);
114 void __sanitizer_cov_trace_const_cmp1(u8 arg1, u8 arg2);
115 void __sanitizer_cov_trace_const_cmp2(u16 arg1, u16 arg2);
116 void __sanitizer_cov_trace_const_cmp4(u32 arg1, u32 arg2);
117 void __sanitizer_cov_trace_const_cmp8(kcov_u64 arg1, kcov_u64 arg2);
118 void __sanitizer_cov_trace_switch(kcov_u64 val, void *cases);
119
120 #else
121
kcov_task_init(struct task_struct * t)122 static inline void kcov_task_init(struct task_struct *t) {}
kcov_task_exit(struct task_struct * t)123 static inline void kcov_task_exit(struct task_struct *t) {}
kcov_prepare_switch(struct task_struct * t)124 static inline void kcov_prepare_switch(struct task_struct *t) {}
kcov_finish_switch(struct task_struct * t)125 static inline void kcov_finish_switch(struct task_struct *t) {}
kcov_remote_start(u64 handle)126 static inline void kcov_remote_start(u64 handle) {}
kcov_remote_stop(void)127 static inline void kcov_remote_stop(void) {}
kcov_common_handle(void)128 static inline u64 kcov_common_handle(void)
129 {
130 return 0;
131 }
kcov_remote_start_common(u64 id)132 static inline void kcov_remote_start_common(u64 id) {}
kcov_remote_start_usb(u64 id)133 static inline void kcov_remote_start_usb(u64 id) {}
kcov_remote_start_usb_softirq(u64 id)134 static inline unsigned long kcov_remote_start_usb_softirq(u64 id)
135 {
136 return 0;
137 }
kcov_remote_stop_softirq(unsigned long flags)138 static inline void kcov_remote_stop_softirq(unsigned long flags) {}
139
140 #endif /* CONFIG_KCOV */
141 #endif /* _LINUX_KCOV_H */
142