2 # Copyright (c) 2011-2013, ARM Limited. All rights reserved.
4 # This program and the accompanying materials
5 # are licensed and made available under the terms and conditions of the BSD License
6 # which accompanies this distribution. The full text of the license may be found at
7 # http://opensource.org/licenses/bsd-license.php
9 # THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
10 # WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
13 from arm_ds
.debugger_v1
import DebugException
20 class EfiFileSection(object):
21 EFI_SECTION_PE32
= 0x10
22 EFI_SECTION_PIC
= 0x11
25 EFI_IMAGE_DEBUG_TYPE_CODEVIEW
= 0x2
27 SIZEOF_EFI_FFS_FILE_HEADER
= 0x28
29 def __init__(self
, ec
, base
):
34 return "FileSection(type:0x%X, size:0x%x)" % (self
.get_type(), self
.get_size())
40 return struct
.unpack("B", self
.ec
.getMemoryService().read(self
.base
+ 0x3, 1, 8))[0]
43 return (struct
.unpack("<I", self
.ec
.getMemoryService().read(self
.base
, 4, 32))[0] & 0x00ffffff)
45 def get_debug_filepath(self
):
46 type = self
.get_type()
47 if type == EfiFileSection
.EFI_SECTION_TE
:
48 section
= EfiSectionTE(self
, ec
, self
.base
+ 0x4)
49 elif type == EfiFileSection
.EFI_SECTION_PE32
:
50 section
= EfiSectionPE32(self
, ec
, self
.base
+ 0x4)
52 raise Exception("EfiFileSection", "No debug section")
53 return section
.get_debug_filepath()
56 SIZEOF_EFI_TE_IMAGE_HEADER
= 0x28
57 EFI_TE_IMAGE_SIGNATURE
= ('V','Z')
59 def __init__(self
, ec
, base_te
):
61 self
.base_te
= int(base_te
)
62 te_sig
= struct
.unpack("cc", self
.ec
.getMemoryService().read(self
.base_te
, 2, 32))
63 if te_sig
!= EfiSectionTE
.EFI_TE_IMAGE_SIGNATURE
:
64 raise Exception("EfiFileSectionTE","TE Signature incorrect")
66 def get_debug_filepath(self
):
67 stripped_size
= struct
.unpack("<H", self
.ec
.getMemoryService().read(self
.base_te
+ 0x6, 2, 32))[0]
68 stripped_size
-= EfiSectionTE
.SIZEOF_EFI_TE_IMAGE_HEADER
70 debug_dir_entry_rva
= self
.ec
.getMemoryService().readMemory32(self
.base_te
+ 0x20)
71 if debug_dir_entry_rva
== 0:
72 raise Exception("EfiFileSectionTE","No debug directory for image")
73 debug_dir_entry_rva
-= stripped_size
75 debug_type
= self
.ec
.getMemoryService().readMemory32(self
.base_te
+ debug_dir_entry_rva
+ 0xC)
76 if (debug_type
!= 0xdf) and (debug_type
!= EfiFileSection
.EFI_IMAGE_DEBUG_TYPE_CODEVIEW
):
77 raise Exception("EfiFileSectionTE","Debug type is not dwarf")
79 debug_rva
= self
.ec
.getMemoryService().readMemory32(self
.base_te
+ debug_dir_entry_rva
+ 0x14)
80 debug_rva
-= stripped_size
82 dwarf_sig
= struct
.unpack("cccc", self
.ec
.getMemoryService().read(self
.base_te
+ debug_rva
, 4, 32))
83 if (dwarf_sig
!= 0x66727764) and (dwarf_sig
!= FirmwareFile
.CONST_NB10_SIGNATURE
):
84 raise Exception("EfiFileSectionTE","Dwarf debug signature not found")
86 if dwarf_sig
== 0x66727764:
87 filename
= self
.base_te
+ debug_rva
+ 0xc
89 filename
= self
.base_te
+ debug_rva
+ 0x10
90 filename
= struct
.unpack("200s", self
.ec
.getMemoryService().read(filename
, 200, 32))[0]
91 return filename
[0:string
.find(filename
,'\0')]
93 def get_debug_elfbase(self
):
94 stripped_size
= struct
.unpack("<H", self
.ec
.getMemoryService().read(self
.base_te
+ 0x6, 2, 32))[0]
95 stripped_size
-= EfiSectionTE
.SIZEOF_EFI_TE_IMAGE_HEADER
97 return self
.base_te
- stripped_size
100 def __init__(self
, ec
, base_pe32
):
102 self
.base_pe32
= base_pe32
104 def get_debug_filepath(self
):
105 # Offset from dos hdr to PE file hdr
106 file_header_offset
= self
.ec
.getMemoryService().readMemory32(self
.base_pe32
+ 0x3C)
108 # Offset to debug dir in PE hdrs
109 debug_dir_entry_rva
= self
.ec
.getMemoryService().readMemory32(self
.base_pe32
+ file_header_offset
+ 0xA8)
110 if debug_dir_entry_rva
== 0:
111 raise Exception("EfiFileSectionPE32","No Debug Directory")
113 debug_type
= self
.ec
.getMemoryService().readMemory32(self
.base_pe32
+ debug_dir_entry_rva
+ 0xC)
114 if (debug_type
!= 0xdf) and (debug_type
!= EfiFileSection
.EFI_IMAGE_DEBUG_TYPE_CODEVIEW
):
115 raise Exception("EfiFileSectionPE32","Debug type is not dwarf")
118 debug_rva
= self
.ec
.getMemoryService().readMemory32(self
.base_pe32
+ debug_dir_entry_rva
+ 0x14)
120 dwarf_sig
= struct
.unpack("cccc", self
.ec
.getMemoryService().read(str(self
.base_pe32
+ debug_rva
), 4, 32))
121 if (dwarf_sig
!= 0x66727764) and (dwarf_sig
!= FirmwareFile
.CONST_NB10_SIGNATURE
):
122 raise Exception("EfiFileSectionPE32","Dwarf debug signature not found")
124 if dwarf_sig
== 0x66727764:
125 filename
= self
.base_pe32
+ debug_rva
+ 0xc
127 filename
= self
.base_pe32
+ debug_rva
+ 0x10
128 filename
= struct
.unpack("200s", self
.ec
.getMemoryService().read(str(filename
), 200, 32))[0]
129 return filename
[0:string
.find(filename
,'\0')]
131 def get_debug_elfbase(self
):
132 return self
.base_pe32
134 class EfiSectionPE64
:
135 def __init__(self
, ec
, base_pe64
):
137 self
.base_pe64
= base_pe64
139 def get_debug_filepath(self
):
140 # Offset from dos hdr to PE file hdr (EFI_IMAGE_NT_HEADERS64)
141 #file_header_offset = self.ec.getMemoryService().readMemory32(self.base_pe64 + 0x3C)
142 file_header_offset
= 0x0
144 # Offset to debug dir in PE hdrs
145 debug_dir_entry_rva
= self
.ec
.getMemoryService().readMemory32(self
.base_pe64
+ file_header_offset
+ 0x138)
146 if debug_dir_entry_rva
== 0:
147 raise Exception("EfiFileSectionPE64","No Debug Directory")
149 debug_type
= self
.ec
.getMemoryService().readMemory32(self
.base_pe64
+ debug_dir_entry_rva
+ 0xC)
150 if (debug_type
!= 0xdf) and (debug_type
!= EfiFileSection
.EFI_IMAGE_DEBUG_TYPE_CODEVIEW
):
151 raise Exception("EfiFileSectionPE64","Debug type is not dwarf")
154 debug_rva
= self
.ec
.getMemoryService().readMemory32(self
.base_pe64
+ debug_dir_entry_rva
+ 0x14)
156 dwarf_sig
= struct
.unpack("cccc", self
.ec
.getMemoryService().read(str(self
.base_pe64
+ debug_rva
), 4, 32))
157 if (dwarf_sig
!= 0x66727764) and (dwarf_sig
!= FirmwareFile
.CONST_NB10_SIGNATURE
):
158 raise Exception("EfiFileSectionPE64","Dwarf debug signature not found")
160 if dwarf_sig
== 0x66727764:
161 filename
= self
.base_pe64
+ debug_rva
+ 0xc
163 filename
= self
.base_pe64
+ debug_rva
+ 0x10
164 filename
= struct
.unpack("200s", self
.ec
.getMemoryService().read(str(filename
), 200, 32))[0]
165 return filename
[0:string
.find(filename
,'\0')]
167 def get_debug_elfbase(self
):
168 return self
.base_pe64
171 EFI_FV_FILETYPE_RAW
= 0x01
172 EFI_FV_FILETYPE_FREEFORM
= 0x02
173 EFI_FV_FILETYPE_SECURITY_CORE
= 0x03
174 EFI_FV_FILETYPE_PEI_CORE
= 0x04
175 EFI_FV_FILETYPE_DXE_CORE
= 0x05
176 EFI_FV_FILETYPE_PEIM
= 0x06
177 EFI_FV_FILETYPE_DRIVER
= 0x07
178 EFI_FV_FILETYPE_COMBINED_PEIM_DRIVER
= 0x08
179 EFI_FV_FILETYPE_APPLICATION
= 0x09
180 EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE
= 0x0B
181 EFI_FV_FILETYPE_FFS_MIN
= 0xF0
183 CONST_NB10_SIGNATURE
= ('N','B','1','0')
185 def __init__(self
, fv
, base
, ec
):
191 return "FFS(state:0x%x, type:0x%X, size:0x%x)" % (self
.get_state(), self
.get_type(), self
.get_size())
197 size
= (self
.ec
.getMemoryService().readMemory32(self
.base
+ 0x14) & 0x00ffffff)
199 # Occupied size is the size considering the alignment
200 return size
+ ((0x8 - (size
& 0x7)) & 0x7)
203 return self
.ec
.getMemoryService().readMemory8(self
.base
+ 0x12)
206 state
= self
.ec
.getMemoryService().readMemory8(self
.base
+ 0x17)
208 polarity
= self
.fv
.get_polarity()
213 while (highest_bit
!= 0) and ((highest_bit
& state
) == 0):
218 def get_next_section(self
, section
=None):
220 if self
.get_type() != FirmwareFile
.EFI_FV_FILETYPE_FFS_MIN
:
221 section_base
= self
.get_base() + 0x18;
225 section_base
= int(section
.get_base() + section
.get_size())
227 # Align to next 4 byte boundary
228 if (section_base
& 0x3) != 0:
229 section_base
= section_base
+ 0x4 - (section_base
& 0x3)
231 if section_base
< self
.get_base() + self
.get_size():
232 return EfiFileSection(self
.ec
, section_base
)
236 class FirmwareVolume
:
237 CONST_FV_SIGNATURE
= ('_','F','V','H')
238 EFI_FVB2_ERASE_POLARITY
= 0x800
242 def __init__(self
, ec
, fv_base
, fv_size
):
244 self
.fv_base
= fv_base
245 self
.fv_size
= fv_size
248 signature
= struct
.unpack("cccc", self
.ec
.getMemoryService().read(fv_base
+ 0x28, 4, 32))
249 except DebugException
:
250 raise Exception("FirmwareVolume", "Not possible to access the defined firmware volume at [0x%X,0x%X]. Could be the used build report does not correspond to your current debugging context." % (int(fv_base
),int(fv_base
+fv_size
)))
251 if signature
!= FirmwareVolume
.CONST_FV_SIGNATURE
:
252 raise Exception("FirmwareVolume", "This is not a valid firmware volume")
255 return self
.ec
.getMemoryService().readMemory32(self
.fv_base
+ 0x20)
257 def get_attributes(self
):
258 return self
.ec
.getMemoryService().readMemory32(self
.fv_base
+ 0x2C)
260 def get_polarity(self
):
261 attributes
= self
.get_attributes()
262 if attributes
& FirmwareVolume
.EFI_FVB2_ERASE_POLARITY
:
267 def get_next_ffs(self
, ffs
=None):
269 # Get the offset of the first FFS file from the FV header
270 ffs_base
= self
.fv_base
+ self
.ec
.getMemoryService().readMemory16(self
.fv_base
+ 0x30)
272 # Goto the next FFS file
273 ffs_base
= int(ffs
.get_base() + ffs
.get_size())
275 # Align to next 8 byte boundary
276 if (ffs_base
& 0x7) != 0:
277 ffs_base
= ffs_base
+ 0x8 - (ffs_base
& 0x7)
279 if ffs_base
< self
.fv_base
+ self
.get_size():
280 return FirmwareFile(self
, ffs_base
, self
.ec
)
284 def get_debug_info(self
):
287 ffs
= self
.get_next_ffs()
289 section
= ffs
.get_next_section()
290 while section
!= None:
291 type = section
.get_type()
292 if (type == EfiFileSection
.EFI_SECTION_TE
) or (type == EfiFileSection
.EFI_SECTION_PE32
):
293 self
.DebugInfos
.append((section
.get_base(), section
.get_size(), section
.get_type()))
294 section
= ffs
.get_next_section(section
)
295 ffs
= self
.get_next_ffs(ffs
)
297 def load_symbols_at(self
, addr
, verbose
= False):
298 if self
.DebugInfos
== []:
299 self
.get_debug_info()
301 for debug_info
in self
.DebugInfos
:
302 if (addr
>= debug_info
[0]) and (addr
< debug_info
[0] + debug_info
[1]):
303 if debug_info
[2] == EfiFileSection
.EFI_SECTION_TE
:
304 section
= EfiSectionTE(self
.ec
, debug_info
[0] + 0x4)
305 elif debug_info
[2] == EfiFileSection
.EFI_SECTION_PE32
:
306 section
= EfiSectionPE32(self
.ec
, debug_info
[0] + 0x4)
308 raise Exception('FirmwareVolume','Section Type not supported')
311 edk2_debugger
.load_symbol_from_file(self
.ec
, section
.get_debug_filepath(), section
.get_debug_elfbase(), verbose
)
312 except Exception, (ErrorClass
, ErrorMessage
):
314 print "Error while loading a symbol file (%s: %s)" % (ErrorClass
, ErrorMessage
)
318 def load_all_symbols(self
, verbose
= False):
319 if self
.DebugInfos
== []:
320 self
.get_debug_info()
322 for debug_info
in self
.DebugInfos
:
323 if debug_info
[2] == EfiFileSection
.EFI_SECTION_TE
:
324 section
= EfiSectionTE(self
.ec
, debug_info
[0] + 0x4)
325 elif debug_info
[2] == EfiFileSection
.EFI_SECTION_PE32
:
326 section
= EfiSectionPE32(self
.ec
, debug_info
[0] + 0x4)
331 edk2_debugger
.load_symbol_from_file(self
.ec
, section
.get_debug_filepath(), section
.get_debug_elfbase(), verbose
)
332 except Exception, (ErrorClass
, ErrorMessage
):
334 print "Error while loading a symbol file (%s: %s)" % (ErrorClass
, ErrorMessage
)