Spawn a new process to register and unregister DLLs.

chrome_frame/test_utils.h

-// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// 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.
 
 #ifndef CHROME_FRAME_TEST_UTILS_H_
 #define CHROME_FRAME_TEST_UTILS_H_
 
 #include <string>
 
 #include <atlbase.h>
 #include <atlcom.h>
 
+#include "base/string16.h"
+
 class FilePath;
 
 extern const wchar_t kChromeFrameDllName[];
 extern const wchar_t kChromeLauncherExeName[];
 
 // Helper class used to register different chrome frame DLLs while running
 // tests. The default constructor registers the DLL found in the build path.
-
+// Programs that use this class MUST include a call to the class's
+// RegisterAndExitProcessIfDirected method at the top of their main entrypoint.
+//
 // At destruction, again registers the DLL found in the build path if another
 // DLL has since been registered. Triggers GTEST asserts on failure.
 //
 // TODO(robertshield): Ideally, make this class restore the originally
 // registered chrome frame DLL (e.g. by looking in HKCR) on destruction.
 class ScopedChromeFrameRegistrar {
  public:
   enum RegistrationType {
     PER_USER,
     SYSTEM_LEVEL,
@@ skipping 10 matching lines @@
 
   std::wstring GetChromeFrameDllPath() const;
 
   static void RegisterAtPath(const std::wstring& path,
                              RegistrationType registration_type);
   static void UnregisterAtPath(const std::wstring& path,
                                RegistrationType registration_type);
   static void RegisterDefaults();
   static FilePath GetReferenceChromeFrameDllPath();
 
+  // Registers or unregisters a COM DLL and exits the process if the process's
+  // command line is:
+  // this.exe --call-registration-entrypoint path_to_dll entrypoint
+  // Otherwise simply returns. This method should be invoked at the start of the
+  // entrypoint in each executable that uses ScopedChromeFrameRegistrar to
+  // register or unregister DLLs.
+  static void RegisterAndExitProcessIfDirected();
+
  private:
+  enum RegistrationOperation {
+    REGISTER,
+    UNREGISTER,
+  };
+
+  // The string "--call-registration-entrypoint".
+  static const wchar_t kCallRegistrationEntrypointSwitch[];
+
+  static void DoRegistration(const string16& path,
+                             RegistrationType registration_type,
+                             RegistrationOperation registration_operation);
+
   // Contains the path of the most recently registered Chrome Frame DLL.
   std::wstring new_chrome_frame_dll_path_;
 
   // Contains the path of the Chrome Frame DLL to be registered at destruction.
   std::wstring original_dll_path_;
 
   // Indicates whether per user or per machine registration is needed.
   RegistrationType registration_type_;
   // We need to register the chrome path provider only once per process. This
   // flag keeps track of that.
@@ skipping 43 matching lines @@
 // Kills all running processes named |process_name| that have the string
 // |argument| on their command line. Useful for killing all Chrome Frame
 // instances of Chrome that all have --chrome-frame in their command line.
 bool KillAllNamedProcessesWithArgument(const std::wstring& process_name,
                                        const std::wstring& argument);
 
 // If the workstation is locked and cannot receive user input.
 bool IsWorkstationLocked();
 
 #endif  // CHROME_FRAME_TEST_UTILS_H_