maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

ELF2: LinkerScript: Implement INCLUDE directive. 

llvm-svn: 249960
This commit is contained in:
Rui Ueyama 2015-10-11 01:31:55 +00:00
parent bdca0b1fac
commit 31aa1f83de
2 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
lld/ELF/LinkerScript.cpp
View File

@ -39,12 +39,14 @@ private:
void readAsNeeded();
void readEntry();
void readGroup();
void readInclude();
void readOutput();
void readOutputFormat();
void readSearchDir();
std::vector<StringRef> Tokens;
size_t Pos = 0;
static std::vector<std::unique_ptr<MemoryBuffer>> OwningMBs;
};
}
@ -55,6 +57,8 @@ void LinkerScript::run() {
readEntry();
} else if (Tok == "GROUP") {
readGroup();
} else if (Tok == "INCLUDE") {
readInclude();
} else if (Tok == "OUTPUT") {
readOutput();
} else if (Tok == "OUTPUT_FORMAT") {
@ -160,6 +164,16 @@ void LinkerScript::readGroup() {
}
}
void LinkerScript::readInclude() {
StringRef Tok = next();
auto MBOrErr = MemoryBuffer::getFile(Tok);
error(MBOrErr, Twine("cannot open ") + Tok);
std::unique_ptr<MemoryBuffer> &MB = *MBOrErr;
std::vector<StringRef> V = tokenize(MB->getMemBufferRef().getBuffer());
Tokens.insert(Tokens.begin() + Pos, V.begin(), V.end());
OwningMBs.push_back(std::move(MB)); // keep ownership of MB
}
void LinkerScript::readOutput() {
// -o <file> takes predecence over OUTPUT(<file>).
expect("(");
@ -182,6 +196,8 @@ void LinkerScript::readSearchDir() {
expect(")");
}
std::vector<std::unique_ptr<MemoryBuffer>> LinkerScript::OwningMBs;
// Entry point. The other functions or classes are private to this file.
void lld::elf2::readLinkerScript(MemoryBufferRef MB) {
LinkerScript(MB.getBuffer()).run();

5
lld/test/elf2/linkerscript.s
View File

@ -47,6 +47,11 @@
# RUN: ld.lld2 %t.script %t
# RUN: llvm-readobj %t.out > /dev/null
# RUN: echo "INCLUDE " %t.script2 "OUTPUT(" %t.out ")" > %t.script1
# RUN: echo "GROUP(" %t ")" > %t.script2
# RUN: ld.lld2 %t.script1
# RUN: llvm-readobj %t2 > /dev/null
# RUN: echo "FOO(BAR)" > %t.script
# RUN: not ld.lld2 -o foo %t.script > %t.log 2>&1
# RUN: FileCheck -check-prefix=ERR1 %s < %t.log