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 3// found in the LICENSE file. 4 5#include <windows.h> 6#include <tchar.h> 7 8#include <string> 9 10#include "base/at_exit.h" 11#include "base/command_line.h" 12#include "base/files/file_path.h" 13#include "base/win/windows_version.h" 14#include "chrome/app/client_util.h" 15#include "chrome/browser/chrome_process_finder_win.h" 16#include "chrome/browser/policy/policy_path_parser.h" 17#include "chrome/common/chrome_constants.h" 18#include "chrome/common/chrome_paths_internal.h" 19#include "chrome/common/chrome_switches.h" 20#include "chrome_elf/chrome_elf_main.h" 21#include "components/startup_metric_utils/startup_metric_utils.h" 22#include "content/public/common/result_codes.h" 23#include "ui/gfx/win/dpi.h" 24 25namespace { 26 27void CheckSafeModeLaunch() { 28 unsigned short k1 = ::GetAsyncKeyState(VK_CONTROL); 29 unsigned short k2 = ::GetAsyncKeyState(VK_MENU); 30 const unsigned short kPressedMask = 0x8000; 31 if ((k1 & kPressedMask) && (k2 & kPressedMask)) 32 ::SetEnvironmentVariableA(chrome::kSafeModeEnvVar, "1"); 33} 34 35// List of switches that it's safe to rendezvous early with. Fast start should 36// not be done if a command line contains a switch not in this set. 37// Note this is currently stored as a list of two because it's probably faster 38// to iterate over this small array than building a map for constant time 39// lookups. 40const char* const kFastStartSwitches[] = { 41 switches::kProfileDirectory, 42 switches::kShowAppList, 43}; 44 45bool IsFastStartSwitch(const std::string& command_line_switch) { 46 for (size_t i = 0; i < arraysize(kFastStartSwitches); ++i) { 47 if (command_line_switch == kFastStartSwitches[i]) 48 return true; 49 } 50 return false; 51} 52 53bool ContainsNonFastStartFlag(const CommandLine& command_line) { 54 const CommandLine::SwitchMap& switches = command_line.GetSwitches(); 55 if (switches.size() > arraysize(kFastStartSwitches)) 56 return true; 57 for (CommandLine::SwitchMap::const_iterator it = switches.begin(); 58 it != switches.end(); ++it) { 59 if (!IsFastStartSwitch(it->first)) 60 return true; 61 } 62 return false; 63} 64 65bool AttemptFastNotify(const CommandLine& command_line) { 66 if (ContainsNonFastStartFlag(command_line)) 67 return false; 68 69 base::FilePath user_data_dir; 70 if (!chrome::GetDefaultUserDataDirectory(&user_data_dir)) 71 return false; 72 policy::path_parser::CheckUserDataDirPolicy(&user_data_dir); 73 74 HWND chrome = chrome::FindRunningChromeWindow(user_data_dir); 75 if (!chrome) 76 return false; 77 return chrome::AttemptToNotifyRunningChrome(chrome, true) == 78 chrome::NOTIFY_SUCCESS; 79} 80 81} // namespace 82 83#if !defined(ADDRESS_SANITIZER) 84int APIENTRY wWinMain(HINSTANCE instance, HINSTANCE prev, wchar_t*, int) { 85#else 86// The AddressSanitizer build should be a console program as it prints out stuff 87// on stderr. 88int main() { 89 HINSTANCE instance = GetModuleHandle(NULL); 90#endif 91 startup_metric_utils::RecordExeMainEntryTime(); 92 93 // Signal Chrome Elf that Chrome has begun to start. 94 SignalChromeElf(); 95 96 // Initialize the commandline singleton from the environment. 97 CommandLine::Init(0, NULL); 98 // The exit manager is in charge of calling the dtors of singletons. 99 base::AtExitManager exit_manager; 100 101 // We don't want to set DPI awareness on Vista because we don't support 102 // DirectWrite there. GDI fonts are kerned very badly, so better to leave 103 // DPI-unaware and at effective 1.0. See also ShouldUseDirectWrite(). 104 if (base::win::GetVersion() > base::win::VERSION_VISTA) 105 gfx::EnableHighDPISupport(); 106 107 if (AttemptFastNotify(*CommandLine::ForCurrentProcess())) 108 return 0; 109 110 CheckSafeModeLaunch(); 111 112 // Load and launch the chrome dll. *Everything* happens inside. 113 MainDllLoader* loader = MakeMainDllLoader(); 114 int rc = loader->Launch(instance); 115 loader->RelaunchChromeBrowserWithNewCommandLineIfNeeded(); 116 delete loader; 117 return rc; 118} 119