Add iomem parser

Signed-off-by: quic_wya <[email protected]>

Author: quic-wya

CMakeLists.txt

@@ -33,7 +33,8 @@ list(APPEND PLUGIN_SOURCES
         binder/binder.cpp
         memory/cma.cpp
 	memory/reserved.cpp
-        memory/memblock.cpp
+	memory/memblock.cpp
+	memory/iomem.cpp
         procrank/procrank.cpp
         devicetree/dts.cpp
         devicetree/devicetree.cpp
@@ -77,4 +78,9 @@ add_library(reserved SHARED
             memory/reserved.cpp
             devicetree/devicetree.cpp)
 set_target_properties(reserved PROPERTIES PREFIX "")
+
+add_library(iomem SHARED
+            ${PLUGIN_SOURCES}
+            memory/iomem.cpp)
+set_target_properties(iomem PROPERTIES PREFIX "")
 endif()

README.md

@@ -66,6 +66,7 @@ Supprot command:
 | memblock   | √         | √         | parser memblock info                                 |
 | workqueue  | √         | √         | parser workqueue info                                |
 | reserved   | √         | √         | parser reserved memory info                          |
+| iomem      | √         | √         | parser memory layout info                            |
 | buddy      | x         | x         |                                                      |
 | vmalloc    | x         | x         |                                                      |
 | dmabuf     | x         | x         |                                                      |

memory/iomem.cpp

@@ -0,0 +1,107 @@
+// Copyright (c) 2024-2025 Qualcomm Innovation Center, Inc. All rights reserved.
+// SPDX-License-Identifier: BSD-3-Clause-Clear
+
+#include "iomem.h"
+
+#pragma GCC diagnostic push
+#pragma GCC diagnostic ignored "-Wpointer-arith"
+
+#ifndef BUILD_TARGET_TOGETHER
+DEFINE_PLUGIN_COMMAND(IoMem)
+#endif
+
+void IoMem::cmd_main(void) {
+    int c;
+    if (argcnt < 2) cmd_usage(pc->curcmd, SYNOPSIS);
+    if(iomem_list.size() == 0){
+        parser_iomem();
+    }
+    while ((c = getopt(argcnt, args, "a")) != EOF) {
+        switch(c) {
+            case 'a':
+                print_iomem(iomem_list,0);
+                break;
+            default:
+                argerrs++;
+                break;
+        }
+    }
+    if (argerrs)
+        cmd_usage(pc->curcmd, SYNOPSIS);
+}
+
+IoMem::IoMem(){
+    field_init(resource,start);
+    field_init(resource,end);
+    field_init(resource,name);
+    field_init(resource,flags);
+    field_init(resource,parent);
+    field_init(resource,sibling);
+    field_init(resource,child);
+    struct_init(resource);
+    cmd_name = "iomem";
+    help_str_list={
+        "iomem",                /* command name */
+        "dump io memory information",    /* short description */
+        "-a \n"
+            "\n",
+        "EXAMPLES",
+        "  Display io memory info:",
+        "    %s> iomem -a",
+        "    0x0-0xffffffff  4.00Gb     : PCI mem",
+        "        0x500000-0x7fffff  3.00Mb     : 500000.pinctrl pinctrl@500000",
+        "        0x1400000-0x15dffff  1.87Mb     : 1400000.clock-controller cc_base",
+        "        0x1612000-0x1612003  3b         : 1613000.hsphy eud_enable_reg",
+        "        0x1613000-0x161311f  287b       : 1613000.hsphy hsusb_phy_base",
+        "        0x1880000-0x18de1ff  376.50Kb   : 1880000.interconnect interconnect@1880000",
+        "\n",
+    };
+    initialize();
+}
+
+void IoMem::print_iomem(std::vector<std::shared_ptr<resource>>& res_list,int level){
+    char buf_size[BUFSIZE];
+    for (const auto& res_ptr : res_list) {
+        for (int i = 0; i < level; i++) {
+            fprintf(fp, "\t");
+        }
+        convert_size((res_ptr->end - res_ptr->start),buf_size);
+        fprintf(fp, "0x%lx-0x%lx  %s : %s\n",
+            res_ptr->start,res_ptr->end,
+            mkstring(buf_size, 10, LJUST, buf_size),
+            res_ptr->name.c_str());
+        if(res_ptr->childs.size() > 0){
+            print_iomem(res_ptr->childs,(level+1));
+        }
+    }
+}
+
+void IoMem::parser_iomem(){
+    if (csymbol_exists("iomem_resource")){
+        parser_resource(csymbol_value("iomem_resource"),iomem_list);
+    }
+}
+
+void IoMem::parser_resource(ulong addr,std::vector<std::shared_ptr<resource>>& res_list){
+    if (!is_kvaddr(addr)) return;
+    void *res_buf = read_struct(addr,"resource");
+    if(res_buf == nullptr) return;
+    std::shared_ptr<resource> res_ptr = std::make_shared<resource>();
+    res_ptr->addr = addr;
+    ulong name_addr = ULONG(res_buf + field_offset(resource,name));
+    res_ptr->name = read_cstring(name_addr,64, "resource_name");
+    res_ptr->start = ULONG(res_buf + field_offset(resource,start));
+    res_ptr->end = ULONG(res_buf + field_offset(resource,end));
+    res_ptr->flags = ULONG(res_buf + field_offset(resource,flags));
+    res_list.push_back(res_ptr);
+    ulong sibling = ULONG(res_buf + field_offset(resource,sibling));
+    if (is_kvaddr(sibling)){
+        parser_resource(sibling,res_list);
+    }
+    ulong child = ULONG(res_buf + field_offset(resource,child));
+    if (is_kvaddr(child)){
+        parser_resource(child,res_ptr->childs);
+    }
+    FREEBUF(res_buf);
+}
+#pragma GCC diagnostic pop

memory/iomem.h

@@ -0,0 +1,31 @@
+// Copyright (c) 2024-2025 Qualcomm Innovation Center, Inc. All rights reserved.
+// SPDX-License-Identifier: BSD-3-Clause-Clear
+
+#ifndef IOMEM_DEFS_H_
+#define IOMEM_DEFS_H_
+
+#include "plugin.h"
+#include "devicetree/devicetree.h"
+
+struct resource {
+    ulong addr;
+    ulong start;
+    ulong end;
+    std::string name;
+    ulong flags;
+    std::vector<std::shared_ptr<resource>> childs;
+};
+
+class IoMem : public PaserPlugin {
+public:
+    std::vector<std::shared_ptr<resource>> iomem_list;
+    IoMem();
+    void cmd_main(void) override;
+    void parser_iomem();
+    void print_iomem(std::vector<std::shared_ptr<resource>>& res_list,int level);
+    void parser_resource(ulong addr,std::vector<std::shared_ptr<resource>>& res_list);
+    DEFINE_PLUGIN_INSTANCE(IoMem)
+};
+
+
+#endif // IOMEM_DEFS_H_

plugins.cpp

@@ -6,6 +6,7 @@
 #include "devicetree/dts.h"
 #include "memory/memblock.h"
 #include "memory/reserved.h"
+#include "memory/iomem.h"
 #include "workqueue/workqueue.h"
 
 #pragma GCC diagnostic push
@@ -22,6 +23,7 @@ std::unique_ptr<Dts>        Dts::instance = nullptr;
 std::unique_ptr<Memblock>   Memblock::instance = nullptr;
 std::unique_ptr<Workqueue>  Workqueue::instance = nullptr;
 std::unique_ptr<Reserved>   Reserved::instance = nullptr;
+std::unique_ptr<IoMem>      IoMem::instance = nullptr;
 
 extern "C" void __attribute__((constructor)) plugin_init(void) {
     // fprintf(fp, "plugin_init\n");
@@ -32,6 +34,8 @@ extern "C" void __attribute__((constructor)) plugin_init(void) {
     Memblock::instance = std::make_unique<Memblock>();
     Workqueue::instance = std::make_unique<Workqueue>();
     Reserved::instance = std::make_unique<Reserved>();
+    IoMem::instance = std::make_unique<IoMem>();
+
     static struct command_table_entry command_table[] = {
         { &Binder::instance->cmd_name[0], &Binder::wrapper_func, Binder::instance->cmd_help, 0 },
         { &Procrank::instance->cmd_name[0], &Procrank::wrapper_func, Procrank::instance->cmd_help, 0 },
@@ -40,6 +44,7 @@ extern "C" void __attribute__((constructor)) plugin_init(void) {
         { &Memblock::instance->cmd_name[0], &Memblock::wrapper_func, Memblock::instance->cmd_help, 0 },
         { &Workqueue::instance->cmd_name[0], &Workqueue::wrapper_func, Workqueue::instance->cmd_help, 0 },
         { &Reserved::instance->cmd_name[0], &Reserved::wrapper_func, Reserved::instance->cmd_help, 0 },
+        { &IoMem::instance->cmd_name[0], &IoMem::wrapper_func, IoMem::instance->cmd_help, 0 },
         { NULL }
     };
     register_extension(command_table);
@@ -54,6 +59,7 @@ extern "C" void __attribute__((destructor)) plugin_fini(void) {
     Memblock::instance.reset();
     Workqueue::instance.reset();
     Reserved::instance.reset();
+    IoMem::instance.reset();
 }
 
 #endif // BUILD_TARGET_TOGETHER