mk-rootfs.sh

hc

2024-03-26 e0728245c89800c2038c23308f2d88969d5b41c8

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
# 
#  Copyright (c) 2021, Arm Limited. All rights reserved. 
# 
#  SPDX-License-Identifier: BSD-2-Clause-Patent 
# 
 
from arm_ds.debugger_v1 import DebugException 
 
import subprocess, os, edk2_debugger, re 
 
def get_module_name(line): 
    path = line.rsplit(' ')[1] 
    return os.path.splitext(os.path.basename(path))[0] 
 
def get_module_path(line): 
    return line.rsplit(' ')[1] 
 
def get_module_entrypoint(list, module_name): 
    line = [i for i in list if module_name in i and re.search(r'\b'+module_name+r'\b', i)] 
    if len(line) == 0: 
        # Module was not loaded using DxeDispatcher or PeiDispatcher. It is a SEC module 
        # Symbols for these modules are loaded from FV, not from console log 
        return None 
 
    entrypoint_str =  line[0].rsplit(' ')[4] 
    return entrypoint_str.rsplit('=')[1] 
 
def load_symbol_from_console(ec, console_file, objdump, verbose): 
    if objdump is None: 
        print "Error: A path to objdump tool is not specified, but -i parameter is provided" 
    elif not os.path.exists(objdump): 
        print "Error: Provided path to objdump is invalid: %s" % objdump 
    elif not os.path.exists(console_file): 
        print "Error: UEFI console file is not found: %s" % console_file 
    else: 
 
        full_list = open(console_file).read().splitlines() 
 
        efi_list = [i for i in full_list if "EntryPoint=" in i] 
 
        full_list = dict.fromkeys(full_list) 
        full_list = [i for i in full_list if "add-symbol-file" in i] 
 
        module_dict = {} 
 
        for line in full_list: 
            name = get_module_name(line) 
            module_dict[name] = (get_module_path(line), get_module_entrypoint(efi_list, name)) 
 
        for module in module_dict: 
            entrypoint_addr = module_dict[module][1] 
 
            if entrypoint_addr is not None: 
                path = module_dict[module][0] 
                if not os.path.exists(path): 
                    print "Module not found: " + path + ". Skipping..." 
                    continue 
 
                sp = subprocess.Popen([objdump,'-S', path], stdout = subprocess.PIPE) 
 
                objdump_out = sp.stdout.readlines() 
                entrypoint_record = [i for i in objdump_out if "<_ModuleEntryPoint>" in i] 
 
                entrypoint_offset = entrypoint_record[0].split(' ')[0] 
 
                load_addr = int(entrypoint_addr, 16) - int(entrypoint_offset, 16) 
 
                edk2_debugger.load_symbol_from_file(ec, path, load_addr, verbose)