1 //===-- ObjectFilePECOFF.h --------------------------------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #ifndef liblldb_ObjectFilePECOFF_h_
11 #define liblldb_ObjectFilePECOFF_h_
12 
13 // C Includes
14 // C++ Includes
15 #include <vector>
16 
17 // Other libraries and framework includes
18 // Project includes
19 #include "lldb/Symbol/ObjectFile.h"
20 
21 class ObjectFilePECOFF : public lldb_private::ObjectFile {
22 public:
23   typedef enum MachineType {
24     MachineUnknown = 0x0,
25     MachineAm33 = 0x1d3,
26     MachineAmd64 = 0x8664,
27     MachineArm = 0x1c0,
28     MachineArmNt = 0x1c4,
29     MachineArm64 = 0xaa64,
30     MachineEbc = 0xebc,
31     MachineX86 = 0x14c,
32     MachineIA64 = 0x200,
33     MachineM32R = 0x9041,
34     MachineMips16 = 0x266,
35     MachineMipsFpu = 0x366,
36     MachineMipsFpu16 = 0x466,
37     MachinePowerPc = 0x1f0,
38     MachinePowerPcfp = 0x1f1,
39     MachineR4000 = 0x166,
40     MachineSh3 = 0x1a2,
41     MachineSh3dsp = 0x1a3,
42     MachineSh4 = 0x1a6,
43     MachineSh5 = 0x1a8,
44     MachineThumb = 0x1c2,
45     MachineWcemIpsv2 = 0x169
46   } MachineType;
47 
48   ObjectFilePECOFF(const lldb::ModuleSP &module_sp, lldb::DataBufferSP &data_sp,
49                    lldb::offset_t data_offset,
50                    const lldb_private::FileSpec *file,
51                    lldb::offset_t file_offset, lldb::offset_t length);
52 
53   ~ObjectFilePECOFF() override;
54 
55   //------------------------------------------------------------------
56   // Static Functions
57   //------------------------------------------------------------------
58   static void Initialize();
59 
60   static void Terminate();
61 
62   static lldb_private::ConstString GetPluginNameStatic();
63 
64   static const char *GetPluginDescriptionStatic();
65 
66   static ObjectFile *
67   CreateInstance(const lldb::ModuleSP &module_sp, lldb::DataBufferSP &data_sp,
68                  lldb::offset_t data_offset, const lldb_private::FileSpec *file,
69                  lldb::offset_t offset, lldb::offset_t length);
70 
71   static lldb_private::ObjectFile *CreateMemoryInstance(
72       const lldb::ModuleSP &module_sp, lldb::DataBufferSP &data_sp,
73       const lldb::ProcessSP &process_sp, lldb::addr_t header_addr);
74 
75   static size_t GetModuleSpecifications(const lldb_private::FileSpec &file,
76                                         lldb::DataBufferSP &data_sp,
77                                         lldb::offset_t data_offset,
78                                         lldb::offset_t file_offset,
79                                         lldb::offset_t length,
80                                         lldb_private::ModuleSpecList &specs);
81 
82   static bool SaveCore(const lldb::ProcessSP &process_sp,
83                        const lldb_private::FileSpec &outfile,
84                        lldb_private::Error &error);
85 
86   static bool MagicBytesMatch(lldb::DataBufferSP &data_sp);
87 
88   static lldb::SymbolType MapSymbolType(uint16_t coff_symbol_type);
89 
90   bool ParseHeader() override;
91 
92   bool SetLoadAddress(lldb_private::Target &target, lldb::addr_t value,
93                       bool value_is_offset) override;
94 
95   lldb::ByteOrder GetByteOrder() const override;
96 
97   bool IsExecutable() const override;
98 
99   uint32_t GetAddressByteSize() const override;
100 
101   //    virtual lldb_private::AddressClass
102   //    GetAddressClass (lldb::addr_t file_addr);
103 
104   lldb_private::Symtab *GetSymtab() override;
105 
106   bool IsStripped() override;
107 
108   void CreateSections(lldb_private::SectionList &unified_section_list) override;
109 
110   void Dump(lldb_private::Stream *s) override;
111 
112   bool GetArchitecture(lldb_private::ArchSpec &arch) override;
113 
114   bool GetUUID(lldb_private::UUID *uuid) override;
115 
116   uint32_t GetDependentModules(lldb_private::FileSpecList &files) override;
117 
118   virtual lldb_private::Address GetEntryPointAddress() override;
119 
120   ObjectFile::Type CalculateType() override;
121 
122   ObjectFile::Strata CalculateStrata() override;
123 
124   //------------------------------------------------------------------
125   // PluginInterface protocol
126   //------------------------------------------------------------------
127   lldb_private::ConstString GetPluginName() override;
128 
129   uint32_t GetPluginVersion() override;
130 
131   bool IsWindowsSubsystem();
132 
133 protected:
134   bool NeedsEndianSwap() const;
135 
136   typedef struct dos_header { // DOS .EXE header
137     uint16_t e_magic;         // Magic number
138     uint16_t e_cblp;          // Bytes on last page of file
139     uint16_t e_cp;            // Pages in file
140     uint16_t e_crlc;          // Relocations
141     uint16_t e_cparhdr;       // Size of header in paragraphs
142     uint16_t e_minalloc;      // Minimum extra paragraphs needed
143     uint16_t e_maxalloc;      // Maximum extra paragraphs needed
144     uint16_t e_ss;            // Initial (relative) SS value
145     uint16_t e_sp;            // Initial SP value
146     uint16_t e_csum;          // Checksum
147     uint16_t e_ip;            // Initial IP value
148     uint16_t e_cs;            // Initial (relative) CS value
149     uint16_t e_lfarlc;        // File address of relocation table
150     uint16_t e_ovno;          // Overlay number
151     uint16_t e_res[4];        // Reserved words
152     uint16_t e_oemid;         // OEM identifier (for e_oeminfo)
153     uint16_t e_oeminfo;       // OEM information; e_oemid specific
154     uint16_t e_res2[10];      // Reserved words
155     uint32_t e_lfanew;        // File address of new exe header
156   } dos_header_t;
157 
158   typedef struct coff_header {
159     uint16_t machine;
160     uint16_t nsects;
161     uint32_t modtime;
162     uint32_t symoff;
163     uint32_t nsyms;
164     uint16_t hdrsize;
165     uint16_t flags;
166   } coff_header_t;
167 
168   typedef struct data_directory {
169     uint32_t vmaddr;
170     uint32_t vmsize;
171   } data_directory_t;
172 
173   typedef struct coff_opt_header {
174     uint16_t magic;
175     uint8_t major_linker_version;
176     uint8_t minor_linker_version;
177     uint32_t code_size;
178     uint32_t data_size;
179     uint32_t bss_size;
180     uint32_t entry;
181     uint32_t code_offset;
182     uint32_t data_offset;
183 
184     uint64_t image_base;
185     uint32_t sect_alignment;
186     uint32_t file_alignment;
187     uint16_t major_os_system_version;
188     uint16_t minor_os_system_version;
189     uint16_t major_image_version;
190     uint16_t minor_image_version;
191     uint16_t major_subsystem_version;
192     uint16_t minor_subsystem_version;
193     uint32_t reserved1;
194     uint32_t image_size;
195     uint32_t header_size;
196     uint32_t checksum;
197     uint16_t subsystem;
198     uint16_t dll_flags;
199     uint64_t stack_reserve_size;
200     uint64_t stack_commit_size;
201     uint64_t heap_reserve_size;
202     uint64_t heap_commit_size;
203     uint32_t loader_flags;
204     //    uint32_t	num_data_dir_entries;
205     std::vector<data_directory>
206         data_dirs; // will contain num_data_dir_entries entries
207   } coff_opt_header_t;
208 
209   typedef enum coff_data_dir_type {
210     coff_data_dir_export_table = 0,
211     coff_data_dir_import_table = 1,
212   } coff_data_dir_type;
213 
214   typedef struct section_header {
215     char name[8];
216     uint32_t vmsize;  // Virtual Size
217     uint32_t vmaddr;  // Virtual Addr
218     uint32_t size;    // File size
219     uint32_t offset;  // File offset
220     uint32_t reloff;  // Offset to relocations
221     uint32_t lineoff; // Offset to line table entries
222     uint16_t nreloc;  // Number of relocation entries
223     uint16_t nline;   // Number of line table entries
224     uint32_t flags;
225   } section_header_t;
226 
227   typedef struct coff_symbol {
228     char name[8];
229     uint32_t value;
230     uint16_t sect;
231     uint16_t type;
232     uint8_t storage;
233     uint8_t naux;
234   } coff_symbol_t;
235 
236   typedef struct export_directory_entry {
237     uint32_t characteristics;
238     uint32_t time_date_stamp;
239     uint16_t major_version;
240     uint16_t minor_version;
241     uint32_t name;
242     uint32_t base;
243     uint32_t number_of_functions;
244     uint32_t number_of_names;
245     uint32_t address_of_functions;
246     uint32_t address_of_names;
247     uint32_t address_of_name_ordinals;
248   } export_directory_entry;
249 
250   static bool ParseDOSHeader(lldb_private::DataExtractor &data,
251                              dos_header_t &dos_header);
252   static bool ParseCOFFHeader(lldb_private::DataExtractor &data,
253                               lldb::offset_t *offset_ptr,
254                               coff_header_t &coff_header);
255   bool ParseCOFFOptionalHeader(lldb::offset_t *offset_ptr);
256   bool ParseSectionHeaders(uint32_t offset);
257 
258   static void DumpDOSHeader(lldb_private::Stream *s,
259                             const dos_header_t &header);
260   static void DumpCOFFHeader(lldb_private::Stream *s,
261                              const coff_header_t &header);
262   static void DumpOptCOFFHeader(lldb_private::Stream *s,
263                                 const coff_opt_header_t &header);
264   void DumpSectionHeaders(lldb_private::Stream *s);
265   void DumpSectionHeader(lldb_private::Stream *s, const section_header_t &sh);
266   bool GetSectionName(std::string &sect_name, const section_header_t &sect);
267 
268   typedef std::vector<section_header_t> SectionHeaderColl;
269   typedef SectionHeaderColl::iterator SectionHeaderCollIter;
270   typedef SectionHeaderColl::const_iterator SectionHeaderCollConstIter;
271 
272 private:
273   dos_header_t m_dos_header;
274   coff_header_t m_coff_header;
275   coff_opt_header_t m_coff_header_opt;
276   SectionHeaderColl m_sect_headers;
277   lldb::addr_t m_image_base;
278   lldb_private::Address m_entry_point_address;
279 };
280 
281 #endif // liblldb_ObjectFilePECOFF_h_
282