linux/include/linux/kcov.h

/* SPDX-License-Identifier: GPL-2.0 */
#ifndef _LINUX_KCOV_H
#define _LINUX_KCOV_H

#include <linux/sched.h>
#include <uapi/linux/kcov.h>

struct task_struct;

#ifdef CONFIG_KCOV

enum kcov_mode { … };

#define KCOV_IN_CTXSW …

void kcov_task_init(struct task_struct *t);
void kcov_task_exit(struct task_struct *t);

#define kcov_prepare_switch(t) …

#define kcov_finish_switch(t) …

/* See Documentation/dev-tools/kcov.rst for usage details. */
void kcov_remote_start(u64 handle);
void kcov_remote_stop(void);
u64 kcov_common_handle(void);

static inline void kcov_remote_start_common(u64 id)
{ … }

static inline void kcov_remote_start_usb(u64 id)
{ … }

/*
 * The softirq flavor of kcov_remote_*() functions is introduced as a temporary
 * workaround for KCOV's lack of nested remote coverage sections support.
 *
 * Adding support is tracked in https://bugzilla.kernel.org/show_bug.cgi?id=210337.
 *
 * kcov_remote_start_usb_softirq():
 *
 * 1. Only collects coverage when called in the softirq context. This allows
 *    avoiding nested remote coverage collection sections in the task context.
 *    For example, USB/IP calls usb_hcd_giveback_urb() in the task context
 *    within an existing remote coverage collection section. Thus, KCOV should
 *    not attempt to start collecting coverage within the coverage collection
 *    section in __usb_hcd_giveback_urb() in this case.
 *
 * 2. Disables interrupts for the duration of the coverage collection section.
 *    This allows avoiding nested remote coverage collection sections in the
 *    softirq context (a softirq might occur during the execution of a work in
 *    the BH workqueue, which runs with in_serving_softirq() > 0).
 *    For example, usb_giveback_urb_bh() runs in the BH workqueue with
 *    interrupts enabled, so __usb_hcd_giveback_urb() might be interrupted in
 *    the middle of its remote coverage collection section, and the interrupt
 *    handler might invoke __usb_hcd_giveback_urb() again.
 */

static inline unsigned long kcov_remote_start_usb_softirq(u64 id)
{ … }

static inline void kcov_remote_stop_softirq(unsigned long flags)
{ … }

#ifdef CONFIG_64BIT
kcov_u64;
#else
typedef unsigned long long kcov_u64;
#endif

void __sanitizer_cov_trace_pc(void);
void __sanitizer_cov_trace_cmp1(u8 arg1, u8 arg2);
void __sanitizer_cov_trace_cmp2(u16 arg1, u16 arg2);
void __sanitizer_cov_trace_cmp4(u32 arg1, u32 arg2);
void __sanitizer_cov_trace_cmp8(kcov_u64 arg1, kcov_u64 arg2);
void __sanitizer_cov_trace_const_cmp1(u8 arg1, u8 arg2);
void __sanitizer_cov_trace_const_cmp2(u16 arg1, u16 arg2);
void __sanitizer_cov_trace_const_cmp4(u32 arg1, u32 arg2);
void __sanitizer_cov_trace_const_cmp8(kcov_u64 arg1, kcov_u64 arg2);
void __sanitizer_cov_trace_switch(kcov_u64 val, void *cases);

#else

static inline void kcov_task_init(struct task_struct *t) {}
static inline void kcov_task_exit(struct task_struct *t) {}
static inline void kcov_prepare_switch(struct task_struct *t) {}
static inline void kcov_finish_switch(struct task_struct *t) {}
static inline void kcov_remote_start(u64 handle) {}
static inline void kcov_remote_stop(void) {}
static inline u64 kcov_common_handle(void)
{
	return 0;
}
static inline void kcov_remote_start_common(u64 id) {}
static inline void kcov_remote_start_usb(u64 id) {}
static inline unsigned long kcov_remote_start_usb_softirq(u64 id)
{
	return 0;
}
static inline void kcov_remote_stop_softirq(unsigned long flags) {}

#endif /* CONFIG_KCOV */
#endif /* _LINUX_KCOV_H */