1*a9ac8606Spatrick //===-- APINotesTest.cpp - API Notes Testing Tool ------------------ C++ --===//
2*a9ac8606Spatrick //
3*a9ac8606Spatrick // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*a9ac8606Spatrick // See https://llvm.org/LICENSE.txt for license information.
5*a9ac8606Spatrick // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*a9ac8606Spatrick //
7*a9ac8606Spatrick //===----------------------------------------------------------------------===//
8*a9ac8606Spatrick
9*a9ac8606Spatrick #include "clang/APINotes/APINotesYAMLCompiler.h"
10*a9ac8606Spatrick #include "llvm/Support/CommandLine.h"
11*a9ac8606Spatrick #include "llvm/Support/FileSystem.h"
12*a9ac8606Spatrick #include "llvm/Support/MemoryBuffer.h"
13*a9ac8606Spatrick #include "llvm/Support/Signals.h"
14*a9ac8606Spatrick #include "llvm/Support/ToolOutputFile.h"
15*a9ac8606Spatrick #include "llvm/Support/WithColor.h"
16*a9ac8606Spatrick
17*a9ac8606Spatrick static llvm::cl::list<std::string> APINotes(llvm::cl::Positional,
18*a9ac8606Spatrick llvm::cl::desc("[<apinotes> ...]"),
19*a9ac8606Spatrick llvm::cl::Required);
20*a9ac8606Spatrick
21*a9ac8606Spatrick static llvm::cl::opt<std::string>
22*a9ac8606Spatrick OutputFileName("o", llvm::cl::desc("output filename"),
23*a9ac8606Spatrick llvm::cl::value_desc("filename"), llvm::cl::init("-"));
24*a9ac8606Spatrick
main(int argc,const char ** argv)25*a9ac8606Spatrick int main(int argc, const char **argv) {
26*a9ac8606Spatrick const bool DisableCrashReporting = true;
27*a9ac8606Spatrick llvm::sys::PrintStackTraceOnErrorSignal(argv[0], DisableCrashReporting);
28*a9ac8606Spatrick llvm::cl::ParseCommandLineOptions(argc, argv);
29*a9ac8606Spatrick
30*a9ac8606Spatrick auto Error = [](const llvm::Twine &Msg) {
31*a9ac8606Spatrick llvm::WithColor::error(llvm::errs(), "apinotes-test") << Msg << '\n';
32*a9ac8606Spatrick };
33*a9ac8606Spatrick
34*a9ac8606Spatrick std::error_code EC;
35*a9ac8606Spatrick auto Out = std::make_unique<llvm::ToolOutputFile>(OutputFileName, EC,
36*a9ac8606Spatrick llvm::sys::fs::OF_None);
37*a9ac8606Spatrick if (EC) {
38*a9ac8606Spatrick Error("failed to open '" + OutputFileName + "': " + EC.message());
39*a9ac8606Spatrick return EXIT_FAILURE;
40*a9ac8606Spatrick }
41*a9ac8606Spatrick
42*a9ac8606Spatrick for (const std::string &Notes : APINotes) {
43*a9ac8606Spatrick llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> NotesOrError =
44*a9ac8606Spatrick llvm::MemoryBuffer::getFileOrSTDIN(Notes);
45*a9ac8606Spatrick if (std::error_code EC = NotesOrError.getError()) {
46*a9ac8606Spatrick llvm::errs() << EC.message() << '\n';
47*a9ac8606Spatrick return EXIT_FAILURE;
48*a9ac8606Spatrick }
49*a9ac8606Spatrick
50*a9ac8606Spatrick clang::api_notes::parseAndDumpAPINotes((*NotesOrError)->getBuffer(),
51*a9ac8606Spatrick Out->os());
52*a9ac8606Spatrick }
53*a9ac8606Spatrick
54*a9ac8606Spatrick return EXIT_SUCCESS;
55*a9ac8606Spatrick }
56