| //===-- llvm-modextract.cpp - LLVM module extractor utility ---------------===// |
| // |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| // |
| //===----------------------------------------------------------------------===// |
| // |
| // This program is for testing features that rely on multi-module bitcode files. |
| // It takes a multi-module bitcode file, extracts one of the modules and writes |
| // it to the output file. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "llvm/Bitcode/BitcodeReader.h" |
| #include "llvm/Bitcode/BitcodeWriter.h" |
| #include "llvm/Support/CommandLine.h" |
| #include "llvm/Support/Error.h" |
| #include "llvm/Support/FileSystem.h" |
| #include "llvm/Support/ToolOutputFile.h" |
| |
| using namespace llvm; |
| |
| static cl::opt<bool> |
| BinaryExtract("b", cl::desc("Whether to perform binary extraction")); |
| |
| static cl::opt<std::string> OutputFilename("o", cl::Required, |
| cl::desc("Output filename"), |
| cl::value_desc("filename")); |
| |
| static cl::opt<std::string> |
| InputFilename(cl::Positional, cl::desc("<input bitcode>"), cl::init("-")); |
| |
| static cl::opt<unsigned> ModuleIndex("n", cl::Required, |
| cl::desc("Index of module to extract"), |
| cl::value_desc("index")); |
| |
| int main(int argc, char **argv) { |
| cl::ParseCommandLineOptions(argc, argv, "Module extractor"); |
| |
| ExitOnError ExitOnErr("llvm-modextract: error: "); |
| |
| std::unique_ptr<MemoryBuffer> MB = |
| ExitOnErr(errorOrToExpected(MemoryBuffer::getFileOrSTDIN(InputFilename))); |
| std::vector<BitcodeModule> Ms = ExitOnErr(getBitcodeModuleList(*MB)); |
| |
| LLVMContext Context; |
| if (ModuleIndex >= Ms.size()) { |
| errs() << "llvm-modextract: error: module index out of range; bitcode file " |
| "contains " |
| << Ms.size() << " module(s)\n"; |
| return 1; |
| } |
| |
| std::error_code EC; |
| std::unique_ptr<ToolOutputFile> Out( |
| new ToolOutputFile(OutputFilename, EC, sys::fs::OF_None)); |
| ExitOnErr(errorCodeToError(EC)); |
| |
| if (BinaryExtract) { |
| SmallVector<char, 0> Result; |
| BitcodeWriter Writer(Result); |
| Result.append(Ms[ModuleIndex].getBuffer().begin(), |
| Ms[ModuleIndex].getBuffer().end()); |
| Writer.copyStrtab(Ms[ModuleIndex].getStrtab()); |
| Out->os() << Result; |
| Out->keep(); |
| return 0; |
| } |
| |
| std::unique_ptr<Module> M = ExitOnErr(Ms[ModuleIndex].parseModule(Context)); |
| WriteBitcodeToFile(*M, Out->os()); |
| |
| Out->keep(); |
| return 0; |
| } |