libbpf-tools/memleak: add mremap uprobe (#5151)

Add mremap uprobe. Signed-off-by: Xiaobo Liu <[email protected]>
iovisor · Nov 23, 2024 · 378d287 · 378d287
1 parent fb62cc3
commit 378d287
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/libbpf-tools/memleak.bpf.c b/libbpf-tools/memleak.bpf.c
@@ -223,6 +223,20 @@ int BPF_UPROBE(munmap_enter, void *address)
 	return gen_free_enter(address);
 }
 
+SEC("uprobe")
+int BPF_UPROBE(mremap_enter, void *old_address, size_t old_size, size_t new_size, int flags)
+{
+	gen_free_enter(old_address);
+
+	return gen_alloc_enter(new_size);
+}
+
+SEC("uretprobe")
+int BPF_URETPROBE(mremap_exit)
+{
+	return gen_alloc_exit(ctx);
+}
+
 SEC("uprobe")
 int BPF_UPROBE(posix_memalign_enter, void **memptr, size_t alignment, size_t size)
 {

diff --git a/libbpf-tools/memleak.c b/libbpf-tools/memleak.c
@@ -1060,9 +1060,15 @@ int attach_uprobes(struct memleak_bpf *skel)
 	if (strlen(env.symbols_prefix)) {
 		ATTACH_UPROBE(skel, mmap, mmap_enter);
 		ATTACH_URETPROBE(skel, mmap, mmap_exit);
+
+		ATTACH_UPROBE(skel, mremap, mmap_enter);
+		ATTACH_URETPROBE(skel, mremap, mmap_exit);
 	} else {
 		ATTACH_UPROBE_CHECKED(skel, mmap, mmap_enter);
 		ATTACH_URETPROBE_CHECKED(skel, mmap, mmap_exit);
+
+		ATTACH_UPROBE_CHECKED(skel, mremap, mremap_enter);
+		ATTACH_URETPROBE_CHECKED(skel, mremap, mremap_exit);
 	}
 
 	ATTACH_UPROBE_CHECKED(skel, posix_memalign, posix_memalign_enter);