X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=tools%2Fllvm-link%2Fllvm-link.cpp;h=3fb7ba42cd05727e12e08222954b11bca1806596;hb=adf01b3f18442ae8db6b8948e70d82d9df415119;hp=0ae39228d9cbbe060b849449e678c3cbf699c7a6;hpb=d6b65255fec99708c8317cde601b12ec09f538d9;p=oota-llvm.git diff --git a/tools/llvm-link/llvm-link.cpp b/tools/llvm-link/llvm-link.cpp index 0ae39228d9c..3fb7ba42cd0 100644 --- a/tools/llvm-link/llvm-link.cpp +++ b/tools/llvm-link/llvm-link.cpp @@ -1,85 +1,141 @@ +//===- llvm-link.cpp - Low-level LLVM linker ------------------------------===// +// +// The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// //===----------------------------------------------------------------------===// -// LLVM 'LINK' UTILITY // // This utility may be invoked in the following manner: -// link a.bc b.bc c.bc -o x.bc -// -// Alternatively, this can be used as an 'ar' tool as well. If invoked as -// either 'ar' or 'llvm-ar', it accepts a 'cr' parameter as well. +// llvm-link a.bc b.bc c.bc -o x.bc // //===----------------------------------------------------------------------===// -#include "llvm/Transforms/Linker.h" -#include "llvm/Bytecode/Reader.h" -#include "llvm/Bytecode/Writer.h" -#include "llvm/Assembly/Writer.h" -#include "llvm/Support/CommandLine.h" +#include "llvm/Linker.h" +#include "llvm/LLVMContext.h" #include "llvm/Module.h" -#include "llvm/Method.h" -#include +#include "llvm/Analysis/Verifier.h" +#include "llvm/Bitcode/ReaderWriter.h" +#include "llvm/Support/CommandLine.h" +#include "llvm/Support/ManagedStatic.h" +#include "llvm/Support/PrettyStackTrace.h" +#include "llvm/Support/ToolOutputFile.h" +#include "llvm/Support/SystemUtils.h" +#include "llvm/Support/IRReader.h" +#include "llvm/Support/Signals.h" +#include "llvm/Support/Path.h" #include +using namespace llvm; + +static cl::list +InputFilenames(cl::Positional, cl::OneOrMore, + cl::desc("")); +static cl::opt +OutputFilename("o", cl::desc("Override output filename"), cl::init("-"), + cl::value_desc("filename")); -cl::StringList InputFilenames("", "Load files, linking them together", - cl::OneOrMore); -cl::String OutputFilename("o", "Override output filename", cl::NoFlags, "-"); -cl::Flag Force ("f", "Overwrite output files", cl::NoFlags, false); -cl::Flag Verbose ("v", "Print information about actions taken"); -cl::Flag DumpAsm ("d", "Print assembly as linked", cl::Hidden, false); +static cl::opt +Force("f", cl::desc("Enable binary output on terminals")); + +static cl::opt +OutputAssembly("S", + cl::desc("Write output as LLVM assembly"), cl::Hidden); + +static cl::opt +Verbose("v", cl::desc("Print information about actions taken")); + +static cl::opt +DumpAsm("d", cl::desc("Print assembly as linked"), cl::Hidden); + +// LoadFile - Read the specified bitcode file in and return it. This routine +// searches the link path for the specified file to try to find it... +// +static inline std::auto_ptr LoadFile(const char *argv0, + const std::string &FN, + LLVMContext& Context) { + sys::Path Filename; + if (!Filename.set(FN)) { + errs() << "Invalid file name: '" << FN << "'\n"; + return std::auto_ptr(); + } + SMDiagnostic Err; + if (Verbose) errs() << "Loading '" << Filename.c_str() << "'\n"; + Module* Result = 0; + + const std::string &FNStr = Filename.str(); + Result = ParseIRFile(FNStr, Err, Context); + if (Result) return std::auto_ptr(Result); // Load successful! + + Err.Print(argv0, errs()); + return std::auto_ptr(); +} int main(int argc, char **argv) { - cl::ParseCommandLineOptions(argc, argv, " llvm linker\n"); - assert(InputFilenames.size() > 0 && "OneOrMore is not working"); - - // TODO: TEST argv[0] - string ErrorMessage; - - if (Verbose) cerr << "Loading '" << InputFilenames[0] << "'\n"; - std::auto_ptr Composite(ParseBytecodeFile(InputFilenames[0], - &ErrorMessage)); + // Print a stack trace if we signal out. + sys::PrintStackTraceOnErrorSignal(); + PrettyStackTraceProgram X(argc, argv); + + LLVMContext &Context = getGlobalContext(); + llvm_shutdown_obj Y; // Call llvm_shutdown() on exit. + cl::ParseCommandLineOptions(argc, argv, "llvm linker\n"); + + unsigned BaseArg = 0; + std::string ErrorMessage; + + std::auto_ptr Composite(LoadFile(argv[0], + InputFilenames[BaseArg], Context)); if (Composite.get() == 0) { - cerr << "Error opening bytecode file: '" << InputFilenames[0] << "'"; - if (ErrorMessage.size()) cerr << ": " << ErrorMessage; - cerr << endl; + errs() << argv[0] << ": error loading file '" + << InputFilenames[BaseArg] << "'\n"; return 1; } - for (unsigned i = 1; i < InputFilenames.size(); ++i) { - if (Verbose) cerr << "Loading '" << InputFilenames[i] << "'\n"; - auto_ptr M(ParseBytecodeFile(InputFilenames[i], &ErrorMessage)); + for (unsigned i = BaseArg+1; i < InputFilenames.size(); ++i) { + std::auto_ptr M(LoadFile(argv[0], + InputFilenames[i], Context)); if (M.get() == 0) { - cerr << "Error opening bytecode file: '" << InputFilenames[i] << "'"; - if (ErrorMessage.size()) cerr << ": " << ErrorMessage; - cerr << endl; + errs() << argv[0] << ": error loading file '" <good()) { - cerr << "Error opening '" << OutputFilename << "'!\n"; - return 1; - } + if (DumpAsm) errs() << "Here's the assembly:\n" << *Composite; + + std::string ErrorInfo; + tool_output_file Out(OutputFilename.c_str(), ErrorInfo, + raw_fd_ostream::F_Binary); + if (!ErrorInfo.empty()) { + errs() << ErrorInfo << '\n'; + return 1; + } + + if (verifyModule(*Composite)) { + errs() << argv[0] << ": linked module is broken!\n"; + return 1; } - if (Verbose) cerr << "Writing bytecode...\n"; - WriteBytecodeToFile(Composite.get(), *Out); + if (Verbose) errs() << "Writing bitcode...\n"; + if (OutputAssembly) { + Out.os() << *Composite; + } else if (Force || !CheckBitcodeOutputToConsole(Out.os(), true)) + WriteBitcodeToFile(Composite.get(), Out.os()); + + // Declare success. + Out.keep(); - if (Out != &cout) delete Out; return 0; }