X-Git-Url: http://plrg.eecs.uci.edu/git/?a=blobdiff_plain;f=tools%2Flto%2Flto.cpp;h=8f62929b326bb66dae91008b3d2ce8edd9fb2532;hb=97c9f0cae01cdb52212c8c427cc94b7c8f929f13;hp=bc752340d50a7d7931e7c6e61bf626570837abf6;hpb=e3611871cb5430f9c958cab1be7ef2e0778b3241;p=oota-llvm.git diff --git a/tools/lto/lto.cpp b/tools/lto/lto.cpp index bc752340d50..8f62929b326 100644 --- a/tools/lto/lto.cpp +++ b/tools/lto/lto.cpp @@ -2,498 +2,369 @@ // // The LLVM Compiler Infrastructure // -// This file was developed by Devang Patel and is distributed under -// the University of Illinois Open Source License. See LICENSE.TXT for details. -// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// //===----------------------------------------------------------------------===// // -// This file implements the Link Time Optimization library. This library is +// This file implements the Link Time Optimization library. This library is // intended to be used by linker to optimize code at link time. // //===----------------------------------------------------------------------===// -#include "llvm/Module.h" -#include "llvm/PassManager.h" -#include "llvm/Linker.h" -#include "llvm/Constants.h" -#include "llvm/DerivedTypes.h" -#include "llvm/ModuleProvider.h" -#include "llvm/Bitcode/ReaderWriter.h" -#include "llvm/Support/CommandLine.h" -#include "llvm/Support/FileUtilities.h" -#include "llvm/Support/SystemUtils.h" -#include "llvm/Support/Mangler.h" +#include "llvm-c/lto.h" +#include "llvm/ADT/STLExtras.h" +#include "llvm/CodeGen/CommandFlags.h" +#include "llvm/IR/LLVMContext.h" +#include "llvm/LTO/LTOCodeGenerator.h" +#include "llvm/LTO/LTOModule.h" #include "llvm/Support/MemoryBuffer.h" -#include "llvm/System/Program.h" -#include "llvm/System/Signals.h" -#include "llvm/Analysis/Passes.h" -#include "llvm/Analysis/LoopPass.h" -#include "llvm/Analysis/Verifier.h" -#include "llvm/CodeGen/FileWriters.h" -#include "llvm/Target/SubtargetFeature.h" -#include "llvm/Target/TargetData.h" -#include "llvm/Target/TargetMachine.h" -#include "llvm/Target/TargetMachineRegistry.h" -#include "llvm/Target/TargetAsmInfo.h" -#include "llvm/Transforms/IPO.h" -#include "llvm/Transforms/Scalar.h" -#include "llvm/Analysis/LoadValueNumbering.h" -#include "llvm/Support/MathExtras.h" -#include "llvm/LinkTimeOptimizer.h" -#include -#include -using namespace llvm; - -extern "C" -llvm::LinkTimeOptimizer *createLLVMOptimizer(unsigned VERSION) -{ - if (VERSION != LLVM_LTO_VERSION) - return NULL; - - llvm::LTO *l = new llvm::LTO(); - return l; -} - -/// If symbol is not used then make it internal and let optimizer takes -/// care of it. -void LLVMSymbol::mayBeNotUsed() { - gv->setLinkage(GlobalValue::InternalLinkage); -} - -// Map LLVM LinkageType to LTO LinakgeType -static LTOLinkageTypes -getLTOLinkageType(GlobalValue *v) -{ - LTOLinkageTypes lt; - if (v->hasExternalLinkage()) - lt = LTOExternalLinkage; - else if (v->hasLinkOnceLinkage()) - lt = LTOLinkOnceLinkage; - else if (v->hasWeakLinkage()) - lt = LTOWeakLinkage; - else - // Otherwise it is internal linkage for link time optimizer - lt = LTOInternalLinkage; - return lt; -} - -// Find exeternal symbols referenced by VALUE. This is a recursive function. -static void -findExternalRefs(Value *value, std::set &references, - Mangler &mangler) { - - if (GlobalValue *gv = dyn_cast(value)) { - LTOLinkageTypes lt = getLTOLinkageType(gv); - if (lt != LTOInternalLinkage && strncmp (gv->getName().c_str(), "llvm.", 5)) - references.insert(mangler.getValueName(gv)); +#include "llvm/Support/Signals.h" +#include "llvm/Support/TargetSelect.h" + +// extra command-line flags needed for LTOCodeGenerator +static cl::opt +OptLevel("O", + cl::desc("Optimization level. [-O0, -O1, -O2, or -O3] " + "(default = '-O2')"), + cl::Prefix, + cl::ZeroOrMore, + cl::init('2')); + +static cl::opt +DisableInline("disable-inlining", cl::init(false), + cl::desc("Do not run the inliner pass")); + +static cl::opt +DisableGVNLoadPRE("disable-gvn-loadpre", cl::init(false), + cl::desc("Do not run the GVN load PRE pass")); + +static cl::opt +DisableLTOVectorization("disable-lto-vectorization", cl::init(false), + cl::desc("Do not run loop or slp vectorization during LTO")); + +// Holds most recent error string. +// *** Not thread safe *** +static std::string sLastErrorString; + +// Holds the initialization state of the LTO module. +// *** Not thread safe *** +static bool initialized = false; + +// Holds the command-line option parsing state of the LTO module. +static bool parsedOptions = false; + +// Initialize the configured targets if they have not been initialized. +static void lto_initialize() { + if (!initialized) { +#ifdef LLVM_ON_WIN32 + // Dialog box on crash disabling doesn't work across DLL boundaries, so do + // it here. + llvm::sys::DisableSystemDialogsOnCrash(); +#endif + + InitializeAllTargetInfos(); + InitializeAllTargets(); + InitializeAllTargetMCs(); + InitializeAllAsmParsers(); + InitializeAllAsmPrinters(); + InitializeAllDisassemblers(); + initialized = true; } +} - // GlobalValue, even with InternalLinkage type, may have operands with - // ExternalLinkage type. Do not ignore these operands. - if (Constant *c = dyn_cast(value)) - // Handle ConstantExpr, ConstantStruct, ConstantArry etc.. - for (unsigned i = 0, e = c->getNumOperands(); i != e; ++i) - findExternalRefs(c->getOperand(i), references, mangler); -} - -/// If Module with InputFilename is available then remove it from allModules -/// and call delete on it. -void -LTO::removeModule (const std::string &InputFilename) -{ - NameToModuleMap::iterator pos = allModules.find(InputFilename.c_str()); - if (pos == allModules.end()) - return; - - Module *m = pos->second; - allModules.erase(pos); - delete m; -} - -/// InputFilename is a LLVM bitcode file. If Module with InputFilename is -/// available then return it. Otherwise parseInputFilename. -Module * -LTO::getModule(const std::string &InputFilename) -{ - Module *m = NULL; - - NameToModuleMap::iterator pos = allModules.find(InputFilename.c_str()); - if (pos != allModules.end()) - m = allModules[InputFilename.c_str()]; - else { - if (MemoryBuffer *Buffer - = MemoryBuffer::getFile(&InputFilename[0], InputFilename.size())) { - m = ParseBitcodeFile(Buffer); - delete Buffer; - } - allModules[InputFilename.c_str()] = m; - } - return m; -} - -/// InputFilename is a LLVM bitcode file. Reade this bitcode file and -/// set corresponding target triplet string. -void -LTO::getTargetTriple(const std::string &InputFilename, - std::string &targetTriple) -{ - Module *m = getModule(InputFilename); - if (m) - targetTriple = m->getTargetTriple(); -} - -/// InputFilename is a LLVM bitcode file. Read it using bitcode reader. -/// Collect global functions and symbol names in symbols vector. -/// Collect external references in references vector. -/// Return LTO_READ_SUCCESS if there is no error. -enum LTOStatus -LTO::readLLVMObjectFile(const std::string &InputFilename, - NameToSymbolMap &symbols, - std::set &references) -{ - Module *m = getModule(InputFilename); - if (!m) - return LTO_READ_FAILURE; - - // Collect Target info - getTarget(m); - - if (!Target) - return LTO_READ_FAILURE; - - // Use mangler to add GlobalPrefix to names to match linker names. - // FIXME : Instead of hard coding "-" use GlobalPrefix. - Mangler mangler(*m, Target->getTargetAsmInfo()->getGlobalPrefix()); - modules.push_back(m); - - for (Module::iterator f = m->begin(), e = m->end(); f != e; ++f) { - - LTOLinkageTypes lt = getLTOLinkageType(f); - - if (!f->isDeclaration() && lt != LTOInternalLinkage - && strncmp (f->getName().c_str(), "llvm.", 5)) { - int alignment = ( 16 > f->getAlignment() ? 16 : f->getAlignment()); - LLVMSymbol *newSymbol = new LLVMSymbol(lt, f, f->getName(), - mangler.getValueName(f), - Log2_32(alignment)); - symbols[newSymbol->getMangledName()] = newSymbol; - allSymbols[newSymbol->getMangledName()] = newSymbol; - } +namespace { - // Collect external symbols referenced by this function. - for (Function::iterator b = f->begin(), fe = f->end(); b != fe; ++b) - for (BasicBlock::iterator i = b->begin(), be = b->end(); - i != be; ++i) - for (unsigned count = 0, total = i->getNumOperands(); - count != total; ++count) - findExternalRefs(i->getOperand(count), references, mangler); - } - - for (Module::global_iterator v = m->global_begin(), e = m->global_end(); - v != e; ++v) { - LTOLinkageTypes lt = getLTOLinkageType(v); - if (!v->isDeclaration() && lt != LTOInternalLinkage - && strncmp (v->getName().c_str(), "llvm.", 5)) { - const TargetData *TD = Target->getTargetData(); - LLVMSymbol *newSymbol = new LLVMSymbol(lt, v, v->getName(), - mangler.getValueName(v), - TD->getPreferredAlignmentLog(v)); - symbols[newSymbol->getMangledName()] = newSymbol; - allSymbols[newSymbol->getMangledName()] = newSymbol; - - for (unsigned count = 0, total = v->getNumOperands(); - count != total; ++count) - findExternalRefs(v->getOperand(count), references, mangler); +// This derived class owns the native object file. This helps implement the +// libLTO API semantics, which require that the code generator owns the object +// file. +struct LibLTOCodeGenerator : LTOCodeGenerator { + LibLTOCodeGenerator() {} + LibLTOCodeGenerator(std::unique_ptr Context) + : LTOCodeGenerator(std::move(Context)) {} + std::unique_ptr NativeObjectFile; +}; + +} + +DEFINE_SIMPLE_CONVERSION_FUNCTIONS(LibLTOCodeGenerator, lto_code_gen_t) +DEFINE_SIMPLE_CONVERSION_FUNCTIONS(LTOModule, lto_module_t) + +// Convert the subtarget features into a string to pass to LTOCodeGenerator. +static void lto_add_attrs(lto_code_gen_t cg) { + LTOCodeGenerator *CG = unwrap(cg); + if (MAttrs.size()) { + std::string attrs; + for (unsigned i = 0; i < MAttrs.size(); ++i) { + if (i > 0) + attrs.append(","); + attrs.append(MAttrs[i]); } - } - - return LTO_READ_SUCCESS; -} - -/// Get TargetMachine. -/// Use module M to find appropriate Target. -void -LTO::getTarget (Module *M) { - - if (Target) - return; - - std::string Err; - const TargetMachineRegistry::entry* March = - TargetMachineRegistry::getClosestStaticTargetForModule(*M, Err); - - if (March == 0) - return; - - // Create target - std::string Features; - Target = March->CtorFn(*M, Features); -} - -/// Optimize module M using various IPO passes. Use exportList to -/// internalize selected symbols. Target platform is selected -/// based on information available to module M. No new target -/// features are selected. -enum LTOStatus -LTO::optimize(Module *M, std::ostream &Out, - std::vector &exportList) -{ - // Instantiate the pass manager to organize the passes. - PassManager Passes; - - // Collect Target info - getTarget(M); - - if (!Target) - return LTO_NO_TARGET; - - // Start off with a verification pass. - Passes.add(createVerifierPass()); - - // Add an appropriate TargetData instance for this module... - Passes.add(new TargetData(*Target->getTargetData())); - - // Internalize symbols if export list is nonemty - if (!exportList.empty()) - Passes.add(createInternalizePass(exportList)); - - // Now that we internalized some globals, see if we can hack on them! - Passes.add(createGlobalOptimizerPass()); - - // Linking modules together can lead to duplicated global constants, only - // keep one copy of each constant... - Passes.add(createConstantMergePass()); - - // If the -s command line option was specified, strip the symbols out of the - // resulting program to make it smaller. -s is a GLD option that we are - // supporting. - Passes.add(createStripSymbolsPass()); - - // Propagate constants at call sites into the functions they call. - Passes.add(createIPConstantPropagationPass()); - - // Remove unused arguments from functions... - Passes.add(createDeadArgEliminationPass()); - - Passes.add(createFunctionInliningPass()); // Inline small functions - - Passes.add(createPruneEHPass()); // Remove dead EH info - - Passes.add(createGlobalDCEPass()); // Remove dead functions - - // If we didn't decide to inline a function, check to see if we can - // transform it to pass arguments by value instead of by reference. - Passes.add(createArgumentPromotionPass()); - - // The IPO passes may leave cruft around. Clean up after them. - Passes.add(createInstructionCombiningPass()); - - Passes.add(createScalarReplAggregatesPass()); // Break up allocas - - // Run a few AA driven optimizations here and now, to cleanup the code. - Passes.add(createGlobalsModRefPass()); // IP alias analysis - - Passes.add(createLICMPass()); // Hoist loop invariants - Passes.add(createLoadValueNumberingPass()); // GVN for load instrs - Passes.add(createGCSEPass()); // Remove common subexprs - Passes.add(createDeadStoreEliminationPass()); // Nuke dead stores - - // Cleanup and simplify the code after the scalar optimizations. - Passes.add(createInstructionCombiningPass()); - - // Delete basic blocks, which optimization passes may have killed... - Passes.add(createCFGSimplificationPass()); - - // Now that we have optimized the program, discard unreachable functions... - Passes.add(createGlobalDCEPass()); - - // Make sure everything is still good. - Passes.add(createVerifierPass()); - - FunctionPassManager *CodeGenPasses = - new FunctionPassManager(new ExistingModuleProvider(M)); - - CodeGenPasses->add(new TargetData(*Target->getTargetData())); - - MachineCodeEmitter *MCE = 0; - - switch (Target->addPassesToEmitFile(*CodeGenPasses, Out, - TargetMachine::AssemblyFile, true)) { - default: - case FileModel::Error: - return LTO_WRITE_FAILURE; - case FileModel::AsmFile: - break; - case FileModel::MachOFile: - MCE = AddMachOWriter(*CodeGenPasses, Out, *Target); - break; - case FileModel::ElfFile: - MCE = AddELFWriter(*CodeGenPasses, Out, *Target); - break; + + CG->setAttr(attrs.c_str()); } - if (Target->addPassesToEmitFileFinish(*CodeGenPasses, MCE, true)) - return LTO_WRITE_FAILURE; + if (OptLevel < '0' || OptLevel > '3') + report_fatal_error("Optimization level must be between 0 and 3"); + CG->setOptLevel(OptLevel - '0'); +} - // Run our queue of passes all at once now, efficiently. - Passes.run(*M); +extern const char* lto_get_version() { + return LTOCodeGenerator::getVersionString(); +} - // Run the code generator, if present. - CodeGenPasses->doInitialization(); - for (Module::iterator I = M->begin(), E = M->end(); I != E; ++I) { - if (!I->isDeclaration()) - CodeGenPasses->run(*I); - } - CodeGenPasses->doFinalization(); - - return LTO_OPT_SUCCESS; -} - -///Link all modules together and optimize them using IPO. Generate -/// native object file using OutputFilename -/// Return appropriate LTOStatus. -enum LTOStatus -LTO::optimizeModules(const std::string &OutputFilename, - std::vector &exportList, - std::string &targetTriple, - bool saveTemps, - const char *FinalOutputFilename) -{ - if (modules.empty()) - return LTO_NO_WORK; - - std::ios::openmode io_mode = - std::ios::out | std::ios::trunc | std::ios::binary; - std::string *errMsg = NULL; - Module *bigOne = modules[0]; - Linker theLinker("LinkTimeOptimizer", bigOne, false); - for (unsigned i = 1, e = modules.size(); i != e; ++i) - if (theLinker.LinkModules(bigOne, modules[i], errMsg)) - return LTO_MODULE_MERGE_FAILURE; - // all modules have been handed off to the linker. - modules.clear(); - - sys::Path FinalOutputPath(FinalOutputFilename); - FinalOutputPath.eraseSuffix(); - - if (saveTemps) { - std::string tempFileName(FinalOutputPath.c_str()); - tempFileName += "0.bc"; - std::ofstream Out(tempFileName.c_str(), io_mode); - WriteBitcodeToFile(bigOne, Out); - } +const char* lto_get_error_message() { + return sLastErrorString.c_str(); +} - // Strip leading underscore because it was added to match names - // seen by linker. - for (unsigned i = 0, e = exportList.size(); i != e; ++i) { - const char *name = exportList[i]; - NameToSymbolMap::iterator itr = allSymbols.find(name); - if (itr != allSymbols.end()) - exportList[i] = allSymbols[name]->getName(); - } +bool lto_module_is_object_file(const char* path) { + return LTOModule::isBitcodeFile(path); +} +bool lto_module_is_object_file_for_target(const char* path, + const char* target_triplet_prefix) { + ErrorOr> Buffer = MemoryBuffer::getFile(path); + if (!Buffer) + return false; + return LTOModule::isBitcodeForTarget(Buffer->get(), target_triplet_prefix); +} - std::string ErrMsg; - sys::Path TempDir = sys::Path::GetTemporaryDirectory(&ErrMsg); - if (TempDir.isEmpty()) { - cerr << "lto: " << ErrMsg << "\n"; - return LTO_WRITE_FAILURE; - } - sys::Path tmpAsmFilePath(TempDir); - if (!tmpAsmFilePath.appendComponent("lto")) { - cerr << "lto: " << ErrMsg << "\n"; - TempDir.eraseFromDisk(true); - return LTO_WRITE_FAILURE; - } - if (tmpAsmFilePath.createTemporaryFileOnDisk(true, &ErrMsg)) { - cerr << "lto: " << ErrMsg << "\n"; - TempDir.eraseFromDisk(true); - return LTO_WRITE_FAILURE; - } - sys::RemoveFileOnSignal(tmpAsmFilePath); +bool lto_module_is_object_file_in_memory(const void* mem, size_t length) { + return LTOModule::isBitcodeFile(mem, length); +} - std::ofstream asmFile(tmpAsmFilePath.c_str(), io_mode); - if (!asmFile.is_open() || asmFile.bad()) { - if (tmpAsmFilePath.exists()) { - tmpAsmFilePath.eraseFromDisk(); - TempDir.eraseFromDisk(true); - } - return LTO_WRITE_FAILURE; - } +bool +lto_module_is_object_file_in_memory_for_target(const void* mem, + size_t length, + const char* target_triplet_prefix) { + std::unique_ptr buffer(LTOModule::makeBuffer(mem, length)); + if (!buffer) + return false; + return LTOModule::isBitcodeForTarget(buffer.get(), target_triplet_prefix); +} - enum LTOStatus status = optimize(bigOne, asmFile, exportList); - asmFile.close(); - if (status != LTO_OPT_SUCCESS) { - tmpAsmFilePath.eraseFromDisk(); - TempDir.eraseFromDisk(true); - return status; - } +lto_module_t lto_module_create(const char* path) { + lto_initialize(); + llvm::TargetOptions Options = InitTargetOptionsFromCodeGenFlags(); + return wrap(LTOModule::createFromFile(path, Options, sLastErrorString)); +} - if (saveTemps) { - std::string tempFileName(FinalOutputPath.c_str()); - tempFileName += "1.bc"; - std::ofstream Out(tempFileName.c_str(), io_mode); - WriteBitcodeToFile(bigOne, Out); - } +lto_module_t lto_module_create_from_fd(int fd, const char *path, size_t size) { + lto_initialize(); + llvm::TargetOptions Options = InitTargetOptionsFromCodeGenFlags(); + return wrap( + LTOModule::createFromOpenFile(fd, path, size, Options, sLastErrorString)); +} + +lto_module_t lto_module_create_from_fd_at_offset(int fd, const char *path, + size_t file_size, + size_t map_size, + off_t offset) { + lto_initialize(); + llvm::TargetOptions Options = InitTargetOptionsFromCodeGenFlags(); + return wrap(LTOModule::createFromOpenFileSlice(fd, path, map_size, offset, + Options, sLastErrorString)); +} + +lto_module_t lto_module_create_from_memory(const void* mem, size_t length) { + lto_initialize(); + llvm::TargetOptions Options = InitTargetOptionsFromCodeGenFlags(); + return wrap(LTOModule::createFromBuffer(mem, length, Options, sLastErrorString)); +} + +lto_module_t lto_module_create_from_memory_with_path(const void* mem, + size_t length, + const char *path) { + lto_initialize(); + llvm::TargetOptions Options = InitTargetOptionsFromCodeGenFlags(); + return wrap( + LTOModule::createFromBuffer(mem, length, Options, sLastErrorString, path)); +} + +lto_module_t lto_module_create_in_local_context(const void *mem, size_t length, + const char *path) { + lto_initialize(); + llvm::TargetOptions Options = InitTargetOptionsFromCodeGenFlags(); + return wrap(LTOModule::createInLocalContext(mem, length, Options, + sLastErrorString, path)); +} + +lto_module_t lto_module_create_in_codegen_context(const void *mem, + size_t length, + const char *path, + lto_code_gen_t cg) { + lto_initialize(); + llvm::TargetOptions Options = InitTargetOptionsFromCodeGenFlags(); + return wrap(LTOModule::createInContext(mem, length, Options, sLastErrorString, + path, &unwrap(cg)->getContext())); +} + +void lto_module_dispose(lto_module_t mod) { delete unwrap(mod); } + +const char* lto_module_get_target_triple(lto_module_t mod) { + return unwrap(mod)->getTargetTriple().c_str(); +} - targetTriple = bigOne->getTargetTriple(); - - // Run GCC to assemble and link the program into native code. - // - // Note: - // We can't just assemble and link the file with the system assembler - // and linker because we don't know where to put the _start symbol. - // GCC mysteriously knows how to do it. - const sys::Path gcc = sys::Program::FindProgramByName("gcc"); - if (gcc.isEmpty()) { - tmpAsmFilePath.eraseFromDisk(); - TempDir.eraseFromDisk(true); - return LTO_ASM_FAILURE; +void lto_module_set_target_triple(lto_module_t mod, const char *triple) { + return unwrap(mod)->setTargetTriple(triple); +} + +unsigned int lto_module_get_num_symbols(lto_module_t mod) { + return unwrap(mod)->getSymbolCount(); +} + +const char* lto_module_get_symbol_name(lto_module_t mod, unsigned int index) { + return unwrap(mod)->getSymbolName(index); +} + +lto_symbol_attributes lto_module_get_symbol_attribute(lto_module_t mod, + unsigned int index) { + return unwrap(mod)->getSymbolAttributes(index); +} + +const char* lto_module_get_linkeropts(lto_module_t mod) { + return unwrap(mod)->getLinkerOpts(); +} + +void lto_codegen_set_diagnostic_handler(lto_code_gen_t cg, + lto_diagnostic_handler_t diag_handler, + void *ctxt) { + unwrap(cg)->setDiagnosticHandler(diag_handler, ctxt); +} + +static lto_code_gen_t createCodeGen(bool InLocalContext) { + lto_initialize(); + + TargetOptions Options = InitTargetOptionsFromCodeGenFlags(); + + LibLTOCodeGenerator *CodeGen = + InLocalContext ? new LibLTOCodeGenerator(make_unique()) + : new LibLTOCodeGenerator(); + CodeGen->setTargetOptions(Options); + return wrap(CodeGen); +} + +lto_code_gen_t lto_codegen_create(void) { + return createCodeGen(/* InLocalContext */ false); +} + +lto_code_gen_t lto_codegen_create_in_local_context(void) { + return createCodeGen(/* InLocalContext */ true); +} + +void lto_codegen_dispose(lto_code_gen_t cg) { delete unwrap(cg); } + +bool lto_codegen_add_module(lto_code_gen_t cg, lto_module_t mod) { + return !unwrap(cg)->addModule(unwrap(mod)); +} + +void lto_codegen_set_module(lto_code_gen_t cg, lto_module_t mod) { + unwrap(cg)->setModule(std::unique_ptr(unwrap(mod))); +} + +bool lto_codegen_set_debug_model(lto_code_gen_t cg, lto_debug_model debug) { + unwrap(cg)->setDebugInfo(debug); + return false; +} + +bool lto_codegen_set_pic_model(lto_code_gen_t cg, lto_codegen_model model) { + switch (model) { + case LTO_CODEGEN_PIC_MODEL_STATIC: + unwrap(cg)->setCodePICModel(Reloc::Static); + return false; + case LTO_CODEGEN_PIC_MODEL_DYNAMIC: + unwrap(cg)->setCodePICModel(Reloc::PIC_); + return false; + case LTO_CODEGEN_PIC_MODEL_DYNAMIC_NO_PIC: + unwrap(cg)->setCodePICModel(Reloc::DynamicNoPIC); + return false; + case LTO_CODEGEN_PIC_MODEL_DEFAULT: + unwrap(cg)->setCodePICModel(Reloc::Default); + return false; } + sLastErrorString = "Unknown PIC model"; + return true; +} + +void lto_codegen_set_cpu(lto_code_gen_t cg, const char *cpu) { + return unwrap(cg)->setCpu(cpu); +} + +void lto_codegen_set_assembler_path(lto_code_gen_t cg, const char *path) { + // In here only for backwards compatibility. We use MC now. +} - std::vector args; - args.push_back(gcc.c_str()); - args.push_back("-c"); - args.push_back("-x"); - args.push_back("assembler"); - args.push_back("-o"); - args.push_back(OutputFilename.c_str()); - args.push_back(tmpAsmFilePath.c_str()); - args.push_back(0); - - if (sys::Program::ExecuteAndWait(gcc, &args[0], 0, 0, 1, 0, &ErrMsg)) { - cerr << "lto: " << ErrMsg << "\n"; - return LTO_ASM_FAILURE; +void lto_codegen_set_assembler_args(lto_code_gen_t cg, const char **args, + int nargs) { + // In here only for backwards compatibility. We use MC now. +} + +void lto_codegen_add_must_preserve_symbol(lto_code_gen_t cg, + const char *symbol) { + unwrap(cg)->addMustPreserveSymbol(symbol); +} + +static void maybeParseOptions(lto_code_gen_t cg) { + if (!parsedOptions) { + unwrap(cg)->parseCodeGenDebugOptions(); + lto_add_attrs(cg); + parsedOptions = true; } +} - tmpAsmFilePath.eraseFromDisk(); - TempDir.eraseFromDisk(true); +bool lto_codegen_write_merged_modules(lto_code_gen_t cg, const char *path) { + maybeParseOptions(cg); + return !unwrap(cg)->writeMergedModules(path, sLastErrorString); +} - return LTO_OPT_SUCCESS; +const void *lto_codegen_compile(lto_code_gen_t cg, size_t *length) { + maybeParseOptions(cg); + LibLTOCodeGenerator *CG = unwrap(cg); + CG->NativeObjectFile = CG->compile(DisableInline, DisableGVNLoadPRE, + DisableLTOVectorization, sLastErrorString); + if (!CG->NativeObjectFile) + return nullptr; + *length = CG->NativeObjectFile->getBufferSize(); + return CG->NativeObjectFile->getBufferStart(); } -void LTO::printVersion() { - cl::PrintVersionMessage(); +bool lto_codegen_optimize(lto_code_gen_t cg) { + maybeParseOptions(cg); + return !unwrap(cg)->optimize(DisableInline, + DisableGVNLoadPRE, DisableLTOVectorization, + sLastErrorString); } -/// Unused pure-virtual destructor. Must remain empty. -LinkTimeOptimizer::~LinkTimeOptimizer() {} +const void *lto_codegen_compile_optimized(lto_code_gen_t cg, size_t *length) { + maybeParseOptions(cg); + LibLTOCodeGenerator *CG = unwrap(cg); + CG->NativeObjectFile = CG->compileOptimized(sLastErrorString); + if (!CG->NativeObjectFile) + return nullptr; + *length = CG->NativeObjectFile->getBufferSize(); + return CG->NativeObjectFile->getBufferStart(); +} -/// Destruct LTO. Delete all modules, symbols and target. -LTO::~LTO() { - - for (std::vector::iterator itr = modules.begin(), e = modules.end(); - itr != e; ++itr) - delete *itr; +bool lto_codegen_compile_to_file(lto_code_gen_t cg, const char **name) { + maybeParseOptions(cg); + return !unwrap(cg)->compile_to_file( + name, DisableInline, DisableGVNLoadPRE, + DisableLTOVectorization, sLastErrorString); +} - modules.clear(); +void lto_codegen_debug_options(lto_code_gen_t cg, const char *opt) { + unwrap(cg)->setCodeGenDebugOptions(opt); +} - for (NameToSymbolMap::iterator itr = allSymbols.begin(), e = allSymbols.end(); - itr != e; ++itr) - delete itr->second; +unsigned int lto_api_version() { return LTO_API_VERSION; } - allSymbols.clear(); +void lto_codegen_set_should_internalize(lto_code_gen_t cg, + bool ShouldInternalize) { + unwrap(cg)->setShouldInternalize(ShouldInternalize); +} - delete Target; +void lto_codegen_set_should_embed_uselists(lto_code_gen_t cg, + lto_bool_t ShouldEmbedUselists) { + unwrap(cg)->setShouldEmbedUselists(ShouldEmbedUselists); }