Skip to content

Commit 31c11f1

Browse files
Delyan KratunovNobody
authored andcommitted
bpftool: explicit errno handling in skeletons
Andrii noticed that since f97b8b9 (bpftool: Fix a bug in subskeleton code generation) the subskeleton code allows bpf_object__destroy_subskeleton to overwrite the errno that subskeleton__open would return with. While this is not currently an issue, let's make it future-proof. This patch explicitly tracks err in subskeleton__open and skeleton__create (i.e. calloc failure is explicitly ENOMEM) and ensures that errno is -err on the error return path. The skeleton code had to be changed since maps and progs codegen is shared with subskeletons. Signed-off-by: Delyan Kratunov <[email protected]>
1 parent 8009508 commit 31c11f1

File tree

1 file changed

+15
-7
lines changed

1 file changed

+15
-7
lines changed

tools/bpf/bpftool/gen.c

Lines changed: 15 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -828,8 +828,10 @@ codegen_maps_skeleton(struct bpf_object *obj, size_t map_cnt, bool mmaped)
828828
s->map_cnt = %zu; \n\
829829
s->map_skel_sz = sizeof(*s->maps); \n\
830830
s->maps = (struct bpf_map_skeleton *)calloc(s->map_cnt, s->map_skel_sz);\n\
831-
if (!s->maps) \n\
831+
if (!s->maps) { \n\
832+
err = -ENOMEM; \n\
832833
goto err; \n\
834+
} \n\
833835
",
834836
map_cnt
835837
);
@@ -870,8 +872,10 @@ codegen_progs_skeleton(struct bpf_object *obj, size_t prog_cnt, bool populate_li
870872
s->prog_cnt = %zu; \n\
871873
s->prog_skel_sz = sizeof(*s->progs); \n\
872874
s->progs = (struct bpf_prog_skeleton *)calloc(s->prog_cnt, s->prog_skel_sz);\n\
873-
if (!s->progs) \n\
875+
if (!s->progs) { \n\
876+
err = -ENOMEM; \n\
874877
goto err; \n\
878+
} \n\
875879
",
876880
prog_cnt
877881
);
@@ -1182,10 +1186,13 @@ static int do_skeleton(int argc, char **argv)
11821186
%1$s__create_skeleton(struct %1$s *obj) \n\
11831187
{ \n\
11841188
struct bpf_object_skeleton *s; \n\
1189+
int err; \n\
11851190
\n\
11861191
s = (struct bpf_object_skeleton *)calloc(1, sizeof(*s));\n\
1187-
if (!s) \n\
1192+
if (!s) { \n\
1193+
err = -ENOMEM; \n\
11881194
goto err; \n\
1195+
} \n\
11891196
\n\
11901197
s->sz = sizeof(*s); \n\
11911198
s->name = \"%1$s\"; \n\
@@ -1206,7 +1213,7 @@ static int do_skeleton(int argc, char **argv)
12061213
return 0; \n\
12071214
err: \n\
12081215
bpf_object__destroy_skeleton(s); \n\
1209-
return -ENOMEM; \n\
1216+
return err; \n\
12101217
} \n\
12111218
\n\
12121219
static inline const void *%2$s__elf_bytes(size_t *sz) \n\
@@ -1466,12 +1473,12 @@ static int do_subskeleton(int argc, char **argv)
14661473
\n\
14671474
obj = (struct %1$s *)calloc(1, sizeof(*obj)); \n\
14681475
if (!obj) { \n\
1469-
errno = ENOMEM; \n\
1476+
err = -ENOMEM; \n\
14701477
goto err; \n\
14711478
} \n\
14721479
s = (struct bpf_object_subskeleton *)calloc(1, sizeof(*s));\n\
14731480
if (!s) { \n\
1474-
errno = ENOMEM; \n\
1481+
err = -ENOMEM; \n\
14751482
goto err; \n\
14761483
} \n\
14771484
s->sz = sizeof(*s); \n\
@@ -1483,7 +1490,7 @@ static int do_subskeleton(int argc, char **argv)
14831490
s->var_cnt = %2$d; \n\
14841491
s->vars = (struct bpf_var_skeleton *)calloc(%2$d, sizeof(*s->vars));\n\
14851492
if (!s->vars) { \n\
1486-
errno = ENOMEM; \n\
1493+
err = -ENOMEM; \n\
14871494
goto err; \n\
14881495
} \n\
14891496
",
@@ -1538,6 +1545,7 @@ static int do_subskeleton(int argc, char **argv)
15381545
return obj; \n\
15391546
err: \n\
15401547
%1$s__destroy(obj); \n\
1548+
errno = -err; \n\
15411549
return NULL; \n\
15421550
} \n\
15431551
\n\

0 commit comments

Comments
 (0)