Move crash keys for command-line switches to components/crash so they can be set

components/crash/core/common/crash_keys.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/crash/core/common/crash_keys.h"
 
+#include "base/command_line.h"
 #include "base/debug/crash_logging.h"
 #include "base/format_macros.h"
+#include "base/logging.h"
 #include "base/strings/string_piece.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
+#include "base/strings/utf_string_conversions.h"
 
 namespace crash_keys {
 
 #if defined(OS_MACOSX)
 // Crashpad owns the "guid" key. Chrome's metrics client ID is a separate ID
 // carried in a distinct "metrics_client_id" field.
 const char kMetricsClientId[] = "metrics_client_id";
 #else
 const char kClientId[] = "guid";
 #endif
 
 const char kChannel[] = "channel";
 
 const char kNumVariations[] = "num-experiments";
 const char kVariations[] = "variations";
 
+const char kSwitchFormat[] = "switch-%" PRIuS;
+const char kNumSwitches[] = "num-switches";
+
 const char kBug464926CrashKey[] = "bug-464926-info";
 
 #if defined(OS_MACOSX)
 namespace mac {
 
 const char kZombie[] = "zombie";
 const char kZombieTrace[] = "zombie_dealloc_bt";
 
 }  // namespace mac
 #endif
@@ skipping 47 matching lines @@
     // Do not truncate an individual experiment.
     if (variations_string.size() + variation.size() >= kLargeSize)
       break;
     variations_string += variation;
     variations_string += ",";
   }
 
   base::debug::SetCrashKeyValue(kVariations, variations_string);
 }
 
+void GetCrashKeysForCommandLineSwitches(
+    std::vector<base::debug::CrashKey>* keys) {
+  DCHECK(keys);
+  base::debug::CrashKey crash_key = { kNumSwitches, kSmallSize };
+  keys->push_back(crash_key);
+
+  // Use static storage for formatted key names, since they will persist for
+  // the duration of the program.
+  static char formatted_keys[kSwitchesMaxCount][sizeof(kSwitchFormat) + 1] =
+      {{ 0 }};
+  const size_t formatted_key_len = sizeof(formatted_keys[0]);
+
+  // sizeof(kSwitchFormat) is platform-dependent, so make sure formatted_keys
+  // actually have space for a 2-digit switch number plus null-terminator.
+  static_assert(formatted_key_len >= 10,
+                "insufficient space for \"switch-NN\"");
+
+  for (size_t i = 0; i < kSwitchesMaxCount; ++i) {
+    // Name the keys using 1-based indexing.
+    int n = base::snprintf(formatted_keys[i], formatted_key_len, kSwitchFormat,
+                           i + 1);
+    DCHECK_GT(n, 0);
+    base::debug::CrashKey crash_key = { formatted_keys[i], kSmallSize };
+    keys->push_back(crash_key);
+  }
+}
+
+void SetSwitchesFromCommandLine(const base::CommandLine& command_line,
+                                SwitchFilterFunction skip_filter) {
+  const base::CommandLine::StringVector& argv = command_line.argv();
+
+  // Set the number of switches in case size > kNumSwitches.
+  base::debug::SetCrashKeyValue(kNumSwitches,
+      base::StringPrintf("%" PRIuS, argv.size() - 1));
+
+  size_t key_i = 1;  // Key names are 1-indexed.
+
+  // Go through the argv, skipping the exec path. Stop if there are too many
+  // switches to hold in crash keys.
+  for (size_t i = 1; i < argv.size() && key_i <= crash_keys::kSwitchesMaxCount;
+       ++i) {
+#if defined(OS_WIN)
+    std::string switch_str = base::WideToUTF8(argv[i]);
+#else
+    std::string switch_str = argv[i];
+#endif
+
+    // Skip uninteresting switches.
+    if (skip_filter && (*skip_filter)(switch_str))
+      continue;
+
+    std::string key = base::StringPrintf(kSwitchFormat, key_i++);
+    base::debug::SetCrashKeyValue(key, switch_str);
+  }
+
+  // Clear any remaining switches.
+  for (; key_i <= kSwitchesMaxCount; ++key_i)
+    base::debug::ClearCrashKey(base::StringPrintf(kSwitchFormat, key_i));
+}
+
 }  // namespace crash_keys