ArmPlatformPkg/Scripts: Infer dll load address from console output

[mirror_edk2.git] / ArmPlatformPkg / Scripts / Ds5 / cmd_load_symbols.py

#\r
#  Copyright (c) 2011-2021, Arm Limited. All rights reserved.\r
#\r
#  SPDX-License-Identifier: BSD-2-Clause-Patent\r
#\r
\r
from arm_ds.debugger_v1 import Debugger\r
from arm_ds.debugger_v1 import DebugException\r
\r
from console_loader import load_symbol_from_console\r
\r
import re, sys, getopt\r
\r
import edk2_debugger\r
\r
# Reload external classes\r
reload(edk2_debugger)\r
\r
def usage():\r
    print "-v,--verbose"\r
    print "-a,--all: Load all symbols"\r
    print "-l,--report=: Filename for the EDK2 report log"\r
    print "-m,--sysmem=(base,size): System Memory region"\r
    print "-f,--fv=(base,size): Firmware region"\r
    print "-r,--rom=(base,size): ROM region"\r
    print "-i,--input=: Filename for the EDK2 console output"\r
    print "-o,--objdump=: Path to the objdump tool"\r
\r
verbose = False\r
load_all = False\r
report_file = None\r
input_file = None\r
objdump = None\r
regions = []\r
opts,args = getopt.getopt(sys.argv[1:], "hvar:i:o:vm:vr:vf:v", ["help","verbose","all","report=","sysmem=","rom=","fv=","input=","objdump="])\r
if (opts is None) or (not opts):\r
    report_file = '../../../report.log'\r
else:\r
    region_reg = re.compile("\((.*),(.*)\)")\r
    base_reg = re.compile("(.*)")\r
\r
    for o,a in opts:\r
        region_type = None\r
        regex = None\r
        m = None\r
        if o in ("-h","--help"):\r
            usage()\r
            sys.exit()\r
        elif o in ("-v","--verbose"):\r
            verbose = True\r
        elif o in ("-a","--all"):\r
            load_all = True\r
        elif o in ("-l","--report"):\r
            report_file = a\r
        elif o in ("-m","--sysmem"):\r
            region_type = edk2_debugger.ArmPlatformDebugger.REGION_TYPE_SYSMEM\r
            regex = region_reg\r
        elif o in ("-f","--fv"):\r
            region_type = edk2_debugger.ArmPlatformDebugger.REGION_TYPE_FV\r
            regex = region_reg\r
        elif o in ("-r","--rom"):\r
            region_type = edk2_debugger.ArmPlatformDebugger.REGION_TYPE_ROM\r
            regex = region_reg\r
        elif o in ("-i","--input"):\r
            input_file = a\r
        elif o in ("-o", "--objdump"):\r
            objdump = a\r
        else:\r
            assert False, "Unhandled option (%s)" % o\r
\r
        if region_type:\r
            m = regex.match(a)\r
            if m:\r
                if regex.groups == 1:\r
                    regions.append((region_type,int(m.group(1),0),0))\r
                else:\r
                    regions.append((region_type,int(m.group(1),0),int(m.group(2),0)))\r
            else:\r
                if regex.groups == 1:\r
                    raise Exception('cmd_load_symbols', "Expect a base address")\r
                else:\r
                    raise Exception('cmd_load_symbols', "Expect a region format as (base,size)")\r
\r
# Debugger object for accessing the debugger\r
debugger = Debugger()\r
\r
# Initialisation commands\r
ec = debugger.getExecutionContext(0)\r
ec.getExecutionService().stop()\r
ec.getExecutionService().waitForStop()\r
# in case the execution context reference is out of date\r
ec = debugger.getExecutionContext(0)\r
\r
try:\r
    armplatform_debugger = edk2_debugger.ArmPlatformDebugger(ec, report_file, regions, verbose)\r
\r
    if load_all:\r
        armplatform_debugger.load_all_symbols()\r
    else:\r
        armplatform_debugger.load_current_symbols()\r
except IOError, (ErrorNumber, ErrorMessage):\r
    print "Error: %s" % ErrorMessage\r
except Exception, (ErrorClass, ErrorMessage):\r
    print "Error(%s): %s" % (ErrorClass, ErrorMessage)\r
except DebugException, de:\r
    print "DebugError: %s" % (de.getMessage())\r
\r
if input_file:\r
    load_symbol_from_console(ec, input_file, objdump, verbose)\r