51 lines
1.0 KiB
C
51 lines
1.0 KiB
C
/* SPDX-License-Identifier: GPL-2.0 */
|
|
#ifndef _LINUX_BPF_SCHED_H
|
|
#define _LINUX_BPF_SCHED_H
|
|
|
|
#include <linux/bpf.h>
|
|
|
|
#ifdef CONFIG_BPF_SCHED
|
|
|
|
#include <linux/jump_label.h>
|
|
|
|
#define BPF_SCHED_HOOK(RET, DEFAULT, NAME, ...) \
|
|
RET bpf_sched_##NAME(__VA_ARGS__);
|
|
#include <linux/sched_hook_defs.h>
|
|
#undef BPF_SCHED_HOOK
|
|
|
|
int bpf_sched_verify_prog(struct bpf_verifier_log *vlog,
|
|
const struct bpf_prog *prog);
|
|
|
|
DECLARE_STATIC_KEY_FALSE(bpf_sched_enabled_key);
|
|
|
|
static inline bool bpf_sched_enabled(void)
|
|
{
|
|
return static_branch_unlikely(&bpf_sched_enabled_key);
|
|
}
|
|
|
|
static inline void bpf_sched_inc(void)
|
|
{
|
|
static_branch_inc(&bpf_sched_enabled_key);
|
|
}
|
|
|
|
static inline void bpf_sched_dec(void)
|
|
{
|
|
static_branch_dec(&bpf_sched_enabled_key);
|
|
}
|
|
|
|
#else /* !CONFIG_BPF_SCHED */
|
|
|
|
static inline int bpf_sched_verify_prog(struct bpf_verifier_log *vlog,
|
|
const struct bpf_prog *prog)
|
|
{
|
|
return -EOPNOTSUPP;
|
|
}
|
|
|
|
static inline bool bpf_sched_enabled(void)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
#endif /* CONFIG_BPF_SCHED */
|
|
#endif /* _LINUX_BPF_SCHED_H */
|