llvm-project/lldb/source/Plugins/Process/MacOSX-Kernel/ProcessKDPLog.cpp

//===-- ProcessKDPLog.cpp ---------------------------------------*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#include "ProcessKDPLog.h"

#include "lldb/Core/StreamFile.h"
#include "lldb/Interpreter/Args.h"

#include "ProcessKDP.h"

using namespace lldb;
using namespace lldb_private;

// We want to avoid global constructors where code needs to be run so here we
// control access to our static g_log_sp by hiding it in a singleton function
// that will construct the static g_log_sp the first time this function is
// called.
static bool g_log_enabled = false;
static Log *g_log = NULL;
static Log *GetLog() {
  if (!g_log_enabled)
    return NULL;
  return g_log;
}

Log *ProcessKDPLog::GetLogIfAllCategoriesSet(uint32_t mask) {
  Log *log(GetLog());
  if (log && mask) {
    uint32_t log_mask = log->GetMask().Get();
    if ((log_mask & mask) != mask)
      return NULL;
  }
  return log;
}

void ProcessKDPLog::DisableLog(const char **categories, Stream *feedback_strm) {
  Log *log(GetLog());
  if (log) {
    uint32_t flag_bits = 0;

    if (categories[0] != NULL) {
      flag_bits = log->GetMask().Get();
      for (size_t i = 0; categories[i] != NULL; ++i) {
        const char *arg = categories[i];

        if (::strcasecmp(arg, "all") == 0)
          flag_bits &= ~KDP_LOG_ALL;
        else if (::strcasecmp(arg, "async") == 0)
          flag_bits &= ~KDP_LOG_ASYNC;
        else if (::strncasecmp(arg, "break", 5) == 0)
          flag_bits &= ~KDP_LOG_BREAKPOINTS;
        else if (::strncasecmp(arg, "comm", 4) == 0)
          flag_bits &= ~KDP_LOG_COMM;
        else if (::strcasecmp(arg, "default") == 0)
          flag_bits &= ~KDP_LOG_DEFAULT;
        else if (::strcasecmp(arg, "packets") == 0)
          flag_bits &= ~KDP_LOG_PACKETS;
        else if (::strcasecmp(arg, "memory") == 0)
          flag_bits &= ~KDP_LOG_MEMORY;
        else if (::strcasecmp(arg, "data-short") == 0)
          flag_bits &= ~KDP_LOG_MEMORY_DATA_SHORT;
        else if (::strcasecmp(arg, "data-long") == 0)
          flag_bits &= ~KDP_LOG_MEMORY_DATA_LONG;
        else if (::strcasecmp(arg, "process") == 0)
          flag_bits &= ~KDP_LOG_PROCESS;
        else if (::strcasecmp(arg, "step") == 0)
          flag_bits &= ~KDP_LOG_STEP;
        else if (::strcasecmp(arg, "thread") == 0)
          flag_bits &= ~KDP_LOG_THREAD;
        else if (::strncasecmp(arg, "watch", 5) == 0)
          flag_bits &= ~KDP_LOG_WATCHPOINTS;
        else {
          feedback_strm->Printf("error: unrecognized log category '%s'\n", arg);
          ListLogCategories(feedback_strm);
        }
      }
    }

    log->GetMask().Reset(flag_bits);
    if (flag_bits == 0)
      g_log_enabled = false;
  }

  return;
}

Log *ProcessKDPLog::EnableLog(
    const std::shared_ptr<llvm::raw_ostream> &log_stream_sp,
    uint32_t log_options, const char **categories, Stream *feedback_strm) {
  // Try see if there already is a log - that way we can reuse its settings.
  // We could reuse the log in toto, but we don't know that the stream is the
  // same.
  uint32_t flag_bits = 0;
  if (g_log)
    flag_bits = g_log->GetMask().Get();

  // Now make a new log with this stream if one was provided
  if (log_stream_sp) {
    if (g_log)
      g_log->SetStream(log_stream_sp);
    else
      g_log = new Log(log_stream_sp);
  }

  if (g_log) {
    bool got_unknown_category = false;
    for (size_t i = 0; categories[i] != NULL; ++i) {
      const char *arg = categories[i];

      if (::strcasecmp(arg, "all") == 0)
        flag_bits |= KDP_LOG_ALL;
      else if (::strcasecmp(arg, "async") == 0)
        flag_bits |= KDP_LOG_ASYNC;
      else if (::strncasecmp(arg, "break", 5) == 0)
        flag_bits |= KDP_LOG_BREAKPOINTS;
      else if (::strncasecmp(arg, "comm", 4) == 0)
        flag_bits |= KDP_LOG_COMM;
      else if (::strcasecmp(arg, "default") == 0)
        flag_bits |= KDP_LOG_DEFAULT;
      else if (::strcasecmp(arg, "packets") == 0)
        flag_bits |= KDP_LOG_PACKETS;
      else if (::strcasecmp(arg, "memory") == 0)
        flag_bits |= KDP_LOG_MEMORY;
      else if (::strcasecmp(arg, "data-short") == 0)
        flag_bits |= KDP_LOG_MEMORY_DATA_SHORT;
      else if (::strcasecmp(arg, "data-long") == 0)
        flag_bits |= KDP_LOG_MEMORY_DATA_LONG;
      else if (::strcasecmp(arg, "process") == 0)
        flag_bits |= KDP_LOG_PROCESS;
      else if (::strcasecmp(arg, "step") == 0)
        flag_bits |= KDP_LOG_STEP;
      else if (::strcasecmp(arg, "thread") == 0)
        flag_bits |= KDP_LOG_THREAD;
      else if (::strncasecmp(arg, "watch", 5) == 0)
        flag_bits |= KDP_LOG_WATCHPOINTS;
      else {
        feedback_strm->Printf("error: unrecognized log category '%s'\n", arg);
        if (got_unknown_category == false) {
          got_unknown_category = true;
          ListLogCategories(feedback_strm);
        }
      }
    }
    if (flag_bits == 0)
      flag_bits = KDP_LOG_DEFAULT;
    g_log->GetMask().Reset(flag_bits);
    g_log->GetOptions().Reset(log_options);
  }
  g_log_enabled = true;
  return g_log;
}

void ProcessKDPLog::ListLogCategories(Stream *strm) {
  strm->Printf(
      "Logging categories for '%s':\n"
      "  all - turn on all available logging categories\n"
      "  async - log asynchronous activity\n"
      "  break - log breakpoints\n"
      "  communication - log communication activity\n"
      "  default - enable the default set of logging categories for liblldb\n"
      "  packets - log gdb remote packets\n"
      "  memory - log memory reads and writes\n"
      "  data-short - log memory bytes for memory reads and writes for short "
      "transactions only\n"
      "  data-long - log memory bytes for memory reads and writes for all "
      "transactions\n"
      "  process - log process events and activities\n"
      "  thread - log thread events and activities\n"
      "  step - log step related activities\n"
      "  verbose - enable verbose logging\n"
      "  watch - log watchpoint related activities\n",
      ProcessKDP::GetPluginNameStatic().GetCString());
}

void ProcessKDPLog::LogIf(uint32_t mask, const char *format, ...) {
  Log *log(ProcessKDPLog::GetLogIfAllCategoriesSet(mask));
  if (log) {
    va_list args;
    va_start(args, format);
    log->VAPrintf(format, args);
    va_end(args);
  }
}