Implement the persistent metro mode setting (win8)

win8/delegate_execute/command_execute_impl.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.
 // Implementation of the CommandExecuteImpl class which implements the
 // IExecuteCommand and related interfaces for handling ShellExecute based
 // launches of the Chrome browser.
 
 #include "win8/delegate_execute/command_execute_impl.h"
 
 #include "base/file_util.h"
 #include "base/path_service.h"
 #include "base/utf_string_conversions.h"
+#include "base/win/registry.h"
 #include "base/win/scoped_co_mem.h"
 #include "base/win/scoped_handle.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
 #include "win8/delegate_execute/chrome_util.h"
 
 // CommandExecuteImpl is resposible for activating chrome in Windows 8. The
 // flow is complicated and this tries to highlight the important events.
 // The current approach is to have a single instance of chrome either
@@ skipping 15 matching lines @@
 //    CommandExecuteImpl::SetParameter
 //    CommandExecuteImpl::SetNoShowUI
 //    CommandExecuteImpl::SetSelection
 //    CommandExecuteImpl::Initialize
 //    Up to this point the code basically just gathers values passed in, like
 //    the launch scheme (or url) and the activation verb.
 // 5- Windows calls CommandExecuteImpl::Getvalue()
 //    Here we need to return AHE_IMMERSIVE or AHE_DESKTOP. That depends on:
 //    a) if run in high-integrity return AHE_DESKTOP
 //    b) if chrome is running return the AHE_ mode of chrome
-//    c) if the current process is inmmersive return AHE_IMMERSIVE
-//    d) if the protocol is file and IExecuteCommandHost::GetUIMode() is not
-//       ECHUIM_DESKTOP then return AHE_IMMERSIVE
-//    e) if none of the above return AHE_DESKTOP
+//    c) else we return what GetLaunchMode() tells us, which is:
+//       i) if the command line --force-xxx is present return that
+//       ii) if the registry 'launch_mode' exists return that
+//       iii) if IsMachineATablet() is true return AHE_IMMERSIVE
+//       iv) else return AHE_DESKTOP
 // 6- If we returned AHE_DESKTOP in step 5 then CommandExecuteImpl::Execute()
-//    is called, here we call GetLaunchMode() which:
-//    a) if chrome is running return the mode of chrome or
-//    b) return IExecuteCommandHost::GetUIMode()
+//    is called, here we call GetLaunchMode() which returns the cached answer
+//    computed at step 5c.
 // 7- If GetLaunchMode() returns
 //    a) ECHUIM_DESKTOP we call LaunchDestopChrome() that calls ::CreateProcess
 //    b) else we call one of the IApplicationActivationManager activation
 //       functions depending on the parameters passed in step 4.
 //
-//  Some examples will help clarify the common cases.
-//
-//  I - No chrome running, taskbar icon launch:
-//      a) Scheme is 'file', Verb is 'open'
-//      b) GetValue() returns at e) step : AHE_DESKTOP
-//      c) Execute() calls LaunchDestopChrome()
-//      --> desktop chrome runs
-//  II- No chrome running, tile activation launch:
-//      a) Scheme is 'file', Verb is 'open'
-//      b) GetValue() returns at d) step : AHE_IMMERSIVE
-//      c) Windows does not call us back and just activates chrome
-//      --> metro chrome runs
-//  III- No chrome running, url link click on metro app:
-//      a) Scheme is 'http', Verb is 'open'
-//      b) Getvalue() returns at e) step : AHE_DESKTOP
-//      c) Execute() calls IApplicationActivationManager::ActivateForProtocol
-//      --> metro chrome runs
+// Note that if a command line --force-xxx is present we write that launch mode
+// in the registry so next time the logic reaches 5c-ii it will use the same
+// mode again.
 //
 CommandExecuteImpl::CommandExecuteImpl()
     : integrity_level_(base::INTEGRITY_UNKNOWN),
       launch_scheme_(INTERNET_SCHEME_DEFAULT),
       chrome_mode_(ECHUIM_SYSTEM_LAUNCHER) {
   memset(&start_info_, 0, sizeof(start_info_));
   start_info_.cb = sizeof(start_info_);
   // We need to query the user data dir of chrome so we need chrome's
   // path provider.
   chrome::RegisterPathProvider();
@@ skipping 82 matching lines @@
       chrome_mode_ = ECHUIM_IMMERSIVE;
       *pahe = AHE_IMMERSIVE;
     } else {
       AtlTrace("Chrome [%d] is Desktop, AHE_DESKTOP\n");
       chrome_mode_ = ECHUIM_DESKTOP;
       *pahe = AHE_DESKTOP;
     }
     return S_OK;
   }
 
