mirror of
https://github.com/torvalds/linux.git
synced 2024-10-25 14:40:58 +00:00
82c450803a
This implements the retrieval and printing, as well the help output. Signed-off-by: Kumar Kartikeya Dwivedi <memxor@gmail.com> Signed-off-by: Alexei Starovoitov <ast@kernel.org> Link: https://lore.kernel.org/bpf/20210821002010.845777-8-memxor@gmail.com
85 lines
3.1 KiB
C
85 lines
3.1 KiB
C
// SPDX-License-Identifier: GPL-2.0-only
|
|
#ifndef XDP_SAMPLE_USER_H
|
|
#define XDP_SAMPLE_USER_H
|
|
|
|
#include <bpf/libbpf.h>
|
|
#include <linux/compiler.h>
|
|
|
|
#include "xdp_sample_shared.h"
|
|
|
|
enum stats_mask {
|
|
_SAMPLE_REDIRECT_MAP = 1U << 0,
|
|
SAMPLE_RX_CNT = 1U << 1,
|
|
SAMPLE_REDIRECT_ERR_CNT = 1U << 2,
|
|
SAMPLE_EXCEPTION_CNT = 1U << 5,
|
|
SAMPLE_REDIRECT_CNT = 1U << 7,
|
|
SAMPLE_REDIRECT_MAP_CNT = SAMPLE_REDIRECT_CNT | _SAMPLE_REDIRECT_MAP,
|
|
SAMPLE_REDIRECT_ERR_MAP_CNT = SAMPLE_REDIRECT_ERR_CNT | _SAMPLE_REDIRECT_MAP,
|
|
};
|
|
|
|
/* Exit return codes */
|
|
#define EXIT_OK 0
|
|
#define EXIT_FAIL 1
|
|
#define EXIT_FAIL_OPTION 2
|
|
#define EXIT_FAIL_XDP 3
|
|
#define EXIT_FAIL_BPF 4
|
|
#define EXIT_FAIL_MEM 5
|
|
|
|
int sample_setup_maps(struct bpf_map **maps);
|
|
int __sample_init(int mask);
|
|
void sample_exit(int status);
|
|
int sample_run(int interval, void (*post_cb)(void *), void *ctx);
|
|
|
|
void sample_switch_mode(void);
|
|
int sample_install_xdp(struct bpf_program *xdp_prog, int ifindex, bool generic,
|
|
bool force);
|
|
void sample_usage(char *argv[], const struct option *long_options,
|
|
const char *doc, int mask, bool error);
|
|
|
|
const char *get_driver_name(int ifindex);
|
|
int get_mac_addr(int ifindex, void *mac_addr);
|
|
|
|
#pragma GCC diagnostic push
|
|
#pragma GCC diagnostic ignored "-Wstringop-truncation"
|
|
__attribute__((unused))
|
|
static inline char *safe_strncpy(char *dst, const char *src, size_t size)
|
|
{
|
|
if (!size)
|
|
return dst;
|
|
strncpy(dst, src, size - 1);
|
|
dst[size - 1] = '\0';
|
|
return dst;
|
|
}
|
|
#pragma GCC diagnostic pop
|
|
|
|
#define __attach_tp(name) \
|
|
({ \
|
|
if (!bpf_program__is_tracing(skel->progs.name)) \
|
|
return -EINVAL; \
|
|
skel->links.name = bpf_program__attach(skel->progs.name); \
|
|
if (!skel->links.name) \
|
|
return -errno; \
|
|
})
|
|
|
|
#define DEFINE_SAMPLE_INIT(name) \
|
|
static int sample_init(struct name *skel, int mask) \
|
|
{ \
|
|
int ret; \
|
|
ret = __sample_init(mask); \
|
|
if (ret < 0) \
|
|
return ret; \
|
|
if (mask & SAMPLE_REDIRECT_MAP_CNT) \
|
|
__attach_tp(tp_xdp_redirect_map); \
|
|
if (mask & SAMPLE_REDIRECT_CNT) \
|
|
__attach_tp(tp_xdp_redirect); \
|
|
if (mask & SAMPLE_REDIRECT_ERR_MAP_CNT) \
|
|
__attach_tp(tp_xdp_redirect_map_err); \
|
|
if (mask & SAMPLE_REDIRECT_ERR_CNT) \
|
|
__attach_tp(tp_xdp_redirect_err); \
|
|
if (mask & SAMPLE_EXCEPTION_CNT) \
|
|
__attach_tp(tp_xdp_exception); \
|
|
return 0; \
|
|
}
|
|
|
|
#endif
|