lunaix-os/scripts/gdb/lunadbg/mem.py

   1 from .commands import LunadbgCommand
   2 from .pp import MyPrettyPrinter
   3 from .profiling.pmstat import PhysicalMemProfile
   4 from .structs.pagetable import PageTable
   5
   6 class MMStats(LunadbgCommand):
   7     def __init__(self) -> None:
   8         super().__init__("mm")
   9         subparsers = self._parser.add_subparsers(dest="cmd")
  10
  11         stats = subparsers.add_parser("stats")
  12         stats.add_argument("state_type")
  13         stats.add_argument("-g", "--granule", type=int, default=512)
  14         stats.add_argument("--cols", type=int, default=32)
  15
  16         lookup = subparsers.add_parser("lookup")
  17         lookup.add_argument("mem_type")
  18         lookup.add_argument("address")
  19         lookup.add_argument("-l", "--level", type=int, default=-1)
  20         lookup.add_argument("-n", type=int, default=0)
  21         lookup.add_argument("-t", "--to", dest="to_addr", default='0')
  22
  23         self.__ptw = PageTable()
  24
  25     def print_pmem_stats(self, pp: MyPrettyPrinter, optn):
  26         pmem = PhysicalMemProfile()
  27         pmem.rescan_pmem(optn.granule)
  28
  29         pp.printf("Total: %dKiB (%d@4K)",
  30                   pmem.max_mem_sz / 1024, pmem.max_mem_pg)
  31
  32         pp.printf("Used:  %dKiB (%d@4K) ~%.2f%%",
  33                   pmem.consumed_pg * 4096 / 1024,
  34                   pmem.consumed_pg, pmem.utilisation * 100)
  35
  36         pp.printf("Fragmentations: %d ~%.2f%%", pmem.discontig, pmem.fragmentation * 100)
  37         pp.print()
  38
  39         pp.print("Distribution")
  40         pp.print("( . = empty, * = full, [0-9]0% full )")
  41         pp2 = pp.next_level(2)
  42         row = []
  43         for i in range(0, len(pmem.mem_distr)):
  44             ratio = pmem.mem_distr[i] / pmem.page_per_granule
  45             cat = int(ratio * 9)
  46             if ratio == 0:
  47                 row.append('.')
  48             elif ratio == 1:
  49                 row.append('*')
  50             else:
  51                 row.append(str(cat))
  52
  53             if (i + 1) % optn.cols == 0:
  54                 pp2.print(''.join(row))
  55                 row.clear()
  56         if len(row) > 0:
  57             pp2.print(''.join(row))
  58
  59         pp.printf("(granule: %d, density: %d@4K)", optn.granule, pmem.page_per_granule)
  60
  61     def vm_lookup(self, pp, va, optn):
  62         to_addr = int(optn.to_addr, 0)
  63         if not optn.n and not to_addr:
  64             pp.print(self.__ptw.get_pte(va, level=optn.level))
  65         else:
  66             if to_addr:
  67                 self.__ptw.print_ptes_between(pp, va, to_addr, optn.level)
  68             else:
  69                 self.__ptw.print_ptes(pp, va, optn.n, optn.level)
  70
  71     def __do_stats(self, pp, optn):
  72         if optn.state_type == "pmem":
  73             self.print_pmem_stats(pp, optn)
  74         else:
  75             print("unknow stats type:", optn.state_type)
  76
  77     def __do_lookup(self, pp, address, optn):
  78         if optn.mem_type == "vm":
  79             self.vm_lookup(pp, int(address, base=0), optn)
  80         else:
  81             print("unknow mem type:", optn.state_type)
  82
  83     def on_execute(self, optn, gdb_args, from_tty) -> None:
  84         pp = MyPrettyPrinter()
  85
  86         if optn.cmd == 'stats':
  87             self.__do_stats(pp, optn)
  88         elif optn.cmd == 'lookup':
  89             self.__do_lookup(pp, optn.address, optn)
  90         else:
  91             print("unknown command:", optn.cmd)