[PATCHv4 00/14] uprobe, bpf: Add session support

Jiri Olsa
2024-09-17T08:50:32Z

hi,
this patchset is adding support for session uprobe attachment and
using it through bpf link for bpf programs.

The session means that the uprobe consumer is executed on entry
and return of probed function with additional control:
- entry callback can control execution of the return callback
- entry and return callbacks can share data/cookie

On more details please see patch #2.

It's based on Peter's perf/core:
git://git.kernel.org/pub/scm/linux/kernel/git/peterz/queue.git perf/core

v4 changes:
- rework the uprobe consumer allocation based on Oleg's suggestions [Oleg]
- moved docs about handler return values to uprobes.h header [Oleg]
- use libbpf's attach_uprobe_multi for session attach [Andrii]
- make verifier to check session return values [Andrii]
- various small review changes
- added more tests
- added acks

patch #6 is already in bpf-next tree, but we need that as dependency

thanks,
jirka

---
Jiri Olsa (14):
uprobe: Add data pointer to consumer handlers
uprobe: Add support for session consumer
bpf: Add support for uprobe multi session attach
bpf: Add support for uprobe multi session context
bpf: Allow return values 0 and 1 for uprobe/kprobe session
libbpf: Fix uretprobe.multi.s programs auto attachment
libbpf: Add support for uprobe multi session attach
selftests/bpf: Add uprobe session test
selftests/bpf: Add uprobe session cookie test
selftests/bpf: Add uprobe session recursive test
selftests/bpf: Add uprobe session verifier test for return value
selftests/bpf: Add kprobe session verifier test for return value
selftests/bpf: Add uprobe session single consumer test
selftests/bpf: Add consumers stress test on single uprobe

include/linux/uprobes.h | 29 +++++++++-
include/uapi/linux/bpf.h | 1 +
kernel/bpf/syscall.c | 9 ++-
kernel/bpf/verifier.c | 10 ++++
kernel/events/uprobes.c | 150 ++++++++++++++++++++++++++++++++++++++++----------
kernel/trace/bpf_trace.c | 66 ++++++++++++++++------
kernel/trace/trace_uprobe.c | 12 ++--
tools/include/uapi/linux/bpf.h | 1 +
tools/lib/bpf/bpf.c | 1 +
tools/lib/bpf/libbpf.c | 22 +++++++-
tools/lib/bpf/libbpf.h | 4 +-
tools/testing/selftests/bpf/bpf_testmod/bpf_testmod.c | 2 +-
tools/testing/selftests/bpf/prog_tests/kprobe_multi_test.c | 2 +
tools/testing/selftests/bpf/prog_tests/uprobe_multi_test.c | 252 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
tools/testing/selftests/bpf/progs/kprobe_multi_verifier.c | 31 +++++++++++
tools/testing/selftests/bpf/progs/uprobe_multi_consumer_stress.c | 29 ++++++++++
tools/testing/selftests/bpf/progs/uprobe_multi_session.c | 71 ++++++++++++++++++++++++
tools/testing/selftests/bpf/progs/uprobe_multi_session_cookie.c | 48 ++++++++++++++++
tools/testing/selftests/bpf/progs/uprobe_multi_session_recursive.c | 44 +++++++++++++++
tools/testing/selftests/bpf/progs/uprobe_multi_session_single.c | 44 +++++++++++++++
tools/testing/selftests/bpf/progs/uprobe_multi_verifier.c | 31 +++++++++++
21 files changed, 798 insertions(+), 61 deletions(-)
create mode 100644 tools/testing/selftests/bpf/progs/kprobe_multi_verifier.c
create mode 100644 tools/testing/selftests/bpf/progs/uprobe_multi_consumer_stress.c
create mode 100644 tools/testing/selftests/bpf/progs/uprobe_multi_session.c
create mode 100644 tools/testing/selftests/bpf/progs/uprobe_multi_session_cookie.c
create mode 100644 tools/testing/selftests/bpf/progs/uprobe_multi_session_recursive.c
create mode 100644 tools/testing/selftests/bpf/progs/uprobe_multi_session_single.c
create mode 100644 tools/testing/selftests/bpf/progs/uprobe_multi_verifier.c