xref: /freebsd/contrib/llvm-project/lld/COFF/DLL.cpp (revision 0b57cec536236d46e3dba9bd041533462f33dbb7)
1 //===- DLL.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 // This file defines various types of chunks for the DLL import or export
10 // descriptor tables. They are inherently Windows-specific.
11 // You need to read Microsoft PE/COFF spec to understand details
12 // about the data structures.
13 //
14 // If you are not particularly interested in linking against Windows
15 // DLL, you can skip this file, and you should still be able to
16 // understand the rest of the linker.
17 //
18 //===----------------------------------------------------------------------===//
19 
20 #include "DLL.h"
21 #include "Chunks.h"
22 #include "llvm/Object/COFF.h"
23 #include "llvm/Support/Endian.h"
24 #include "llvm/Support/Path.h"
25 
26 using namespace llvm;
27 using namespace llvm::object;
28 using namespace llvm::support::endian;
29 using namespace llvm::COFF;
30 
31 namespace lld {
32 namespace coff {
33 namespace {
34 
35 // Import table
36 
37 // A chunk for the import descriptor table.
38 class HintNameChunk : public NonSectionChunk {
39 public:
40   HintNameChunk(StringRef n, uint16_t h) : name(n), hint(h) {}
41 
42   size_t getSize() const override {
43     // Starts with 2 byte Hint field, followed by a null-terminated string,
44     // ends with 0 or 1 byte padding.
45     return alignTo(name.size() + 3, 2);
46   }
47 
48   void writeTo(uint8_t *buf) const override {
49     memset(buf, 0, getSize());
50     write16le(buf, hint);
51     memcpy(buf + 2, name.data(), name.size());
52   }
53 
54 private:
55   StringRef name;
56   uint16_t hint;
57 };
58 
59 // A chunk for the import descriptor table.
60 class LookupChunk : public NonSectionChunk {
61 public:
62   explicit LookupChunk(Chunk *c) : hintName(c) {
63     setAlignment(config->wordsize);
64   }
65   size_t getSize() const override { return config->wordsize; }
66 
67   void writeTo(uint8_t *buf) const override {
68     if (config->is64())
69       write64le(buf, hintName->getRVA());
70     else
71       write32le(buf, hintName->getRVA());
72   }
73 
74   Chunk *hintName;
75 };
76 
77 // A chunk for the import descriptor table.
78 // This chunk represent import-by-ordinal symbols.
79 // See Microsoft PE/COFF spec 7.1. Import Header for details.
80 class OrdinalOnlyChunk : public NonSectionChunk {
81 public:
82   explicit OrdinalOnlyChunk(uint16_t v) : ordinal(v) {
83     setAlignment(config->wordsize);
84   }
85   size_t getSize() const override { return config->wordsize; }
86 
87   void writeTo(uint8_t *buf) const override {
88     // An import-by-ordinal slot has MSB 1 to indicate that
89     // this is import-by-ordinal (and not import-by-name).
90     if (config->is64()) {
91       write64le(buf, (1ULL << 63) | ordinal);
92     } else {
93       write32le(buf, (1ULL << 31) | ordinal);
94     }
95   }
96 
97   uint16_t ordinal;
98 };
99 
100 // A chunk for the import descriptor table.
101 class ImportDirectoryChunk : public NonSectionChunk {
102 public:
103   explicit ImportDirectoryChunk(Chunk *n) : dllName(n) {}
104   size_t getSize() const override { return sizeof(ImportDirectoryTableEntry); }
105 
106   void writeTo(uint8_t *buf) const override {
107     memset(buf, 0, getSize());
108 
109     auto *e = (coff_import_directory_table_entry *)(buf);
110     e->ImportLookupTableRVA = lookupTab->getRVA();
111     e->NameRVA = dllName->getRVA();
112     e->ImportAddressTableRVA = addressTab->getRVA();
113   }
114 
115   Chunk *dllName;
116   Chunk *lookupTab;
117   Chunk *addressTab;
118 };
119 
120 // A chunk representing null terminator in the import table.
121 // Contents of this chunk is always null bytes.
122 class NullChunk : public NonSectionChunk {
123 public:
124   explicit NullChunk(size_t n) : size(n) { hasData = false; }
125   size_t getSize() const override { return size; }
126 
127   void writeTo(uint8_t *buf) const override {
128     memset(buf, 0, size);
129   }
130 
131 private:
132   size_t size;
133 };
134 
135 static std::vector<std::vector<DefinedImportData *>>
136 binImports(const std::vector<DefinedImportData *> &imports) {
137   // Group DLL-imported symbols by DLL name because that's how
138   // symbols are layed out in the import descriptor table.
139   auto less = [](const std::string &a, const std::string &b) {
140     return config->dllOrder[a] < config->dllOrder[b];
141   };
142   std::map<std::string, std::vector<DefinedImportData *>,
143            bool(*)(const std::string &, const std::string &)> m(less);
144   for (DefinedImportData *sym : imports)
145     m[sym->getDLLName().lower()].push_back(sym);
146 
147   std::vector<std::vector<DefinedImportData *>> v;
148   for (auto &kv : m) {
149     // Sort symbols by name for each group.
150     std::vector<DefinedImportData *> &syms = kv.second;
151     std::sort(syms.begin(), syms.end(),
152               [](DefinedImportData *a, DefinedImportData *b) {
153                 return a->getName() < b->getName();
154               });
155     v.push_back(std::move(syms));
156   }
157   return v;
158 }
159 
160 // Export table
161 // See Microsoft PE/COFF spec 4.3 for details.
162 
163 // A chunk for the delay import descriptor table etnry.
164 class DelayDirectoryChunk : public NonSectionChunk {
165 public:
166   explicit DelayDirectoryChunk(Chunk *n) : dllName(n) {}
167 
168   size_t getSize() const override {
169     return sizeof(delay_import_directory_table_entry);
170   }
171 
172   void writeTo(uint8_t *buf) const override {
173     memset(buf, 0, getSize());
174 
175     auto *e = (delay_import_directory_table_entry *)(buf);
176     e->Attributes = 1;
177     e->Name = dllName->getRVA();
178     e->ModuleHandle = moduleHandle->getRVA();
179     e->DelayImportAddressTable = addressTab->getRVA();
180     e->DelayImportNameTable = nameTab->getRVA();
181   }
182 
183   Chunk *dllName;
184   Chunk *moduleHandle;
185   Chunk *addressTab;
186   Chunk *nameTab;
187 };
188 
189 // Initial contents for delay-loaded functions.
190 // This code calls __delayLoadHelper2 function to resolve a symbol
191 // and then overwrites its jump table slot with the result
192 // for subsequent function calls.
193 static const uint8_t thunkX64[] = {
194     0x48, 0x8D, 0x05, 0, 0, 0, 0,       // lea     rax, [__imp_<FUNCNAME>]
195     0xE9, 0, 0, 0, 0,                   // jmp     __tailMerge_<lib>
196 };
197 
198 static const uint8_t tailMergeX64[] = {
199     0x51,                               // push    rcx
200     0x52,                               // push    rdx
201     0x41, 0x50,                         // push    r8
202     0x41, 0x51,                         // push    r9
203     0x48, 0x83, 0xEC, 0x48,             // sub     rsp, 48h
204     0x66, 0x0F, 0x7F, 0x04, 0x24,       // movdqa  xmmword ptr [rsp], xmm0
205     0x66, 0x0F, 0x7F, 0x4C, 0x24, 0x10, // movdqa  xmmword ptr [rsp+10h], xmm1
206     0x66, 0x0F, 0x7F, 0x54, 0x24, 0x20, // movdqa  xmmword ptr [rsp+20h], xmm2
207     0x66, 0x0F, 0x7F, 0x5C, 0x24, 0x30, // movdqa  xmmword ptr [rsp+30h], xmm3
208     0x48, 0x8B, 0xD0,                   // mov     rdx, rax
209     0x48, 0x8D, 0x0D, 0, 0, 0, 0,       // lea     rcx, [___DELAY_IMPORT_...]
210     0xE8, 0, 0, 0, 0,                   // call    __delayLoadHelper2
211     0x66, 0x0F, 0x6F, 0x04, 0x24,       // movdqa  xmm0, xmmword ptr [rsp]
212     0x66, 0x0F, 0x6F, 0x4C, 0x24, 0x10, // movdqa  xmm1, xmmword ptr [rsp+10h]
213     0x66, 0x0F, 0x6F, 0x54, 0x24, 0x20, // movdqa  xmm2, xmmword ptr [rsp+20h]
214     0x66, 0x0F, 0x6F, 0x5C, 0x24, 0x30, // movdqa  xmm3, xmmword ptr [rsp+30h]
215     0x48, 0x83, 0xC4, 0x48,             // add     rsp, 48h
216     0x41, 0x59,                         // pop     r9
217     0x41, 0x58,                         // pop     r8
218     0x5A,                               // pop     rdx
219     0x59,                               // pop     rcx
220     0xFF, 0xE0,                         // jmp     rax
221 };
222 
223 static const uint8_t thunkX86[] = {
224     0xB8, 0, 0, 0, 0,  // mov   eax, offset ___imp__<FUNCNAME>
225     0xE9, 0, 0, 0, 0,  // jmp   __tailMerge_<lib>
226 };
227 
228 static const uint8_t tailMergeX86[] = {
229     0x51,              // push  ecx
230     0x52,              // push  edx
231     0x50,              // push  eax
232     0x68, 0, 0, 0, 0,  // push  offset ___DELAY_IMPORT_DESCRIPTOR_<DLLNAME>_dll
233     0xE8, 0, 0, 0, 0,  // call  ___delayLoadHelper2@8
234     0x5A,              // pop   edx
235     0x59,              // pop   ecx
236     0xFF, 0xE0,        // jmp   eax
237 };
238 
239 static const uint8_t thunkARM[] = {
240     0x40, 0xf2, 0x00, 0x0c, // mov.w   ip, #0 __imp_<FUNCNAME>
241     0xc0, 0xf2, 0x00, 0x0c, // mov.t   ip, #0 __imp_<FUNCNAME>
242     0x00, 0xf0, 0x00, 0xb8, // b.w     __tailMerge_<lib>
243 };
244 
245 static const uint8_t tailMergeARM[] = {
246     0x2d, 0xe9, 0x0f, 0x48, // push.w  {r0, r1, r2, r3, r11, lr}
247     0x0d, 0xf2, 0x10, 0x0b, // addw    r11, sp, #16
248     0x2d, 0xed, 0x10, 0x0b, // vpush   {d0, d1, d2, d3, d4, d5, d6, d7}
249     0x61, 0x46,             // mov     r1, ip
250     0x40, 0xf2, 0x00, 0x00, // mov.w   r0, #0 DELAY_IMPORT_DESCRIPTOR
251     0xc0, 0xf2, 0x00, 0x00, // mov.t   r0, #0 DELAY_IMPORT_DESCRIPTOR
252     0x00, 0xf0, 0x00, 0xd0, // bl      #0 __delayLoadHelper2
253     0x84, 0x46,             // mov     ip, r0
254     0xbd, 0xec, 0x10, 0x0b, // vpop    {d0, d1, d2, d3, d4, d5, d6, d7}
255     0xbd, 0xe8, 0x0f, 0x48, // pop.w   {r0, r1, r2, r3, r11, lr}
256     0x60, 0x47,             // bx      ip
257 };
258 
259 static const uint8_t thunkARM64[] = {
260     0x11, 0x00, 0x00, 0x90, // adrp    x17, #0      __imp_<FUNCNAME>
261     0x31, 0x02, 0x00, 0x91, // add     x17, x17, #0 :lo12:__imp_<FUNCNAME>
262     0x00, 0x00, 0x00, 0x14, // b       __tailMerge_<lib>
263 };
264 
265 static const uint8_t tailMergeARM64[] = {
266     0xfd, 0x7b, 0xb3, 0xa9, // stp     x29, x30, [sp, #-208]!
267     0xfd, 0x03, 0x00, 0x91, // mov     x29, sp
268     0xe0, 0x07, 0x01, 0xa9, // stp     x0, x1, [sp, #16]
269     0xe2, 0x0f, 0x02, 0xa9, // stp     x2, x3, [sp, #32]
270     0xe4, 0x17, 0x03, 0xa9, // stp     x4, x5, [sp, #48]
271     0xe6, 0x1f, 0x04, 0xa9, // stp     x6, x7, [sp, #64]
272     0xe0, 0x87, 0x02, 0xad, // stp     q0, q1, [sp, #80]
273     0xe2, 0x8f, 0x03, 0xad, // stp     q2, q3, [sp, #112]
274     0xe4, 0x97, 0x04, 0xad, // stp     q4, q5, [sp, #144]
275     0xe6, 0x9f, 0x05, 0xad, // stp     q6, q7, [sp, #176]
276     0xe1, 0x03, 0x11, 0xaa, // mov     x1, x17
277     0x00, 0x00, 0x00, 0x90, // adrp    x0, #0     DELAY_IMPORT_DESCRIPTOR
278     0x00, 0x00, 0x00, 0x91, // add     x0, x0, #0 :lo12:DELAY_IMPORT_DESCRIPTOR
279     0x00, 0x00, 0x00, 0x94, // bl      #0 __delayLoadHelper2
280     0xf0, 0x03, 0x00, 0xaa, // mov     x16, x0
281     0xe6, 0x9f, 0x45, 0xad, // ldp     q6, q7, [sp, #176]
282     0xe4, 0x97, 0x44, 0xad, // ldp     q4, q5, [sp, #144]
283     0xe2, 0x8f, 0x43, 0xad, // ldp     q2, q3, [sp, #112]
284     0xe0, 0x87, 0x42, 0xad, // ldp     q0, q1, [sp, #80]
285     0xe6, 0x1f, 0x44, 0xa9, // ldp     x6, x7, [sp, #64]
286     0xe4, 0x17, 0x43, 0xa9, // ldp     x4, x5, [sp, #48]
287     0xe2, 0x0f, 0x42, 0xa9, // ldp     x2, x3, [sp, #32]
288     0xe0, 0x07, 0x41, 0xa9, // ldp     x0, x1, [sp, #16]
289     0xfd, 0x7b, 0xcd, 0xa8, // ldp     x29, x30, [sp], #208
290     0x00, 0x02, 0x1f, 0xd6, // br      x16
291 };
292 
293 // A chunk for the delay import thunk.
294 class ThunkChunkX64 : public NonSectionChunk {
295 public:
296   ThunkChunkX64(Defined *i, Chunk *tm) : imp(i), tailMerge(tm) {}
297 
298   size_t getSize() const override { return sizeof(thunkX64); }
299 
300   void writeTo(uint8_t *buf) const override {
301     memcpy(buf, thunkX64, sizeof(thunkX64));
302     write32le(buf + 3, imp->getRVA() - rva - 7);
303     write32le(buf + 8, tailMerge->getRVA() - rva - 12);
304   }
305 
306   Defined *imp = nullptr;
307   Chunk *tailMerge = nullptr;
308 };
309 
310 class TailMergeChunkX64 : public NonSectionChunk {
311 public:
312   TailMergeChunkX64(Chunk *d, Defined *h) : desc(d), helper(h) {}
313 
314   size_t getSize() const override { return sizeof(tailMergeX64); }
315 
316   void writeTo(uint8_t *buf) const override {
317     memcpy(buf, tailMergeX64, sizeof(tailMergeX64));
318     write32le(buf + 39, desc->getRVA() - rva - 43);
319     write32le(buf + 44, helper->getRVA() - rva - 48);
320   }
321 
322   Chunk *desc = nullptr;
323   Defined *helper = nullptr;
324 };
325 
326 class ThunkChunkX86 : public NonSectionChunk {
327 public:
328   ThunkChunkX86(Defined *i, Chunk *tm) : imp(i), tailMerge(tm) {}
329 
330   size_t getSize() const override { return sizeof(thunkX86); }
331 
332   void writeTo(uint8_t *buf) const override {
333     memcpy(buf, thunkX86, sizeof(thunkX86));
334     write32le(buf + 1, imp->getRVA() + config->imageBase);
335     write32le(buf + 6, tailMerge->getRVA() - rva - 10);
336   }
337 
338   void getBaserels(std::vector<Baserel> *res) override {
339     res->emplace_back(rva + 1);
340   }
341 
342   Defined *imp = nullptr;
343   Chunk *tailMerge = nullptr;
344 };
345 
346 class TailMergeChunkX86 : public NonSectionChunk {
347 public:
348   TailMergeChunkX86(Chunk *d, Defined *h) : desc(d), helper(h) {}
349 
350   size_t getSize() const override { return sizeof(tailMergeX86); }
351 
352   void writeTo(uint8_t *buf) const override {
353     memcpy(buf, tailMergeX86, sizeof(tailMergeX86));
354     write32le(buf + 4, desc->getRVA() + config->imageBase);
355     write32le(buf + 9, helper->getRVA() - rva - 13);
356   }
357 
358   void getBaserels(std::vector<Baserel> *res) override {
359     res->emplace_back(rva + 4);
360   }
361 
362   Chunk *desc = nullptr;
363   Defined *helper = nullptr;
364 };
365 
366 class ThunkChunkARM : public NonSectionChunk {
367 public:
368   ThunkChunkARM(Defined *i, Chunk *tm) : imp(i), tailMerge(tm) {}
369 
370   size_t getSize() const override { return sizeof(thunkARM); }
371 
372   void writeTo(uint8_t *buf) const override {
373     memcpy(buf, thunkARM, sizeof(thunkARM));
374     applyMOV32T(buf + 0, imp->getRVA() + config->imageBase);
375     applyBranch24T(buf + 8, tailMerge->getRVA() - rva - 12);
376   }
377 
378   void getBaserels(std::vector<Baserel> *res) override {
379     res->emplace_back(rva + 0, IMAGE_REL_BASED_ARM_MOV32T);
380   }
381 
382   Defined *imp = nullptr;
383   Chunk *tailMerge = nullptr;
384 };
385 
386 class TailMergeChunkARM : public NonSectionChunk {
387 public:
388   TailMergeChunkARM(Chunk *d, Defined *h) : desc(d), helper(h) {}
389 
390   size_t getSize() const override { return sizeof(tailMergeARM); }
391 
392   void writeTo(uint8_t *buf) const override {
393     memcpy(buf, tailMergeARM, sizeof(tailMergeARM));
394     applyMOV32T(buf + 14, desc->getRVA() + config->imageBase);
395     applyBranch24T(buf + 22, helper->getRVA() - rva - 26);
396   }
397 
398   void getBaserels(std::vector<Baserel> *res) override {
399     res->emplace_back(rva + 14, IMAGE_REL_BASED_ARM_MOV32T);
400   }
401 
402   Chunk *desc = nullptr;
403   Defined *helper = nullptr;
404 };
405 
406 class ThunkChunkARM64 : public NonSectionChunk {
407 public:
408   ThunkChunkARM64(Defined *i, Chunk *tm) : imp(i), tailMerge(tm) {}
409 
410   size_t getSize() const override { return sizeof(thunkARM64); }
411 
412   void writeTo(uint8_t *buf) const override {
413     memcpy(buf, thunkARM64, sizeof(thunkARM64));
414     applyArm64Addr(buf + 0, imp->getRVA(), rva + 0, 12);
415     applyArm64Imm(buf + 4, imp->getRVA() & 0xfff, 0);
416     applyArm64Branch26(buf + 8, tailMerge->getRVA() - rva - 8);
417   }
418 
419   Defined *imp = nullptr;
420   Chunk *tailMerge = nullptr;
421 };
422 
423 class TailMergeChunkARM64 : public NonSectionChunk {
424 public:
425   TailMergeChunkARM64(Chunk *d, Defined *h) : desc(d), helper(h) {}
426 
427   size_t getSize() const override { return sizeof(tailMergeARM64); }
428 
429   void writeTo(uint8_t *buf) const override {
430     memcpy(buf, tailMergeARM64, sizeof(tailMergeARM64));
431     applyArm64Addr(buf + 44, desc->getRVA(), rva + 44, 12);
432     applyArm64Imm(buf + 48, desc->getRVA() & 0xfff, 0);
433     applyArm64Branch26(buf + 52, helper->getRVA() - rva - 52);
434   }
435 
436   Chunk *desc = nullptr;
437   Defined *helper = nullptr;
438 };
439 
440 // A chunk for the import descriptor table.
441 class DelayAddressChunk : public NonSectionChunk {
442 public:
443   explicit DelayAddressChunk(Chunk *c) : thunk(c) {
444     setAlignment(config->wordsize);
445   }
446   size_t getSize() const override { return config->wordsize; }
447 
448   void writeTo(uint8_t *buf) const override {
449     if (config->is64()) {
450       write64le(buf, thunk->getRVA() + config->imageBase);
451     } else {
452       uint32_t bit = 0;
453       // Pointer to thumb code must have the LSB set, so adjust it.
454       if (config->machine == ARMNT)
455         bit = 1;
456       write32le(buf, (thunk->getRVA() + config->imageBase) | bit);
457     }
458   }
459 
460   void getBaserels(std::vector<Baserel> *res) override {
461     res->emplace_back(rva);
462   }
463 
464   Chunk *thunk;
465 };
466 
467 // Export table
468 // Read Microsoft PE/COFF spec 5.3 for details.
469 
470 // A chunk for the export descriptor table.
471 class ExportDirectoryChunk : public NonSectionChunk {
472 public:
473   ExportDirectoryChunk(int i, int j, Chunk *d, Chunk *a, Chunk *n, Chunk *o)
474       : maxOrdinal(i), nameTabSize(j), dllName(d), addressTab(a), nameTab(n),
475         ordinalTab(o) {}
476 
477   size_t getSize() const override {
478     return sizeof(export_directory_table_entry);
479   }
480 
481   void writeTo(uint8_t *buf) const override {
482     memset(buf, 0, getSize());
483 
484     auto *e = (export_directory_table_entry *)(buf);
485     e->NameRVA = dllName->getRVA();
486     e->OrdinalBase = 0;
487     e->AddressTableEntries = maxOrdinal + 1;
488     e->NumberOfNamePointers = nameTabSize;
489     e->ExportAddressTableRVA = addressTab->getRVA();
490     e->NamePointerRVA = nameTab->getRVA();
491     e->OrdinalTableRVA = ordinalTab->getRVA();
492   }
493 
494   uint16_t maxOrdinal;
495   uint16_t nameTabSize;
496   Chunk *dllName;
497   Chunk *addressTab;
498   Chunk *nameTab;
499   Chunk *ordinalTab;
500 };
501 
502 class AddressTableChunk : public NonSectionChunk {
503 public:
504   explicit AddressTableChunk(size_t maxOrdinal) : size(maxOrdinal + 1) {}
505   size_t getSize() const override { return size * 4; }
506 
507   void writeTo(uint8_t *buf) const override {
508     memset(buf, 0, getSize());
509 
510     for (const Export &e : config->exports) {
511       uint8_t *p = buf + e.ordinal * 4;
512       uint32_t bit = 0;
513       // Pointer to thumb code must have the LSB set, so adjust it.
514       if (config->machine == ARMNT && !e.data)
515         bit = 1;
516       if (e.forwardChunk) {
517         write32le(p, e.forwardChunk->getRVA() | bit);
518       } else {
519         write32le(p, cast<Defined>(e.sym)->getRVA() | bit);
520       }
521     }
522   }
523 
524 private:
525   size_t size;
526 };
527 
528 class NamePointersChunk : public NonSectionChunk {
529 public:
530   explicit NamePointersChunk(std::vector<Chunk *> &v) : chunks(v) {}
531   size_t getSize() const override { return chunks.size() * 4; }
532 
533   void writeTo(uint8_t *buf) const override {
534     for (Chunk *c : chunks) {
535       write32le(buf, c->getRVA());
536       buf += 4;
537     }
538   }
539 
540 private:
541   std::vector<Chunk *> chunks;
542 };
543 
544 class ExportOrdinalChunk : public NonSectionChunk {
545 public:
546   explicit ExportOrdinalChunk(size_t i) : size(i) {}
547   size_t getSize() const override { return size * 2; }
548 
549   void writeTo(uint8_t *buf) const override {
550     for (Export &e : config->exports) {
551       if (e.noname)
552         continue;
553       write16le(buf, e.ordinal);
554       buf += 2;
555     }
556   }
557 
558 private:
559   size_t size;
560 };
561 
562 } // anonymous namespace
563 
564 void IdataContents::create() {
565   std::vector<std::vector<DefinedImportData *>> v = binImports(imports);
566 
567   // Create .idata contents for each DLL.
568   for (std::vector<DefinedImportData *> &syms : v) {
569     // Create lookup and address tables. If they have external names,
570     // we need to create hintName chunks to store the names.
571     // If they don't (if they are import-by-ordinals), we store only
572     // ordinal values to the table.
573     size_t base = lookups.size();
574     for (DefinedImportData *s : syms) {
575       uint16_t ord = s->getOrdinal();
576       if (s->getExternalName().empty()) {
577         lookups.push_back(make<OrdinalOnlyChunk>(ord));
578         addresses.push_back(make<OrdinalOnlyChunk>(ord));
579         continue;
580       }
581       auto *c = make<HintNameChunk>(s->getExternalName(), ord);
582       lookups.push_back(make<LookupChunk>(c));
583       addresses.push_back(make<LookupChunk>(c));
584       hints.push_back(c);
585     }
586     // Terminate with null values.
587     lookups.push_back(make<NullChunk>(config->wordsize));
588     addresses.push_back(make<NullChunk>(config->wordsize));
589 
590     for (int i = 0, e = syms.size(); i < e; ++i)
591       syms[i]->setLocation(addresses[base + i]);
592 
593     // Create the import table header.
594     dllNames.push_back(make<StringChunk>(syms[0]->getDLLName()));
595     auto *dir = make<ImportDirectoryChunk>(dllNames.back());
596     dir->lookupTab = lookups[base];
597     dir->addressTab = addresses[base];
598     dirs.push_back(dir);
599   }
600   // Add null terminator.
601   dirs.push_back(make<NullChunk>(sizeof(ImportDirectoryTableEntry)));
602 }
603 
604 std::vector<Chunk *> DelayLoadContents::getChunks() {
605   std::vector<Chunk *> v;
606   v.insert(v.end(), dirs.begin(), dirs.end());
607   v.insert(v.end(), names.begin(), names.end());
608   v.insert(v.end(), hintNames.begin(), hintNames.end());
609   v.insert(v.end(), dllNames.begin(), dllNames.end());
610   return v;
611 }
612 
613 std::vector<Chunk *> DelayLoadContents::getDataChunks() {
614   std::vector<Chunk *> v;
615   v.insert(v.end(), moduleHandles.begin(), moduleHandles.end());
616   v.insert(v.end(), addresses.begin(), addresses.end());
617   return v;
618 }
619 
620 uint64_t DelayLoadContents::getDirSize() {
621   return dirs.size() * sizeof(delay_import_directory_table_entry);
622 }
623 
624 void DelayLoadContents::create(Defined *h) {
625   helper = h;
626   std::vector<std::vector<DefinedImportData *>> v = binImports(imports);
627 
628   // Create .didat contents for each DLL.
629   for (std::vector<DefinedImportData *> &syms : v) {
630     // Create the delay import table header.
631     dllNames.push_back(make<StringChunk>(syms[0]->getDLLName()));
632     auto *dir = make<DelayDirectoryChunk>(dllNames.back());
633 
634     size_t base = addresses.size();
635     Chunk *tm = newTailMergeChunk(dir);
636     for (DefinedImportData *s : syms) {
637       Chunk *t = newThunkChunk(s, tm);
638       auto *a = make<DelayAddressChunk>(t);
639       addresses.push_back(a);
640       thunks.push_back(t);
641       StringRef extName = s->getExternalName();
642       if (extName.empty()) {
643         names.push_back(make<OrdinalOnlyChunk>(s->getOrdinal()));
644       } else {
645         auto *c = make<HintNameChunk>(extName, 0);
646         names.push_back(make<LookupChunk>(c));
647         hintNames.push_back(c);
648       }
649     }
650     thunks.push_back(tm);
651     // Terminate with null values.
652     addresses.push_back(make<NullChunk>(8));
653     names.push_back(make<NullChunk>(8));
654 
655     for (int i = 0, e = syms.size(); i < e; ++i)
656       syms[i]->setLocation(addresses[base + i]);
657     auto *mh = make<NullChunk>(8);
658     mh->setAlignment(8);
659     moduleHandles.push_back(mh);
660 
661     // Fill the delay import table header fields.
662     dir->moduleHandle = mh;
663     dir->addressTab = addresses[base];
664     dir->nameTab = names[base];
665     dirs.push_back(dir);
666   }
667   // Add null terminator.
668   dirs.push_back(make<NullChunk>(sizeof(delay_import_directory_table_entry)));
669 }
670 
671 Chunk *DelayLoadContents::newTailMergeChunk(Chunk *dir) {
672   switch (config->machine) {
673   case AMD64:
674     return make<TailMergeChunkX64>(dir, helper);
675   case I386:
676     return make<TailMergeChunkX86>(dir, helper);
677   case ARMNT:
678     return make<TailMergeChunkARM>(dir, helper);
679   case ARM64:
680     return make<TailMergeChunkARM64>(dir, helper);
681   default:
682     llvm_unreachable("unsupported machine type");
683   }
684 }
685 
686 Chunk *DelayLoadContents::newThunkChunk(DefinedImportData *s,
687                                         Chunk *tailMerge) {
688   switch (config->machine) {
689   case AMD64:
690     return make<ThunkChunkX64>(s, tailMerge);
691   case I386:
692     return make<ThunkChunkX86>(s, tailMerge);
693   case ARMNT:
694     return make<ThunkChunkARM>(s, tailMerge);
695   case ARM64:
696     return make<ThunkChunkARM64>(s, tailMerge);
697   default:
698     llvm_unreachable("unsupported machine type");
699   }
700 }
701 
702 EdataContents::EdataContents() {
703   uint16_t maxOrdinal = 0;
704   for (Export &e : config->exports)
705     maxOrdinal = std::max(maxOrdinal, e.ordinal);
706 
707   auto *dllName = make<StringChunk>(sys::path::filename(config->outputFile));
708   auto *addressTab = make<AddressTableChunk>(maxOrdinal);
709   std::vector<Chunk *> names;
710   for (Export &e : config->exports)
711     if (!e.noname)
712       names.push_back(make<StringChunk>(e.exportName));
713 
714   std::vector<Chunk *> forwards;
715   for (Export &e : config->exports) {
716     if (e.forwardTo.empty())
717       continue;
718     e.forwardChunk = make<StringChunk>(e.forwardTo);
719     forwards.push_back(e.forwardChunk);
720   }
721 
722   auto *nameTab = make<NamePointersChunk>(names);
723   auto *ordinalTab = make<ExportOrdinalChunk>(names.size());
724   auto *dir = make<ExportDirectoryChunk>(maxOrdinal, names.size(), dllName,
725                                          addressTab, nameTab, ordinalTab);
726   chunks.push_back(dir);
727   chunks.push_back(dllName);
728   chunks.push_back(addressTab);
729   chunks.push_back(nameTab);
730   chunks.push_back(ordinalTab);
731   chunks.insert(chunks.end(), names.begin(), names.end());
732   chunks.insert(chunks.end(), forwards.begin(), forwards.end());
733 }
734 
735 } // namespace coff
736 } // namespace lld
737