Print ioeventfd, iobus, vmstat and vcpustat information in kvm module.

Orabug: 37713468
Signed-off-by: Siddhi Katage <[email protected]>

Author: SiddhiK29

drgn_tools/kvm.py

@@ -8,13 +8,15 @@
 from drgn import cast
 from drgn import Object
 from drgn import Program
+from drgn.helpers.common.type import enum_type_to_class
 from drgn.helpers.linux.list import list_for_each_entry
 from drgn.helpers.linux.pid import find_task
 from drgn.helpers.linux.sched import task_state_to_char
 from drgn.helpers.linux.xarray import xa_for_each
 
 from drgn_tools.corelens import CorelensModule
 from drgn_tools.table import print_table
+from drgn_tools.util import enum_name_get
 from drgn_tools.util import has_member
 
 
@@ -268,6 +270,203 @@ def print_memslot_info(prog: Program) -> None:
         print("\n## Total Pages: %d ##" % (nr_pages))
 
 
+def print_ioeventfd_info(prog: Program) -> None:
+    """
+    Print VM's ioeventfd information
+    """
+    vm_list = for_each_vm(prog)
+
+    rows = [["KVM", "IOEVENTFD", "ADDR", "EVENTFD_CTX", "KVM_IO_DEV"]]
+    nr_ioeventfds = 0
+
+    for vm in vm_list:
+        iofd = list_for_each_entry(
+            "struct _ioeventfd", vm.ioeventfds.address_of_(), "list"
+        )
+        for fd in iofd:
+            addr = hex(fd.addr.value_())
+            ioeventfd = hex(fd.eventfd.value_())
+            dev = fd.dev.address_of_()
+            nr_ioeventfds = nr_ioeventfds + 1
+            rows.append(
+                [hex(vm.value_()), hex(fd.value_()), addr, ioeventfd, hex(dev)]
+            )
+    print("=============<< IOEVENTFDS >>=============")
+    print_table(rows)
+    print("\n## Total ioeventfds: ##", nr_ioeventfds)
+
+
+def print_iobus_info(prog: Program) -> None:
+    """
+    Print iobus information of VM
+    """
+    rows = [
+        ["KVM", "IOBUS", "BUS", "DEV_COUNT", "EVTFD_COUNT", "KVM_IO_RANGE"]
+    ]
+
+    vm_list = for_each_vm(prog)
+    kvm_bus_type = enum_type_to_class(
+        prog.type("enum kvm_bus"), "kvm_bus_type"
+    )
+
+    for vm in vm_list:
+        iobus_iterator = iter(vm.buses)
+        for i, bus in enumerate(iobus_iterator):
+            iobus = hex(bus.value_())
+            dev_count = bus.dev_count.value_()
+            eventfd_count = bus.ioeventfd_count.value_()
+            bus_name = enum_name_get(
+                kvm_bus_type,
+                i,
+                "UNKNOWN",
+            )
+            range = hex(bus.range.address_of_())
+            rows.append(
+                [
+                    hex(vm.value_()),
+                    iobus,
+                    str(bus_name),
+                    dev_count,
+                    eventfd_count,
+                    range,
+                ]
+            )
+    print("=============<< IOBUS >>=============")
+    print_table(rows)
+
+
+def print_kvmstat_info(prog: Program) -> None:
+    """
+    print vmstat and vcpustat information of a VM
+    """
+    vm_list = for_each_vm(prog)
+
+    for vm in vm_list:
+        stat = vm.stat
+        if has_member(stat, "generic"):
+            rtlab_flush = stat.generic.remote_tlb_flush.value_()
+        else:
+            rtlab_flush = stat.remote_tlb_flush.value_()
+        if has_member(stat, "lpages"):
+            lpages = stat.lpages.value_()
+        else:
+            lpages = "NA"
+        if has_member(stat, "pages_1g"):
+            pages_1g = stat.pages_1g.counter.value_()
+        else:
+            pages_1g = "NA"
+        if has_member(stat, "pages_2m"):
+            pages_2m = stat.pages_2m.counter.value_()
+        else:
+            pages_2m = "NA"
+        if has_member(stat, "pages_4k"):
+            pages_4k = stat.pages_4k.counter.value_()
+        else:
+            pages_4k = "NA"
+
+        print("=============<< VMSTAT >>============= \n")
+        rows_mm = [
+            ["KVM", hex(vm.value_())],
+            ["KVM_STAT", hex(vm.stat.address_of_())],
+            ["Remote TLB Flush", rtlab_flush],
+            ["MMU Shadow Zapped", stat.mmu_shadow_zapped.value_()],
+            ["MMU PTE Write", stat.mmu_pte_write.value_()],
+            ["MMU PDE Zapped", stat.mmu_pde_zapped.value_()],
+            ["MMU Floaded", stat.mmu_flooded.value_()],
+            ["MMU Recycled", stat.mmu_recycled.value_()],
+            ["MMU Cache Miss", stat.mmu_cache_miss.value_()],
+            ["MMU Unsync", stat.mmu_unsync.value_()],
+            ["Lpages", lpages],
+            ["NX Lpage Splits", stat.nx_lpage_splits.value_()],
+            [
+                "Max MMU Page Hash Collisions",
+                stat.max_mmu_page_hash_collisions.value_(),
+            ],
+            ["Pages_1G", pages_1g],
+            ["Pgaes_2M", pages_2m],
+            ["Pages_4k", pages_4k],
+        ]
+        print_table(rows_mm)
+
+        print("\n=============<< VCPU STAT >>============ \n")
+        for vcpu in for_each_vcpu(vm):
+            vcpu_stat = vcpu.stat
+            rows_vcpu = [
+                [
+                    "VCPU:",
+                    vcpu.vcpu_id.value_(),
+                    "HALT_SUC:",
+                    vcpu_stat.generic.halt_successful_poll.value_(),
+                    "HALT_ATTMPT:",
+                    vcpu_stat.generic.halt_attempted_poll.value_(),
+                    "HALT_INV:",
+                    vcpu_stat.generic.halt_poll_invalid.value_(),
+                ],
+                [
+                    "PF_FIXED:",
+                    vcpu_stat.pf_fixed.value_(),
+                    "PF_GUEST:",
+                    vcpu_stat.pf_guest.value_(),
+                    "TLB_FLUSH:",
+                    vcpu_stat.tlb_flush.value_(),
+                    "INVLPG:",
+                    vcpu_stat.invlpg.value_(),
+                ],
+                [
+                    "EXITS:",
+                    vcpu_stat.exits.value_(),
+                    "IO_EXIT:",
+                    vcpu_stat.io_exits.value_(),
+                    "MMIO_EXIT:",
+                    vcpu_stat.mmio_exits.value_(),
+                    "SIG_EXIT:",
+                    vcpu_stat.signal_exits.value_(),
+                ],
+                [
+                    "IRQ_WIN_EXIT:",
+                    vcpu_stat.irq_window_exits.value_(),
+                    "NMI_WIN_EXIT:",
+                    vcpu_stat.nmi_window_exits.value_(),
+                    "L1D_FLUSH:",
+                    vcpu_stat.l1d_flush.value_(),
+                    "HALT_EXIT:",
+                    vcpu_stat.halt_exits.value_(),
+                ],
+                [
+                    "REQ_IRQ_EXIT:",
+                    vcpu_stat.request_irq_exits.value_(),
+                    "IRQ_EXITS:",
+                    vcpu_stat.irq_exits.value_(),
+                    "HOST_STATE_RL:",
+                    vcpu_stat.host_state_reload.value_(),
+                    "FPU_RL:",
+                    vcpu_stat.fpu_reload.value_(),
+                ],
+                [
+                    "INSN_EMUL:",
+                    vcpu_stat.insn_emulation.value_(),
+                    "INSN_EMUL_FAIL:",
+                    vcpu_stat.insn_emulation_fail.value_(),
+                    "HYPERCALLS:",
+                    vcpu_stat.hypercalls.value_(),
+                    "IRQ_INJ:",
+                    vcpu_stat.irq_injections.value_(),
+                ],
+                [
+                    "NMI_INJ:",
+                    vcpu_stat.nmi_injections.value_(),
+                    "REQ_EVENT:",
+                    vcpu_stat.req_event.value_(),
+                    "PREEMPT_RPT:",
+                    vcpu_stat.preemption_reported.value_(),
+                    "PREEMT_OTH:",
+                    vcpu_stat.preemption_other.value_(),
+                ],
+            ]
+            print()
+            print_table(rows_vcpu)
+
+
 class KvmUtil(CorelensModule):
     """
     Show all the VM related info from KVM host side
@@ -277,9 +476,7 @@ class KvmUtil(CorelensModule):
 
     default_args = [
         [
-            "--vms",
-            "--vcpu",
-            "--mmslot",
+            "--all",
         ]
     ]
 
@@ -302,12 +499,41 @@ def add_args(self, parser: argparse.ArgumentParser) -> None:
             action="store_true",
             help="show all memslot info",
         )
+        parser.add_argument(
+            "--ioeventfd",
+            dest="ioeventfds",
+            action="store_true",
+            help="show all ioeventfds info",
+        )
+        parser.add_argument(
+            "--iobus",
+            dest="iobus",
+            action="store_true",
+            help="show all iobus info",
+        )
+        parser.add_argument(
+            "--kvmstat",
+            dest="kvmstat",
+            action="store_true",
+            help="show all iobus info",
+        )
+        parser.add_argument(
+            "--all",
+            action="store_true",
+            help="show all of the above info",
+        )
 
     def run(self, prog: Program, args: argparse.Namespace) -> None:
-        if args.list_vm:
+        if args.list_vm or args.all:
             print_vm_list(prog)
-        if args.vcpu_list:
+        if args.vcpu_list or args.all:
             print_vcpu_list(prog)
-        if args.memslot:
+        if args.memslot or args.all:
             print_memslot_info(prog)
+        if args.ioeventfds or args.all:
+            print_ioeventfd_info(prog)
+        if args.iobus or args.all:
+            print_iobus_info(prog)
+        if args.kvmstat or args.all:
+            print_kvmstat_info(prog)
         return

tests/test_kvm.py

@@ -7,3 +7,5 @@ def test_kvmutil(prog):
     kvm.print_vm_list(prog)
     kvm.print_vcpu_list(prog)
     kvm.print_memslot_info(prog)
+    kvm.print_ioeventfd_info(prog)
+    kvm.print_kvmstat_info(prog)