diff --git a/lld/ELF/ScriptLexer.h b/lld/ELF/ScriptLexer.h --- a/lld/ELF/ScriptLexer.h +++ b/lld/ELF/ScriptLexer.h @@ -40,13 +40,14 @@ bool inExpr = false; size_t pos = 0; +protected: + MemoryBufferRef getCurrentMB(); + private: void maybeSplitExpr(); StringRef getLine(); size_t getLineNumber(); size_t getColumnNumber(); - - MemoryBufferRef getCurrentMB(); }; } // namespace elf diff --git a/lld/ELF/ScriptParser.cpp b/lld/ELF/ScriptParser.cpp --- a/lld/ELF/ScriptParser.cpp +++ b/lld/ELF/ScriptParser.cpp @@ -290,22 +290,40 @@ } if (s.startswith("/")) { + // Case 1: s is an absolute path. Just open it. driver->addFile(s, /*withLOption=*/false); } else if (s.startswith("=")) { + // Case 2: relative to the sysroot. if (config->sysroot.empty()) driver->addFile(s.substr(1), /*withLOption=*/false); else driver->addFile(saver.save(config->sysroot + "/" + s.substr(1)), /*withLOption=*/false); } else if (s.startswith("-l")) { + // Case 3: search in the list of library paths. driver->addLibrary(s.substr(2)); - } else if (sys::fs::exists(s)) { - driver->addFile(s, /*withLOption=*/false); } else { - if (Optional path = findFromSearchPaths(s)) - driver->addFile(saver.save(*path), /*withLOption=*/true); - else - setError("unable to find " + s); + // Case 4: s is a relative path. Search in the directory of the script file. + std::string filename = std::string(getCurrentMB().getBufferIdentifier()); + StringRef directory = sys::path::parent_path(filename); + if (!directory.empty()) { + SmallString<0> path(directory); + sys::path::append(path, s); + if (sys::fs::exists(path)) { + driver->addFile(path, /*withLOption=*/false); + return; + } + } + // Then search in the current working directory. + if (sys::fs::exists(s)) { + driver->addFile(s, /*withLOption=*/false); + } else { + // Finally, search in the list of library paths. + if (Optional path = findFromSearchPaths(s)) + driver->addFile(saver.save(*path), /*withLOption=*/true); + else + setError("unable to find " + s); + } } } diff --git a/lld/test/ELF/linkerscript/input-relative.s b/lld/test/ELF/linkerscript/input-relative.s new file mode 100644 --- /dev/null +++ b/lld/test/ELF/linkerscript/input-relative.s @@ -0,0 +1,46 @@ +# REQUIRES: x86 +## For a relative pathname in INPUT() or GROUP(), the parent directory of +## the current linker script has priority over current working directory and -L. + +# RUN: rm -rf %t.dir +# RUN: mkdir %t.dir +# RUN: cd %t.dir + +# RUN: mkdir dir +# RUN: llvm-mc -filetype=obj -triple=x86_64 %s -o a.o +# RUN: echo '.globl b, cwd; b: cwd:' | llvm-mc -filetype=obj -triple=x86_64 - -o b.o +# RUN: echo '.globl b, dir; b: dir:' | llvm-mc -filetype=obj -triple=x86_64 - -o dir/b.o +# RUN: llvm-ar rc libb.a b.o +# RUN: llvm-ar rc dir/libb.a dir/b.o + +## A relative pathname is relative to the parent directory of the current linker script. +## The directory has priority over current working directory and -L. +# RUN: echo 'INPUT(libb.a)' > dir/relative.lds +# RUN: ld.lld -L. a.o dir/relative.lds -o relative +# RUN: llvm-nm relative | FileCheck --check-prefix=DIR %s +## GROUP() uses the same search order. +# RUN: echo 'GROUP(libb.a)' > dir/relative1.lds +# RUN: ld.lld -L. a.o dir/relative1.lds -o relative1 +# RUN: llvm-nm relative1 | FileCheck --check-prefix=DIR %s + +# DIR: T dir + +## -l does not use the special rule. +# RUN: echo 'INPUT(-lb)' > dir/cwd.lds +# RUN: ld.lld -L. a.o dir/cwd.lds -o cwd +# RUN: llvm-nm cwd | FileCheck --check-prefix=CWD %s +# RUN: echo 'GROUP(-lb)' > dir/cwd1.lds +# RUN: ld.lld -L. a.o dir/cwd1.lds -o cwd1 +# RUN: llvm-nm cwd1 | FileCheck --check-prefix=CWD %s + +# CWD: T cwd + +## If the parent directory of the current linker script does not contain the file, +## fall back to the current working directory. +# RUN: cp libb.a libc.a +# RUN: echo 'INPUT(libc.a)' > dir/fallback.lds +# RUN: ld.lld a.o dir/fallback.lds -o /dev/null + +.globl _start +_start: + call b