tools/slabratetop: detect the current memory allocator

Currently, slabratetop always included slub_def.h. For the system uses SLAB, the definition of 'struct kmem_cache' is different and the eBPF program would fetch the wrong field and print garbages. Signed-off-by: Gary Lin <[email protected]>
drcwr · Mar 14, 2018 · fa10345 · fa10345
1 parent 7cfb73b
commit fa10345
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tools/slabratetop.py b/tools/slabratetop.py
@@ -63,7 +63,11 @@ def signal_ignore(signal, frame):
 #include <uapi/linux/ptrace.h>
 #include <linux/mm.h>
 #include <linux/slab.h>
+#ifdef CONFIG_SLUB
 #include <linux/slub_def.h>
+#else
+#include <linux/slab_def.h>
+#endif
 
 #define CACHE_NAME_SIZE 32