tools/filetop: Add directory filter

Add support to filtop to filter by directory.

Signed-off-by: Srivathsa Dara <[email protected]>

Author: srivathsa729

tools/filetop.py

@@ -18,6 +18,8 @@
 from bcc import BPF
 from time import sleep, strftime
 import argparse
+import os
+import stat
 from subprocess import call
 
 # arguments
@@ -55,6 +57,9 @@
     help="number of outputs")
 parser.add_argument("--ebpf", action="store_true",
     help=argparse.SUPPRESS)
+parser.add_argument("-d", "--directory", type=str,
+    help="trace this directory only")
+
 args = parser.parse_args()
 interval = int(args.interval)
 countdown = int(args.count)
@@ -109,6 +114,10 @@
     if (d_name.len == 0 || TYPE_FILTER)
         return 0;
 
+    // skip if not in the specified directory
+    if (DIRECTORY_FILTER)
+        return 0;
+
     // store counts and sizes by pid & file
     struct info_t info = {
         .pid = pid,
@@ -163,6 +172,16 @@
     bpf_text = bpf_text.replace('TYPE_FILTER', '0')
 else:
     bpf_text = bpf_text.replace('TYPE_FILTER', '!S_ISREG(mode)')
+if args.directory:
+    try:
+        directory_inode = os.lstat(args.directory)[stat.ST_INO]
+        print(f'Tracing directory: {args.directory} (Inode: {directory_inode})')
+        bpf_text = bpf_text.replace('DIRECTORY_FILTER',  'file->f_path.dentry->d_parent->d_inode->i_ino != %d' % directory_inode)
+    except (FileNotFoundError, PermissionError) as e:
+        print(f'Error accessing directory {args.directory}: {e}')
+        exit(1)
+else:
+    bpf_text = bpf_text.replace('DIRECTORY_FILTER', '0')
 
 if debug or args.ebpf:
     print(bpf_text)