12ec34544SRui Ueyama //===- ScriptParser.cpp ---------------------------------------------------===//
22ec34544SRui Ueyama //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
62ec34544SRui Ueyama //
72ec34544SRui Ueyama //===----------------------------------------------------------------------===//
805f6b852SRui Ueyama //
905f6b852SRui Ueyama // This file contains a recursive-descendent parser for linker scripts.
1005f6b852SRui Ueyama // Parsed results are stored to Config and Script global objects.
1105f6b852SRui Ueyama //
1205f6b852SRui Ueyama //===----------------------------------------------------------------------===//
132ec34544SRui Ueyama 
142ec34544SRui Ueyama #include "ScriptParser.h"
152ec34544SRui Ueyama #include "Config.h"
162ec34544SRui Ueyama #include "Driver.h"
172ec34544SRui Ueyama #include "InputSection.h"
182ec34544SRui Ueyama #include "LinkerScript.h"
192ec34544SRui Ueyama #include "OutputSections.h"
202ec34544SRui Ueyama #include "ScriptLexer.h"
212ec34544SRui Ueyama #include "Symbols.h"
222ec34544SRui Ueyama #include "Target.h"
232017d52bSRui Ueyama #include "lld/Common/Memory.h"
242ec34544SRui Ueyama #include "llvm/ADT/SmallString.h"
252ec34544SRui Ueyama #include "llvm/ADT/StringRef.h"
260440be4aSRui Ueyama #include "llvm/ADT/StringSet.h"
272ec34544SRui Ueyama #include "llvm/ADT/StringSwitch.h"
28264b5d9eSZachary Turner #include "llvm/BinaryFormat/ELF.h"
292ec34544SRui Ueyama #include "llvm/Support/Casting.h"
302ec34544SRui Ueyama #include "llvm/Support/ErrorHandling.h"
312ec34544SRui Ueyama #include "llvm/Support/FileSystem.h"
322ec34544SRui Ueyama #include "llvm/Support/Path.h"
332ec34544SRui Ueyama #include <cassert>
342ec34544SRui Ueyama #include <limits>
352ec34544SRui Ueyama #include <vector>
362ec34544SRui Ueyama 
372ec34544SRui Ueyama using namespace llvm;
382ec34544SRui Ueyama using namespace llvm::ELF;
39b58079d4SRui Ueyama using namespace llvm::support::endian;
402ec34544SRui Ueyama using namespace lld;
412ec34544SRui Ueyama using namespace lld::elf;
422ec34544SRui Ueyama 
432ec34544SRui Ueyama static bool isUnderSysroot(StringRef Path);
442ec34544SRui Ueyama 
4596b3fe02SRui Ueyama namespace {
4696b3fe02SRui Ueyama class ScriptParser final : ScriptLexer {
472ec34544SRui Ueyama public:
482ec34544SRui Ueyama   ScriptParser(MemoryBufferRef MB)
492ec34544SRui Ueyama       : ScriptLexer(MB),
502ec34544SRui Ueyama         IsUnderSysroot(isUnderSysroot(MB.getBufferIdentifier())) {}
512ec34544SRui Ueyama 
522ec34544SRui Ueyama   void readLinkerScript();
532ec34544SRui Ueyama   void readVersionScript();
542ec34544SRui Ueyama   void readDynamicList();
558c7e8cceSPetr Hosek   void readDefsym(StringRef Name);
562ec34544SRui Ueyama 
572ec34544SRui Ueyama private:
582ec34544SRui Ueyama   void addFile(StringRef Path);
592ec34544SRui Ueyama 
602ec34544SRui Ueyama   void readAsNeeded();
612ec34544SRui Ueyama   void readEntry();
622ec34544SRui Ueyama   void readExtern();
632ec34544SRui Ueyama   void readGroup();
642ec34544SRui Ueyama   void readInclude();
651d92aa73SRui Ueyama   void readInput();
662ec34544SRui Ueyama   void readMemory();
672ec34544SRui Ueyama   void readOutput();
682ec34544SRui Ueyama   void readOutputArch();
692ec34544SRui Ueyama   void readOutputFormat();
702ec34544SRui Ueyama   void readPhdrs();
715f37541cSGeorge Rimar   void readRegionAlias();
722ec34544SRui Ueyama   void readSearchDir();
732ec34544SRui Ueyama   void readSections();
74e262bb1aSRui Ueyama   void readTarget();
752ec34544SRui Ueyama   void readVersion();
762ec34544SRui Ueyama   void readVersionScriptCommand();
772ec34544SRui Ueyama 
78d30a78b3SGeorge Rimar   SymbolAssignment *readSymbolAssignment(StringRef Name);
79f0403c60SRui Ueyama   ByteCommand *readByteCommand(StringRef Tok);
80b0486051SSimon Atanasyan   std::array<uint8_t, 4> readFill();
81b0486051SSimon Atanasyan   std::array<uint8_t, 4> parseFill(StringRef Tok);
82a46d08ebSGeorge Rimar   bool readSectionDirective(OutputSection *Cmd, StringRef Tok1, StringRef Tok2);
838c022ca7SRafael Espindola   void readSectionAddressType(OutputSection *Cmd);
84a582419aSGeorge Rimar   OutputSection *readOverlaySectionDescription();
858c022ca7SRafael Espindola   OutputSection *readOutputSectionDescription(StringRef OutSec);
86a582419aSGeorge Rimar   std::vector<BaseCommand *> readOverlay();
872ec34544SRui Ueyama   std::vector<StringRef> readOutputSectionPhdrs();
882ec34544SRui Ueyama   InputSectionDescription *readInputSectionDescription(StringRef Tok);
892ec34544SRui Ueyama   StringMatcher readFilePatterns();
902ec34544SRui Ueyama   std::vector<SectionPattern> readInputSectionsList();
912ec34544SRui Ueyama   InputSectionDescription *readInputSectionRules(StringRef FilePattern);
922ec34544SRui Ueyama   unsigned readPhdrType();
932ec34544SRui Ueyama   SortSectionPolicy readSortKind();
942ec34544SRui Ueyama   SymbolAssignment *readProvideHidden(bool Provide, bool Hidden);
95d30a78b3SGeorge Rimar   SymbolAssignment *readAssignment(StringRef Tok);
9669c61200SSimon Atanasyan   std::tuple<ELFKind, uint16_t, bool> readBfdName();
972ec34544SRui Ueyama   void readSort();
98d30a78b3SGeorge Rimar   Expr readAssert();
995fb17128SGeorge Rimar   Expr readConstant();
1005fb17128SGeorge Rimar   Expr getPageSize();
1012ec34544SRui Ueyama 
1022ec34544SRui Ueyama   uint64_t readMemoryAssignment(StringRef, StringRef, StringRef);
1032ec34544SRui Ueyama   std::pair<uint32_t, uint32_t> readMemoryAttributes();
1042ec34544SRui Ueyama 
1057b91e213SGeorge Rimar   Expr combine(StringRef Op, Expr L, Expr R);
1062ec34544SRui Ueyama   Expr readExpr();
1072ec34544SRui Ueyama   Expr readExpr1(Expr Lhs, int MinPrec);
1082ec34544SRui Ueyama   StringRef readParenLiteral();
1092ec34544SRui Ueyama   Expr readPrimary();
1102ec34544SRui Ueyama   Expr readTernary(Expr Cond);
1112ec34544SRui Ueyama   Expr readParenExpr();
1122ec34544SRui Ueyama 
1132ec34544SRui Ueyama   // For parsing version script.
1142ec34544SRui Ueyama   std::vector<SymbolVersion> readVersionExtern();
1152ec34544SRui Ueyama   void readAnonymousDeclaration();
1162ec34544SRui Ueyama   void readVersionDeclaration(StringRef VerStr);
1172ec34544SRui Ueyama 
1182ec34544SRui Ueyama   std::pair<std::vector<SymbolVersion>, std::vector<SymbolVersion>>
1192ec34544SRui Ueyama   readSymbols();
1202ec34544SRui Ueyama 
121fd06b025SRui Ueyama   // True if a script being read is in a subdirectory specified by -sysroot.
1222ec34544SRui Ueyama   bool IsUnderSysroot;
1230440be4aSRui Ueyama 
1240440be4aSRui Ueyama   // A set to detect an INCLUDE() cycle.
1250440be4aSRui Ueyama   StringSet<> Seen;
1262ec34544SRui Ueyama };
12796b3fe02SRui Ueyama } // namespace
1282ec34544SRui Ueyama 
1291e77ad14SRui Ueyama static StringRef unquote(StringRef S) {
1301e77ad14SRui Ueyama   if (S.startswith("\""))
1311e77ad14SRui Ueyama     return S.substr(1, S.size() - 2);
1321e77ad14SRui Ueyama   return S;
1331e77ad14SRui Ueyama }
1341e77ad14SRui Ueyama 
1352ec34544SRui Ueyama static bool isUnderSysroot(StringRef Path) {
1362ec34544SRui Ueyama   if (Config->Sysroot == "")
1372ec34544SRui Ueyama     return false;
1382ec34544SRui Ueyama   for (; !Path.empty(); Path = sys::path::parent_path(Path))
1392ec34544SRui Ueyama     if (sys::fs::equivalent(Config->Sysroot, Path))
1402ec34544SRui Ueyama       return true;
1412ec34544SRui Ueyama   return false;
1422ec34544SRui Ueyama }
1432ec34544SRui Ueyama 
1442ec34544SRui Ueyama // Some operations only support one non absolute value. Move the
1452ec34544SRui Ueyama // absolute one to the right hand side for convenience.
1462ec34544SRui Ueyama static void moveAbsRight(ExprValue &A, ExprValue &B) {
14723be5e8dSRafael Espindola   if (A.Sec == nullptr || (A.ForceAbsolute && !B.isAbsolute()))
1482ec34544SRui Ueyama     std::swap(A, B);
1492ec34544SRui Ueyama   if (!B.isAbsolute())
15041c7ab4aSGeorge Rimar     error(A.Loc + ": at least one side of the expression must be absolute");
1512ec34544SRui Ueyama }
1522ec34544SRui Ueyama 
1532ec34544SRui Ueyama static ExprValue add(ExprValue A, ExprValue B) {
1542ec34544SRui Ueyama   moveAbsRight(A, B);
155a6acd23cSRafael Espindola   return {A.Sec, A.ForceAbsolute, A.getSectionOffset() + B.getValue(), A.Loc};
1562ec34544SRui Ueyama }
1572ec34544SRui Ueyama 
1582ec34544SRui Ueyama static ExprValue sub(ExprValue A, ExprValue B) {
15963a4a98eSRafael Espindola   // The distance between two symbols in sections is absolute.
1609cbb6dd1SRafael Espindola   if (!A.isAbsolute() && !B.isAbsolute())
1619cbb6dd1SRafael Espindola     return A.getValue() - B.getValue();
1624fbe3518SRui Ueyama   return {A.Sec, false, A.getSectionOffset() - B.getValue(), A.Loc};
1632ec34544SRui Ueyama }
1642ec34544SRui Ueyama 
1652ec34544SRui Ueyama static ExprValue bitAnd(ExprValue A, ExprValue B) {
1662ec34544SRui Ueyama   moveAbsRight(A, B);
1672ec34544SRui Ueyama   return {A.Sec, A.ForceAbsolute,
16841c7ab4aSGeorge Rimar           (A.getValue() & B.getValue()) - A.getSecAddr(), A.Loc};
1692ec34544SRui Ueyama }
1702ec34544SRui Ueyama 
1712ec34544SRui Ueyama static ExprValue bitOr(ExprValue A, ExprValue B) {
1722ec34544SRui Ueyama   moveAbsRight(A, B);
1732ec34544SRui Ueyama   return {A.Sec, A.ForceAbsolute,
17441c7ab4aSGeorge Rimar           (A.getValue() | B.getValue()) - A.getSecAddr(), A.Loc};
1752ec34544SRui Ueyama }
1762ec34544SRui Ueyama 
1772ec34544SRui Ueyama void ScriptParser::readDynamicList() {
1788016bdfdSRafael Espindola   Config->HasDynamicList = true;
1792ec34544SRui Ueyama   expect("{");
180d72d97b3SRafael Espindola   std::vector<SymbolVersion> Locals;
181d72d97b3SRafael Espindola   std::vector<SymbolVersion> Globals;
182d72d97b3SRafael Espindola   std::tie(Locals, Globals) = readSymbols();
183d72d97b3SRafael Espindola   expect(";");
184d72d97b3SRafael Espindola 
185d72d97b3SRafael Espindola   if (!atEOF()) {
1862ec34544SRui Ueyama     setError("EOF expected, but got " + next());
187d72d97b3SRafael Espindola     return;
188d72d97b3SRafael Espindola   }
189d72d97b3SRafael Espindola   if (!Locals.empty()) {
190d72d97b3SRafael Espindola     setError("\"local:\" scope not supported in --dynamic-list");
191d72d97b3SRafael Espindola     return;
192d72d97b3SRafael Espindola   }
193d72d97b3SRafael Espindola 
194d72d97b3SRafael Espindola   for (SymbolVersion V : Globals)
195d72d97b3SRafael Espindola     Config->DynamicList.push_back(V);
1962ec34544SRui Ueyama }
1972ec34544SRui Ueyama 
1982ec34544SRui Ueyama void ScriptParser::readVersionScript() {
1992ec34544SRui Ueyama   readVersionScriptCommand();
2002ec34544SRui Ueyama   if (!atEOF())
2012ec34544SRui Ueyama     setError("EOF expected, but got " + next());
2022ec34544SRui Ueyama }
2032ec34544SRui Ueyama 
2042ec34544SRui Ueyama void ScriptParser::readVersionScriptCommand() {
2052ec34544SRui Ueyama   if (consume("{")) {
2062ec34544SRui Ueyama     readAnonymousDeclaration();
2072ec34544SRui Ueyama     return;
2082ec34544SRui Ueyama   }
2092ec34544SRui Ueyama 
210b8a59c8aSBob Haarman   while (!atEOF() && !errorCount() && peek() != "}") {
2112ec34544SRui Ueyama     StringRef VerStr = next();
2122ec34544SRui Ueyama     if (VerStr == "{") {
2132ec34544SRui Ueyama       setError("anonymous version definition is used in "
2142ec34544SRui Ueyama                "combination with other version definitions");
2152ec34544SRui Ueyama       return;
2162ec34544SRui Ueyama     }
2172ec34544SRui Ueyama     expect("{");
2182ec34544SRui Ueyama     readVersionDeclaration(VerStr);
2192ec34544SRui Ueyama   }
2202ec34544SRui Ueyama }
2212ec34544SRui Ueyama 
2222ec34544SRui Ueyama void ScriptParser::readVersion() {
2232ec34544SRui Ueyama   expect("{");
2242ec34544SRui Ueyama   readVersionScriptCommand();
2252ec34544SRui Ueyama   expect("}");
2262ec34544SRui Ueyama }
2272ec34544SRui Ueyama 
2282ec34544SRui Ueyama void ScriptParser::readLinkerScript() {
2292ec34544SRui Ueyama   while (!atEOF()) {
2302ec34544SRui Ueyama     StringRef Tok = next();
2312ec34544SRui Ueyama     if (Tok == ";")
2322ec34544SRui Ueyama       continue;
2332ec34544SRui Ueyama 
234d30a78b3SGeorge Rimar     if (Tok == "ENTRY") {
2352ec34544SRui Ueyama       readEntry();
2362ec34544SRui Ueyama     } else if (Tok == "EXTERN") {
2372ec34544SRui Ueyama       readExtern();
2381d92aa73SRui Ueyama     } else if (Tok == "GROUP") {
2392ec34544SRui Ueyama       readGroup();
2402ec34544SRui Ueyama     } else if (Tok == "INCLUDE") {
2412ec34544SRui Ueyama       readInclude();
2421d92aa73SRui Ueyama     } else if (Tok == "INPUT") {
2431d92aa73SRui Ueyama       readInput();
2442ec34544SRui Ueyama     } else if (Tok == "MEMORY") {
2452ec34544SRui Ueyama       readMemory();
2462ec34544SRui Ueyama     } else if (Tok == "OUTPUT") {
2472ec34544SRui Ueyama       readOutput();
2482ec34544SRui Ueyama     } else if (Tok == "OUTPUT_ARCH") {
2492ec34544SRui Ueyama       readOutputArch();
2502ec34544SRui Ueyama     } else if (Tok == "OUTPUT_FORMAT") {
2512ec34544SRui Ueyama       readOutputFormat();
2522ec34544SRui Ueyama     } else if (Tok == "PHDRS") {
2532ec34544SRui Ueyama       readPhdrs();
2545f37541cSGeorge Rimar     } else if (Tok == "REGION_ALIAS") {
2555f37541cSGeorge Rimar       readRegionAlias();
2562ec34544SRui Ueyama     } else if (Tok == "SEARCH_DIR") {
2572ec34544SRui Ueyama       readSearchDir();
2582ec34544SRui Ueyama     } else if (Tok == "SECTIONS") {
2592ec34544SRui Ueyama       readSections();
260e262bb1aSRui Ueyama     } else if (Tok == "TARGET") {
261e262bb1aSRui Ueyama       readTarget();
2622ec34544SRui Ueyama     } else if (Tok == "VERSION") {
2632ec34544SRui Ueyama       readVersion();
264d30a78b3SGeorge Rimar     } else if (SymbolAssignment *Cmd = readAssignment(Tok)) {
2656b394caaSRui Ueyama       Script->SectionCommands.push_back(Cmd);
2662ec34544SRui Ueyama     } else {
2672ec34544SRui Ueyama       setError("unknown directive: " + Tok);
2682ec34544SRui Ueyama     }
2692ec34544SRui Ueyama   }
2702ec34544SRui Ueyama }
2712ec34544SRui Ueyama 
2728c7e8cceSPetr Hosek void ScriptParser::readDefsym(StringRef Name) {
273c1522816SGeorge Rimar   if (errorCount())
274c1522816SGeorge Rimar     return;
2758c7e8cceSPetr Hosek   Expr E = readExpr();
2768c7e8cceSPetr Hosek   if (!atEOF())
2778c7e8cceSPetr Hosek     setError("EOF expected, but got " + next());
278e88b76a9SGeorge Rimar   SymbolAssignment *Cmd = make<SymbolAssignment>(Name, E, getCurrentLocation());
2798c7e8cceSPetr Hosek   Script->SectionCommands.push_back(Cmd);
2808c7e8cceSPetr Hosek }
2818c7e8cceSPetr Hosek 
2822ec34544SRui Ueyama void ScriptParser::addFile(StringRef S) {
2832ec34544SRui Ueyama   if (IsUnderSysroot && S.startswith("/")) {
2842ec34544SRui Ueyama     SmallString<128> PathData;
2852ec34544SRui Ueyama     StringRef Path = (Config->Sysroot + S).toStringRef(PathData);
2862ec34544SRui Ueyama     if (sys::fs::exists(Path)) {
287a76349bfSEvgeniy Stepanov       Driver->addFile(Saver.save(Path), /*WithLOption=*/false);
2882ec34544SRui Ueyama       return;
2892ec34544SRui Ueyama     }
2902ec34544SRui Ueyama   }
2912ec34544SRui Ueyama 
292875ae82bSRui Ueyama   if (S.startswith("/")) {
293a76349bfSEvgeniy Stepanov     Driver->addFile(S, /*WithLOption=*/false);
2942ec34544SRui Ueyama   } else if (S.startswith("=")) {
2952ec34544SRui Ueyama     if (Config->Sysroot.empty())
296a76349bfSEvgeniy Stepanov       Driver->addFile(S.substr(1), /*WithLOption=*/false);
2972ec34544SRui Ueyama     else
298a76349bfSEvgeniy Stepanov       Driver->addFile(Saver.save(Config->Sysroot + "/" + S.substr(1)),
299a76349bfSEvgeniy Stepanov                       /*WithLOption=*/false);
3002ec34544SRui Ueyama   } else if (S.startswith("-l")) {
3012ec34544SRui Ueyama     Driver->addLibrary(S.substr(2));
3022ec34544SRui Ueyama   } else if (sys::fs::exists(S)) {
303a76349bfSEvgeniy Stepanov     Driver->addFile(S, /*WithLOption=*/false);
3042ec34544SRui Ueyama   } else {
3052ec34544SRui Ueyama     if (Optional<std::string> Path = findFromSearchPaths(S))
306a76349bfSEvgeniy Stepanov       Driver->addFile(Saver.save(*Path), /*WithLOption=*/true);
3072ec34544SRui Ueyama     else
3082ec34544SRui Ueyama       setError("unable to find " + S);
3092ec34544SRui Ueyama   }
3102ec34544SRui Ueyama }
3112ec34544SRui Ueyama 
3122ec34544SRui Ueyama void ScriptParser::readAsNeeded() {
3132ec34544SRui Ueyama   expect("(");
3142ec34544SRui Ueyama   bool Orig = Config->AsNeeded;
3152ec34544SRui Ueyama   Config->AsNeeded = true;
316b8a59c8aSBob Haarman   while (!errorCount() && !consume(")"))
3172ec34544SRui Ueyama     addFile(unquote(next()));
3182ec34544SRui Ueyama   Config->AsNeeded = Orig;
3192ec34544SRui Ueyama }
3202ec34544SRui Ueyama 
3212ec34544SRui Ueyama void ScriptParser::readEntry() {
3222ec34544SRui Ueyama   // -e <symbol> takes predecence over ENTRY(<symbol>).
3232ec34544SRui Ueyama   expect("(");
3242ec34544SRui Ueyama   StringRef Tok = next();
3252ec34544SRui Ueyama   if (Config->Entry.empty())
3262ec34544SRui Ueyama     Config->Entry = Tok;
3272ec34544SRui Ueyama   expect(")");
3282ec34544SRui Ueyama }
3292ec34544SRui Ueyama 
3302ec34544SRui Ueyama void ScriptParser::readExtern() {
3312ec34544SRui Ueyama   expect("(");
332b8a59c8aSBob Haarman   while (!errorCount() && !consume(")"))
3332ec34544SRui Ueyama     Config->Undefined.push_back(next());
3342ec34544SRui Ueyama }
3352ec34544SRui Ueyama 
3362ec34544SRui Ueyama void ScriptParser::readGroup() {
3371d92aa73SRui Ueyama   bool Orig = InputFile::IsInGroup;
3381d92aa73SRui Ueyama   InputFile::IsInGroup = true;
3391d92aa73SRui Ueyama   readInput();
3401d92aa73SRui Ueyama   InputFile::IsInGroup = Orig;
341b72daf00SFangrui Song   if (!Orig)
342b72daf00SFangrui Song     ++InputFile::NextGroupId;
3432ec34544SRui Ueyama }
3442ec34544SRui Ueyama 
3452ec34544SRui Ueyama void ScriptParser::readInclude() {
3462ec34544SRui Ueyama   StringRef Tok = unquote(next());
3472ec34544SRui Ueyama 
3480440be4aSRui Ueyama   if (!Seen.insert(Tok).second) {
3490440be4aSRui Ueyama     setError("there is a cycle in linker script INCLUDEs");
3500440be4aSRui Ueyama     return;
3510440be4aSRui Ueyama   }
3520440be4aSRui Ueyama 
353c60f85d0SFangrui Song   if (Optional<std::string> Path = searchScript(Tok)) {
3542ec34544SRui Ueyama     if (Optional<MemoryBufferRef> MB = readFile(*Path))
3552ec34544SRui Ueyama       tokenize(*MB);
3562ec34544SRui Ueyama     return;
3572ec34544SRui Ueyama   }
3581de78471SAlexander Richardson   setError("cannot find linker script " + Tok);
3592ec34544SRui Ueyama }
3602ec34544SRui Ueyama 
3611d92aa73SRui Ueyama void ScriptParser::readInput() {
3621d92aa73SRui Ueyama   expect("(");
3631d92aa73SRui Ueyama   while (!errorCount() && !consume(")")) {
3641d92aa73SRui Ueyama     if (consume("AS_NEEDED"))
3651d92aa73SRui Ueyama       readAsNeeded();
3661d92aa73SRui Ueyama     else
3671d92aa73SRui Ueyama       addFile(unquote(next()));
3681d92aa73SRui Ueyama   }
3691d92aa73SRui Ueyama }
3701d92aa73SRui Ueyama 
3712ec34544SRui Ueyama void ScriptParser::readOutput() {
3722ec34544SRui Ueyama   // -o <file> takes predecence over OUTPUT(<file>).
3732ec34544SRui Ueyama   expect("(");
3742ec34544SRui Ueyama   StringRef Tok = next();
3752ec34544SRui Ueyama   if (Config->OutputFile.empty())
3762ec34544SRui Ueyama     Config->OutputFile = unquote(Tok);
3772ec34544SRui Ueyama   expect(")");
3782ec34544SRui Ueyama }
3792ec34544SRui Ueyama 
3802ec34544SRui Ueyama void ScriptParser::readOutputArch() {
3812ec34544SRui Ueyama   // OUTPUT_ARCH is ignored for now.
3822ec34544SRui Ueyama   expect("(");
383b8a59c8aSBob Haarman   while (!errorCount() && !consume(")"))
3842ec34544SRui Ueyama     skip();
3852ec34544SRui Ueyama }
3862ec34544SRui Ueyama 
387*0068d223SRui Ueyama static std::pair<ELFKind, uint16_t> parseBfdName(StringRef S) {
388*0068d223SRui Ueyama   return StringSwitch<std::pair<ELFKind, uint16_t>>(S)
3894f8c8228SRui Ueyama       .Case("elf32-i386", {ELF32LEKind, EM_386})
3904f8c8228SRui Ueyama       .Case("elf32-iamcu", {ELF32LEKind, EM_IAMCU})
3914f8c8228SRui Ueyama       .Case("elf32-littlearm", {ELF32LEKind, EM_ARM})
3924f8c8228SRui Ueyama       .Case("elf32-x86-64", {ELF32LEKind, EM_X86_64})
3934f8c8228SRui Ueyama       .Case("elf64-littleaarch64", {ELF64LEKind, EM_AARCH64})
3944f8c8228SRui Ueyama       .Case("elf64-powerpc", {ELF64BEKind, EM_PPC64})
3954f8c8228SRui Ueyama       .Case("elf64-powerpcle", {ELF64LEKind, EM_PPC64})
3964f8c8228SRui Ueyama       .Case("elf64-x86-64", {ELF64LEKind, EM_X86_64})
3974f8c8228SRui Ueyama       .Case("elf32-tradbigmips", {ELF32BEKind, EM_MIPS})
3984f8c8228SRui Ueyama       .Case("elf32-ntradbigmips", {ELF32BEKind, EM_MIPS})
3994f8c8228SRui Ueyama       .Case("elf32-tradlittlemips", {ELF32LEKind, EM_MIPS})
4004f8c8228SRui Ueyama       .Case("elf32-ntradlittlemips", {ELF32LEKind, EM_MIPS})
4014f8c8228SRui Ueyama       .Case("elf64-tradbigmips", {ELF64BEKind, EM_MIPS})
4024f8c8228SRui Ueyama       .Case("elf64-tradlittlemips", {ELF64LEKind, EM_MIPS})
4034f8c8228SRui Ueyama       .Default({ELFNoneKind, EM_NONE});
404ea8cd00aSRui Ueyama }
405ea8cd00aSRui Ueyama 
406ea8cd00aSRui Ueyama // Parse OUTPUT_FORMAT(bfdname) or OUTPUT_FORMAT(bfdname, big, little).
407ea8cd00aSRui Ueyama // Currently we ignore big and little parameters.
4082ec34544SRui Ueyama void ScriptParser::readOutputFormat() {
4092ec34544SRui Ueyama   expect("(");
410ea8cd00aSRui Ueyama 
4114f8c8228SRui Ueyama   StringRef S = unquote(next());
4124f8c8228SRui Ueyama 
4134f8c8228SRui Ueyama   std::tie(Config->EKind, Config->EMachine) = parseBfdName(S);
4144f8c8228SRui Ueyama   if (Config->EMachine == EM_NONE)
4154f8c8228SRui Ueyama     setError("unknown output format name: " + S);
4164f8c8228SRui Ueyama   if (S == "elf32-ntradlittlemips" || S == "elf32-ntradbigmips")
4174f8c8228SRui Ueyama     Config->MipsN32Abi = true;
418ea8cd00aSRui Ueyama 
419b579c439SRui Ueyama   if (consume(")"))
4202ec34544SRui Ueyama     return;
421b579c439SRui Ueyama   expect(",");
4222ec34544SRui Ueyama   skip();
4232ec34544SRui Ueyama   expect(",");
4242ec34544SRui Ueyama   skip();
4252ec34544SRui Ueyama   expect(")");
4262ec34544SRui Ueyama }
4272ec34544SRui Ueyama 
4282ec34544SRui Ueyama void ScriptParser::readPhdrs() {
4292ec34544SRui Ueyama   expect("{");
4302ec34544SRui Ueyama 
431b8a59c8aSBob Haarman   while (!errorCount() && !consume("}")) {
4320ae2c24cSRui Ueyama     PhdrsCommand Cmd;
4330ae2c24cSRui Ueyama     Cmd.Name = next();
4340ae2c24cSRui Ueyama     Cmd.Type = readPhdrType();
435b579c439SRui Ueyama 
436b8a59c8aSBob Haarman     while (!errorCount() && !consume(";")) {
437b579c439SRui Ueyama       if (consume("FILEHDR"))
4380ae2c24cSRui Ueyama         Cmd.HasFilehdr = true;
439b579c439SRui Ueyama       else if (consume("PHDRS"))
4400ae2c24cSRui Ueyama         Cmd.HasPhdrs = true;
441b579c439SRui Ueyama       else if (consume("AT"))
4420ae2c24cSRui Ueyama         Cmd.LMAExpr = readParenExpr();
443b579c439SRui Ueyama       else if (consume("FLAGS"))
4440ae2c24cSRui Ueyama         Cmd.Flags = readParenExpr()().getValue();
445b579c439SRui Ueyama       else
446b579c439SRui Ueyama         setError("unexpected header attribute: " + next());
447b579c439SRui Ueyama     }
4480ae2c24cSRui Ueyama 
449ac27de9dSRui Ueyama     Script->PhdrsCommands.push_back(Cmd);
4502ec34544SRui Ueyama   }
4512ec34544SRui Ueyama }
4522ec34544SRui Ueyama 
4535f37541cSGeorge Rimar void ScriptParser::readRegionAlias() {
4545f37541cSGeorge Rimar   expect("(");
4555f37541cSGeorge Rimar   StringRef Alias = unquote(next());
4565f37541cSGeorge Rimar   expect(",");
4575f37541cSGeorge Rimar   StringRef Name = next();
4585f37541cSGeorge Rimar   expect(")");
4595f37541cSGeorge Rimar 
460ac27de9dSRui Ueyama   if (Script->MemoryRegions.count(Alias))
4615f37541cSGeorge Rimar     setError("redefinition of memory region '" + Alias + "'");
462ac27de9dSRui Ueyama   if (!Script->MemoryRegions.count(Name))
4635f37541cSGeorge Rimar     setError("memory region '" + Name + "' is not defined");
4648c825db2SGeorge Rimar   Script->MemoryRegions.insert({Alias, Script->MemoryRegions[Name]});
4655f37541cSGeorge Rimar }
4665f37541cSGeorge Rimar 
4672ec34544SRui Ueyama void ScriptParser::readSearchDir() {
4682ec34544SRui Ueyama   expect("(");
4692ec34544SRui Ueyama   StringRef Tok = next();
4702ec34544SRui Ueyama   if (!Config->Nostdlib)
4712ec34544SRui Ueyama     Config->SearchPaths.push_back(unquote(Tok));
4722ec34544SRui Ueyama   expect(")");
4732ec34544SRui Ueyama }
4742ec34544SRui Ueyama 
475a582419aSGeorge Rimar // This reads an overlay description. Overlays are used to describe output
476a582419aSGeorge Rimar // sections that use the same virtual memory range and normally would trigger
477a582419aSGeorge Rimar // linker's sections sanity check failures.
478a582419aSGeorge Rimar // https://sourceware.org/binutils/docs/ld/Overlay-Description.html#Overlay-Description
479a582419aSGeorge Rimar std::vector<BaseCommand *> ScriptParser::readOverlay() {
480a582419aSGeorge Rimar   // VA and LMA expressions are optional, though for simplicity of
481a582419aSGeorge Rimar   // implementation we assume they are not. That is what OVERLAY was designed
482a582419aSGeorge Rimar   // for first of all: to allow sections with overlapping VAs at different LMAs.
483a582419aSGeorge Rimar   Expr AddrExpr = readExpr();
484a582419aSGeorge Rimar   expect(":");
485a582419aSGeorge Rimar   expect("AT");
486a582419aSGeorge Rimar   Expr LMAExpr = readParenExpr();
487a582419aSGeorge Rimar   expect("{");
488a582419aSGeorge Rimar 
489a582419aSGeorge Rimar   std::vector<BaseCommand *> V;
490a582419aSGeorge Rimar   OutputSection *Prev = nullptr;
491a582419aSGeorge Rimar   while (!errorCount() && !consume("}")) {
492a582419aSGeorge Rimar     // VA is the same for all sections. The LMAs are consecutive in memory
493a582419aSGeorge Rimar     // starting from the base load address specified.
494a582419aSGeorge Rimar     OutputSection *OS = readOverlaySectionDescription();
495a582419aSGeorge Rimar     OS->AddrExpr = AddrExpr;
496a582419aSGeorge Rimar     if (Prev)
497a582419aSGeorge Rimar       OS->LMAExpr = [=] { return Prev->getLMA() + Prev->Size; };
498a582419aSGeorge Rimar     else
499a582419aSGeorge Rimar       OS->LMAExpr = LMAExpr;
500a582419aSGeorge Rimar     V.push_back(OS);
501a582419aSGeorge Rimar     Prev = OS;
502a582419aSGeorge Rimar   }
503a582419aSGeorge Rimar 
504a582419aSGeorge Rimar   // According to the specification, at the end of the overlay, the location
505a582419aSGeorge Rimar   // counter should be equal to the overlay base address plus size of the
506a582419aSGeorge Rimar   // largest section seen in the overlay.
507a582419aSGeorge Rimar   // Here we want to create the Dot assignment command to achieve that.
508a582419aSGeorge Rimar   Expr MoveDot = [=] {
509a582419aSGeorge Rimar     uint64_t Max = 0;
510a582419aSGeorge Rimar     for (BaseCommand *Cmd : V)
511a582419aSGeorge Rimar       Max = std::max(Max, cast<OutputSection>(Cmd)->Size);
512a582419aSGeorge Rimar     return AddrExpr().getValue() + Max;
513a582419aSGeorge Rimar   };
514a582419aSGeorge Rimar   V.push_back(make<SymbolAssignment>(".", MoveDot, getCurrentLocation()));
515a582419aSGeorge Rimar   return V;
516a582419aSGeorge Rimar }
517a582419aSGeorge Rimar 
5182ec34544SRui Ueyama void ScriptParser::readSections() {
519a323e2a7SRui Ueyama   Script->HasSectionsCommand = true;
520b579c439SRui Ueyama 
5212ec34544SRui Ueyama   // -no-rosegment is used to avoid placing read only non-executable sections in
5222ec34544SRui Ueyama   // their own segment. We do the same if SECTIONS command is present in linker
5232ec34544SRui Ueyama   // script. See comment for computeFlags().
5242ec34544SRui Ueyama   Config->SingleRoRx = true;
5252ec34544SRui Ueyama 
5262ec34544SRui Ueyama   expect("{");
5279e2c8a9dSGeorge Rimar   std::vector<BaseCommand *> V;
528b8a59c8aSBob Haarman   while (!errorCount() && !consume("}")) {
5292ec34544SRui Ueyama     StringRef Tok = next();
530a582419aSGeorge Rimar     if (Tok == "OVERLAY") {
531a582419aSGeorge Rimar       for (BaseCommand *Cmd : readOverlay())
532a582419aSGeorge Rimar         V.push_back(Cmd);
533a582419aSGeorge Rimar       continue;
5342e9d40d5SRui Ueyama     } else if (Tok == "INCLUDE") {
5352e9d40d5SRui Ueyama       readInclude();
5362e9d40d5SRui Ueyama       continue;
537a582419aSGeorge Rimar     }
538a582419aSGeorge Rimar 
539d30a78b3SGeorge Rimar     if (BaseCommand *Cmd = readAssignment(Tok))
5409e2c8a9dSGeorge Rimar       V.push_back(Cmd);
541d30a78b3SGeorge Rimar     else
542d30a78b3SGeorge Rimar       V.push_back(readOutputSectionDescription(Tok));
5432ec34544SRui Ueyama   }
5449e2c8a9dSGeorge Rimar 
5459e2c8a9dSGeorge Rimar   if (!atEOF() && consume("INSERT")) {
546796684b4SGeorge Rimar     std::vector<BaseCommand *> *Dest = nullptr;
547796684b4SGeorge Rimar     if (consume("AFTER"))
548796684b4SGeorge Rimar       Dest = &Script->InsertAfterCommands[next()];
549796684b4SGeorge Rimar     else if (consume("BEFORE"))
550796684b4SGeorge Rimar       Dest = &Script->InsertBeforeCommands[next()];
551796684b4SGeorge Rimar     else
552796684b4SGeorge Rimar       setError("expected AFTER/BEFORE, but got '" + next() + "'");
553796684b4SGeorge Rimar     if (Dest)
554796684b4SGeorge Rimar       Dest->insert(Dest->end(), V.begin(), V.end());
5559e2c8a9dSGeorge Rimar     return;
5569e2c8a9dSGeorge Rimar   }
5579e2c8a9dSGeorge Rimar 
5589e2c8a9dSGeorge Rimar   Script->SectionCommands.insert(Script->SectionCommands.end(), V.begin(),
5599e2c8a9dSGeorge Rimar                                  V.end());
5602ec34544SRui Ueyama }
5612ec34544SRui Ueyama 
562e262bb1aSRui Ueyama void ScriptParser::readTarget() {
563e262bb1aSRui Ueyama   // TARGET(foo) is an alias for "--format foo". Unlike GNU linkers,
564e262bb1aSRui Ueyama   // we accept only a limited set of BFD names (i.e. "elf" or "binary")
565e262bb1aSRui Ueyama   // for --format. We recognize only /^elf/ and "binary" in the linker
566e262bb1aSRui Ueyama   // script as well.
567e262bb1aSRui Ueyama   expect("(");
568e262bb1aSRui Ueyama   StringRef Tok = next();
569e262bb1aSRui Ueyama   expect(")");
570e262bb1aSRui Ueyama 
571e262bb1aSRui Ueyama   if (Tok.startswith("elf"))
572e262bb1aSRui Ueyama     Config->FormatBinary = false;
573e262bb1aSRui Ueyama   else if (Tok == "binary")
574e262bb1aSRui Ueyama     Config->FormatBinary = true;
575e262bb1aSRui Ueyama   else
576e262bb1aSRui Ueyama     setError("unknown target: " + Tok);
577e262bb1aSRui Ueyama }
578e262bb1aSRui Ueyama 
5792ec34544SRui Ueyama static int precedence(StringRef Op) {
5802ec34544SRui Ueyama   return StringSwitch<int>(Op)
581a5005482SGeorge Rimar       .Cases("*", "/", "%", 8)
582a5005482SGeorge Rimar       .Cases("+", "-", 7)
583a5005482SGeorge Rimar       .Cases("<<", ">>", 6)
584a5005482SGeorge Rimar       .Cases("<", "<=", ">", ">=", "==", "!=", 5)
585a5005482SGeorge Rimar       .Case("&", 4)
586a5005482SGeorge Rimar       .Case("|", 3)
587a5005482SGeorge Rimar       .Case("&&", 2)
588a5005482SGeorge Rimar       .Case("||", 1)
5892ec34544SRui Ueyama       .Default(-1);
5902ec34544SRui Ueyama }
5912ec34544SRui Ueyama 
5922ec34544SRui Ueyama StringMatcher ScriptParser::readFilePatterns() {
5932ec34544SRui Ueyama   std::vector<StringRef> V;
594b8a59c8aSBob Haarman   while (!errorCount() && !consume(")"))
5952ec34544SRui Ueyama     V.push_back(next());
5962ec34544SRui Ueyama   return StringMatcher(V);
5972ec34544SRui Ueyama }
5982ec34544SRui Ueyama 
5992ec34544SRui Ueyama SortSectionPolicy ScriptParser::readSortKind() {
6002ec34544SRui Ueyama   if (consume("SORT") || consume("SORT_BY_NAME"))
6012ec34544SRui Ueyama     return SortSectionPolicy::Name;
6022ec34544SRui Ueyama   if (consume("SORT_BY_ALIGNMENT"))
6032ec34544SRui Ueyama     return SortSectionPolicy::Alignment;
6042ec34544SRui Ueyama   if (consume("SORT_BY_INIT_PRIORITY"))
6052ec34544SRui Ueyama     return SortSectionPolicy::Priority;
6062ec34544SRui Ueyama   if (consume("SORT_NONE"))
6072ec34544SRui Ueyama     return SortSectionPolicy::None;
6082ec34544SRui Ueyama   return SortSectionPolicy::Default;
6092ec34544SRui Ueyama }
6102ec34544SRui Ueyama 
61103fc8d1eSRui Ueyama // Reads SECTIONS command contents in the following form:
61203fc8d1eSRui Ueyama //
61303fc8d1eSRui Ueyama // <contents> ::= <elem>*
61403fc8d1eSRui Ueyama // <elem>     ::= <exclude>? <glob-pattern>
61503fc8d1eSRui Ueyama // <exclude>  ::= "EXCLUDE_FILE" "(" <glob-pattern>+ ")"
61603fc8d1eSRui Ueyama //
61703fc8d1eSRui Ueyama // For example,
61803fc8d1eSRui Ueyama //
61903fc8d1eSRui Ueyama // *(.foo EXCLUDE_FILE (a.o) .bar EXCLUDE_FILE (b.o) .baz)
62003fc8d1eSRui Ueyama //
62103fc8d1eSRui Ueyama // is parsed as ".foo", ".bar" with "a.o", and ".baz" with "b.o".
62203fc8d1eSRui Ueyama // The semantics of that is section .foo in any file, section .bar in
62303fc8d1eSRui Ueyama // any file but a.o, and section .baz in any file but b.o.
6242ec34544SRui Ueyama std::vector<SectionPattern> ScriptParser::readInputSectionsList() {
6252ec34544SRui Ueyama   std::vector<SectionPattern> Ret;
626b8a59c8aSBob Haarman   while (!errorCount() && peek() != ")") {
6272ec34544SRui Ueyama     StringMatcher ExcludeFilePat;
6282ec34544SRui Ueyama     if (consume("EXCLUDE_FILE")) {
6292ec34544SRui Ueyama       expect("(");
6302ec34544SRui Ueyama       ExcludeFilePat = readFilePatterns();
6312ec34544SRui Ueyama     }
6322ec34544SRui Ueyama 
6332ec34544SRui Ueyama     std::vector<StringRef> V;
634b8a59c8aSBob Haarman     while (!errorCount() && peek() != ")" && peek() != "EXCLUDE_FILE")
6352ec34544SRui Ueyama       V.push_back(next());
6362ec34544SRui Ueyama 
6372ec34544SRui Ueyama     if (!V.empty())
6382ec34544SRui Ueyama       Ret.push_back({std::move(ExcludeFilePat), StringMatcher(V)});
6392ec34544SRui Ueyama     else
6402ec34544SRui Ueyama       setError("section pattern is expected");
6412ec34544SRui Ueyama   }
6422ec34544SRui Ueyama   return Ret;
6432ec34544SRui Ueyama }
6442ec34544SRui Ueyama 
6452ec34544SRui Ueyama // Reads contents of "SECTIONS" directive. That directive contains a
6462ec34544SRui Ueyama // list of glob patterns for input sections. The grammar is as follows.
6472ec34544SRui Ueyama //
6482ec34544SRui Ueyama // <patterns> ::= <section-list>
6492ec34544SRui Ueyama //              | <sort> "(" <section-list> ")"
6502ec34544SRui Ueyama //              | <sort> "(" <sort> "(" <section-list> ")" ")"
6512ec34544SRui Ueyama //
6522ec34544SRui Ueyama // <sort>     ::= "SORT" | "SORT_BY_NAME" | "SORT_BY_ALIGNMENT"
6532ec34544SRui Ueyama //              | "SORT_BY_INIT_PRIORITY" | "SORT_NONE"
6542ec34544SRui Ueyama //
6552ec34544SRui Ueyama // <section-list> is parsed by readInputSectionsList().
6562ec34544SRui Ueyama InputSectionDescription *
6572ec34544SRui Ueyama ScriptParser::readInputSectionRules(StringRef FilePattern) {
6582ec34544SRui Ueyama   auto *Cmd = make<InputSectionDescription>(FilePattern);
6592ec34544SRui Ueyama   expect("(");
6602ec34544SRui Ueyama 
661b8a59c8aSBob Haarman   while (!errorCount() && !consume(")")) {
6622ec34544SRui Ueyama     SortSectionPolicy Outer = readSortKind();
6632ec34544SRui Ueyama     SortSectionPolicy Inner = SortSectionPolicy::Default;
6642ec34544SRui Ueyama     std::vector<SectionPattern> V;
6652ec34544SRui Ueyama     if (Outer != SortSectionPolicy::Default) {
6662ec34544SRui Ueyama       expect("(");
6672ec34544SRui Ueyama       Inner = readSortKind();
6682ec34544SRui Ueyama       if (Inner != SortSectionPolicy::Default) {
6692ec34544SRui Ueyama         expect("(");
6702ec34544SRui Ueyama         V = readInputSectionsList();
6712ec34544SRui Ueyama         expect(")");
6722ec34544SRui Ueyama       } else {
6732ec34544SRui Ueyama         V = readInputSectionsList();
6742ec34544SRui Ueyama       }
6752ec34544SRui Ueyama       expect(")");
6762ec34544SRui Ueyama     } else {
6772ec34544SRui Ueyama       V = readInputSectionsList();
6782ec34544SRui Ueyama     }
6792ec34544SRui Ueyama 
6802ec34544SRui Ueyama     for (SectionPattern &Pat : V) {
6812ec34544SRui Ueyama       Pat.SortInner = Inner;
6822ec34544SRui Ueyama       Pat.SortOuter = Outer;
6832ec34544SRui Ueyama     }
6842ec34544SRui Ueyama 
6852ec34544SRui Ueyama     std::move(V.begin(), V.end(), std::back_inserter(Cmd->SectionPatterns));
6862ec34544SRui Ueyama   }
6872ec34544SRui Ueyama   return Cmd;
6882ec34544SRui Ueyama }
6892ec34544SRui Ueyama 
6902ec34544SRui Ueyama InputSectionDescription *
6912ec34544SRui Ueyama ScriptParser::readInputSectionDescription(StringRef Tok) {
6922ec34544SRui Ueyama   // Input section wildcard can be surrounded by KEEP.
6932ec34544SRui Ueyama   // https://sourceware.org/binutils/docs/ld/Input-Section-Keep.html#Input-Section-Keep
6942ec34544SRui Ueyama   if (Tok == "KEEP") {
6952ec34544SRui Ueyama     expect("(");
6962ec34544SRui Ueyama     StringRef FilePattern = next();
6972ec34544SRui Ueyama     InputSectionDescription *Cmd = readInputSectionRules(FilePattern);
6982ec34544SRui Ueyama     expect(")");
699ac27de9dSRui Ueyama     Script->KeptSections.push_back(Cmd);
7002ec34544SRui Ueyama     return Cmd;
7012ec34544SRui Ueyama   }
7022ec34544SRui Ueyama   return readInputSectionRules(Tok);
7032ec34544SRui Ueyama }
7042ec34544SRui Ueyama 
7052ec34544SRui Ueyama void ScriptParser::readSort() {
7062ec34544SRui Ueyama   expect("(");
7072ec34544SRui Ueyama   expect("CONSTRUCTORS");
7082ec34544SRui Ueyama   expect(")");
7092ec34544SRui Ueyama }
7102ec34544SRui Ueyama 
711d30a78b3SGeorge Rimar Expr ScriptParser::readAssert() {
7122ec34544SRui Ueyama   expect("(");
7132ec34544SRui Ueyama   Expr E = readExpr();
7142ec34544SRui Ueyama   expect(",");
7152ec34544SRui Ueyama   StringRef Msg = unquote(next());
7162ec34544SRui Ueyama   expect(")");
717b579c439SRui Ueyama 
7182ec34544SRui Ueyama   return [=] {
7192ec34544SRui Ueyama     if (!E().getValue())
7202ec34544SRui Ueyama       error(Msg);
7212ec34544SRui Ueyama     return Script->getDot();
7222ec34544SRui Ueyama   };
7232ec34544SRui Ueyama }
7242ec34544SRui Ueyama 
7252ec34544SRui Ueyama // Reads a FILL(expr) command. We handle the FILL command as an
7262ec34544SRui Ueyama // alias for =fillexp section attribute, which is different from
7272ec34544SRui Ueyama // what GNU linkers do.
7282ec34544SRui Ueyama // https://sourceware.org/binutils/docs/ld/Output-Section-Data.html
729b0486051SSimon Atanasyan std::array<uint8_t, 4> ScriptParser::readFill() {
7302ec34544SRui Ueyama   expect("(");
731b0486051SSimon Atanasyan   std::array<uint8_t, 4> V = parseFill(next());
7322ec34544SRui Ueyama   expect(")");
7332ec34544SRui Ueyama   return V;
7342ec34544SRui Ueyama }
7352ec34544SRui Ueyama 
736a46d08ebSGeorge Rimar // Tries to read the special directive for an output section definition which
737a46d08ebSGeorge Rimar // can be one of following: "(NOLOAD)", "(COPY)", "(INFO)" or "(OVERLAY)".
738a46d08ebSGeorge Rimar // Tok1 and Tok2 are next 2 tokens peeked. See comment for readSectionAddressType below.
739a46d08ebSGeorge Rimar bool ScriptParser::readSectionDirective(OutputSection *Cmd, StringRef Tok1, StringRef Tok2) {
740a46d08ebSGeorge Rimar   if (Tok1 != "(")
741a46d08ebSGeorge Rimar     return false;
742a46d08ebSGeorge Rimar   if (Tok2 != "NOLOAD" && Tok2 != "COPY" && Tok2 != "INFO" && Tok2 != "OVERLAY")
743a46d08ebSGeorge Rimar     return false;
744a46d08ebSGeorge Rimar 
745a46d08ebSGeorge Rimar   expect("(");
746a46d08ebSGeorge Rimar   if (consume("NOLOAD")) {
747a46d08ebSGeorge Rimar     Cmd->Noload = true;
748a46d08ebSGeorge Rimar   } else {
749a46d08ebSGeorge Rimar     skip(); // This is "COPY", "INFO" or "OVERLAY".
750a46d08ebSGeorge Rimar     Cmd->NonAlloc = true;
751a46d08ebSGeorge Rimar   }
752a46d08ebSGeorge Rimar   expect(")");
753a46d08ebSGeorge Rimar   return true;
754a46d08ebSGeorge Rimar }
755a46d08ebSGeorge Rimar 
7561c08e9f5SGeorge Rimar // Reads an expression and/or the special directive for an output
7571c08e9f5SGeorge Rimar // section definition. Directive is one of following: "(NOLOAD)",
7581c08e9f5SGeorge Rimar // "(COPY)", "(INFO)" or "(OVERLAY)".
7593271d370SRui Ueyama //
7603271d370SRui Ueyama // An output section name can be followed by an address expression
7611c08e9f5SGeorge Rimar // and/or directive. This grammar is not LL(1) because "(" can be
76297f4d158SGeorge Rimar // interpreted as either the beginning of some expression or beginning
7631c08e9f5SGeorge Rimar // of directive.
7643271d370SRui Ueyama //
765b579c439SRui Ueyama // https://sourceware.org/binutils/docs/ld/Output-Section-Address.html
766fbb0463fSGeorge Rimar // https://sourceware.org/binutils/docs/ld/Output-Section-Type.html
7678c022ca7SRafael Espindola void ScriptParser::readSectionAddressType(OutputSection *Cmd) {
768a46d08ebSGeorge Rimar   if (readSectionDirective(Cmd, peek(), peek2()))
7693271d370SRui Ueyama     return;
7703271d370SRui Ueyama 
771a46d08ebSGeorge Rimar   Cmd->AddrExpr = readExpr();
772a46d08ebSGeorge Rimar   if (peek() == "(" && !readSectionDirective(Cmd, "(", peek2()))
773a46d08ebSGeorge Rimar     setError("unknown section directive: " + peek2());
774fbb0463fSGeorge Rimar }
775fbb0463fSGeorge Rimar 
776f22ec9ddSGeorge Rimar static Expr checkAlignment(Expr E, std::string &Loc) {
777f22ec9ddSGeorge Rimar   return [=] {
778f22ec9ddSGeorge Rimar     uint64_t Alignment = std::max((uint64_t)1, E().getValue());
779f22ec9ddSGeorge Rimar     if (!isPowerOf2_64(Alignment)) {
780f22ec9ddSGeorge Rimar       error(Loc + ": alignment must be power of 2");
781f22ec9ddSGeorge Rimar       return (uint64_t)1; // Return a dummy value.
782f22ec9ddSGeorge Rimar     }
783f22ec9ddSGeorge Rimar     return Alignment;
784f22ec9ddSGeorge Rimar   };
785f22ec9ddSGeorge Rimar }
786f22ec9ddSGeorge Rimar 
787a582419aSGeorge Rimar OutputSection *ScriptParser::readOverlaySectionDescription() {
788a582419aSGeorge Rimar   OutputSection *Cmd =
789a582419aSGeorge Rimar       Script->createOutputSection(next(), getCurrentLocation());
790a582419aSGeorge Rimar   Cmd->InOverlay = true;
791a582419aSGeorge Rimar   expect("{");
792a582419aSGeorge Rimar   while (!errorCount() && !consume("}"))
793a582419aSGeorge Rimar     Cmd->SectionCommands.push_back(readInputSectionRules(next()));
794a582419aSGeorge Rimar   Cmd->Phdrs = readOutputSectionPhdrs();
795a582419aSGeorge Rimar   return Cmd;
796a582419aSGeorge Rimar }
797a582419aSGeorge Rimar 
7988c022ca7SRafael Espindola OutputSection *ScriptParser::readOutputSectionDescription(StringRef OutSec) {
7998c022ca7SRafael Espindola   OutputSection *Cmd =
8008c022ca7SRafael Espindola       Script->createOutputSection(OutSec, getCurrentLocation());
8013271d370SRui Ueyama 
802c4df670dSGeorge Rimar   size_t SymbolsReferenced = Script->ReferencedSymbols.size();
803c4df670dSGeorge Rimar 
8043271d370SRui Ueyama   if (peek() != ":")
8053271d370SRui Ueyama     readSectionAddressType(Cmd);
8062ec34544SRui Ueyama   expect(":");
8072ec34544SRui Ueyama 
808f22ec9ddSGeorge Rimar   std::string Location = getCurrentLocation();
8092ec34544SRui Ueyama   if (consume("AT"))
8102ec34544SRui Ueyama     Cmd->LMAExpr = readParenExpr();
8112ec34544SRui Ueyama   if (consume("ALIGN"))
812f22ec9ddSGeorge Rimar     Cmd->AlignExpr = checkAlignment(readParenExpr(), Location);
8132ec34544SRui Ueyama   if (consume("SUBALIGN"))
814f22ec9ddSGeorge Rimar     Cmd->SubalignExpr = checkAlignment(readParenExpr(), Location);
8152ec34544SRui Ueyama 
8162ec34544SRui Ueyama   // Parse constraints.
8172ec34544SRui Ueyama   if (consume("ONLY_IF_RO"))
8182ec34544SRui Ueyama     Cmd->Constraint = ConstraintKind::ReadOnly;
8192ec34544SRui Ueyama   if (consume("ONLY_IF_RW"))
8202ec34544SRui Ueyama     Cmd->Constraint = ConstraintKind::ReadWrite;
8212ec34544SRui Ueyama   expect("{");
8222ec34544SRui Ueyama 
823b8a59c8aSBob Haarman   while (!errorCount() && !consume("}")) {
8242ec34544SRui Ueyama     StringRef Tok = next();
8252ec34544SRui Ueyama     if (Tok == ";") {
8262ec34544SRui Ueyama       // Empty commands are allowed. Do nothing here.
827d30a78b3SGeorge Rimar     } else if (SymbolAssignment *Assign = readAssignment(Tok)) {
8286b394caaSRui Ueyama       Cmd->SectionCommands.push_back(Assign);
829f0403c60SRui Ueyama     } else if (ByteCommand *Data = readByteCommand(Tok)) {
8306b394caaSRui Ueyama       Cmd->SectionCommands.push_back(Data);
8312ec34544SRui Ueyama     } else if (Tok == "CONSTRUCTORS") {
8322ec34544SRui Ueyama       // CONSTRUCTORS is a keyword to make the linker recognize C++ ctors/dtors
8332ec34544SRui Ueyama       // by name. This is for very old file formats such as ECOFF/XCOFF.
8342ec34544SRui Ueyama       // For ELF, we should ignore.
8352ec34544SRui Ueyama     } else if (Tok == "FILL") {
8362ec34544SRui Ueyama       Cmd->Filler = readFill();
8372ec34544SRui Ueyama     } else if (Tok == "SORT") {
8382ec34544SRui Ueyama       readSort();
8392e9d40d5SRui Ueyama     } else if (Tok == "INCLUDE") {
8402e9d40d5SRui Ueyama       readInclude();
8412ec34544SRui Ueyama     } else if (peek() == "(") {
8426b394caaSRui Ueyama       Cmd->SectionCommands.push_back(readInputSectionDescription(Tok));
8432ec34544SRui Ueyama     } else {
844f49fe218SGeorge Rimar       // We have a file name and no input sections description. It is not a
845f49fe218SGeorge Rimar       // commonly used syntax, but still acceptable. In that case, all sections
846f49fe218SGeorge Rimar       // from the file will be included.
847e1fd3f94SGeorge Rimar       auto *ISD = make<InputSectionDescription>(Tok);
848f49fe218SGeorge Rimar       ISD->SectionPatterns.push_back({{}, StringMatcher({"*"})});
849f49fe218SGeorge Rimar       Cmd->SectionCommands.push_back(ISD);
8502ec34544SRui Ueyama     }
8512ec34544SRui Ueyama   }
8522ec34544SRui Ueyama 
8532ec34544SRui Ueyama   if (consume(">"))
8542ec34544SRui Ueyama     Cmd->MemoryRegionName = next();
8552ec34544SRui Ueyama 
8565d01a8beSGeorge Rimar   if (consume("AT")) {
8575d01a8beSGeorge Rimar     expect(">");
8585d01a8beSGeorge Rimar     Cmd->LMARegionName = next();
8595d01a8beSGeorge Rimar   }
8605d01a8beSGeorge Rimar 
8615d01a8beSGeorge Rimar   if (Cmd->LMAExpr && !Cmd->LMARegionName.empty())
8625d01a8beSGeorge Rimar     error("section can't have both LMA and a load region");
8635d01a8beSGeorge Rimar 
8642ec34544SRui Ueyama   Cmd->Phdrs = readOutputSectionPhdrs();
8652ec34544SRui Ueyama 
8662ec34544SRui Ueyama   if (consume("="))
8678acbf1ccSRui Ueyama     Cmd->Filler = parseFill(next());
8682ec34544SRui Ueyama   else if (peek().startswith("="))
8698acbf1ccSRui Ueyama     Cmd->Filler = parseFill(next().drop_front());
8702ec34544SRui Ueyama 
8712ec34544SRui Ueyama   // Consume optional comma following output section command.
8722ec34544SRui Ueyama   consume(",");
8732ec34544SRui Ueyama 
874c4df670dSGeorge Rimar   if (Script->ReferencedSymbols.size() > SymbolsReferenced)
875c4df670dSGeorge Rimar     Cmd->ExpressionsUseSymbols = true;
8762ec34544SRui Ueyama   return Cmd;
8772ec34544SRui Ueyama }
8782ec34544SRui Ueyama 
8798acbf1ccSRui Ueyama // Parses a given string as a octal/decimal/hexadecimal number and
8808acbf1ccSRui Ueyama // returns it as a big-endian number. Used for `=<fillexp>`.
8812ec34544SRui Ueyama // https://sourceware.org/binutils/docs/ld/Output-Section-Fill.html
8822ec34544SRui Ueyama //
8838acbf1ccSRui Ueyama // When reading a hexstring, ld.bfd handles it as a blob of arbitrary
8848acbf1ccSRui Ueyama // size, while ld.gold always handles it as a 32-bit big-endian number.
8858acbf1ccSRui Ueyama // We are compatible with ld.gold because it's easier to implement.
886b0486051SSimon Atanasyan std::array<uint8_t, 4> ScriptParser::parseFill(StringRef Tok) {
887b58079d4SRui Ueyama   uint32_t V = 0;
888ab94768cSGeorge Rimar   if (!to_integer(Tok, V))
8892ec34544SRui Ueyama     setError("invalid filler expression: " + Tok);
890b58079d4SRui Ueyama 
891b0486051SSimon Atanasyan   std::array<uint8_t, 4> Buf;
892b0486051SSimon Atanasyan   write32be(Buf.data(), V);
893b58079d4SRui Ueyama   return Buf;
8942ec34544SRui Ueyama }
8952ec34544SRui Ueyama 
8962ec34544SRui Ueyama SymbolAssignment *ScriptParser::readProvideHidden(bool Provide, bool Hidden) {
8972ec34544SRui Ueyama   expect("(");
898d30a78b3SGeorge Rimar   SymbolAssignment *Cmd = readSymbolAssignment(next());
8992ec34544SRui Ueyama   Cmd->Provide = Provide;
9002ec34544SRui Ueyama   Cmd->Hidden = Hidden;
9012ec34544SRui Ueyama   expect(")");
9022ec34544SRui Ueyama   return Cmd;
9032ec34544SRui Ueyama }
9042ec34544SRui Ueyama 
905d30a78b3SGeorge Rimar SymbolAssignment *ScriptParser::readAssignment(StringRef Tok) {
906d30a78b3SGeorge Rimar   // Assert expression returns Dot, so this is equal to ".=."
907d30a78b3SGeorge Rimar   if (Tok == "ASSERT")
908d30a78b3SGeorge Rimar     return make<SymbolAssignment>(".", readAssert(), getCurrentLocation());
909d30a78b3SGeorge Rimar 
910e88b76a9SGeorge Rimar   size_t OldPos = Pos;
9112ec34544SRui Ueyama   SymbolAssignment *Cmd = nullptr;
912e88b76a9SGeorge Rimar   if (peek() == "=" || peek() == "+=")
913d30a78b3SGeorge Rimar     Cmd = readSymbolAssignment(Tok);
914e88b76a9SGeorge Rimar   else if (Tok == "PROVIDE")
9152ec34544SRui Ueyama     Cmd = readProvideHidden(true, false);
916e88b76a9SGeorge Rimar   else if (Tok == "HIDDEN")
9172ec34544SRui Ueyama     Cmd = readProvideHidden(false, true);
918e88b76a9SGeorge Rimar   else if (Tok == "PROVIDE_HIDDEN")
9192ec34544SRui Ueyama     Cmd = readProvideHidden(true, true);
920e88b76a9SGeorge Rimar 
921e88b76a9SGeorge Rimar   if (Cmd) {
922e88b76a9SGeorge Rimar     Cmd->CommandString =
923e88b76a9SGeorge Rimar         Tok.str() + " " +
924e88b76a9SGeorge Rimar         llvm::join(Tokens.begin() + OldPos, Tokens.begin() + Pos, " ");
925e88b76a9SGeorge Rimar     expect(";");
9262ec34544SRui Ueyama   }
9272ec34544SRui Ueyama   return Cmd;
9282ec34544SRui Ueyama }
9292ec34544SRui Ueyama 
930d30a78b3SGeorge Rimar SymbolAssignment *ScriptParser::readSymbolAssignment(StringRef Name) {
9312ec34544SRui Ueyama   StringRef Op = next();
9322ec34544SRui Ueyama   assert(Op == "=" || Op == "+=");
9332ec34544SRui Ueyama   Expr E = readExpr();
9342ec34544SRui Ueyama   if (Op == "+=") {
9352ec34544SRui Ueyama     std::string Loc = getCurrentLocation();
936722221f5SRui Ueyama     E = [=] { return add(Script->getSymbolValue(Name, Loc), E()); };
9372ec34544SRui Ueyama   }
938e88b76a9SGeorge Rimar   return make<SymbolAssignment>(Name, E, getCurrentLocation());
9392ec34544SRui Ueyama }
9402ec34544SRui Ueyama 
9412ec34544SRui Ueyama // This is an operator-precedence parser to parse a linker
9422ec34544SRui Ueyama // script expression.
9432ec34544SRui Ueyama Expr ScriptParser::readExpr() {
9442ec34544SRui Ueyama   // Our lexer is context-aware. Set the in-expression bit so that
9452ec34544SRui Ueyama   // they apply different tokenization rules.
9462ec34544SRui Ueyama   bool Orig = InExpr;
9472ec34544SRui Ueyama   InExpr = true;
9482ec34544SRui Ueyama   Expr E = readExpr1(readPrimary(), 0);
9492ec34544SRui Ueyama   InExpr = Orig;
9502ec34544SRui Ueyama   return E;
9512ec34544SRui Ueyama }
9522ec34544SRui Ueyama 
9537b91e213SGeorge Rimar Expr ScriptParser::combine(StringRef Op, Expr L, Expr R) {
9542ec34544SRui Ueyama   if (Op == "+")
9552ec34544SRui Ueyama     return [=] { return add(L(), R()); };
9562ec34544SRui Ueyama   if (Op == "-")
9572ec34544SRui Ueyama     return [=] { return sub(L(), R()); };
958b579c439SRui Ueyama   if (Op == "*")
9591d20222aSRui Ueyama     return [=] { return L().getValue() * R().getValue(); };
9607b91e213SGeorge Rimar   if (Op == "/") {
9617b91e213SGeorge Rimar     std::string Loc = getCurrentLocation();
9627b91e213SGeorge Rimar     return [=]() -> uint64_t {
9637b91e213SGeorge Rimar       if (uint64_t RV = R().getValue())
9647b91e213SGeorge Rimar         return L().getValue() / RV;
9657b91e213SGeorge Rimar       error(Loc + ": division by zero");
966067617f9SRui Ueyama       return 0;
9677b91e213SGeorge Rimar     };
9687b91e213SGeorge Rimar   }
9697b91e213SGeorge Rimar   if (Op == "%") {
9707b91e213SGeorge Rimar     std::string Loc = getCurrentLocation();
9717b91e213SGeorge Rimar     return [=]() -> uint64_t {
9727b91e213SGeorge Rimar       if (uint64_t RV = R().getValue())
9737b91e213SGeorge Rimar         return L().getValue() % RV;
9747b91e213SGeorge Rimar       error(Loc + ": modulo by zero");
975067617f9SRui Ueyama       return 0;
9767b91e213SGeorge Rimar     };
9777b91e213SGeorge Rimar   }
9782ec34544SRui Ueyama   if (Op == "<<")
9797e915511SRui Ueyama     return [=] { return L().getValue() << R().getValue(); };
9802ec34544SRui Ueyama   if (Op == ">>")
9817e915511SRui Ueyama     return [=] { return L().getValue() >> R().getValue(); };
9822ec34544SRui Ueyama   if (Op == "<")
9832ec34544SRui Ueyama     return [=] { return L().getValue() < R().getValue(); };
9842ec34544SRui Ueyama   if (Op == ">")
9852ec34544SRui Ueyama     return [=] { return L().getValue() > R().getValue(); };
9862ec34544SRui Ueyama   if (Op == ">=")
9872ec34544SRui Ueyama     return [=] { return L().getValue() >= R().getValue(); };
9882ec34544SRui Ueyama   if (Op == "<=")
9892ec34544SRui Ueyama     return [=] { return L().getValue() <= R().getValue(); };
9902ec34544SRui Ueyama   if (Op == "==")
9912ec34544SRui Ueyama     return [=] { return L().getValue() == R().getValue(); };
9922ec34544SRui Ueyama   if (Op == "!=")
9932ec34544SRui Ueyama     return [=] { return L().getValue() != R().getValue(); };
994a5005482SGeorge Rimar   if (Op == "||")
995a5005482SGeorge Rimar     return [=] { return L().getValue() || R().getValue(); };
996a5005482SGeorge Rimar   if (Op == "&&")
997a5005482SGeorge Rimar     return [=] { return L().getValue() && R().getValue(); };
9982ec34544SRui Ueyama   if (Op == "&")
9992ec34544SRui Ueyama     return [=] { return bitAnd(L(), R()); };
10002ec34544SRui Ueyama   if (Op == "|")
10012ec34544SRui Ueyama     return [=] { return bitOr(L(), R()); };
10022ec34544SRui Ueyama   llvm_unreachable("invalid operator");
10032ec34544SRui Ueyama }
10042ec34544SRui Ueyama 
10052ec34544SRui Ueyama // This is a part of the operator-precedence parser. This function
10062ec34544SRui Ueyama // assumes that the remaining token stream starts with an operator.
10072ec34544SRui Ueyama Expr ScriptParser::readExpr1(Expr Lhs, int MinPrec) {
1008b8a59c8aSBob Haarman   while (!atEOF() && !errorCount()) {
10092ec34544SRui Ueyama     // Read an operator and an expression.
10102ec34544SRui Ueyama     if (consume("?"))
10112ec34544SRui Ueyama       return readTernary(Lhs);
10122ec34544SRui Ueyama     StringRef Op1 = peek();
10132ec34544SRui Ueyama     if (precedence(Op1) < MinPrec)
10142ec34544SRui Ueyama       break;
10152ec34544SRui Ueyama     skip();
10162ec34544SRui Ueyama     Expr Rhs = readPrimary();
10172ec34544SRui Ueyama 
10182ec34544SRui Ueyama     // Evaluate the remaining part of the expression first if the
10192ec34544SRui Ueyama     // next operator has greater precedence than the previous one.
10202ec34544SRui Ueyama     // For example, if we have read "+" and "3", and if the next
10212ec34544SRui Ueyama     // operator is "*", then we'll evaluate 3 * ... part first.
10222ec34544SRui Ueyama     while (!atEOF()) {
10232ec34544SRui Ueyama       StringRef Op2 = peek();
10242ec34544SRui Ueyama       if (precedence(Op2) <= precedence(Op1))
10252ec34544SRui Ueyama         break;
10262ec34544SRui Ueyama       Rhs = readExpr1(Rhs, precedence(Op2));
10272ec34544SRui Ueyama     }
10282ec34544SRui Ueyama 
10292ec34544SRui Ueyama     Lhs = combine(Op1, Lhs, Rhs);
10302ec34544SRui Ueyama   }
10312ec34544SRui Ueyama   return Lhs;
10322ec34544SRui Ueyama }
10332ec34544SRui Ueyama 
10345fb17128SGeorge Rimar Expr ScriptParser::getPageSize() {
10355fb17128SGeorge Rimar   std::string Location = getCurrentLocation();
10365fb17128SGeorge Rimar   return [=]() -> uint64_t {
10375fb17128SGeorge Rimar     if (Target)
10382ec34544SRui Ueyama       return Target->PageSize;
10395fb17128SGeorge Rimar     error(Location + ": unable to calculate page size");
10405fb17128SGeorge Rimar     return 4096; // Return a dummy value.
10415fb17128SGeorge Rimar   };
10425fb17128SGeorge Rimar }
10435fb17128SGeorge Rimar 
10445fb17128SGeorge Rimar Expr ScriptParser::readConstant() {
10455fb17128SGeorge Rimar   StringRef S = readParenLiteral();
10465fb17128SGeorge Rimar   if (S == "COMMONPAGESIZE")
10475fb17128SGeorge Rimar     return getPageSize();
10482ec34544SRui Ueyama   if (S == "MAXPAGESIZE")
10495fb17128SGeorge Rimar     return [] { return Config->MaxPageSize; };
10505fb17128SGeorge Rimar   setError("unknown constant: " + S);
1051b068b037SGeorge Rimar   return [] { return 0; };
10522ec34544SRui Ueyama }
10532ec34544SRui Ueyama 
10545c65088fSRui Ueyama // Parses Tok as an integer. It recognizes hexadecimal (prefixed with
10555c65088fSRui Ueyama // "0x" or suffixed with "H") and decimal numbers. Decimal numbers may
10565c65088fSRui Ueyama // have "K" (Ki) or "M" (Mi) suffixes.
10575c65088fSRui Ueyama static Optional<uint64_t> parseInt(StringRef Tok) {
10582ec34544SRui Ueyama   // Hexadecimal
10595c65088fSRui Ueyama   uint64_t Val;
10604092016bSRui Ueyama   if (Tok.startswith_lower("0x")) {
10614092016bSRui Ueyama     if (!to_integer(Tok.substr(2), Val, 16))
10624092016bSRui Ueyama       return None;
10635c65088fSRui Ueyama     return Val;
10644092016bSRui Ueyama   }
10654092016bSRui Ueyama   if (Tok.endswith_lower("H")) {
10664092016bSRui Ueyama     if (!to_integer(Tok.drop_back(), Val, 16))
10674092016bSRui Ueyama       return None;
10685c65088fSRui Ueyama     return Val;
10694092016bSRui Ueyama   }
10702ec34544SRui Ueyama 
10712ec34544SRui Ueyama   // Decimal
10722ec34544SRui Ueyama   if (Tok.endswith_lower("K")) {
1073ab94768cSGeorge Rimar     if (!to_integer(Tok.drop_back(), Val, 10))
10745c65088fSRui Ueyama       return None;
10755c65088fSRui Ueyama     return Val * 1024;
10762ec34544SRui Ueyama   }
10775c65088fSRui Ueyama   if (Tok.endswith_lower("M")) {
1078ab94768cSGeorge Rimar     if (!to_integer(Tok.drop_back(), Val, 10))
10795c65088fSRui Ueyama       return None;
10805c65088fSRui Ueyama     return Val * 1024 * 1024;
10815c65088fSRui Ueyama   }
1082ab94768cSGeorge Rimar   if (!to_integer(Tok, Val, 10))
10835c65088fSRui Ueyama     return None;
10845c65088fSRui Ueyama   return Val;
10852ec34544SRui Ueyama }
10862ec34544SRui Ueyama 
1087f0403c60SRui Ueyama ByteCommand *ScriptParser::readByteCommand(StringRef Tok) {
1088b579c439SRui Ueyama   int Size = StringSwitch<int>(Tok)
10892ec34544SRui Ueyama                  .Case("BYTE", 1)
10902ec34544SRui Ueyama                  .Case("SHORT", 2)
10912ec34544SRui Ueyama                  .Case("LONG", 4)
10922ec34544SRui Ueyama                  .Case("QUAD", 8)
10932ec34544SRui Ueyama                  .Default(-1);
10942ec34544SRui Ueyama   if (Size == -1)
10952ec34544SRui Ueyama     return nullptr;
109684bcabcbSGeorge Rimar 
109784bcabcbSGeorge Rimar   size_t OldPos = Pos;
109884bcabcbSGeorge Rimar   Expr E = readParenExpr();
109984bcabcbSGeorge Rimar   std::string CommandString =
110084bcabcbSGeorge Rimar       Tok.str() + " " +
110184bcabcbSGeorge Rimar       llvm::join(Tokens.begin() + OldPos, Tokens.begin() + Pos, " ");
110284bcabcbSGeorge Rimar   return make<ByteCommand>(E, Size, CommandString);
11032ec34544SRui Ueyama }
11042ec34544SRui Ueyama 
11052ec34544SRui Ueyama StringRef ScriptParser::readParenLiteral() {
11062ec34544SRui Ueyama   expect("(");
11075e9c7762SRafael Espindola   bool Orig = InExpr;
11085e9c7762SRafael Espindola   InExpr = false;
11092ec34544SRui Ueyama   StringRef Tok = next();
11105e9c7762SRafael Espindola   InExpr = Orig;
11112ec34544SRui Ueyama   expect(")");
11122ec34544SRui Ueyama   return Tok;
11132ec34544SRui Ueyama }
11142ec34544SRui Ueyama 
1115617e2f98SRui Ueyama static void checkIfExists(OutputSection *Cmd, StringRef Location) {
111605c4f67cSRafael Espindola   if (Cmd->Location.empty() && Script->ErrorOnMissingSection)
111705c4f67cSRafael Espindola     error(Location + ": undefined section " + Cmd->Name);
111805c4f67cSRafael Espindola }
111905c4f67cSRafael Espindola 
11202ec34544SRui Ueyama Expr ScriptParser::readPrimary() {
11212ec34544SRui Ueyama   if (peek() == "(")
11222ec34544SRui Ueyama     return readParenExpr();
11232ec34544SRui Ueyama 
11245c65088fSRui Ueyama   if (consume("~")) {
11252ec34544SRui Ueyama     Expr E = readPrimary();
1126b2fb84a1SRui Ueyama     return [=] { return ~E().getValue(); };
11272ec34544SRui Ueyama   }
11286f1d954eSHafiz Abid Qadeer   if (consume("!")) {
11296f1d954eSHafiz Abid Qadeer     Expr E = readPrimary();
11306f1d954eSHafiz Abid Qadeer     return [=] { return !E().getValue(); };
11316f1d954eSHafiz Abid Qadeer   }
11325c65088fSRui Ueyama   if (consume("-")) {
11332ec34544SRui Ueyama     Expr E = readPrimary();
1134b2fb84a1SRui Ueyama     return [=] { return -E().getValue(); };
11352ec34544SRui Ueyama   }
11362ec34544SRui Ueyama 
11375c65088fSRui Ueyama   StringRef Tok = next();
11385c65088fSRui Ueyama   std::string Location = getCurrentLocation();
11395c65088fSRui Ueyama 
11402ec34544SRui Ueyama   // Built-in functions are parsed here.
11412ec34544SRui Ueyama   // https://sourceware.org/binutils/docs/ld/Builtin-Functions.html.
11422ec34544SRui Ueyama   if (Tok == "ABSOLUTE") {
11432ec34544SRui Ueyama     Expr Inner = readParenExpr();
11442ec34544SRui Ueyama     return [=] {
11452ec34544SRui Ueyama       ExprValue I = Inner();
11462ec34544SRui Ueyama       I.ForceAbsolute = true;
11472ec34544SRui Ueyama       return I;
11482ec34544SRui Ueyama     };
11492ec34544SRui Ueyama   }
11502ec34544SRui Ueyama   if (Tok == "ADDR") {
11512ec34544SRui Ueyama     StringRef Name = readParenLiteral();
11524fbe3518SRui Ueyama     OutputSection *Sec = Script->getOrCreateOutputSection(Name);
115341c7ab4aSGeorge Rimar     return [=]() -> ExprValue {
11544fbe3518SRui Ueyama       checkIfExists(Sec, Location);
11554fbe3518SRui Ueyama       return {Sec, false, 0, Location};
115641c7ab4aSGeorge Rimar     };
11572ec34544SRui Ueyama   }
11582ec34544SRui Ueyama   if (Tok == "ALIGN") {
11592ec34544SRui Ueyama     expect("(");
11602ec34544SRui Ueyama     Expr E = readExpr();
1161f22ec9ddSGeorge Rimar     if (consume(")")) {
1162f22ec9ddSGeorge Rimar       E = checkAlignment(E, Location);
1163f22ec9ddSGeorge Rimar       return [=] { return alignTo(Script->getDot(), E().getValue()); };
1164f22ec9ddSGeorge Rimar     }
1165b579c439SRui Ueyama     expect(",");
1166f22ec9ddSGeorge Rimar     Expr E2 = checkAlignment(readExpr(), Location);
11672ec34544SRui Ueyama     expect(")");
11683c6de1a6SPetr Hosek     return [=] {
11693c6de1a6SPetr Hosek       ExprValue V = E();
1170f22ec9ddSGeorge Rimar       V.Alignment = E2().getValue();
11713c6de1a6SPetr Hosek       return V;
11723c6de1a6SPetr Hosek     };
11732ec34544SRui Ueyama   }
11742ec34544SRui Ueyama   if (Tok == "ALIGNOF") {
11752ec34544SRui Ueyama     StringRef Name = readParenLiteral();
11768c022ca7SRafael Espindola     OutputSection *Cmd = Script->getOrCreateOutputSection(Name);
1177617e2f98SRui Ueyama     return [=] {
1178617e2f98SRui Ueyama       checkIfExists(Cmd, Location);
1179617e2f98SRui Ueyama       return Cmd->Alignment;
1180617e2f98SRui Ueyama     };
11812ec34544SRui Ueyama   }
11822ec34544SRui Ueyama   if (Tok == "ASSERT")
1183d30a78b3SGeorge Rimar     return readAssert();
11845fb17128SGeorge Rimar   if (Tok == "CONSTANT")
11855fb17128SGeorge Rimar     return readConstant();
11862ec34544SRui Ueyama   if (Tok == "DATA_SEGMENT_ALIGN") {
11872ec34544SRui Ueyama     expect("(");
11882ec34544SRui Ueyama     Expr E = readExpr();
11892ec34544SRui Ueyama     expect(",");
11902ec34544SRui Ueyama     readExpr();
11912ec34544SRui Ueyama     expect(")");
119260833f6eSGeorge Rimar     return [=] {
119360833f6eSGeorge Rimar       return alignTo(Script->getDot(), std::max((uint64_t)1, E().getValue()));
119460833f6eSGeorge Rimar     };
11952ec34544SRui Ueyama   }
11962ec34544SRui Ueyama   if (Tok == "DATA_SEGMENT_END") {
11972ec34544SRui Ueyama     expect("(");
11982ec34544SRui Ueyama     expect(".");
11992ec34544SRui Ueyama     expect(")");
12002ec34544SRui Ueyama     return [] { return Script->getDot(); };
12012ec34544SRui Ueyama   }
12022ec34544SRui Ueyama   if (Tok == "DATA_SEGMENT_RELRO_END") {
12032ec34544SRui Ueyama     // GNU linkers implements more complicated logic to handle
12042ec34544SRui Ueyama     // DATA_SEGMENT_RELRO_END. We instead ignore the arguments and
12052ec34544SRui Ueyama     // just align to the next page boundary for simplicity.
12062ec34544SRui Ueyama     expect("(");
12072ec34544SRui Ueyama     readExpr();
12082ec34544SRui Ueyama     expect(",");
12092ec34544SRui Ueyama     readExpr();
12102ec34544SRui Ueyama     expect(")");
12115fb17128SGeorge Rimar     Expr E = getPageSize();
12125fb17128SGeorge Rimar     return [=] { return alignTo(Script->getDot(), E().getValue()); };
12132ec34544SRui Ueyama   }
12142ec34544SRui Ueyama   if (Tok == "DEFINED") {
12152ec34544SRui Ueyama     StringRef Name = readParenLiteral();
12169b18f50fSRui Ueyama     return [=] { return Symtab->find(Name) ? 1 : 0; };
12172ec34544SRui Ueyama   }
121891b95b61SRui Ueyama   if (Tok == "LENGTH") {
121991b95b61SRui Ueyama     StringRef Name = readParenLiteral();
1220b068b037SGeorge Rimar     if (Script->MemoryRegions.count(Name) == 0) {
122191b95b61SRui Ueyama       setError("memory region not defined: " + Name);
1222b068b037SGeorge Rimar       return [] { return 0; };
1223b068b037SGeorge Rimar     }
1224ac27de9dSRui Ueyama     return [=] { return Script->MemoryRegions[Name]->Length; };
122591b95b61SRui Ueyama   }
12262ec34544SRui Ueyama   if (Tok == "LOADADDR") {
12272ec34544SRui Ueyama     StringRef Name = readParenLiteral();
12288c022ca7SRafael Espindola     OutputSection *Cmd = Script->getOrCreateOutputSection(Name);
1229617e2f98SRui Ueyama     return [=] {
1230617e2f98SRui Ueyama       checkIfExists(Cmd, Location);
1231617e2f98SRui Ueyama       return Cmd->getLMA();
1232617e2f98SRui Ueyama     };
12332ec34544SRui Ueyama   }
1234fd11560fSGeorge Rimar   if (Tok == "MAX" || Tok == "MIN") {
1235fd11560fSGeorge Rimar     expect("(");
1236fd11560fSGeorge Rimar     Expr A = readExpr();
1237fd11560fSGeorge Rimar     expect(",");
1238fd11560fSGeorge Rimar     Expr B = readExpr();
1239fd11560fSGeorge Rimar     expect(")");
1240fd11560fSGeorge Rimar     if (Tok == "MIN")
1241fd11560fSGeorge Rimar       return [=] { return std::min(A().getValue(), B().getValue()); };
1242fd11560fSGeorge Rimar     return [=] { return std::max(A().getValue(), B().getValue()); };
1243fd11560fSGeorge Rimar   }
124491b95b61SRui Ueyama   if (Tok == "ORIGIN") {
124591b95b61SRui Ueyama     StringRef Name = readParenLiteral();
1246b068b037SGeorge Rimar     if (Script->MemoryRegions.count(Name) == 0) {
124791b95b61SRui Ueyama       setError("memory region not defined: " + Name);
1248b068b037SGeorge Rimar       return [] { return 0; };
1249b068b037SGeorge Rimar     }
1250ac27de9dSRui Ueyama     return [=] { return Script->MemoryRegions[Name]->Origin; };
125191b95b61SRui Ueyama   }
12522ec34544SRui Ueyama   if (Tok == "SEGMENT_START") {
12532ec34544SRui Ueyama     expect("(");
12542ec34544SRui Ueyama     skip();
12552ec34544SRui Ueyama     expect(",");
12562ec34544SRui Ueyama     Expr E = readExpr();
12572ec34544SRui Ueyama     expect(")");
12582ec34544SRui Ueyama     return [=] { return E(); };
12592ec34544SRui Ueyama   }
12602ec34544SRui Ueyama   if (Tok == "SIZEOF") {
12612ec34544SRui Ueyama     StringRef Name = readParenLiteral();
12628c022ca7SRafael Espindola     OutputSection *Cmd = Script->getOrCreateOutputSection(Name);
126305c4f67cSRafael Espindola     // Linker script does not create an output section if its content is empty.
126405c4f67cSRafael Espindola     // We want to allow SIZEOF(.foo) where .foo is a section which happened to
126505c4f67cSRafael Espindola     // be empty.
12668c022ca7SRafael Espindola     return [=] { return Cmd->Size; };
12672ec34544SRui Ueyama   }
12682ec34544SRui Ueyama   if (Tok == "SIZEOF_HEADERS")
12692ec34544SRui Ueyama     return [=] { return elf::getHeaderSize(); };
12702ec34544SRui Ueyama 
12714eb2eccbSRui Ueyama   // Tok is the dot.
12724eb2eccbSRui Ueyama   if (Tok == ".")
1273722221f5SRui Ueyama     return [=] { return Script->getSymbolValue(Tok, Location); };
12744eb2eccbSRui Ueyama 
12752ec34544SRui Ueyama   // Tok is a literal number.
12765c65088fSRui Ueyama   if (Optional<uint64_t> Val = parseInt(Tok))
12775c65088fSRui Ueyama     return [=] { return *Val; };
12782ec34544SRui Ueyama 
12792ec34544SRui Ueyama   // Tok is a symbol name.
12802ec34544SRui Ueyama   if (!isValidCIdentifier(Tok))
12812ec34544SRui Ueyama     setError("malformed number: " + Tok);
1282ac27de9dSRui Ueyama   Script->ReferencedSymbols.push_back(Tok);
1283722221f5SRui Ueyama   return [=] { return Script->getSymbolValue(Tok, Location); };
12842ec34544SRui Ueyama }
12852ec34544SRui Ueyama 
12862ec34544SRui Ueyama Expr ScriptParser::readTernary(Expr Cond) {
12872ec34544SRui Ueyama   Expr L = readExpr();
12882ec34544SRui Ueyama   expect(":");
12892ec34544SRui Ueyama   Expr R = readExpr();
12902ec34544SRui Ueyama   return [=] { return Cond().getValue() ? L() : R(); };
12912ec34544SRui Ueyama }
12922ec34544SRui Ueyama 
12932ec34544SRui Ueyama Expr ScriptParser::readParenExpr() {
12942ec34544SRui Ueyama   expect("(");
12952ec34544SRui Ueyama   Expr E = readExpr();
12962ec34544SRui Ueyama   expect(")");
12972ec34544SRui Ueyama   return E;
12982ec34544SRui Ueyama }
12992ec34544SRui Ueyama 
13002ec34544SRui Ueyama std::vector<StringRef> ScriptParser::readOutputSectionPhdrs() {
13012ec34544SRui Ueyama   std::vector<StringRef> Phdrs;
1302b8a59c8aSBob Haarman   while (!errorCount() && peek().startswith(":")) {
13032ec34544SRui Ueyama     StringRef Tok = next();
13042ec34544SRui Ueyama     Phdrs.push_back((Tok.size() == 1) ? next() : Tok.substr(1));
13052ec34544SRui Ueyama   }
13062ec34544SRui Ueyama   return Phdrs;
13072ec34544SRui Ueyama }
13082ec34544SRui Ueyama 
13092ec34544SRui Ueyama // Read a program header type name. The next token must be a
13102ec34544SRui Ueyama // name of a program header type or a constant (e.g. "0x3").
13112ec34544SRui Ueyama unsigned ScriptParser::readPhdrType() {
13122ec34544SRui Ueyama   StringRef Tok = next();
13135c65088fSRui Ueyama   if (Optional<uint64_t> Val = parseInt(Tok))
13145c65088fSRui Ueyama     return *Val;
13152ec34544SRui Ueyama 
13162ec34544SRui Ueyama   unsigned Ret = StringSwitch<unsigned>(Tok)
13172ec34544SRui Ueyama                      .Case("PT_NULL", PT_NULL)
13182ec34544SRui Ueyama                      .Case("PT_LOAD", PT_LOAD)
13192ec34544SRui Ueyama                      .Case("PT_DYNAMIC", PT_DYNAMIC)
13202ec34544SRui Ueyama                      .Case("PT_INTERP", PT_INTERP)
13212ec34544SRui Ueyama                      .Case("PT_NOTE", PT_NOTE)
13222ec34544SRui Ueyama                      .Case("PT_SHLIB", PT_SHLIB)
13232ec34544SRui Ueyama                      .Case("PT_PHDR", PT_PHDR)
13242ec34544SRui Ueyama                      .Case("PT_TLS", PT_TLS)
13252ec34544SRui Ueyama                      .Case("PT_GNU_EH_FRAME", PT_GNU_EH_FRAME)
13262ec34544SRui Ueyama                      .Case("PT_GNU_STACK", PT_GNU_STACK)
13272ec34544SRui Ueyama                      .Case("PT_GNU_RELRO", PT_GNU_RELRO)
13282ec34544SRui Ueyama                      .Case("PT_OPENBSD_RANDOMIZE", PT_OPENBSD_RANDOMIZE)
13292ec34544SRui Ueyama                      .Case("PT_OPENBSD_WXNEEDED", PT_OPENBSD_WXNEEDED)
13302ec34544SRui Ueyama                      .Case("PT_OPENBSD_BOOTDATA", PT_OPENBSD_BOOTDATA)
13312ec34544SRui Ueyama                      .Default(-1);
13322ec34544SRui Ueyama 
13332ec34544SRui Ueyama   if (Ret == (unsigned)-1) {
13342ec34544SRui Ueyama     setError("invalid program header type: " + Tok);
13352ec34544SRui Ueyama     return PT_NULL;
13362ec34544SRui Ueyama   }
13372ec34544SRui Ueyama   return Ret;
13382ec34544SRui Ueyama }
13392ec34544SRui Ueyama 
13402ec34544SRui Ueyama // Reads an anonymous version declaration.
13412ec34544SRui Ueyama void ScriptParser::readAnonymousDeclaration() {
13422ec34544SRui Ueyama   std::vector<SymbolVersion> Locals;
13432ec34544SRui Ueyama   std::vector<SymbolVersion> Globals;
13442ec34544SRui Ueyama   std::tie(Locals, Globals) = readSymbols();
13452ec34544SRui Ueyama 
13462ec34544SRui Ueyama   for (SymbolVersion V : Locals) {
13472ec34544SRui Ueyama     if (V.Name == "*")
13482ec34544SRui Ueyama       Config->DefaultSymbolVersion = VER_NDX_LOCAL;
13492ec34544SRui Ueyama     else
13502ec34544SRui Ueyama       Config->VersionScriptLocals.push_back(V);
13512ec34544SRui Ueyama   }
13522ec34544SRui Ueyama 
13532ec34544SRui Ueyama   for (SymbolVersion V : Globals)
13542ec34544SRui Ueyama     Config->VersionScriptGlobals.push_back(V);
13552ec34544SRui Ueyama 
13562ec34544SRui Ueyama   expect(";");
13572ec34544SRui Ueyama }
13582ec34544SRui Ueyama 
13592ec34544SRui Ueyama // Reads a non-anonymous version definition,
13602ec34544SRui Ueyama // e.g. "VerStr { global: foo; bar; local: *; };".
13612ec34544SRui Ueyama void ScriptParser::readVersionDeclaration(StringRef VerStr) {
13622ec34544SRui Ueyama   // Read a symbol list.
13632ec34544SRui Ueyama   std::vector<SymbolVersion> Locals;
13642ec34544SRui Ueyama   std::vector<SymbolVersion> Globals;
13652ec34544SRui Ueyama   std::tie(Locals, Globals) = readSymbols();
13662ec34544SRui Ueyama 
13672ec34544SRui Ueyama   for (SymbolVersion V : Locals) {
13682ec34544SRui Ueyama     if (V.Name == "*")
13692ec34544SRui Ueyama       Config->DefaultSymbolVersion = VER_NDX_LOCAL;
13702ec34544SRui Ueyama     else
13712ec34544SRui Ueyama       Config->VersionScriptLocals.push_back(V);
13722ec34544SRui Ueyama   }
13732ec34544SRui Ueyama 
13742ec34544SRui Ueyama   // Create a new version definition and add that to the global symbols.
13752ec34544SRui Ueyama   VersionDefinition Ver;
13762ec34544SRui Ueyama   Ver.Name = VerStr;
13772ec34544SRui Ueyama   Ver.Globals = Globals;
13782ec34544SRui Ueyama 
13792ec34544SRui Ueyama   // User-defined version number starts from 2 because 0 and 1 are
13802ec34544SRui Ueyama   // reserved for VER_NDX_LOCAL and VER_NDX_GLOBAL, respectively.
13812ec34544SRui Ueyama   Ver.Id = Config->VersionDefinitions.size() + 2;
13822ec34544SRui Ueyama   Config->VersionDefinitions.push_back(Ver);
13832ec34544SRui Ueyama 
13842ec34544SRui Ueyama   // Each version may have a parent version. For example, "Ver2"
13852ec34544SRui Ueyama   // defined as "Ver2 { global: foo; local: *; } Ver1;" has "Ver1"
13862ec34544SRui Ueyama   // as a parent. This version hierarchy is, probably against your
13872ec34544SRui Ueyama   // instinct, purely for hint; the runtime doesn't care about it
13882ec34544SRui Ueyama   // at all. In LLD, we simply ignore it.
13892ec34544SRui Ueyama   if (peek() != ";")
13902ec34544SRui Ueyama     skip();
13912ec34544SRui Ueyama   expect(";");
13922ec34544SRui Ueyama }
13932ec34544SRui Ueyama 
13941e77ad14SRui Ueyama static bool hasWildcard(StringRef S) {
13951e77ad14SRui Ueyama   return S.find_first_of("?*[") != StringRef::npos;
13961e77ad14SRui Ueyama }
13971e77ad14SRui Ueyama 
13982ec34544SRui Ueyama // Reads a list of symbols, e.g. "{ global: foo; bar; local: *; };".
13992ec34544SRui Ueyama std::pair<std::vector<SymbolVersion>, std::vector<SymbolVersion>>
14002ec34544SRui Ueyama ScriptParser::readSymbols() {
14012ec34544SRui Ueyama   std::vector<SymbolVersion> Locals;
14022ec34544SRui Ueyama   std::vector<SymbolVersion> Globals;
14032ec34544SRui Ueyama   std::vector<SymbolVersion> *V = &Globals;
14042ec34544SRui Ueyama 
1405b8a59c8aSBob Haarman   while (!errorCount()) {
14062ec34544SRui Ueyama     if (consume("}"))
14072ec34544SRui Ueyama       break;
14082ec34544SRui Ueyama     if (consumeLabel("local")) {
14092ec34544SRui Ueyama       V = &Locals;
14102ec34544SRui Ueyama       continue;
14112ec34544SRui Ueyama     }
14122ec34544SRui Ueyama     if (consumeLabel("global")) {
14132ec34544SRui Ueyama       V = &Globals;
14142ec34544SRui Ueyama       continue;
14152ec34544SRui Ueyama     }
14162ec34544SRui Ueyama 
14172ec34544SRui Ueyama     if (consume("extern")) {
14182ec34544SRui Ueyama       std::vector<SymbolVersion> Ext = readVersionExtern();
14192ec34544SRui Ueyama       V->insert(V->end(), Ext.begin(), Ext.end());
14202ec34544SRui Ueyama     } else {
14212ec34544SRui Ueyama       StringRef Tok = next();
14222ec34544SRui Ueyama       V->push_back({unquote(Tok), false, hasWildcard(Tok)});
14232ec34544SRui Ueyama     }
14242ec34544SRui Ueyama     expect(";");
14252ec34544SRui Ueyama   }
14262ec34544SRui Ueyama   return {Locals, Globals};
14272ec34544SRui Ueyama }
14282ec34544SRui Ueyama 
14292ec34544SRui Ueyama // Reads an "extern C++" directive, e.g.,
14302ec34544SRui Ueyama // "extern "C++" { ns::*; "f(int, double)"; };"
143117324d8bSRui Ueyama //
143217324d8bSRui Ueyama // The last semicolon is optional. E.g. this is OK:
143317324d8bSRui Ueyama // "extern "C++" { ns::*; "f(int, double)" };"
14342ec34544SRui Ueyama std::vector<SymbolVersion> ScriptParser::readVersionExtern() {
14352ec34544SRui Ueyama   StringRef Tok = next();
14362ec34544SRui Ueyama   bool IsCXX = Tok == "\"C++\"";
14372ec34544SRui Ueyama   if (!IsCXX && Tok != "\"C\"")
14382ec34544SRui Ueyama     setError("Unknown language");
14392ec34544SRui Ueyama   expect("{");
14402ec34544SRui Ueyama 
14412ec34544SRui Ueyama   std::vector<SymbolVersion> Ret;
1442b8a59c8aSBob Haarman   while (!errorCount() && peek() != "}") {
14432ec34544SRui Ueyama     StringRef Tok = next();
14442ec34544SRui Ueyama     bool HasWildcard = !Tok.startswith("\"") && hasWildcard(Tok);
14452ec34544SRui Ueyama     Ret.push_back({unquote(Tok), IsCXX, HasWildcard});
144617324d8bSRui Ueyama     if (consume("}"))
144717324d8bSRui Ueyama       return Ret;
14482ec34544SRui Ueyama     expect(";");
14492ec34544SRui Ueyama   }
14502ec34544SRui Ueyama 
14512ec34544SRui Ueyama   expect("}");
14522ec34544SRui Ueyama   return Ret;
14532ec34544SRui Ueyama }
14542ec34544SRui Ueyama 
14552ec34544SRui Ueyama uint64_t ScriptParser::readMemoryAssignment(StringRef S1, StringRef S2,
14562ec34544SRui Ueyama                                             StringRef S3) {
1457b579c439SRui Ueyama   if (!consume(S1) && !consume(S2) && !consume(S3)) {
14582ec34544SRui Ueyama     setError("expected one of: " + S1 + ", " + S2 + ", or " + S3);
14592ec34544SRui Ueyama     return 0;
14602ec34544SRui Ueyama   }
14612ec34544SRui Ueyama   expect("=");
1462040af7deSRui Ueyama   return readExpr()().getValue();
14632ec34544SRui Ueyama }
14642ec34544SRui Ueyama 
14652ec34544SRui Ueyama // Parse the MEMORY command as specified in:
14662ec34544SRui Ueyama // https://sourceware.org/binutils/docs/ld/MEMORY.html
14672ec34544SRui Ueyama //
14682ec34544SRui Ueyama // MEMORY { name [(attr)] : ORIGIN = origin, LENGTH = len ... }
14692ec34544SRui Ueyama void ScriptParser::readMemory() {
14702ec34544SRui Ueyama   expect("{");
1471b8a59c8aSBob Haarman   while (!errorCount() && !consume("}")) {
14722e9d40d5SRui Ueyama     StringRef Tok = next();
14732e9d40d5SRui Ueyama     if (Tok == "INCLUDE") {
14742e9d40d5SRui Ueyama       readInclude();
14752e9d40d5SRui Ueyama       continue;
14762e9d40d5SRui Ueyama     }
14772ec34544SRui Ueyama 
14782ec34544SRui Ueyama     uint32_t Flags = 0;
14792ec34544SRui Ueyama     uint32_t NegFlags = 0;
14802ec34544SRui Ueyama     if (consume("(")) {
14812ec34544SRui Ueyama       std::tie(Flags, NegFlags) = readMemoryAttributes();
14822ec34544SRui Ueyama       expect(")");
14832ec34544SRui Ueyama     }
14842ec34544SRui Ueyama     expect(":");
14852ec34544SRui Ueyama 
14862ec34544SRui Ueyama     uint64_t Origin = readMemoryAssignment("ORIGIN", "org", "o");
14872ec34544SRui Ueyama     expect(",");
14882ec34544SRui Ueyama     uint64_t Length = readMemoryAssignment("LENGTH", "len", "l");
14892ec34544SRui Ueyama 
14905f37541cSGeorge Rimar     // Add the memory region to the region map.
14912e9d40d5SRui Ueyama     MemoryRegion *MR = make<MemoryRegion>(Tok, Origin, Length, Flags, NegFlags);
14922e9d40d5SRui Ueyama     if (!Script->MemoryRegions.insert({Tok, MR}).second)
14932e9d40d5SRui Ueyama       setError("region '" + Tok + "' already defined");
14942ec34544SRui Ueyama   }
14952ec34544SRui Ueyama }
14962ec34544SRui Ueyama 
14972ec34544SRui Ueyama // This function parses the attributes used to match against section
14982ec34544SRui Ueyama // flags when placing output sections in a memory region. These flags
14992ec34544SRui Ueyama // are only used when an explicit memory region name is not used.
15002ec34544SRui Ueyama std::pair<uint32_t, uint32_t> ScriptParser::readMemoryAttributes() {
15012ec34544SRui Ueyama   uint32_t Flags = 0;
15022ec34544SRui Ueyama   uint32_t NegFlags = 0;
15032ec34544SRui Ueyama   bool Invert = false;
15042ec34544SRui Ueyama 
15052ec34544SRui Ueyama   for (char C : next().lower()) {
15062ec34544SRui Ueyama     uint32_t Flag = 0;
15072ec34544SRui Ueyama     if (C == '!')
15082ec34544SRui Ueyama       Invert = !Invert;
15092ec34544SRui Ueyama     else if (C == 'w')
15102ec34544SRui Ueyama       Flag = SHF_WRITE;
15112ec34544SRui Ueyama     else if (C == 'x')
15122ec34544SRui Ueyama       Flag = SHF_EXECINSTR;
15132ec34544SRui Ueyama     else if (C == 'a')
15142ec34544SRui Ueyama       Flag = SHF_ALLOC;
15152ec34544SRui Ueyama     else if (C != 'r')
15162ec34544SRui Ueyama       setError("invalid memory region attribute");
15172ec34544SRui Ueyama 
15182ec34544SRui Ueyama     if (Invert)
15192ec34544SRui Ueyama       NegFlags |= Flag;
15202ec34544SRui Ueyama     else
15212ec34544SRui Ueyama       Flags |= Flag;
15222ec34544SRui Ueyama   }
15232ec34544SRui Ueyama   return {Flags, NegFlags};
15242ec34544SRui Ueyama }
15252ec34544SRui Ueyama 
15262ec34544SRui Ueyama void elf::readLinkerScript(MemoryBufferRef MB) {
15272ec34544SRui Ueyama   ScriptParser(MB).readLinkerScript();
15282ec34544SRui Ueyama }
15292ec34544SRui Ueyama 
15302ec34544SRui Ueyama void elf::readVersionScript(MemoryBufferRef MB) {
15312ec34544SRui Ueyama   ScriptParser(MB).readVersionScript();
15322ec34544SRui Ueyama }
15332ec34544SRui Ueyama 
15342ec34544SRui Ueyama void elf::readDynamicList(MemoryBufferRef MB) {
15352ec34544SRui Ueyama   ScriptParser(MB).readDynamicList();
15362ec34544SRui Ueyama }
15378c7e8cceSPetr Hosek 
15388c7e8cceSPetr Hosek void elf::readDefsym(StringRef Name, MemoryBufferRef MB) {
15398c7e8cceSPetr Hosek   ScriptParser(MB).readDefsym(Name);
15408c7e8cceSPetr Hosek }
1541