Register chrome as a browser when setting the default browser in Win 10

chrome/browser/shell_integration_win.cc

 // Copyright (c) 2012 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 "chrome/browser/shell_integration.h"
 
 #include <windows.h>
 #include <shlwapi.h>
 #include <shobjidl.h>
 #include <propkey.h>  // Needs to come after shobjidl.h.
@@ skipping 267 matching lines @@
   const std::string group_name =
       base::FieldTrialList::FindFullName("AsyncSetAsDefault");
   if (CommandLine::ForCurrentProcess()->HasSwitch(kDisableAsyncSetAsDefault))
     return false;
   if (CommandLine::ForCurrentProcess()->HasSwitch(kEnableAsyncSetAsDefault))
     return true;
 
   return base::StartsWith(group_name, "Enabled", base::CompareCase::SENSITIVE);
 }
 
+bool RegisterBrowser() {
+  base::FilePath chrome_exe;
+  if (!PathService::Get(base::FILE_EXE, &chrome_exe)) {
+    NOTREACHED() << "Error getting app exe path";
+    return false;
+  }
+  BrowserDistribution* dist = BrowserDistribution::GetDistribution();
+
+  return ShellUtil::RegisterChromeBrowser(dist, chrome_exe, base::string16(),
+                                          true);
+}
+
 }  // namespace
 
 // static
 bool ShellIntegration::IsSetAsDefaultAsynchronous() {
   return base::win::GetVersion() >= base::win::VERSION_WIN10 &&
          IsAsyncSetAsDefaultEnabled();
 }
 
 bool ShellIntegration::SetAsDefaultBrowser() {
   base::FilePath chrome_exe;
@@ skipping 350 matching lines @@
 bool ShellIntegration::DefaultBrowserWorker::InitializeSetAsDefault() {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 
   if (!IsSetAsDefaultAsynchronous())
     return false;
 
   // On Windows 10+, there is no official way to prompt the user to set a
   // default browser. This is the workaround:
   // 1. Unregister the default browser.
   // 2. Open "How to make Chrome my default browser" link with openwith.exe.
-  // 3. Windows will prompt the user with "How would you link to open this?".
+  // 3. Windows will prompt the user with "How would you like to open this?".
   // 4. If Chrome is selected, we intercept the attempt to open the URL and
   //    instead call OnSetAsDefaultAttemptComplete(), passing true to indicate
   //    success.
   // 5. If Chrome is not selected, the url is opened in the selected browser.
   //    After a certain amount of time, we notify the observer that the
   //    process failed.
 
   if (!StartupBrowserCreator::SetDefaultBrowserCallback(
           base::Bind(&DefaultBrowserWorker::OnSetAsDefaultAttemptComplete, this,
                      AttemptResult::SUCCESS))) {
@@ skipping 29 matching lines @@
   DCHECK(set_as_default_initialized());
 
   async_timer_.reset();
   StartupBrowserCreator::ClearDefaultBrowserCallback();
 }
 
 // static
 bool ShellIntegration::DefaultBrowserWorker::SetAsDefaultBrowserAsynchronous() {
   DCHECK(IsSetAsDefaultAsynchronous());
 
+  // Registers chrome.exe as a browser on Windows to make sure it will be shown
+  // in the "How would you like to open this?" prompt.
+  if (!RegisterBrowser())
+    return false;
+
   ResetDefaultBrowser();
 
   base::CommandLine cmdline(base::FilePath(L"openwith.exe"));
   cmdline.AppendArgNative(StartupBrowserCreator::GetDefaultBrowserUrl());
   return base::LaunchProcess(cmdline, base::LaunchOptions()).IsValid();
 }