10b57cec5SDimitry Andric //===- COFFImportFile.cpp - COFF short import file implementation ---------===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric //
90b57cec5SDimitry Andric // This file defines the writeImportLibrary function.
100b57cec5SDimitry Andric //
110b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
120b57cec5SDimitry Andric 
130b57cec5SDimitry Andric #include "llvm/Object/COFFImportFile.h"
140b57cec5SDimitry Andric #include "llvm/ADT/ArrayRef.h"
1581ad6265SDimitry Andric #include "llvm/ADT/Twine.h"
160b57cec5SDimitry Andric #include "llvm/Object/Archive.h"
170b57cec5SDimitry Andric #include "llvm/Object/ArchiveWriter.h"
180b57cec5SDimitry Andric #include "llvm/Object/COFF.h"
1981ad6265SDimitry Andric #include "llvm/Support/Allocator.h"
2081ad6265SDimitry Andric #include "llvm/Support/Endian.h"
210b57cec5SDimitry Andric #include "llvm/Support/Error.h"
2281ad6265SDimitry Andric #include "llvm/Support/ErrorHandling.h"
230b57cec5SDimitry Andric #include "llvm/Support/Path.h"
240b57cec5SDimitry Andric 
250b57cec5SDimitry Andric #include <cstdint>
260b57cec5SDimitry Andric #include <string>
270b57cec5SDimitry Andric #include <vector>
280b57cec5SDimitry Andric 
290b57cec5SDimitry Andric using namespace llvm::COFF;
300b57cec5SDimitry Andric using namespace llvm::object;
310b57cec5SDimitry Andric using namespace llvm;
320b57cec5SDimitry Andric 
330b57cec5SDimitry Andric namespace llvm {
340b57cec5SDimitry Andric namespace object {
350b57cec5SDimitry Andric 
getFileFormatName() const367a6dacacSDimitry Andric StringRef COFFImportFile::getFileFormatName() const {
377a6dacacSDimitry Andric   switch (getMachine()) {
387a6dacacSDimitry Andric   case COFF::IMAGE_FILE_MACHINE_I386:
397a6dacacSDimitry Andric     return "COFF-import-file-i386";
407a6dacacSDimitry Andric   case COFF::IMAGE_FILE_MACHINE_AMD64:
417a6dacacSDimitry Andric     return "COFF-import-file-x86-64";
427a6dacacSDimitry Andric   case COFF::IMAGE_FILE_MACHINE_ARMNT:
437a6dacacSDimitry Andric     return "COFF-import-file-ARM";
447a6dacacSDimitry Andric   case COFF::IMAGE_FILE_MACHINE_ARM64:
457a6dacacSDimitry Andric     return "COFF-import-file-ARM64";
467a6dacacSDimitry Andric   case COFF::IMAGE_FILE_MACHINE_ARM64EC:
477a6dacacSDimitry Andric     return "COFF-import-file-ARM64EC";
487a6dacacSDimitry Andric   case COFF::IMAGE_FILE_MACHINE_ARM64X:
497a6dacacSDimitry Andric     return "COFF-import-file-ARM64X";
507a6dacacSDimitry Andric   default:
517a6dacacSDimitry Andric     return "COFF-import-file-<unknown arch>";
527a6dacacSDimitry Andric   }
537a6dacacSDimitry Andric }
547a6dacacSDimitry Andric 
getExportName() const55439352acSDimitry Andric StringRef COFFImportFile::getExportName() const {
56439352acSDimitry Andric   const coff_import_header *hdr = getCOFFImportHeader();
57439352acSDimitry Andric   StringRef name = Data.getBuffer().substr(sizeof(*hdr)).split('\0').first;
58439352acSDimitry Andric 
59439352acSDimitry Andric   auto ltrim1 = [](StringRef s, StringRef chars) {
60439352acSDimitry Andric     return !s.empty() && chars.contains(s[0]) ? s.substr(1) : s;
61439352acSDimitry Andric   };
62439352acSDimitry Andric 
63439352acSDimitry Andric   switch (hdr->getNameType()) {
64439352acSDimitry Andric   case IMPORT_ORDINAL:
65439352acSDimitry Andric     name = "";
66439352acSDimitry Andric     break;
67439352acSDimitry Andric   case IMPORT_NAME_NOPREFIX:
68439352acSDimitry Andric     name = ltrim1(name, "?@_");
69439352acSDimitry Andric     break;
70439352acSDimitry Andric   case IMPORT_NAME_UNDECORATE:
71439352acSDimitry Andric     name = ltrim1(name, "?@_");
72439352acSDimitry Andric     name = name.substr(0, name.find('@'));
73439352acSDimitry Andric     break;
74439352acSDimitry Andric   case IMPORT_NAME_EXPORTAS: {
75439352acSDimitry Andric     // Skip DLL name
76439352acSDimitry Andric     name = Data.getBuffer().substr(sizeof(*hdr) + name.size() + 1);
77439352acSDimitry Andric     name = name.split('\0').second.split('\0').first;
78439352acSDimitry Andric     break;
79439352acSDimitry Andric   }
80439352acSDimitry Andric   default:
81439352acSDimitry Andric     break;
82439352acSDimitry Andric   }
83439352acSDimitry Andric 
84439352acSDimitry Andric   return name;
85439352acSDimitry Andric }
86439352acSDimitry Andric 
getImgRelRelocation(MachineTypes Machine)870b57cec5SDimitry Andric static uint16_t getImgRelRelocation(MachineTypes Machine) {
880b57cec5SDimitry Andric   switch (Machine) {
890b57cec5SDimitry Andric   default:
900b57cec5SDimitry Andric     llvm_unreachable("unsupported machine");
910b57cec5SDimitry Andric   case IMAGE_FILE_MACHINE_AMD64:
920b57cec5SDimitry Andric     return IMAGE_REL_AMD64_ADDR32NB;
930b57cec5SDimitry Andric   case IMAGE_FILE_MACHINE_ARMNT:
940b57cec5SDimitry Andric     return IMAGE_REL_ARM_ADDR32NB;
950b57cec5SDimitry Andric   case IMAGE_FILE_MACHINE_ARM64:
96bdd1243dSDimitry Andric   case IMAGE_FILE_MACHINE_ARM64EC:
9706c3fb27SDimitry Andric   case IMAGE_FILE_MACHINE_ARM64X:
980b57cec5SDimitry Andric     return IMAGE_REL_ARM64_ADDR32NB;
990b57cec5SDimitry Andric   case IMAGE_FILE_MACHINE_I386:
1000b57cec5SDimitry Andric     return IMAGE_REL_I386_DIR32NB;
1010b57cec5SDimitry Andric   }
1020b57cec5SDimitry Andric }
1030b57cec5SDimitry Andric 
append(std::vector<uint8_t> & B,const T & Data)1040b57cec5SDimitry Andric template <class T> static void append(std::vector<uint8_t> &B, const T &Data) {
1050b57cec5SDimitry Andric   size_t S = B.size();
1060b57cec5SDimitry Andric   B.resize(S + sizeof(T));
1070b57cec5SDimitry Andric   memcpy(&B[S], &Data, sizeof(T));
1080b57cec5SDimitry Andric }
1090b57cec5SDimitry Andric 
writeStringTable(std::vector<uint8_t> & B,ArrayRef<const std::string_view> Strings)1100b57cec5SDimitry Andric static void writeStringTable(std::vector<uint8_t> &B,
111439352acSDimitry Andric                              ArrayRef<const std::string_view> Strings) {
1120b57cec5SDimitry Andric   // The COFF string table consists of a 4-byte value which is the size of the
1130b57cec5SDimitry Andric   // table, including the length field itself.  This value is followed by the
1140b57cec5SDimitry Andric   // string content itself, which is an array of null-terminated C-style
1150b57cec5SDimitry Andric   // strings.  The termination is important as they are referenced to by offset
1160b57cec5SDimitry Andric   // by the symbol entity in the file format.
1170b57cec5SDimitry Andric 
1180b57cec5SDimitry Andric   size_t Pos = B.size();
1190b57cec5SDimitry Andric   size_t Offset = B.size();
1200b57cec5SDimitry Andric 
1210b57cec5SDimitry Andric   // Skip over the length field, we will fill it in later as we will have
1220b57cec5SDimitry Andric   // computed the length while emitting the string content itself.
1230b57cec5SDimitry Andric   Pos += sizeof(uint32_t);
1240b57cec5SDimitry Andric 
1250b57cec5SDimitry Andric   for (const auto &S : Strings) {
1260b57cec5SDimitry Andric     B.resize(Pos + S.length() + 1);
12706c3fb27SDimitry Andric     std::copy(S.begin(), S.end(), std::next(B.begin(), Pos));
12806c3fb27SDimitry Andric     B[Pos + S.length()] = 0;
1290b57cec5SDimitry Andric     Pos += S.length() + 1;
1300b57cec5SDimitry Andric   }
1310b57cec5SDimitry Andric 
1320b57cec5SDimitry Andric   // Backfill the length of the table now that it has been computed.
1330b57cec5SDimitry Andric   support::ulittle32_t Length(B.size() - Offset);
1340b57cec5SDimitry Andric   support::endian::write32le(&B[Offset], Length);
1350b57cec5SDimitry Andric }
1360b57cec5SDimitry Andric 
getNameType(StringRef Sym,StringRef ExtName,MachineTypes Machine,bool MinGW)1370b57cec5SDimitry Andric static ImportNameType getNameType(StringRef Sym, StringRef ExtName,
1380b57cec5SDimitry Andric                                   MachineTypes Machine, bool MinGW) {
1390b57cec5SDimitry Andric   // A decorated stdcall function in MSVC is exported with the
1400b57cec5SDimitry Andric   // type IMPORT_NAME, and the exported function name includes the
1410b57cec5SDimitry Andric   // the leading underscore. In MinGW on the other hand, a decorated
1420b57cec5SDimitry Andric   // stdcall function still omits the underscore (IMPORT_NAME_NOPREFIX).
1430b57cec5SDimitry Andric   // See the comment in isDecorated in COFFModuleDefinition.cpp for more
1440b57cec5SDimitry Andric   // details.
1455f757f3fSDimitry Andric   if (ExtName.starts_with("_") && ExtName.contains('@') && !MinGW)
1460b57cec5SDimitry Andric     return IMPORT_NAME;
1470b57cec5SDimitry Andric   if (Sym != ExtName)
1480b57cec5SDimitry Andric     return IMPORT_NAME_UNDECORATE;
1495f757f3fSDimitry Andric   if (Machine == IMAGE_FILE_MACHINE_I386 && Sym.starts_with("_"))
1500b57cec5SDimitry Andric     return IMPORT_NAME_NOPREFIX;
1510b57cec5SDimitry Andric   return IMPORT_NAME;
1520b57cec5SDimitry Andric }
1530b57cec5SDimitry Andric 
replace(StringRef S,StringRef From,StringRef To)1540b57cec5SDimitry Andric static Expected<std::string> replace(StringRef S, StringRef From,
1550b57cec5SDimitry Andric                                      StringRef To) {
1560b57cec5SDimitry Andric   size_t Pos = S.find(From);
1570b57cec5SDimitry Andric 
1580b57cec5SDimitry Andric   // From and To may be mangled, but substrings in S may not.
1595f757f3fSDimitry Andric   if (Pos == StringRef::npos && From.starts_with("_") && To.starts_with("_")) {
1600b57cec5SDimitry Andric     From = From.substr(1);
1610b57cec5SDimitry Andric     To = To.substr(1);
1620b57cec5SDimitry Andric     Pos = S.find(From);
1630b57cec5SDimitry Andric   }
1640b57cec5SDimitry Andric 
1650b57cec5SDimitry Andric   if (Pos == StringRef::npos) {
1660b57cec5SDimitry Andric     return make_error<StringError>(
1670b57cec5SDimitry Andric       StringRef(Twine(S + ": replacing '" + From +
1680b57cec5SDimitry Andric         "' with '" + To + "' failed").str()), object_error::parse_failed);
1690b57cec5SDimitry Andric   }
1700b57cec5SDimitry Andric 
1710b57cec5SDimitry Andric   return (Twine(S.substr(0, Pos)) + To + S.substr(Pos + From.size())).str();
1720b57cec5SDimitry Andric }
1730b57cec5SDimitry Andric 
1740b57cec5SDimitry Andric namespace {
1750b57cec5SDimitry Andric // This class constructs various small object files necessary to support linking
1760b57cec5SDimitry Andric // symbols imported from a DLL.  The contents are pretty strictly defined and
1770b57cec5SDimitry Andric // nearly entirely static.  The details of the structures files are defined in
1780b57cec5SDimitry Andric // WINNT.h and the PE/COFF specification.
1790b57cec5SDimitry Andric class ObjectFactory {
1800b57cec5SDimitry Andric   using u16 = support::ulittle16_t;
1810b57cec5SDimitry Andric   using u32 = support::ulittle32_t;
1827a6dacacSDimitry Andric   MachineTypes NativeMachine;
1830b57cec5SDimitry Andric   BumpPtrAllocator Alloc;
1840b57cec5SDimitry Andric   StringRef ImportName;
1850b57cec5SDimitry Andric   StringRef Library;
1860b57cec5SDimitry Andric   std::string ImportDescriptorSymbolName;
1870b57cec5SDimitry Andric   std::string NullThunkSymbolName;
1880b57cec5SDimitry Andric 
1890b57cec5SDimitry Andric public:
ObjectFactory(StringRef S,MachineTypes M)1900b57cec5SDimitry Andric   ObjectFactory(StringRef S, MachineTypes M)
1917a6dacacSDimitry Andric       : NativeMachine(M), ImportName(S), Library(llvm::sys::path::stem(S)),
192439352acSDimitry Andric         ImportDescriptorSymbolName((ImportDescriptorPrefix + Library).str()),
193439352acSDimitry Andric         NullThunkSymbolName(
194439352acSDimitry Andric             (NullThunkDataPrefix + Library + NullThunkDataSuffix).str()) {}
1950b57cec5SDimitry Andric 
1960b57cec5SDimitry Andric   // Creates an Import Descriptor.  This is a small object file which contains a
1970b57cec5SDimitry Andric   // reference to the terminators and contains the library name (entry) for the
1980b57cec5SDimitry Andric   // import name table.  It will force the linker to construct the necessary
1990b57cec5SDimitry Andric   // structure to import symbols from the DLL.
2000b57cec5SDimitry Andric   NewArchiveMember createImportDescriptor(std::vector<uint8_t> &Buffer);
2010b57cec5SDimitry Andric 
2020b57cec5SDimitry Andric   // Creates a NULL import descriptor.  This is a small object file whcih
2030b57cec5SDimitry Andric   // contains a NULL import descriptor.  It is used to terminate the imports
2040b57cec5SDimitry Andric   // from a specific DLL.
2050b57cec5SDimitry Andric   NewArchiveMember createNullImportDescriptor(std::vector<uint8_t> &Buffer);
2060b57cec5SDimitry Andric 
2070b57cec5SDimitry Andric   // Create a NULL Thunk Entry.  This is a small object file which contains a
2080b57cec5SDimitry Andric   // NULL Import Address Table entry and a NULL Import Lookup Table Entry.  It
2090b57cec5SDimitry Andric   // is used to terminate the IAT and ILT.
2100b57cec5SDimitry Andric   NewArchiveMember createNullThunk(std::vector<uint8_t> &Buffer);
2110b57cec5SDimitry Andric 
2120b57cec5SDimitry Andric   // Create a short import file which is described in PE/COFF spec 7. Import
2130b57cec5SDimitry Andric   // Library Format.
2140b57cec5SDimitry Andric   NewArchiveMember createShortImport(StringRef Sym, uint16_t Ordinal,
2157a6dacacSDimitry Andric                                      ImportType Type, ImportNameType NameType,
216439352acSDimitry Andric                                      StringRef ExportName,
2177a6dacacSDimitry Andric                                      MachineTypes Machine);
2180b57cec5SDimitry Andric 
2190b57cec5SDimitry Andric   // Create a weak external file which is described in PE/COFF Aux Format 3.
2207a6dacacSDimitry Andric   NewArchiveMember createWeakExternal(StringRef Sym, StringRef Weak, bool Imp,
2217a6dacacSDimitry Andric                                       MachineTypes Machine);
2227a6dacacSDimitry Andric 
is64Bit() const2237a6dacacSDimitry Andric   bool is64Bit() const { return COFF::is64Bit(NativeMachine); }
2240b57cec5SDimitry Andric };
2250b57cec5SDimitry Andric } // namespace
2260b57cec5SDimitry Andric 
2270b57cec5SDimitry Andric NewArchiveMember
createImportDescriptor(std::vector<uint8_t> & Buffer)2280b57cec5SDimitry Andric ObjectFactory::createImportDescriptor(std::vector<uint8_t> &Buffer) {
2290b57cec5SDimitry Andric   const uint32_t NumberOfSections = 2;
2300b57cec5SDimitry Andric   const uint32_t NumberOfSymbols = 7;
2310b57cec5SDimitry Andric   const uint32_t NumberOfRelocations = 3;
2320b57cec5SDimitry Andric 
2330b57cec5SDimitry Andric   // COFF Header
2340b57cec5SDimitry Andric   coff_file_header Header{
2357a6dacacSDimitry Andric       u16(NativeMachine),
2360b57cec5SDimitry Andric       u16(NumberOfSections),
2370b57cec5SDimitry Andric       u32(0),
2380b57cec5SDimitry Andric       u32(sizeof(Header) + (NumberOfSections * sizeof(coff_section)) +
2390b57cec5SDimitry Andric           // .idata$2
2400b57cec5SDimitry Andric           sizeof(coff_import_directory_table_entry) +
2410b57cec5SDimitry Andric           NumberOfRelocations * sizeof(coff_relocation) +
2420b57cec5SDimitry Andric           // .idata$4
2430b57cec5SDimitry Andric           (ImportName.size() + 1)),
2440b57cec5SDimitry Andric       u32(NumberOfSymbols),
2450b57cec5SDimitry Andric       u16(0),
2467a6dacacSDimitry Andric       u16(is64Bit() ? C_Invalid : IMAGE_FILE_32BIT_MACHINE),
2470b57cec5SDimitry Andric   };
2480b57cec5SDimitry Andric   append(Buffer, Header);
2490b57cec5SDimitry Andric 
2500b57cec5SDimitry Andric   // Section Header Table
2510b57cec5SDimitry Andric   const coff_section SectionTable[NumberOfSections] = {
2520b57cec5SDimitry Andric       {{'.', 'i', 'd', 'a', 't', 'a', '$', '2'},
2530b57cec5SDimitry Andric        u32(0),
2540b57cec5SDimitry Andric        u32(0),
2550b57cec5SDimitry Andric        u32(sizeof(coff_import_directory_table_entry)),
2560b57cec5SDimitry Andric        u32(sizeof(coff_file_header) + NumberOfSections * sizeof(coff_section)),
2570b57cec5SDimitry Andric        u32(sizeof(coff_file_header) + NumberOfSections * sizeof(coff_section) +
2580b57cec5SDimitry Andric            sizeof(coff_import_directory_table_entry)),
2590b57cec5SDimitry Andric        u32(0),
2600b57cec5SDimitry Andric        u16(NumberOfRelocations),
2610b57cec5SDimitry Andric        u16(0),
2620b57cec5SDimitry Andric        u32(IMAGE_SCN_ALIGN_4BYTES | IMAGE_SCN_CNT_INITIALIZED_DATA |
2630b57cec5SDimitry Andric            IMAGE_SCN_MEM_READ | IMAGE_SCN_MEM_WRITE)},
2640b57cec5SDimitry Andric       {{'.', 'i', 'd', 'a', 't', 'a', '$', '6'},
2650b57cec5SDimitry Andric        u32(0),
2660b57cec5SDimitry Andric        u32(0),
2670b57cec5SDimitry Andric        u32(ImportName.size() + 1),
2680b57cec5SDimitry Andric        u32(sizeof(coff_file_header) + NumberOfSections * sizeof(coff_section) +
2690b57cec5SDimitry Andric            sizeof(coff_import_directory_table_entry) +
2700b57cec5SDimitry Andric            NumberOfRelocations * sizeof(coff_relocation)),
2710b57cec5SDimitry Andric        u32(0),
2720b57cec5SDimitry Andric        u32(0),
2730b57cec5SDimitry Andric        u16(0),
2740b57cec5SDimitry Andric        u16(0),
2750b57cec5SDimitry Andric        u32(IMAGE_SCN_ALIGN_2BYTES | IMAGE_SCN_CNT_INITIALIZED_DATA |
2760b57cec5SDimitry Andric            IMAGE_SCN_MEM_READ | IMAGE_SCN_MEM_WRITE)},
2770b57cec5SDimitry Andric   };
2780b57cec5SDimitry Andric   append(Buffer, SectionTable);
2790b57cec5SDimitry Andric 
2800b57cec5SDimitry Andric   // .idata$2
2810b57cec5SDimitry Andric   const coff_import_directory_table_entry ImportDescriptor{
2820b57cec5SDimitry Andric       u32(0), u32(0), u32(0), u32(0), u32(0),
2830b57cec5SDimitry Andric   };
2840b57cec5SDimitry Andric   append(Buffer, ImportDescriptor);
2850b57cec5SDimitry Andric 
2860b57cec5SDimitry Andric   const coff_relocation RelocationTable[NumberOfRelocations] = {
2870b57cec5SDimitry Andric       {u32(offsetof(coff_import_directory_table_entry, NameRVA)), u32(2),
2887a6dacacSDimitry Andric        u16(getImgRelRelocation(NativeMachine))},
2890b57cec5SDimitry Andric       {u32(offsetof(coff_import_directory_table_entry, ImportLookupTableRVA)),
2907a6dacacSDimitry Andric        u32(3), u16(getImgRelRelocation(NativeMachine))},
2910b57cec5SDimitry Andric       {u32(offsetof(coff_import_directory_table_entry, ImportAddressTableRVA)),
2927a6dacacSDimitry Andric        u32(4), u16(getImgRelRelocation(NativeMachine))},
2930b57cec5SDimitry Andric   };
2940b57cec5SDimitry Andric   append(Buffer, RelocationTable);
2950b57cec5SDimitry Andric 
2960b57cec5SDimitry Andric   // .idata$6
2970b57cec5SDimitry Andric   auto S = Buffer.size();
2980b57cec5SDimitry Andric   Buffer.resize(S + ImportName.size() + 1);
2990b57cec5SDimitry Andric   memcpy(&Buffer[S], ImportName.data(), ImportName.size());
3000b57cec5SDimitry Andric   Buffer[S + ImportName.size()] = '\0';
3010b57cec5SDimitry Andric 
3020b57cec5SDimitry Andric   // Symbol Table
3030b57cec5SDimitry Andric   coff_symbol16 SymbolTable[NumberOfSymbols] = {
3040b57cec5SDimitry Andric       {{{0, 0, 0, 0, 0, 0, 0, 0}},
3050b57cec5SDimitry Andric        u32(0),
3060b57cec5SDimitry Andric        u16(1),
3070b57cec5SDimitry Andric        u16(0),
3080b57cec5SDimitry Andric        IMAGE_SYM_CLASS_EXTERNAL,
3090b57cec5SDimitry Andric        0},
3100b57cec5SDimitry Andric       {{{'.', 'i', 'd', 'a', 't', 'a', '$', '2'}},
3110b57cec5SDimitry Andric        u32(0),
3120b57cec5SDimitry Andric        u16(1),
3130b57cec5SDimitry Andric        u16(0),
3140b57cec5SDimitry Andric        IMAGE_SYM_CLASS_SECTION,
3150b57cec5SDimitry Andric        0},
3160b57cec5SDimitry Andric       {{{'.', 'i', 'd', 'a', 't', 'a', '$', '6'}},
3170b57cec5SDimitry Andric        u32(0),
3180b57cec5SDimitry Andric        u16(2),
3190b57cec5SDimitry Andric        u16(0),
3200b57cec5SDimitry Andric        IMAGE_SYM_CLASS_STATIC,
3210b57cec5SDimitry Andric        0},
3220b57cec5SDimitry Andric       {{{'.', 'i', 'd', 'a', 't', 'a', '$', '4'}},
3230b57cec5SDimitry Andric        u32(0),
3240b57cec5SDimitry Andric        u16(0),
3250b57cec5SDimitry Andric        u16(0),
3260b57cec5SDimitry Andric        IMAGE_SYM_CLASS_SECTION,
3270b57cec5SDimitry Andric        0},
3280b57cec5SDimitry Andric       {{{'.', 'i', 'd', 'a', 't', 'a', '$', '5'}},
3290b57cec5SDimitry Andric        u32(0),
3300b57cec5SDimitry Andric        u16(0),
3310b57cec5SDimitry Andric        u16(0),
3320b57cec5SDimitry Andric        IMAGE_SYM_CLASS_SECTION,
3330b57cec5SDimitry Andric        0},
3340b57cec5SDimitry Andric       {{{0, 0, 0, 0, 0, 0, 0, 0}},
3350b57cec5SDimitry Andric        u32(0),
3360b57cec5SDimitry Andric        u16(0),
3370b57cec5SDimitry Andric        u16(0),
3380b57cec5SDimitry Andric        IMAGE_SYM_CLASS_EXTERNAL,
3390b57cec5SDimitry Andric        0},
3400b57cec5SDimitry Andric       {{{0, 0, 0, 0, 0, 0, 0, 0}},
3410b57cec5SDimitry Andric        u32(0),
3420b57cec5SDimitry Andric        u16(0),
3430b57cec5SDimitry Andric        u16(0),
3440b57cec5SDimitry Andric        IMAGE_SYM_CLASS_EXTERNAL,
3450b57cec5SDimitry Andric        0},
3460b57cec5SDimitry Andric   };
3470b57cec5SDimitry Andric   // TODO: Name.Offset.Offset here and in the all similar places below
3480b57cec5SDimitry Andric   // suggests a names refactoring. Maybe StringTableOffset.Value?
3490b57cec5SDimitry Andric   SymbolTable[0].Name.Offset.Offset =
3500b57cec5SDimitry Andric       sizeof(uint32_t);
3510b57cec5SDimitry Andric   SymbolTable[5].Name.Offset.Offset =
3520b57cec5SDimitry Andric       sizeof(uint32_t) + ImportDescriptorSymbolName.length() + 1;
3530b57cec5SDimitry Andric   SymbolTable[6].Name.Offset.Offset =
3540b57cec5SDimitry Andric       sizeof(uint32_t) + ImportDescriptorSymbolName.length() + 1 +
3550b57cec5SDimitry Andric       NullImportDescriptorSymbolName.length() + 1;
3560b57cec5SDimitry Andric   append(Buffer, SymbolTable);
3570b57cec5SDimitry Andric 
3580b57cec5SDimitry Andric   // String Table
3590b57cec5SDimitry Andric   writeStringTable(Buffer,
3600b57cec5SDimitry Andric                    {ImportDescriptorSymbolName, NullImportDescriptorSymbolName,
3610b57cec5SDimitry Andric                     NullThunkSymbolName});
3620b57cec5SDimitry Andric 
3630b57cec5SDimitry Andric   StringRef F{reinterpret_cast<const char *>(Buffer.data()), Buffer.size()};
3640b57cec5SDimitry Andric   return {MemoryBufferRef(F, ImportName)};
3650b57cec5SDimitry Andric }
3660b57cec5SDimitry Andric 
3670b57cec5SDimitry Andric NewArchiveMember
createNullImportDescriptor(std::vector<uint8_t> & Buffer)3680b57cec5SDimitry Andric ObjectFactory::createNullImportDescriptor(std::vector<uint8_t> &Buffer) {
3690b57cec5SDimitry Andric   const uint32_t NumberOfSections = 1;
3700b57cec5SDimitry Andric   const uint32_t NumberOfSymbols = 1;
3710b57cec5SDimitry Andric 
3720b57cec5SDimitry Andric   // COFF Header
3730b57cec5SDimitry Andric   coff_file_header Header{
3747a6dacacSDimitry Andric       u16(NativeMachine),
3750b57cec5SDimitry Andric       u16(NumberOfSections),
3760b57cec5SDimitry Andric       u32(0),
3770b57cec5SDimitry Andric       u32(sizeof(Header) + (NumberOfSections * sizeof(coff_section)) +
3780b57cec5SDimitry Andric           // .idata$3
3790b57cec5SDimitry Andric           sizeof(coff_import_directory_table_entry)),
3800b57cec5SDimitry Andric       u32(NumberOfSymbols),
3810b57cec5SDimitry Andric       u16(0),
3827a6dacacSDimitry Andric       u16(is64Bit() ? C_Invalid : IMAGE_FILE_32BIT_MACHINE),
3830b57cec5SDimitry Andric   };
3840b57cec5SDimitry Andric   append(Buffer, Header);
3850b57cec5SDimitry Andric 
3860b57cec5SDimitry Andric   // Section Header Table
3870b57cec5SDimitry Andric   const coff_section SectionTable[NumberOfSections] = {
3880b57cec5SDimitry Andric       {{'.', 'i', 'd', 'a', 't', 'a', '$', '3'},
3890b57cec5SDimitry Andric        u32(0),
3900b57cec5SDimitry Andric        u32(0),
3910b57cec5SDimitry Andric        u32(sizeof(coff_import_directory_table_entry)),
3920b57cec5SDimitry Andric        u32(sizeof(coff_file_header) +
3930b57cec5SDimitry Andric            (NumberOfSections * sizeof(coff_section))),
3940b57cec5SDimitry Andric        u32(0),
3950b57cec5SDimitry Andric        u32(0),
3960b57cec5SDimitry Andric        u16(0),
3970b57cec5SDimitry Andric        u16(0),
3980b57cec5SDimitry Andric        u32(IMAGE_SCN_ALIGN_4BYTES | IMAGE_SCN_CNT_INITIALIZED_DATA |
3990b57cec5SDimitry Andric            IMAGE_SCN_MEM_READ | IMAGE_SCN_MEM_WRITE)},
4000b57cec5SDimitry Andric   };
4010b57cec5SDimitry Andric   append(Buffer, SectionTable);
4020b57cec5SDimitry Andric 
4030b57cec5SDimitry Andric   // .idata$3
4040b57cec5SDimitry Andric   const coff_import_directory_table_entry ImportDescriptor{
4050b57cec5SDimitry Andric       u32(0), u32(0), u32(0), u32(0), u32(0),
4060b57cec5SDimitry Andric   };
4070b57cec5SDimitry Andric   append(Buffer, ImportDescriptor);
4080b57cec5SDimitry Andric 
4090b57cec5SDimitry Andric   // Symbol Table
4100b57cec5SDimitry Andric   coff_symbol16 SymbolTable[NumberOfSymbols] = {
4110b57cec5SDimitry Andric       {{{0, 0, 0, 0, 0, 0, 0, 0}},
4120b57cec5SDimitry Andric        u32(0),
4130b57cec5SDimitry Andric        u16(1),
4140b57cec5SDimitry Andric        u16(0),
4150b57cec5SDimitry Andric        IMAGE_SYM_CLASS_EXTERNAL,
4160b57cec5SDimitry Andric        0},
4170b57cec5SDimitry Andric   };
4180b57cec5SDimitry Andric   SymbolTable[0].Name.Offset.Offset = sizeof(uint32_t);
4190b57cec5SDimitry Andric   append(Buffer, SymbolTable);
4200b57cec5SDimitry Andric 
4210b57cec5SDimitry Andric   // String Table
4220b57cec5SDimitry Andric   writeStringTable(Buffer, {NullImportDescriptorSymbolName});
4230b57cec5SDimitry Andric 
4240b57cec5SDimitry Andric   StringRef F{reinterpret_cast<const char *>(Buffer.data()), Buffer.size()};
4250b57cec5SDimitry Andric   return {MemoryBufferRef(F, ImportName)};
4260b57cec5SDimitry Andric }
4270b57cec5SDimitry Andric 
createNullThunk(std::vector<uint8_t> & Buffer)4280b57cec5SDimitry Andric NewArchiveMember ObjectFactory::createNullThunk(std::vector<uint8_t> &Buffer) {
4290b57cec5SDimitry Andric   const uint32_t NumberOfSections = 2;
4300b57cec5SDimitry Andric   const uint32_t NumberOfSymbols = 1;
4317a6dacacSDimitry Andric   uint32_t VASize = is64Bit() ? 8 : 4;
4320b57cec5SDimitry Andric 
4330b57cec5SDimitry Andric   // COFF Header
4340b57cec5SDimitry Andric   coff_file_header Header{
4357a6dacacSDimitry Andric       u16(NativeMachine),
4360b57cec5SDimitry Andric       u16(NumberOfSections),
4370b57cec5SDimitry Andric       u32(0),
4380b57cec5SDimitry Andric       u32(sizeof(Header) + (NumberOfSections * sizeof(coff_section)) +
4390b57cec5SDimitry Andric           // .idata$5
4400b57cec5SDimitry Andric           VASize +
4410b57cec5SDimitry Andric           // .idata$4
4420b57cec5SDimitry Andric           VASize),
4430b57cec5SDimitry Andric       u32(NumberOfSymbols),
4440b57cec5SDimitry Andric       u16(0),
4457a6dacacSDimitry Andric       u16(is64Bit() ? C_Invalid : IMAGE_FILE_32BIT_MACHINE),
4460b57cec5SDimitry Andric   };
4470b57cec5SDimitry Andric   append(Buffer, Header);
4480b57cec5SDimitry Andric 
4490b57cec5SDimitry Andric   // Section Header Table
4500b57cec5SDimitry Andric   const coff_section SectionTable[NumberOfSections] = {
4510b57cec5SDimitry Andric       {{'.', 'i', 'd', 'a', 't', 'a', '$', '5'},
4520b57cec5SDimitry Andric        u32(0),
4530b57cec5SDimitry Andric        u32(0),
4540b57cec5SDimitry Andric        u32(VASize),
4550b57cec5SDimitry Andric        u32(sizeof(coff_file_header) + NumberOfSections * sizeof(coff_section)),
4560b57cec5SDimitry Andric        u32(0),
4570b57cec5SDimitry Andric        u32(0),
4580b57cec5SDimitry Andric        u16(0),
4590b57cec5SDimitry Andric        u16(0),
4607a6dacacSDimitry Andric        u32((is64Bit() ? IMAGE_SCN_ALIGN_8BYTES : IMAGE_SCN_ALIGN_4BYTES) |
4610b57cec5SDimitry Andric            IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ |
4620b57cec5SDimitry Andric            IMAGE_SCN_MEM_WRITE)},
4630b57cec5SDimitry Andric       {{'.', 'i', 'd', 'a', 't', 'a', '$', '4'},
4640b57cec5SDimitry Andric        u32(0),
4650b57cec5SDimitry Andric        u32(0),
4660b57cec5SDimitry Andric        u32(VASize),
4670b57cec5SDimitry Andric        u32(sizeof(coff_file_header) + NumberOfSections * sizeof(coff_section) +
4680b57cec5SDimitry Andric            VASize),
4690b57cec5SDimitry Andric        u32(0),
4700b57cec5SDimitry Andric        u32(0),
4710b57cec5SDimitry Andric        u16(0),
4720b57cec5SDimitry Andric        u16(0),
4737a6dacacSDimitry Andric        u32((is64Bit() ? IMAGE_SCN_ALIGN_8BYTES : IMAGE_SCN_ALIGN_4BYTES) |
4740b57cec5SDimitry Andric            IMAGE_SCN_CNT_INITIALIZED_DATA | IMAGE_SCN_MEM_READ |
4750b57cec5SDimitry Andric            IMAGE_SCN_MEM_WRITE)},
4760b57cec5SDimitry Andric   };
4770b57cec5SDimitry Andric   append(Buffer, SectionTable);
4780b57cec5SDimitry Andric 
4790b57cec5SDimitry Andric   // .idata$5, ILT
4800b57cec5SDimitry Andric   append(Buffer, u32(0));
4817a6dacacSDimitry Andric   if (is64Bit())
4820b57cec5SDimitry Andric     append(Buffer, u32(0));
4830b57cec5SDimitry Andric 
4840b57cec5SDimitry Andric   // .idata$4, IAT
4850b57cec5SDimitry Andric   append(Buffer, u32(0));
4867a6dacacSDimitry Andric   if (is64Bit())
4870b57cec5SDimitry Andric     append(Buffer, u32(0));
4880b57cec5SDimitry Andric 
4890b57cec5SDimitry Andric   // Symbol Table
4900b57cec5SDimitry Andric   coff_symbol16 SymbolTable[NumberOfSymbols] = {
4910b57cec5SDimitry Andric       {{{0, 0, 0, 0, 0, 0, 0, 0}},
4920b57cec5SDimitry Andric        u32(0),
4930b57cec5SDimitry Andric        u16(1),
4940b57cec5SDimitry Andric        u16(0),
4950b57cec5SDimitry Andric        IMAGE_SYM_CLASS_EXTERNAL,
4960b57cec5SDimitry Andric        0},
4970b57cec5SDimitry Andric   };
4980b57cec5SDimitry Andric   SymbolTable[0].Name.Offset.Offset = sizeof(uint32_t);
4990b57cec5SDimitry Andric   append(Buffer, SymbolTable);
5000b57cec5SDimitry Andric 
5010b57cec5SDimitry Andric   // String Table
5020b57cec5SDimitry Andric   writeStringTable(Buffer, {NullThunkSymbolName});
5030b57cec5SDimitry Andric 
5040b57cec5SDimitry Andric   StringRef F{reinterpret_cast<const char *>(Buffer.data()), Buffer.size()};
5050b57cec5SDimitry Andric   return {MemoryBufferRef{F, ImportName}};
5060b57cec5SDimitry Andric }
5070b57cec5SDimitry Andric 
508439352acSDimitry Andric NewArchiveMember
createShortImport(StringRef Sym,uint16_t Ordinal,ImportType ImportType,ImportNameType NameType,StringRef ExportName,MachineTypes Machine)509439352acSDimitry Andric ObjectFactory::createShortImport(StringRef Sym, uint16_t Ordinal,
510439352acSDimitry Andric                                  ImportType ImportType, ImportNameType NameType,
511439352acSDimitry Andric                                  StringRef ExportName, MachineTypes Machine) {
5120b57cec5SDimitry Andric   size_t ImpSize = ImportName.size() + Sym.size() + 2; // +2 for NULs
513439352acSDimitry Andric   if (!ExportName.empty())
514439352acSDimitry Andric     ImpSize += ExportName.size() + 1;
5150b57cec5SDimitry Andric   size_t Size = sizeof(coff_import_header) + ImpSize;
5160b57cec5SDimitry Andric   char *Buf = Alloc.Allocate<char>(Size);
5170b57cec5SDimitry Andric   memset(Buf, 0, Size);
5180b57cec5SDimitry Andric   char *P = Buf;
5190b57cec5SDimitry Andric 
5200b57cec5SDimitry Andric   // Write short import library.
5210b57cec5SDimitry Andric   auto *Imp = reinterpret_cast<coff_import_header *>(P);
5220b57cec5SDimitry Andric   P += sizeof(*Imp);
5230b57cec5SDimitry Andric   Imp->Sig2 = 0xFFFF;
5240b57cec5SDimitry Andric   Imp->Machine = Machine;
5250b57cec5SDimitry Andric   Imp->SizeOfData = ImpSize;
5260b57cec5SDimitry Andric   if (Ordinal > 0)
5270b57cec5SDimitry Andric     Imp->OrdinalHint = Ordinal;
5280b57cec5SDimitry Andric   Imp->TypeInfo = (NameType << 2) | ImportType;
5290b57cec5SDimitry Andric 
5300b57cec5SDimitry Andric   // Write symbol name and DLL name.
5310b57cec5SDimitry Andric   memcpy(P, Sym.data(), Sym.size());
5320b57cec5SDimitry Andric   P += Sym.size() + 1;
5330b57cec5SDimitry Andric   memcpy(P, ImportName.data(), ImportName.size());
534439352acSDimitry Andric   if (!ExportName.empty()) {
535439352acSDimitry Andric     P += ImportName.size() + 1;
536439352acSDimitry Andric     memcpy(P, ExportName.data(), ExportName.size());
537439352acSDimitry Andric   }
5380b57cec5SDimitry Andric 
5390b57cec5SDimitry Andric   return {MemoryBufferRef(StringRef(Buf, Size), ImportName)};
5400b57cec5SDimitry Andric }
5410b57cec5SDimitry Andric 
createWeakExternal(StringRef Sym,StringRef Weak,bool Imp,MachineTypes Machine)5420b57cec5SDimitry Andric NewArchiveMember ObjectFactory::createWeakExternal(StringRef Sym,
5437a6dacacSDimitry Andric                                                    StringRef Weak, bool Imp,
5447a6dacacSDimitry Andric                                                    MachineTypes Machine) {
5450b57cec5SDimitry Andric   std::vector<uint8_t> Buffer;
5460b57cec5SDimitry Andric   const uint32_t NumberOfSections = 1;
5470b57cec5SDimitry Andric   const uint32_t NumberOfSymbols = 5;
5480b57cec5SDimitry Andric 
5490b57cec5SDimitry Andric   // COFF Header
5500b57cec5SDimitry Andric   coff_file_header Header{
5510b57cec5SDimitry Andric       u16(Machine),
5520b57cec5SDimitry Andric       u16(NumberOfSections),
5530b57cec5SDimitry Andric       u32(0),
5540b57cec5SDimitry Andric       u32(sizeof(Header) + (NumberOfSections * sizeof(coff_section))),
5550b57cec5SDimitry Andric       u32(NumberOfSymbols),
5560b57cec5SDimitry Andric       u16(0),
5570b57cec5SDimitry Andric       u16(0),
5580b57cec5SDimitry Andric   };
5590b57cec5SDimitry Andric   append(Buffer, Header);
5600b57cec5SDimitry Andric 
5610b57cec5SDimitry Andric   // Section Header Table
5620b57cec5SDimitry Andric   const coff_section SectionTable[NumberOfSections] = {
5630b57cec5SDimitry Andric       {{'.', 'd', 'r', 'e', 'c', 't', 'v', 'e'},
5640b57cec5SDimitry Andric        u32(0),
5650b57cec5SDimitry Andric        u32(0),
5660b57cec5SDimitry Andric        u32(0),
5670b57cec5SDimitry Andric        u32(0),
5680b57cec5SDimitry Andric        u32(0),
5690b57cec5SDimitry Andric        u32(0),
5700b57cec5SDimitry Andric        u16(0),
5710b57cec5SDimitry Andric        u16(0),
5720b57cec5SDimitry Andric        u32(IMAGE_SCN_LNK_INFO | IMAGE_SCN_LNK_REMOVE)}};
5730b57cec5SDimitry Andric   append(Buffer, SectionTable);
5740b57cec5SDimitry Andric 
5750b57cec5SDimitry Andric   // Symbol Table
5760b57cec5SDimitry Andric   coff_symbol16 SymbolTable[NumberOfSymbols] = {
5770b57cec5SDimitry Andric       {{{'@', 'c', 'o', 'm', 'p', '.', 'i', 'd'}},
5780b57cec5SDimitry Andric        u32(0),
5790b57cec5SDimitry Andric        u16(0xFFFF),
5800b57cec5SDimitry Andric        u16(0),
5810b57cec5SDimitry Andric        IMAGE_SYM_CLASS_STATIC,
5820b57cec5SDimitry Andric        0},
5830b57cec5SDimitry Andric       {{{'@', 'f', 'e', 'a', 't', '.', '0', '0'}},
5840b57cec5SDimitry Andric        u32(0),
5850b57cec5SDimitry Andric        u16(0xFFFF),
5860b57cec5SDimitry Andric        u16(0),
5870b57cec5SDimitry Andric        IMAGE_SYM_CLASS_STATIC,
5880b57cec5SDimitry Andric        0},
5890b57cec5SDimitry Andric       {{{0, 0, 0, 0, 0, 0, 0, 0}},
5900b57cec5SDimitry Andric        u32(0),
5910b57cec5SDimitry Andric        u16(0),
5920b57cec5SDimitry Andric        u16(0),
5930b57cec5SDimitry Andric        IMAGE_SYM_CLASS_EXTERNAL,
5940b57cec5SDimitry Andric        0},
5950b57cec5SDimitry Andric       {{{0, 0, 0, 0, 0, 0, 0, 0}},
5960b57cec5SDimitry Andric        u32(0),
5970b57cec5SDimitry Andric        u16(0),
5980b57cec5SDimitry Andric        u16(0),
5990b57cec5SDimitry Andric        IMAGE_SYM_CLASS_WEAK_EXTERNAL,
6000b57cec5SDimitry Andric        1},
6010b57cec5SDimitry Andric       {{{2, 0, 0, 0, IMAGE_WEAK_EXTERN_SEARCH_ALIAS, 0, 0, 0}},
6020b57cec5SDimitry Andric        u32(0),
6030b57cec5SDimitry Andric        u16(0),
6040b57cec5SDimitry Andric        u16(0),
6050b57cec5SDimitry Andric        IMAGE_SYM_CLASS_NULL,
6060b57cec5SDimitry Andric        0},
6070b57cec5SDimitry Andric   };
6080b57cec5SDimitry Andric   SymbolTable[2].Name.Offset.Offset = sizeof(uint32_t);
6090b57cec5SDimitry Andric 
6100b57cec5SDimitry Andric   //__imp_ String Table
6110b57cec5SDimitry Andric   StringRef Prefix = Imp ? "__imp_" : "";
6120b57cec5SDimitry Andric   SymbolTable[3].Name.Offset.Offset =
6130b57cec5SDimitry Andric       sizeof(uint32_t) + Sym.size() + Prefix.size() + 1;
6140b57cec5SDimitry Andric   append(Buffer, SymbolTable);
6150b57cec5SDimitry Andric   writeStringTable(Buffer, {(Prefix + Sym).str(),
6160b57cec5SDimitry Andric                             (Prefix + Weak).str()});
6170b57cec5SDimitry Andric 
6180b57cec5SDimitry Andric   // Copied here so we can still use writeStringTable
6190b57cec5SDimitry Andric   char *Buf = Alloc.Allocate<char>(Buffer.size());
6200b57cec5SDimitry Andric   memcpy(Buf, Buffer.data(), Buffer.size());
6210b57cec5SDimitry Andric   return {MemoryBufferRef(StringRef(Buf, Buffer.size()), ImportName)};
6220b57cec5SDimitry Andric }
6230b57cec5SDimitry Andric 
writeImportLibrary(StringRef ImportName,StringRef Path,ArrayRef<COFFShortExport> Exports,MachineTypes Machine,bool MinGW)6240b57cec5SDimitry Andric Error writeImportLibrary(StringRef ImportName, StringRef Path,
6250b57cec5SDimitry Andric                          ArrayRef<COFFShortExport> Exports,
6260b57cec5SDimitry Andric                          MachineTypes Machine, bool MinGW) {
6270b57cec5SDimitry Andric 
6287a6dacacSDimitry Andric   MachineTypes NativeMachine =
6297a6dacacSDimitry Andric       isArm64EC(Machine) ? IMAGE_FILE_MACHINE_ARM64 : Machine;
6307a6dacacSDimitry Andric 
6310b57cec5SDimitry Andric   std::vector<NewArchiveMember> Members;
6327a6dacacSDimitry Andric   ObjectFactory OF(llvm::sys::path::filename(ImportName), NativeMachine);
6330b57cec5SDimitry Andric 
6340b57cec5SDimitry Andric   std::vector<uint8_t> ImportDescriptor;
6350b57cec5SDimitry Andric   Members.push_back(OF.createImportDescriptor(ImportDescriptor));
6360b57cec5SDimitry Andric 
6370b57cec5SDimitry Andric   std::vector<uint8_t> NullImportDescriptor;
6380b57cec5SDimitry Andric   Members.push_back(OF.createNullImportDescriptor(NullImportDescriptor));
6390b57cec5SDimitry Andric 
6400b57cec5SDimitry Andric   std::vector<uint8_t> NullThunk;
6410b57cec5SDimitry Andric   Members.push_back(OF.createNullThunk(NullThunk));
6420b57cec5SDimitry Andric 
6435f757f3fSDimitry Andric   for (const COFFShortExport &E : Exports) {
6440b57cec5SDimitry Andric     if (E.Private)
6450b57cec5SDimitry Andric       continue;
6460b57cec5SDimitry Andric 
6470b57cec5SDimitry Andric     ImportType ImportType = IMPORT_CODE;
6480b57cec5SDimitry Andric     if (E.Data)
6490b57cec5SDimitry Andric       ImportType = IMPORT_DATA;
6500b57cec5SDimitry Andric     if (E.Constant)
6510b57cec5SDimitry Andric       ImportType = IMPORT_CONST;
6520b57cec5SDimitry Andric 
6530b57cec5SDimitry Andric     StringRef SymbolName = E.SymbolName.empty() ? E.Name : E.SymbolName;
654439352acSDimitry Andric     std::string Name;
6550b57cec5SDimitry Andric 
656439352acSDimitry Andric     if (E.ExtName.empty()) {
657439352acSDimitry Andric       Name = std::string(SymbolName);
658439352acSDimitry Andric     } else {
659439352acSDimitry Andric       Expected<std::string> ReplacedName =
660439352acSDimitry Andric           replace(SymbolName, E.Name, E.ExtName);
661439352acSDimitry Andric       if (!ReplacedName)
662439352acSDimitry Andric         return ReplacedName.takeError();
663439352acSDimitry Andric       Name.swap(*ReplacedName);
664439352acSDimitry Andric     }
6650b57cec5SDimitry Andric 
666439352acSDimitry Andric     if (!E.AliasTarget.empty() && Name != E.AliasTarget) {
6677a6dacacSDimitry Andric       Members.push_back(
668439352acSDimitry Andric           OF.createWeakExternal(E.AliasTarget, Name, false, Machine));
6697a6dacacSDimitry Andric       Members.push_back(
670439352acSDimitry Andric           OF.createWeakExternal(E.AliasTarget, Name, true, Machine));
6710b57cec5SDimitry Andric       continue;
6720b57cec5SDimitry Andric     }
6730b57cec5SDimitry Andric 
674439352acSDimitry Andric     ImportNameType NameType;
675439352acSDimitry Andric     std::string ExportName;
676439352acSDimitry Andric     if (E.Noname) {
677439352acSDimitry Andric       NameType = IMPORT_ORDINAL;
678439352acSDimitry Andric     } else {
679439352acSDimitry Andric       NameType = getNameType(SymbolName, E.Name, Machine, MinGW);
680439352acSDimitry Andric     }
681439352acSDimitry Andric 
682439352acSDimitry Andric     // On ARM64EC, use EXPORTAS to import demangled name for mangled symbols.
683439352acSDimitry Andric     if (ImportType == IMPORT_CODE && isArm64EC(Machine)) {
684439352acSDimitry Andric       if (std::optional<std::string> MangledName =
685439352acSDimitry Andric               getArm64ECMangledFunctionName(Name)) {
686439352acSDimitry Andric         if (ExportName.empty()) {
687439352acSDimitry Andric           NameType = IMPORT_NAME_EXPORTAS;
688439352acSDimitry Andric           ExportName.swap(Name);
689439352acSDimitry Andric         }
690439352acSDimitry Andric         Name = std::move(*MangledName);
691439352acSDimitry Andric       } else if (ExportName.empty()) {
692439352acSDimitry Andric         NameType = IMPORT_NAME_EXPORTAS;
693439352acSDimitry Andric         ExportName = std::move(*getArm64ECDemangledFunctionName(Name));
694439352acSDimitry Andric       }
695439352acSDimitry Andric     }
696439352acSDimitry Andric 
697439352acSDimitry Andric     Members.push_back(OF.createShortImport(Name, E.Ordinal, ImportType,
698439352acSDimitry Andric                                            NameType, ExportName, Machine));
6990b57cec5SDimitry Andric   }
7000b57cec5SDimitry Andric 
7015f757f3fSDimitry Andric   return writeArchive(Path, Members, SymtabWritingMode::NormalSymtab,
7025f757f3fSDimitry Andric                       MinGW ? object::Archive::K_GNU : object::Archive::K_COFF,
7035f757f3fSDimitry Andric                       /*Deterministic*/ true, /*Thin*/ false,
7045f757f3fSDimitry Andric                       /*OldArchiveBuf*/ nullptr, isArm64EC(Machine));
7050b57cec5SDimitry Andric }
7060b57cec5SDimitry Andric 
7070b57cec5SDimitry Andric } // namespace object
7080b57cec5SDimitry Andric } // namespace llvm
709