# SPDX-License-Identifier: GPL-2.0
#
# Copyright (c) 2023 MediaTek Inc.
#
# Authors:
#  Kuan-Ying Lee <[email protected]>
#

import gdb
import re
from linux import lists, utils, stackdepot, constants, mm

if constants.LX_CONFIG_MMU:
    vmap_area_type = utils.CachedType('struct vmap_area')
    vmap_area_ptr_type = vmap_area_type.get_type().pointer()

def is_vmalloc_addr(x):
    pg_ops = mm.page_ops().ops
    addr = pg_ops.kasan_reset_tag(x)
    return addr >= pg_ops.VMALLOC_START and addr < pg_ops.VMALLOC_END

class LxVmallocInfo(gdb.Command):
    """Show vmallocinfo"""

    def __init__(self):
        super(LxVmallocInfo, self).__init__("lx-vmallocinfo", gdb.COMMAND_DATA)

    def invoke(self, arg, from_tty):
        if not constants.LX_CONFIG_MMU:
            raise gdb.GdbError("Requires MMU support")

        vmap_area_list = gdb.parse_and_eval('vmap_area_list')
        for vmap_area in lists.list_for_each_entry(vmap_area_list, vmap_area_ptr_type, "list"):
            if not vmap_area['vm']:
                gdb.write("0x%x-0x%x %10d vm_map_ram\n" % (vmap_area['va_start'], vmap_area['va_end'],
                    vmap_area['va_end'] - vmap_area['va_start']))
                continue
            v = vmap_area['vm']
            gdb.write("0x%x-0x%x %10d" % (v['addr'], v['addr'] + v['size'], v['size']))
            if v['caller']:
                gdb.write(" %s" % str(v['caller']).split(' ')[-1])
            if v['nr_pages']:
                gdb.write(" pages=%d" % v['nr_pages'])
            if v['phys_addr']:
                gdb.write(" phys=0x%x" % v['phys_addr'])
            if v['flags'] & constants.LX_VM_IOREMAP:
                gdb.write(" ioremap")
            if v['flags'] & constants.LX_VM_ALLOC:
                gdb.write(" vmalloc")
            if v['flags'] & constants.LX_VM_MAP:
                gdb.write(" vmap")
            if v['flags'] & constants.LX_VM_USERMAP:
                gdb.write(" user")
            if v['flags'] & constants.LX_VM_DMA_COHERENT:
                gdb.write(" dma-coherent")
            if v['flags'] & constants.LX_VM_RANDOM:
                gdb.write(" random")
            if v['flags'] & constants.LX_VM_ALIGN_END:
                gdb.write(" ralign")
            if is_vmalloc_addr(v['pages']):
                gdb.write(" vpages")
            gdb.write("\n")

LxVmallocInfo()