1/* SPDX-License-Identifier: GPL-2.0 */
  2#ifndef __PERF_BPF_COUNTER_H
  3#define __PERF_BPF_COUNTER_H 1
  4
  5#include <linux/list.h>
  6#include <sys/resource.h>
  7
  8#ifdef HAVE_LIBBPF_SUPPORT
  9#include <bpf/bpf.h>
 10#include <bpf/btf.h>
 11#include <bpf/libbpf.h>
 12#endif
 13
 14struct evsel;
 15struct target;
 16struct bpf_counter;
 17
 18typedef int (*bpf_counter_evsel_op)(struct evsel *evsel);
 19typedef int (*bpf_counter_evsel_target_op)(struct evsel *evsel,
 20					   struct target *target);
 21typedef int (*bpf_counter_evsel_install_pe_op)(struct evsel *evsel,
 22					       int cpu_map_idx,
 23					       int fd);
 24
 25struct bpf_counter_ops {
 26	bpf_counter_evsel_target_op load;
 27	bpf_counter_evsel_op enable;
 28	bpf_counter_evsel_op disable;
 29	bpf_counter_evsel_op read;
 30	bpf_counter_evsel_op destroy;
 31	bpf_counter_evsel_install_pe_op install_pe;
 32};
 33
 34struct bpf_counter {
 35	void *skel;
 36	struct list_head list;
 37};
 38
 39#ifdef HAVE_BPF_SKEL
 40
 41int bpf_counter__load(struct evsel *evsel, struct target *target);
 42int bpf_counter__enable(struct evsel *evsel);
 43int bpf_counter__disable(struct evsel *evsel);
 44int bpf_counter__read(struct evsel *evsel);
 45void bpf_counter__destroy(struct evsel *evsel);
 46int bpf_counter__install_pe(struct evsel *evsel, int cpu_map_idx, int fd);
 47
 48#else /* HAVE_BPF_SKEL */
 49
 50#include <linux/err.h>
 51
 52static inline int bpf_counter__load(struct evsel *evsel __maybe_unused,
 53				    struct target *target __maybe_unused)
 54{
 55	return 0;
 56}
 57
 58static inline int bpf_counter__enable(struct evsel *evsel __maybe_unused)
 59{
 60	return 0;
 61}
 62
 63static inline int bpf_counter__disable(struct evsel *evsel __maybe_unused)
 64{
 65	return 0;
 66}
 67
 68static inline int bpf_counter__read(struct evsel *evsel __maybe_unused)
 69{
 70	return -EAGAIN;
 71}
 72
 73static inline void bpf_counter__destroy(struct evsel *evsel __maybe_unused)
 74{
 75}
 76
 77static inline int bpf_counter__install_pe(struct evsel *evsel __maybe_unused,
 78					  int cpu __maybe_unused,
 79					  int fd __maybe_unused)
 80{
 81	return 0;
 82}
 83
 84#endif /* HAVE_BPF_SKEL */
 85
 86static inline void set_max_rlimit(void)
 87{
 88	struct rlimit rinf = { RLIM_INFINITY, RLIM_INFINITY };
 89
 90	setrlimit(RLIMIT_MEMLOCK, &rinf);
 91}
 92
 93#ifdef HAVE_BPF_SKEL
 94
 95static inline __u32 bpf_link_get_id(int fd)
 96{
 97	struct bpf_link_info link_info = { .id = 0, };
 98	__u32 link_info_len = sizeof(link_info);
 99
100	bpf_obj_get_info_by_fd(fd, &link_info, &link_info_len);
101	return link_info.id;
102}
103
104static inline __u32 bpf_link_get_prog_id(int fd)
105{
106	struct bpf_link_info link_info = { .id = 0, };
107	__u32 link_info_len = sizeof(link_info);
108
109	bpf_obj_get_info_by_fd(fd, &link_info, &link_info_len);
110	return link_info.prog_id;
111}
112
113static inline __u32 bpf_map_get_id(int fd)
114{
115	struct bpf_map_info map_info = { .id = 0, };
116	__u32 map_info_len = sizeof(map_info);
117
118	bpf_obj_get_info_by_fd(fd, &map_info, &map_info_len);
119	return map_info.id;
120}
121
122/* trigger the leader program on a cpu */
123static inline int bperf_trigger_reading(int prog_fd, int cpu)
124{
125	DECLARE_LIBBPF_OPTS(bpf_test_run_opts, opts,
126			    .ctx_in = NULL,
127			    .ctx_size_in = 0,
128			    .flags = BPF_F_TEST_RUN_ON_CPU,
129			    .cpu = cpu,
130			    .retval = 0,
131		);
132
133	return bpf_prog_test_run_opts(prog_fd, &opts);
134}
135#endif /* HAVE_BPF_SKEL */
136
137#endif /* __PERF_BPF_COUNTER_H */