llvm-link.cpp revision bb37a69bfcf6ec663c84c3df9d77db39c46426a0
1bb37a69bfcf6ec663c84c3df9d77db39c46426a0Chris Lattner//===- llvm-link.cpp - Low-level LLVM linker ------------------------------===// 2075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner// 3075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner// This utility may be invoked in the following manner: 43ef3beb1d79f8f05fa55dac2c2f58822bad5e348Misha Brukman// llvm-link a.bc b.bc c.bc -o x.bc 5075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner// 6075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner//===----------------------------------------------------------------------===// 7075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner 8a55c4b1ce71bf62b669d5e533544a29739d56a98Chris Lattner#include "llvm/Module.h" 9a55c4b1ce71bf62b669d5e533544a29739d56a98Chris Lattner#include "llvm/Analysis/Verifier.h" 10075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner#include "llvm/Bytecode/Reader.h" 11075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner#include "llvm/Bytecode/Writer.h" 12a55c4b1ce71bf62b669d5e533544a29739d56a98Chris Lattner#include "llvm/Transforms/Utils/Linker.h" 13cee8f9ae67104576b2028125b56e9ba4856a1d66Chris Lattner#include "Support/CommandLine.h" 1476d12299b5333d71bfc079614031f97c97aa5148Chris Lattner#include "Support/Signals.h" 15cee8f9ae67104576b2028125b56e9ba4856a1d66Chris Lattner#include <fstream> 16075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner#include <memory> 17952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner#include <sys/types.h> // For FileExists 18952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner#include <sys/stat.h> 19075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner 205ff62e90d0bc321206023897edc1e2691cb0fbb6Chris Lattnerstatic cl::list<std::string> 215ff62e90d0bc321206023897edc1e2691cb0fbb6Chris LattnerInputFilenames(cl::Positional, cl::OneOrMore, 2254e05afa3c1e8e2ac1580e25c49a7d1bc9377727Chris Lattner cl::desc("<input bytecode files>")); 235ff62e90d0bc321206023897edc1e2691cb0fbb6Chris Lattner 245ff62e90d0bc321206023897edc1e2691cb0fbb6Chris Lattnerstatic cl::opt<std::string> 255ff62e90d0bc321206023897edc1e2691cb0fbb6Chris LattnerOutputFilename("o", cl::desc("Override output filename"), cl::init("-"), 265ff62e90d0bc321206023897edc1e2691cb0fbb6Chris Lattner cl::value_desc("filename")); 275ff62e90d0bc321206023897edc1e2691cb0fbb6Chris Lattner 285ff62e90d0bc321206023897edc1e2691cb0fbb6Chris Lattnerstatic cl::opt<bool> Force("f", cl::desc("Overwrite output files")); 295ff62e90d0bc321206023897edc1e2691cb0fbb6Chris Lattner 305ff62e90d0bc321206023897edc1e2691cb0fbb6Chris Lattnerstatic cl::opt<bool> 315ff62e90d0bc321206023897edc1e2691cb0fbb6Chris LattnerVerbose("v", cl::desc("Print information about actions taken")); 325ff62e90d0bc321206023897edc1e2691cb0fbb6Chris Lattner 335ff62e90d0bc321206023897edc1e2691cb0fbb6Chris Lattnerstatic cl::opt<bool> 345ff62e90d0bc321206023897edc1e2691cb0fbb6Chris LattnerDumpAsm("d", cl::desc("Print assembly as linked"), cl::Hidden); 355ff62e90d0bc321206023897edc1e2691cb0fbb6Chris Lattner 365ff62e90d0bc321206023897edc1e2691cb0fbb6Chris Lattnerstatic cl::list<std::string> 375ff62e90d0bc321206023897edc1e2691cb0fbb6Chris LattnerLibPaths("L", cl::desc("Specify a library search path"), cl::ZeroOrMore, 385ff62e90d0bc321206023897edc1e2691cb0fbb6Chris Lattner cl::value_desc("directory"), cl::Prefix); 39952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner 40952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner// FileExists - Return true if the specified string is an openable file... 41697954c15da58bd8b186dbafdedd8b06db770201Chris Lattnerstatic inline bool FileExists(const std::string &FN) { 42952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner struct stat StatBuf; 43952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner return stat(FN.c_str(), &StatBuf) != -1; 44952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner} 45952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner 46952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner// LoadFile - Read the specified bytecode file in and return it. This routine 47952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner// searches the link path for the specified file to try to find it... 48952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner// 49697954c15da58bd8b186dbafdedd8b06db770201Chris Lattnerstatic inline std::auto_ptr<Module> LoadFile(const std::string &FN) { 50697954c15da58bd8b186dbafdedd8b06db770201Chris Lattner std::string Filename = FN; 51697954c15da58bd8b186dbafdedd8b06db770201Chris Lattner std::string ErrorMessage; 5265be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner 5365be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner unsigned NextLibPathIdx = 0; 54952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner bool FoundAFile = false; 5565be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner 5665be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner while (1) { 576c8103f7ddb734d128739a463bcd0667d73c62aaChris Lattner if (Verbose) std::cerr << "Loading '" << Filename << "'\n"; 58952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner if (FileExists(Filename)) FoundAFile = true; 5965be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner Module *Result = ParseBytecodeFile(Filename, &ErrorMessage); 6065be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner if (Result) return std::auto_ptr<Module>(Result); // Load successful! 6165be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner 6265be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner if (Verbose) { 636c8103f7ddb734d128739a463bcd0667d73c62aaChris Lattner std::cerr << "Error opening bytecode file: '" << Filename << "'"; 646c8103f7ddb734d128739a463bcd0667d73c62aaChris Lattner if (ErrorMessage.size()) std::cerr << ": " << ErrorMessage; 656c8103f7ddb734d128739a463bcd0667d73c62aaChris Lattner std::cerr << "\n"; 6665be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner } 6765be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner 6865be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner if (NextLibPathIdx == LibPaths.size()) break; 6965be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner Filename = LibPaths[NextLibPathIdx++] + "/" + FN; 7065be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner } 7165be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner 72952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner if (FoundAFile) 736c8103f7ddb734d128739a463bcd0667d73c62aaChris Lattner std::cerr << "Bytecode file '" << FN << "' corrupt! " 743ef3beb1d79f8f05fa55dac2c2f58822bad5e348Misha Brukman << "Use 'llvm-link -v ...' for more info.\n"; 75952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner else 766c8103f7ddb734d128739a463bcd0667d73c62aaChris Lattner std::cerr << "Could not locate bytecode file: '" << FN << "'\n"; 7765be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner return std::auto_ptr<Module>(); 7865be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner} 79075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner 80952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner 81952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner 82952d365a3a446ebfbf14a8db27e26c5c2abec651Chris Lattner 83075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattnerint main(int argc, char **argv) { 845ff62e90d0bc321206023897edc1e2691cb0fbb6Chris Lattner cl::ParseCommandLineOptions(argc, argv, " llvm linker\n"); 85075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner assert(InputFilenames.size() > 0 && "OneOrMore is not working"); 86075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner 8765be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner unsigned BaseArg = 0; 88697954c15da58bd8b186dbafdedd8b06db770201Chris Lattner std::string ErrorMessage; 8965be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner 9065be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner std::auto_ptr<Module> Composite(LoadFile(InputFilenames[BaseArg])); 9165be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner if (Composite.get() == 0) return 1; 9265be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner 9365be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner for (unsigned i = BaseArg+1; i < InputFilenames.size(); ++i) { 94697954c15da58bd8b186dbafdedd8b06db770201Chris Lattner std::auto_ptr<Module> M(LoadFile(InputFilenames[i])); 9565be321fdb92cb8330a22db6a16d5e8b0f3ec165Chris Lattner if (M.get() == 0) return 1; 96b81adf14fbdf9d57b7101bbc710752f864f7ea21Chris Lattner 976c8103f7ddb734d128739a463bcd0667d73c62aaChris Lattner if (Verbose) std::cerr << "Linking in '" << InputFilenames[i] << "'\n"; 98b81adf14fbdf9d57b7101bbc710752f864f7ea21Chris Lattner 99075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner if (LinkModules(Composite.get(), M.get(), &ErrorMessage)) { 1006c8103f7ddb734d128739a463bcd0667d73c62aaChris Lattner std::cerr << argv[0] << ": error linking in '" << InputFilenames[i] 1016c8103f7ddb734d128739a463bcd0667d73c62aaChris Lattner << "': " << ErrorMessage << "\n"; 102075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner return 1; 103075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner } 104075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner } 105075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner 1066c8103f7ddb734d128739a463bcd0667d73c62aaChris Lattner if (DumpAsm) std::cerr << "Here's the assembly:\n" << Composite.get(); 107164cb69e045f91900410781605cff7e5594460f2Chris Lattner 108cf17bcc7e73cf9dc1c8c74dae50a6694fb124f0fAnand Shukla std::ostream *Out = &std::cout; // Default to printing to stdout... 1094a4dabaac05f06a94e1f537f3f987ce1bd9c96e1Chris Lattner if (OutputFilename != "-") { 110888912dbe01c715aa5a0ddec19da6ef12f382ebfChris Lattner if (!Force && std::ifstream(OutputFilename.c_str())) { 111697954c15da58bd8b186dbafdedd8b06db770201Chris Lattner // If force is not specified, make sure not to overwrite a file! 1126c8103f7ddb734d128739a463bcd0667d73c62aaChris Lattner std::cerr << argv[0] << ": error opening '" << OutputFilename 1136c8103f7ddb734d128739a463bcd0667d73c62aaChris Lattner << "': file exists!\n" 1146c8103f7ddb734d128739a463bcd0667d73c62aaChris Lattner << "Use -f command line argument to force output\n"; 115697954c15da58bd8b186dbafdedd8b06db770201Chris Lattner return 1; 116697954c15da58bd8b186dbafdedd8b06db770201Chris Lattner } 117697954c15da58bd8b186dbafdedd8b06db770201Chris Lattner Out = new std::ofstream(OutputFilename.c_str()); 118075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner if (!Out->good()) { 1196c8103f7ddb734d128739a463bcd0667d73c62aaChris Lattner std::cerr << argv[0] << ": error opening '" << OutputFilename << "'!\n"; 120075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner return 1; 121075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner } 12276d12299b5333d71bfc079614031f97c97aa5148Chris Lattner 12376d12299b5333d71bfc079614031f97c97aa5148Chris Lattner // Make sure that the Out file gets unlink'd from the disk if we get a 12476d12299b5333d71bfc079614031f97c97aa5148Chris Lattner // SIGINT 12576d12299b5333d71bfc079614031f97c97aa5148Chris Lattner RemoveFileOnSignal(OutputFilename); 126075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner } 127075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner 128a55c4b1ce71bf62b669d5e533544a29739d56a98Chris Lattner if (verifyModule(*Composite.get())) { 129a55c4b1ce71bf62b669d5e533544a29739d56a98Chris Lattner std::cerr << argv[0] << ": linked module is broken!\n"; 130a55c4b1ce71bf62b669d5e533544a29739d56a98Chris Lattner return 1; 131a55c4b1ce71bf62b669d5e533544a29739d56a98Chris Lattner } 132a55c4b1ce71bf62b669d5e533544a29739d56a98Chris Lattner 1336c8103f7ddb734d128739a463bcd0667d73c62aaChris Lattner if (Verbose) std::cerr << "Writing bytecode...\n"; 134075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner WriteBytecodeToFile(Composite.get(), *Out); 135075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner 136697954c15da58bd8b186dbafdedd8b06db770201Chris Lattner if (Out != &std::cout) delete Out; 137075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner return 0; 138075a0b771e1d76ce64f4628b7a680ef6a0486a4bChris Lattner} 139