On 08/11, Andrii Nakryiko wrote:
>
> On Sun, Aug 11, 2024 at 5:35 AM Oleg Nesterov <[email protected]> wrote:
> >
> > Something like below on top of perf/core. But I don't like the usage of
> > "i" in the +error_unregister path...
> >
>
> Wouldn't the below be cleaner?
>
> diff --git a/kernel/trace/bpf_trace.c b/kernel/trace/bpf_trace.c
> index cd098846e251..3ca65454f888 100644
> --- a/kernel/trace/bpf_trace.c
> +++ b/kernel/trace/bpf_trace.c
> @@ -3491,8 +3491,10 @@ int bpf_uprobe_multi_link_attach(const union
> bpf_attr *attr, struct bpf_prog *pr
>         }
>
>         err = bpf_link_prime(&link->link, &link_primer);
> -       if (err)
> +       if (err) {
> +               bpf_uprobe_unregister(&path, uprobes, cnt);

I disagree. This code already uses the "goto error_xxx" pattern, why
duplicate bpf_uprobe_unregister() ? What if another "can fail" code
comes between register and bpf_link_prime() ?

See the patch below, on top of perf/core.

> We should probably route this through the bpf tree, I don't think it
> will conflict with your changes, right?

It will conflict, and in this sense it is even worse than the "#syz test"
patch I sent in https://lore.kernel.org/all/[email protected]/

Because with your version above the necessary change

        -       bpf_uprobe_unregister(&path, uprobes, cnt);
        +       bpf_uprobe_unregister(uprobes, cnt);

won't be noticed during the merge, I guess.

So can we route this fix through the perf/core ? I'll add "cc: stable",
in the next merge window the Greg's scripts will report the "FAILED"
status of the -stable patch, I'll send the trivial backport in reply.

No?

Oleg.
---

diff --git a/kernel/trace/bpf_trace.c b/kernel/trace/bpf_trace.c
index 4e391daafa64..90cd30e9723e 100644
--- a/kernel/trace/bpf_trace.c
+++ b/kernel/trace/bpf_trace.c
@@ -3484,17 +3484,20 @@ int bpf_uprobe_multi_link_attach(const union bpf_attr 
*attr, struct bpf_prog *pr
                                                    &uprobes[i].consumer);
                if (IS_ERR(uprobes[i].uprobe)) {
                        err = PTR_ERR(uprobes[i].uprobe);
-                       bpf_uprobe_unregister(uprobes, i);
-                       goto error_free;
+                       link->cnt = i;
+                       goto error_unregister;
                }
        }
 
        err = bpf_link_prime(&link->link, &link_primer);
        if (err)
-               goto error_free;
+               goto error_unregister;
 
        return bpf_link_settle(&link_primer);
 
+error_unregister:
+       bpf_uprobe_unregister(uprobes, link->cnt);
+
 error_free:
        kvfree(uprobes);
        kfree(link);


Reply via email to