Re: [PATCHv4 02/14] uprobe: Add support for session consumer

From: Oleg Nesterov
Date: Tue Sep 17 2024 - 08:03:55 EST


I don't see anything wrong after a quick glance, but I don't
really understand the UPROBE_HANDLER_IGNORE logic, see below.

On 09/17, Jiri Olsa wrote:
>
> + * UPROBE_HANDLER_IWANTMYCOOKIE
> + * - Store cookie and pass it to ret_handler (if defined).

Cough ;) yes it was me who used this name in the previous discussion, but maybe

UPROBE_HANDLER_COOKIE

will look a bit better? Feel free to ignore.

> static void handler_chain(struct uprobe *uprobe, struct pt_regs *regs)
...
> + if (!uc->ret_handler || rc == UPROBE_HANDLER_REMOVE)
> + continue;
> +
> + /*
> + * If alloc_return_instance and push_consumer fail, the return probe
> + * won't be prepared, but we'll finish to execute all entry handlers.
> + *
> + * We need to store handler's return value in case the return uprobe
> + * gets installed and contains consumers that need to be ignored.
> + */
> + if (!ri)
> + ri = alloc_return_instance();
> +
> + if (rc == UPROBE_HANDLER_IWANTMYCOOKIE || rc == UPROBE_HANDLER_IGNORE)
> + ri = push_consumer(ri, push_idx++, uc->id, cookie, rc);

So this code allocates ri (which implies prepare_uretprobe!) and calls push_consumer()
even if rc == UPROBE_HANDLER_IGNORE.

Why? The comment in uprobes.h says:

UPROBE_HANDLER_IGNORE
- Ignore ret_handler callback for this consumer

but the ret_handler callback won't be ignored?

To me this code should do:

if (!uc->ret_handler || UPROBE_HANDLER_REMOVE || UPROBE_HANDLER_IGNORE)
continue;

if (!ri)
ri = alloc_return_instance();

if (rc == UPROBE_HANDLER_IWANTMYCOOKIE)
ri = push_consumer(...);

And,

> handle_uretprobe_chain(struct return_instance *ri, struct pt_regs *regs)
...
> list_for_each_entry_srcu(uc, &uprobe->consumers, cons_node,
> srcu_read_lock_held(&uprobes_srcu)) {
> + ric = return_consumer_find(ri, &ric_idx, uc->id);
> + if (ric && ric->rc == UPROBE_HANDLER_IGNORE)
> + continue;
> if (uc->ret_handler)
> - uc->ret_handler(uc, ri->func, regs);
> + uc->ret_handler(uc, ri->func, regs, ric ? &ric->cookie : NULL);
> }

the UPROBE_HANDLER_IGNORE check above and the new ric->rc member should die,

if (!uc->ret_handler)
continue;

ric = return_consumer_find(...);
uc->ret_handler(..., ric ? &ric->cookie : NULL);

as we have already discussed, the session ret_handler(data) can simply do

// my ->handler() wasn't called or it didn't return
// UPROBE_HANDLER_IWANTMYCOOKIE
if (!data)
return;

at the start.

Could you explain why this can't work?

Oleg.