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