1 //===-- ObjectFilePECOFF.cpp ----------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include "ObjectFilePECOFF.h"
10 #include "PECallFrameInfo.h"
11 #include "WindowsMiniDump.h"
12 
13 #include "lldb/Core/FileSpecList.h"
14 #include "lldb/Core/Module.h"
15 #include "lldb/Core/ModuleSpec.h"
16 #include "lldb/Core/PluginManager.h"
17 #include "lldb/Core/Section.h"
18 #include "lldb/Core/StreamFile.h"
19 #include "lldb/Symbol/ObjectFile.h"
20 #include "lldb/Target/Process.h"
21 #include "lldb/Target/SectionLoadList.h"
22 #include "lldb/Target/Target.h"
23 #include "lldb/Utility/ArchSpec.h"
24 #include "lldb/Utility/DataBufferHeap.h"
25 #include "lldb/Utility/FileSpec.h"
26 #include "lldb/Utility/Log.h"
27 #include "lldb/Utility/StreamString.h"
28 #include "lldb/Utility/Timer.h"
29 #include "lldb/Utility/UUID.h"
30 #include "llvm/BinaryFormat/COFF.h"
31 
32 #include "llvm/Object/COFFImportFile.h"
33 #include "llvm/Support/Error.h"
34 #include "llvm/Support/MemoryBuffer.h"
35 
36 #define IMAGE_DOS_SIGNATURE 0x5A4D    // MZ
37 #define IMAGE_NT_SIGNATURE 0x00004550 // PE00
38 #define OPT_HEADER_MAGIC_PE32 0x010b
39 #define OPT_HEADER_MAGIC_PE32_PLUS 0x020b
40 
41 using namespace lldb;
42 using namespace lldb_private;
43 
44 LLDB_PLUGIN_DEFINE(ObjectFilePECOFF)
45 
46 struct CVInfoPdb70 {
47   // 16-byte GUID
48   struct _Guid {
49     llvm::support::ulittle32_t Data1;
50     llvm::support::ulittle16_t Data2;
51     llvm::support::ulittle16_t Data3;
52     uint8_t Data4[8];
53   } Guid;
54 
55   llvm::support::ulittle32_t Age;
56 };
57 
58 static UUID GetCoffUUID(llvm::object::COFFObjectFile &coff_obj) {
59   const llvm::codeview::DebugInfo *pdb_info = nullptr;
60   llvm::StringRef pdb_file;
61 
62   // This part is similar with what has done in minidump parser.
63   if (!coff_obj.getDebugPDBInfo(pdb_info, pdb_file) && pdb_info) {
64     if (pdb_info->PDB70.CVSignature == llvm::OMF::Signature::PDB70) {
65       using llvm::support::endian::read16be;
66       using llvm::support::endian::read32be;
67 
68       const uint8_t *sig = pdb_info->PDB70.Signature;
69       struct CVInfoPdb70 info;
70       info.Guid.Data1 = read32be(sig);
71       sig += 4;
72       info.Guid.Data2 = read16be(sig);
73       sig += 2;
74       info.Guid.Data3 = read16be(sig);
75       sig += 2;
76       memcpy(info.Guid.Data4, sig, 8);
77 
78       // Return 20-byte UUID if the Age is not zero
79       if (pdb_info->PDB70.Age) {
80         info.Age = read32be(&pdb_info->PDB70.Age);
81         return UUID::fromOptionalData(&info, sizeof(info));
82       }
83       // Otherwise return 16-byte GUID
84       return UUID::fromOptionalData(&info.Guid, sizeof(info.Guid));
85     }
86   }
87 
88   return UUID();
89 }
90 
91 char ObjectFilePECOFF::ID;
92 
93 void ObjectFilePECOFF::Initialize() {
94   PluginManager::RegisterPlugin(
95       GetPluginNameStatic(), GetPluginDescriptionStatic(), CreateInstance,
96       CreateMemoryInstance, GetModuleSpecifications, SaveCore);
97 }
98 
99 void ObjectFilePECOFF::Terminate() {
100   PluginManager::UnregisterPlugin(CreateInstance);
101 }
102 
103 lldb_private::ConstString ObjectFilePECOFF::GetPluginNameStatic() {
104   static ConstString g_name("pe-coff");
105   return g_name;
106 }
107 
108 const char *ObjectFilePECOFF::GetPluginDescriptionStatic() {
109   return "Portable Executable and Common Object File Format object file reader "
110          "(32 and 64 bit)";
111 }
112 
113 ObjectFile *ObjectFilePECOFF::CreateInstance(const lldb::ModuleSP &module_sp,
114                                              DataBufferSP &data_sp,
115                                              lldb::offset_t data_offset,
116                                              const lldb_private::FileSpec *file_p,
117                                              lldb::offset_t file_offset,
118                                              lldb::offset_t length) {
119   FileSpec file = file_p ? *file_p : FileSpec();
120   if (!data_sp) {
121     data_sp = MapFileData(file, length, file_offset);
122     if (!data_sp)
123       return nullptr;
124     data_offset = 0;
125   }
126 
127   if (!ObjectFilePECOFF::MagicBytesMatch(data_sp))
128     return nullptr;
129 
130   // Update the data to contain the entire file if it doesn't already
131   if (data_sp->GetByteSize() < length) {
132     data_sp = MapFileData(file, length, file_offset);
133     if (!data_sp)
134       return nullptr;
135   }
136 
137   auto objfile_up = std::make_unique<ObjectFilePECOFF>(
138       module_sp, data_sp, data_offset, file_p, file_offset, length);
139   if (!objfile_up || !objfile_up->ParseHeader())
140     return nullptr;
141 
142   // Cache coff binary.
143   if (!objfile_up->CreateBinary())
144     return nullptr;
145 
146   return objfile_up.release();
147 }
148 
149 ObjectFile *ObjectFilePECOFF::CreateMemoryInstance(
150     const lldb::ModuleSP &module_sp, lldb::DataBufferSP &data_sp,
151     const lldb::ProcessSP &process_sp, lldb::addr_t header_addr) {
152   if (!data_sp || !ObjectFilePECOFF::MagicBytesMatch(data_sp))
153     return nullptr;
154   auto objfile_up = std::make_unique<ObjectFilePECOFF>(
155       module_sp, data_sp, process_sp, header_addr);
156   if (objfile_up.get() && objfile_up->ParseHeader()) {
157     return objfile_up.release();
158   }
159   return nullptr;
160 }
161 
162 size_t ObjectFilePECOFF::GetModuleSpecifications(
163     const lldb_private::FileSpec &file, lldb::DataBufferSP &data_sp,
164     lldb::offset_t data_offset, lldb::offset_t file_offset,
165     lldb::offset_t length, lldb_private::ModuleSpecList &specs) {
166   const size_t initial_count = specs.GetSize();
167   if (!data_sp || !ObjectFilePECOFF::MagicBytesMatch(data_sp))
168     return initial_count;
169 
170   Log *log(GetLogIfAllCategoriesSet(LIBLLDB_LOG_OBJECT));
171 
172   if (data_sp->GetByteSize() < length)
173     if (DataBufferSP full_sp = MapFileData(file, -1, file_offset))
174       data_sp = std::move(full_sp);
175   auto binary = llvm::object::createBinary(llvm::MemoryBufferRef(
176       toStringRef(data_sp->GetData()), file.GetFilename().GetStringRef()));
177 
178   if (!binary) {
179     LLDB_LOG_ERROR(log, binary.takeError(),
180                    "Failed to create binary for file ({1}): {0}", file);
181     return initial_count;
182   }
183 
184   auto *COFFObj = llvm::dyn_cast<llvm::object::COFFObjectFile>(binary->get());
185   if (!COFFObj)
186     return initial_count;
187 
188   ModuleSpec module_spec(file);
189   ArchSpec &spec = module_spec.GetArchitecture();
190   lldb_private::UUID &uuid = module_spec.GetUUID();
191   if (!uuid.IsValid())
192     uuid = GetCoffUUID(*COFFObj);
193 
194   switch (COFFObj->getMachine()) {
195   case MachineAmd64:
196     spec.SetTriple("x86_64-pc-windows");
197     specs.Append(module_spec);
198     break;
199   case MachineX86:
200     spec.SetTriple("i386-pc-windows");
201     specs.Append(module_spec);
202     spec.SetTriple("i686-pc-windows");
203     specs.Append(module_spec);
204     break;
205   case MachineArmNt:
206     spec.SetTriple("armv7-pc-windows");
207     specs.Append(module_spec);
208     break;
209   case MachineArm64:
210     spec.SetTriple("aarch64-pc-windows");
211     specs.Append(module_spec);
212     break;
213   default:
214     break;
215   }
216 
217   return specs.GetSize() - initial_count;
218 }
219 
220 bool ObjectFilePECOFF::SaveCore(const lldb::ProcessSP &process_sp,
221                                 const lldb_private::FileSpec &outfile,
222                                 lldb_private::Status &error) {
223   return SaveMiniDump(process_sp, outfile, error);
224 }
225 
226 bool ObjectFilePECOFF::MagicBytesMatch(DataBufferSP &data_sp) {
227   DataExtractor data(data_sp, eByteOrderLittle, 4);
228   lldb::offset_t offset = 0;
229   uint16_t magic = data.GetU16(&offset);
230   return magic == IMAGE_DOS_SIGNATURE;
231 }
232 
233 lldb::SymbolType ObjectFilePECOFF::MapSymbolType(uint16_t coff_symbol_type) {
234   // TODO:  We need to complete this mapping of COFF symbol types to LLDB ones.
235   // For now, here's a hack to make sure our function have types.
236   const auto complex_type =
237       coff_symbol_type >> llvm::COFF::SCT_COMPLEX_TYPE_SHIFT;
238   if (complex_type == llvm::COFF::IMAGE_SYM_DTYPE_FUNCTION) {
239     return lldb::eSymbolTypeCode;
240   }
241   return lldb::eSymbolTypeInvalid;
242 }
243 
244 bool ObjectFilePECOFF::CreateBinary() {
245   if (m_binary)
246     return true;
247 
248   Log *log(GetLogIfAllCategoriesSet(LIBLLDB_LOG_OBJECT));
249 
250   auto binary = llvm::object::createBinary(llvm::MemoryBufferRef(
251       toStringRef(m_data.GetData()), m_file.GetFilename().GetStringRef()));
252   if (!binary) {
253     LLDB_LOG_ERROR(log, binary.takeError(),
254                    "Failed to create binary for file ({1}): {0}", m_file);
255     return false;
256   }
257 
258   // Make sure we only handle COFF format.
259   m_binary =
260       llvm::unique_dyn_cast<llvm::object::COFFObjectFile>(std::move(*binary));
261   if (!m_binary)
262     return false;
263 
264   LLDB_LOG(log, "this = {0}, module = {1} ({2}), file = {3}, binary = {4}",
265            this, GetModule().get(), GetModule()->GetSpecificationDescription(),
266            m_file.GetPath(), m_binary.get());
267   return true;
268 }
269 
270 ObjectFilePECOFF::ObjectFilePECOFF(const lldb::ModuleSP &module_sp,
271                                    DataBufferSP &data_sp,
272                                    lldb::offset_t data_offset,
273                                    const FileSpec *file,
274                                    lldb::offset_t file_offset,
275                                    lldb::offset_t length)
276     : ObjectFile(module_sp, file, file_offset, length, data_sp, data_offset),
277       m_dos_header(), m_coff_header(), m_sect_headers(),
278       m_entry_point_address(), m_deps_filespec() {
279   ::memset(&m_dos_header, 0, sizeof(m_dos_header));
280   ::memset(&m_coff_header, 0, sizeof(m_coff_header));
281 }
282 
283 ObjectFilePECOFF::ObjectFilePECOFF(const lldb::ModuleSP &module_sp,
284                                    DataBufferSP &header_data_sp,
285                                    const lldb::ProcessSP &process_sp,
286                                    addr_t header_addr)
287     : ObjectFile(module_sp, process_sp, header_addr, header_data_sp),
288       m_dos_header(), m_coff_header(), m_sect_headers(),
289       m_entry_point_address(), m_deps_filespec() {
290   ::memset(&m_dos_header, 0, sizeof(m_dos_header));
291   ::memset(&m_coff_header, 0, sizeof(m_coff_header));
292 }
293 
294 ObjectFilePECOFF::~ObjectFilePECOFF() {}
295 
296 bool ObjectFilePECOFF::ParseHeader() {
297   ModuleSP module_sp(GetModule());
298   if (module_sp) {
299     std::lock_guard<std::recursive_mutex> guard(module_sp->GetMutex());
300     m_sect_headers.clear();
301     m_data.SetByteOrder(eByteOrderLittle);
302     lldb::offset_t offset = 0;
303 
304     if (ParseDOSHeader(m_data, m_dos_header)) {
305       offset = m_dos_header.e_lfanew;
306       uint32_t pe_signature = m_data.GetU32(&offset);
307       if (pe_signature != IMAGE_NT_SIGNATURE)
308         return false;
309       if (ParseCOFFHeader(m_data, &offset, m_coff_header)) {
310         if (m_coff_header.hdrsize > 0)
311           ParseCOFFOptionalHeader(&offset);
312         ParseSectionHeaders(offset);
313       }
314       m_data.SetAddressByteSize(GetAddressByteSize());
315       return true;
316     }
317   }
318   return false;
319 }
320 
321 bool ObjectFilePECOFF::SetLoadAddress(Target &target, addr_t value,
322                                       bool value_is_offset) {
323   bool changed = false;
324   ModuleSP module_sp = GetModule();
325   if (module_sp) {
326     size_t num_loaded_sections = 0;
327     SectionList *section_list = GetSectionList();
328     if (section_list) {
329       if (!value_is_offset) {
330         value -= m_image_base;
331       }
332 
333       const size_t num_sections = section_list->GetSize();
334       size_t sect_idx = 0;
335 
336       for (sect_idx = 0; sect_idx < num_sections; ++sect_idx) {
337         // Iterate through the object file sections to find all of the sections
338         // that have SHF_ALLOC in their flag bits.
339         SectionSP section_sp(section_list->GetSectionAtIndex(sect_idx));
340         if (section_sp && !section_sp->IsThreadSpecific()) {
341           if (target.GetSectionLoadList().SetSectionLoadAddress(
342                   section_sp, section_sp->GetFileAddress() + value))
343             ++num_loaded_sections;
344         }
345       }
346       changed = num_loaded_sections > 0;
347     }
348   }
349   return changed;
350 }
351 
352 ByteOrder ObjectFilePECOFF::GetByteOrder() const { return eByteOrderLittle; }
353 
354 bool ObjectFilePECOFF::IsExecutable() const {
355   return (m_coff_header.flags & llvm::COFF::IMAGE_FILE_DLL) == 0;
356 }
357 
358 uint32_t ObjectFilePECOFF::GetAddressByteSize() const {
359   if (m_coff_header_opt.magic == OPT_HEADER_MAGIC_PE32_PLUS)
360     return 8;
361   else if (m_coff_header_opt.magic == OPT_HEADER_MAGIC_PE32)
362     return 4;
363   return 4;
364 }
365 
366 // NeedsEndianSwap
367 //
368 // Return true if an endian swap needs to occur when extracting data from this
369 // file.
370 bool ObjectFilePECOFF::NeedsEndianSwap() const {
371 #if defined(__LITTLE_ENDIAN__)
372   return false;
373 #else
374   return true;
375 #endif
376 }
377 // ParseDOSHeader
378 bool ObjectFilePECOFF::ParseDOSHeader(DataExtractor &data,
379                                       dos_header_t &dos_header) {
380   bool success = false;
381   lldb::offset_t offset = 0;
382   success = data.ValidOffsetForDataOfSize(0, sizeof(dos_header));
383 
384   if (success) {
385     dos_header.e_magic = data.GetU16(&offset); // Magic number
386     success = dos_header.e_magic == IMAGE_DOS_SIGNATURE;
387 
388     if (success) {
389       dos_header.e_cblp = data.GetU16(&offset); // Bytes on last page of file
390       dos_header.e_cp = data.GetU16(&offset);   // Pages in file
391       dos_header.e_crlc = data.GetU16(&offset); // Relocations
392       dos_header.e_cparhdr =
393           data.GetU16(&offset); // Size of header in paragraphs
394       dos_header.e_minalloc =
395           data.GetU16(&offset); // Minimum extra paragraphs needed
396       dos_header.e_maxalloc =
397           data.GetU16(&offset);               // Maximum extra paragraphs needed
398       dos_header.e_ss = data.GetU16(&offset); // Initial (relative) SS value
399       dos_header.e_sp = data.GetU16(&offset); // Initial SP value
400       dos_header.e_csum = data.GetU16(&offset); // Checksum
401       dos_header.e_ip = data.GetU16(&offset);   // Initial IP value
402       dos_header.e_cs = data.GetU16(&offset);   // Initial (relative) CS value
403       dos_header.e_lfarlc =
404           data.GetU16(&offset); // File address of relocation table
405       dos_header.e_ovno = data.GetU16(&offset); // Overlay number
406 
407       dos_header.e_res[0] = data.GetU16(&offset); // Reserved words
408       dos_header.e_res[1] = data.GetU16(&offset); // Reserved words
409       dos_header.e_res[2] = data.GetU16(&offset); // Reserved words
410       dos_header.e_res[3] = data.GetU16(&offset); // Reserved words
411 
412       dos_header.e_oemid =
413           data.GetU16(&offset); // OEM identifier (for e_oeminfo)
414       dos_header.e_oeminfo =
415           data.GetU16(&offset); // OEM information; e_oemid specific
416       dos_header.e_res2[0] = data.GetU16(&offset); // Reserved words
417       dos_header.e_res2[1] = data.GetU16(&offset); // Reserved words
418       dos_header.e_res2[2] = data.GetU16(&offset); // Reserved words
419       dos_header.e_res2[3] = data.GetU16(&offset); // Reserved words
420       dos_header.e_res2[4] = data.GetU16(&offset); // Reserved words
421       dos_header.e_res2[5] = data.GetU16(&offset); // Reserved words
422       dos_header.e_res2[6] = data.GetU16(&offset); // Reserved words
423       dos_header.e_res2[7] = data.GetU16(&offset); // Reserved words
424       dos_header.e_res2[8] = data.GetU16(&offset); // Reserved words
425       dos_header.e_res2[9] = data.GetU16(&offset); // Reserved words
426 
427       dos_header.e_lfanew =
428           data.GetU32(&offset); // File address of new exe header
429     }
430   }
431   if (!success)
432     memset(&dos_header, 0, sizeof(dos_header));
433   return success;
434 }
435 
436 // ParserCOFFHeader
437 bool ObjectFilePECOFF::ParseCOFFHeader(DataExtractor &data,
438                                        lldb::offset_t *offset_ptr,
439                                        coff_header_t &coff_header) {
440   bool success =
441       data.ValidOffsetForDataOfSize(*offset_ptr, sizeof(coff_header));
442   if (success) {
443     coff_header.machine = data.GetU16(offset_ptr);
444     coff_header.nsects = data.GetU16(offset_ptr);
445     coff_header.modtime = data.GetU32(offset_ptr);
446     coff_header.symoff = data.GetU32(offset_ptr);
447     coff_header.nsyms = data.GetU32(offset_ptr);
448     coff_header.hdrsize = data.GetU16(offset_ptr);
449     coff_header.flags = data.GetU16(offset_ptr);
450   }
451   if (!success)
452     memset(&coff_header, 0, sizeof(coff_header));
453   return success;
454 }
455 
456 bool ObjectFilePECOFF::ParseCOFFOptionalHeader(lldb::offset_t *offset_ptr) {
457   bool success = false;
458   const lldb::offset_t end_offset = *offset_ptr + m_coff_header.hdrsize;
459   if (*offset_ptr < end_offset) {
460     success = true;
461     m_coff_header_opt.magic = m_data.GetU16(offset_ptr);
462     m_coff_header_opt.major_linker_version = m_data.GetU8(offset_ptr);
463     m_coff_header_opt.minor_linker_version = m_data.GetU8(offset_ptr);
464     m_coff_header_opt.code_size = m_data.GetU32(offset_ptr);
465     m_coff_header_opt.data_size = m_data.GetU32(offset_ptr);
466     m_coff_header_opt.bss_size = m_data.GetU32(offset_ptr);
467     m_coff_header_opt.entry = m_data.GetU32(offset_ptr);
468     m_coff_header_opt.code_offset = m_data.GetU32(offset_ptr);
469 
470     const uint32_t addr_byte_size = GetAddressByteSize();
471 
472     if (*offset_ptr < end_offset) {
473       if (m_coff_header_opt.magic == OPT_HEADER_MAGIC_PE32) {
474         // PE32 only
475         m_coff_header_opt.data_offset = m_data.GetU32(offset_ptr);
476       } else
477         m_coff_header_opt.data_offset = 0;
478 
479       if (*offset_ptr < end_offset) {
480         m_coff_header_opt.image_base =
481             m_data.GetMaxU64(offset_ptr, addr_byte_size);
482         m_coff_header_opt.sect_alignment = m_data.GetU32(offset_ptr);
483         m_coff_header_opt.file_alignment = m_data.GetU32(offset_ptr);
484         m_coff_header_opt.major_os_system_version = m_data.GetU16(offset_ptr);
485         m_coff_header_opt.minor_os_system_version = m_data.GetU16(offset_ptr);
486         m_coff_header_opt.major_image_version = m_data.GetU16(offset_ptr);
487         m_coff_header_opt.minor_image_version = m_data.GetU16(offset_ptr);
488         m_coff_header_opt.major_subsystem_version = m_data.GetU16(offset_ptr);
489         m_coff_header_opt.minor_subsystem_version = m_data.GetU16(offset_ptr);
490         m_coff_header_opt.reserved1 = m_data.GetU32(offset_ptr);
491         m_coff_header_opt.image_size = m_data.GetU32(offset_ptr);
492         m_coff_header_opt.header_size = m_data.GetU32(offset_ptr);
493         m_coff_header_opt.checksum = m_data.GetU32(offset_ptr);
494         m_coff_header_opt.subsystem = m_data.GetU16(offset_ptr);
495         m_coff_header_opt.dll_flags = m_data.GetU16(offset_ptr);
496         m_coff_header_opt.stack_reserve_size =
497             m_data.GetMaxU64(offset_ptr, addr_byte_size);
498         m_coff_header_opt.stack_commit_size =
499             m_data.GetMaxU64(offset_ptr, addr_byte_size);
500         m_coff_header_opt.heap_reserve_size =
501             m_data.GetMaxU64(offset_ptr, addr_byte_size);
502         m_coff_header_opt.heap_commit_size =
503             m_data.GetMaxU64(offset_ptr, addr_byte_size);
504         m_coff_header_opt.loader_flags = m_data.GetU32(offset_ptr);
505         uint32_t num_data_dir_entries = m_data.GetU32(offset_ptr);
506         m_coff_header_opt.data_dirs.clear();
507         m_coff_header_opt.data_dirs.resize(num_data_dir_entries);
508         uint32_t i;
509         for (i = 0; i < num_data_dir_entries; i++) {
510           m_coff_header_opt.data_dirs[i].vmaddr = m_data.GetU32(offset_ptr);
511           m_coff_header_opt.data_dirs[i].vmsize = m_data.GetU32(offset_ptr);
512         }
513 
514         m_image_base = m_coff_header_opt.image_base;
515       }
516     }
517   }
518   // Make sure we are on track for section data which follows
519   *offset_ptr = end_offset;
520   return success;
521 }
522 
523 uint32_t ObjectFilePECOFF::GetRVA(const Address &addr) const {
524   return addr.GetFileAddress() - m_image_base;
525 }
526 
527 Address ObjectFilePECOFF::GetAddress(uint32_t rva) {
528   SectionList *sect_list = GetSectionList();
529   if (!sect_list)
530     return Address(GetFileAddress(rva));
531 
532   return Address(GetFileAddress(rva), sect_list);
533 }
534 
535 lldb::addr_t ObjectFilePECOFF::GetFileAddress(uint32_t rva) const {
536   return m_image_base + rva;
537 }
538 
539 DataExtractor ObjectFilePECOFF::ReadImageData(uint32_t offset, size_t size) {
540   if (!size)
541     return {};
542 
543   if (m_data.ValidOffsetForDataOfSize(offset, size))
544     return DataExtractor(m_data, offset, size);
545 
546   ProcessSP process_sp(m_process_wp.lock());
547   DataExtractor data;
548   if (process_sp) {
549     auto data_up = std::make_unique<DataBufferHeap>(size, 0);
550     Status readmem_error;
551     size_t bytes_read =
552         process_sp->ReadMemory(m_image_base + offset, data_up->GetBytes(),
553                                data_up->GetByteSize(), readmem_error);
554     if (bytes_read == size) {
555       DataBufferSP buffer_sp(data_up.release());
556       data.SetData(buffer_sp, 0, buffer_sp->GetByteSize());
557     }
558   }
559   return data;
560 }
561 
562 DataExtractor ObjectFilePECOFF::ReadImageDataByRVA(uint32_t rva, size_t size) {
563   Address addr = GetAddress(rva);
564   SectionSP sect = addr.GetSection();
565   if (!sect)
566     return {};
567   rva = sect->GetFileOffset() + addr.GetOffset();
568 
569   return ReadImageData(rva, size);
570 }
571 
572 // ParseSectionHeaders
573 bool ObjectFilePECOFF::ParseSectionHeaders(
574     uint32_t section_header_data_offset) {
575   const uint32_t nsects = m_coff_header.nsects;
576   m_sect_headers.clear();
577 
578   if (nsects > 0) {
579     const size_t section_header_byte_size = nsects * sizeof(section_header_t);
580     DataExtractor section_header_data =
581         ReadImageData(section_header_data_offset, section_header_byte_size);
582 
583     lldb::offset_t offset = 0;
584     if (section_header_data.ValidOffsetForDataOfSize(
585             offset, section_header_byte_size)) {
586       m_sect_headers.resize(nsects);
587 
588       for (uint32_t idx = 0; idx < nsects; ++idx) {
589         const void *name_data = section_header_data.GetData(&offset, 8);
590         if (name_data) {
591           memcpy(m_sect_headers[idx].name, name_data, 8);
592           m_sect_headers[idx].vmsize = section_header_data.GetU32(&offset);
593           m_sect_headers[idx].vmaddr = section_header_data.GetU32(&offset);
594           m_sect_headers[idx].size = section_header_data.GetU32(&offset);
595           m_sect_headers[idx].offset = section_header_data.GetU32(&offset);
596           m_sect_headers[idx].reloff = section_header_data.GetU32(&offset);
597           m_sect_headers[idx].lineoff = section_header_data.GetU32(&offset);
598           m_sect_headers[idx].nreloc = section_header_data.GetU16(&offset);
599           m_sect_headers[idx].nline = section_header_data.GetU16(&offset);
600           m_sect_headers[idx].flags = section_header_data.GetU32(&offset);
601         }
602       }
603     }
604   }
605 
606   return !m_sect_headers.empty();
607 }
608 
609 llvm::StringRef ObjectFilePECOFF::GetSectionName(const section_header_t &sect) {
610   llvm::StringRef hdr_name(sect.name, llvm::array_lengthof(sect.name));
611   hdr_name = hdr_name.split('\0').first;
612   if (hdr_name.consume_front("/")) {
613     lldb::offset_t stroff;
614     if (!to_integer(hdr_name, stroff, 10))
615       return "";
616     lldb::offset_t string_file_offset =
617         m_coff_header.symoff + (m_coff_header.nsyms * 18) + stroff;
618     if (const char *name = m_data.GetCStr(&string_file_offset))
619       return name;
620     return "";
621   }
622   return hdr_name;
623 }
624 
625 // GetNListSymtab
626 Symtab *ObjectFilePECOFF::GetSymtab() {
627   ModuleSP module_sp(GetModule());
628   if (module_sp) {
629     std::lock_guard<std::recursive_mutex> guard(module_sp->GetMutex());
630     if (m_symtab_up == nullptr) {
631       SectionList *sect_list = GetSectionList();
632       m_symtab_up = std::make_unique<Symtab>(this);
633       std::lock_guard<std::recursive_mutex> guard(m_symtab_up->GetMutex());
634 
635       const uint32_t num_syms = m_coff_header.nsyms;
636 
637       if (m_file && num_syms > 0 && m_coff_header.symoff > 0) {
638         const uint32_t symbol_size = 18;
639         const size_t symbol_data_size = num_syms * symbol_size;
640         // Include the 4-byte string table size at the end of the symbols
641         DataExtractor symtab_data =
642             ReadImageData(m_coff_header.symoff, symbol_data_size + 4);
643         lldb::offset_t offset = symbol_data_size;
644         const uint32_t strtab_size = symtab_data.GetU32(&offset);
645         if (strtab_size > 0) {
646           DataExtractor strtab_data = ReadImageData(
647               m_coff_header.symoff + symbol_data_size, strtab_size);
648 
649           offset = 0;
650           std::string symbol_name;
651           Symbol *symbols = m_symtab_up->Resize(num_syms);
652           for (uint32_t i = 0; i < num_syms; ++i) {
653             coff_symbol_t symbol;
654             const uint32_t symbol_offset = offset;
655             const char *symbol_name_cstr = nullptr;
656             // If the first 4 bytes of the symbol string are zero, then they
657             // are followed by a 4-byte string table offset. Else these
658             // 8 bytes contain the symbol name
659             if (symtab_data.GetU32(&offset) == 0) {
660               // Long string that doesn't fit into the symbol table name, so
661               // now we must read the 4 byte string table offset
662               uint32_t strtab_offset = symtab_data.GetU32(&offset);
663               symbol_name_cstr = strtab_data.PeekCStr(strtab_offset);
664               symbol_name.assign(symbol_name_cstr);
665             } else {
666               // Short string that fits into the symbol table name which is 8
667               // bytes
668               offset += sizeof(symbol.name) - 4; // Skip remaining
669               symbol_name_cstr = symtab_data.PeekCStr(symbol_offset);
670               if (symbol_name_cstr == nullptr)
671                 break;
672               symbol_name.assign(symbol_name_cstr, sizeof(symbol.name));
673             }
674             symbol.value = symtab_data.GetU32(&offset);
675             symbol.sect = symtab_data.GetU16(&offset);
676             symbol.type = symtab_data.GetU16(&offset);
677             symbol.storage = symtab_data.GetU8(&offset);
678             symbol.naux = symtab_data.GetU8(&offset);
679             symbols[i].GetMangled().SetValue(ConstString(symbol_name.c_str()));
680             if ((int16_t)symbol.sect >= 1) {
681               Address symbol_addr(sect_list->FindSectionByID(symbol.sect),
682                                   symbol.value);
683               symbols[i].GetAddressRef() = symbol_addr;
684               symbols[i].SetType(MapSymbolType(symbol.type));
685             }
686 
687             if (symbol.naux > 0) {
688               i += symbol.naux;
689               offset += symbol.naux * symbol_size;
690             }
691           }
692         }
693       }
694 
695       // Read export header
696       if (coff_data_dir_export_table < m_coff_header_opt.data_dirs.size() &&
697           m_coff_header_opt.data_dirs[coff_data_dir_export_table].vmsize > 0 &&
698           m_coff_header_opt.data_dirs[coff_data_dir_export_table].vmaddr > 0) {
699         export_directory_entry export_table;
700         uint32_t data_start =
701             m_coff_header_opt.data_dirs[coff_data_dir_export_table].vmaddr;
702 
703         DataExtractor symtab_data = ReadImageDataByRVA(
704             data_start, m_coff_header_opt.data_dirs[0].vmsize);
705         lldb::offset_t offset = 0;
706 
707         // Read export_table header
708         export_table.characteristics = symtab_data.GetU32(&offset);
709         export_table.time_date_stamp = symtab_data.GetU32(&offset);
710         export_table.major_version = symtab_data.GetU16(&offset);
711         export_table.minor_version = symtab_data.GetU16(&offset);
712         export_table.name = symtab_data.GetU32(&offset);
713         export_table.base = symtab_data.GetU32(&offset);
714         export_table.number_of_functions = symtab_data.GetU32(&offset);
715         export_table.number_of_names = symtab_data.GetU32(&offset);
716         export_table.address_of_functions = symtab_data.GetU32(&offset);
717         export_table.address_of_names = symtab_data.GetU32(&offset);
718         export_table.address_of_name_ordinals = symtab_data.GetU32(&offset);
719 
720         bool has_ordinal = export_table.address_of_name_ordinals != 0;
721 
722         lldb::offset_t name_offset = export_table.address_of_names - data_start;
723         lldb::offset_t name_ordinal_offset =
724             export_table.address_of_name_ordinals - data_start;
725 
726         Symbol *symbols = m_symtab_up->Resize(export_table.number_of_names);
727 
728         std::string symbol_name;
729 
730         // Read each export table entry
731         for (size_t i = 0; i < export_table.number_of_names; ++i) {
732           uint32_t name_ordinal =
733               has_ordinal ? symtab_data.GetU16(&name_ordinal_offset) : i;
734           uint32_t name_address = symtab_data.GetU32(&name_offset);
735 
736           const char *symbol_name_cstr =
737               symtab_data.PeekCStr(name_address - data_start);
738           symbol_name.assign(symbol_name_cstr);
739 
740           lldb::offset_t function_offset = export_table.address_of_functions -
741                                            data_start +
742                                            sizeof(uint32_t) * name_ordinal;
743           uint32_t function_rva = symtab_data.GetU32(&function_offset);
744 
745           Address symbol_addr(m_coff_header_opt.image_base + function_rva,
746                               sect_list);
747           symbols[i].GetMangled().SetValue(ConstString(symbol_name.c_str()));
748           symbols[i].GetAddressRef() = symbol_addr;
749           symbols[i].SetType(lldb::eSymbolTypeCode);
750           symbols[i].SetDebug(true);
751         }
752       }
753       m_symtab_up->CalculateSymbolSizes();
754     }
755   }
756   return m_symtab_up.get();
757 }
758 
759 std::unique_ptr<CallFrameInfo> ObjectFilePECOFF::CreateCallFrameInfo() {
760   if (coff_data_dir_exception_table >= m_coff_header_opt.data_dirs.size())
761     return {};
762 
763   data_directory data_dir_exception =
764       m_coff_header_opt.data_dirs[coff_data_dir_exception_table];
765   if (!data_dir_exception.vmaddr)
766     return {};
767 
768   if (m_coff_header.machine != llvm::COFF::IMAGE_FILE_MACHINE_AMD64)
769     return {};
770 
771   return std::make_unique<PECallFrameInfo>(*this, data_dir_exception.vmaddr,
772                                            data_dir_exception.vmsize);
773 }
774 
775 bool ObjectFilePECOFF::IsStripped() {
776   // TODO: determine this for COFF
777   return false;
778 }
779 
780 SectionType ObjectFilePECOFF::GetSectionType(llvm::StringRef sect_name,
781                                              const section_header_t &sect) {
782   ConstString const_sect_name(sect_name);
783   static ConstString g_code_sect_name(".code");
784   static ConstString g_CODE_sect_name("CODE");
785   static ConstString g_data_sect_name(".data");
786   static ConstString g_DATA_sect_name("DATA");
787   static ConstString g_bss_sect_name(".bss");
788   static ConstString g_BSS_sect_name("BSS");
789 
790   if (sect.flags & llvm::COFF::IMAGE_SCN_CNT_CODE &&
791       ((const_sect_name == g_code_sect_name) ||
792        (const_sect_name == g_CODE_sect_name))) {
793     return eSectionTypeCode;
794   }
795   if (sect.flags & llvm::COFF::IMAGE_SCN_CNT_INITIALIZED_DATA &&
796              ((const_sect_name == g_data_sect_name) ||
797               (const_sect_name == g_DATA_sect_name))) {
798     if (sect.size == 0 && sect.offset == 0)
799       return eSectionTypeZeroFill;
800     else
801       return eSectionTypeData;
802   }
803   if (sect.flags & llvm::COFF::IMAGE_SCN_CNT_UNINITIALIZED_DATA &&
804              ((const_sect_name == g_bss_sect_name) ||
805               (const_sect_name == g_BSS_sect_name))) {
806     if (sect.size == 0)
807       return eSectionTypeZeroFill;
808     else
809       return eSectionTypeData;
810   }
811 
812   SectionType section_type =
813       llvm::StringSwitch<SectionType>(sect_name)
814           .Case(".debug", eSectionTypeDebug)
815           .Case(".stabstr", eSectionTypeDataCString)
816           .Case(".reloc", eSectionTypeOther)
817           .Case(".debug_abbrev", eSectionTypeDWARFDebugAbbrev)
818           .Case(".debug_aranges", eSectionTypeDWARFDebugAranges)
819           .Case(".debug_frame", eSectionTypeDWARFDebugFrame)
820           .Case(".debug_info", eSectionTypeDWARFDebugInfo)
821           .Case(".debug_line", eSectionTypeDWARFDebugLine)
822           .Case(".debug_loc", eSectionTypeDWARFDebugLoc)
823           .Case(".debug_loclists", eSectionTypeDWARFDebugLocLists)
824           .Case(".debug_macinfo", eSectionTypeDWARFDebugMacInfo)
825           .Case(".debug_names", eSectionTypeDWARFDebugNames)
826           .Case(".debug_pubnames", eSectionTypeDWARFDebugPubNames)
827           .Case(".debug_pubtypes", eSectionTypeDWARFDebugPubTypes)
828           .Case(".debug_ranges", eSectionTypeDWARFDebugRanges)
829           .Case(".debug_str", eSectionTypeDWARFDebugStr)
830           .Case(".debug_types", eSectionTypeDWARFDebugTypes)
831           // .eh_frame can be truncated to 8 chars.
832           .Cases(".eh_frame", ".eh_fram", eSectionTypeEHFrame)
833           .Case(".gosymtab", eSectionTypeGoSymtab)
834           .Default(eSectionTypeInvalid);
835   if (section_type != eSectionTypeInvalid)
836     return section_type;
837 
838   if (sect.flags & llvm::COFF::IMAGE_SCN_CNT_CODE)
839     return eSectionTypeCode;
840   if (sect.flags & llvm::COFF::IMAGE_SCN_CNT_INITIALIZED_DATA)
841     return eSectionTypeData;
842   if (sect.flags & llvm::COFF::IMAGE_SCN_CNT_UNINITIALIZED_DATA) {
843     if (sect.size == 0)
844       return eSectionTypeZeroFill;
845     else
846       return eSectionTypeData;
847   }
848   return eSectionTypeOther;
849 }
850 
851 void ObjectFilePECOFF::CreateSections(SectionList &unified_section_list) {
852   if (m_sections_up)
853     return;
854   m_sections_up = std::make_unique<SectionList>();
855 
856   ModuleSP module_sp(GetModule());
857   if (module_sp) {
858     std::lock_guard<std::recursive_mutex> guard(module_sp->GetMutex());
859 
860     SectionSP header_sp = std::make_shared<Section>(
861         module_sp, this, ~user_id_t(0), ConstString("PECOFF header"),
862         eSectionTypeOther, m_coff_header_opt.image_base,
863         m_coff_header_opt.header_size,
864         /*file_offset*/ 0, m_coff_header_opt.header_size,
865         m_coff_header_opt.sect_alignment,
866         /*flags*/ 0);
867     header_sp->SetPermissions(ePermissionsReadable);
868     m_sections_up->AddSection(header_sp);
869     unified_section_list.AddSection(header_sp);
870 
871     const uint32_t nsects = m_sect_headers.size();
872     ModuleSP module_sp(GetModule());
873     for (uint32_t idx = 0; idx < nsects; ++idx) {
874       llvm::StringRef sect_name = GetSectionName(m_sect_headers[idx]);
875       ConstString const_sect_name(sect_name);
876       SectionType section_type = GetSectionType(sect_name, m_sect_headers[idx]);
877 
878       SectionSP section_sp(new Section(
879           module_sp,       // Module to which this section belongs
880           this,            // Object file to which this section belongs
881           idx + 1,         // Section ID is the 1 based section index.
882           const_sect_name, // Name of this section
883           section_type,
884           m_coff_header_opt.image_base +
885               m_sect_headers[idx].vmaddr, // File VM address == addresses as
886                                           // they are found in the object file
887           m_sect_headers[idx].vmsize,     // VM size in bytes of this section
888           m_sect_headers[idx]
889               .offset, // Offset to the data for this section in the file
890           m_sect_headers[idx]
891               .size, // Size in bytes of this section as found in the file
892           m_coff_header_opt.sect_alignment, // Section alignment
893           m_sect_headers[idx].flags));      // Flags for this section
894 
895       uint32_t permissions = 0;
896       if (m_sect_headers[idx].flags & llvm::COFF::IMAGE_SCN_MEM_EXECUTE)
897         permissions |= ePermissionsExecutable;
898       if (m_sect_headers[idx].flags & llvm::COFF::IMAGE_SCN_MEM_READ)
899         permissions |= ePermissionsReadable;
900       if (m_sect_headers[idx].flags & llvm::COFF::IMAGE_SCN_MEM_WRITE)
901         permissions |= ePermissionsWritable;
902       section_sp->SetPermissions(permissions);
903 
904       m_sections_up->AddSection(section_sp);
905       unified_section_list.AddSection(section_sp);
906     }
907   }
908 }
909 
910 UUID ObjectFilePECOFF::GetUUID() {
911   if (m_uuid.IsValid())
912     return m_uuid;
913 
914   if (!CreateBinary())
915     return UUID();
916 
917   m_uuid = GetCoffUUID(*m_binary);
918   return m_uuid;
919 }
920 
921 uint32_t ObjectFilePECOFF::ParseDependentModules() {
922   ModuleSP module_sp(GetModule());
923   if (!module_sp)
924     return 0;
925 
926   std::lock_guard<std::recursive_mutex> guard(module_sp->GetMutex());
927   if (m_deps_filespec)
928     return m_deps_filespec->GetSize();
929 
930   // Cache coff binary if it is not done yet.
931   if (!CreateBinary())
932     return 0;
933 
934   Log *log(GetLogIfAllCategoriesSet(LIBLLDB_LOG_OBJECT));
935   LLDB_LOG(log, "this = {0}, module = {1} ({2}), file = {3}, binary = {4}",
936            this, GetModule().get(), GetModule()->GetSpecificationDescription(),
937            m_file.GetPath(), m_binary.get());
938 
939   m_deps_filespec = FileSpecList();
940 
941   for (const auto &entry : m_binary->import_directories()) {
942     llvm::StringRef dll_name;
943     // Report a bogus entry.
944     if (llvm::Error e = entry.getName(dll_name)) {
945       LLDB_LOGF(log,
946                 "ObjectFilePECOFF::ParseDependentModules() - failed to get "
947                 "import directory entry name: %s",
948                 llvm::toString(std::move(e)).c_str());
949       continue;
950     }
951 
952     // At this moment we only have the base name of the DLL. The full path can
953     // only be seen after the dynamic loading.  Our best guess is Try to get it
954     // with the help of the object file's directory.
955     llvm::SmallString<128> dll_fullpath;
956     FileSpec dll_specs(dll_name);
957     dll_specs.GetDirectory().SetString(m_file.GetDirectory().GetCString());
958 
959     if (!llvm::sys::fs::real_path(dll_specs.GetPath(), dll_fullpath))
960       m_deps_filespec->EmplaceBack(dll_fullpath);
961     else {
962       // Known DLLs or DLL not found in the object file directory.
963       m_deps_filespec->EmplaceBack(dll_name);
964     }
965   }
966   return m_deps_filespec->GetSize();
967 }
968 
969 uint32_t ObjectFilePECOFF::GetDependentModules(FileSpecList &files) {
970   auto num_modules = ParseDependentModules();
971   auto original_size = files.GetSize();
972 
973   for (unsigned i = 0; i < num_modules; ++i)
974     files.AppendIfUnique(m_deps_filespec->GetFileSpecAtIndex(i));
975 
976   return files.GetSize() - original_size;
977 }
978 
979 lldb_private::Address ObjectFilePECOFF::GetEntryPointAddress() {
980   if (m_entry_point_address.IsValid())
981     return m_entry_point_address;
982 
983   if (!ParseHeader() || !IsExecutable())
984     return m_entry_point_address;
985 
986   SectionList *section_list = GetSectionList();
987   addr_t file_addr = m_coff_header_opt.entry + m_coff_header_opt.image_base;
988 
989   if (!section_list)
990     m_entry_point_address.SetOffset(file_addr);
991   else
992     m_entry_point_address.ResolveAddressUsingFileSections(file_addr,
993                                                           section_list);
994   return m_entry_point_address;
995 }
996 
997 Address ObjectFilePECOFF::GetBaseAddress() {
998   return Address(GetSectionList()->GetSectionAtIndex(0), 0);
999 }
1000 
1001 // Dump
1002 //
1003 // Dump the specifics of the runtime file container (such as any headers
1004 // segments, sections, etc).
1005 void ObjectFilePECOFF::Dump(Stream *s) {
1006   ModuleSP module_sp(GetModule());
1007   if (module_sp) {
1008     std::lock_guard<std::recursive_mutex> guard(module_sp->GetMutex());
1009     s->Printf("%p: ", static_cast<void *>(this));
1010     s->Indent();
1011     s->PutCString("ObjectFilePECOFF");
1012 
1013     ArchSpec header_arch = GetArchitecture();
1014 
1015     *s << ", file = '" << m_file
1016        << "', arch = " << header_arch.GetArchitectureName() << "\n";
1017 
1018     SectionList *sections = GetSectionList();
1019     if (sections)
1020       sections->Dump(s->AsRawOstream(), s->GetIndentLevel(), nullptr, true,
1021                      UINT32_MAX);
1022 
1023     if (m_symtab_up)
1024       m_symtab_up->Dump(s, nullptr, eSortOrderNone);
1025 
1026     if (m_dos_header.e_magic)
1027       DumpDOSHeader(s, m_dos_header);
1028     if (m_coff_header.machine) {
1029       DumpCOFFHeader(s, m_coff_header);
1030       if (m_coff_header.hdrsize)
1031         DumpOptCOFFHeader(s, m_coff_header_opt);
1032     }
1033     s->EOL();
1034     DumpSectionHeaders(s);
1035     s->EOL();
1036 
1037     DumpDependentModules(s);
1038     s->EOL();
1039   }
1040 }
1041 
1042 // DumpDOSHeader
1043 //
1044 // Dump the MS-DOS header to the specified output stream
1045 void ObjectFilePECOFF::DumpDOSHeader(Stream *s, const dos_header_t &header) {
1046   s->PutCString("MSDOS Header\n");
1047   s->Printf("  e_magic    = 0x%4.4x\n", header.e_magic);
1048   s->Printf("  e_cblp     = 0x%4.4x\n", header.e_cblp);
1049   s->Printf("  e_cp       = 0x%4.4x\n", header.e_cp);
1050   s->Printf("  e_crlc     = 0x%4.4x\n", header.e_crlc);
1051   s->Printf("  e_cparhdr  = 0x%4.4x\n", header.e_cparhdr);
1052   s->Printf("  e_minalloc = 0x%4.4x\n", header.e_minalloc);
1053   s->Printf("  e_maxalloc = 0x%4.4x\n", header.e_maxalloc);
1054   s->Printf("  e_ss       = 0x%4.4x\n", header.e_ss);
1055   s->Printf("  e_sp       = 0x%4.4x\n", header.e_sp);
1056   s->Printf("  e_csum     = 0x%4.4x\n", header.e_csum);
1057   s->Printf("  e_ip       = 0x%4.4x\n", header.e_ip);
1058   s->Printf("  e_cs       = 0x%4.4x\n", header.e_cs);
1059   s->Printf("  e_lfarlc   = 0x%4.4x\n", header.e_lfarlc);
1060   s->Printf("  e_ovno     = 0x%4.4x\n", header.e_ovno);
1061   s->Printf("  e_res[4]   = { 0x%4.4x, 0x%4.4x, 0x%4.4x, 0x%4.4x }\n",
1062             header.e_res[0], header.e_res[1], header.e_res[2], header.e_res[3]);
1063   s->Printf("  e_oemid    = 0x%4.4x\n", header.e_oemid);
1064   s->Printf("  e_oeminfo  = 0x%4.4x\n", header.e_oeminfo);
1065   s->Printf("  e_res2[10] = { 0x%4.4x, 0x%4.4x, 0x%4.4x, 0x%4.4x, 0x%4.4x, "
1066             "0x%4.4x, 0x%4.4x, 0x%4.4x, 0x%4.4x, 0x%4.4x }\n",
1067             header.e_res2[0], header.e_res2[1], header.e_res2[2],
1068             header.e_res2[3], header.e_res2[4], header.e_res2[5],
1069             header.e_res2[6], header.e_res2[7], header.e_res2[8],
1070             header.e_res2[9]);
1071   s->Printf("  e_lfanew   = 0x%8.8x\n", header.e_lfanew);
1072 }
1073 
1074 // DumpCOFFHeader
1075 //
1076 // Dump the COFF header to the specified output stream
1077 void ObjectFilePECOFF::DumpCOFFHeader(Stream *s, const coff_header_t &header) {
1078   s->PutCString("COFF Header\n");
1079   s->Printf("  machine = 0x%4.4x\n", header.machine);
1080   s->Printf("  nsects  = 0x%4.4x\n", header.nsects);
1081   s->Printf("  modtime = 0x%8.8x\n", header.modtime);
1082   s->Printf("  symoff  = 0x%8.8x\n", header.symoff);
1083   s->Printf("  nsyms   = 0x%8.8x\n", header.nsyms);
1084   s->Printf("  hdrsize = 0x%4.4x\n", header.hdrsize);
1085 }
1086 
1087 // DumpOptCOFFHeader
1088 //
1089 // Dump the optional COFF header to the specified output stream
1090 void ObjectFilePECOFF::DumpOptCOFFHeader(Stream *s,
1091                                          const coff_opt_header_t &header) {
1092   s->PutCString("Optional COFF Header\n");
1093   s->Printf("  magic                   = 0x%4.4x\n", header.magic);
1094   s->Printf("  major_linker_version    = 0x%2.2x\n",
1095             header.major_linker_version);
1096   s->Printf("  minor_linker_version    = 0x%2.2x\n",
1097             header.minor_linker_version);
1098   s->Printf("  code_size               = 0x%8.8x\n", header.code_size);
1099   s->Printf("  data_size               = 0x%8.8x\n", header.data_size);
1100   s->Printf("  bss_size                = 0x%8.8x\n", header.bss_size);
1101   s->Printf("  entry                   = 0x%8.8x\n", header.entry);
1102   s->Printf("  code_offset             = 0x%8.8x\n", header.code_offset);
1103   s->Printf("  data_offset             = 0x%8.8x\n", header.data_offset);
1104   s->Printf("  image_base              = 0x%16.16" PRIx64 "\n",
1105             header.image_base);
1106   s->Printf("  sect_alignment          = 0x%8.8x\n", header.sect_alignment);
1107   s->Printf("  file_alignment          = 0x%8.8x\n", header.file_alignment);
1108   s->Printf("  major_os_system_version = 0x%4.4x\n",
1109             header.major_os_system_version);
1110   s->Printf("  minor_os_system_version = 0x%4.4x\n",
1111             header.minor_os_system_version);
1112   s->Printf("  major_image_version     = 0x%4.4x\n",
1113             header.major_image_version);
1114   s->Printf("  minor_image_version     = 0x%4.4x\n",
1115             header.minor_image_version);
1116   s->Printf("  major_subsystem_version = 0x%4.4x\n",
1117             header.major_subsystem_version);
1118   s->Printf("  minor_subsystem_version = 0x%4.4x\n",
1119             header.minor_subsystem_version);
1120   s->Printf("  reserved1               = 0x%8.8x\n", header.reserved1);
1121   s->Printf("  image_size              = 0x%8.8x\n", header.image_size);
1122   s->Printf("  header_size             = 0x%8.8x\n", header.header_size);
1123   s->Printf("  checksum                = 0x%8.8x\n", header.checksum);
1124   s->Printf("  subsystem               = 0x%4.4x\n", header.subsystem);
1125   s->Printf("  dll_flags               = 0x%4.4x\n", header.dll_flags);
1126   s->Printf("  stack_reserve_size      = 0x%16.16" PRIx64 "\n",
1127             header.stack_reserve_size);
1128   s->Printf("  stack_commit_size       = 0x%16.16" PRIx64 "\n",
1129             header.stack_commit_size);
1130   s->Printf("  heap_reserve_size       = 0x%16.16" PRIx64 "\n",
1131             header.heap_reserve_size);
1132   s->Printf("  heap_commit_size        = 0x%16.16" PRIx64 "\n",
1133             header.heap_commit_size);
1134   s->Printf("  loader_flags            = 0x%8.8x\n", header.loader_flags);
1135   s->Printf("  num_data_dir_entries    = 0x%8.8x\n",
1136             (uint32_t)header.data_dirs.size());
1137   uint32_t i;
1138   for (i = 0; i < header.data_dirs.size(); i++) {
1139     s->Printf("  data_dirs[%2u] vmaddr = 0x%8.8x, vmsize = 0x%8.8x\n", i,
1140               header.data_dirs[i].vmaddr, header.data_dirs[i].vmsize);
1141   }
1142 }
1143 // DumpSectionHeader
1144 //
1145 // Dump a single ELF section header to the specified output stream
1146 void ObjectFilePECOFF::DumpSectionHeader(Stream *s,
1147                                          const section_header_t &sh) {
1148   std::string name = std::string(GetSectionName(sh));
1149   s->Printf("%-16s 0x%8.8x 0x%8.8x 0x%8.8x 0x%8.8x 0x%8.8x 0x%8.8x 0x%4.4x "
1150             "0x%4.4x 0x%8.8x\n",
1151             name.c_str(), sh.vmaddr, sh.vmsize, sh.offset, sh.size, sh.reloff,
1152             sh.lineoff, sh.nreloc, sh.nline, sh.flags);
1153 }
1154 
1155 // DumpSectionHeaders
1156 //
1157 // Dump all of the ELF section header to the specified output stream
1158 void ObjectFilePECOFF::DumpSectionHeaders(Stream *s) {
1159 
1160   s->PutCString("Section Headers\n");
1161   s->PutCString("IDX  name             vm addr    vm size    file off   file "
1162                 "size  reloc off  line off   nreloc nline  flags\n");
1163   s->PutCString("==== ---------------- ---------- ---------- ---------- "
1164                 "---------- ---------- ---------- ------ ------ ----------\n");
1165 
1166   uint32_t idx = 0;
1167   SectionHeaderCollIter pos, end = m_sect_headers.end();
1168 
1169   for (pos = m_sect_headers.begin(); pos != end; ++pos, ++idx) {
1170     s->Printf("[%2u] ", idx);
1171     ObjectFilePECOFF::DumpSectionHeader(s, *pos);
1172   }
1173 }
1174 
1175 // DumpDependentModules
1176 //
1177 // Dump all of the dependent modules to the specified output stream
1178 void ObjectFilePECOFF::DumpDependentModules(lldb_private::Stream *s) {
1179   auto num_modules = ParseDependentModules();
1180   if (num_modules > 0) {
1181     s->PutCString("Dependent Modules\n");
1182     for (unsigned i = 0; i < num_modules; ++i) {
1183       auto spec = m_deps_filespec->GetFileSpecAtIndex(i);
1184       s->Printf("  %s\n", spec.GetFilename().GetCString());
1185     }
1186   }
1187 }
1188 
1189 bool ObjectFilePECOFF::IsWindowsSubsystem() {
1190   switch (m_coff_header_opt.subsystem) {
1191   case llvm::COFF::IMAGE_SUBSYSTEM_NATIVE:
1192   case llvm::COFF::IMAGE_SUBSYSTEM_WINDOWS_GUI:
1193   case llvm::COFF::IMAGE_SUBSYSTEM_WINDOWS_CUI:
1194   case llvm::COFF::IMAGE_SUBSYSTEM_NATIVE_WINDOWS:
1195   case llvm::COFF::IMAGE_SUBSYSTEM_WINDOWS_CE_GUI:
1196   case llvm::COFF::IMAGE_SUBSYSTEM_XBOX:
1197   case llvm::COFF::IMAGE_SUBSYSTEM_WINDOWS_BOOT_APPLICATION:
1198     return true;
1199   default:
1200     return false;
1201   }
1202 }
1203 
1204 ArchSpec ObjectFilePECOFF::GetArchitecture() {
1205   uint16_t machine = m_coff_header.machine;
1206   switch (machine) {
1207   default:
1208     break;
1209   case llvm::COFF::IMAGE_FILE_MACHINE_AMD64:
1210   case llvm::COFF::IMAGE_FILE_MACHINE_I386:
1211   case llvm::COFF::IMAGE_FILE_MACHINE_POWERPC:
1212   case llvm::COFF::IMAGE_FILE_MACHINE_POWERPCFP:
1213   case llvm::COFF::IMAGE_FILE_MACHINE_ARM:
1214   case llvm::COFF::IMAGE_FILE_MACHINE_ARMNT:
1215   case llvm::COFF::IMAGE_FILE_MACHINE_THUMB:
1216   case llvm::COFF::IMAGE_FILE_MACHINE_ARM64:
1217     ArchSpec arch;
1218     arch.SetArchitecture(eArchTypeCOFF, machine, LLDB_INVALID_CPUTYPE,
1219                          IsWindowsSubsystem() ? llvm::Triple::Win32
1220                                               : llvm::Triple::UnknownOS);
1221     return arch;
1222   }
1223   return ArchSpec();
1224 }
1225 
1226 ObjectFile::Type ObjectFilePECOFF::CalculateType() {
1227   if (m_coff_header.machine != 0) {
1228     if ((m_coff_header.flags & llvm::COFF::IMAGE_FILE_DLL) == 0)
1229       return eTypeExecutable;
1230     else
1231       return eTypeSharedLibrary;
1232   }
1233   return eTypeExecutable;
1234 }
1235 
1236 ObjectFile::Strata ObjectFilePECOFF::CalculateStrata() { return eStrataUser; }
1237 
1238 // PluginInterface protocol
1239 ConstString ObjectFilePECOFF::GetPluginName() { return GetPluginNameStatic(); }
1240 
1241 uint32_t ObjectFilePECOFF::GetPluginVersion() { return 1; }
1242