On Thu, Mar 21, 2024 at 11:17:51AM +0100, Oleg Nesterov wrote:
> On 03/21, Jiri Olsa wrote:
> >
> > On Wed, Mar 20, 2024 at 04:28:48PM +0100, Oleg Nesterov wrote:
> >
> > SNIP
> >
> > > SYSCALL_DEFINE0(uretprobe)
> > > {
> > > struct pt_regs *regs = task_pt_regs(current);
> > > unsigned long err, ip, sp, r11_cx_ax[3];
> > >
> > > err = copy_from_user(r11_cx_ax, (void __user*)regs->sp, sizeof(r11_cx_ax));
> > > WARN_ON_ONCE(err);
> > >
> > > // Q1: apart from ax, do we really care?
> > > // expose the "right" values of r11/cx/ax/sp to uprobe_consumer's
> > > regs->r11 = r11_cx_ax[0];
> > > regs->cx = r11_cx_ax[1];
> > > regs->ax = r11_cx_ax[2];
> > > regs->sp += sizeof(r11_cx_ax);
> > > regs->orig_ax = -1;
> > >
> > > ip = regs->ip;
> > > sp = regs->sp;
> > >
> > > uprobe_handle_trampoline(regs);
> > >
> > > // Q2: is it possible? do we care?
> > > // uprobe_consumer has changed sp, we can do nothing,
> > > // just return via iret.
> > > if (regs->sp != sp)
> > > return regs->ax;
> > > regs->sp -= sizeof(r11_cx_ax);
> > >
> > > // Q3: is it possible? do we care?
> > > // for the case uprobe_consumer has changed r11/cx
> > > r11_cx_ax[0] = regs->r11;
> > > r11_cx_ax[1] = regs->cx;
> >
> > I wonder we could add test for this as well, and check we return
> > proper register values in case the consuer changed them, will check
> >
> > >
> > > // comment to explain this hack
> > > r11_cx_ax[2] = regs->ip;
> > > regs->ip = ip;
> >
> > we still need restore regs->ip in case do_syscall_64 decides to do
> > iret for some reason, right?
>
> I don't understand... could you spell?
I was wondering why to restore regs->ip for sysret path, but do_syscall_64
can decide to do iret return (for which we need proper regs->ip) even if we
prepare cx/r11 registers for sysexit
>
> AFAICS everything should work correctly even if do_syscall_64() returns F
> and entry_SYSCALL_64() returns via iret. No?
>
> > overall lgtm, thanks
>
> OK, great, feel free to update this code according to your preferences and
> use it in V2.
will do, thanks
jirka
