15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <windows.h> 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <tchar.h> 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include <string> 9868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/at_exit.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/command_line.h" 12868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/files/file_path.h" 13ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch#include "base/lazy_instance.h" 14ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch#include "chrome/app/chrome_breakpad_client.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/app/breakpad_win.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/app/client_util.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/app/metro_driver_win.h" 18868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "chrome/browser/chrome_process_finder_win.h" 19868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "chrome/browser/policy/policy_path_parser.h" 20868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "chrome/common/chrome_paths_internal.h" 21868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "chrome/common/chrome_switches.h" 22ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch#include "components/breakpad/breakpad_client.h" 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/app/startup_helper_win.h" 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/common/result_codes.h" 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "sandbox/win/src/sandbox_factory.h" 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 27868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)namespace { 28868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 29ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdochbase::LazyInstance<chrome::ChromeBreakpadClient>::Leaky 30ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch g_chrome_breakpad_client = LAZY_INSTANCE_INITIALIZER; 31ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int RunChrome(HINSTANCE instance) { 33ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch breakpad::SetBreakpadClient(g_chrome_breakpad_client.Pointer()); 34ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool exit_now = true; 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We restarted because of a previous crash. Ask user if we should relaunch. 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (ShowRestartDialogIfCrashed(&exit_now)) { 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (exit_now) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return content::RESULT_CODE_NORMAL_EXIT; 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Initialize the sandbox services. 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sandbox::SandboxInterfaceInfo sandbox_info = {0}; 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::InitializeSandboxInfo(&sandbox_info); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Load and launch the chrome dll. *Everything* happens inside. 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MainDllLoader* loader = MakeMainDllLoader(); 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int rc = loader->Launch(instance, &sandbox_info); 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) loader->RelaunchChromeBrowserWithNewCommandLineIfNeeded(); 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delete loader; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return rc; 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 54868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)// List of switches that it's safe to rendezvous early with. Fast start should 55868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)// not be done if a command line contains a switch not in this set. 56868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)// Note this is currently stored as a list of two because it's probably faster 57868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)// to iterate over this small array than building a map for constant time 58868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)// lookups. 59868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)const char* const kFastStartSwitches[] = { 60868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) switches::kProfileDirectory, 61868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) switches::kShowAppList, 62868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)}; 63868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 64868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)bool IsFastStartSwitch(const std::string& command_line_switch) { 65868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) for (size_t i = 0; i < arraysize(kFastStartSwitches); ++i) { 66868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (command_line_switch == kFastStartSwitches[i]) 67868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) return true; 68868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) } 69868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) return false; 70868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)} 71868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 72868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)bool ContainsNonFastStartFlag(const CommandLine& command_line) { 73868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) const CommandLine::SwitchMap& switches = command_line.GetSwitches(); 74868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (switches.size() > arraysize(kFastStartSwitches)) 75868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) return true; 76868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) for (CommandLine::SwitchMap::const_iterator it = switches.begin(); 77868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) it != switches.end(); ++it) { 78868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (!IsFastStartSwitch(it->first)) 79868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) return true; 80868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) } 81868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) return false; 82868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)} 83868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 84868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)bool AttemptFastNotify(const CommandLine& command_line) { 85868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (ContainsNonFastStartFlag(command_line)) 86868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) return false; 87868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 88868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) base::FilePath user_data_dir; 89868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (!chrome::GetDefaultUserDataDirectory(&user_data_dir)) 90868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) return false; 91868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) policy::path_parser::CheckUserDataDirPolicy(&user_data_dir); 92868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 93868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) HWND chrome = chrome::FindRunningChromeWindow(user_data_dir); 94eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch if (!chrome) 95eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch return false; 96eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch return chrome::AttemptToNotifyRunningChrome(chrome, true) == 97eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch chrome::NOTIFY_SUCCESS; 98868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)} 99868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 100868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)} // namespace 101868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int APIENTRY wWinMain(HINSTANCE instance, HINSTANCE prev, wchar_t*, int) { 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Initialize the commandline singleton from the environment. 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CommandLine::Init(0, NULL); 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The exit manager is in charge of calling the dtors of singletons. 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::AtExitManager exit_manager; 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 108868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) if (AttemptFastNotify(*CommandLine::ForCurrentProcess())) 109868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) return 0; 110868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MetroDriver metro_driver; 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (metro_driver.in_metro_mode()) 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return metro_driver.RunInMetro(instance, &RunChrome); 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Not in metro mode, proceed as normal. 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return RunChrome(instance); 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 117