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