132 lines
3.0 KiB
C
132 lines
3.0 KiB
C
/* SPDX-License-Identifier: GPL-2.0 */
|
|
#ifndef __PERF_BPF_COUNTER_H
|
|
#define __PERF_BPF_COUNTER_H 1
|
|
|
|
#include <linux/list.h>
|
|
#include <sys/resource.h>
|
|
#include <bpf/bpf.h>
|
|
#include <bpf/btf.h>
|
|
#include <bpf/libbpf.h>
|
|
|
|
struct evsel;
|
|
struct target;
|
|
struct bpf_counter;
|
|
|
|
typedef int (*bpf_counter_evsel_op)(struct evsel *evsel);
|
|
typedef int (*bpf_counter_evsel_target_op)(struct evsel *evsel,
|
|
struct target *target);
|
|
typedef int (*bpf_counter_evsel_install_pe_op)(struct evsel *evsel,
|
|
int cpu_map_idx,
|
|
int fd);
|
|
|
|
struct bpf_counter_ops {
|
|
bpf_counter_evsel_target_op load;
|
|
bpf_counter_evsel_op enable;
|
|
bpf_counter_evsel_op disable;
|
|
bpf_counter_evsel_op read;
|
|
bpf_counter_evsel_op destroy;
|
|
bpf_counter_evsel_install_pe_op install_pe;
|
|
};
|
|
|
|
struct bpf_counter {
|
|
void *skel;
|
|
struct list_head list;
|
|
};
|
|
|
|
#ifdef HAVE_BPF_SKEL
|
|
|
|
int bpf_counter__load(struct evsel *evsel, struct target *target);
|
|
int bpf_counter__enable(struct evsel *evsel);
|
|
int bpf_counter__disable(struct evsel *evsel);
|
|
int bpf_counter__read(struct evsel *evsel);
|
|
void bpf_counter__destroy(struct evsel *evsel);
|
|
int bpf_counter__install_pe(struct evsel *evsel, int cpu_map_idx, int fd);
|
|
|
|
#else /* HAVE_BPF_SKEL */
|
|
|
|
#include <linux/err.h>
|
|
|
|
static inline int bpf_counter__load(struct evsel *evsel __maybe_unused,
|
|
struct target *target __maybe_unused)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
static inline int bpf_counter__enable(struct evsel *evsel __maybe_unused)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
static inline int bpf_counter__disable(struct evsel *evsel __maybe_unused)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
static inline int bpf_counter__read(struct evsel *evsel __maybe_unused)
|
|
{
|
|
return -EAGAIN;
|
|
}
|
|
|
|
static inline void bpf_counter__destroy(struct evsel *evsel __maybe_unused)
|
|
{
|
|
}
|
|
|
|
static inline int bpf_counter__install_pe(struct evsel *evsel __maybe_unused,
|
|
int cpu __maybe_unused,
|
|
int fd __maybe_unused)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
#endif /* HAVE_BPF_SKEL */
|
|
|
|
static inline void set_max_rlimit(void)
|
|
{
|
|
struct rlimit rinf = { RLIM_INFINITY, RLIM_INFINITY };
|
|
|
|
setrlimit(RLIMIT_MEMLOCK, &rinf);
|
|
}
|
|
|
|
static inline __u32 bpf_link_get_id(int fd)
|
|
{
|
|
struct bpf_link_info link_info = { .id = 0, };
|
|
__u32 link_info_len = sizeof(link_info);
|
|
|
|
bpf_obj_get_info_by_fd(fd, &link_info, &link_info_len);
|
|
return link_info.id;
|
|
}
|
|
|
|
static inline __u32 bpf_link_get_prog_id(int fd)
|
|
{
|
|
struct bpf_link_info link_info = { .id = 0, };
|
|
__u32 link_info_len = sizeof(link_info);
|
|
|
|
bpf_obj_get_info_by_fd(fd, &link_info, &link_info_len);
|
|
return link_info.prog_id;
|
|
}
|
|
|
|
static inline __u32 bpf_map_get_id(int fd)
|
|
{
|
|
struct bpf_map_info map_info = { .id = 0, };
|
|
__u32 map_info_len = sizeof(map_info);
|
|
|
|
bpf_obj_get_info_by_fd(fd, &map_info, &map_info_len);
|
|
return map_info.id;
|
|
}
|
|
|
|
/* trigger the leader program on a cpu */
|
|
static inline int bperf_trigger_reading(int prog_fd, int cpu)
|
|
{
|
|
DECLARE_LIBBPF_OPTS(bpf_test_run_opts, opts,
|
|
.ctx_in = NULL,
|
|
.ctx_size_in = 0,
|
|
.flags = BPF_F_TEST_RUN_ON_CPU,
|
|
.cpu = cpu,
|
|
.retval = 0,
|
|
);
|
|
|
|
return bpf_prog_test_run_opts(prog_fd, &opts);
|
|
}
|
|
|
|
#endif /* __PERF_BPF_COUNTER_H */
|