tools/swapin: Support swap_read_folio()

kernel[0] commit c9bdf768dd93("mm: convert swap_readpage() to swap_read_folio()")
convert swap_readpage() to swap_read_folio().

[0] https://github.com/torvalds/linux.git

Signed-off-by: Xue Yuehua <2482887395@qq.com>
Signed-off-by: Rong Tao <rongtao@cestc.cn>

Author: Rtoax

tools/swapin.py

@@ -13,6 +13,7 @@
 # When copying or porting, include this comment.
 #
 # 03-Jul-2019   Brendan Gregg   Ported from bpftrace to BCC.
+# 31-May-2024   Rong Tao        Support folio
 
 from __future__ import print_function
 from bcc import BPF
@@ -46,7 +47,7 @@
 
 BPF_HASH(counts, struct key_t, u64);
 
-int kprobe__swap_readpage(struct pt_regs *ctx)
+int trace_swap_read(struct pt_regs *ctx)
 {
     u64 *val, zero = 0;
     u32 tgid = bpf_get_current_pid_tgid() >> 32;
@@ -66,6 +67,16 @@
 htab_batch_ops = True if BPF.kernel_struct_has_field(b'bpf_map_ops',
         b'map_lookup_and_delete_batch') == 1 else False
 
+if b.get_kprobe_functions(b"swap_readpage"):
+    b.attach_kprobe(event="swap_readpage", fn_name="trace_swap_read")
+elif b.get_kprobe_functions(b"swap_read_folio"):
+    b.attach_kprobe(event="swap_read_folio", fn_name="trace_swap_read")
+else:
+    print("ERROR: swap_readpage() and swap_read_folio() kernel function"
+          " not found or traceable. "
+          "The kernel might be too old or the the function has been inlined.")
+    exit()
+
 print("Counting swap ins. Ctrl-C to end.");
 
 # output