Commit 813847a3 authored by Andrii Nakryiko's avatar Andrii Nakryiko Committed by Daniel Borkmann

libbpf: Streamline bpf_attr and perf_event_attr initialization

Make sure that entire libbpf code base is initializing bpf_attr and
perf_event_attr with memset(0). Also for bpf_attr make sure we
clear and pass to kernel only relevant parts of bpf_attr. bpf_attr is
a huge union of independent sub-command attributes, so there is no need
to clear and pass entire union bpf_attr, which over time grows quite
a lot and for most commands this growth is completely irrelevant.

Few cases where we were relying on compiler initialization of BPF UAPI
structs (like bpf_prog_info, bpf_map_info, etc) with `= {};` were
switched to memset(0) pattern for future-proofing.
Signed-off-by: default avatarAndrii Nakryiko <andrii@kernel.org>
Signed-off-by: default avatarDaniel Borkmann <daniel@iogearbox.net>
Acked-by: default avatarHao Luo <haoluo@google.com>
Link: https://lore.kernel.org/bpf/20220816001929.369487-3-andrii@kernel.org
parent d4e6d684
This diff is collapsed.
...@@ -4287,11 +4287,12 @@ int bpf_map__set_autocreate(struct bpf_map *map, bool autocreate) ...@@ -4287,11 +4287,12 @@ int bpf_map__set_autocreate(struct bpf_map *map, bool autocreate)
int bpf_map__reuse_fd(struct bpf_map *map, int fd) int bpf_map__reuse_fd(struct bpf_map *map, int fd)
{ {
struct bpf_map_info info = {}; struct bpf_map_info info;
__u32 len = sizeof(info), name_len; __u32 len = sizeof(info), name_len;
int new_fd, err; int new_fd, err;
char *new_name; char *new_name;
memset(&info, 0, len);
err = bpf_obj_get_info_by_fd(fd, &info, &len); err = bpf_obj_get_info_by_fd(fd, &info, &len);
if (err && errno == EINVAL) if (err && errno == EINVAL)
err = bpf_get_map_info_from_fdinfo(fd, &info); err = bpf_get_map_info_from_fdinfo(fd, &info);
...@@ -4833,13 +4834,12 @@ bool kernel_supports(const struct bpf_object *obj, enum kern_feature_id feat_id) ...@@ -4833,13 +4834,12 @@ bool kernel_supports(const struct bpf_object *obj, enum kern_feature_id feat_id)
static bool map_is_reuse_compat(const struct bpf_map *map, int map_fd) static bool map_is_reuse_compat(const struct bpf_map *map, int map_fd)
{ {
struct bpf_map_info map_info = {}; struct bpf_map_info map_info;
char msg[STRERR_BUFSIZE]; char msg[STRERR_BUFSIZE];
__u32 map_info_len; __u32 map_info_len = sizeof(map_info);
int err; int err;
map_info_len = sizeof(map_info); memset(&map_info, 0, map_info_len);
err = bpf_obj_get_info_by_fd(map_fd, &map_info, &map_info_len); err = bpf_obj_get_info_by_fd(map_fd, &map_info, &map_info_len);
if (err && errno == EINVAL) if (err && errno == EINVAL)
err = bpf_get_map_info_from_fdinfo(map_fd, &map_info); err = bpf_get_map_info_from_fdinfo(map_fd, &map_info);
...@@ -9007,11 +9007,12 @@ int libbpf_find_vmlinux_btf_id(const char *name, ...@@ -9007,11 +9007,12 @@ int libbpf_find_vmlinux_btf_id(const char *name,
static int libbpf_find_prog_btf_id(const char *name, __u32 attach_prog_fd) static int libbpf_find_prog_btf_id(const char *name, __u32 attach_prog_fd)
{ {
struct bpf_prog_info info = {}; struct bpf_prog_info info;
__u32 info_len = sizeof(info); __u32 info_len = sizeof(info);
struct btf *btf; struct btf *btf;
int err; int err;
memset(&info, 0, info_len);
err = bpf_obj_get_info_by_fd(attach_prog_fd, &info, &info_len); err = bpf_obj_get_info_by_fd(attach_prog_fd, &info, &info_len);
if (err) { if (err) {
pr_warn("failed bpf_obj_get_info_by_fd for FD %d: %d\n", pr_warn("failed bpf_obj_get_info_by_fd for FD %d: %d\n",
...@@ -9839,13 +9840,16 @@ static int determine_uprobe_retprobe_bit(void) ...@@ -9839,13 +9840,16 @@ static int determine_uprobe_retprobe_bit(void)
static int perf_event_open_probe(bool uprobe, bool retprobe, const char *name, static int perf_event_open_probe(bool uprobe, bool retprobe, const char *name,
uint64_t offset, int pid, size_t ref_ctr_off) uint64_t offset, int pid, size_t ref_ctr_off)
{ {
struct perf_event_attr attr = {}; const size_t attr_sz = sizeof(struct perf_event_attr);
struct perf_event_attr attr;
char errmsg[STRERR_BUFSIZE]; char errmsg[STRERR_BUFSIZE];
int type, pfd; int type, pfd;
if (ref_ctr_off >= (1ULL << PERF_UPROBE_REF_CTR_OFFSET_BITS)) if (ref_ctr_off >= (1ULL << PERF_UPROBE_REF_CTR_OFFSET_BITS))
return -EINVAL; return -EINVAL;
memset(&attr, 0, attr_sz);
type = uprobe ? determine_uprobe_perf_type() type = uprobe ? determine_uprobe_perf_type()
: determine_kprobe_perf_type(); : determine_kprobe_perf_type();
if (type < 0) { if (type < 0) {
...@@ -9866,7 +9870,7 @@ static int perf_event_open_probe(bool uprobe, bool retprobe, const char *name, ...@@ -9866,7 +9870,7 @@ static int perf_event_open_probe(bool uprobe, bool retprobe, const char *name,
} }
attr.config |= 1 << bit; attr.config |= 1 << bit;
} }
attr.size = sizeof(attr); attr.size = attr_sz;
attr.type = type; attr.type = type;
attr.config |= (__u64)ref_ctr_off << PERF_UPROBE_REF_CTR_OFFSET_SHIFT; attr.config |= (__u64)ref_ctr_off << PERF_UPROBE_REF_CTR_OFFSET_SHIFT;
attr.config1 = ptr_to_u64(name); /* kprobe_func or uprobe_path */ attr.config1 = ptr_to_u64(name); /* kprobe_func or uprobe_path */
...@@ -9965,7 +9969,8 @@ static int determine_kprobe_perf_type_legacy(const char *probe_name, bool retpro ...@@ -9965,7 +9969,8 @@ static int determine_kprobe_perf_type_legacy(const char *probe_name, bool retpro
static int perf_event_kprobe_open_legacy(const char *probe_name, bool retprobe, static int perf_event_kprobe_open_legacy(const char *probe_name, bool retprobe,
const char *kfunc_name, size_t offset, int pid) const char *kfunc_name, size_t offset, int pid)
{ {
struct perf_event_attr attr = {}; const size_t attr_sz = sizeof(struct perf_event_attr);
struct perf_event_attr attr;
char errmsg[STRERR_BUFSIZE]; char errmsg[STRERR_BUFSIZE];
int type, pfd, err; int type, pfd, err;
...@@ -9984,7 +9989,9 @@ static int perf_event_kprobe_open_legacy(const char *probe_name, bool retprobe, ...@@ -9984,7 +9989,9 @@ static int perf_event_kprobe_open_legacy(const char *probe_name, bool retprobe,
libbpf_strerror_r(err, errmsg, sizeof(errmsg))); libbpf_strerror_r(err, errmsg, sizeof(errmsg)));
goto err_clean_legacy; goto err_clean_legacy;
} }
attr.size = sizeof(attr);
memset(&attr, 0, attr_sz);
attr.size = attr_sz;
attr.config = type; attr.config = type;
attr.type = PERF_TYPE_TRACEPOINT; attr.type = PERF_TYPE_TRACEPOINT;
...@@ -10441,6 +10448,7 @@ static int determine_uprobe_perf_type_legacy(const char *probe_name, bool retpro ...@@ -10441,6 +10448,7 @@ static int determine_uprobe_perf_type_legacy(const char *probe_name, bool retpro
static int perf_event_uprobe_open_legacy(const char *probe_name, bool retprobe, static int perf_event_uprobe_open_legacy(const char *probe_name, bool retprobe,
const char *binary_path, size_t offset, int pid) const char *binary_path, size_t offset, int pid)
{ {
const size_t attr_sz = sizeof(struct perf_event_attr);
struct perf_event_attr attr; struct perf_event_attr attr;
int type, pfd, err; int type, pfd, err;
...@@ -10458,8 +10466,8 @@ static int perf_event_uprobe_open_legacy(const char *probe_name, bool retprobe, ...@@ -10458,8 +10466,8 @@ static int perf_event_uprobe_open_legacy(const char *probe_name, bool retprobe,
goto err_clean_legacy; goto err_clean_legacy;
} }
memset(&attr, 0, sizeof(attr)); memset(&attr, 0, attr_sz);
attr.size = sizeof(attr); attr.size = attr_sz;
attr.config = type; attr.config = type;
attr.type = PERF_TYPE_TRACEPOINT; attr.type = PERF_TYPE_TRACEPOINT;
...@@ -10998,7 +11006,8 @@ static int determine_tracepoint_id(const char *tp_category, ...@@ -10998,7 +11006,8 @@ static int determine_tracepoint_id(const char *tp_category,
static int perf_event_open_tracepoint(const char *tp_category, static int perf_event_open_tracepoint(const char *tp_category,
const char *tp_name) const char *tp_name)
{ {
struct perf_event_attr attr = {}; const size_t attr_sz = sizeof(struct perf_event_attr);
struct perf_event_attr attr;
char errmsg[STRERR_BUFSIZE]; char errmsg[STRERR_BUFSIZE];
int tp_id, pfd, err; int tp_id, pfd, err;
...@@ -11010,8 +11019,9 @@ static int perf_event_open_tracepoint(const char *tp_category, ...@@ -11010,8 +11019,9 @@ static int perf_event_open_tracepoint(const char *tp_category,
return tp_id; return tp_id;
} }
memset(&attr, 0, attr_sz);
attr.type = PERF_TYPE_TRACEPOINT; attr.type = PERF_TYPE_TRACEPOINT;
attr.size = sizeof(attr); attr.size = attr_sz;
attr.config = tp_id; attr.config = tp_id;
pfd = syscall(__NR_perf_event_open, &attr, -1 /* pid */, 0 /* cpu */, pfd = syscall(__NR_perf_event_open, &attr, -1 /* pid */, 0 /* cpu */,
...@@ -11631,12 +11641,15 @@ struct perf_buffer *perf_buffer__new(int map_fd, size_t page_cnt, ...@@ -11631,12 +11641,15 @@ struct perf_buffer *perf_buffer__new(int map_fd, size_t page_cnt,
void *ctx, void *ctx,
const struct perf_buffer_opts *opts) const struct perf_buffer_opts *opts)
{ {
const size_t attr_sz = sizeof(struct perf_event_attr);
struct perf_buffer_params p = {}; struct perf_buffer_params p = {};
struct perf_event_attr attr = {}; struct perf_event_attr attr;
if (!OPTS_VALID(opts, perf_buffer_opts)) if (!OPTS_VALID(opts, perf_buffer_opts))
return libbpf_err_ptr(-EINVAL); return libbpf_err_ptr(-EINVAL);
memset(&attr, 0, attr_sz);
attr.size = attr_sz;
attr.config = PERF_COUNT_SW_BPF_OUTPUT; attr.config = PERF_COUNT_SW_BPF_OUTPUT;
attr.type = PERF_TYPE_SOFTWARE; attr.type = PERF_TYPE_SOFTWARE;
attr.sample_type = PERF_SAMPLE_RAW; attr.sample_type = PERF_SAMPLE_RAW;
......
...@@ -587,11 +587,12 @@ static int get_tc_info(struct nlmsghdr *nh, libbpf_dump_nlmsg_t fn, ...@@ -587,11 +587,12 @@ static int get_tc_info(struct nlmsghdr *nh, libbpf_dump_nlmsg_t fn,
static int tc_add_fd_and_name(struct libbpf_nla_req *req, int fd) static int tc_add_fd_and_name(struct libbpf_nla_req *req, int fd)
{ {
struct bpf_prog_info info = {}; struct bpf_prog_info info;
__u32 info_len = sizeof(info); __u32 info_len = sizeof(info);
char name[256]; char name[256];
int len, ret; int len, ret;
memset(&info, 0, info_len);
ret = bpf_obj_get_info_by_fd(fd, &info, &info_len); ret = bpf_obj_get_info_by_fd(fd, &info, &info_len);
if (ret < 0) if (ret < 0)
return ret; return ret;
......
...@@ -285,6 +285,8 @@ static inline int skel_link_create(int prog_fd, int target_fd, ...@@ -285,6 +285,8 @@ static inline int skel_link_create(int prog_fd, int target_fd,
static inline int bpf_load_and_run(struct bpf_load_and_run_opts *opts) static inline int bpf_load_and_run(struct bpf_load_and_run_opts *opts)
{ {
const size_t prog_load_attr_sz = offsetofend(union bpf_attr, fd_array);
const size_t test_run_attr_sz = offsetofend(union bpf_attr, test);
int map_fd = -1, prog_fd = -1, key = 0, err; int map_fd = -1, prog_fd = -1, key = 0, err;
union bpf_attr attr; union bpf_attr attr;
...@@ -302,7 +304,7 @@ static inline int bpf_load_and_run(struct bpf_load_and_run_opts *opts) ...@@ -302,7 +304,7 @@ static inline int bpf_load_and_run(struct bpf_load_and_run_opts *opts)
goto out; goto out;
} }
memset(&attr, 0, sizeof(attr)); memset(&attr, 0, prog_load_attr_sz);
attr.prog_type = BPF_PROG_TYPE_SYSCALL; attr.prog_type = BPF_PROG_TYPE_SYSCALL;
attr.insns = (long) opts->insns; attr.insns = (long) opts->insns;
attr.insn_cnt = opts->insns_sz / sizeof(struct bpf_insn); attr.insn_cnt = opts->insns_sz / sizeof(struct bpf_insn);
...@@ -313,18 +315,18 @@ static inline int bpf_load_and_run(struct bpf_load_and_run_opts *opts) ...@@ -313,18 +315,18 @@ static inline int bpf_load_and_run(struct bpf_load_and_run_opts *opts)
attr.log_size = opts->ctx->log_size; attr.log_size = opts->ctx->log_size;
attr.log_buf = opts->ctx->log_buf; attr.log_buf = opts->ctx->log_buf;
attr.prog_flags = BPF_F_SLEEPABLE; attr.prog_flags = BPF_F_SLEEPABLE;
err = prog_fd = skel_sys_bpf(BPF_PROG_LOAD, &attr, sizeof(attr)); err = prog_fd = skel_sys_bpf(BPF_PROG_LOAD, &attr, prog_load_attr_sz);
if (prog_fd < 0) { if (prog_fd < 0) {
opts->errstr = "failed to load loader prog"; opts->errstr = "failed to load loader prog";
set_err; set_err;
goto out; goto out;
} }
memset(&attr, 0, sizeof(attr)); memset(&attr, 0, test_run_attr_sz);
attr.test.prog_fd = prog_fd; attr.test.prog_fd = prog_fd;
attr.test.ctx_in = (long) opts->ctx; attr.test.ctx_in = (long) opts->ctx;
attr.test.ctx_size_in = opts->ctx->sz; attr.test.ctx_size_in = opts->ctx->sz;
err = skel_sys_bpf(BPF_PROG_RUN, &attr, sizeof(attr)); err = skel_sys_bpf(BPF_PROG_RUN, &attr, test_run_attr_sz);
if (err < 0 || (int)attr.test.retval < 0) { if (err < 0 || (int)attr.test.retval < 0) {
opts->errstr = "failed to execute loader prog"; opts->errstr = "failed to execute loader prog";
if (err < 0) { if (err < 0) {
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment