| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #ifndef CHROME_FRAME_TEST_UTILS_H_ | |
| 6 #define CHROME_FRAME_TEST_UTILS_H_ | |
| 7 | |
| 8 #include <string> | |
| 9 | |
| 10 #include <atlbase.h> | |
| 11 #include <atlcom.h> | |
| 12 | |
| 13 #include "base/strings/string16.h" | |
| 14 | |
| 15 namespace base { | |
| 16 class FilePath; | |
| 17 } | |
| 18 | |
| 19 extern const wchar_t kChromeFrameDllName[]; | |
| 20 extern const wchar_t kChromeLauncherExeName[]; | |
| 21 | |
| 22 // Helper class used to register different chrome frame DLLs while running | |
| 23 // tests. The default constructor registers the DLL found in the build path. | |
| 24 // Programs that use this class MUST include a call to the class's | |
| 25 // RegisterAndExitProcessIfDirected method at the top of their main entrypoint. | |
| 26 // | |
| 27 // At destruction, again registers the DLL found in the build path if another | |
| 28 // DLL has since been registered. Triggers GTEST asserts on failure. | |
| 29 // | |
| 30 // TODO(robertshield): Ideally, make this class restore the originally | |
| 31 // registered chrome frame DLL (e.g. by looking in HKCR) on destruction. | |
| 32 class ScopedChromeFrameRegistrar { | |
| 33 public: | |
| 34 enum RegistrationType { | |
| 35 PER_USER, | |
| 36 SYSTEM_LEVEL, | |
| 37 }; | |
| 38 | |
| 39 explicit ScopedChromeFrameRegistrar(RegistrationType registration_type); | |
| 40 ScopedChromeFrameRegistrar(const std::wstring& path, | |
| 41 RegistrationType registration_type); | |
| 42 virtual ~ScopedChromeFrameRegistrar(); | |
| 43 | |
| 44 void RegisterChromeFrameAtPath(const std::wstring& path); | |
| 45 void UnegisterChromeFrameAtPath(const std::wstring& path); | |
| 46 void RegisterReferenceChromeFrameBuild(); | |
| 47 | |
| 48 std::wstring GetChromeFrameDllPath() const; | |
| 49 | |
| 50 static void RegisterAtPath(const std::wstring& path, | |
| 51 RegistrationType registration_type); | |
| 52 static void UnregisterAtPath(const std::wstring& path, | |
| 53 RegistrationType registration_type); | |
| 54 static void RegisterDefaults(); | |
| 55 static base::FilePath GetReferenceChromeFrameDllPath(); | |
| 56 | |
| 57 // Registers or unregisters a COM DLL and exits the process if the process's | |
| 58 // command line is: | |
| 59 // this.exe --call-registration-entrypoint path_to_dll entrypoint | |
| 60 // Otherwise simply returns. This method should be invoked at the start of the | |
| 61 // entrypoint in each executable that uses ScopedChromeFrameRegistrar to | |
| 62 // register or unregister DLLs. | |
| 63 static void RegisterAndExitProcessIfDirected(); | |
| 64 | |
| 65 private: | |
| 66 enum RegistrationOperation { | |
| 67 REGISTER, | |
| 68 UNREGISTER, | |
| 69 }; | |
| 70 | |
| 71 // The string "--call-registration-entrypoint". | |
| 72 static const wchar_t kCallRegistrationEntrypointSwitch[]; | |
| 73 | |
| 74 static void DoRegistration(const base::string16& path, | |
| 75 RegistrationType registration_type, | |
| 76 RegistrationOperation registration_operation); | |
| 77 | |
| 78 // Contains the path of the most recently registered Chrome Frame DLL. | |
| 79 std::wstring new_chrome_frame_dll_path_; | |
| 80 | |
| 81 // Contains the path of the Chrome Frame DLL to be registered at destruction. | |
| 82 std::wstring original_dll_path_; | |
| 83 | |
| 84 // Indicates whether per user or per machine registration is needed. | |
| 85 RegistrationType registration_type_; | |
| 86 // We need to register the chrome path provider only once per process. This | |
| 87 // flag keeps track of that. | |
| 88 static bool register_chrome_path_provider_; | |
| 89 }; | |
| 90 | |
| 91 // Returns the path to the Chrome Frame DLL in the build directory. Assumes | |
| 92 // that the test executable is running from the build folder or a similar | |
| 93 // folder structure. | |
| 94 base::FilePath GetChromeFrameBuildPath(); | |
| 95 | |
| 96 // Callback description for onload, onloaderror, onmessage | |
| 97 static _ATL_FUNC_INFO g_single_param = {CC_STDCALL, VT_EMPTY, 1, {VT_VARIANT}}; | |
| 98 // Simple class that forwards the callbacks. | |
| 99 template <typename T> | |
| 100 class DispCallback | |
| 101 : public IDispEventSimpleImpl<1, DispCallback<T>, &IID_IDispatch> { | |
| 102 public: | |
| 103 typedef HRESULT (T::*Method)(const VARIANT* param); | |
| 104 | |
| 105 DispCallback(T* owner, Method method) : owner_(owner), method_(method) { | |
| 106 } | |
| 107 | |
| 108 BEGIN_SINK_MAP(DispCallback) | |
| 109 SINK_ENTRY_INFO(1, IID_IDispatch, DISPID_VALUE, OnCallback, &g_single_param) | |
| 110 END_SINK_MAP() | |
| 111 | |
| 112 virtual ULONG STDMETHODCALLTYPE AddRef() { | |
| 113 return owner_->AddRef(); | |
| 114 } | |
| 115 virtual ULONG STDMETHODCALLTYPE Release() { | |
| 116 return owner_->Release(); | |
| 117 } | |
| 118 | |
| 119 STDMETHOD(OnCallback)(VARIANT param) { | |
| 120 return (owner_->*method_)(¶m); | |
| 121 } | |
| 122 | |
| 123 IDispatch* ToDispatch() { | |
| 124 return reinterpret_cast<IDispatch*>(this); | |
| 125 } | |
| 126 | |
| 127 T* owner_; | |
| 128 Method method_; | |
| 129 }; | |
| 130 | |
| 131 // If the workstation is locked and cannot receive user input. | |
| 132 bool IsWorkstationLocked(); | |
| 133 | |
| 134 #endif // CHROME_FRAME_TEST_UTILS_H_ | |
| OLD | NEW |