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 8struct task_struct; 9 10#ifdef CONFIG_KCOV 11 12enum 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}; 25 26#define KCOV_IN_CTXSW (1 << 30) 27 28void kcov_task_init(struct task_struct *t); 29void kcov_task_exit(struct task_struct *t); 30 31#define kcov_prepare_switch(t) \ 32do { \ 33 (t)->kcov_mode |= KCOV_IN_CTXSW; \ 34} while (0) 35 36#define kcov_finish_switch(t) \ 37do { \ 38 (t)->kcov_mode &= ~KCOV_IN_CTXSW; \ 39} while (0) 40 41/* See Documentation/dev-tools/kcov.rst for usage details. */ 42void kcov_remote_start(u64 handle); 43void kcov_remote_stop(void); 44u64 kcov_common_handle(void); 45 46static inline void kcov_remote_start_common(u64 id) 47{ 48 kcov_remote_start(kcov_remote_handle(KCOV_SUBSYSTEM_COMMON, id)); 49} 50 51static inline void kcov_remote_start_usb(u64 id) 52{ 53 kcov_remote_start(kcov_remote_handle(KCOV_SUBSYSTEM_USB, id)); 54} 55 56/* 57 * The softirq flavor of kcov_remote_*() functions is introduced as a temporary 58 * work around for kcov's lack of nested remote coverage sections support in 59 * task context. Adding support for nested sections is tracked in: 60 * https://bugzilla.kernel.org/show_bug.cgi?id=210337 61 */ 62 63static inline void kcov_remote_start_usb_softirq(u64 id) 64{ 65 if (in_serving_softirq()) 66 kcov_remote_start_usb(id); 67} 68 69static inline void kcov_remote_stop_softirq(void) 70{ 71 if (in_serving_softirq()) 72 kcov_remote_stop(); 73} 74 75#ifdef CONFIG_64BIT 76typedef unsigned long kcov_u64; 77#else 78typedef unsigned long long kcov_u64; 79#endif 80 81void __sanitizer_cov_trace_pc(void); 82void __sanitizer_cov_trace_cmp1(u8 arg1, u8 arg2); 83void __sanitizer_cov_trace_cmp2(u16 arg1, u16 arg2); 84void __sanitizer_cov_trace_cmp4(u32 arg1, u32 arg2); 85void __sanitizer_cov_trace_cmp8(kcov_u64 arg1, kcov_u64 arg2); 86void __sanitizer_cov_trace_const_cmp1(u8 arg1, u8 arg2); 87void __sanitizer_cov_trace_const_cmp2(u16 arg1, u16 arg2); 88void __sanitizer_cov_trace_const_cmp4(u32 arg1, u32 arg2); 89void __sanitizer_cov_trace_const_cmp8(kcov_u64 arg1, kcov_u64 arg2); 90void __sanitizer_cov_trace_switch(kcov_u64 val, void *cases); 91 92#else 93 94static inline void kcov_task_init(struct task_struct *t) {} 95static inline void kcov_task_exit(struct task_struct *t) {} 96static inline void kcov_prepare_switch(struct task_struct *t) {} 97static inline void kcov_finish_switch(struct task_struct *t) {} 98static inline void kcov_remote_start(u64 handle) {} 99static inline void kcov_remote_stop(void) {} 100static inline u64 kcov_common_handle(void) 101{ 102 return 0; 103} 104static inline void kcov_remote_start_common(u64 id) {} 105static inline void kcov_remote_start_usb(u64 id) {} 106static inline void kcov_remote_start_usb_softirq(u64 id) {} 107static inline void kcov_remote_stop_softirq(void) {} 108 109#endif /* CONFIG_KCOV */ 110#endif /* _LINUX_KCOV_H */ 111