| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <windows.h> | 5 #include <windows.h> |
| 6 #include <shlwapi.h> | 6 #include <shlwapi.h> |
| 7 | 7 |
| 8 #include "base/command_line.h" | 8 #include "base/command_line.h" |
| 9 #include "base/debug/trace_event.h" | 9 #include "base/debug/trace_event.h" |
| 10 #include "base/environment.h" | 10 #include "base/environment.h" |
| 11 #include "base/file_util.h" | 11 #include "base/file_util.h" |
| 12 #include "base/logging.h" | 12 #include "base/logging.h" |
| 13 #include "base/memory/scoped_ptr.h" | 13 #include "base/memory/scoped_ptr.h" |
| 14 #include "base/win/registry.h" | 14 #include "base/win/registry.h" |
| 15 #include "base/string_util.h" | 15 #include "base/string_util.h" |
| 16 #include "base/utf_string_conversions.h" | 16 #include "base/utf_string_conversions.h" |
| 17 #include "base/version.h" | 17 #include "base/version.h" |
| 18 #include "chrome/app/breakpad_win.h" | 18 #include "chrome/app/breakpad_win.h" |
| 19 #include "chrome/app/client_util.h" | 19 #include "chrome/app/client_util.h" |
| 20 #include "chrome/common/chrome_constants.h" | 20 #include "chrome/common/chrome_constants.h" |
| 21 #include "chrome/common/chrome_result_codes.h" |
| 21 #include "chrome/common/chrome_switches.h" | 22 #include "chrome/common/chrome_switches.h" |
| 22 #include "chrome/installer/util/browser_distribution.h" | 23 #include "chrome/installer/util/browser_distribution.h" |
| 23 #include "chrome/installer/util/channel_info.h" | 24 #include "chrome/installer/util/channel_info.h" |
| 24 #include "chrome/installer/util/install_util.h" | 25 #include "chrome/installer/util/install_util.h" |
| 25 #include "chrome/installer/util/google_update_constants.h" | 26 #include "chrome/installer/util/google_update_constants.h" |
| 26 #include "chrome/installer/util/google_update_settings.h" | 27 #include "chrome/installer/util/google_update_settings.h" |
| 27 #include "chrome/installer/util/util_constants.h" | 28 #include "chrome/installer/util/util_constants.h" |
| 28 #include "content/common/result_codes.h" | |
| 29 | 29 |
| 30 namespace { | 30 namespace { |
| 31 // The entry point signature of chrome.dll. | 31 // The entry point signature of chrome.dll. |
| 32 typedef int (*DLL_MAIN)(HINSTANCE, sandbox::SandboxInterfaceInfo*, wchar_t*); | 32 typedef int (*DLL_MAIN)(HINSTANCE, sandbox::SandboxInterfaceInfo*, wchar_t*); |
| 33 | 33 |
| 34 typedef void (*RelaunchChromeBrowserWithNewCommandLineIfNeededFunc)(); | 34 typedef void (*RelaunchChromeBrowserWithNewCommandLineIfNeededFunc)(); |
| 35 | 35 |
| 36 // Gets chrome version according to the load path. |exe_path| must be the | 36 // Gets chrome version according to the load path. |exe_path| must be the |
| 37 // backslash terminated directory of the current chrome.exe. | 37 // backslash terminated directory of the current chrome.exe. |
| 38 bool GetVersion(const wchar_t* exe_path, const wchar_t* key_path, | 38 bool GetVersion(const wchar_t* exe_path, const wchar_t* key_path, |
| (...skipping 192 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 231 | 231 |
| 232 // Launching is a matter of loading the right dll, setting the CHROME_VERSION | 232 // Launching is a matter of loading the right dll, setting the CHROME_VERSION |
| 233 // environment variable and just calling the entry point. Derived classes can | 233 // environment variable and just calling the entry point. Derived classes can |
| 234 // add custom code in the OnBeforeLaunch callback. | 234 // add custom code in the OnBeforeLaunch callback. |
| 235 int MainDllLoader::Launch(HINSTANCE instance, | 235 int MainDllLoader::Launch(HINSTANCE instance, |
| 236 sandbox::SandboxInterfaceInfo* sbox_info) { | 236 sandbox::SandboxInterfaceInfo* sbox_info) { |
| 237 std::wstring version; | 237 std::wstring version; |
| 238 std::wstring file; | 238 std::wstring file; |
| 239 dll_ = Load(&version, &file); | 239 dll_ = Load(&version, &file); |
| 240 if (!dll_) | 240 if (!dll_) |
| 241 return ResultCodes::MISSING_DATA; | 241 return chrome::RESULT_CODE_MISSING_DATA; |
| 242 | 242 |
| 243 scoped_ptr<base::Environment> env(base::Environment::Create()); | 243 scoped_ptr<base::Environment> env(base::Environment::Create()); |
| 244 env->SetVar(chrome::kChromeVersionEnvVar, WideToUTF8(version)); | 244 env->SetVar(chrome::kChromeVersionEnvVar, WideToUTF8(version)); |
| 245 | 245 |
| 246 InitCrashReporterWithDllPath(file); | 246 InitCrashReporterWithDllPath(file); |
| 247 OnBeforeLaunch(file); | 247 OnBeforeLaunch(file); |
| 248 | 248 |
| 249 DLL_MAIN entry_point = | 249 DLL_MAIN entry_point = |
| 250 reinterpret_cast<DLL_MAIN>(::GetProcAddress(dll_, "ChromeMain")); | 250 reinterpret_cast<DLL_MAIN>(::GetProcAddress(dll_, "ChromeMain")); |
| 251 if (!entry_point) | 251 if (!entry_point) |
| 252 return ResultCodes::BAD_PROCESS_TYPE; | 252 return chrome::RESULT_CODE_BAD_PROCESS_TYPE; |
| 253 | 253 |
| 254 int rc = entry_point(instance, sbox_info, ::GetCommandLineW()); | 254 int rc = entry_point(instance, sbox_info, ::GetCommandLineW()); |
| 255 return OnBeforeExit(rc, file); | 255 return OnBeforeExit(rc, file); |
| 256 } | 256 } |
| 257 | 257 |
| 258 void MainDllLoader::RelaunchChromeBrowserWithNewCommandLineIfNeeded() { | 258 void MainDllLoader::RelaunchChromeBrowserWithNewCommandLineIfNeeded() { |
| 259 RelaunchChromeBrowserWithNewCommandLineIfNeededFunc relaunch_function = | 259 RelaunchChromeBrowserWithNewCommandLineIfNeededFunc relaunch_function = |
| 260 reinterpret_cast<RelaunchChromeBrowserWithNewCommandLineIfNeededFunc>( | 260 reinterpret_cast<RelaunchChromeBrowserWithNewCommandLineIfNeededFunc>( |
| 261 ::GetProcAddress(dll_, | 261 ::GetProcAddress(dll_, |
| 262 "RelaunchChromeBrowserWithNewCommandLineIfNeeded")); | 262 "RelaunchChromeBrowserWithNewCommandLineIfNeeded")); |
| (...skipping 17 matching lines...) Expand all Loading... |
| 280 } | 280 } |
| 281 | 281 |
| 282 virtual void OnBeforeLaunch(const std::wstring& dll_path) { | 282 virtual void OnBeforeLaunch(const std::wstring& dll_path) { |
| 283 RecordDidRun(dll_path); | 283 RecordDidRun(dll_path); |
| 284 } | 284 } |
| 285 | 285 |
| 286 virtual int OnBeforeExit(int return_code, const std::wstring& dll_path) { | 286 virtual int OnBeforeExit(int return_code, const std::wstring& dll_path) { |
| 287 // NORMAL_EXIT_CANCEL is used for experiments when the user cancels | 287 // NORMAL_EXIT_CANCEL is used for experiments when the user cancels |
| 288 // so we need to reset the did_run signal so omaha does not count | 288 // so we need to reset the did_run signal so omaha does not count |
| 289 // this run as active usage. | 289 // this run as active usage. |
| 290 if (ResultCodes::NORMAL_EXIT_CANCEL == return_code) { | 290 if (chrome::RESULT_CODE_NORMAL_EXIT_CANCEL == return_code) { |
| 291 ClearDidRun(dll_path); | 291 ClearDidRun(dll_path); |
| 292 } | 292 } |
| 293 return return_code; | 293 return return_code; |
| 294 } | 294 } |
| 295 }; | 295 }; |
| 296 | 296 |
| 297 //============================================================================= | 297 //============================================================================= |
| 298 | 298 |
| 299 class ChromiumDllLoader : public MainDllLoader { | 299 class ChromiumDllLoader : public MainDllLoader { |
| 300 public: | 300 public: |
| 301 virtual std::wstring GetRegistryPath() { | 301 virtual std::wstring GetRegistryPath() { |
| 302 BrowserDistribution* dist = BrowserDistribution::GetDistribution(); | 302 BrowserDistribution* dist = BrowserDistribution::GetDistribution(); |
| 303 return dist->GetVersionKey(); | 303 return dist->GetVersionKey(); |
| 304 } | 304 } |
| 305 }; | 305 }; |
| 306 | 306 |
| 307 MainDllLoader* MakeMainDllLoader() { | 307 MainDllLoader* MakeMainDllLoader() { |
| 308 #if defined(GOOGLE_CHROME_BUILD) | 308 #if defined(GOOGLE_CHROME_BUILD) |
| 309 return new ChromeDllLoader(); | 309 return new ChromeDllLoader(); |
| 310 #else | 310 #else |
| 311 return new ChromiumDllLoader(); | 311 return new ChromiumDllLoader(); |
| 312 #endif | 312 #endif |
| 313 } | 313 } |
| OLD | NEW |