Index: llvm/test/tools/llvm-link/archive-bad.ll =================================================================== --- llvm/test/tools/llvm-link/archive-bad.ll +++ llvm/test/tools/llvm-link/archive-bad.ll @@ -1,7 +1,7 @@ -# RUN: cp %S/Inputs/f.ll %t.fg.a +# RUN: echo -e '!\nwith invalid contents' > %t.fg.a # RUN: not llvm-link %S/Inputs/h.ll %t.fg.a -o %t.linked.bc 2>&1 | FileCheck %s # RUN: rm -f %t.fg.a # RUN: rm -f %t.linked.bc -# CHECK: file too small to be an archive +# CHECK: truncated or malformed archive Index: llvm/test/tools/llvm-link/archive.ll =================================================================== --- llvm/test/tools/llvm-link/archive.ll +++ llvm/test/tools/llvm-link/archive.ll @@ -1,8 +1,8 @@ # RUN: llvm-as %S/Inputs/f.ll -o %t.f.bc # RUN: llvm-as %S/Inputs/g.ll -o %t.g.bc # RUN: llvm-ar cr %t.fg.a %t.f.bc %t.g.bc -# RUN: llvm-ar cr %t.empty.a -# RUN: llvm-link %S/Inputs/h.ll %t.fg.a %t.empty.a -o %t.linked.bc +# RUN: llvm-ar cr %t.empty.lib +# RUN: llvm-link %S/Inputs/h.ll %t.fg.a %t.empty.lib -o %t.linked.bc # RUN: llvm-nm %t.linked.bc | FileCheck %s Index: llvm/tools/llvm-link/CMakeLists.txt =================================================================== --- llvm/tools/llvm-link/CMakeLists.txt +++ llvm/tools/llvm-link/CMakeLists.txt @@ -1,4 +1,5 @@ set(LLVM_LINK_COMPONENTS + BinaryFormat BitReader BitWriter Core Index: llvm/tools/llvm-link/llvm-link.cpp =================================================================== --- llvm/tools/llvm-link/llvm-link.cpp +++ llvm/tools/llvm-link/llvm-link.cpp @@ -11,8 +11,8 @@ // //===----------------------------------------------------------------------===// -#include "llvm/Object/Archive.h" #include "llvm/ADT/STLExtras.h" +#include "llvm/BinaryFormat/Magic.h" #include "llvm/Bitcode/BitcodeReader.h" #include "llvm/Bitcode/BitcodeWriter.h" #include "llvm/IR/AutoUpgrade.h" @@ -24,6 +24,7 @@ #include "llvm/IR/Verifier.h" #include "llvm/IRReader/IRReader.h" #include "llvm/Linker/Linker.h" +#include "llvm/Object/Archive.h" #include "llvm/Support/CommandLine.h" #include "llvm/Support/FileSystem.h" #include "llvm/Support/InitLLVM.h" @@ -115,17 +116,18 @@ // link path for the specified file to try to find it... // static std::unique_ptr loadFile(const char *argv0, - const std::string &FN, + std::unique_ptr Buffer, LLVMContext &Context, bool MaterializeMetadata = true) { SMDiagnostic Err; if (Verbose) - errs() << "Loading '" << FN << "'\n"; + errs() << "Loading '" << Buffer->getBufferIdentifier() << "'\n"; std::unique_ptr Result; if (DisableLazyLoad) - Result = parseIRFile(FN, Err, Context); + Result = parseIR(*Buffer, Err, Context); else - Result = getLazyIRFileModule(FN, Err, Context, !MaterializeMetadata); + Result = + getLazyIRModule(std::move(Buffer), Err, Context, !MaterializeMetadata); if (!Result) { Err.print(argv0, errs()); @@ -141,19 +143,17 @@ } static std::unique_ptr loadArFile(const char *Argv0, - const std::string &ArchiveName, + std::unique_ptr Buffer, LLVMContext &Context, Linker &L, unsigned OrigFlags, unsigned ApplicableFlags) { std::unique_ptr Result(new Module("ArchiveModule", Context)); + StringRef ArchiveName = Buffer->getBufferIdentifier(); if (Verbose) errs() << "Reading library archive file '" << ArchiveName << "' to memory\n"; - ErrorOr> Buf = - MemoryBuffer::getFile(ArchiveName, -1, false); - ExitOnErr(errorCodeToError(Buf.getError())); Error Err = Error::success(); - object::Archive Archive(Buf.get()->getMemBufferRef(), Err); + object::Archive Archive(*Buffer, Err); ExitOnErr(std::move(Err)); for (const object::Archive::Child &C : Archive.children(Err)) { Expected Ename = C.getName(); @@ -287,7 +287,9 @@ auto ModuleLoader = [&DestModule](const char *argv0, const std::string &Identifier) { - return loadFile(argv0, Identifier, DestModule.getContext(), false); + std::unique_ptr Buffer = + ExitOnErr(errorOrToExpected(MemoryBuffer::getFileOrSTDIN(Identifier))); + return loadFile(argv0, std::move(Buffer), DestModule.getContext(), false); }; ModuleLazyLoaderCache ModuleLoaderCache(ModuleLoader); @@ -349,10 +351,14 @@ // Similar to some flags, internalization doesn't apply to the first file. bool InternalizeLinkedSymbols = false; for (const auto &File : Files) { + std::unique_ptr Buffer = + ExitOnErr(errorOrToExpected(MemoryBuffer::getFileOrSTDIN(File))); + std::unique_ptr M = - (llvm::sys::path::extension(File) == ".a") - ? loadArFile(argv0, File, Context, L, Flags, ApplicableFlags) - : loadFile(argv0, File, Context); + identify_magic(Buffer->getBuffer()) == file_magic::archive + ? loadArFile(argv0, std::move(Buffer), Context, L, Flags, + ApplicableFlags) + : loadFile(argv0, std::move(Buffer), Context); if (!M.get()) { errs() << argv0 << ": "; WithColor::error() << " loading file '" << File << "'\n";