Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 1 | //===- LinkerScript.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 the parser/evaluator of the linker script. |
Rui Ueyama | 629e0aa5 | 2016-07-21 19:45:22 +0000 | [diff] [blame] | 11 | // It parses a linker script and write the result to Config or ScriptConfig |
| 12 | // objects. |
| 13 | // |
| 14 | // If SECTIONS command is used, a ScriptConfig contains an AST |
| 15 | // of the command which will later be consumed by createSections() and |
| 16 | // assignAddresses(). |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 17 | // |
| 18 | //===----------------------------------------------------------------------===// |
| 19 | |
Rui Ueyama | 717677a | 2016-02-11 21:17:59 +0000 | [diff] [blame] | 20 | #include "LinkerScript.h" |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 21 | #include "Config.h" |
| 22 | #include "Driver.h" |
Rui Ueyama | 1ebc8ed | 2016-02-12 21:47:28 +0000 | [diff] [blame] | 23 | #include "InputSection.h" |
George Rimar | 652852c | 2016-04-16 10:10:32 +0000 | [diff] [blame] | 24 | #include "OutputSections.h" |
Adhemerval Zanella | e77b5bf | 2016-04-06 20:59:11 +0000 | [diff] [blame] | 25 | #include "ScriptParser.h" |
Rui Ueyama | 93c9af4 | 2016-06-29 08:01:32 +0000 | [diff] [blame] | 26 | #include "Strings.h" |
Eugene Leviant | eda81a1 | 2016-07-12 06:39:48 +0000 | [diff] [blame] | 27 | #include "Symbols.h" |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 28 | #include "SymbolTable.h" |
Eugene Leviant | 467c4d5 | 2016-07-01 10:27:36 +0000 | [diff] [blame] | 29 | #include "Target.h" |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 30 | #include "Writer.h" |
Rui Ueyama | 960504b | 2016-04-19 18:58:11 +0000 | [diff] [blame] | 31 | #include "llvm/ADT/StringSwitch.h" |
George Rimar | 652852c | 2016-04-16 10:10:32 +0000 | [diff] [blame] | 32 | #include "llvm/Support/ELF.h" |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 33 | #include "llvm/Support/FileSystem.h" |
| 34 | #include "llvm/Support/MemoryBuffer.h" |
Rui Ueyama | f03f3cc | 2015-10-13 00:09:21 +0000 | [diff] [blame] | 35 | #include "llvm/Support/Path.h" |
Rui Ueyama | a47ee68 | 2015-10-11 01:53:04 +0000 | [diff] [blame] | 36 | #include "llvm/Support/StringSaver.h" |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 37 | |
| 38 | using namespace llvm; |
George Rimar | 652852c | 2016-04-16 10:10:32 +0000 | [diff] [blame] | 39 | using namespace llvm::ELF; |
Rui Ueyama | 1ebc8ed | 2016-02-12 21:47:28 +0000 | [diff] [blame] | 40 | using namespace llvm::object; |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 41 | using namespace lld; |
Rafael Espindola | e0df00b | 2016-02-28 00:25:54 +0000 | [diff] [blame] | 42 | using namespace lld::elf; |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 43 | |
George Rimar | 884e786 | 2016-09-08 08:19:13 +0000 | [diff] [blame] | 44 | LinkerScriptBase *elf::ScriptBase; |
Rui Ueyama | 07320e4 | 2016-04-20 20:13:41 +0000 | [diff] [blame] | 45 | ScriptConfiguration *elf::ScriptConfig; |
Rui Ueyama | 717677a | 2016-02-11 21:17:59 +0000 | [diff] [blame] | 46 | |
George Rimar | 6c55f0e | 2016-09-08 08:20:30 +0000 | [diff] [blame] | 47 | template <class ELFT> static void addRegular(SymbolAssignment *Cmd) { |
Rui Ueyama | 1602421 | 2016-08-11 23:22:52 +0000 | [diff] [blame] | 48 | Symbol *Sym = Symtab<ELFT>::X->addRegular(Cmd->Name, STB_GLOBAL, STV_DEFAULT); |
| 49 | Sym->Visibility = Cmd->Hidden ? STV_HIDDEN : STV_DEFAULT; |
| 50 | Cmd->Sym = Sym->body(); |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 51 | } |
| 52 | |
Rui Ueyama | 0c70d3c | 2016-08-12 03:31:09 +0000 | [diff] [blame] | 53 | template <class ELFT> static void addSynthetic(SymbolAssignment *Cmd) { |
George Rimar | e1937bb | 2016-08-19 15:36:32 +0000 | [diff] [blame] | 54 | Symbol *Sym = Symtab<ELFT>::X->addSynthetic( |
| 55 | Cmd->Name, nullptr, 0, Cmd->Hidden ? STV_HIDDEN : STV_DEFAULT); |
Rui Ueyama | 1602421 | 2016-08-11 23:22:52 +0000 | [diff] [blame] | 56 | Cmd->Sym = Sym->body(); |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 57 | } |
| 58 | |
Eugene Leviant | db741e7 | 2016-09-07 07:08:43 +0000 | [diff] [blame] | 59 | template <class ELFT> static void addSymbol(SymbolAssignment *Cmd) { |
| 60 | if (Cmd->IsAbsolute) |
| 61 | addRegular<ELFT>(Cmd); |
| 62 | else |
| 63 | addSynthetic<ELFT>(Cmd); |
| 64 | } |
Rui Ueyama | 1602421 | 2016-08-11 23:22:52 +0000 | [diff] [blame] | 65 | // If a symbol was in PROVIDE(), we need to define it only when |
| 66 | // it is an undefined symbol. |
| 67 | template <class ELFT> static bool shouldDefine(SymbolAssignment *Cmd) { |
| 68 | if (Cmd->Name == ".") |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 69 | return false; |
Rui Ueyama | 1602421 | 2016-08-11 23:22:52 +0000 | [diff] [blame] | 70 | if (!Cmd->Provide) |
| 71 | return true; |
| 72 | SymbolBody *B = Symtab<ELFT>::X->find(Cmd->Name); |
| 73 | return B && B->isUndefined(); |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 74 | } |
| 75 | |
George Rimar | 076fe15 | 2016-07-21 06:43:01 +0000 | [diff] [blame] | 76 | bool SymbolAssignment::classof(const BaseCommand *C) { |
| 77 | return C->Kind == AssignmentKind; |
| 78 | } |
| 79 | |
| 80 | bool OutputSectionCommand::classof(const BaseCommand *C) { |
| 81 | return C->Kind == OutputSectionKind; |
| 82 | } |
| 83 | |
George Rimar | eea3114 | 2016-07-21 14:26:59 +0000 | [diff] [blame] | 84 | bool InputSectionDescription::classof(const BaseCommand *C) { |
| 85 | return C->Kind == InputSectionKind; |
| 86 | } |
| 87 | |
George Rimar | eefa758 | 2016-08-04 09:29:31 +0000 | [diff] [blame] | 88 | bool AssertCommand::classof(const BaseCommand *C) { |
| 89 | return C->Kind == AssertKind; |
| 90 | } |
| 91 | |
Rui Ueyama | 36a153c | 2016-07-23 14:09:58 +0000 | [diff] [blame] | 92 | template <class ELFT> static bool isDiscarded(InputSectionBase<ELFT> *S) { |
George Rimar | eea3114 | 2016-07-21 14:26:59 +0000 | [diff] [blame] | 93 | return !S || !S->Live; |
Rui Ueyama | 717677a | 2016-02-11 21:17:59 +0000 | [diff] [blame] | 94 | } |
| 95 | |
Rui Ueyama | f34d0e0 | 2016-08-12 01:24:53 +0000 | [diff] [blame] | 96 | template <class ELFT> LinkerScript<ELFT>::LinkerScript() {} |
| 97 | template <class ELFT> LinkerScript<ELFT>::~LinkerScript() {} |
| 98 | |
Rui Ueyama | 07320e4 | 2016-04-20 20:13:41 +0000 | [diff] [blame] | 99 | template <class ELFT> |
| 100 | bool LinkerScript<ELFT>::shouldKeep(InputSectionBase<ELFT> *S) { |
George Rimar | c91930a | 2016-09-02 21:17:20 +0000 | [diff] [blame] | 101 | for (Regex *Re : Opt.KeptSections) |
Rafael Espindola | 042a3f2 | 2016-09-08 14:06:08 +0000 | [diff] [blame] | 102 | if (Re->match(S->Name)) |
George Rimar | eea3114 | 2016-07-21 14:26:59 +0000 | [diff] [blame] | 103 | return true; |
| 104 | return false; |
| 105 | } |
| 106 | |
George Rimar | 0659800 | 2016-07-28 21:51:30 +0000 | [diff] [blame] | 107 | static bool fileMatches(const InputSectionDescription *Desc, |
| 108 | StringRef Filename) { |
George Rimar | c91930a | 2016-09-02 21:17:20 +0000 | [diff] [blame] | 109 | return const_cast<Regex &>(Desc->FileRe).match(Filename) && |
| 110 | !const_cast<Regex &>(Desc->ExcludedFileRe).match(Filename); |
George Rimar | 0659800 | 2016-07-28 21:51:30 +0000 | [diff] [blame] | 111 | } |
| 112 | |
Rui Ueyama | 6b27481 | 2016-07-25 22:51:07 +0000 | [diff] [blame] | 113 | // Returns input sections filtered by given glob patterns. |
| 114 | template <class ELFT> |
| 115 | std::vector<InputSectionBase<ELFT> *> |
Rui Ueyama | ad10c3d | 2016-07-28 21:05:04 +0000 | [diff] [blame] | 116 | LinkerScript<ELFT>::getInputSections(const InputSectionDescription *I) { |
George Rimar | c91930a | 2016-09-02 21:17:20 +0000 | [diff] [blame] | 117 | const Regex &Re = I->SectionRe; |
Rui Ueyama | 6b27481 | 2016-07-25 22:51:07 +0000 | [diff] [blame] | 118 | std::vector<InputSectionBase<ELFT> *> Ret; |
| 119 | for (const std::unique_ptr<ObjectFile<ELFT>> &F : |
George Rimar | 0659800 | 2016-07-28 21:51:30 +0000 | [diff] [blame] | 120 | Symtab<ELFT>::X->getObjectFiles()) { |
| 121 | if (fileMatches(I, sys::path::filename(F->getName()))) |
| 122 | for (InputSectionBase<ELFT> *S : F->getSections()) |
| 123 | if (!isDiscarded(S) && !S->OutSec && |
Rafael Espindola | 042a3f2 | 2016-09-08 14:06:08 +0000 | [diff] [blame] | 124 | const_cast<Regex &>(Re).match(S->Name)) |
Davide Italiano | e728279 | 2016-07-27 01:44:01 +0000 | [diff] [blame] | 125 | Ret.push_back(S); |
George Rimar | 0659800 | 2016-07-28 21:51:30 +0000 | [diff] [blame] | 126 | } |
Eugene Leviant | 3e6b027 | 2016-07-28 19:24:13 +0000 | [diff] [blame] | 127 | |
George Rimar | c91930a | 2016-09-02 21:17:20 +0000 | [diff] [blame] | 128 | if (const_cast<Regex &>(Re).match("COMMON")) |
Rui Ueyama | ad10c3d | 2016-07-28 21:05:04 +0000 | [diff] [blame] | 129 | Ret.push_back(CommonInputSection<ELFT>::X); |
Rui Ueyama | 6b27481 | 2016-07-25 22:51:07 +0000 | [diff] [blame] | 130 | return Ret; |
| 131 | } |
| 132 | |
Rafael Espindola | c0028d3 | 2016-09-08 20:47:52 +0000 | [diff] [blame] | 133 | static bool compareName(InputSectionData *A, InputSectionData *B) { |
Rafael Espindola | 042a3f2 | 2016-09-08 14:06:08 +0000 | [diff] [blame] | 134 | return A->Name < B->Name; |
Rui Ueyama | 742c383 | 2016-08-04 22:27:00 +0000 | [diff] [blame] | 135 | } |
George Rimar | 350ece4 | 2016-08-03 08:35:59 +0000 | [diff] [blame] | 136 | |
Rafael Espindola | c0028d3 | 2016-09-08 20:47:52 +0000 | [diff] [blame] | 137 | static bool compareAlignment(InputSectionData *A, InputSectionData *B) { |
Rui Ueyama | 742c383 | 2016-08-04 22:27:00 +0000 | [diff] [blame] | 138 | // ">" is not a mistake. Larger alignments are placed before smaller |
| 139 | // alignments in order to reduce the amount of padding necessary. |
| 140 | // This is compatible with GNU. |
| 141 | return A->Alignment > B->Alignment; |
| 142 | } |
George Rimar | 350ece4 | 2016-08-03 08:35:59 +0000 | [diff] [blame] | 143 | |
Rafael Espindola | c0028d3 | 2016-09-08 20:47:52 +0000 | [diff] [blame] | 144 | static std::function<bool(InputSectionData *, InputSectionData *)> |
Rui Ueyama | 742c383 | 2016-08-04 22:27:00 +0000 | [diff] [blame] | 145 | getComparator(SortKind K) { |
| 146 | if (K == SortByName) |
Rafael Espindola | c0028d3 | 2016-09-08 20:47:52 +0000 | [diff] [blame] | 147 | return compareName; |
| 148 | return compareAlignment; |
Rui Ueyama | 742c383 | 2016-08-04 22:27:00 +0000 | [diff] [blame] | 149 | } |
George Rimar | 0702c4e | 2016-07-29 15:32:46 +0000 | [diff] [blame] | 150 | |
| 151 | template <class ELFT> |
Rafael Espindola | 7bd3787 | 2016-09-12 16:05:16 +0000 | [diff] [blame^] | 152 | void LinkerScript<ELFT>::discard(ArrayRef<InputSectionBase<ELFT> *> V) { |
| 153 | for (InputSectionBase<ELFT> *S : V) { |
| 154 | S->Live = false; |
| 155 | reportDiscarded(S); |
Rui Ueyama | 48c3f1c | 2016-08-12 00:27:23 +0000 | [diff] [blame] | 156 | } |
| 157 | } |
| 158 | |
George Rimar | 8f66df9 | 2016-08-12 20:38:20 +0000 | [diff] [blame] | 159 | static bool checkConstraint(uint64_t Flags, ConstraintKind Kind) { |
| 160 | bool RO = (Kind == ConstraintKind::ReadOnly); |
| 161 | bool RW = (Kind == ConstraintKind::ReadWrite); |
| 162 | bool Writable = Flags & SHF_WRITE; |
Rui Ueyama | adcdb66 | 2016-09-06 22:50:48 +0000 | [diff] [blame] | 163 | return !(RO && Writable) && !(RW && !Writable); |
George Rimar | 8f66df9 | 2016-08-12 20:38:20 +0000 | [diff] [blame] | 164 | } |
| 165 | |
Rui Ueyama | 48c3f1c | 2016-08-12 00:27:23 +0000 | [diff] [blame] | 166 | template <class ELFT> |
George Rimar | 06ae683 | 2016-08-12 09:07:57 +0000 | [diff] [blame] | 167 | static bool matchConstraints(ArrayRef<InputSectionBase<ELFT> *> Sections, |
| 168 | ConstraintKind Kind) { |
George Rimar | 8f66df9 | 2016-08-12 20:38:20 +0000 | [diff] [blame] | 169 | if (Kind == ConstraintKind::NoConstraint) |
| 170 | return true; |
| 171 | return llvm::all_of(Sections, [=](InputSectionBase<ELFT> *Sec) { |
| 172 | return checkConstraint(Sec->getSectionHdr()->sh_flags, Kind); |
George Rimar | 06ae683 | 2016-08-12 09:07:57 +0000 | [diff] [blame] | 173 | }); |
| 174 | } |
| 175 | |
| 176 | template <class ELFT> |
Rui Ueyama | 0b9ce6a | 2016-08-12 03:16:56 +0000 | [diff] [blame] | 177 | std::vector<InputSectionBase<ELFT> *> |
George Rimar | 06ae683 | 2016-08-12 09:07:57 +0000 | [diff] [blame] | 178 | LinkerScript<ELFT>::createInputSectionList(OutputSectionCommand &OutCmd) { |
Rui Ueyama | 0b9ce6a | 2016-08-12 03:16:56 +0000 | [diff] [blame] | 179 | std::vector<InputSectionBase<ELFT> *> Ret; |
Eugene Leviant | 97403d1 | 2016-09-01 09:55:57 +0000 | [diff] [blame] | 180 | DenseSet<InputSectionBase<ELFT> *> SectionIndex; |
Rui Ueyama | e7f912c | 2016-08-03 21:12:09 +0000 | [diff] [blame] | 181 | |
George Rimar | 06ae683 | 2016-08-12 09:07:57 +0000 | [diff] [blame] | 182 | for (const std::unique_ptr<BaseCommand> &Base : OutCmd.Commands) { |
| 183 | if (auto *OutCmd = dyn_cast<SymbolAssignment>(Base.get())) { |
| 184 | if (shouldDefine<ELFT>(OutCmd)) |
Eugene Leviant | db741e7 | 2016-09-07 07:08:43 +0000 | [diff] [blame] | 185 | addSymbol<ELFT>(OutCmd); |
Eugene Leviant | 97403d1 | 2016-09-01 09:55:57 +0000 | [diff] [blame] | 186 | OutCmd->GoesAfter = Ret.empty() ? nullptr : Ret.back(); |
Rui Ueyama | 0b9ce6a | 2016-08-12 03:16:56 +0000 | [diff] [blame] | 187 | continue; |
| 188 | } |
| 189 | |
| 190 | auto *Cmd = cast<InputSectionDescription>(Base.get()); |
| 191 | std::vector<InputSectionBase<ELFT> *> V = getInputSections(Cmd); |
George Rimar | 06ae683 | 2016-08-12 09:07:57 +0000 | [diff] [blame] | 192 | if (!matchConstraints<ELFT>(V, OutCmd.Constraint)) |
| 193 | continue; |
Rui Ueyama | 0b9ce6a | 2016-08-12 03:16:56 +0000 | [diff] [blame] | 194 | if (Cmd->SortInner) |
Rafael Espindola | c0028d3 | 2016-09-08 20:47:52 +0000 | [diff] [blame] | 195 | std::stable_sort(V.begin(), V.end(), getComparator(Cmd->SortInner)); |
Rui Ueyama | 0b9ce6a | 2016-08-12 03:16:56 +0000 | [diff] [blame] | 196 | if (Cmd->SortOuter) |
Rafael Espindola | c0028d3 | 2016-09-08 20:47:52 +0000 | [diff] [blame] | 197 | std::stable_sort(V.begin(), V.end(), getComparator(Cmd->SortOuter)); |
Eugene Leviant | 97403d1 | 2016-09-01 09:55:57 +0000 | [diff] [blame] | 198 | |
| 199 | // Add all input sections corresponding to rule 'Cmd' to |
| 200 | // resulting vector. We do not add duplicate input sections. |
| 201 | for (InputSectionBase<ELFT> *S : V) |
| 202 | if (SectionIndex.insert(S).second) |
| 203 | Ret.push_back(S); |
Rui Ueyama | 0b9ce6a | 2016-08-12 03:16:56 +0000 | [diff] [blame] | 204 | } |
| 205 | return Ret; |
| 206 | } |
| 207 | |
George Rimar | 6c55f0e | 2016-09-08 08:20:30 +0000 | [diff] [blame] | 208 | template <class ELFT> void LinkerScript<ELFT>::createAssignments() { |
Petr Hosek | e5d3ca5 | 2016-08-31 15:31:17 +0000 | [diff] [blame] | 209 | for (const std::unique_ptr<SymbolAssignment> &Cmd : Opt.Assignments) { |
| 210 | if (shouldDefine<ELFT>(Cmd.get())) |
| 211 | addRegular<ELFT>(Cmd.get()); |
| 212 | if (Cmd->Sym) |
| 213 | cast<DefinedRegular<ELFT>>(Cmd->Sym)->Value = Cmd->Expression(0); |
| 214 | } |
| 215 | } |
| 216 | |
| 217 | template <class ELFT> |
Rui Ueyama | 0b9ce6a | 2016-08-12 03:16:56 +0000 | [diff] [blame] | 218 | void LinkerScript<ELFT>::createSections(OutputSectionFactory<ELFT> &Factory) { |
Rui Ueyama | 48c3f1c | 2016-08-12 00:27:23 +0000 | [diff] [blame] | 219 | for (const std::unique_ptr<BaseCommand> &Base1 : Opt.Commands) { |
Rui Ueyama | 2ab5f73 | 2016-08-12 03:33:04 +0000 | [diff] [blame] | 220 | if (auto *Cmd = dyn_cast<SymbolAssignment>(Base1.get())) { |
| 221 | if (shouldDefine<ELFT>(Cmd)) |
| 222 | addRegular<ELFT>(Cmd); |
| 223 | continue; |
| 224 | } |
| 225 | |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 226 | if (auto *Cmd = dyn_cast<OutputSectionCommand>(Base1.get())) { |
Rafael Espindola | 7bd3787 | 2016-09-12 16:05:16 +0000 | [diff] [blame^] | 227 | std::vector<InputSectionBase<ELFT> *> V = createInputSectionList(*Cmd); |
| 228 | |
Rui Ueyama | 48c3f1c | 2016-08-12 00:27:23 +0000 | [diff] [blame] | 229 | if (Cmd->Name == "/DISCARD/") { |
Rafael Espindola | 7bd3787 | 2016-09-12 16:05:16 +0000 | [diff] [blame^] | 230 | discard(V); |
Rui Ueyama | 48c3f1c | 2016-08-12 00:27:23 +0000 | [diff] [blame] | 231 | continue; |
| 232 | } |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 233 | |
Eugene Leviant | 97403d1 | 2016-09-01 09:55:57 +0000 | [diff] [blame] | 234 | if (V.empty()) |
Rui Ueyama | 0b9ce6a | 2016-08-12 03:16:56 +0000 | [diff] [blame] | 235 | continue; |
| 236 | |
George Rimar | db24d9c | 2016-08-19 15:18:23 +0000 | [diff] [blame] | 237 | for (InputSectionBase<ELFT> *Sec : V) { |
George Rimar | a14b13d | 2016-09-07 10:46:07 +0000 | [diff] [blame] | 238 | OutputSectionBase<ELFT> *OutSec; |
| 239 | bool IsNew; |
| 240 | std::tie(OutSec, IsNew) = Factory.create(Sec, Cmd->Name); |
| 241 | if (IsNew) |
| 242 | OutputSections->push_back(OutSec); |
| 243 | |
| 244 | uint32_t Subalign = Cmd->SubalignExpr ? Cmd->SubalignExpr(0) : 0; |
| 245 | |
George Rimar | db24d9c | 2016-08-19 15:18:23 +0000 | [diff] [blame] | 246 | if (Subalign) |
| 247 | Sec->Alignment = Subalign; |
Eugene Leviant | 97403d1 | 2016-09-01 09:55:57 +0000 | [diff] [blame] | 248 | OutSec->addSection(Sec); |
George Rimar | db24d9c | 2016-08-19 15:18:23 +0000 | [diff] [blame] | 249 | } |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 250 | } |
Rui Ueyama | 48c3f1c | 2016-08-12 00:27:23 +0000 | [diff] [blame] | 251 | } |
Eugene Leviant | e63d81b | 2016-07-20 14:43:20 +0000 | [diff] [blame] | 252 | |
Rui Ueyama | 0b9ce6a | 2016-08-12 03:16:56 +0000 | [diff] [blame] | 253 | // Add orphan sections. |
Rui Ueyama | 6b27481 | 2016-07-25 22:51:07 +0000 | [diff] [blame] | 254 | for (const std::unique_ptr<ObjectFile<ELFT>> &F : |
Rui Ueyama | 0b9ce6a | 2016-08-12 03:16:56 +0000 | [diff] [blame] | 255 | Symtab<ELFT>::X->getObjectFiles()) { |
| 256 | for (InputSectionBase<ELFT> *S : F->getSections()) { |
Rui Ueyama | 2ab5f73 | 2016-08-12 03:33:04 +0000 | [diff] [blame] | 257 | if (isDiscarded(S) || S->OutSec) |
| 258 | continue; |
| 259 | OutputSectionBase<ELFT> *OutSec; |
| 260 | bool IsNew; |
| 261 | std::tie(OutSec, IsNew) = Factory.create(S, getOutputSectionName(S)); |
| 262 | if (IsNew) |
| 263 | OutputSections->push_back(OutSec); |
| 264 | OutSec->addSection(S); |
Rui Ueyama | 0b9ce6a | 2016-08-12 03:16:56 +0000 | [diff] [blame] | 265 | } |
| 266 | } |
Eugene Leviant | e63d81b | 2016-07-20 14:43:20 +0000 | [diff] [blame] | 267 | } |
| 268 | |
Eugene Leviant | db741e7 | 2016-09-07 07:08:43 +0000 | [diff] [blame] | 269 | // Sets value of a section-defined symbol. Two kinds of |
| 270 | // symbols are processed: synthetic symbols, whose value |
| 271 | // is an offset from beginning of section and regular |
| 272 | // symbols whose value is absolute. |
| 273 | template <class ELFT> |
| 274 | static void assignSectionSymbol(SymbolAssignment *Cmd, |
| 275 | OutputSectionBase<ELFT> *Sec, |
| 276 | typename ELFT::uint Off) { |
| 277 | if (!Cmd->Sym) |
| 278 | return; |
| 279 | |
| 280 | if (auto *Body = dyn_cast<DefinedSynthetic<ELFT>>(Cmd->Sym)) { |
| 281 | Body->Section = Sec; |
| 282 | Body->Value = Cmd->Expression(Sec->getVA() + Off) - Sec->getVA(); |
| 283 | return; |
| 284 | } |
| 285 | auto *Body = cast<DefinedRegular<ELFT>>(Cmd->Sym); |
| 286 | Body->Value = Cmd->Expression(Sec->getVA() + Off); |
| 287 | } |
| 288 | |
Eugene Leviant | 20889c5 | 2016-08-31 08:13:33 +0000 | [diff] [blame] | 289 | // Linker script may define start and end symbols for special section types, |
| 290 | // like .got, .eh_frame_hdr, .eh_frame and others. Those sections are not a list |
| 291 | // of regular input input sections, therefore our way of defining symbols for |
| 292 | // regular sections will not work. The approach we use for special section types |
| 293 | // is not perfect - it handles only start and end symbols. |
| 294 | template <class ELFT> |
| 295 | void addStartEndSymbols(OutputSectionCommand *Cmd, |
| 296 | OutputSectionBase<ELFT> *Sec) { |
| 297 | bool Start = true; |
| 298 | BaseCommand *PrevCmd = nullptr; |
| 299 | |
| 300 | for (std::unique_ptr<BaseCommand> &Base : Cmd->Commands) { |
| 301 | if (auto *AssignCmd = dyn_cast<SymbolAssignment>(Base.get())) { |
Eugene Leviant | db741e7 | 2016-09-07 07:08:43 +0000 | [diff] [blame] | 302 | assignSectionSymbol<ELFT>(AssignCmd, Sec, Start ? 0 : Sec->getSize()); |
Eugene Leviant | 20889c5 | 2016-08-31 08:13:33 +0000 | [diff] [blame] | 303 | } else { |
| 304 | if (!Start && isa<SymbolAssignment>(PrevCmd)) |
| 305 | error("section '" + Sec->getName() + |
| 306 | "' supports only start and end symbols"); |
| 307 | Start = false; |
| 308 | } |
| 309 | PrevCmd = Base.get(); |
| 310 | } |
| 311 | } |
| 312 | |
| 313 | template <class ELFT> |
| 314 | void assignOffsets(OutputSectionCommand *Cmd, OutputSectionBase<ELFT> *Sec) { |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 315 | auto *OutSec = dyn_cast<OutputSection<ELFT>>(Sec); |
Rui Ueyama | 2de509c | 2016-08-12 00:55:08 +0000 | [diff] [blame] | 316 | if (!OutSec) { |
| 317 | Sec->assignOffsets(); |
Eugene Leviant | 20889c5 | 2016-08-31 08:13:33 +0000 | [diff] [blame] | 318 | // This section is not regular output section. However linker script may |
| 319 | // have defined start/end symbols for it. This case is handled below. |
| 320 | addStartEndSymbols(Cmd, Sec); |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 321 | return; |
Rui Ueyama | 2de509c | 2016-08-12 00:55:08 +0000 | [diff] [blame] | 322 | } |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 323 | typedef typename ELFT::uint uintX_t; |
| 324 | uintX_t Off = 0; |
Eugene Leviant | 97403d1 | 2016-09-01 09:55:57 +0000 | [diff] [blame] | 325 | auto ItCmd = Cmd->Commands.begin(); |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 326 | |
Eugene Leviant | 97403d1 | 2016-09-01 09:55:57 +0000 | [diff] [blame] | 327 | // Assigns values to all symbols following the given |
| 328 | // input section 'D' in output section 'Sec'. When symbols |
| 329 | // are in the beginning of output section the value of 'D' |
| 330 | // is nullptr. |
| 331 | auto AssignSuccessors = [&](InputSectionData *D) { |
| 332 | for (; ItCmd != Cmd->Commands.end(); ++ItCmd) { |
| 333 | auto *AssignCmd = dyn_cast<SymbolAssignment>(ItCmd->get()); |
| 334 | if (!AssignCmd) |
| 335 | continue; |
| 336 | if (D != AssignCmd->GoesAfter) |
| 337 | break; |
| 338 | |
Eugene Leviant | 97403d1 | 2016-09-01 09:55:57 +0000 | [diff] [blame] | 339 | if (AssignCmd->Name == ".") { |
| 340 | // Update to location counter means update to section size. |
Eugene Leviant | db741e7 | 2016-09-07 07:08:43 +0000 | [diff] [blame] | 341 | Off = AssignCmd->Expression(Sec->getVA() + Off) - Sec->getVA(); |
Eugene Leviant | 97403d1 | 2016-09-01 09:55:57 +0000 | [diff] [blame] | 342 | Sec->setSize(Off); |
| 343 | continue; |
| 344 | } |
Eugene Leviant | db741e7 | 2016-09-07 07:08:43 +0000 | [diff] [blame] | 345 | assignSectionSymbol<ELFT>(AssignCmd, Sec, Off); |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 346 | } |
Eugene Leviant | 97403d1 | 2016-09-01 09:55:57 +0000 | [diff] [blame] | 347 | }; |
| 348 | |
| 349 | AssignSuccessors(nullptr); |
| 350 | for (InputSection<ELFT> *I : OutSec->Sections) { |
| 351 | Off = alignTo(Off, I->Alignment); |
| 352 | I->OutSecOff = Off; |
| 353 | Off += I->getSize(); |
Rui Ueyama | f4a30a5 | 2016-08-11 21:30:42 +0000 | [diff] [blame] | 354 | // Update section size inside for-loop, so that SIZEOF |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 355 | // works correctly in the case below: |
| 356 | // .foo { *(.aaa) a = SIZEOF(.foo); *(.bbb) } |
| 357 | Sec->setSize(Off); |
Eugene Leviant | 97403d1 | 2016-09-01 09:55:57 +0000 | [diff] [blame] | 358 | // Add symbols following current input section. |
| 359 | AssignSuccessors(I); |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 360 | } |
| 361 | } |
| 362 | |
George Rimar | 8f66df9 | 2016-08-12 20:38:20 +0000 | [diff] [blame] | 363 | template <class ELFT> |
George Rimar | a14b13d | 2016-09-07 10:46:07 +0000 | [diff] [blame] | 364 | static std::vector<OutputSectionBase<ELFT> *> |
| 365 | findSections(OutputSectionCommand &Cmd, |
| 366 | ArrayRef<OutputSectionBase<ELFT> *> Sections) { |
| 367 | std::vector<OutputSectionBase<ELFT> *> Ret; |
| 368 | for (OutputSectionBase<ELFT> *Sec : Sections) |
| 369 | if (Sec->getName() == Cmd.Name && |
| 370 | checkConstraint(Sec->getFlags(), Cmd.Constraint)) |
| 371 | Ret.push_back(Sec); |
| 372 | return Ret; |
George Rimar | 8f66df9 | 2016-08-12 20:38:20 +0000 | [diff] [blame] | 373 | } |
| 374 | |
Rafael Espindola | a4b41dc | 2016-08-04 12:13:05 +0000 | [diff] [blame] | 375 | template <class ELFT> void LinkerScript<ELFT>::assignAddresses() { |
George Rimar | 652852c | 2016-04-16 10:10:32 +0000 | [diff] [blame] | 376 | // Orphan sections are sections present in the input files which |
Rui Ueyama | 7c18c28 | 2016-04-18 21:00:40 +0000 | [diff] [blame] | 377 | // are not explicitly placed into the output file by the linker script. |
| 378 | // We place orphan sections at end of file. |
| 379 | // Other linkers places them using some heuristics as described in |
George Rimar | 652852c | 2016-04-16 10:10:32 +0000 | [diff] [blame] | 380 | // https://sourceware.org/binutils/docs/ld/Orphan-Sections.html#Orphan-Sections. |
Rui Ueyama | e5cc668 | 2016-08-12 00:36:56 +0000 | [diff] [blame] | 381 | for (OutputSectionBase<ELFT> *Sec : *OutputSections) { |
George Rimar | 652852c | 2016-04-16 10:10:32 +0000 | [diff] [blame] | 382 | StringRef Name = Sec->getName(); |
Rui Ueyama | c3e2a4b | 2016-04-21 20:30:00 +0000 | [diff] [blame] | 383 | if (getSectionIndex(Name) == INT_MAX) |
George Rimar | 076fe15 | 2016-07-21 06:43:01 +0000 | [diff] [blame] | 384 | Opt.Commands.push_back(llvm::make_unique<OutputSectionCommand>(Name)); |
George Rimar | 652852c | 2016-04-16 10:10:32 +0000 | [diff] [blame] | 385 | } |
George Rimar | 652852c | 2016-04-16 10:10:32 +0000 | [diff] [blame] | 386 | |
Rui Ueyama | 7c18c28 | 2016-04-18 21:00:40 +0000 | [diff] [blame] | 387 | // Assign addresses as instructed by linker script SECTIONS sub-commands. |
Rui Ueyama | 4f7500b | 2016-08-12 04:00:22 +0000 | [diff] [blame] | 388 | Dot = getHeaderSize(); |
Eugene Leviant | 467c4d5 | 2016-07-01 10:27:36 +0000 | [diff] [blame] | 389 | uintX_t MinVA = std::numeric_limits<uintX_t>::max(); |
George Rimar | 652852c | 2016-04-16 10:10:32 +0000 | [diff] [blame] | 390 | uintX_t ThreadBssOffset = 0; |
George Rimar | 652852c | 2016-04-16 10:10:32 +0000 | [diff] [blame] | 391 | |
George Rimar | 076fe15 | 2016-07-21 06:43:01 +0000 | [diff] [blame] | 392 | for (const std::unique_ptr<BaseCommand> &Base : Opt.Commands) { |
| 393 | if (auto *Cmd = dyn_cast<SymbolAssignment>(Base.get())) { |
Rui Ueyama | 8d083e6 | 2016-07-29 05:48:39 +0000 | [diff] [blame] | 394 | if (Cmd->Name == ".") { |
| 395 | Dot = Cmd->Expression(Dot); |
| 396 | } else if (Cmd->Sym) { |
| 397 | cast<DefinedRegular<ELFT>>(Cmd->Sym)->Value = Cmd->Expression(Dot); |
| 398 | } |
George Rimar | 652852c | 2016-04-16 10:10:32 +0000 | [diff] [blame] | 399 | continue; |
| 400 | } |
| 401 | |
George Rimar | eefa758 | 2016-08-04 09:29:31 +0000 | [diff] [blame] | 402 | if (auto *Cmd = dyn_cast<AssertCommand>(Base.get())) { |
| 403 | Cmd->Expression(Dot); |
| 404 | continue; |
| 405 | } |
| 406 | |
George Rimar | 076fe15 | 2016-07-21 06:43:01 +0000 | [diff] [blame] | 407 | auto *Cmd = cast<OutputSectionCommand>(Base.get()); |
George Rimar | a14b13d | 2016-09-07 10:46:07 +0000 | [diff] [blame] | 408 | for (OutputSectionBase<ELFT> *Sec : |
| 409 | findSections<ELFT>(*Cmd, *OutputSections)) { |
George Rimar | 652852c | 2016-04-16 10:10:32 +0000 | [diff] [blame] | 410 | |
George Rimar | a14b13d | 2016-09-07 10:46:07 +0000 | [diff] [blame] | 411 | if (Cmd->AddrExpr) |
| 412 | Dot = Cmd->AddrExpr(Dot); |
George Rimar | 58e5c4d | 2016-07-25 08:29:46 +0000 | [diff] [blame] | 413 | |
George Rimar | a14b13d | 2016-09-07 10:46:07 +0000 | [diff] [blame] | 414 | if (Cmd->AlignExpr) |
| 415 | Sec->updateAlignment(Cmd->AlignExpr(Dot)); |
George Rimar | 630c617 | 2016-07-26 18:06:29 +0000 | [diff] [blame] | 416 | |
George Rimar | a14b13d | 2016-09-07 10:46:07 +0000 | [diff] [blame] | 417 | if ((Sec->getFlags() & SHF_TLS) && Sec->getType() == SHT_NOBITS) { |
| 418 | uintX_t TVA = Dot + ThreadBssOffset; |
| 419 | TVA = alignTo(TVA, Sec->getAlignment()); |
| 420 | Sec->setVA(TVA); |
| 421 | assignOffsets(Cmd, Sec); |
| 422 | ThreadBssOffset = TVA - Dot + Sec->getSize(); |
| 423 | continue; |
| 424 | } |
| 425 | |
| 426 | if (!(Sec->getFlags() & SHF_ALLOC)) { |
| 427 | assignOffsets(Cmd, Sec); |
| 428 | continue; |
| 429 | } |
| 430 | |
| 431 | Dot = alignTo(Dot, Sec->getAlignment()); |
| 432 | Sec->setVA(Dot); |
Eugene Leviant | 20889c5 | 2016-08-31 08:13:33 +0000 | [diff] [blame] | 433 | assignOffsets(Cmd, Sec); |
George Rimar | a14b13d | 2016-09-07 10:46:07 +0000 | [diff] [blame] | 434 | MinVA = std::min(MinVA, Dot); |
| 435 | Dot += Sec->getSize(); |
George Rimar | 652852c | 2016-04-16 10:10:32 +0000 | [diff] [blame] | 436 | } |
| 437 | } |
Rui Ueyama | 52c4e17 | 2016-07-01 10:42:25 +0000 | [diff] [blame] | 438 | |
Rafael Espindola | 64c32d6 | 2016-07-07 14:28:47 +0000 | [diff] [blame] | 439 | // ELF and Program headers need to be right before the first section in |
George Rimar | b91e711 | 2016-07-19 07:42:07 +0000 | [diff] [blame] | 440 | // memory. Set their addresses accordingly. |
Eugene Leviant | 467c4d5 | 2016-07-01 10:27:36 +0000 | [diff] [blame] | 441 | MinVA = alignDown(MinVA - Out<ELFT>::ElfHeader->getSize() - |
| 442 | Out<ELFT>::ProgramHeaders->getSize(), |
| 443 | Target->PageSize); |
| 444 | Out<ELFT>::ElfHeader->setVA(MinVA); |
| 445 | Out<ELFT>::ProgramHeaders->setVA(Out<ELFT>::ElfHeader->getSize() + MinVA); |
George Rimar | 652852c | 2016-04-16 10:10:32 +0000 | [diff] [blame] | 446 | } |
| 447 | |
Rui Ueyama | 464daad | 2016-08-22 04:55:20 +0000 | [diff] [blame] | 448 | // Creates program headers as instructed by PHDRS linker script command. |
Rui Ueyama | 07320e4 | 2016-04-20 20:13:41 +0000 | [diff] [blame] | 449 | template <class ELFT> |
Rafael Espindola | a4b41dc | 2016-08-04 12:13:05 +0000 | [diff] [blame] | 450 | std::vector<PhdrEntry<ELFT>> LinkerScript<ELFT>::createPhdrs() { |
Rui Ueyama | edebbdf | 2016-07-24 23:47:31 +0000 | [diff] [blame] | 451 | std::vector<PhdrEntry<ELFT>> Ret; |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 452 | |
Rui Ueyama | 464daad | 2016-08-22 04:55:20 +0000 | [diff] [blame] | 453 | // Process PHDRS and FILEHDR keywords because they are not |
| 454 | // real output sections and cannot be added in the following loop. |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 455 | for (const PhdrsCommand &Cmd : Opt.PhdrsCommands) { |
Rui Ueyama | edebbdf | 2016-07-24 23:47:31 +0000 | [diff] [blame] | 456 | Ret.emplace_back(Cmd.Type, Cmd.Flags == UINT_MAX ? PF_R : Cmd.Flags); |
| 457 | PhdrEntry<ELFT> &Phdr = Ret.back(); |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 458 | |
| 459 | if (Cmd.HasFilehdr) |
Rui Ueyama | adca245 | 2016-07-23 14:18:48 +0000 | [diff] [blame] | 460 | Phdr.add(Out<ELFT>::ElfHeader); |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 461 | if (Cmd.HasPhdrs) |
Rui Ueyama | adca245 | 2016-07-23 14:18:48 +0000 | [diff] [blame] | 462 | Phdr.add(Out<ELFT>::ProgramHeaders); |
Eugene Leviant | 56b21c8 | 2016-09-09 09:46:16 +0000 | [diff] [blame] | 463 | |
| 464 | if (Cmd.LMAExpr) { |
| 465 | Phdr.H.p_paddr = Cmd.LMAExpr(0); |
| 466 | Phdr.HasLMA = true; |
| 467 | } |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 468 | } |
| 469 | |
Rui Ueyama | 464daad | 2016-08-22 04:55:20 +0000 | [diff] [blame] | 470 | // Add output sections to program headers. |
Rui Ueyama | edebbdf | 2016-07-24 23:47:31 +0000 | [diff] [blame] | 471 | PhdrEntry<ELFT> *Load = nullptr; |
| 472 | uintX_t Flags = PF_R; |
Rui Ueyama | 464daad | 2016-08-22 04:55:20 +0000 | [diff] [blame] | 473 | for (OutputSectionBase<ELFT> *Sec : *OutputSections) { |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 474 | if (!(Sec->getFlags() & SHF_ALLOC)) |
| 475 | break; |
| 476 | |
Rui Ueyama | edebbdf | 2016-07-24 23:47:31 +0000 | [diff] [blame] | 477 | std::vector<size_t> PhdrIds = getPhdrIndices(Sec->getName()); |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 478 | if (!PhdrIds.empty()) { |
| 479 | // Assign headers specified by linker script |
| 480 | for (size_t Id : PhdrIds) { |
Rui Ueyama | edebbdf | 2016-07-24 23:47:31 +0000 | [diff] [blame] | 481 | Ret[Id].add(Sec); |
Eugene Leviant | 865bf86 | 2016-07-21 10:43:25 +0000 | [diff] [blame] | 482 | if (Opt.PhdrsCommands[Id].Flags == UINT_MAX) |
Rafael Espindola | 0b11367 | 2016-07-27 14:10:56 +0000 | [diff] [blame] | 483 | Ret[Id].H.p_flags |= Sec->getPhdrFlags(); |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 484 | } |
| 485 | } else { |
| 486 | // If we have no load segment or flags've changed then we want new load |
| 487 | // segment. |
Rafael Espindola | 0b11367 | 2016-07-27 14:10:56 +0000 | [diff] [blame] | 488 | uintX_t NewFlags = Sec->getPhdrFlags(); |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 489 | if (Load == nullptr || Flags != NewFlags) { |
Rui Ueyama | edebbdf | 2016-07-24 23:47:31 +0000 | [diff] [blame] | 490 | Load = &*Ret.emplace(Ret.end(), PT_LOAD, NewFlags); |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 491 | Flags = NewFlags; |
| 492 | } |
Rui Ueyama | 18f084f | 2016-07-20 19:36:41 +0000 | [diff] [blame] | 493 | Load->add(Sec); |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 494 | } |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 495 | } |
Rui Ueyama | edebbdf | 2016-07-24 23:47:31 +0000 | [diff] [blame] | 496 | return Ret; |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 497 | } |
| 498 | |
Eugene Leviant | f9bc3bd | 2016-08-16 06:40:58 +0000 | [diff] [blame] | 499 | template <class ELFT> bool LinkerScript<ELFT>::ignoreInterpSection() { |
| 500 | // Ignore .interp section in case we have PHDRS specification |
| 501 | // and PT_INTERP isn't listed. |
| 502 | return !Opt.PhdrsCommands.empty() && |
| 503 | llvm::find_if(Opt.PhdrsCommands, [](const PhdrsCommand &Cmd) { |
| 504 | return Cmd.Type == PT_INTERP; |
| 505 | }) == Opt.PhdrsCommands.end(); |
| 506 | } |
| 507 | |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 508 | template <class ELFT> |
Rui Ueyama | 07320e4 | 2016-04-20 20:13:41 +0000 | [diff] [blame] | 509 | ArrayRef<uint8_t> LinkerScript<ELFT>::getFiller(StringRef Name) { |
George Rimar | f6c3cce | 2016-07-21 07:48:54 +0000 | [diff] [blame] | 510 | for (const std::unique_ptr<BaseCommand> &Base : Opt.Commands) |
| 511 | if (auto *Cmd = dyn_cast<OutputSectionCommand>(Base.get())) |
| 512 | if (Cmd->Name == Name) |
| 513 | return Cmd->Filler; |
| 514 | return {}; |
George Rimar | e2ee72b | 2016-02-26 14:48:31 +0000 | [diff] [blame] | 515 | } |
| 516 | |
George Rimar | 206fffa | 2016-08-17 08:16:57 +0000 | [diff] [blame] | 517 | template <class ELFT> Expr LinkerScript<ELFT>::getLma(StringRef Name) { |
George Rimar | 8ceadb3 | 2016-08-17 07:44:19 +0000 | [diff] [blame] | 518 | for (const std::unique_ptr<BaseCommand> &Base : Opt.Commands) |
| 519 | if (auto *Cmd = dyn_cast<OutputSectionCommand>(Base.get())) |
| 520 | if (Cmd->LmaExpr && Cmd->Name == Name) |
| 521 | return Cmd->LmaExpr; |
| 522 | return {}; |
| 523 | } |
| 524 | |
Rui Ueyama | c3e2a4b | 2016-04-21 20:30:00 +0000 | [diff] [blame] | 525 | // Returns the index of the given section name in linker script |
| 526 | // SECTIONS commands. Sections are laid out as the same order as they |
| 527 | // were in the script. If a given name did not appear in the script, |
| 528 | // it returns INT_MAX, so that it will be laid out at end of file. |
George Rimar | 076fe15 | 2016-07-21 06:43:01 +0000 | [diff] [blame] | 529 | template <class ELFT> int LinkerScript<ELFT>::getSectionIndex(StringRef Name) { |
Rui Ueyama | f510fa6 | 2016-07-26 00:21:15 +0000 | [diff] [blame] | 530 | int I = 0; |
| 531 | for (std::unique_ptr<BaseCommand> &Base : Opt.Commands) { |
| 532 | if (auto *Cmd = dyn_cast<OutputSectionCommand>(Base.get())) |
| 533 | if (Cmd->Name == Name) |
| 534 | return I; |
| 535 | ++I; |
| 536 | } |
| 537 | return INT_MAX; |
George Rimar | 71b26e9 | 2016-04-21 10:22:02 +0000 | [diff] [blame] | 538 | } |
| 539 | |
| 540 | // A compartor to sort output sections. Returns -1 or 1 if |
| 541 | // A or B are mentioned in linker script. Otherwise, returns 0. |
Rui Ueyama | 07320e4 | 2016-04-20 20:13:41 +0000 | [diff] [blame] | 542 | template <class ELFT> |
| 543 | int LinkerScript<ELFT>::compareSections(StringRef A, StringRef B) { |
Rui Ueyama | c3e2a4b | 2016-04-21 20:30:00 +0000 | [diff] [blame] | 544 | int I = getSectionIndex(A); |
| 545 | int J = getSectionIndex(B); |
| 546 | if (I == INT_MAX && J == INT_MAX) |
Rui Ueyama | 717677a | 2016-02-11 21:17:59 +0000 | [diff] [blame] | 547 | return 0; |
| 548 | return I < J ? -1 : 1; |
| 549 | } |
| 550 | |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 551 | template <class ELFT> bool LinkerScript<ELFT>::hasPhdrsCommands() { |
| 552 | return !Opt.PhdrsCommands.empty(); |
| 553 | } |
| 554 | |
George Rimar | 9e69450 | 2016-07-29 16:18:47 +0000 | [diff] [blame] | 555 | template <class ELFT> |
George Rimar | 884e786 | 2016-09-08 08:19:13 +0000 | [diff] [blame] | 556 | uint64_t LinkerScript<ELFT>::getOutputSectionAddress(StringRef Name) { |
George Rimar | 96659df | 2016-08-30 09:54:01 +0000 | [diff] [blame] | 557 | for (OutputSectionBase<ELFT> *Sec : *OutputSections) |
| 558 | if (Sec->getName() == Name) |
| 559 | return Sec->getVA(); |
| 560 | error("undefined section " + Name); |
| 561 | return 0; |
| 562 | } |
| 563 | |
| 564 | template <class ELFT> |
George Rimar | 884e786 | 2016-09-08 08:19:13 +0000 | [diff] [blame] | 565 | uint64_t LinkerScript<ELFT>::getOutputSectionSize(StringRef Name) { |
George Rimar | 9e69450 | 2016-07-29 16:18:47 +0000 | [diff] [blame] | 566 | for (OutputSectionBase<ELFT> *Sec : *OutputSections) |
| 567 | if (Sec->getName() == Name) |
| 568 | return Sec->getSize(); |
| 569 | error("undefined section " + Name); |
| 570 | return 0; |
| 571 | } |
| 572 | |
Eugene Leviant | 36fac7f | 2016-09-08 09:08:30 +0000 | [diff] [blame] | 573 | template <class ELFT> |
| 574 | uint64_t LinkerScript<ELFT>::getOutputSectionAlign(StringRef Name) { |
| 575 | for (OutputSectionBase<ELFT> *Sec : *OutputSections) |
| 576 | if (Sec->getName() == Name) |
| 577 | return Sec->getAlignment(); |
| 578 | error("undefined section " + Name); |
| 579 | return 0; |
| 580 | } |
| 581 | |
George Rimar | 884e786 | 2016-09-08 08:19:13 +0000 | [diff] [blame] | 582 | template <class ELFT> uint64_t LinkerScript<ELFT>::getHeaderSize() { |
George Rimar | e32a359 | 2016-08-10 07:59:34 +0000 | [diff] [blame] | 583 | return Out<ELFT>::ElfHeader->getSize() + Out<ELFT>::ProgramHeaders->getSize(); |
| 584 | } |
| 585 | |
George Rimar | 884e786 | 2016-09-08 08:19:13 +0000 | [diff] [blame] | 586 | template <class ELFT> uint64_t LinkerScript<ELFT>::getSymbolValue(StringRef S) { |
| 587 | if (SymbolBody *B = Symtab<ELFT>::X->find(S)) |
| 588 | return B->getVA<ELFT>(); |
| 589 | error("symbol not found: " + S); |
| 590 | return 0; |
| 591 | } |
| 592 | |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 593 | // Returns indices of ELF headers containing specific section, identified |
| 594 | // by Name. Each index is a zero based number of ELF header listed within |
| 595 | // PHDRS {} script block. |
| 596 | template <class ELFT> |
Rui Ueyama | edebbdf | 2016-07-24 23:47:31 +0000 | [diff] [blame] | 597 | std::vector<size_t> LinkerScript<ELFT>::getPhdrIndices(StringRef SectionName) { |
George Rimar | 076fe15 | 2016-07-21 06:43:01 +0000 | [diff] [blame] | 598 | for (const std::unique_ptr<BaseCommand> &Base : Opt.Commands) { |
| 599 | auto *Cmd = dyn_cast<OutputSectionCommand>(Base.get()); |
Rui Ueyama | edebbdf | 2016-07-24 23:47:31 +0000 | [diff] [blame] | 600 | if (!Cmd || Cmd->Name != SectionName) |
George Rimar | 31d842f | 2016-07-20 16:43:03 +0000 | [diff] [blame] | 601 | continue; |
| 602 | |
Rui Ueyama | 29c5a2a | 2016-07-26 00:27:36 +0000 | [diff] [blame] | 603 | std::vector<size_t> Ret; |
| 604 | for (StringRef PhdrName : Cmd->Phdrs) |
| 605 | Ret.push_back(getPhdrIndex(PhdrName)); |
| 606 | return Ret; |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 607 | } |
George Rimar | 31d842f | 2016-07-20 16:43:03 +0000 | [diff] [blame] | 608 | return {}; |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 609 | } |
| 610 | |
Rui Ueyama | 29c5a2a | 2016-07-26 00:27:36 +0000 | [diff] [blame] | 611 | template <class ELFT> |
| 612 | size_t LinkerScript<ELFT>::getPhdrIndex(StringRef PhdrName) { |
| 613 | size_t I = 0; |
| 614 | for (PhdrsCommand &Cmd : Opt.PhdrsCommands) { |
| 615 | if (Cmd.Name == PhdrName) |
| 616 | return I; |
| 617 | ++I; |
| 618 | } |
| 619 | error("section header '" + PhdrName + "' is not listed in PHDRS"); |
| 620 | return 0; |
| 621 | } |
| 622 | |
Rui Ueyama | 07320e4 | 2016-04-20 20:13:41 +0000 | [diff] [blame] | 623 | class elf::ScriptParser : public ScriptParserBase { |
George Rimar | c3794e5 | 2016-02-24 09:21:47 +0000 | [diff] [blame] | 624 | typedef void (ScriptParser::*Handler)(); |
| 625 | |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 626 | public: |
Rui Ueyama | 07320e4 | 2016-04-20 20:13:41 +0000 | [diff] [blame] | 627 | ScriptParser(StringRef S, bool B) : ScriptParserBase(S), IsUnderSysroot(B) {} |
George Rimar | f23b232 | 2016-02-19 10:45:45 +0000 | [diff] [blame] | 628 | |
George Rimar | 20b6598 | 2016-08-31 09:08:26 +0000 | [diff] [blame] | 629 | void readLinkerScript(); |
| 630 | void readVersionScript(); |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 631 | |
| 632 | private: |
Rui Ueyama | 52a1509 | 2015-10-11 03:28:42 +0000 | [diff] [blame] | 633 | void addFile(StringRef Path); |
| 634 | |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 635 | void readAsNeeded(); |
Denis Protivensky | 90c5099 | 2015-10-08 06:48:38 +0000 | [diff] [blame] | 636 | void readEntry(); |
George Rimar | 83f406c | 2015-10-19 17:35:12 +0000 | [diff] [blame] | 637 | void readExtern(); |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 638 | void readGroup(); |
Rui Ueyama | 31aa1f8 | 2015-10-11 01:31:55 +0000 | [diff] [blame] | 639 | void readInclude(); |
Rui Ueyama | ee59282 | 2015-10-07 00:25:09 +0000 | [diff] [blame] | 640 | void readOutput(); |
Davide Italiano | 9159ce9 | 2015-10-12 21:50:08 +0000 | [diff] [blame] | 641 | void readOutputArch(); |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 642 | void readOutputFormat(); |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 643 | void readPhdrs(); |
Davide Italiano | 68a39a6 | 2015-10-08 17:51:41 +0000 | [diff] [blame] | 644 | void readSearchDir(); |
Denis Protivensky | 8e3b38a | 2015-11-12 09:52:08 +0000 | [diff] [blame] | 645 | void readSections(); |
Rui Ueyama | 95769b4 | 2016-08-31 20:03:54 +0000 | [diff] [blame] | 646 | void readVersion(); |
| 647 | void readVersionScriptCommand(); |
Denis Protivensky | 8e3b38a | 2015-11-12 09:52:08 +0000 | [diff] [blame] | 648 | |
Rui Ueyama | 113cdec | 2016-07-24 23:05:57 +0000 | [diff] [blame] | 649 | SymbolAssignment *readAssignment(StringRef Name); |
George Rimar | ff1f29e | 2016-09-06 13:51:57 +0000 | [diff] [blame] | 650 | std::vector<uint8_t> readFill(); |
Rui Ueyama | 1041656 | 2016-08-04 02:03:27 +0000 | [diff] [blame] | 651 | OutputSectionCommand *readOutputSectionDescription(StringRef OutSec); |
George Rimar | ff1f29e | 2016-09-06 13:51:57 +0000 | [diff] [blame] | 652 | std::vector<uint8_t> readOutputSectionFiller(StringRef Tok); |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 653 | std::vector<StringRef> readOutputSectionPhdrs(); |
George Rimar | a2496cb | 2016-08-30 09:46:59 +0000 | [diff] [blame] | 654 | InputSectionDescription *readInputSectionDescription(StringRef Tok); |
George Rimar | c91930a | 2016-09-02 21:17:20 +0000 | [diff] [blame] | 655 | Regex readFilePatterns(); |
George Rimar | a2496cb | 2016-08-30 09:46:59 +0000 | [diff] [blame] | 656 | InputSectionDescription *readInputSectionRules(StringRef FilePattern); |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 657 | unsigned readPhdrType(); |
Rui Ueyama | 742c383 | 2016-08-04 22:27:00 +0000 | [diff] [blame] | 658 | SortKind readSortKind(); |
Petr Hosek | a35e39c | 2016-08-16 01:11:16 +0000 | [diff] [blame] | 659 | SymbolAssignment *readProvideHidden(bool Provide, bool Hidden); |
Eugene Leviant | db741e7 | 2016-09-07 07:08:43 +0000 | [diff] [blame] | 660 | SymbolAssignment *readProvideOrAssignment(StringRef Tok, bool MakeAbsolute); |
George Rimar | 03fc010 | 2016-07-28 07:18:23 +0000 | [diff] [blame] | 661 | void readSort(); |
George Rimar | eefa758 | 2016-08-04 09:29:31 +0000 | [diff] [blame] | 662 | Expr readAssert(); |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 663 | |
| 664 | Expr readExpr(); |
| 665 | Expr readExpr1(Expr Lhs, int MinPrec); |
| 666 | Expr readPrimary(); |
| 667 | Expr readTernary(Expr Cond); |
Rui Ueyama | 6ad7dfc | 2016-08-17 18:59:16 +0000 | [diff] [blame] | 668 | Expr readParenExpr(); |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 669 | |
George Rimar | 20b6598 | 2016-08-31 09:08:26 +0000 | [diff] [blame] | 670 | // For parsing version script. |
| 671 | void readExtern(std::vector<SymbolVersion> *Globals); |
Rui Ueyama | 95769b4 | 2016-08-31 20:03:54 +0000 | [diff] [blame] | 672 | void readVersionDeclaration(StringRef VerStr); |
George Rimar | 20b6598 | 2016-08-31 09:08:26 +0000 | [diff] [blame] | 673 | void readGlobal(StringRef VerStr); |
| 674 | void readLocal(); |
| 675 | |
Rui Ueyama | 07320e4 | 2016-04-20 20:13:41 +0000 | [diff] [blame] | 676 | ScriptConfiguration &Opt = *ScriptConfig; |
| 677 | StringSaver Saver = {ScriptConfig->Alloc}; |
Simon Atanasyan | 16b0cc9 | 2015-11-26 05:53:00 +0000 | [diff] [blame] | 678 | bool IsUnderSysroot; |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 679 | }; |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 680 | |
George Rimar | 20b6598 | 2016-08-31 09:08:26 +0000 | [diff] [blame] | 681 | void ScriptParser::readVersionScript() { |
Rui Ueyama | 95769b4 | 2016-08-31 20:03:54 +0000 | [diff] [blame] | 682 | readVersionScriptCommand(); |
| 683 | if (!atEOF()) |
| 684 | setError("EOF expected, but got " + next()); |
| 685 | } |
| 686 | |
| 687 | void ScriptParser::readVersionScriptCommand() { |
George Rimar | 20b6598 | 2016-08-31 09:08:26 +0000 | [diff] [blame] | 688 | if (skip("{")) { |
Rui Ueyama | 95769b4 | 2016-08-31 20:03:54 +0000 | [diff] [blame] | 689 | readVersionDeclaration(""); |
George Rimar | 20b6598 | 2016-08-31 09:08:26 +0000 | [diff] [blame] | 690 | return; |
| 691 | } |
| 692 | |
Rui Ueyama | 95769b4 | 2016-08-31 20:03:54 +0000 | [diff] [blame] | 693 | while (!atEOF() && !Error && peek() != "}") { |
George Rimar | 20b6598 | 2016-08-31 09:08:26 +0000 | [diff] [blame] | 694 | StringRef VerStr = next(); |
| 695 | if (VerStr == "{") { |
Rui Ueyama | 95769b4 | 2016-08-31 20:03:54 +0000 | [diff] [blame] | 696 | setError("anonymous version definition is used in " |
| 697 | "combination with other version definitions"); |
George Rimar | 20b6598 | 2016-08-31 09:08:26 +0000 | [diff] [blame] | 698 | return; |
| 699 | } |
| 700 | expect("{"); |
Rui Ueyama | 95769b4 | 2016-08-31 20:03:54 +0000 | [diff] [blame] | 701 | readVersionDeclaration(VerStr); |
George Rimar | 20b6598 | 2016-08-31 09:08:26 +0000 | [diff] [blame] | 702 | } |
| 703 | } |
| 704 | |
Rui Ueyama | 95769b4 | 2016-08-31 20:03:54 +0000 | [diff] [blame] | 705 | void ScriptParser::readVersion() { |
| 706 | expect("{"); |
| 707 | readVersionScriptCommand(); |
| 708 | expect("}"); |
| 709 | } |
| 710 | |
George Rimar | 20b6598 | 2016-08-31 09:08:26 +0000 | [diff] [blame] | 711 | void ScriptParser::readLinkerScript() { |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 712 | while (!atEOF()) { |
| 713 | StringRef Tok = next(); |
Rui Ueyama | a27eecc | 2016-09-02 18:52:41 +0000 | [diff] [blame] | 714 | if (Tok == ";") |
| 715 | continue; |
| 716 | |
| 717 | if (Tok == "ENTRY") { |
| 718 | readEntry(); |
| 719 | } else if (Tok == "EXTERN") { |
| 720 | readExtern(); |
| 721 | } else if (Tok == "GROUP" || Tok == "INPUT") { |
| 722 | readGroup(); |
| 723 | } else if (Tok == "INCLUDE") { |
| 724 | readInclude(); |
| 725 | } else if (Tok == "OUTPUT") { |
| 726 | readOutput(); |
| 727 | } else if (Tok == "OUTPUT_ARCH") { |
| 728 | readOutputArch(); |
| 729 | } else if (Tok == "OUTPUT_FORMAT") { |
| 730 | readOutputFormat(); |
| 731 | } else if (Tok == "PHDRS") { |
| 732 | readPhdrs(); |
| 733 | } else if (Tok == "SEARCH_DIR") { |
| 734 | readSearchDir(); |
| 735 | } else if (Tok == "SECTIONS") { |
| 736 | readSections(); |
| 737 | } else if (Tok == "VERSION") { |
| 738 | readVersion(); |
Eugene Leviant | db741e7 | 2016-09-07 07:08:43 +0000 | [diff] [blame] | 739 | } else if (SymbolAssignment *Cmd = readProvideOrAssignment(Tok, true)) { |
Petr Hosek | e5d3ca5 | 2016-08-31 15:31:17 +0000 | [diff] [blame] | 740 | if (Opt.HasContents) |
| 741 | Opt.Commands.emplace_back(Cmd); |
| 742 | else |
| 743 | Opt.Assignments.emplace_back(Cmd); |
| 744 | } else { |
George Rimar | 5761042 | 2016-03-11 14:43:02 +0000 | [diff] [blame] | 745 | setError("unknown directive: " + Tok); |
Petr Hosek | e5d3ca5 | 2016-08-31 15:31:17 +0000 | [diff] [blame] | 746 | } |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 747 | } |
| 748 | } |
| 749 | |
Rui Ueyama | 717677a | 2016-02-11 21:17:59 +0000 | [diff] [blame] | 750 | void ScriptParser::addFile(StringRef S) { |
Simon Atanasyan | 16b0cc9 | 2015-11-26 05:53:00 +0000 | [diff] [blame] | 751 | if (IsUnderSysroot && S.startswith("/")) { |
| 752 | SmallString<128> Path; |
| 753 | (Config->Sysroot + S).toStringRef(Path); |
| 754 | if (sys::fs::exists(Path)) { |
| 755 | Driver->addFile(Saver.save(Path.str())); |
| 756 | return; |
| 757 | } |
| 758 | } |
| 759 | |
Rui Ueyama | f03f3cc | 2015-10-13 00:09:21 +0000 | [diff] [blame] | 760 | if (sys::path::is_absolute(S)) { |
Rui Ueyama | 52a1509 | 2015-10-11 03:28:42 +0000 | [diff] [blame] | 761 | Driver->addFile(S); |
| 762 | } else if (S.startswith("=")) { |
| 763 | if (Config->Sysroot.empty()) |
| 764 | Driver->addFile(S.substr(1)); |
| 765 | else |
| 766 | Driver->addFile(Saver.save(Config->Sysroot + "/" + S.substr(1))); |
| 767 | } else if (S.startswith("-l")) { |
Rui Ueyama | 21eecb4 | 2016-02-02 21:13:09 +0000 | [diff] [blame] | 768 | Driver->addLibrary(S.substr(2)); |
Simon Atanasyan | a1b8fc3 | 2015-11-26 20:23:46 +0000 | [diff] [blame] | 769 | } else if (sys::fs::exists(S)) { |
| 770 | Driver->addFile(S); |
Rui Ueyama | 52a1509 | 2015-10-11 03:28:42 +0000 | [diff] [blame] | 771 | } else { |
| 772 | std::string Path = findFromSearchPaths(S); |
| 773 | if (Path.empty()) |
George Rimar | 777f963 | 2016-03-12 08:31:34 +0000 | [diff] [blame] | 774 | setError("unable to find " + S); |
Rui Ueyama | 025d59b | 2016-02-02 20:27:59 +0000 | [diff] [blame] | 775 | else |
| 776 | Driver->addFile(Saver.save(Path)); |
Rui Ueyama | 52a1509 | 2015-10-11 03:28:42 +0000 | [diff] [blame] | 777 | } |
| 778 | } |
| 779 | |
Rui Ueyama | 717677a | 2016-02-11 21:17:59 +0000 | [diff] [blame] | 780 | void ScriptParser::readAsNeeded() { |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 781 | expect("("); |
Rui Ueyama | 35da9b6 | 2015-10-11 20:59:12 +0000 | [diff] [blame] | 782 | bool Orig = Config->AsNeeded; |
| 783 | Config->AsNeeded = true; |
Rui Ueyama | a2acc93 | 2016-08-05 01:25:45 +0000 | [diff] [blame] | 784 | while (!Error && !skip(")")) |
George Rimar | cd574a5 | 2016-09-09 14:35:36 +0000 | [diff] [blame] | 785 | addFile(unquote(next())); |
Rui Ueyama | 35da9b6 | 2015-10-11 20:59:12 +0000 | [diff] [blame] | 786 | Config->AsNeeded = Orig; |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 787 | } |
| 788 | |
Rui Ueyama | 717677a | 2016-02-11 21:17:59 +0000 | [diff] [blame] | 789 | void ScriptParser::readEntry() { |
Denis Protivensky | 90c5099 | 2015-10-08 06:48:38 +0000 | [diff] [blame] | 790 | // -e <symbol> takes predecence over ENTRY(<symbol>). |
| 791 | expect("("); |
| 792 | StringRef Tok = next(); |
| 793 | if (Config->Entry.empty()) |
| 794 | Config->Entry = Tok; |
| 795 | expect(")"); |
| 796 | } |
| 797 | |
Rui Ueyama | 717677a | 2016-02-11 21:17:59 +0000 | [diff] [blame] | 798 | void ScriptParser::readExtern() { |
George Rimar | 83f406c | 2015-10-19 17:35:12 +0000 | [diff] [blame] | 799 | expect("("); |
Rui Ueyama | a2acc93 | 2016-08-05 01:25:45 +0000 | [diff] [blame] | 800 | while (!Error && !skip(")")) |
| 801 | Config->Undefined.push_back(next()); |
George Rimar | 83f406c | 2015-10-19 17:35:12 +0000 | [diff] [blame] | 802 | } |
| 803 | |
Rui Ueyama | 717677a | 2016-02-11 21:17:59 +0000 | [diff] [blame] | 804 | void ScriptParser::readGroup() { |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 805 | expect("("); |
Rui Ueyama | a2acc93 | 2016-08-05 01:25:45 +0000 | [diff] [blame] | 806 | while (!Error && !skip(")")) { |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 807 | StringRef Tok = next(); |
Rui Ueyama | a2acc93 | 2016-08-05 01:25:45 +0000 | [diff] [blame] | 808 | if (Tok == "AS_NEEDED") |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 809 | readAsNeeded(); |
Rui Ueyama | a2acc93 | 2016-08-05 01:25:45 +0000 | [diff] [blame] | 810 | else |
George Rimar | cd574a5 | 2016-09-09 14:35:36 +0000 | [diff] [blame] | 811 | addFile(unquote(Tok)); |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 812 | } |
| 813 | } |
| 814 | |
Rui Ueyama | 717677a | 2016-02-11 21:17:59 +0000 | [diff] [blame] | 815 | void ScriptParser::readInclude() { |
Rui Ueyama | 31aa1f8 | 2015-10-11 01:31:55 +0000 | [diff] [blame] | 816 | StringRef Tok = next(); |
George Rimar | cd574a5 | 2016-09-09 14:35:36 +0000 | [diff] [blame] | 817 | auto MBOrErr = MemoryBuffer::getFile(unquote(Tok)); |
Rui Ueyama | 025d59b | 2016-02-02 20:27:59 +0000 | [diff] [blame] | 818 | if (!MBOrErr) { |
George Rimar | 5761042 | 2016-03-11 14:43:02 +0000 | [diff] [blame] | 819 | setError("cannot open " + Tok); |
Rui Ueyama | 025d59b | 2016-02-02 20:27:59 +0000 | [diff] [blame] | 820 | return; |
| 821 | } |
Rui Ueyama | 31aa1f8 | 2015-10-11 01:31:55 +0000 | [diff] [blame] | 822 | std::unique_ptr<MemoryBuffer> &MB = *MBOrErr; |
Rui Ueyama | a47ee68 | 2015-10-11 01:53:04 +0000 | [diff] [blame] | 823 | StringRef S = Saver.save(MB->getMemBufferRef().getBuffer()); |
| 824 | std::vector<StringRef> V = tokenize(S); |
Rui Ueyama | 31aa1f8 | 2015-10-11 01:31:55 +0000 | [diff] [blame] | 825 | Tokens.insert(Tokens.begin() + Pos, V.begin(), V.end()); |
Rui Ueyama | 31aa1f8 | 2015-10-11 01:31:55 +0000 | [diff] [blame] | 826 | } |
| 827 | |
Rui Ueyama | 717677a | 2016-02-11 21:17:59 +0000 | [diff] [blame] | 828 | void ScriptParser::readOutput() { |
Rui Ueyama | ee59282 | 2015-10-07 00:25:09 +0000 | [diff] [blame] | 829 | // -o <file> takes predecence over OUTPUT(<file>). |
| 830 | expect("("); |
| 831 | StringRef Tok = next(); |
| 832 | if (Config->OutputFile.empty()) |
George Rimar | cd574a5 | 2016-09-09 14:35:36 +0000 | [diff] [blame] | 833 | Config->OutputFile = unquote(Tok); |
Rui Ueyama | ee59282 | 2015-10-07 00:25:09 +0000 | [diff] [blame] | 834 | expect(")"); |
| 835 | } |
| 836 | |
Rui Ueyama | 717677a | 2016-02-11 21:17:59 +0000 | [diff] [blame] | 837 | void ScriptParser::readOutputArch() { |
Davide Italiano | 9159ce9 | 2015-10-12 21:50:08 +0000 | [diff] [blame] | 838 | // Error checking only for now. |
| 839 | expect("("); |
| 840 | next(); |
| 841 | expect(")"); |
| 842 | } |
| 843 | |
Rui Ueyama | 717677a | 2016-02-11 21:17:59 +0000 | [diff] [blame] | 844 | void ScriptParser::readOutputFormat() { |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 845 | // Error checking only for now. |
| 846 | expect("("); |
| 847 | next(); |
Davide Italiano | 6836c61 | 2015-10-12 21:08:41 +0000 | [diff] [blame] | 848 | StringRef Tok = next(); |
| 849 | if (Tok == ")") |
George Rimar | 6c55f0e | 2016-09-08 08:20:30 +0000 | [diff] [blame] | 850 | return; |
Rui Ueyama | 025d59b | 2016-02-02 20:27:59 +0000 | [diff] [blame] | 851 | if (Tok != ",") { |
George Rimar | 5761042 | 2016-03-11 14:43:02 +0000 | [diff] [blame] | 852 | setError("unexpected token: " + Tok); |
Rui Ueyama | 025d59b | 2016-02-02 20:27:59 +0000 | [diff] [blame] | 853 | return; |
| 854 | } |
Davide Italiano | 6836c61 | 2015-10-12 21:08:41 +0000 | [diff] [blame] | 855 | next(); |
| 856 | expect(","); |
| 857 | next(); |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 858 | expect(")"); |
| 859 | } |
| 860 | |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 861 | void ScriptParser::readPhdrs() { |
| 862 | expect("{"); |
| 863 | while (!Error && !skip("}")) { |
| 864 | StringRef Tok = next(); |
Eugene Leviant | 56b21c8 | 2016-09-09 09:46:16 +0000 | [diff] [blame] | 865 | Opt.PhdrsCommands.push_back( |
| 866 | {Tok, PT_NULL, false, false, UINT_MAX, nullptr}); |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 867 | PhdrsCommand &PhdrCmd = Opt.PhdrsCommands.back(); |
| 868 | |
| 869 | PhdrCmd.Type = readPhdrType(); |
| 870 | do { |
| 871 | Tok = next(); |
| 872 | if (Tok == ";") |
| 873 | break; |
| 874 | if (Tok == "FILEHDR") |
| 875 | PhdrCmd.HasFilehdr = true; |
| 876 | else if (Tok == "PHDRS") |
| 877 | PhdrCmd.HasPhdrs = true; |
Eugene Leviant | 56b21c8 | 2016-09-09 09:46:16 +0000 | [diff] [blame] | 878 | else if (Tok == "AT") |
| 879 | PhdrCmd.LMAExpr = readParenExpr(); |
Eugene Leviant | 865bf86 | 2016-07-21 10:43:25 +0000 | [diff] [blame] | 880 | else if (Tok == "FLAGS") { |
| 881 | expect("("); |
Rafael Espindola | eb685cd | 2016-08-02 22:14:57 +0000 | [diff] [blame] | 882 | // Passing 0 for the value of dot is a bit of a hack. It means that |
| 883 | // we accept expressions like ".|1". |
| 884 | PhdrCmd.Flags = readExpr()(0); |
Eugene Leviant | 865bf86 | 2016-07-21 10:43:25 +0000 | [diff] [blame] | 885 | expect(")"); |
| 886 | } else |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 887 | setError("unexpected header attribute: " + Tok); |
| 888 | } while (!Error); |
| 889 | } |
| 890 | } |
| 891 | |
Rui Ueyama | 717677a | 2016-02-11 21:17:59 +0000 | [diff] [blame] | 892 | void ScriptParser::readSearchDir() { |
Davide Italiano | 68a39a6 | 2015-10-08 17:51:41 +0000 | [diff] [blame] | 893 | expect("("); |
Rui Ueyama | 86c5fb8 | 2016-09-08 23:26:54 +0000 | [diff] [blame] | 894 | StringRef Tok = next(); |
Rui Ueyama | 6c7ad13 | 2016-09-02 19:20:33 +0000 | [diff] [blame] | 895 | if (!Config->Nostdlib) |
George Rimar | cd574a5 | 2016-09-09 14:35:36 +0000 | [diff] [blame] | 896 | Config->SearchPaths.push_back(unquote(Tok)); |
Davide Italiano | 68a39a6 | 2015-10-08 17:51:41 +0000 | [diff] [blame] | 897 | expect(")"); |
| 898 | } |
| 899 | |
Rui Ueyama | 717677a | 2016-02-11 21:17:59 +0000 | [diff] [blame] | 900 | void ScriptParser::readSections() { |
Rui Ueyama | 3de0a33 | 2016-07-29 03:31:09 +0000 | [diff] [blame] | 901 | Opt.HasContents = true; |
Denis Protivensky | 8e3b38a | 2015-11-12 09:52:08 +0000 | [diff] [blame] | 902 | expect("{"); |
George Rimar | 652852c | 2016-04-16 10:10:32 +0000 | [diff] [blame] | 903 | while (!Error && !skip("}")) { |
Rui Ueyama | 113cdec | 2016-07-24 23:05:57 +0000 | [diff] [blame] | 904 | StringRef Tok = next(); |
Eugene Leviant | db741e7 | 2016-09-07 07:08:43 +0000 | [diff] [blame] | 905 | BaseCommand *Cmd = readProvideOrAssignment(Tok, true); |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 906 | if (!Cmd) { |
| 907 | if (Tok == "ASSERT") |
| 908 | Cmd = new AssertCommand(readAssert()); |
| 909 | else |
| 910 | Cmd = readOutputSectionDescription(Tok); |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 911 | } |
Rui Ueyama | 1041656 | 2016-08-04 02:03:27 +0000 | [diff] [blame] | 912 | Opt.Commands.emplace_back(Cmd); |
George Rimar | 652852c | 2016-04-16 10:10:32 +0000 | [diff] [blame] | 913 | } |
Denis Protivensky | 8e3b38a | 2015-11-12 09:52:08 +0000 | [diff] [blame] | 914 | } |
| 915 | |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 916 | static int precedence(StringRef Op) { |
| 917 | return StringSwitch<int>(Op) |
| 918 | .Case("*", 4) |
| 919 | .Case("/", 4) |
| 920 | .Case("+", 3) |
| 921 | .Case("-", 3) |
| 922 | .Case("<", 2) |
| 923 | .Case(">", 2) |
| 924 | .Case(">=", 2) |
| 925 | .Case("<=", 2) |
| 926 | .Case("==", 2) |
| 927 | .Case("!=", 2) |
| 928 | .Case("&", 1) |
Rafael Espindola | cc3dd62 | 2016-08-22 21:33:35 +0000 | [diff] [blame] | 929 | .Case("|", 1) |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 930 | .Default(-1); |
| 931 | } |
| 932 | |
George Rimar | c91930a | 2016-09-02 21:17:20 +0000 | [diff] [blame] | 933 | Regex ScriptParser::readFilePatterns() { |
Rui Ueyama | 1041656 | 2016-08-04 02:03:27 +0000 | [diff] [blame] | 934 | std::vector<StringRef> V; |
| 935 | while (!Error && !skip(")")) |
| 936 | V.push_back(next()); |
George Rimar | c91930a | 2016-09-02 21:17:20 +0000 | [diff] [blame] | 937 | return compileGlobPatterns(V); |
George Rimar | 0702c4e | 2016-07-29 15:32:46 +0000 | [diff] [blame] | 938 | } |
| 939 | |
Rui Ueyama | 742c383 | 2016-08-04 22:27:00 +0000 | [diff] [blame] | 940 | SortKind ScriptParser::readSortKind() { |
| 941 | if (skip("SORT") || skip("SORT_BY_NAME")) |
| 942 | return SortByName; |
| 943 | if (skip("SORT_BY_ALIGNMENT")) |
| 944 | return SortByAlignment; |
| 945 | return SortNone; |
| 946 | } |
| 947 | |
George Rimar | a2496cb | 2016-08-30 09:46:59 +0000 | [diff] [blame] | 948 | InputSectionDescription * |
| 949 | ScriptParser::readInputSectionRules(StringRef FilePattern) { |
George Rimar | c91930a | 2016-09-02 21:17:20 +0000 | [diff] [blame] | 950 | auto *Cmd = new InputSectionDescription(FilePattern); |
Davide Italiano | 0ed42b0 | 2016-07-25 21:47:13 +0000 | [diff] [blame] | 951 | expect("("); |
Davide Italiano | e728279 | 2016-07-27 01:44:01 +0000 | [diff] [blame] | 952 | |
Rui Ueyama | 742c383 | 2016-08-04 22:27:00 +0000 | [diff] [blame] | 953 | // Read EXCLUDE_FILE(). |
Davide Italiano | e728279 | 2016-07-27 01:44:01 +0000 | [diff] [blame] | 954 | if (skip("EXCLUDE_FILE")) { |
| 955 | expect("("); |
George Rimar | c91930a | 2016-09-02 21:17:20 +0000 | [diff] [blame] | 956 | Cmd->ExcludedFileRe = readFilePatterns(); |
Davide Italiano | 0ed42b0 | 2016-07-25 21:47:13 +0000 | [diff] [blame] | 957 | } |
George Rimar | 0659800 | 2016-07-28 21:51:30 +0000 | [diff] [blame] | 958 | |
Rui Ueyama | 742c383 | 2016-08-04 22:27:00 +0000 | [diff] [blame] | 959 | // Read SORT(). |
| 960 | if (SortKind K1 = readSortKind()) { |
| 961 | Cmd->SortOuter = K1; |
George Rimar | 0702c4e | 2016-07-29 15:32:46 +0000 | [diff] [blame] | 962 | expect("("); |
Rui Ueyama | 742c383 | 2016-08-04 22:27:00 +0000 | [diff] [blame] | 963 | if (SortKind K2 = readSortKind()) { |
| 964 | Cmd->SortInner = K2; |
George Rimar | 350ece4 | 2016-08-03 08:35:59 +0000 | [diff] [blame] | 965 | expect("("); |
George Rimar | c91930a | 2016-09-02 21:17:20 +0000 | [diff] [blame] | 966 | Cmd->SectionRe = readFilePatterns(); |
George Rimar | 350ece4 | 2016-08-03 08:35:59 +0000 | [diff] [blame] | 967 | expect(")"); |
| 968 | } else { |
George Rimar | c91930a | 2016-09-02 21:17:20 +0000 | [diff] [blame] | 969 | Cmd->SectionRe = readFilePatterns(); |
George Rimar | 350ece4 | 2016-08-03 08:35:59 +0000 | [diff] [blame] | 970 | } |
George Rimar | 0702c4e | 2016-07-29 15:32:46 +0000 | [diff] [blame] | 971 | expect(")"); |
Rui Ueyama | 1041656 | 2016-08-04 02:03:27 +0000 | [diff] [blame] | 972 | return Cmd; |
George Rimar | 0659800 | 2016-07-28 21:51:30 +0000 | [diff] [blame] | 973 | } |
George Rimar | 0702c4e | 2016-07-29 15:32:46 +0000 | [diff] [blame] | 974 | |
George Rimar | c91930a | 2016-09-02 21:17:20 +0000 | [diff] [blame] | 975 | Cmd->SectionRe = readFilePatterns(); |
Rui Ueyama | 1041656 | 2016-08-04 02:03:27 +0000 | [diff] [blame] | 976 | return Cmd; |
Davide Italiano | e728279 | 2016-07-27 01:44:01 +0000 | [diff] [blame] | 977 | } |
| 978 | |
George Rimar | a2496cb | 2016-08-30 09:46:59 +0000 | [diff] [blame] | 979 | InputSectionDescription * |
| 980 | ScriptParser::readInputSectionDescription(StringRef Tok) { |
George Rimar | 0659800 | 2016-07-28 21:51:30 +0000 | [diff] [blame] | 981 | // Input section wildcard can be surrounded by KEEP. |
| 982 | // https://sourceware.org/binutils/docs/ld/Input-Section-Keep.html#Input-Section-Keep |
George Rimar | a2496cb | 2016-08-30 09:46:59 +0000 | [diff] [blame] | 983 | if (Tok == "KEEP") { |
George Rimar | 0659800 | 2016-07-28 21:51:30 +0000 | [diff] [blame] | 984 | expect("("); |
George Rimar | a2496cb | 2016-08-30 09:46:59 +0000 | [diff] [blame] | 985 | StringRef FilePattern = next(); |
| 986 | InputSectionDescription *Cmd = readInputSectionRules(FilePattern); |
George Rimar | 0659800 | 2016-07-28 21:51:30 +0000 | [diff] [blame] | 987 | expect(")"); |
George Rimar | c91930a | 2016-09-02 21:17:20 +0000 | [diff] [blame] | 988 | Opt.KeptSections.push_back(&Cmd->SectionRe); |
Rui Ueyama | 1041656 | 2016-08-04 02:03:27 +0000 | [diff] [blame] | 989 | return Cmd; |
George Rimar | 0659800 | 2016-07-28 21:51:30 +0000 | [diff] [blame] | 990 | } |
George Rimar | a2496cb | 2016-08-30 09:46:59 +0000 | [diff] [blame] | 991 | return readInputSectionRules(Tok); |
Davide Italiano | 0ed42b0 | 2016-07-25 21:47:13 +0000 | [diff] [blame] | 992 | } |
| 993 | |
George Rimar | 03fc010 | 2016-07-28 07:18:23 +0000 | [diff] [blame] | 994 | void ScriptParser::readSort() { |
| 995 | expect("("); |
| 996 | expect("CONSTRUCTORS"); |
| 997 | expect(")"); |
| 998 | } |
| 999 | |
George Rimar | eefa758 | 2016-08-04 09:29:31 +0000 | [diff] [blame] | 1000 | Expr ScriptParser::readAssert() { |
| 1001 | expect("("); |
| 1002 | Expr E = readExpr(); |
| 1003 | expect(","); |
George Rimar | cd574a5 | 2016-09-09 14:35:36 +0000 | [diff] [blame] | 1004 | StringRef Msg = unquote(next()); |
George Rimar | eefa758 | 2016-08-04 09:29:31 +0000 | [diff] [blame] | 1005 | expect(")"); |
| 1006 | return [=](uint64_t Dot) { |
| 1007 | uint64_t V = E(Dot); |
| 1008 | if (!V) |
| 1009 | error(Msg); |
| 1010 | return V; |
| 1011 | }; |
| 1012 | } |
| 1013 | |
Rui Ueyama | 25150e8 | 2016-09-06 17:46:43 +0000 | [diff] [blame] | 1014 | // Reads a FILL(expr) command. We handle the FILL command as an |
| 1015 | // alias for =fillexp section attribute, which is different from |
| 1016 | // what GNU linkers do. |
| 1017 | // https://sourceware.org/binutils/docs/ld/Output-Section-Data.html |
George Rimar | ff1f29e | 2016-09-06 13:51:57 +0000 | [diff] [blame] | 1018 | std::vector<uint8_t> ScriptParser::readFill() { |
| 1019 | expect("("); |
| 1020 | std::vector<uint8_t> V = readOutputSectionFiller(next()); |
| 1021 | expect(")"); |
| 1022 | expect(";"); |
| 1023 | return V; |
| 1024 | } |
| 1025 | |
Rui Ueyama | 1041656 | 2016-08-04 02:03:27 +0000 | [diff] [blame] | 1026 | OutputSectionCommand * |
| 1027 | ScriptParser::readOutputSectionDescription(StringRef OutSec) { |
George Rimar | 076fe15 | 2016-07-21 06:43:01 +0000 | [diff] [blame] | 1028 | OutputSectionCommand *Cmd = new OutputSectionCommand(OutSec); |
George Rimar | 58e5c4d | 2016-07-25 08:29:46 +0000 | [diff] [blame] | 1029 | |
| 1030 | // Read an address expression. |
| 1031 | // https://sourceware.org/binutils/docs/ld/Output-Section-Address.html#Output-Section-Address |
| 1032 | if (peek() != ":") |
| 1033 | Cmd->AddrExpr = readExpr(); |
| 1034 | |
Denis Protivensky | 8e3b38a | 2015-11-12 09:52:08 +0000 | [diff] [blame] | 1035 | expect(":"); |
Davide Italiano | 246f681 | 2016-07-22 03:36:24 +0000 | [diff] [blame] | 1036 | |
George Rimar | 8ceadb3 | 2016-08-17 07:44:19 +0000 | [diff] [blame] | 1037 | if (skip("AT")) |
Rui Ueyama | 6ad7dfc | 2016-08-17 18:59:16 +0000 | [diff] [blame] | 1038 | Cmd->LmaExpr = readParenExpr(); |
George Rimar | 630c617 | 2016-07-26 18:06:29 +0000 | [diff] [blame] | 1039 | if (skip("ALIGN")) |
Rui Ueyama | 6ad7dfc | 2016-08-17 18:59:16 +0000 | [diff] [blame] | 1040 | Cmd->AlignExpr = readParenExpr(); |
George Rimar | db24d9c | 2016-08-19 15:18:23 +0000 | [diff] [blame] | 1041 | if (skip("SUBALIGN")) |
| 1042 | Cmd->SubalignExpr = readParenExpr(); |
George Rimar | 630c617 | 2016-07-26 18:06:29 +0000 | [diff] [blame] | 1043 | |
Davide Italiano | 246f681 | 2016-07-22 03:36:24 +0000 | [diff] [blame] | 1044 | // Parse constraints. |
| 1045 | if (skip("ONLY_IF_RO")) |
Rui Ueyama | efc4066 | 2016-07-25 22:00:10 +0000 | [diff] [blame] | 1046 | Cmd->Constraint = ConstraintKind::ReadOnly; |
Davide Italiano | 246f681 | 2016-07-22 03:36:24 +0000 | [diff] [blame] | 1047 | if (skip("ONLY_IF_RW")) |
Rui Ueyama | efc4066 | 2016-07-25 22:00:10 +0000 | [diff] [blame] | 1048 | Cmd->Constraint = ConstraintKind::ReadWrite; |
Denis Protivensky | 8e3b38a | 2015-11-12 09:52:08 +0000 | [diff] [blame] | 1049 | expect("{"); |
Rui Ueyama | 8ec77e6 | 2016-04-21 22:00:51 +0000 | [diff] [blame] | 1050 | |
Rui Ueyama | 025d59b | 2016-02-02 20:27:59 +0000 | [diff] [blame] | 1051 | while (!Error && !skip("}")) { |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 1052 | StringRef Tok = next(); |
Eugene Leviant | db741e7 | 2016-09-07 07:08:43 +0000 | [diff] [blame] | 1053 | if (SymbolAssignment *Assignment = readProvideOrAssignment(Tok, false)) |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 1054 | Cmd->Commands.emplace_back(Assignment); |
George Rimar | ff1f29e | 2016-09-06 13:51:57 +0000 | [diff] [blame] | 1055 | else if (Tok == "FILL") |
| 1056 | Cmd->Filler = readFill(); |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 1057 | else if (Tok == "SORT") |
George Rimar | 03fc010 | 2016-07-28 07:18:23 +0000 | [diff] [blame] | 1058 | readSort(); |
George Rimar | a2496cb | 2016-08-30 09:46:59 +0000 | [diff] [blame] | 1059 | else if (peek() == "(") |
| 1060 | Cmd->Commands.emplace_back(readInputSectionDescription(Tok)); |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 1061 | else |
| 1062 | setError("unknown command " + Tok); |
Denis Protivensky | 8e3b38a | 2015-11-12 09:52:08 +0000 | [diff] [blame] | 1063 | } |
George Rimar | 076fe15 | 2016-07-21 06:43:01 +0000 | [diff] [blame] | 1064 | Cmd->Phdrs = readOutputSectionPhdrs(); |
George Rimar | ff1f29e | 2016-09-06 13:51:57 +0000 | [diff] [blame] | 1065 | if (peek().startswith("=")) |
| 1066 | Cmd->Filler = readOutputSectionFiller(next().drop_front()); |
Rui Ueyama | 1041656 | 2016-08-04 02:03:27 +0000 | [diff] [blame] | 1067 | return Cmd; |
Rui Ueyama | f71caa2 | 2016-07-29 06:14:07 +0000 | [diff] [blame] | 1068 | } |
Rui Ueyama | 8ec77e6 | 2016-04-21 22:00:51 +0000 | [diff] [blame] | 1069 | |
Rui Ueyama | 2c8f1f0 | 2016-08-29 22:01:21 +0000 | [diff] [blame] | 1070 | // Read "=<number>" where <number> is an octal/decimal/hexadecimal number. |
| 1071 | // https://sourceware.org/binutils/docs/ld/Output-Section-Fill.html |
| 1072 | // |
| 1073 | // ld.gold is not fully compatible with ld.bfd. ld.bfd handles |
| 1074 | // hexstrings as blobs of arbitrary sizes, while ld.gold handles them |
| 1075 | // as 32-bit big-endian values. We will do the same as ld.gold does |
| 1076 | // because it's simpler than what ld.bfd does. |
George Rimar | ff1f29e | 2016-09-06 13:51:57 +0000 | [diff] [blame] | 1077 | std::vector<uint8_t> ScriptParser::readOutputSectionFiller(StringRef Tok) { |
Rui Ueyama | 965827d | 2016-08-03 23:25:15 +0000 | [diff] [blame] | 1078 | uint32_t V; |
George Rimar | ff1f29e | 2016-09-06 13:51:57 +0000 | [diff] [blame] | 1079 | if (Tok.getAsInteger(0, V)) { |
Rui Ueyama | 965827d | 2016-08-03 23:25:15 +0000 | [diff] [blame] | 1080 | setError("invalid filler expression: " + Tok); |
Rui Ueyama | f71caa2 | 2016-07-29 06:14:07 +0000 | [diff] [blame] | 1081 | return {}; |
George Rimar | e2ee72b | 2016-02-26 14:48:31 +0000 | [diff] [blame] | 1082 | } |
Rui Ueyama | 2c8f1f0 | 2016-08-29 22:01:21 +0000 | [diff] [blame] | 1083 | return {uint8_t(V >> 24), uint8_t(V >> 16), uint8_t(V >> 8), uint8_t(V)}; |
Denis Protivensky | 8e3b38a | 2015-11-12 09:52:08 +0000 | [diff] [blame] | 1084 | } |
| 1085 | |
Petr Hosek | a35e39c | 2016-08-16 01:11:16 +0000 | [diff] [blame] | 1086 | SymbolAssignment *ScriptParser::readProvideHidden(bool Provide, bool Hidden) { |
Eugene Leviant | a31c91b | 2016-07-22 07:38:40 +0000 | [diff] [blame] | 1087 | expect("("); |
Rui Ueyama | 174e0a1 | 2016-07-29 00:29:25 +0000 | [diff] [blame] | 1088 | SymbolAssignment *Cmd = readAssignment(next()); |
Petr Hosek | a35e39c | 2016-08-16 01:11:16 +0000 | [diff] [blame] | 1089 | Cmd->Provide = Provide; |
Rui Ueyama | 174e0a1 | 2016-07-29 00:29:25 +0000 | [diff] [blame] | 1090 | Cmd->Hidden = Hidden; |
Eugene Leviant | a31c91b | 2016-07-22 07:38:40 +0000 | [diff] [blame] | 1091 | expect(")"); |
| 1092 | expect(";"); |
Rui Ueyama | 1041656 | 2016-08-04 02:03:27 +0000 | [diff] [blame] | 1093 | return Cmd; |
Eugene Leviant | eda81a1 | 2016-07-12 06:39:48 +0000 | [diff] [blame] | 1094 | } |
| 1095 | |
Eugene Leviant | db741e7 | 2016-09-07 07:08:43 +0000 | [diff] [blame] | 1096 | SymbolAssignment *ScriptParser::readProvideOrAssignment(StringRef Tok, |
| 1097 | bool MakeAbsolute) { |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 1098 | SymbolAssignment *Cmd = nullptr; |
| 1099 | if (peek() == "=" || peek() == "+=") { |
| 1100 | Cmd = readAssignment(Tok); |
| 1101 | expect(";"); |
| 1102 | } else if (Tok == "PROVIDE") { |
Petr Hosek | a35e39c | 2016-08-16 01:11:16 +0000 | [diff] [blame] | 1103 | Cmd = readProvideHidden(true, false); |
| 1104 | } else if (Tok == "HIDDEN") { |
| 1105 | Cmd = readProvideHidden(false, true); |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 1106 | } else if (Tok == "PROVIDE_HIDDEN") { |
Petr Hosek | a35e39c | 2016-08-16 01:11:16 +0000 | [diff] [blame] | 1107 | Cmd = readProvideHidden(true, true); |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 1108 | } |
Eugene Leviant | db741e7 | 2016-09-07 07:08:43 +0000 | [diff] [blame] | 1109 | if (Cmd && MakeAbsolute) |
| 1110 | Cmd->IsAbsolute = true; |
Eugene Leviant | ceabe80 | 2016-08-11 07:56:43 +0000 | [diff] [blame] | 1111 | return Cmd; |
| 1112 | } |
| 1113 | |
George Rimar | 30835ea | 2016-07-28 21:08:56 +0000 | [diff] [blame] | 1114 | static uint64_t getSymbolValue(StringRef S, uint64_t Dot) { |
| 1115 | if (S == ".") |
| 1116 | return Dot; |
George Rimar | 884e786 | 2016-09-08 08:19:13 +0000 | [diff] [blame] | 1117 | return ScriptBase->getSymbolValue(S); |
George Rimar | e32a359 | 2016-08-10 07:59:34 +0000 | [diff] [blame] | 1118 | } |
| 1119 | |
George Rimar | 30835ea | 2016-07-28 21:08:56 +0000 | [diff] [blame] | 1120 | SymbolAssignment *ScriptParser::readAssignment(StringRef Name) { |
| 1121 | StringRef Op = next(); |
Eugene Leviant | db741e7 | 2016-09-07 07:08:43 +0000 | [diff] [blame] | 1122 | bool IsAbsolute = false; |
| 1123 | Expr E; |
George Rimar | 30835ea | 2016-07-28 21:08:56 +0000 | [diff] [blame] | 1124 | assert(Op == "=" || Op == "+="); |
Eugene Leviant | db741e7 | 2016-09-07 07:08:43 +0000 | [diff] [blame] | 1125 | if (skip("ABSOLUTE")) { |
| 1126 | E = readParenExpr(); |
| 1127 | IsAbsolute = true; |
| 1128 | } else { |
| 1129 | E = readExpr(); |
| 1130 | } |
George Rimar | 30835ea | 2016-07-28 21:08:56 +0000 | [diff] [blame] | 1131 | if (Op == "+=") |
| 1132 | E = [=](uint64_t Dot) { return getSymbolValue(Name, Dot) + E(Dot); }; |
Eugene Leviant | db741e7 | 2016-09-07 07:08:43 +0000 | [diff] [blame] | 1133 | return new SymbolAssignment(Name, E, IsAbsolute); |
George Rimar | 30835ea | 2016-07-28 21:08:56 +0000 | [diff] [blame] | 1134 | } |
| 1135 | |
| 1136 | // This is an operator-precedence parser to parse a linker |
| 1137 | // script expression. |
| 1138 | Expr ScriptParser::readExpr() { return readExpr1(readPrimary(), 0); } |
| 1139 | |
Rui Ueyama | 36c1cd2 | 2016-08-05 01:04:59 +0000 | [diff] [blame] | 1140 | static Expr combine(StringRef Op, Expr L, Expr R) { |
| 1141 | if (Op == "*") |
| 1142 | return [=](uint64_t Dot) { return L(Dot) * R(Dot); }; |
| 1143 | if (Op == "/") { |
| 1144 | return [=](uint64_t Dot) -> uint64_t { |
| 1145 | uint64_t RHS = R(Dot); |
| 1146 | if (RHS == 0) { |
| 1147 | error("division by zero"); |
| 1148 | return 0; |
| 1149 | } |
| 1150 | return L(Dot) / RHS; |
| 1151 | }; |
| 1152 | } |
| 1153 | if (Op == "+") |
| 1154 | return [=](uint64_t Dot) { return L(Dot) + R(Dot); }; |
| 1155 | if (Op == "-") |
| 1156 | return [=](uint64_t Dot) { return L(Dot) - R(Dot); }; |
| 1157 | if (Op == "<") |
| 1158 | return [=](uint64_t Dot) { return L(Dot) < R(Dot); }; |
| 1159 | if (Op == ">") |
| 1160 | return [=](uint64_t Dot) { return L(Dot) > R(Dot); }; |
| 1161 | if (Op == ">=") |
| 1162 | return [=](uint64_t Dot) { return L(Dot) >= R(Dot); }; |
| 1163 | if (Op == "<=") |
| 1164 | return [=](uint64_t Dot) { return L(Dot) <= R(Dot); }; |
| 1165 | if (Op == "==") |
| 1166 | return [=](uint64_t Dot) { return L(Dot) == R(Dot); }; |
| 1167 | if (Op == "!=") |
| 1168 | return [=](uint64_t Dot) { return L(Dot) != R(Dot); }; |
| 1169 | if (Op == "&") |
| 1170 | return [=](uint64_t Dot) { return L(Dot) & R(Dot); }; |
Rafael Espindola | cc3dd62 | 2016-08-22 21:33:35 +0000 | [diff] [blame] | 1171 | if (Op == "|") |
| 1172 | return [=](uint64_t Dot) { return L(Dot) | R(Dot); }; |
Rui Ueyama | 36c1cd2 | 2016-08-05 01:04:59 +0000 | [diff] [blame] | 1173 | llvm_unreachable("invalid operator"); |
| 1174 | } |
| 1175 | |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 1176 | // This is a part of the operator-precedence parser. This function |
| 1177 | // assumes that the remaining token stream starts with an operator. |
| 1178 | Expr ScriptParser::readExpr1(Expr Lhs, int MinPrec) { |
| 1179 | while (!atEOF() && !Error) { |
| 1180 | // Read an operator and an expression. |
| 1181 | StringRef Op1 = peek(); |
| 1182 | if (Op1 == "?") |
| 1183 | return readTernary(Lhs); |
| 1184 | if (precedence(Op1) < MinPrec) |
Eugene Leviant | eda81a1 | 2016-07-12 06:39:48 +0000 | [diff] [blame] | 1185 | break; |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 1186 | next(); |
| 1187 | Expr Rhs = readPrimary(); |
| 1188 | |
| 1189 | // Evaluate the remaining part of the expression first if the |
| 1190 | // next operator has greater precedence than the previous one. |
| 1191 | // For example, if we have read "+" and "3", and if the next |
| 1192 | // operator is "*", then we'll evaluate 3 * ... part first. |
| 1193 | while (!atEOF()) { |
| 1194 | StringRef Op2 = peek(); |
| 1195 | if (precedence(Op2) <= precedence(Op1)) |
| 1196 | break; |
| 1197 | Rhs = readExpr1(Rhs, precedence(Op2)); |
| 1198 | } |
| 1199 | |
| 1200 | Lhs = combine(Op1, Lhs, Rhs); |
Eugene Leviant | eda81a1 | 2016-07-12 06:39:48 +0000 | [diff] [blame] | 1201 | } |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 1202 | return Lhs; |
| 1203 | } |
| 1204 | |
| 1205 | uint64_t static getConstant(StringRef S) { |
Michael J. Spencer | e2cc07b | 2016-08-17 02:10:51 +0000 | [diff] [blame] | 1206 | if (S == "COMMONPAGESIZE") |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 1207 | return Target->PageSize; |
Michael J. Spencer | e2cc07b | 2016-08-17 02:10:51 +0000 | [diff] [blame] | 1208 | if (S == "MAXPAGESIZE") |
| 1209 | return Target->MaxPageSize; |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 1210 | error("unknown constant: " + S); |
| 1211 | return 0; |
| 1212 | } |
| 1213 | |
Rui Ueyama | 626e0b0 | 2016-09-02 18:19:00 +0000 | [diff] [blame] | 1214 | // Parses Tok as an integer. Returns true if successful. |
| 1215 | // It recognizes hexadecimal (prefixed with "0x" or suffixed with "H") |
| 1216 | // and decimal numbers. Decimal numbers may have "K" (kilo) or |
| 1217 | // "M" (mega) prefixes. |
George Rimar | 9f2f7ad | 2016-09-02 16:01:42 +0000 | [diff] [blame] | 1218 | static bool readInteger(StringRef Tok, uint64_t &Result) { |
Simon Atanasyan | eaeafb2 | 2016-09-02 21:54:35 +0000 | [diff] [blame] | 1219 | if (Tok.startswith("-")) { |
| 1220 | if (!readInteger(Tok.substr(1), Result)) |
| 1221 | return false; |
| 1222 | Result = -Result; |
| 1223 | return true; |
| 1224 | } |
George Rimar | 9f2f7ad | 2016-09-02 16:01:42 +0000 | [diff] [blame] | 1225 | if (Tok.startswith_lower("0x")) |
| 1226 | return !Tok.substr(2).getAsInteger(16, Result); |
| 1227 | if (Tok.endswith_lower("H")) |
| 1228 | return !Tok.drop_back().getAsInteger(16, Result); |
| 1229 | |
| 1230 | int Suffix = 1; |
| 1231 | if (Tok.endswith_lower("K")) { |
| 1232 | Suffix = 1024; |
| 1233 | Tok = Tok.drop_back(); |
| 1234 | } else if (Tok.endswith_lower("M")) { |
| 1235 | Suffix = 1024 * 1024; |
| 1236 | Tok = Tok.drop_back(); |
| 1237 | } |
| 1238 | if (Tok.getAsInteger(10, Result)) |
| 1239 | return false; |
| 1240 | Result *= Suffix; |
| 1241 | return true; |
| 1242 | } |
| 1243 | |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 1244 | Expr ScriptParser::readPrimary() { |
Rui Ueyama | 6ad7dfc | 2016-08-17 18:59:16 +0000 | [diff] [blame] | 1245 | if (peek() == "(") |
| 1246 | return readParenExpr(); |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 1247 | |
Rui Ueyama | 6ad7dfc | 2016-08-17 18:59:16 +0000 | [diff] [blame] | 1248 | StringRef Tok = next(); |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 1249 | |
Simon Atanasyan | eaeafb2 | 2016-09-02 21:54:35 +0000 | [diff] [blame] | 1250 | if (Tok == "~") { |
| 1251 | Expr E = readPrimary(); |
| 1252 | return [=](uint64_t Dot) { return ~E(Dot); }; |
| 1253 | } |
| 1254 | if (Tok == "-") { |
| 1255 | Expr E = readPrimary(); |
| 1256 | return [=](uint64_t Dot) { return -E(Dot); }; |
| 1257 | } |
| 1258 | |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 1259 | // Built-in functions are parsed here. |
| 1260 | // https://sourceware.org/binutils/docs/ld/Builtin-Functions.html. |
George Rimar | 96659df | 2016-08-30 09:54:01 +0000 | [diff] [blame] | 1261 | if (Tok == "ADDR") { |
| 1262 | expect("("); |
| 1263 | StringRef Name = next(); |
| 1264 | expect(")"); |
George Rimar | 884e786 | 2016-09-08 08:19:13 +0000 | [diff] [blame] | 1265 | return |
| 1266 | [=](uint64_t Dot) { return ScriptBase->getOutputSectionAddress(Name); }; |
George Rimar | 96659df | 2016-08-30 09:54:01 +0000 | [diff] [blame] | 1267 | } |
George Rimar | eefa758 | 2016-08-04 09:29:31 +0000 | [diff] [blame] | 1268 | if (Tok == "ASSERT") |
| 1269 | return readAssert(); |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 1270 | if (Tok == "ALIGN") { |
Rui Ueyama | 6ad7dfc | 2016-08-17 18:59:16 +0000 | [diff] [blame] | 1271 | Expr E = readParenExpr(); |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 1272 | return [=](uint64_t Dot) { return alignTo(Dot, E(Dot)); }; |
| 1273 | } |
| 1274 | if (Tok == "CONSTANT") { |
| 1275 | expect("("); |
| 1276 | StringRef Tok = next(); |
| 1277 | expect(")"); |
| 1278 | return [=](uint64_t Dot) { return getConstant(Tok); }; |
| 1279 | } |
Rafael Espindola | 54c145c | 2016-07-28 18:16:24 +0000 | [diff] [blame] | 1280 | if (Tok == "SEGMENT_START") { |
| 1281 | expect("("); |
| 1282 | next(); |
| 1283 | expect(","); |
| 1284 | uint64_t Val; |
| 1285 | next().getAsInteger(0, Val); |
| 1286 | expect(")"); |
| 1287 | return [=](uint64_t Dot) { return Val; }; |
| 1288 | } |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 1289 | if (Tok == "DATA_SEGMENT_ALIGN") { |
| 1290 | expect("("); |
| 1291 | Expr E = readExpr(); |
| 1292 | expect(","); |
| 1293 | readExpr(); |
| 1294 | expect(")"); |
Rui Ueyama | f7791bb | 2016-07-26 19:34:10 +0000 | [diff] [blame] | 1295 | return [=](uint64_t Dot) { return alignTo(Dot, E(Dot)); }; |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 1296 | } |
| 1297 | if (Tok == "DATA_SEGMENT_END") { |
| 1298 | expect("("); |
| 1299 | expect("."); |
| 1300 | expect(")"); |
| 1301 | return [](uint64_t Dot) { return Dot; }; |
| 1302 | } |
George Rimar | 276b4e6 | 2016-07-26 17:58:44 +0000 | [diff] [blame] | 1303 | // GNU linkers implements more complicated logic to handle |
| 1304 | // DATA_SEGMENT_RELRO_END. We instead ignore the arguments and just align to |
| 1305 | // the next page boundary for simplicity. |
| 1306 | if (Tok == "DATA_SEGMENT_RELRO_END") { |
| 1307 | expect("("); |
| 1308 | next(); |
| 1309 | expect(","); |
| 1310 | readExpr(); |
| 1311 | expect(")"); |
| 1312 | return [](uint64_t Dot) { return alignTo(Dot, Target->PageSize); }; |
| 1313 | } |
George Rimar | 9e69450 | 2016-07-29 16:18:47 +0000 | [diff] [blame] | 1314 | if (Tok == "SIZEOF") { |
| 1315 | expect("("); |
| 1316 | StringRef Name = next(); |
| 1317 | expect(")"); |
George Rimar | 884e786 | 2016-09-08 08:19:13 +0000 | [diff] [blame] | 1318 | return [=](uint64_t Dot) { return ScriptBase->getOutputSectionSize(Name); }; |
George Rimar | 9e69450 | 2016-07-29 16:18:47 +0000 | [diff] [blame] | 1319 | } |
Eugene Leviant | 36fac7f | 2016-09-08 09:08:30 +0000 | [diff] [blame] | 1320 | if (Tok == "ALIGNOF") { |
| 1321 | expect("("); |
| 1322 | StringRef Name = next(); |
| 1323 | expect(")"); |
| 1324 | return |
| 1325 | [=](uint64_t Dot) { return ScriptBase->getOutputSectionAlign(Name); }; |
| 1326 | } |
George Rimar | e32a359 | 2016-08-10 07:59:34 +0000 | [diff] [blame] | 1327 | if (Tok == "SIZEOF_HEADERS") |
George Rimar | 884e786 | 2016-09-08 08:19:13 +0000 | [diff] [blame] | 1328 | return [=](uint64_t Dot) { return ScriptBase->getHeaderSize(); }; |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 1329 | |
George Rimar | 9f2f7ad | 2016-09-02 16:01:42 +0000 | [diff] [blame] | 1330 | // Tok is a literal number. |
| 1331 | uint64_t V; |
| 1332 | if (readInteger(Tok, V)) |
| 1333 | return [=](uint64_t Dot) { return V; }; |
| 1334 | |
| 1335 | // Tok is a symbol name. |
| 1336 | if (Tok != "." && !isValidCIdentifier(Tok)) |
| 1337 | setError("malformed number: " + Tok); |
| 1338 | return [=](uint64_t Dot) { return getSymbolValue(Tok, Dot); }; |
Rui Ueyama | 708019c | 2016-07-24 18:19:40 +0000 | [diff] [blame] | 1339 | } |
| 1340 | |
| 1341 | Expr ScriptParser::readTernary(Expr Cond) { |
| 1342 | next(); |
| 1343 | Expr L = readExpr(); |
| 1344 | expect(":"); |
| 1345 | Expr R = readExpr(); |
| 1346 | return [=](uint64_t Dot) { return Cond(Dot) ? L(Dot) : R(Dot); }; |
| 1347 | } |
| 1348 | |
Rui Ueyama | 6ad7dfc | 2016-08-17 18:59:16 +0000 | [diff] [blame] | 1349 | Expr ScriptParser::readParenExpr() { |
| 1350 | expect("("); |
| 1351 | Expr E = readExpr(); |
| 1352 | expect(")"); |
| 1353 | return E; |
| 1354 | } |
| 1355 | |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 1356 | std::vector<StringRef> ScriptParser::readOutputSectionPhdrs() { |
| 1357 | std::vector<StringRef> Phdrs; |
| 1358 | while (!Error && peek().startswith(":")) { |
| 1359 | StringRef Tok = next(); |
| 1360 | Tok = (Tok.size() == 1) ? next() : Tok.substr(1); |
| 1361 | if (Tok.empty()) { |
| 1362 | setError("section header name is empty"); |
| 1363 | break; |
| 1364 | } |
Rui Ueyama | 047404f | 2016-07-20 19:36:36 +0000 | [diff] [blame] | 1365 | Phdrs.push_back(Tok); |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 1366 | } |
| 1367 | return Phdrs; |
| 1368 | } |
| 1369 | |
| 1370 | unsigned ScriptParser::readPhdrType() { |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 1371 | StringRef Tok = next(); |
Rui Ueyama | b0f6c59 | 2016-07-20 19:36:38 +0000 | [diff] [blame] | 1372 | unsigned Ret = StringSwitch<unsigned>(Tok) |
George Rimar | 6c55f0e | 2016-09-08 08:20:30 +0000 | [diff] [blame] | 1373 | .Case("PT_NULL", PT_NULL) |
| 1374 | .Case("PT_LOAD", PT_LOAD) |
| 1375 | .Case("PT_DYNAMIC", PT_DYNAMIC) |
| 1376 | .Case("PT_INTERP", PT_INTERP) |
| 1377 | .Case("PT_NOTE", PT_NOTE) |
| 1378 | .Case("PT_SHLIB", PT_SHLIB) |
| 1379 | .Case("PT_PHDR", PT_PHDR) |
| 1380 | .Case("PT_TLS", PT_TLS) |
| 1381 | .Case("PT_GNU_EH_FRAME", PT_GNU_EH_FRAME) |
| 1382 | .Case("PT_GNU_STACK", PT_GNU_STACK) |
| 1383 | .Case("PT_GNU_RELRO", PT_GNU_RELRO) |
| 1384 | .Default(-1); |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 1385 | |
Rui Ueyama | b0f6c59 | 2016-07-20 19:36:38 +0000 | [diff] [blame] | 1386 | if (Ret == (unsigned)-1) { |
| 1387 | setError("invalid program header type: " + Tok); |
| 1388 | return PT_NULL; |
| 1389 | } |
| 1390 | return Ret; |
Eugene Leviant | bbe3860 | 2016-07-19 09:25:43 +0000 | [diff] [blame] | 1391 | } |
| 1392 | |
Rui Ueyama | 95769b4 | 2016-08-31 20:03:54 +0000 | [diff] [blame] | 1393 | void ScriptParser::readVersionDeclaration(StringRef VerStr) { |
George Rimar | 20b6598 | 2016-08-31 09:08:26 +0000 | [diff] [blame] | 1394 | // Identifiers start at 2 because 0 and 1 are reserved |
| 1395 | // for VER_NDX_LOCAL and VER_NDX_GLOBAL constants. |
| 1396 | size_t VersionId = Config->VersionDefinitions.size() + 2; |
| 1397 | Config->VersionDefinitions.push_back({VerStr, VersionId}); |
| 1398 | |
| 1399 | if (skip("global:") || peek() != "local:") |
| 1400 | readGlobal(VerStr); |
| 1401 | if (skip("local:")) |
| 1402 | readLocal(); |
| 1403 | expect("}"); |
| 1404 | |
| 1405 | // Each version may have a parent version. For example, "Ver2" defined as |
| 1406 | // "Ver2 { global: foo; local: *; } Ver1;" has "Ver1" as a parent. This |
| 1407 | // version hierarchy is, probably against your instinct, purely for human; the |
| 1408 | // runtime doesn't care about them at all. In LLD, we simply skip the token. |
| 1409 | if (!VerStr.empty() && peek() != ";") |
| 1410 | next(); |
| 1411 | expect(";"); |
| 1412 | } |
| 1413 | |
| 1414 | void ScriptParser::readLocal() { |
| 1415 | Config->DefaultSymbolVersion = VER_NDX_LOCAL; |
| 1416 | expect("*"); |
| 1417 | expect(";"); |
| 1418 | } |
| 1419 | |
| 1420 | void ScriptParser::readExtern(std::vector<SymbolVersion> *Globals) { |
George Rimar | cd574a5 | 2016-09-09 14:35:36 +0000 | [diff] [blame] | 1421 | expect("\"C++\""); |
George Rimar | 20b6598 | 2016-08-31 09:08:26 +0000 | [diff] [blame] | 1422 | expect("{"); |
| 1423 | |
| 1424 | for (;;) { |
| 1425 | if (peek() == "}" || Error) |
| 1426 | break; |
George Rimar | cd574a5 | 2016-09-09 14:35:36 +0000 | [diff] [blame] | 1427 | bool HasWildcard = !peek().startswith("\"") && hasWildcard(peek()); |
| 1428 | Globals->push_back({unquote(next()), true, HasWildcard}); |
George Rimar | 20b6598 | 2016-08-31 09:08:26 +0000 | [diff] [blame] | 1429 | expect(";"); |
| 1430 | } |
| 1431 | |
| 1432 | expect("}"); |
| 1433 | expect(";"); |
| 1434 | } |
| 1435 | |
| 1436 | void ScriptParser::readGlobal(StringRef VerStr) { |
| 1437 | std::vector<SymbolVersion> *Globals; |
| 1438 | if (VerStr.empty()) |
| 1439 | Globals = &Config->VersionScriptGlobals; |
| 1440 | else |
| 1441 | Globals = &Config->VersionDefinitions.back().Globals; |
| 1442 | |
| 1443 | for (;;) { |
| 1444 | if (skip("extern")) |
| 1445 | readExtern(Globals); |
| 1446 | |
| 1447 | StringRef Cur = peek(); |
| 1448 | if (Cur == "}" || Cur == "local:" || Error) |
| 1449 | return; |
| 1450 | next(); |
George Rimar | cd574a5 | 2016-09-09 14:35:36 +0000 | [diff] [blame] | 1451 | Globals->push_back({unquote(Cur), false, hasWildcard(Cur)}); |
George Rimar | 20b6598 | 2016-08-31 09:08:26 +0000 | [diff] [blame] | 1452 | expect(";"); |
| 1453 | } |
| 1454 | } |
| 1455 | |
Simon Atanasyan | 16b0cc9 | 2015-11-26 05:53:00 +0000 | [diff] [blame] | 1456 | static bool isUnderSysroot(StringRef Path) { |
| 1457 | if (Config->Sysroot == "") |
| 1458 | return false; |
| 1459 | for (; !Path.empty(); Path = sys::path::parent_path(Path)) |
| 1460 | if (sys::fs::equivalent(Config->Sysroot, Path)) |
| 1461 | return true; |
| 1462 | return false; |
| 1463 | } |
| 1464 | |
Rui Ueyama | 07320e4 | 2016-04-20 20:13:41 +0000 | [diff] [blame] | 1465 | void elf::readLinkerScript(MemoryBufferRef MB) { |
Simon Atanasyan | 16b0cc9 | 2015-11-26 05:53:00 +0000 | [diff] [blame] | 1466 | StringRef Path = MB.getBufferIdentifier(); |
George Rimar | 20b6598 | 2016-08-31 09:08:26 +0000 | [diff] [blame] | 1467 | ScriptParser(MB.getBuffer(), isUnderSysroot(Path)).readLinkerScript(); |
| 1468 | } |
| 1469 | |
| 1470 | void elf::readVersionScript(MemoryBufferRef MB) { |
| 1471 | ScriptParser(MB.getBuffer(), false).readVersionScript(); |
Rui Ueyama | f7c5fbb | 2015-09-30 17:23:26 +0000 | [diff] [blame] | 1472 | } |
Rui Ueyama | 1ebc8ed | 2016-02-12 21:47:28 +0000 | [diff] [blame] | 1473 | |
Rui Ueyama | 07320e4 | 2016-04-20 20:13:41 +0000 | [diff] [blame] | 1474 | template class elf::LinkerScript<ELF32LE>; |
| 1475 | template class elf::LinkerScript<ELF32BE>; |
| 1476 | template class elf::LinkerScript<ELF64LE>; |
| 1477 | template class elf::LinkerScript<ELF64BE>; |