contrib/llvm/tools/lldb/source/Commands/CommandObjectStats.cpp

   1 //===-- CommandObjectStats.cpp ----------------------------------*- 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 #include "CommandObjectStats.h"
  11 #include "lldb/Host/Host.h"
  12 #include "lldb/Interpreter/CommandInterpreter.h"
  13 #include "lldb/Interpreter/CommandReturnObject.h"
  14 #include "lldb/Target/Target.h"
  15
  16 using namespace lldb;
  17 using namespace lldb_private;
  18
  19 class CommandObjectStatsEnable : public CommandObjectParsed {
  20 public:
  21   CommandObjectStatsEnable(CommandInterpreter &interpreter)
  22       : CommandObjectParsed(interpreter, "enable",
  23                             "Enable statistics collection", nullptr,
  24                             eCommandProcessMustBePaused) {}
  25
  26   ~CommandObjectStatsEnable() override = default;
  27
  28 protected:
  29   bool DoExecute(Args &command, CommandReturnObject &result) override {
  30     Target *target = GetSelectedOrDummyTarget();
  31
  32     if (target->GetCollectingStats()) {
  33       result.AppendError("statistics already enabled");
  34       result.SetStatus(eReturnStatusFailed);
  35       return false;
  36     }
  37
  38     target->SetCollectingStats(true);
  39     result.SetStatus(eReturnStatusSuccessFinishResult);
  40     return true;
  41   }
  42 };
  43
  44 class CommandObjectStatsDisable : public CommandObjectParsed {
  45 public:
  46   CommandObjectStatsDisable(CommandInterpreter &interpreter)
  47       : CommandObjectParsed(interpreter, "disable",
  48                             "Disable statistics collection", nullptr,
  49                             eCommandProcessMustBePaused) {}
  50
  51   ~CommandObjectStatsDisable() override = default;
  52
  53 protected:
  54   bool DoExecute(Args &command, CommandReturnObject &result) override {
  55     Target *target = GetSelectedOrDummyTarget();
  56
  57     if (!target->GetCollectingStats()) {
  58       result.AppendError("need to enable statistics before disabling them");
  59       result.SetStatus(eReturnStatusFailed);
  60       return false;
  61     }
  62
  63     target->SetCollectingStats(false);
  64     result.SetStatus(eReturnStatusSuccessFinishResult);
  65     return true;
  66   }
  67 };
  68
  69 class CommandObjectStatsDump : public CommandObjectParsed {
  70 public:
  71   CommandObjectStatsDump(CommandInterpreter &interpreter)
  72       : CommandObjectParsed(interpreter, "dump", "Dump statistics results",
  73                             nullptr, eCommandProcessMustBePaused) {}
  74
  75   ~CommandObjectStatsDump() override = default;
  76
  77 protected:
  78   bool DoExecute(Args &command, CommandReturnObject &result) override {
  79     Target *target = GetSelectedOrDummyTarget();
  80
  81     uint32_t i = 0;
  82     for (auto &stat : target->GetStatistics()) {
  83       result.AppendMessageWithFormat(
  84           "%s : %u\n",
  85           lldb_private::GetStatDescription(static_cast<lldb_private::StatisticKind>(i))
  86               .c_str(),
  87           stat);
  88       i += 1;
  89     }
  90     result.SetStatus(eReturnStatusSuccessFinishResult);
  91     return true;
  92   }
  93 };
  94
  95 CommandObjectStats::CommandObjectStats(CommandInterpreter &interpreter)
  96     : CommandObjectMultiword(interpreter, "statistics",
  97                              "Print statistics about a debugging session",
  98                              "statistics <subcommand> [<subcommand-options>]") {
  99   LoadSubCommand("enable",
 100                  CommandObjectSP(new CommandObjectStatsEnable(interpreter)));
 101   LoadSubCommand("disable",
 102                  CommandObjectSP(new CommandObjectStatsDisable(interpreter)));
 103   LoadSubCommand("dump",
 104                  CommandObjectSP(new CommandObjectStatsDump(interpreter)));
 105 }
 106
 107 CommandObjectStats::~CommandObjectStats() = default;