| Rui Ueyama | 3500f66 | 2015-05-28 20:30:06 +0000 | [diff] [blame] | 1 | //===- DriverUtils.cpp ----------------------------------------------------===// | 
|  | 2 | // | 
|  | 3 | //                             The LLVM Linker | 
|  | 4 | // | 
|  | 5 | // This file is distributed under the University of Illinois Open Source | 
|  | 6 | // License. See LICENSE.TXT for details. | 
|  | 7 | // | 
|  | 8 | //===----------------------------------------------------------------------===// | 
|  | 9 | // | 
|  | 10 | // This file contains utility functions for the driver. Because there | 
|  | 11 | // are so many small functions, we created this separate file to make | 
|  | 12 | // Driver.cpp less cluttered. | 
|  | 13 | // | 
|  | 14 | //===----------------------------------------------------------------------===// | 
|  | 15 |  | 
| Rui Ueyama | 8854d8a | 2015-06-04 19:21:24 +0000 | [diff] [blame] | 16 | #include "Config.h" | 
| Rui Ueyama | 3500f66 | 2015-05-28 20:30:06 +0000 | [diff] [blame] | 17 | #include "Driver.h" | 
| Rui Ueyama | 8fd9fb9 | 2015-06-01 02:58:15 +0000 | [diff] [blame] | 18 | #include "Error.h" | 
| Rui Ueyama | 9381eb1 | 2016-12-18 14:06:06 +0000 | [diff] [blame] | 19 | #include "Memory.h" | 
| Rui Ueyama | 8765fba | 2015-07-15 22:21:08 +0000 | [diff] [blame] | 20 | #include "Symbols.h" | 
| Rui Ueyama | 3500f66 | 2015-05-28 20:30:06 +0000 | [diff] [blame] | 21 | #include "llvm/ADT/Optional.h" | 
| Rui Ueyama | 3500f66 | 2015-05-28 20:30:06 +0000 | [diff] [blame] | 22 | #include "llvm/ADT/StringSwitch.h" | 
|  | 23 | #include "llvm/Object/COFF.h" | 
|  | 24 | #include "llvm/Option/Arg.h" | 
|  | 25 | #include "llvm/Option/ArgList.h" | 
|  | 26 | #include "llvm/Option/Option.h" | 
|  | 27 | #include "llvm/Support/CommandLine.h" | 
| Rui Ueyama | 151d862 | 2015-06-17 20:40:43 +0000 | [diff] [blame] | 28 | #include "llvm/Support/FileUtilities.h" | 
| Rui Ueyama | 3500f66 | 2015-05-28 20:30:06 +0000 | [diff] [blame] | 29 | #include "llvm/Support/Process.h" | 
| Rui Ueyama | 2bf6a12 | 2015-06-14 21:50:50 +0000 | [diff] [blame] | 30 | #include "llvm/Support/Program.h" | 
| Rui Ueyama | 3500f66 | 2015-05-28 20:30:06 +0000 | [diff] [blame] | 31 | #include "llvm/Support/raw_ostream.h" | 
|  | 32 | #include <memory> | 
|  | 33 |  | 
|  | 34 | using namespace llvm::COFF; | 
|  | 35 | using namespace llvm; | 
| Rui Ueyama | b51f67a | 2015-06-07 23:00:29 +0000 | [diff] [blame] | 36 | using llvm::cl::ExpandResponseFiles; | 
|  | 37 | using llvm::cl::TokenizeWindowsCommandLine; | 
| Rui Ueyama | 3500f66 | 2015-05-28 20:30:06 +0000 | [diff] [blame] | 38 | using llvm::sys::Process; | 
| Rui Ueyama | 3500f66 | 2015-05-28 20:30:06 +0000 | [diff] [blame] | 39 |  | 
|  | 40 | namespace lld { | 
|  | 41 | namespace coff { | 
| Rui Ueyama | a9c8838 | 2015-06-17 21:01:56 +0000 | [diff] [blame] | 42 | namespace { | 
|  | 43 |  | 
|  | 44 | class Executor { | 
|  | 45 | public: | 
|  | 46 | explicit Executor(StringRef S) : Saver(Alloc), Prog(Saver.save(S)) {} | 
| Rui Ueyama | e6e206d | 2017-02-21 23:22:56 +0000 | [diff] [blame] | 47 | void add(StringRef S) { Args.push_back(Saver.save(S)); } | 
|  | 48 | void add(std::string &S) { Args.push_back(Saver.save(S)); } | 
|  | 49 | void add(Twine S) { Args.push_back(Saver.save(S)); } | 
|  | 50 | void add(const char *S) { Args.push_back(Saver.save(S)); } | 
| Rui Ueyama | a9c8838 | 2015-06-17 21:01:56 +0000 | [diff] [blame] | 51 |  | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 52 | void run() { | 
| Rui Ueyama | 8fe1767 | 2016-12-08 20:50:47 +0000 | [diff] [blame] | 53 | ErrorOr<std::string> ExeOrErr = sys::findProgramByName(Prog); | 
| Rui Ueyama | 0d09a86 | 2016-07-15 00:40:46 +0000 | [diff] [blame] | 54 | if (auto EC = ExeOrErr.getError()) | 
|  | 55 | fatal(EC, "unable to find " + Prog + " in PATH: "); | 
| Rui Ueyama | e6e206d | 2017-02-21 23:22:56 +0000 | [diff] [blame] | 56 | StringRef Exe = Saver.save(*ExeOrErr); | 
| Rui Ueyama | a9c8838 | 2015-06-17 21:01:56 +0000 | [diff] [blame] | 57 | Args.insert(Args.begin(), Exe); | 
| Rui Ueyama | e6e206d | 2017-02-21 23:22:56 +0000 | [diff] [blame] | 58 |  | 
|  | 59 | std::vector<const char *> Vec; | 
|  | 60 | for (StringRef S : Args) | 
|  | 61 | Vec.push_back(S.data()); | 
|  | 62 | Vec.push_back(nullptr); | 
|  | 63 |  | 
|  | 64 | if (sys::ExecuteAndWait(Args[0], Vec.data()) != 0) | 
|  | 65 | fatal("ExecuteAndWait failed: " + | 
|  | 66 | llvm::join(Args.begin(), Args.end(), " ")); | 
| Rui Ueyama | a9c8838 | 2015-06-17 21:01:56 +0000 | [diff] [blame] | 67 | } | 
|  | 68 |  | 
|  | 69 | private: | 
| Rui Ueyama | 8fe1767 | 2016-12-08 20:50:47 +0000 | [diff] [blame] | 70 | BumpPtrAllocator Alloc; | 
|  | 71 | StringSaver Saver; | 
| Rui Ueyama | a9c8838 | 2015-06-17 21:01:56 +0000 | [diff] [blame] | 72 | StringRef Prog; | 
| Rui Ueyama | e6e206d | 2017-02-21 23:22:56 +0000 | [diff] [blame] | 73 | std::vector<StringRef> Args; | 
| Rui Ueyama | a9c8838 | 2015-06-17 21:01:56 +0000 | [diff] [blame] | 74 | }; | 
|  | 75 |  | 
|  | 76 | } // anonymous namespace | 
| Rui Ueyama | 3500f66 | 2015-05-28 20:30:06 +0000 | [diff] [blame] | 77 |  | 
| Rui Ueyama | 3d3e6fb | 2015-05-29 16:06:00 +0000 | [diff] [blame] | 78 | // Returns /machine's value. | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 79 | MachineTypes getMachineType(StringRef S) { | 
| Rui Ueyama | e16a75d5 | 2015-07-08 18:14:51 +0000 | [diff] [blame] | 80 | MachineTypes MT = StringSwitch<MachineTypes>(S.lower()) | 
| Rui Ueyama | c79ecdd | 2016-09-30 22:01:25 +0000 | [diff] [blame] | 81 | .Cases("x64", "amd64", AMD64) | 
|  | 82 | .Cases("x86", "i386", I386) | 
| Rui Ueyama | 5e706b3 | 2015-07-25 21:54:50 +0000 | [diff] [blame] | 83 | .Case("arm", ARMNT) | 
| Rui Ueyama | e16a75d5 | 2015-07-08 18:14:51 +0000 | [diff] [blame] | 84 | .Default(IMAGE_FILE_MACHINE_UNKNOWN); | 
|  | 85 | if (MT != IMAGE_FILE_MACHINE_UNKNOWN) | 
| Rui Ueyama | 3d3e6fb | 2015-05-29 16:06:00 +0000 | [diff] [blame] | 86 | return MT; | 
| Rui Ueyama | 1a3fd13 | 2016-07-14 23:43:36 +0000 | [diff] [blame] | 87 | fatal("unknown /machine argument: " + S); | 
| Rui Ueyama | 3d3e6fb | 2015-05-29 16:06:00 +0000 | [diff] [blame] | 88 | } | 
|  | 89 |  | 
| Rui Ueyama | 5e706b3 | 2015-07-25 21:54:50 +0000 | [diff] [blame] | 90 | StringRef machineToStr(MachineTypes MT) { | 
| Rui Ueyama | 84936e0 | 2015-07-07 23:39:18 +0000 | [diff] [blame] | 91 | switch (MT) { | 
| Rui Ueyama | 5e706b3 | 2015-07-25 21:54:50 +0000 | [diff] [blame] | 92 | case ARMNT: | 
| Rui Ueyama | 84936e0 | 2015-07-07 23:39:18 +0000 | [diff] [blame] | 93 | return "arm"; | 
| Rui Ueyama | 5e706b3 | 2015-07-25 21:54:50 +0000 | [diff] [blame] | 94 | case AMD64: | 
| Rui Ueyama | 84936e0 | 2015-07-07 23:39:18 +0000 | [diff] [blame] | 95 | return "x64"; | 
| Rui Ueyama | 5e706b3 | 2015-07-25 21:54:50 +0000 | [diff] [blame] | 96 | case I386: | 
| Rui Ueyama | 84936e0 | 2015-07-07 23:39:18 +0000 | [diff] [blame] | 97 | return "x86"; | 
|  | 98 | default: | 
|  | 99 | llvm_unreachable("unknown machine type"); | 
|  | 100 | } | 
|  | 101 | } | 
|  | 102 |  | 
| Rui Ueyama | 804a8b6 | 2015-05-29 16:18:15 +0000 | [diff] [blame] | 103 | // Parses a string in the form of "<integer>[,<integer>]". | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 104 | void parseNumbers(StringRef Arg, uint64_t *Addr, uint64_t *Size) { | 
| Rui Ueyama | 804a8b6 | 2015-05-29 16:18:15 +0000 | [diff] [blame] | 105 | StringRef S1, S2; | 
|  | 106 | std::tie(S1, S2) = Arg.split(','); | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 107 | if (S1.getAsInteger(0, *Addr)) | 
| Rui Ueyama | 1a3fd13 | 2016-07-14 23:43:36 +0000 | [diff] [blame] | 108 | fatal("invalid number: " + S1); | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 109 | if (Size && !S2.empty() && S2.getAsInteger(0, *Size)) | 
| Rui Ueyama | 1a3fd13 | 2016-07-14 23:43:36 +0000 | [diff] [blame] | 110 | fatal("invalid number: " + S2); | 
| Rui Ueyama | 804a8b6 | 2015-05-29 16:18:15 +0000 | [diff] [blame] | 111 | } | 
|  | 112 |  | 
| Rui Ueyama | b9dcdb5 | 2015-05-29 16:28:29 +0000 | [diff] [blame] | 113 | // Parses a string in the form of "<integer>[.<integer>]". | 
|  | 114 | // If second number is not present, Minor is set to 0. | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 115 | void parseVersion(StringRef Arg, uint32_t *Major, uint32_t *Minor) { | 
| Rui Ueyama | b9dcdb5 | 2015-05-29 16:28:29 +0000 | [diff] [blame] | 116 | StringRef S1, S2; | 
|  | 117 | std::tie(S1, S2) = Arg.split('.'); | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 118 | if (S1.getAsInteger(0, *Major)) | 
| Rui Ueyama | 1a3fd13 | 2016-07-14 23:43:36 +0000 | [diff] [blame] | 119 | fatal("invalid number: " + S1); | 
| Rui Ueyama | b9dcdb5 | 2015-05-29 16:28:29 +0000 | [diff] [blame] | 120 | *Minor = 0; | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 121 | if (!S2.empty() && S2.getAsInteger(0, *Minor)) | 
| Rui Ueyama | 1a3fd13 | 2016-07-14 23:43:36 +0000 | [diff] [blame] | 122 | fatal("invalid number: " + S2); | 
| Rui Ueyama | b9dcdb5 | 2015-05-29 16:28:29 +0000 | [diff] [blame] | 123 | } | 
|  | 124 |  | 
| Rui Ueyama | 15cc47e | 2015-05-29 16:34:31 +0000 | [diff] [blame] | 125 | // Parses a string in the form of "<subsystem>[,<integer>[.<integer>]]". | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 126 | void parseSubsystem(StringRef Arg, WindowsSubsystem *Sys, uint32_t *Major, | 
|  | 127 | uint32_t *Minor) { | 
| Rui Ueyama | 15cc47e | 2015-05-29 16:34:31 +0000 | [diff] [blame] | 128 | StringRef SysStr, Ver; | 
|  | 129 | std::tie(SysStr, Ver) = Arg.split(','); | 
|  | 130 | *Sys = StringSwitch<WindowsSubsystem>(SysStr.lower()) | 
|  | 131 | .Case("boot_application", IMAGE_SUBSYSTEM_WINDOWS_BOOT_APPLICATION) | 
|  | 132 | .Case("console", IMAGE_SUBSYSTEM_WINDOWS_CUI) | 
|  | 133 | .Case("efi_application", IMAGE_SUBSYSTEM_EFI_APPLICATION) | 
|  | 134 | .Case("efi_boot_service_driver", IMAGE_SUBSYSTEM_EFI_BOOT_SERVICE_DRIVER) | 
|  | 135 | .Case("efi_rom", IMAGE_SUBSYSTEM_EFI_ROM) | 
|  | 136 | .Case("efi_runtime_driver", IMAGE_SUBSYSTEM_EFI_RUNTIME_DRIVER) | 
|  | 137 | .Case("native", IMAGE_SUBSYSTEM_NATIVE) | 
|  | 138 | .Case("posix", IMAGE_SUBSYSTEM_POSIX_CUI) | 
|  | 139 | .Case("windows", IMAGE_SUBSYSTEM_WINDOWS_GUI) | 
|  | 140 | .Default(IMAGE_SUBSYSTEM_UNKNOWN); | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 141 | if (*Sys == IMAGE_SUBSYSTEM_UNKNOWN) | 
| Rui Ueyama | 1a3fd13 | 2016-07-14 23:43:36 +0000 | [diff] [blame] | 142 | fatal("unknown subsystem: " + SysStr); | 
| Rui Ueyama | 15cc47e | 2015-05-29 16:34:31 +0000 | [diff] [blame] | 143 | if (!Ver.empty()) | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 144 | parseVersion(Ver, Major, Minor); | 
| Rui Ueyama | 15cc47e | 2015-05-29 16:34:31 +0000 | [diff] [blame] | 145 | } | 
|  | 146 |  | 
| Rui Ueyama | 2edb35a | 2015-06-18 19:09:30 +0000 | [diff] [blame] | 147 | // Parse a string of the form of "<from>=<to>". | 
|  | 148 | // Results are directly written to Config. | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 149 | void parseAlternateName(StringRef S) { | 
| Rui Ueyama | 2edb35a | 2015-06-18 19:09:30 +0000 | [diff] [blame] | 150 | StringRef From, To; | 
|  | 151 | std::tie(From, To) = S.split('='); | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 152 | if (From.empty() || To.empty()) | 
| Rui Ueyama | 1a3fd13 | 2016-07-14 23:43:36 +0000 | [diff] [blame] | 153 | fatal("/alternatename: invalid argument: " + S); | 
| Rui Ueyama | e8d56b5 | 2015-06-18 23:04:26 +0000 | [diff] [blame] | 154 | auto It = Config->AlternateNames.find(From); | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 155 | if (It != Config->AlternateNames.end() && It->second != To) | 
| Rui Ueyama | 1a3fd13 | 2016-07-14 23:43:36 +0000 | [diff] [blame] | 156 | fatal("/alternatename: conflicts: " + S); | 
| Rui Ueyama | e8d56b5 | 2015-06-18 23:04:26 +0000 | [diff] [blame] | 157 | Config->AlternateNames.insert(It, std::make_pair(From, To)); | 
| Rui Ueyama | 2edb35a | 2015-06-18 19:09:30 +0000 | [diff] [blame] | 158 | } | 
|  | 159 |  | 
| Rui Ueyama | 6600eb1 | 2015-07-04 23:37:32 +0000 | [diff] [blame] | 160 | // Parse a string of the form of "<from>=<to>". | 
|  | 161 | // Results are directly written to Config. | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 162 | void parseMerge(StringRef S) { | 
| Rui Ueyama | 6600eb1 | 2015-07-04 23:37:32 +0000 | [diff] [blame] | 163 | StringRef From, To; | 
|  | 164 | std::tie(From, To) = S.split('='); | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 165 | if (From.empty() || To.empty()) | 
| Rui Ueyama | 1a3fd13 | 2016-07-14 23:43:36 +0000 | [diff] [blame] | 166 | fatal("/merge: invalid argument: " + S); | 
| Rui Ueyama | 6600eb1 | 2015-07-04 23:37:32 +0000 | [diff] [blame] | 167 | auto Pair = Config->Merge.insert(std::make_pair(From, To)); | 
|  | 168 | bool Inserted = Pair.second; | 
| Rui Ueyama | 2b82d5f | 2015-07-04 23:54:52 +0000 | [diff] [blame] | 169 | if (!Inserted) { | 
|  | 170 | StringRef Existing = Pair.first->second; | 
|  | 171 | if (Existing != To) | 
| Rui Ueyama | e6e206d | 2017-02-21 23:22:56 +0000 | [diff] [blame] | 172 | warn(S + ": already merged into " + Existing); | 
| Rui Ueyama | 2b82d5f | 2015-07-04 23:54:52 +0000 | [diff] [blame] | 173 | } | 
| Rui Ueyama | 6600eb1 | 2015-07-04 23:37:32 +0000 | [diff] [blame] | 174 | } | 
|  | 175 |  | 
| Rui Ueyama | 440138c | 2016-06-20 03:39:39 +0000 | [diff] [blame] | 176 | static uint32_t parseSectionAttributes(StringRef S) { | 
|  | 177 | uint32_t Ret = 0; | 
|  | 178 | for (char C : S.lower()) { | 
|  | 179 | switch (C) { | 
|  | 180 | case 'd': | 
|  | 181 | Ret |= IMAGE_SCN_MEM_DISCARDABLE; | 
|  | 182 | break; | 
|  | 183 | case 'e': | 
|  | 184 | Ret |= IMAGE_SCN_MEM_EXECUTE; | 
|  | 185 | break; | 
|  | 186 | case 'k': | 
|  | 187 | Ret |= IMAGE_SCN_MEM_NOT_CACHED; | 
|  | 188 | break; | 
|  | 189 | case 'p': | 
|  | 190 | Ret |= IMAGE_SCN_MEM_NOT_PAGED; | 
|  | 191 | break; | 
|  | 192 | case 'r': | 
|  | 193 | Ret |= IMAGE_SCN_MEM_READ; | 
|  | 194 | break; | 
|  | 195 | case 's': | 
|  | 196 | Ret |= IMAGE_SCN_MEM_SHARED; | 
|  | 197 | break; | 
|  | 198 | case 'w': | 
|  | 199 | Ret |= IMAGE_SCN_MEM_WRITE; | 
|  | 200 | break; | 
|  | 201 | default: | 
| Rui Ueyama | 1a3fd13 | 2016-07-14 23:43:36 +0000 | [diff] [blame] | 202 | fatal("/section: invalid argument: " + S); | 
| Rui Ueyama | 440138c | 2016-06-20 03:39:39 +0000 | [diff] [blame] | 203 | } | 
|  | 204 | } | 
|  | 205 | return Ret; | 
|  | 206 | } | 
|  | 207 |  | 
|  | 208 | // Parses /section option argument. | 
|  | 209 | void parseSection(StringRef S) { | 
|  | 210 | StringRef Name, Attrs; | 
|  | 211 | std::tie(Name, Attrs) = S.split(','); | 
|  | 212 | if (Name.empty() || Attrs.empty()) | 
| Rui Ueyama | 1a3fd13 | 2016-07-14 23:43:36 +0000 | [diff] [blame] | 213 | fatal("/section: invalid argument: " + S); | 
| Rui Ueyama | 440138c | 2016-06-20 03:39:39 +0000 | [diff] [blame] | 214 | Config->Section[Name] = parseSectionAttributes(Attrs); | 
|  | 215 | } | 
|  | 216 |  | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 217 | // Parses a string in the form of "EMBED[,=<integer>]|NO". | 
|  | 218 | // Results are directly written to Config. | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 219 | void parseManifest(StringRef Arg) { | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 220 | if (Arg.equals_lower("no")) { | 
|  | 221 | Config->Manifest = Configuration::No; | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 222 | return; | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 223 | } | 
|  | 224 | if (!Arg.startswith_lower("embed")) | 
| Rui Ueyama | bb57954 | 2016-07-15 01:12:24 +0000 | [diff] [blame] | 225 | fatal("invalid option " + Arg); | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 226 | Config->Manifest = Configuration::Embed; | 
|  | 227 | Arg = Arg.substr(strlen("embed")); | 
|  | 228 | if (Arg.empty()) | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 229 | return; | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 230 | if (!Arg.startswith_lower(",id=")) | 
| Rui Ueyama | bb57954 | 2016-07-15 01:12:24 +0000 | [diff] [blame] | 231 | fatal("invalid option " + Arg); | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 232 | Arg = Arg.substr(strlen(",id=")); | 
|  | 233 | if (Arg.getAsInteger(0, Config->ManifestID)) | 
| Rui Ueyama | bb57954 | 2016-07-15 01:12:24 +0000 | [diff] [blame] | 234 | fatal("invalid option " + Arg); | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 235 | } | 
|  | 236 |  | 
|  | 237 | // Parses a string in the form of "level=<string>|uiAccess=<string>|NO". | 
|  | 238 | // Results are directly written to Config. | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 239 | void parseManifestUAC(StringRef Arg) { | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 240 | if (Arg.equals_lower("no")) { | 
|  | 241 | Config->ManifestUAC = false; | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 242 | return; | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 243 | } | 
|  | 244 | for (;;) { | 
|  | 245 | Arg = Arg.ltrim(); | 
|  | 246 | if (Arg.empty()) | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 247 | return; | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 248 | if (Arg.startswith_lower("level=")) { | 
|  | 249 | Arg = Arg.substr(strlen("level=")); | 
|  | 250 | std::tie(Config->ManifestLevel, Arg) = Arg.split(" "); | 
|  | 251 | continue; | 
|  | 252 | } | 
|  | 253 | if (Arg.startswith_lower("uiaccess=")) { | 
|  | 254 | Arg = Arg.substr(strlen("uiaccess=")); | 
|  | 255 | std::tie(Config->ManifestUIAccess, Arg) = Arg.split(" "); | 
|  | 256 | continue; | 
|  | 257 | } | 
| Rui Ueyama | bb57954 | 2016-07-15 01:12:24 +0000 | [diff] [blame] | 258 | fatal("invalid option " + Arg); | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 259 | } | 
|  | 260 | } | 
|  | 261 |  | 
|  | 262 | // Quote each line with "". Existing double-quote is converted | 
|  | 263 | // to two double-quotes. | 
|  | 264 | static void quoteAndPrint(raw_ostream &Out, StringRef S) { | 
|  | 265 | while (!S.empty()) { | 
|  | 266 | StringRef Line; | 
|  | 267 | std::tie(Line, S) = S.split("\n"); | 
|  | 268 | if (Line.empty()) | 
|  | 269 | continue; | 
|  | 270 | Out << '\"'; | 
|  | 271 | for (int I = 0, E = Line.size(); I != E; ++I) { | 
|  | 272 | if (Line[I] == '\"') { | 
|  | 273 | Out << "\"\""; | 
|  | 274 | } else { | 
|  | 275 | Out << Line[I]; | 
|  | 276 | } | 
|  | 277 | } | 
|  | 278 | Out << "\"\n"; | 
|  | 279 | } | 
|  | 280 | } | 
|  | 281 |  | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 282 | // An RAII temporary file class that automatically removes a temporary file. | 
|  | 283 | namespace { | 
|  | 284 | class TemporaryFile { | 
|  | 285 | public: | 
| Rui Ueyama | 1e0b158 | 2017-02-06 20:47:55 +0000 | [diff] [blame] | 286 | TemporaryFile(StringRef Prefix, StringRef Extn, StringRef Contents = "") { | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 287 | SmallString<128> S; | 
|  | 288 | if (auto EC = sys::fs::createTemporaryFile("lld-" + Prefix, Extn, S)) | 
|  | 289 | fatal(EC, "cannot create a temporary file"); | 
|  | 290 | Path = S.str(); | 
| Rui Ueyama | 1e0b158 | 2017-02-06 20:47:55 +0000 | [diff] [blame] | 291 |  | 
|  | 292 | if (!Contents.empty()) { | 
|  | 293 | std::error_code EC; | 
|  | 294 | raw_fd_ostream OS(Path, EC, sys::fs::F_None); | 
|  | 295 | if (EC) | 
|  | 296 | fatal(EC, "failed to open " + Path); | 
|  | 297 | OS << Contents; | 
|  | 298 | } | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 299 | } | 
|  | 300 |  | 
|  | 301 | TemporaryFile(TemporaryFile &&Obj) { | 
|  | 302 | std::swap(Path, Obj.Path); | 
|  | 303 | } | 
|  | 304 |  | 
|  | 305 | ~TemporaryFile() { | 
|  | 306 | if (Path.empty()) | 
|  | 307 | return; | 
|  | 308 | if (sys::fs::remove(Path)) | 
|  | 309 | fatal("failed to remove " + Path); | 
|  | 310 | } | 
|  | 311 |  | 
|  | 312 | // Returns a memory buffer of this temporary file. | 
|  | 313 | // Note that this function does not leave the file open, | 
|  | 314 | // so it is safe to remove the file immediately after this function | 
|  | 315 | // is called (you cannot remove an opened file on Windows.) | 
|  | 316 | std::unique_ptr<MemoryBuffer> getMemoryBuffer() { | 
|  | 317 | // IsVolatileSize=true forces MemoryBuffer to not use mmap(). | 
|  | 318 | return check(MemoryBuffer::getFile(Path, /*FileSize=*/-1, | 
|  | 319 | /*RequiresNullTerminator=*/false, | 
|  | 320 | /*IsVolatileSize=*/true), | 
|  | 321 | "could not open " + Path); | 
|  | 322 | } | 
|  | 323 |  | 
|  | 324 | std::string Path; | 
|  | 325 | }; | 
|  | 326 | } | 
|  | 327 |  | 
| Rui Ueyama | afb1901 | 2016-04-19 01:21:58 +0000 | [diff] [blame] | 328 | // Create the default manifest file as a temporary file. | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 329 | TemporaryFile createDefaultXml() { | 
| Rui Ueyama | afb1901 | 2016-04-19 01:21:58 +0000 | [diff] [blame] | 330 | // Create a temporary file. | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 331 | TemporaryFile File("defaultxml", "manifest"); | 
| Rui Ueyama | afb1901 | 2016-04-19 01:21:58 +0000 | [diff] [blame] | 332 |  | 
|  | 333 | // Open the temporary file for writing. | 
| Rui Ueyama | 0d09a86 | 2016-07-15 00:40:46 +0000 | [diff] [blame] | 334 | std::error_code EC; | 
| Rui Ueyama | 8fe1767 | 2016-12-08 20:50:47 +0000 | [diff] [blame] | 335 | raw_fd_ostream OS(File.Path, EC, sys::fs::F_Text); | 
| Rui Ueyama | 0d09a86 | 2016-07-15 00:40:46 +0000 | [diff] [blame] | 336 | if (EC) | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 337 | fatal(EC, "failed to open " + File.Path); | 
| Rui Ueyama | afb1901 | 2016-04-19 01:21:58 +0000 | [diff] [blame] | 338 |  | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 339 | // Emit the XML. Note that we do *not* verify that the XML attributes are | 
|  | 340 | // syntactically correct. This is intentional for link.exe compatibility. | 
|  | 341 | OS << "<?xml version=\"1.0\" standalone=\"yes\"?>\n" | 
|  | 342 | << "<assembly xmlns=\"urn:schemas-microsoft-com:asm.v1\"\n" | 
|  | 343 | << "          manifestVersion=\"1.0\">\n"; | 
|  | 344 | if (Config->ManifestUAC) { | 
|  | 345 | OS << "  <trustInfo>\n" | 
|  | 346 | << "    <security>\n" | 
|  | 347 | << "      <requestedPrivileges>\n" | 
|  | 348 | << "         <requestedExecutionLevel level=" << Config->ManifestLevel | 
|  | 349 | << " uiAccess=" << Config->ManifestUIAccess << "/>\n" | 
|  | 350 | << "      </requestedPrivileges>\n" | 
|  | 351 | << "    </security>\n" | 
|  | 352 | << "  </trustInfo>\n"; | 
|  | 353 | if (!Config->ManifestDependency.empty()) { | 
|  | 354 | OS << "  <dependency>\n" | 
|  | 355 | << "    <dependentAssembly>\n" | 
|  | 356 | << "      <assemblyIdentity " << Config->ManifestDependency << " />\n" | 
|  | 357 | << "    </dependentAssembly>\n" | 
|  | 358 | << "  </dependency>\n"; | 
|  | 359 | } | 
|  | 360 | } | 
|  | 361 | OS << "</assembly>\n"; | 
| Rui Ueyama | afb1901 | 2016-04-19 01:21:58 +0000 | [diff] [blame] | 362 | OS.close(); | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 363 | return File; | 
| Rui Ueyama | afb1901 | 2016-04-19 01:21:58 +0000 | [diff] [blame] | 364 | } | 
|  | 365 |  | 
|  | 366 | static std::string readFile(StringRef Path) { | 
| Rui Ueyama | 659a4f2 | 2016-07-15 01:06:38 +0000 | [diff] [blame] | 367 | std::unique_ptr<MemoryBuffer> MB = | 
| Rui Ueyama | bb57954 | 2016-07-15 01:12:24 +0000 | [diff] [blame] | 368 | check(MemoryBuffer::getFile(Path), "could not open " + Path); | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 369 | return MB->getBuffer(); | 
| Rui Ueyama | afb1901 | 2016-04-19 01:21:58 +0000 | [diff] [blame] | 370 | } | 
|  | 371 |  | 
|  | 372 | static std::string createManifestXml() { | 
|  | 373 | // Create the default manifest file. | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 374 | TemporaryFile File1 = createDefaultXml(); | 
| Rui Ueyama | afb1901 | 2016-04-19 01:21:58 +0000 | [diff] [blame] | 375 | if (Config->ManifestInput.empty()) | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 376 | return readFile(File1.Path); | 
| Rui Ueyama | afb1901 | 2016-04-19 01:21:58 +0000 | [diff] [blame] | 377 |  | 
|  | 378 | // If manifest files are supplied by the user using /MANIFESTINPUT | 
|  | 379 | // option, we need to merge them with the default manifest. | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 380 | TemporaryFile File2("user", "manifest"); | 
| Rui Ueyama | afb1901 | 2016-04-19 01:21:58 +0000 | [diff] [blame] | 381 |  | 
|  | 382 | Executor E("mt.exe"); | 
|  | 383 | E.add("/manifest"); | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 384 | E.add(File1.Path); | 
| Rui Ueyama | afb1901 | 2016-04-19 01:21:58 +0000 | [diff] [blame] | 385 | for (StringRef Filename : Config->ManifestInput) { | 
|  | 386 | E.add("/manifest"); | 
|  | 387 | E.add(Filename); | 
|  | 388 | } | 
|  | 389 | E.add("/nologo"); | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 390 | E.add("/out:" + StringRef(File2.Path)); | 
| Rui Ueyama | afb1901 | 2016-04-19 01:21:58 +0000 | [diff] [blame] | 391 | E.run(); | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 392 | return readFile(File2.Path); | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 393 | } | 
|  | 394 |  | 
|  | 395 | // Create a resource file containing a manifest XML. | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 396 | std::unique_ptr<MemoryBuffer> createManifestRes() { | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 397 | // Create a temporary file for the resource script file. | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 398 | TemporaryFile RCFile("manifest", "rc"); | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 399 |  | 
|  | 400 | // Open the temporary file for writing. | 
| Rui Ueyama | 0d09a86 | 2016-07-15 00:40:46 +0000 | [diff] [blame] | 401 | std::error_code EC; | 
| Rui Ueyama | 8fe1767 | 2016-12-08 20:50:47 +0000 | [diff] [blame] | 402 | raw_fd_ostream Out(RCFile.Path, EC, sys::fs::F_Text); | 
| Rui Ueyama | 0d09a86 | 2016-07-15 00:40:46 +0000 | [diff] [blame] | 403 | if (EC) | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 404 | fatal(EC, "failed to open " + RCFile.Path); | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 405 |  | 
|  | 406 | // Write resource script to the RC file. | 
|  | 407 | Out << "#define LANG_ENGLISH 9\n" | 
|  | 408 | << "#define SUBLANG_DEFAULT 1\n" | 
|  | 409 | << "#define APP_MANIFEST " << Config->ManifestID << "\n" | 
|  | 410 | << "#define RT_MANIFEST 24\n" | 
|  | 411 | << "LANGUAGE LANG_ENGLISH, SUBLANG_DEFAULT\n" | 
|  | 412 | << "APP_MANIFEST RT_MANIFEST {\n"; | 
|  | 413 | quoteAndPrint(Out, createManifestXml()); | 
|  | 414 | Out << "}\n"; | 
|  | 415 | Out.close(); | 
|  | 416 |  | 
|  | 417 | // Create output resource file. | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 418 | TemporaryFile ResFile("output-resource", "res"); | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 419 |  | 
|  | 420 | Executor E("rc.exe"); | 
|  | 421 | E.add("/fo"); | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 422 | E.add(ResFile.Path); | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 423 | E.add("/nologo"); | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 424 | E.add(RCFile.Path); | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 425 | E.run(); | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 426 | return ResFile.getMemoryBuffer(); | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 427 | } | 
|  | 428 |  | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 429 | void createSideBySideManifest() { | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 430 | std::string Path = Config->ManifestFile; | 
|  | 431 | if (Path == "") | 
| Rui Ueyama | 1a3fd13 | 2016-07-14 23:43:36 +0000 | [diff] [blame] | 432 | Path = Config->OutputFile + ".manifest"; | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 433 | std::error_code EC; | 
| Rui Ueyama | 8fe1767 | 2016-12-08 20:50:47 +0000 | [diff] [blame] | 434 | raw_fd_ostream Out(Path, EC, sys::fs::F_Text); | 
| Rui Ueyama | 0d09a86 | 2016-07-15 00:40:46 +0000 | [diff] [blame] | 435 | if (EC) | 
|  | 436 | fatal(EC, "failed to create manifest"); | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 437 | Out << createManifestXml(); | 
| Rui Ueyama | 24c5fd0 | 2015-06-18 00:12:42 +0000 | [diff] [blame] | 438 | } | 
|  | 439 |  | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 440 | // Parse a string in the form of | 
| Rui Ueyama | 84425d7 | 2016-01-09 01:22:00 +0000 | [diff] [blame] | 441 | // "<name>[=<internalname>][,@ordinal[,NONAME]][,DATA][,PRIVATE]" | 
|  | 442 | // or "<name>=<dllname>.<name>". | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 443 | // Used for parsing /export arguments. | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 444 | Export parseExport(StringRef Arg) { | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 445 | Export E; | 
|  | 446 | StringRef Rest; | 
|  | 447 | std::tie(E.Name, Rest) = Arg.split(","); | 
|  | 448 | if (E.Name.empty()) | 
|  | 449 | goto err; | 
| Rui Ueyama | 84425d7 | 2016-01-09 01:22:00 +0000 | [diff] [blame] | 450 |  | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 451 | if (E.Name.find('=') != StringRef::npos) { | 
| Rui Ueyama | 84425d7 | 2016-01-09 01:22:00 +0000 | [diff] [blame] | 452 | StringRef X, Y; | 
|  | 453 | std::tie(X, Y) = E.Name.split("="); | 
|  | 454 |  | 
|  | 455 | // If "<name>=<dllname>.<name>". | 
|  | 456 | if (Y.find(".") != StringRef::npos) { | 
|  | 457 | E.Name = X; | 
|  | 458 | E.ForwardTo = Y; | 
|  | 459 | return E; | 
|  | 460 | } | 
|  | 461 |  | 
|  | 462 | E.ExtName = X; | 
|  | 463 | E.Name = Y; | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 464 | if (E.Name.empty()) | 
|  | 465 | goto err; | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 466 | } | 
|  | 467 |  | 
| Rui Ueyama | 84425d7 | 2016-01-09 01:22:00 +0000 | [diff] [blame] | 468 | // If "<name>=<internalname>[,@ordinal[,NONAME]][,DATA][,PRIVATE]" | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 469 | while (!Rest.empty()) { | 
|  | 470 | StringRef Tok; | 
|  | 471 | std::tie(Tok, Rest) = Rest.split(","); | 
|  | 472 | if (Tok.equals_lower("noname")) { | 
|  | 473 | if (E.Ordinal == 0) | 
|  | 474 | goto err; | 
|  | 475 | E.Noname = true; | 
|  | 476 | continue; | 
|  | 477 | } | 
|  | 478 | if (Tok.equals_lower("data")) { | 
|  | 479 | E.Data = true; | 
|  | 480 | continue; | 
|  | 481 | } | 
|  | 482 | if (Tok.equals_lower("private")) { | 
|  | 483 | E.Private = true; | 
|  | 484 | continue; | 
|  | 485 | } | 
|  | 486 | if (Tok.startswith("@")) { | 
|  | 487 | int32_t Ord; | 
|  | 488 | if (Tok.substr(1).getAsInteger(0, Ord)) | 
|  | 489 | goto err; | 
|  | 490 | if (Ord <= 0 || 65535 < Ord) | 
|  | 491 | goto err; | 
|  | 492 | E.Ordinal = Ord; | 
|  | 493 | continue; | 
|  | 494 | } | 
|  | 495 | goto err; | 
|  | 496 | } | 
|  | 497 | return E; | 
|  | 498 |  | 
|  | 499 | err: | 
| Rui Ueyama | 1a3fd13 | 2016-07-14 23:43:36 +0000 | [diff] [blame] | 500 | fatal("invalid /export: " + Arg); | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 501 | } | 
|  | 502 |  | 
| Rui Ueyama | f10a320 | 2015-08-31 08:43:21 +0000 | [diff] [blame] | 503 | static StringRef undecorate(StringRef Sym) { | 
|  | 504 | if (Config->Machine != I386) | 
|  | 505 | return Sym; | 
|  | 506 | return Sym.startswith("_") ? Sym.substr(1) : Sym; | 
|  | 507 | } | 
|  | 508 |  | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 509 | // Performs error checking on all /export arguments. | 
|  | 510 | // It also sets ordinals. | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 511 | void fixupExports() { | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 512 | // Symbol ordinals must be unique. | 
|  | 513 | std::set<uint16_t> Ords; | 
|  | 514 | for (Export &E : Config->Exports) { | 
|  | 515 | if (E.Ordinal == 0) | 
|  | 516 | continue; | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 517 | if (!Ords.insert(E.Ordinal).second) | 
| Rui Ueyama | 6060479 | 2016-07-14 23:37:14 +0000 | [diff] [blame] | 518 | fatal("duplicate export ordinal: " + E.Name); | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 519 | } | 
|  | 520 |  | 
| Rui Ueyama | 8765fba | 2015-07-15 22:21:08 +0000 | [diff] [blame] | 521 | for (Export &E : Config->Exports) { | 
| Peter Collingbourne | 79a5e6b | 2016-12-09 21:55:24 +0000 | [diff] [blame] | 522 | SymbolBody *Sym = E.Sym; | 
| Rui Ueyama | 84425d7 | 2016-01-09 01:22:00 +0000 | [diff] [blame] | 523 | if (!E.ForwardTo.empty()) { | 
|  | 524 | E.SymbolName = E.Name; | 
| Peter Collingbourne | 6baa5d2 | 2016-12-09 23:34:49 +0000 | [diff] [blame] | 525 | } else { | 
|  | 526 | if (auto *U = dyn_cast<Undefined>(Sym)) | 
|  | 527 | if (U->WeakAlias) | 
|  | 528 | Sym = U->WeakAlias; | 
|  | 529 | E.SymbolName = Sym->getName(); | 
|  | 530 | } | 
| Rui Ueyama | 8765fba | 2015-07-15 22:21:08 +0000 | [diff] [blame] | 531 | } | 
|  | 532 |  | 
| Rui Ueyama | 84425d7 | 2016-01-09 01:22:00 +0000 | [diff] [blame] | 533 | for (Export &E : Config->Exports) { | 
|  | 534 | if (!E.ForwardTo.empty()) { | 
|  | 535 | E.ExportName = undecorate(E.Name); | 
|  | 536 | } else { | 
|  | 537 | E.ExportName = undecorate(E.ExtName.empty() ? E.Name : E.ExtName); | 
|  | 538 | } | 
|  | 539 | } | 
| Rui Ueyama | f10a320 | 2015-08-31 08:43:21 +0000 | [diff] [blame] | 540 |  | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 541 | // Uniquefy by name. | 
| Rui Ueyama | 4b8cdd2 | 2015-07-03 23:23:29 +0000 | [diff] [blame] | 542 | std::map<StringRef, Export *> Map; | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 543 | std::vector<Export> V; | 
|  | 544 | for (Export &E : Config->Exports) { | 
| Rui Ueyama | f10a320 | 2015-08-31 08:43:21 +0000 | [diff] [blame] | 545 | auto Pair = Map.insert(std::make_pair(E.ExportName, &E)); | 
| Rui Ueyama | 3126c6c | 2015-07-04 02:00:22 +0000 | [diff] [blame] | 546 | bool Inserted = Pair.second; | 
|  | 547 | if (Inserted) { | 
| Rui Ueyama | 4b8cdd2 | 2015-07-03 23:23:29 +0000 | [diff] [blame] | 548 | V.push_back(E); | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 549 | continue; | 
|  | 550 | } | 
| Rui Ueyama | 3126c6c | 2015-07-04 02:00:22 +0000 | [diff] [blame] | 551 | Export *Existing = Pair.first->second; | 
| Rui Ueyama | 966acb2 | 2015-07-29 22:38:27 +0000 | [diff] [blame] | 552 | if (E == *Existing || E.Name != Existing->Name) | 
| Rui Ueyama | 4b8cdd2 | 2015-07-03 23:23:29 +0000 | [diff] [blame] | 553 | continue; | 
| Rui Ueyama | e6e206d | 2017-02-21 23:22:56 +0000 | [diff] [blame] | 554 | warn("duplicate /export option: " + E.Name); | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 555 | } | 
|  | 556 | Config->Exports = std::move(V); | 
|  | 557 |  | 
|  | 558 | // Sort by name. | 
| Rui Ueyama | 9420dee | 2015-07-28 22:34:24 +0000 | [diff] [blame] | 559 | std::sort(Config->Exports.begin(), Config->Exports.end(), | 
|  | 560 | [](const Export &A, const Export &B) { | 
| Rui Ueyama | f10a320 | 2015-08-31 08:43:21 +0000 | [diff] [blame] | 561 | return A.ExportName < B.ExportName; | 
| Rui Ueyama | 9420dee | 2015-07-28 22:34:24 +0000 | [diff] [blame] | 562 | }); | 
| Rui Ueyama | 8765fba | 2015-07-15 22:21:08 +0000 | [diff] [blame] | 563 | } | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 564 |  | 
| Rui Ueyama | 8765fba | 2015-07-15 22:21:08 +0000 | [diff] [blame] | 565 | void assignExportOrdinals() { | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 566 | // Assign unique ordinals if default (= 0). | 
|  | 567 | uint16_t Max = 0; | 
|  | 568 | for (Export &E : Config->Exports) | 
|  | 569 | Max = std::max(Max, E.Ordinal); | 
|  | 570 | for (Export &E : Config->Exports) | 
|  | 571 | if (E.Ordinal == 0) | 
|  | 572 | E.Ordinal = ++Max; | 
| Rui Ueyama | 97dff9e | 2015-06-17 00:16:33 +0000 | [diff] [blame] | 573 | } | 
|  | 574 |  | 
| Rui Ueyama | 8854d8a | 2015-06-04 19:21:24 +0000 | [diff] [blame] | 575 | // Parses a string in the form of "key=value" and check | 
|  | 576 | // if value matches previous values for the same key. | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 577 | void checkFailIfMismatch(StringRef Arg) { | 
| Rui Ueyama | 75b098b | 2015-06-18 21:23:34 +0000 | [diff] [blame] | 578 | StringRef K, V; | 
|  | 579 | std::tie(K, V) = Arg.split('='); | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 580 | if (K.empty() || V.empty()) | 
| Rui Ueyama | 1a3fd13 | 2016-07-14 23:43:36 +0000 | [diff] [blame] | 581 | fatal("/failifmismatch: invalid argument: " + Arg); | 
| Rui Ueyama | 75b098b | 2015-06-18 21:23:34 +0000 | [diff] [blame] | 582 | StringRef Existing = Config->MustMatch[K]; | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 583 | if (!Existing.empty() && V != Existing) | 
| Rui Ueyama | 1a3fd13 | 2016-07-14 23:43:36 +0000 | [diff] [blame] | 584 | fatal("/failifmismatch: mismatch detected: " + Existing + " and " + V + | 
|  | 585 | " for key " + K); | 
| Rui Ueyama | 75b098b | 2015-06-18 21:23:34 +0000 | [diff] [blame] | 586 | Config->MustMatch[K] = V; | 
| Rui Ueyama | 8854d8a | 2015-06-04 19:21:24 +0000 | [diff] [blame] | 587 | } | 
|  | 588 |  | 
| Rui Ueyama | 2bf6a12 | 2015-06-14 21:50:50 +0000 | [diff] [blame] | 589 | // Convert Windows resource files (.res files) to a .obj file | 
|  | 590 | // using cvtres.exe. | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 591 | std::unique_ptr<MemoryBuffer> | 
| Rui Ueyama | 2bf6a12 | 2015-06-14 21:50:50 +0000 | [diff] [blame] | 592 | convertResToCOFF(const std::vector<MemoryBufferRef> &MBs) { | 
| Rui Ueyama | 2bf6a12 | 2015-06-14 21:50:50 +0000 | [diff] [blame] | 593 | // Create an output file path. | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 594 | TemporaryFile File("resource-file", "obj"); | 
| Rui Ueyama | 2bf6a12 | 2015-06-14 21:50:50 +0000 | [diff] [blame] | 595 |  | 
|  | 596 | // Execute cvtres.exe. | 
| Rui Ueyama | a9c8838 | 2015-06-17 21:01:56 +0000 | [diff] [blame] | 597 | Executor E("cvtres.exe"); | 
| Rui Ueyama | 5e706b3 | 2015-07-25 21:54:50 +0000 | [diff] [blame] | 598 | E.add("/machine:" + machineToStr(Config->Machine)); | 
| Rui Ueyama | a9c8838 | 2015-06-17 21:01:56 +0000 | [diff] [blame] | 599 | E.add("/readonly"); | 
|  | 600 | E.add("/nologo"); | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 601 | E.add("/out:" + Twine(File.Path)); | 
| Rui Ueyama | c3dcf99 | 2016-11-15 21:25:20 +0000 | [diff] [blame] | 602 |  | 
|  | 603 | // We must create new files because the memory buffers we have may have no | 
|  | 604 | // underlying file still existing on the disk. | 
|  | 605 | // It happens if it was created from a TemporaryFile, which usually delete | 
|  | 606 | // the file just after creating the MemoryBuffer. | 
|  | 607 | std::vector<TemporaryFile> ResFiles; | 
|  | 608 | ResFiles.reserve(MBs.size()); | 
|  | 609 | for (MemoryBufferRef MB : MBs) { | 
|  | 610 | // We store the temporary file in a vector to avoid deletion | 
|  | 611 | // before running cvtres | 
|  | 612 | ResFiles.emplace_back("resource-file", "res"); | 
|  | 613 | TemporaryFile& ResFile = ResFiles.back(); | 
|  | 614 | // Write the content of the resource in a temporary file | 
|  | 615 | std::error_code EC; | 
| Rui Ueyama | 8fe1767 | 2016-12-08 20:50:47 +0000 | [diff] [blame] | 616 | raw_fd_ostream OS(ResFile.Path, EC, sys::fs::F_None); | 
| Rui Ueyama | c3dcf99 | 2016-11-15 21:25:20 +0000 | [diff] [blame] | 617 | if (EC) | 
|  | 618 | fatal(EC, "failed to open " + ResFile.Path); | 
|  | 619 | OS << MB.getBuffer(); | 
|  | 620 | OS.close(); | 
|  | 621 |  | 
|  | 622 | E.add(ResFile.Path); | 
|  | 623 | } | 
|  | 624 |  | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 625 | E.run(); | 
| Rui Ueyama | fa12b8b | 2016-09-02 17:34:17 +0000 | [diff] [blame] | 626 | return File.getMemoryBuffer(); | 
| Rui Ueyama | 2bf6a12 | 2015-06-14 21:50:50 +0000 | [diff] [blame] | 627 | } | 
|  | 628 |  | 
| Rui Ueyama | 1e0b158 | 2017-02-06 20:47:55 +0000 | [diff] [blame] | 629 | // Run MSVC link.exe for given in-memory object files. | 
|  | 630 | // Command line options are copied from those given to LLD. | 
|  | 631 | // This is for the /msvclto option. | 
| Rui Ueyama | 85d54b0 | 2017-02-23 00:26:42 +0000 | [diff] [blame] | 632 | void runMSVCLinker(std::string Rsp, ArrayRef<StringRef> Objects) { | 
| Rui Ueyama | 1e0b158 | 2017-02-06 20:47:55 +0000 | [diff] [blame] | 633 | // Write the in-memory object files to disk. | 
|  | 634 | std::vector<TemporaryFile> Temps; | 
| Rui Ueyama | 85d54b0 | 2017-02-23 00:26:42 +0000 | [diff] [blame] | 635 | for (StringRef S : Objects) { | 
| Rui Ueyama | 1e0b158 | 2017-02-06 20:47:55 +0000 | [diff] [blame] | 636 | Temps.emplace_back("lto", "obj", S); | 
| Rui Ueyama | 85d54b0 | 2017-02-23 00:26:42 +0000 | [diff] [blame] | 637 | Rsp += quote(Temps.back().Path) + " "; | 
| Rui Ueyama | 1e0b158 | 2017-02-06 20:47:55 +0000 | [diff] [blame] | 638 | } | 
|  | 639 |  | 
| Rui Ueyama | e6e206d | 2017-02-21 23:22:56 +0000 | [diff] [blame] | 640 | log("link.exe " + Rsp); | 
| Rui Ueyama | 1e0b158 | 2017-02-06 20:47:55 +0000 | [diff] [blame] | 641 |  | 
|  | 642 | // Run MSVC link.exe. | 
|  | 643 | Temps.emplace_back("lto", "rsp", Rsp); | 
|  | 644 | Executor E("link.exe"); | 
|  | 645 | E.add(Twine("@" + Temps.back().Path)); | 
|  | 646 | E.run(); | 
|  | 647 | } | 
|  | 648 |  | 
| Rui Ueyama | 3500f66 | 2015-05-28 20:30:06 +0000 | [diff] [blame] | 649 | // Create OptTable | 
|  | 650 |  | 
|  | 651 | // Create prefix string literals used in Options.td | 
|  | 652 | #define PREFIX(NAME, VALUE) const char *const NAME[] = VALUE; | 
|  | 653 | #include "Options.inc" | 
|  | 654 | #undef PREFIX | 
|  | 655 |  | 
|  | 656 | // Create table mapping all options defined in Options.td | 
|  | 657 | static const llvm::opt::OptTable::Info infoTable[] = { | 
|  | 658 | #define OPTION(X1, X2, ID, KIND, GROUP, ALIAS, X6, X7, X8, X9, X10)    \ | 
|  | 659 | {                                                                    \ | 
|  | 660 | X1, X2, X9, X10, OPT_##ID, llvm::opt::Option::KIND##Class, X8, X7, \ | 
|  | 661 | OPT_##GROUP, OPT_##ALIAS, X6                                       \ | 
|  | 662 | }, | 
|  | 663 | #include "Options.inc" | 
|  | 664 | #undef OPTION | 
|  | 665 | }; | 
|  | 666 |  | 
|  | 667 | class COFFOptTable : public llvm::opt::OptTable { | 
|  | 668 | public: | 
| Craig Topper | f88d2297 | 2015-10-21 16:31:56 +0000 | [diff] [blame] | 669 | COFFOptTable() : OptTable(infoTable, true) {} | 
| Rui Ueyama | 3500f66 | 2015-05-28 20:30:06 +0000 | [diff] [blame] | 670 | }; | 
|  | 671 |  | 
| Rui Ueyama | 115d7c1 | 2015-06-07 02:55:19 +0000 | [diff] [blame] | 672 | // Parses a given list of options. | 
| Rui Ueyama | 8fe1767 | 2016-12-08 20:50:47 +0000 | [diff] [blame] | 673 | opt::InputArgList ArgParser::parse(ArrayRef<const char *> ArgsArr) { | 
| Rui Ueyama | 115d7c1 | 2015-06-07 02:55:19 +0000 | [diff] [blame] | 674 | // First, replace respnose files (@<file>-style options). | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 675 | std::vector<const char *> Argv = replaceResponseFiles(ArgsArr); | 
| Rui Ueyama | 115d7c1 | 2015-06-07 02:55:19 +0000 | [diff] [blame] | 676 |  | 
|  | 677 | // Make InputArgList from string vectors. | 
| Rui Ueyama | 3500f66 | 2015-05-28 20:30:06 +0000 | [diff] [blame] | 678 | COFFOptTable Table; | 
|  | 679 | unsigned MissingIndex; | 
|  | 680 | unsigned MissingCount; | 
| Rui Ueyama | 8fe1767 | 2016-12-08 20:50:47 +0000 | [diff] [blame] | 681 | opt::InputArgList Args = Table.ParseArgs(Argv, MissingIndex, MissingCount); | 
| Rui Ueyama | 2caf407 | 2015-08-26 07:12:08 +0000 | [diff] [blame] | 682 |  | 
|  | 683 | // Print the real command line if response files are expanded. | 
|  | 684 | if (Args.hasArg(OPT_verbose) && ArgsArr.size() != Argv.size()) { | 
| Rui Ueyama | e6e206d | 2017-02-21 23:22:56 +0000 | [diff] [blame] | 685 | std::string Msg = "Command line:"; | 
| Rui Ueyama | 2caf407 | 2015-08-26 07:12:08 +0000 | [diff] [blame] | 686 | for (const char *S : Argv) | 
| Rui Ueyama | e6e206d | 2017-02-21 23:22:56 +0000 | [diff] [blame] | 687 | Msg += " " + std::string(S); | 
|  | 688 | message(Msg); | 
| Rui Ueyama | 2caf407 | 2015-08-26 07:12:08 +0000 | [diff] [blame] | 689 | } | 
|  | 690 |  | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 691 | if (MissingCount) | 
| Rui Ueyama | c910bc7 | 2016-11-29 04:17:31 +0000 | [diff] [blame] | 692 | fatal(Twine(Args.getArgString(MissingIndex)) + ": missing argument"); | 
| David Blaikie | 6521ed9 | 2015-06-22 22:06:52 +0000 | [diff] [blame] | 693 | for (auto *Arg : Args.filtered(OPT_UNKNOWN)) | 
| Rui Ueyama | e6e206d | 2017-02-21 23:22:56 +0000 | [diff] [blame] | 694 | warn("ignoring unknown argument: " + Arg->getSpelling()); | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 695 | return Args; | 
| Rui Ueyama | 3500f66 | 2015-05-28 20:30:06 +0000 | [diff] [blame] | 696 | } | 
|  | 697 |  | 
| Rui Ueyama | e5e407b | 2016-11-29 04:17:30 +0000 | [diff] [blame] | 698 | // link.exe has an interesting feature. If LINK environment exists, | 
|  | 699 | // its contents are handled as a command line string. So you can pass | 
|  | 700 | // extra arguments using the environment variable. | 
| Rui Ueyama | 8fe1767 | 2016-12-08 20:50:47 +0000 | [diff] [blame] | 701 | opt::InputArgList ArgParser::parseLINK(ArrayRef<const char *> Args) { | 
| Rui Ueyama | e5e407b | 2016-11-29 04:17:30 +0000 | [diff] [blame] | 702 | // Concatenate LINK env and command line arguments, and then parse them. | 
| Rui Ueyama | 06cf3df | 2015-06-28 02:35:31 +0000 | [diff] [blame] | 703 | Optional<std::string> Env = Process::GetEnv("LINK"); | 
|  | 704 | if (!Env) | 
|  | 705 | return parse(Args); | 
|  | 706 | std::vector<const char *> V = tokenize(*Env); | 
| Rui Ueyama | 9d72f09 | 2015-06-28 03:05:38 +0000 | [diff] [blame] | 707 | V.insert(V.end(), Args.begin(), Args.end()); | 
| Rui Ueyama | 06cf3df | 2015-06-28 02:35:31 +0000 | [diff] [blame] | 708 | return parse(V); | 
|  | 709 | } | 
|  | 710 |  | 
| Rui Ueyama | 115d7c1 | 2015-06-07 02:55:19 +0000 | [diff] [blame] | 711 | std::vector<const char *> ArgParser::tokenize(StringRef S) { | 
|  | 712 | SmallVector<const char *, 16> Tokens; | 
| Rui Ueyama | 8fe1767 | 2016-12-08 20:50:47 +0000 | [diff] [blame] | 713 | cl::TokenizeWindowsCommandLine(S, Saver, Tokens); | 
| Rui Ueyama | aace577 | 2015-06-07 23:02:50 +0000 | [diff] [blame] | 714 | return std::vector<const char *>(Tokens.begin(), Tokens.end()); | 
| Rui Ueyama | 115d7c1 | 2015-06-07 02:55:19 +0000 | [diff] [blame] | 715 | } | 
|  | 716 |  | 
| Rui Ueyama | b51f67a | 2015-06-07 23:00:29 +0000 | [diff] [blame] | 717 | // Creates a new command line by replacing options starting with '@' | 
| Rui Ueyama | 115d7c1 | 2015-06-07 02:55:19 +0000 | [diff] [blame] | 718 | // character. '@<filename>' is replaced by the file's contents. | 
| Rafael Espindola | b835ae8 | 2015-08-06 14:58:50 +0000 | [diff] [blame] | 719 | std::vector<const char *> | 
| Rui Ueyama | 115d7c1 | 2015-06-07 02:55:19 +0000 | [diff] [blame] | 720 | ArgParser::replaceResponseFiles(std::vector<const char *> Argv) { | 
| Peter Collingbourne | 74ecc89 | 2015-06-19 22:40:05 +0000 | [diff] [blame] | 721 | SmallVector<const char *, 256> Tokens(Argv.data(), Argv.data() + Argv.size()); | 
| Rui Ueyama | b51f67a | 2015-06-07 23:00:29 +0000 | [diff] [blame] | 722 | ExpandResponseFiles(Saver, TokenizeWindowsCommandLine, Tokens); | 
|  | 723 | return std::vector<const char *>(Tokens.begin(), Tokens.end()); | 
| Rui Ueyama | 115d7c1 | 2015-06-07 02:55:19 +0000 | [diff] [blame] | 724 | } | 
|  | 725 |  | 
| Rui Ueyama | 5c72643 | 2015-05-29 16:11:52 +0000 | [diff] [blame] | 726 | void printHelp(const char *Argv0) { | 
|  | 727 | COFFOptTable Table; | 
| Rui Ueyama | 8fe1767 | 2016-12-08 20:50:47 +0000 | [diff] [blame] | 728 | Table.PrintHelp(outs(), Argv0, "LLVM Linker", false); | 
| Rui Ueyama | 5c72643 | 2015-05-29 16:11:52 +0000 | [diff] [blame] | 729 | } | 
|  | 730 |  | 
| Rui Ueyama | 3500f66 | 2015-05-28 20:30:06 +0000 | [diff] [blame] | 731 | } // namespace coff | 
|  | 732 | } // namespace lld |