Add a self-destruct mechanism to user-level Chrome Frame when it detects that system-level Chrome...

chrome_frame/chrome_tab.cc

 // Copyright (c) 2011 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.
 
 // chrome_tab.cc : Implementation of DLL Exports.
 
 // Include without path to make GYP build see it.
 #include "chrome_tab.h"  // NOLINT
 
 #include <atlsecurity.h>
@@ skipping 52 matching lines @@
 
 const wchar_t kProtocolHandlers[] =
     L"Software\\Classes\\Protocols\\Handler";
 
 const wchar_t kRunOnce[] =
     L"Software\\Microsoft\\Windows\\CurrentVersion\\RunOnce";
 
 const wchar_t kRunKeyName[] = L"ChromeFrameHelper";
 
 const wchar_t kChromeFrameHelperExe[] = L"chrome_frame_helper.exe";
+const wchar_t kChromeFrameHelperStartupArg[] = L"--startup";
 
 // Window class and window names.
 // TODO(robertshield): These and other constants need to be refactored into
 // a common chrome_frame_constants.h|cc and built into a separate lib
 // (either chrome_frame_utils or make another one).
 const wchar_t kChromeFrameHelperWindowClassName[] =
     L"ChromeFrameHelperWindowClass";
 const wchar_t kChromeFrameHelperWindowName[] =
     L"ChromeFrameHelperWindowName";
 
@@ skipping 224 matching lines @@
   HRESULT result = E_FAIL;
 
   std::wstring channel_name;
   if (base::win::GetVersion() < base::win::VERSION_VISTA &&
       GoogleUpdateSettings::GetChromeChannelAndModifiers(true, &channel_name)) {
     std::transform(channel_name.begin(), channel_name.end(),
                    channel_name.begin(), tolower);
     // Use this only for the dev channel and CEEE channels.
     if (channel_name.find(L"dev") != std::wstring::npos ||
         channel_name.find(L"ceee") != std::wstring::npos) {
-
       HKEY hive = HKEY_CURRENT_USER;
       if (IsSystemProcess()) {
         // For system installs, our updates will be running as SYSTEM which
         // makes writing to a RunOnce key under HKCU not so terribly useful.
         hive = HKEY_LOCAL_MACHINE;
       }
 
       RegKey run_once;
       LONG ret = run_once.Create(hive, kRunOnce, KEY_READ | KEY_WRITE);
       if (ret == ERROR_SUCCESS) {
@@ skipping 40 matching lines @@
     }
   } else {
     NOTREACHED();
   }
 
   // Find window handle of existing instance.
   HWND old_window = FindWindow(kChromeFrameHelperWindowClassName,
                                kChromeFrameHelperWindowName);
 
   if (file_util::PathExists(helper_path)) {
+    std::wstring helper_path_cmd(L"\"");
+    helper_path_cmd += helper_path.value();
+    helper_path_cmd += L"\" ";
+    helper_path_cmd += kChromeFrameHelperStartupArg;
+
     // Add new run-at-startup entry.
     base::win::AddCommandToAutoRun(HKEY_CURRENT_USER, kRunKeyName,
-                                  helper_path.value());
+                                   helper_path_cmd);
 
     // Start new instance.
     bool launched = base::LaunchApp(helper_path.value(), false, true, NULL);
     if (!launched) {
       NOTREACHED();
       LOG(ERROR) << "Could not launch helper process.";
     }
 
     // Kill old instance using window handle.
     if (IsWindow(old_window)) {
@@ skipping 449 matching lines @@
     sd.GetDacl(&new_dacl);
     new_dacl.AddAllowedAce(token_.GetUser(), GENERIC_WRITE | GENERIC_READ);
     if (AtlSetDacl(object_name.c_str(), SE_REGISTRY_KEY, new_dacl)) {
       result = SetOrDeleteMimeHandlerKey(enable, HKEY_LOCAL_MACHINE);
     }
   }
 
   backup.RestoreSecurity(object_name.c_str());
   return result;
 }