Re-apply r23841 "During uninstall if Chrome is set as default"

BUG=14023
TEST=Make sure the option of choosing a different browser shows up during un

Review URL: http://codereview.chromium.org/174194

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@23915 0039d316-1c4b-4281-b951-d872f2087c98

2 files changed, 32 insertions, 0 deletions

diff --git a/chrome/installer/util/shell_util.cc b/chrome/installer/util/shell_util.cc
index 88674dc..f21aeab 100644
--- a/chrome/installer/util/shell_util.cc
+++ b/chrome/installer/util/shell_util.cc
@@ -583,6 +583,32 @@ bool ShellUtil::GetQuickLaunchPath(bool system_level, std::wstring* path) {
   return true;
 }
 
+void ShellUtil::GetRegisteredBrowsers(std::map<std::wstring,
+                                      std::wstring>* browsers) {
+  std::wstring base_key(ShellUtil::kRegStartMenuInternet);
+  HKEY root = HKEY_LOCAL_MACHINE;
+  for (RegistryKeyIterator iter(root, base_key.c_str()); iter.Valid(); ++iter) {
+    std::wstring key = base_key + L"\\" + iter.Name();
+    RegKey capabilities(root, (key + L"\\Capabilities").c_str());
+    std::wstring name;
+    if (!capabilities.Valid() ||
+        !capabilities.ReadValue(L"ApplicationName", &name)) {
+      RegKey base_key(root, key.c_str());
+      if (!base_key.ReadValue(L"", &name))
+        continue;
+    }
+    RegKey install_info(root, (key + L"\\InstallInfo").c_str());
+    std::wstring command;
+    if (!install_info.Valid() ||
+        !install_info.ReadValue(L"ReinstallCommand", &command))
+      continue;
+    BrowserDistribution* dist = BrowserDistribution::GetDistribution();
+    if (!name.empty() && !command.empty() &&
+        name.find(dist->GetApplicationName()) == std::wstring::npos)
+      (*browsers)[name] = command;
+  }
+}
+
 bool ShellUtil::GetUserSpecificDefaultBrowserSuffix(std::wstring* entry) {
   wchar_t user_name[256];
   DWORD size = _countof(user_name);
diff --git a/chrome/installer/util/shell_util.h b/chrome/installer/util/shell_util.h
index ca6eac7..99f88b8 100644
--- a/chrome/installer/util/shell_util.h
+++ b/chrome/installer/util/shell_util.h
@@ -10,6 +10,7 @@
 #define CHROME_INSTALLER_UTIL_SHELL_UTIL_H_
 
 #include <windows.h>
+#include <map>
 #include <string>
 
 #include "base/basictypes.h"
@@ -139,6 +140,11 @@ class ShellUtil {
   // User's profile only affects any new user profiles (not existing ones).
   static bool GetQuickLaunchPath(bool system_level, std::wstring* path);
 
+  // Gets a mapping of all registered browser (on local machine) names and
+  // thier reinstall command (which usually sets browser as default).
+  static void GetRegisteredBrowsers(std::map<std::wstring,
+                                    std::wstring>* browsers);
+
   // This function gets a suffix (user's login name) that can be added
   // to Chromium default browser entry in the registry to create a unique name
   // if there are multiple users on the machine, each with their own copy of