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