Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===- MachOYAML.h - Mach-O YAMLIO implementation ---------------*- C++ -*-===// |
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 | /// \file |
||
10 | /// This file declares classes for handling the YAML representation |
||
11 | /// of Mach-O. |
||
12 | /// |
||
13 | //===----------------------------------------------------------------------===// |
||
14 | |||
15 | #ifndef LLVM_OBJECTYAML_MACHOYAML_H |
||
16 | #define LLVM_OBJECTYAML_MACHOYAML_H |
||
17 | |||
18 | #include "llvm/ADT/StringRef.h" |
||
19 | #include "llvm/BinaryFormat/MachO.h" |
||
20 | #include "llvm/ObjectYAML/DWARFYAML.h" |
||
21 | #include "llvm/ObjectYAML/YAML.h" |
||
22 | #include "llvm/Support/YAMLTraits.h" |
||
23 | #include <cstdint> |
||
24 | #include <optional> |
||
25 | #include <string> |
||
26 | #include <vector> |
||
27 | |||
28 | namespace llvm { |
||
29 | namespace MachOYAML { |
||
30 | |||
31 | struct Relocation { |
||
32 | // Offset in the section to what is being relocated. |
||
33 | llvm::yaml::Hex32 address; |
||
34 | // Symbol index if r_extern == 1 else section index. |
||
35 | uint32_t symbolnum; |
||
36 | bool is_pcrel; |
||
37 | // Real length = 2 ^ length. |
||
38 | uint8_t length; |
||
39 | bool is_extern; |
||
40 | uint8_t type; |
||
41 | bool is_scattered; |
||
42 | int32_t value; |
||
43 | }; |
||
44 | |||
45 | struct Section { |
||
46 | char sectname[16]; |
||
47 | char segname[16]; |
||
48 | llvm::yaml::Hex64 addr; |
||
49 | uint64_t size; |
||
50 | llvm::yaml::Hex32 offset; |
||
51 | uint32_t align; |
||
52 | llvm::yaml::Hex32 reloff; |
||
53 | uint32_t nreloc; |
||
54 | llvm::yaml::Hex32 flags; |
||
55 | llvm::yaml::Hex32 reserved1; |
||
56 | llvm::yaml::Hex32 reserved2; |
||
57 | llvm::yaml::Hex32 reserved3; |
||
58 | std::optional<llvm::yaml::BinaryRef> content; |
||
59 | std::vector<Relocation> relocations; |
||
60 | }; |
||
61 | |||
62 | struct FileHeader { |
||
63 | llvm::yaml::Hex32 magic; |
||
64 | llvm::yaml::Hex32 cputype; |
||
65 | llvm::yaml::Hex32 cpusubtype; |
||
66 | llvm::yaml::Hex32 filetype; |
||
67 | uint32_t ncmds; |
||
68 | uint32_t sizeofcmds; |
||
69 | llvm::yaml::Hex32 flags; |
||
70 | llvm::yaml::Hex32 reserved; |
||
71 | }; |
||
72 | |||
73 | struct LoadCommand { |
||
74 | virtual ~LoadCommand(); |
||
75 | |||
76 | llvm::MachO::macho_load_command Data; |
||
77 | std::vector<Section> Sections; |
||
78 | std::vector<MachO::build_tool_version> Tools; |
||
79 | std::vector<llvm::yaml::Hex8> PayloadBytes; |
||
80 | std::string Content; |
||
81 | uint64_t ZeroPadBytes; |
||
82 | }; |
||
83 | |||
84 | struct NListEntry { |
||
85 | uint32_t n_strx; |
||
86 | llvm::yaml::Hex8 n_type; |
||
87 | uint8_t n_sect; |
||
88 | uint16_t n_desc; |
||
89 | uint64_t n_value; |
||
90 | }; |
||
91 | |||
92 | struct RebaseOpcode { |
||
93 | MachO::RebaseOpcode Opcode; |
||
94 | uint8_t Imm; |
||
95 | std::vector<yaml::Hex64> ExtraData; |
||
96 | }; |
||
97 | |||
98 | struct BindOpcode { |
||
99 | MachO::BindOpcode Opcode; |
||
100 | uint8_t Imm; |
||
101 | std::vector<yaml::Hex64> ULEBExtraData; |
||
102 | std::vector<int64_t> SLEBExtraData; |
||
103 | StringRef Symbol; |
||
104 | }; |
||
105 | |||
106 | struct ExportEntry { |
||
107 | uint64_t TerminalSize = 0; |
||
108 | uint64_t NodeOffset = 0; |
||
109 | std::string Name; |
||
110 | llvm::yaml::Hex64 Flags = 0; |
||
111 | llvm::yaml::Hex64 Address = 0; |
||
112 | llvm::yaml::Hex64 Other = 0; |
||
113 | std::string ImportName; |
||
114 | std::vector<MachOYAML::ExportEntry> Children; |
||
115 | }; |
||
116 | |||
117 | struct DataInCodeEntry { |
||
118 | llvm::yaml::Hex32 Offset; |
||
119 | uint16_t Length; |
||
120 | llvm::yaml::Hex16 Kind; |
||
121 | }; |
||
122 | |||
123 | struct LinkEditData { |
||
124 | std::vector<MachOYAML::RebaseOpcode> RebaseOpcodes; |
||
125 | std::vector<MachOYAML::BindOpcode> BindOpcodes; |
||
126 | std::vector<MachOYAML::BindOpcode> WeakBindOpcodes; |
||
127 | std::vector<MachOYAML::BindOpcode> LazyBindOpcodes; |
||
128 | MachOYAML::ExportEntry ExportTrie; |
||
129 | std::vector<NListEntry> NameList; |
||
130 | std::vector<StringRef> StringTable; |
||
131 | std::vector<yaml::Hex32> IndirectSymbols; |
||
132 | std::vector<yaml::Hex64> FunctionStarts; |
||
133 | std::vector<DataInCodeEntry> DataInCode; |
||
134 | std::vector<yaml::Hex8> ChainedFixups; |
||
135 | |||
136 | bool isEmpty() const; |
||
137 | }; |
||
138 | |||
139 | struct Object { |
||
140 | bool IsLittleEndian; |
||
141 | FileHeader Header; |
||
142 | std::vector<LoadCommand> LoadCommands; |
||
143 | std::vector<Section> Sections; |
||
144 | LinkEditData LinkEdit; |
||
145 | std::optional<llvm::yaml::BinaryRef> RawLinkEditSegment; |
||
146 | DWARFYAML::Data DWARF; |
||
147 | }; |
||
148 | |||
149 | struct FatHeader { |
||
150 | llvm::yaml::Hex32 magic; |
||
151 | uint32_t nfat_arch; |
||
152 | }; |
||
153 | |||
154 | struct FatArch { |
||
155 | llvm::yaml::Hex32 cputype; |
||
156 | llvm::yaml::Hex32 cpusubtype; |
||
157 | llvm::yaml::Hex64 offset; |
||
158 | uint64_t size; |
||
159 | uint32_t align; |
||
160 | llvm::yaml::Hex32 reserved; |
||
161 | }; |
||
162 | |||
163 | struct UniversalBinary { |
||
164 | FatHeader Header; |
||
165 | std::vector<FatArch> FatArchs; |
||
166 | std::vector<Object> Slices; |
||
167 | }; |
||
168 | |||
169 | } // end namespace MachOYAML |
||
170 | } // end namespace llvm |
||
171 | |||
172 | LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::LoadCommand) |
||
173 | LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::Relocation) |
||
174 | LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::Section) |
||
175 | LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::RebaseOpcode) |
||
176 | LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::BindOpcode) |
||
177 | LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::ExportEntry) |
||
178 | LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::NListEntry) |
||
179 | LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::Object) |
||
180 | LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::FatArch) |
||
181 | LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachOYAML::DataInCodeEntry) |
||
182 | LLVM_YAML_IS_SEQUENCE_VECTOR(llvm::MachO::build_tool_version) |
||
183 | |||
184 | namespace llvm { |
||
185 | |||
186 | class raw_ostream; |
||
187 | |||
188 | namespace yaml { |
||
189 | |||
190 | template <> struct MappingTraits<MachOYAML::FileHeader> { |
||
191 | static void mapping(IO &IO, MachOYAML::FileHeader &FileHeader); |
||
192 | }; |
||
193 | |||
194 | template <> struct MappingTraits<MachOYAML::Object> { |
||
195 | static void mapping(IO &IO, MachOYAML::Object &Object); |
||
196 | }; |
||
197 | |||
198 | template <> struct MappingTraits<MachOYAML::FatHeader> { |
||
199 | static void mapping(IO &IO, MachOYAML::FatHeader &FatHeader); |
||
200 | }; |
||
201 | |||
202 | template <> struct MappingTraits<MachOYAML::FatArch> { |
||
203 | static void mapping(IO &IO, MachOYAML::FatArch &FatArch); |
||
204 | }; |
||
205 | |||
206 | template <> struct MappingTraits<MachOYAML::UniversalBinary> { |
||
207 | static void mapping(IO &IO, MachOYAML::UniversalBinary &UniversalBinary); |
||
208 | }; |
||
209 | |||
210 | template <> struct MappingTraits<MachOYAML::LoadCommand> { |
||
211 | static void mapping(IO &IO, MachOYAML::LoadCommand &LoadCommand); |
||
212 | }; |
||
213 | |||
214 | template <> struct MappingTraits<MachOYAML::LinkEditData> { |
||
215 | static void mapping(IO &IO, MachOYAML::LinkEditData &LinkEditData); |
||
216 | }; |
||
217 | |||
218 | template <> struct MappingTraits<MachOYAML::RebaseOpcode> { |
||
219 | static void mapping(IO &IO, MachOYAML::RebaseOpcode &RebaseOpcode); |
||
220 | }; |
||
221 | |||
222 | template <> struct MappingTraits<MachOYAML::BindOpcode> { |
||
223 | static void mapping(IO &IO, MachOYAML::BindOpcode &BindOpcode); |
||
224 | }; |
||
225 | |||
226 | template <> struct MappingTraits<MachOYAML::ExportEntry> { |
||
227 | static void mapping(IO &IO, MachOYAML::ExportEntry &ExportEntry); |
||
228 | }; |
||
229 | |||
230 | template <> struct MappingTraits<MachOYAML::Relocation> { |
||
231 | static void mapping(IO &IO, MachOYAML::Relocation &R); |
||
232 | }; |
||
233 | |||
234 | template <> struct MappingTraits<MachOYAML::Section> { |
||
235 | static void mapping(IO &IO, MachOYAML::Section &Section); |
||
236 | static std::string validate(IO &io, MachOYAML::Section &Section); |
||
237 | }; |
||
238 | |||
239 | template <> struct MappingTraits<MachOYAML::NListEntry> { |
||
240 | static void mapping(IO &IO, MachOYAML::NListEntry &NListEntry); |
||
241 | }; |
||
242 | |||
243 | template <> struct MappingTraits<MachO::build_tool_version> { |
||
244 | static void mapping(IO &IO, MachO::build_tool_version &tool); |
||
245 | }; |
||
246 | |||
247 | template <> struct MappingTraits<MachOYAML::DataInCodeEntry> { |
||
248 | static void mapping(IO &IO, MachOYAML::DataInCodeEntry &DataInCodeEntry); |
||
249 | }; |
||
250 | |||
251 | #define HANDLE_LOAD_COMMAND(LCName, LCValue, LCStruct) \ |
||
252 | io.enumCase(value, #LCName, MachO::LCName); |
||
253 | |||
254 | template <> struct ScalarEnumerationTraits<MachO::LoadCommandType> { |
||
255 | static void enumeration(IO &io, MachO::LoadCommandType &value) { |
||
256 | #include "llvm/BinaryFormat/MachO.def" |
||
257 | io.enumFallback<Hex32>(value); |
||
258 | } |
||
259 | }; |
||
260 | |||
261 | #define ENUM_CASE(Enum) io.enumCase(value, #Enum, MachO::Enum); |
||
262 | |||
263 | template <> struct ScalarEnumerationTraits<MachO::RebaseOpcode> { |
||
264 | static void enumeration(IO &io, MachO::RebaseOpcode &value) { |
||
265 | ENUM_CASE(REBASE_OPCODE_DONE) |
||
266 | ENUM_CASE(REBASE_OPCODE_SET_TYPE_IMM) |
||
267 | ENUM_CASE(REBASE_OPCODE_SET_SEGMENT_AND_OFFSET_ULEB) |
||
268 | ENUM_CASE(REBASE_OPCODE_ADD_ADDR_ULEB) |
||
269 | ENUM_CASE(REBASE_OPCODE_ADD_ADDR_IMM_SCALED) |
||
270 | ENUM_CASE(REBASE_OPCODE_DO_REBASE_IMM_TIMES) |
||
271 | ENUM_CASE(REBASE_OPCODE_DO_REBASE_ULEB_TIMES) |
||
272 | ENUM_CASE(REBASE_OPCODE_DO_REBASE_ADD_ADDR_ULEB) |
||
273 | ENUM_CASE(REBASE_OPCODE_DO_REBASE_ULEB_TIMES_SKIPPING_ULEB) |
||
274 | io.enumFallback<Hex8>(value); |
||
275 | } |
||
276 | }; |
||
277 | |||
278 | template <> struct ScalarEnumerationTraits<MachO::BindOpcode> { |
||
279 | static void enumeration(IO &io, MachO::BindOpcode &value) { |
||
280 | ENUM_CASE(BIND_OPCODE_DONE) |
||
281 | ENUM_CASE(BIND_OPCODE_SET_DYLIB_ORDINAL_IMM) |
||
282 | ENUM_CASE(BIND_OPCODE_SET_DYLIB_ORDINAL_ULEB) |
||
283 | ENUM_CASE(BIND_OPCODE_SET_DYLIB_SPECIAL_IMM) |
||
284 | ENUM_CASE(BIND_OPCODE_SET_SYMBOL_TRAILING_FLAGS_IMM) |
||
285 | ENUM_CASE(BIND_OPCODE_SET_TYPE_IMM) |
||
286 | ENUM_CASE(BIND_OPCODE_SET_ADDEND_SLEB) |
||
287 | ENUM_CASE(BIND_OPCODE_SET_SEGMENT_AND_OFFSET_ULEB) |
||
288 | ENUM_CASE(BIND_OPCODE_ADD_ADDR_ULEB) |
||
289 | ENUM_CASE(BIND_OPCODE_DO_BIND) |
||
290 | ENUM_CASE(BIND_OPCODE_DO_BIND_ADD_ADDR_ULEB) |
||
291 | ENUM_CASE(BIND_OPCODE_DO_BIND_ADD_ADDR_IMM_SCALED) |
||
292 | ENUM_CASE(BIND_OPCODE_DO_BIND_ULEB_TIMES_SKIPPING_ULEB) |
||
293 | io.enumFallback<Hex8>(value); |
||
294 | } |
||
295 | }; |
||
296 | |||
297 | // This trait is used for 16-byte chars in Mach structures used for strings |
||
298 | using char_16 = char[16]; |
||
299 | |||
300 | template <> struct ScalarTraits<char_16> { |
||
301 | static void output(const char_16 &Val, void *, raw_ostream &Out); |
||
302 | static StringRef input(StringRef Scalar, void *, char_16 &Val); |
||
303 | static QuotingType mustQuote(StringRef S); |
||
304 | }; |
||
305 | |||
306 | // This trait is used for UUIDs. It reads and writes them matching otool's |
||
307 | // formatting style. |
||
308 | using uuid_t = raw_ostream::uuid_t; |
||
309 | |||
310 | template <> struct ScalarTraits<uuid_t> { |
||
311 | static void output(const uuid_t &Val, void *, raw_ostream &Out); |
||
312 | static StringRef input(StringRef Scalar, void *, uuid_t &Val); |
||
313 | static QuotingType mustQuote(StringRef S); |
||
314 | }; |
||
315 | |||
316 | // Load Command struct mapping traits |
||
317 | |||
318 | #define LOAD_COMMAND_STRUCT(LCStruct) \ |
||
319 | template <> struct MappingTraits<MachO::LCStruct> { \ |
||
320 | static void mapping(IO &IO, MachO::LCStruct &LoadCommand); \ |
||
321 | }; |
||
322 | |||
323 | #include "llvm/BinaryFormat/MachO.def" |
||
324 | |||
325 | // Extra structures used by load commands |
||
326 | template <> struct MappingTraits<MachO::dylib> { |
||
327 | static void mapping(IO &IO, MachO::dylib &LoadCommand); |
||
328 | }; |
||
329 | |||
330 | template <> struct MappingTraits<MachO::fvmlib> { |
||
331 | static void mapping(IO &IO, MachO::fvmlib &LoadCommand); |
||
332 | }; |
||
333 | |||
334 | template <> struct MappingTraits<MachO::section> { |
||
335 | static void mapping(IO &IO, MachO::section &LoadCommand); |
||
336 | }; |
||
337 | |||
338 | template <> struct MappingTraits<MachO::section_64> { |
||
339 | static void mapping(IO &IO, MachO::section_64 &LoadCommand); |
||
340 | }; |
||
341 | |||
342 | } // end namespace yaml |
||
343 | |||
344 | } // end namespace llvm |
||
345 | |||
346 | #endif // LLVM_OBJECTYAML_MACHOYAML_H |