tools/cgroup/memcg_shrinker.py

   1 #!/usr/bin/env python3
   2 #
   3 # Copyright (C) 2022 Roman Gushchin <roman.gushchin@linux.dev>
   4 # Copyright (C) 2022 Meta
   5
   6 import os
   7 import argparse
   8 import sys
   9
  10
  11 def scan_cgroups(cgroup_root):
  12     cgroups = {}
  13
  14     for root, subdirs, _ in os.walk(cgroup_root):
  15         for cgroup in subdirs:
  16             path = os.path.join(root, cgroup)
  17             ino = os.stat(path).st_ino
  18             cgroups[ino] = path
  19
  20     # (memcg ino, path)
  21     return cgroups
  22
  23
  24 def scan_shrinkers(shrinker_debugfs):
  25     shrinkers = []
  26
  27     for root, subdirs, _ in os.walk(shrinker_debugfs):
  28         for shrinker in subdirs:
  29             count_path = os.path.join(root, shrinker, "count")
  30             with open(count_path) as f:
  31                 for line in f.readlines():
  32                     items = line.split(' ')
  33                     ino = int(items[0])
  34                     # (count, shrinker, memcg ino)
  35                     shrinkers.append((int(items[1]), shrinker, ino))
  36     return shrinkers
  37
  38
  39 def main():
  40     parser = argparse.ArgumentParser(description='Display biggest shrinkers')
  41     parser.add_argument('-n', '--lines', type=int, help='Number of lines to print')
  42
  43     args = parser.parse_args()
  44
  45     cgroups = scan_cgroups("/sys/fs/cgroup/")
  46     shrinkers = scan_shrinkers("/sys/kernel/debug/shrinker/")
  47     shrinkers = sorted(shrinkers, reverse = True, key = lambda x: x[0])
  48
  49     n = 0
  50     for s in shrinkers:
  51         count, name, ino = (s[0], s[1], s[2])
  52         if count == 0:
  53             break
  54
  55         if ino == 0 or ino == 1:
  56             cg = "/"
  57         else:
  58             try:
  59                 cg = cgroups[ino]
  60             except KeyError:
  61                 cg = "unknown (%d)" % ino
  62
  63         print("%-8s %-20s %s" % (count, name, cg))
  64
  65         n += 1
  66         if args.lines and n >= args.lines:
  67             break
  68
  69
  70 if __name__ == '__main__':
  71     main()