X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=tools%2Fllvm-link%2Fllvm-link.cpp;h=f2ba29cf97d452a449198d581689379b6287d345;hb=867fe8570f299a058f155f98646d85cabc27155b;hp=c4be6715cff6588faf6a7c851b77b927706045d5;hpb=5bdf1614182c679c0d0a50f28f5810ee14110800;p=oota-llvm.git diff --git a/tools/llvm-link/llvm-link.cpp b/tools/llvm-link/llvm-link.cpp index c4be6715cff..f2ba29cf97d 100644 --- a/tools/llvm-link/llvm-link.cpp +++ b/tools/llvm-link/llvm-link.cpp @@ -1,136 +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 'rc' 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/Linker.h" +#include "llvm/LLVMContext.h" #include "llvm/Module.h" -#include "Support/CommandLine.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/System/Signals.h" +#include "llvm/System/Path.h" #include -#include // For FileExists -#include +using namespace llvm; +static cl::list +InputFilenames(cl::Positional, cl::OneOrMore, + cl::desc("")); -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); -cl::StringList LibPaths ("L", "Specify a library search path", cl::ZeroOrMore); -cl::StringList Libraries ("l", "Specify libraries to link to", cl::ZeroOrMore); +static cl::opt +OutputFilename("o", cl::desc("Override output filename"), cl::init("-"), + cl::value_desc("filename")); +static cl::opt +Force("f", cl::desc("Enable binary output on terminals")); -// FileExists - Return true if the specified string is an openable file... -static inline bool FileExists(const std::string &FN) { - struct stat StatBuf; - return stat(FN.c_str(), &StatBuf) != -1; -} - -// LoadFile - Read the specified bytecode 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 std::string &FN) { - std::string Filename = FN; - std::string ErrorMessage; +static cl::opt +OutputAssembly("S", + cl::desc("Write output as LLVM assembly"), cl::Hidden); - unsigned NextLibPathIdx = 0; - bool FoundAFile = false; +static cl::opt +Verbose("v", cl::desc("Print information about actions taken")); - while (1) { - if (Verbose) cerr << "Loading '" << Filename << "'\n"; - if (FileExists(Filename)) FoundAFile = true; - Module *Result = ParseBytecodeFile(Filename, &ErrorMessage); - if (Result) return std::auto_ptr(Result); // Load successful! +static cl::opt +DumpAsm("d", cl::desc("Print assembly as linked"), cl::Hidden); - if (Verbose) { - cerr << "Error opening bytecode file: '" << Filename << "'"; - if (ErrorMessage.size()) cerr << ": " << ErrorMessage; - cerr << "\n"; - } - - if (NextLibPathIdx == LibPaths.size()) break; - Filename = LibPaths[NextLibPathIdx++] + "/" + FN; +// 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(); } - if (FoundAFile) - cerr << "Bytecode file '" << FN << "' corrupt! " - << "Use 'link -v ...' for more info.\n"; - else - cerr << "Could not locate bytecode file: '" << FN << "'\n"; + 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", - cl::EnableSingleLetterArgValue | - cl::DisableSingleLetterArgGrouping); - assert(InputFilenames.size() > 0 && "OneOrMore is not working"); + // 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; - // TODO: TEST argv[0] for llvm-ar forms... for now, this is a huge hack. - if (InputFilenames.size() >= 3 && InputFilenames[0] == "rc" && - OutputFilename == "-") { - BaseArg = 2; - OutputFilename = InputFilenames[1]; + std::auto_ptr Composite(LoadFile(argv[0], + InputFilenames[BaseArg], Context)); + if (Composite.get() == 0) { + errs() << argv[0] << ": error loading file '" + << InputFilenames[BaseArg] << "'\n"; + return 1; } - if (!Libraries.empty()) - cerr << "LLVM Linker Warning: Linking to libraries is unimplemented!\n"; - - std::auto_ptr Composite(LoadFile(InputFilenames[BaseArg])); - if (Composite.get() == 0) return 1; - for (unsigned i = BaseArg+1; i < InputFilenames.size(); ++i) { - std::auto_ptr M(LoadFile(InputFilenames[i])); - if (M.get() == 0) return 1; + std::auto_ptr M(LoadFile(argv[0], + InputFilenames[i], Context)); + if (M.get() == 0) { + errs() << argv[0] << ": error loading file '" <dump(); + // TODO: Iterate over the -l list and link in any modules containing + // global symbols that have not been resolved so far. + + 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; } - ostream *Out = &cout; // Default to printing to stdout... - if (OutputFilename != "-") { - if (!Force && std::ifstream(OutputFilename.c_str())) { - // If force is not specified, make sure not to overwrite a file! - cerr << "Error opening '" << OutputFilename << "': File exists!\n" - << "Use -f command line argument to force output\n"; - return 1; - } - Out = new std::ofstream(OutputFilename.c_str()); - if (!Out->good()) { - cerr << "Error opening '" << OutputFilename << "'!\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 != &std::cout) delete Out; return 0; }