-  if (IsImmersiveProcess(GetCurrentProcess())) {
-    AtlTrace("Current process is inmmersive, AHE_IMMERSIVE\n");
-    *pahe = AHE_IMMERSIVE;
-    return S_OK;
-  }
-
-  if ((launch_scheme_ == INTERNET_SCHEME_FILE) &&
-      (GetLaunchMode() != ECHUIM_DESKTOP)) {
-    AtlTrace("INTERNET_SCHEME_FILE, mode != ECHUIM_DESKTOP, AHE_IMMERSIVE\n");
-    *pahe = AHE_IMMERSIVE;
-    return S_OK;
-  }
-
-  AtlTrace("Fallback is AHE_DESKTOP\n");
-  *pahe = AHE_DESKTOP;
+  EC_HOST_UI_MODE mode = GetLaunchMode();
+  *pahe = (mode == ECHUIM_DESKTOP) ? AHE_DESKTOP : AHE_IMMERSIVE;
   return S_OK;
 }
 
 STDMETHODIMP CommandExecuteImpl::Execute() {
   AtlTrace("In %hs\n", __FUNCTION__);
 
   if (integrity_level_ == base::HIGH_INTEGRITY)
     return LaunchDesktopChrome();
 
   EC_HOST_UI_MODE mode = GetLaunchMode();
@@ skipping 187 matching lines @@
     CloseHandle(proc_info.hProcess);
     CloseHandle(proc_info.hThread);
   } else {
     AtlTrace("Process launch failed, error %d\n", ::GetLastError());
   }
 
   return S_OK;
 }
 
 EC_HOST_UI_MODE CommandExecuteImpl::GetLaunchMode() {
-  // We are to return chrome's mode if chrome exists else we query our embedder
-  // IServiceProvider and learn the mode from them.
+  // See the header file for an explanation of the mode selection logic.
   static bool launch_mode_determined = false;
   static EC_HOST_UI_MODE launch_mode = ECHUIM_DESKTOP;
 
   const char* modes[] = { "Desktop", "Inmmersive", "SysLauncher", "??" };
 
   if (launch_mode_determined)
     return launch_mode;
 
   if (chrome_mode_ != ECHUIM_SYSTEM_LAUNCHER) {
     launch_mode = chrome_mode_;
     AtlTrace("Launch mode is that of chrome, %s\n", modes[launch_mode]);
     launch_mode_determined = true;
     return launch_mode;
   }
 
   if (parameters_ == ASCIIToWide(switches::kForceImmersive)) {
     launch_mode = ECHUIM_IMMERSIVE;
     launch_mode_determined = true;
+    parameters_.clear();
   } else if (parameters_ == ASCIIToWide(switches::kForceDesktop)) {
     launch_mode = ECHUIM_DESKTOP;
     launch_mode_determined = true;
+    parameters_.clear();
+  }
+
+  base::win::RegKey reg_key;
+  if (reg_key.Create(HKEY_CURRENT_USER,
+                     chrome::kMetroRegistryPath,
+                     KEY_ALL_ACCESS) != 0) {

[grt (UTC plus 2), 2012/09/13 04:33:16]

please use ERROR_SUCCESS rather than 0 for consistency with the codebase

[cpu_(ooo_6.6-7.5), 2012/09/13 20:56:56]

Done.

+    AtlTrace("Failed to open HKCU %ls key\n", chrome::kMetroRegistryPath);

[grt (UTC plus 2), 2012/09/13 04:33:16]

might be helpful to log the result of reg_key.Create() here

[cpu_(ooo_6.6-7.5), 2012/09/13 20:56:56]

Done.

+    if (!launch_mode_determined) {
+      launch_mode = ECHUIM_DESKTOP;
+      launch_mode_determined = true;
+    }
+    return launch_mode;
   }
 
   if (launch_mode_determined) {
-    parameters_.clear();
-    AtlTrace("Launch mode forced to %s\n", modes[launch_mode]);
+    AtlTrace("Launch mode forced by cmdline to %s\n", modes[launch_mode]);
+    reg_key.WriteValue(chrome::kLaunchModeValue,

[grt (UTC plus 2), 2012/09/13 04:33:16]

please add code to chrome/installer/setup/uninstall.cc to delete this value, and
the key as well if it's empty.

+                       static_cast<DWORD>(launch_mode));
     return launch_mode;
   }
 
-  CComPtr<IExecuteCommandHost> host;
-  CComQIPtr<IServiceProvider> service_provider = m_spUnkSite;
-  if (service_provider) {
-    service_provider->QueryService(IID_IExecuteCommandHost, &host);
-    if (host) {
-      host->GetUIMode(&launch_mode);
-    } else {
-      AtlTrace("Failed to get IID_IExecuteCommandHost. Assuming desktop\n");
-    }
+  DWORD reg_value;
+  if (reg_key.ReadValueDW(chrome::kLaunchModeValue, &reg_value) != 0) {

[grt (UTC plus 2), 2012/09/13 04:33:16]

please use ERROR_SUCCESS rather than 0

[cpu_(ooo_6.6-7.5), 2012/09/13 20:56:56]

Done.

+    launch_mode = delegate_execute::IsMachineATablet() ?

[grt (UTC plus 2), 2012/09/13 04:33:16]

i'm curious: is this ever called when chrome isn't the default browser?

[cpu_(ooo_6.6-7.5), 2012/09/13 20:56:56]

I'' see if it does.

+                      ECHUIM_IMMERSIVE : ECHUIM_DESKTOP;
+    AtlTrace("Launch mode forced by heuristics to %s\n", modes[launch_mode]);
+  } else if (reg_value > ECHUIM_SYSTEM_LAUNCHER) {

[grt (UTC plus 2), 2012/09/13 04:33:16]

should this be >=?  is it valid for the launch mode to be set to SYSTEM_LAUNCHER
in the registry?

[cpu_(ooo_6.6-7.5), 2012/09/13 20:56:56]

I was thinking on using it experimentally but I'll experiment in my checkout.
Changed.

On 2012/09/13 04:33:16, grt wrote:

+    AtlTrace("Invalid registry launch mode value %u\n", reg_value);
+    launch_mode = ECHUIM_DESKTOP;
   } else {
-      AtlTrace("Failed to get IServiceProvider. Assuming desktop mode\n");
+    AtlTrace("Launch mode forced by registry to %s\n", modes[launch_mode]);
+    launch_mode = static_cast<EC_HOST_UI_MODE>(reg_value);
   }
-  AtlTrace("Launch mode is %s\n", modes[launch_mode]);
+
   launch_mode_determined = true;
   return launch_mode;
 }