1 //===- llvm-mt.cpp - Merge .manifest files ---------------------*- C++ -*-===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===---------------------------------------------------------------------===// 8 // 9 // Merge .manifest files. This is intended to be a platform-independent port 10 // of Microsoft's mt.exe. 11 // 12 //===---------------------------------------------------------------------===// 13 14 #include "llvm/Option/Arg.h" 15 #include "llvm/Option/ArgList.h" 16 #include "llvm/Option/Option.h" 17 #include "llvm/Support/Error.h" 18 #include "llvm/Support/FileOutputBuffer.h" 19 #include "llvm/Support/InitLLVM.h" 20 #include "llvm/Support/ManagedStatic.h" 21 #include "llvm/Support/MemoryBuffer.h" 22 #include "llvm/Support/Path.h" 23 #include "llvm/Support/PrettyStackTrace.h" 24 #include "llvm/Support/Process.h" 25 #include "llvm/Support/Signals.h" 26 #include "llvm/Support/WithColor.h" 27 #include "llvm/Support/raw_ostream.h" 28 #include "llvm/WindowsManifest/WindowsManifestMerger.h" 29 30 #include <system_error> 31 32 using namespace llvm; 33 34 namespace { 35 36 enum ID { 37 OPT_INVALID = 0, // This is not an option ID. 38 #define OPTION(PREFIX, NAME, ID, KIND, GROUP, ALIAS, ALIASARGS, FLAGS, PARAM, \ 39 HELPTEXT, METAVAR, VALUES) \ 40 OPT_##ID, 41 #include "Opts.inc" 42 #undef OPTION 43 }; 44 45 #define PREFIX(NAME, VALUE) const char *const NAME[] = VALUE; 46 #include "Opts.inc" 47 #undef PREFIX 48 49 static const opt::OptTable::Info InfoTable[] = { 50 #define OPTION(PREFIX, NAME, ID, KIND, GROUP, ALIAS, ALIASARGS, FLAGS, PARAM, \ 51 HELPTEXT, METAVAR, VALUES) \ 52 { \ 53 PREFIX, NAME, HELPTEXT, \ 54 METAVAR, OPT_##ID, opt::Option::KIND##Class, \ 55 PARAM, FLAGS, OPT_##GROUP, \ 56 OPT_##ALIAS, ALIASARGS, VALUES}, 57 #include "Opts.inc" 58 #undef OPTION 59 }; 60 61 class CvtResOptTable : public opt::OptTable { 62 public: 63 CvtResOptTable() : OptTable(InfoTable, true) {} 64 }; 65 } // namespace 66 67 LLVM_ATTRIBUTE_NORETURN void reportError(Twine Msg) { 68 WithColor::error(errs(), "llvm-mt") << Msg << '\n'; 69 exit(1); 70 } 71 72 static void reportError(StringRef Input, std::error_code EC) { 73 reportError(Twine(Input) + ": " + EC.message()); 74 } 75 76 void error(std::error_code EC) { 77 if (EC) 78 reportError(EC.message()); 79 } 80 81 void error(Error EC) { 82 if (EC) 83 handleAllErrors(std::move(EC), [&](const ErrorInfoBase &EI) { 84 reportError(EI.message()); 85 }); 86 } 87 88 int main(int Argc, const char **Argv) { 89 InitLLVM X(Argc, Argv); 90 91 CvtResOptTable T; 92 unsigned MAI, MAC; 93 ArrayRef<const char *> ArgsArr = makeArrayRef(Argv + 1, Argc - 1); 94 opt::InputArgList InputArgs = T.ParseArgs(ArgsArr, MAI, MAC); 95 96 for (auto *Arg : InputArgs.filtered(OPT_INPUT)) { 97 auto ArgString = Arg->getAsString(InputArgs); 98 std::string Diag; 99 raw_string_ostream OS(Diag); 100 OS << "invalid option '" << ArgString << "'"; 101 102 std::string Nearest; 103 if (T.findNearest(ArgString, Nearest) < 2) 104 OS << ", did you mean '" << Nearest << "'?"; 105 106 reportError(OS.str()); 107 } 108 109 for (auto &Arg : InputArgs) { 110 if (Arg->getOption().matches(OPT_unsupported)) { 111 outs() << "llvm-mt: ignoring unsupported '" << Arg->getOption().getName() 112 << "' option\n"; 113 } 114 } 115 116 if (InputArgs.hasArg(OPT_help)) { 117 T.PrintHelp(outs(), "llvm-mt [options] file...", "Manifest Tool", false); 118 return 0; 119 } 120 121 std::vector<std::string> InputFiles = InputArgs.getAllArgValues(OPT_manifest); 122 123 if (InputFiles.size() == 0) { 124 reportError("no input file specified"); 125 } 126 127 StringRef OutputFile; 128 if (InputArgs.hasArg(OPT_out)) { 129 OutputFile = InputArgs.getLastArgValue(OPT_out); 130 } else if (InputFiles.size() == 1) { 131 OutputFile = InputFiles[0]; 132 } else { 133 reportError("no output file specified"); 134 } 135 136 windows_manifest::WindowsManifestMerger Merger; 137 138 for (const auto &File : InputFiles) { 139 ErrorOr<std::unique_ptr<MemoryBuffer>> ManifestOrErr = 140 MemoryBuffer::getFile(File); 141 if (!ManifestOrErr) 142 reportError(File, ManifestOrErr.getError()); 143 MemoryBuffer &Manifest = *ManifestOrErr.get(); 144 error(Merger.merge(Manifest)); 145 } 146 147 std::unique_ptr<MemoryBuffer> OutputBuffer = Merger.getMergedManifest(); 148 if (!OutputBuffer) 149 reportError("empty manifest not written"); 150 Expected<std::unique_ptr<FileOutputBuffer>> FileOrErr = 151 FileOutputBuffer::create(OutputFile, OutputBuffer->getBufferSize()); 152 if (!FileOrErr) 153 reportError(OutputFile, errorToErrorCode(FileOrErr.takeError())); 154 std::unique_ptr<FileOutputBuffer> FileBuffer = std::move(*FileOrErr); 155 std::copy(OutputBuffer->getBufferStart(), OutputBuffer->getBufferEnd(), 156 FileBuffer->getBufferStart()); 157 error(FileBuffer->commit()); 158 return 0; 159 } 160