Re: [PATCH bpf] tools/bpf: only set obj->skeleton without err

From: Andrii Nakryiko <andrii.nakryiko@gmail.com>
To: Wei Fu <fuweid89@gmail.com>
Cc: Alexei Starovoitov <ast@kernel.org>,
	Daniel Borkmann <daniel@iogearbox.net>,
	Andrii Nakryiko <andrii@kernel.org>, Martin Lau <kafai@fb.com>,
	Song Liu <songliubraving@fb.com>, Yonghong Song <yhs@fb.com>,
	john fastabend <john.fastabend@gmail.com>,
	KP Singh <kpsingh@kernel.org>, bpf <bpf@vger.kernel.org>,
	Networking <netdev@vger.kernel.org>,
	open list <linux-kernel@vger.kernel.org>
Subject: Re: [PATCH bpf] tools/bpf: only set obj->skeleton without err
Date: Sun, 9 Jan 2022 18:04:45 -0800	[thread overview]
Message-ID: <CAEf4Bzag+qQOs86t2ESmYvTY8xCip+_GTKqXa0m7MQWjDMO5Mg@mail.gmail.com> (raw)
In-Reply-To: <20220108084008.1053111-1-fuweid89@gmail.com>

On Sat, Jan 8, 2022 at 12:40 AM Wei Fu <fuweid89@gmail.com> wrote:
>
> After `bpftool gen skeleton`, the ${bpf_app}.skel.h will provide that
> ${bpf_app_name}__open helper to load bpf. If there is some error
> like ENOMEM, the ${bpf_app_name}__open will rollback(free) the allocated
> object, including `bpf_object_skeleton`.
>
> Since the ${bpf_app_name}__create_skeleton set the obj->skeleton first
> and not rollback it when error, it will cause double-free in
> ${bpf_app_name}__destory at ${bpf_app_name}__open. Therefore, we should
> set the obj->skeleton before return 0;
>
> Signed-off-by: Wei Fu <fuweid89@gmail.com>
> ---
>  tools/bpf/bpftool/gen.c | 3 ++-
>  1 file changed, 2 insertions(+), 1 deletion(-)
>

Great catch! Added (please add it yourself in the future):

Fixes: 5dc7a8b21144 ("bpftool, selftests/bpf: Embed object file inside
skeleton")

Also reworded the subject a bit. Pushed to bpf-next.

> diff --git a/tools/bpf/bpftool/gen.c b/tools/bpf/bpftool/gen.c
> index 5c18351290f0..e61e08f524da 100644
> --- a/tools/bpf/bpftool/gen.c
> +++ b/tools/bpf/bpftool/gen.c
> @@ -928,7 +928,6 @@ static int do_skeleton(int argc, char **argv)
>                         s = (struct bpf_object_skeleton *)calloc(1, sizeof(*s));\n\
>                         if (!s)                                             \n\
>                                 goto err;                                   \n\
> -                       obj->skeleton = s;                                  \n\
>                                                                             \n\
>                         s->sz = sizeof(*s);                                 \n\
>                         s->name = \"%1$s\";                                 \n\
> @@ -1001,6 +1000,8 @@ static int do_skeleton(int argc, char **argv)
>                                                                             \n\
>                         s->data = (void *)%2$s__elf_bytes(&s->data_sz);     \n\
>                                                                             \n\
> +                       obj->skeleton = s;                                  \n\
> +                                                                           \n\
>                         return 0;                                           \n\
>                 err:                                                        \n\
>                         bpf_object__destroy_skeleton(s);                    \n\
> --
> 2.25.1
>