1 //===- llvm-cov.cpp - LLVM coverage tool ----------------------------------===// 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 // llvm-cov is a command line tools to analyze and report coverage information. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #include "llvm/ADT/StringRef.h" 15 #include "llvm/ADT/StringSwitch.h" 16 #include "llvm/Support/CommandLine.h" 17 #include "llvm/Support/ManagedStatic.h" 18 #include "llvm/Support/Path.h" 19 #include "llvm/Support/PrettyStackTrace.h" 20 #include "llvm/Support/Process.h" 21 #include "llvm/Support/Signals.h" 22 #include "llvm/Support/raw_ostream.h" 23 #include <string> 24 25 using namespace llvm; 26 27 /// \brief The main entry point for the 'show' subcommand. 28 int showMain(int argc, const char *argv[]); 29 30 /// \brief The main entry point for the 'report' subcommand. 31 int reportMain(int argc, const char *argv[]); 32 33 /// \brief The main entry point for the 'convert-for-testing' subcommand. 34 int convertForTestingMain(int argc, const char *argv[]); 35 36 /// \brief The main entry point for the gcov compatible coverage tool. 37 int gcovMain(int argc, const char *argv[]); 38 39 /// \brief Top level help. 40 static int helpMain(int argc, const char *argv[]) { 41 errs() << "Usage: llvm-cov {gcov|report|show} [OPTION]...\n\n" 42 << "Shows code coverage information.\n\n" 43 << "Subcommands:\n" 44 << " gcov: Work with the gcov format.\n" 45 << " show: Annotate source files using instrprof style coverage.\n" 46 << " report: Summarize instrprof style coverage information.\n"; 47 return 0; 48 } 49 50 /// \brief Top level version information. 51 static int versionMain(int argc, const char *argv[]) { 52 cl::PrintVersionMessage(); 53 return 0; 54 } 55 56 int main(int argc, const char **argv) { 57 // Print a stack trace if we signal out. 58 sys::PrintStackTraceOnErrorSignal(argv[0]); 59 PrettyStackTraceProgram X(argc, argv); 60 llvm_shutdown_obj Y; // Call llvm_shutdown() on exit. 61 62 // If argv[0] is or ends with 'gcov', always be gcov compatible 63 if (sys::path::stem(argv[0]).endswith_lower("gcov")) 64 return gcovMain(argc, argv); 65 66 // Check if we are invoking a specific tool command. 67 if (argc > 1) { 68 typedef int (*MainFunction)(int, const char *[]); 69 MainFunction Func = StringSwitch<MainFunction>(argv[1]) 70 .Case("convert-for-testing", convertForTestingMain) 71 .Case("gcov", gcovMain) 72 .Case("report", reportMain) 73 .Case("show", showMain) 74 .Cases("-h", "-help", "--help", helpMain) 75 .Cases("-version", "--version", versionMain) 76 .Default(nullptr); 77 78 if (Func) { 79 std::string Invocation = std::string(argv[0]) + " " + argv[1]; 80 argv[1] = Invocation.c_str(); 81 return Func(argc - 1, argv + 1); 82 } 83 } 84 85 if (argc > 1) { 86 if (sys::Process::StandardErrHasColors()) 87 errs().changeColor(raw_ostream::RED); 88 errs() << "Unrecognized command: " << argv[1] << ".\n\n"; 89 if (sys::Process::StandardErrHasColors()) 90 errs().resetColor(); 91 } 92 helpMain(argc, argv); 93 return 1; 94 } 95