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 <tchar.h> | 6 #include <tchar.h> |
7 | 7 |
8 #include <algorithm> | |
grt (UTC plus 2)
2013/05/14 13:49:32
nit: unused
koz (OOO until 15th September)
2013/05/15 03:31:36
Done.
| |
9 #include <string> | |
10 | |
8 #include "base/at_exit.h" | 11 #include "base/at_exit.h" |
9 #include "base/command_line.h" | 12 #include "base/command_line.h" |
13 #include "base/files/file_path.h" | |
10 #include "chrome/app/breakpad_win.h" | 14 #include "chrome/app/breakpad_win.h" |
11 #include "chrome/app/client_util.h" | 15 #include "chrome/app/client_util.h" |
12 #include "chrome/app/metro_driver_win.h" | 16 #include "chrome/app/metro_driver_win.h" |
17 #include "chrome/browser/chrome_process_finder_win.h" | |
18 #include "chrome/browser/policy/policy_path_parser.h" | |
19 #include "chrome/common/chrome_paths_internal.h" | |
20 #include "chrome/common/chrome_switches.h" | |
13 #include "content/public/app/startup_helper_win.h" | 21 #include "content/public/app/startup_helper_win.h" |
14 #include "content/public/common/result_codes.h" | 22 #include "content/public/common/result_codes.h" |
15 #include "sandbox/win/src/sandbox_factory.h" | 23 #include "sandbox/win/src/sandbox_factory.h" |
16 | 24 |
25 namespace { | |
26 | |
17 int RunChrome(HINSTANCE instance) { | 27 int RunChrome(HINSTANCE instance) { |
18 bool exit_now = true; | 28 bool exit_now = true; |
19 // We restarted because of a previous crash. Ask user if we should relaunch. | 29 // We restarted because of a previous crash. Ask user if we should relaunch. |
20 if (ShowRestartDialogIfCrashed(&exit_now)) { | 30 if (ShowRestartDialogIfCrashed(&exit_now)) { |
21 if (exit_now) | 31 if (exit_now) |
22 return content::RESULT_CODE_NORMAL_EXIT; | 32 return content::RESULT_CODE_NORMAL_EXIT; |
23 } | 33 } |
24 | 34 |
25 // Initialize the sandbox services. | 35 // Initialize the sandbox services. |
26 sandbox::SandboxInterfaceInfo sandbox_info = {0}; | 36 sandbox::SandboxInterfaceInfo sandbox_info = {0}; |
27 content::InitializeSandboxInfo(&sandbox_info); | 37 content::InitializeSandboxInfo(&sandbox_info); |
28 | 38 |
29 // Load and launch the chrome dll. *Everything* happens inside. | 39 // Load and launch the chrome dll. *Everything* happens inside. |
30 MainDllLoader* loader = MakeMainDllLoader(); | 40 MainDllLoader* loader = MakeMainDllLoader(); |
31 int rc = loader->Launch(instance, &sandbox_info); | 41 int rc = loader->Launch(instance, &sandbox_info); |
32 loader->RelaunchChromeBrowserWithNewCommandLineIfNeeded(); | 42 loader->RelaunchChromeBrowserWithNewCommandLineIfNeeded(); |
33 delete loader; | 43 delete loader; |
34 return rc; | 44 return rc; |
35 } | 45 } |
36 | 46 |
47 // List of switches that it's safe to rendezvous early with. Fast start should | |
48 // not be done if a command line contains a switch not in this set. | |
49 // Note this is currently stored as a list of two because it's probably faster | |
50 // to iterate over this small array than building a map for constant time | |
51 // lookups. | |
52 static const char* const kFastStartSwitches[] = { | |
grt (UTC plus 2)
2013/05/14 13:49:32
nit: no "static" at namespace scope in the unnamed
koz (OOO until 15th September)
2013/05/15 03:31:36
Done.
| |
53 switches::kProfileDirectory, | |
54 switches::kShowAppList, | |
55 }; | |
56 | |
57 bool IsFastStartSwitch(const std::string& command_line_switch) { | |
58 for (size_t i = 0; i < arraysize(kFastStartSwitches); ++i) { | |
59 if (command_line_switch == kFastStartSwitches[i]) | |
60 return true; | |
61 } | |
62 return false; | |
63 } | |
64 | |
65 bool ContainsNonFastStartFlag(const CommandLine& command_line) { | |
66 const CommandLine::SwitchMap& switches = command_line.GetSwitches(); | |
67 if (switches.size() > arraysize(kFastStartSwitches)) | |
68 return true; | |
69 for (CommandLine::SwitchMap::const_iterator it = switches.begin(); | |
70 it != switches.end(); ++it) { | |
71 if (!IsFastStartSwitch(it->first)) | |
72 return true; | |
73 } | |
74 return false; | |
75 } | |
76 | |
77 bool AttemptFastNotify(CommandLine* command_line) { | |
78 if (ContainsNonFastStartFlag(*command_line)) | |
79 return false; | |
80 | |
81 base::FilePath user_data_dir; | |
82 if (!chrome::GetDefaultUserDataDirectory(&user_data_dir)) | |
83 return false; | |
84 policy::path_parser::CheckUserDataDirPolicy(&user_data_dir); | |
85 | |
86 HWND chrome = chrome::FindRunningChromeWindow(user_data_dir); | |
87 return chrome && chrome::AttemptToNotifyRunningChrome(chrome); | |
88 } | |
89 | |
90 } // namespace | |
91 | |
37 int APIENTRY wWinMain(HINSTANCE instance, HINSTANCE prev, wchar_t*, int) { | 92 int APIENTRY wWinMain(HINSTANCE instance, HINSTANCE prev, wchar_t*, int) { |
38 // Initialize the commandline singleton from the environment. | 93 // Initialize the commandline singleton from the environment. |
39 CommandLine::Init(0, NULL); | 94 CommandLine::Init(0, NULL); |
40 // The exit manager is in charge of calling the dtors of singletons. | 95 // The exit manager is in charge of calling the dtors of singletons. |
41 base::AtExitManager exit_manager; | 96 base::AtExitManager exit_manager; |
42 | 97 |
98 if (AttemptFastNotify(CommandLine::ForCurrentProcess())) | |
99 return 0; | |
100 | |
43 MetroDriver metro_driver; | 101 MetroDriver metro_driver; |
44 if (metro_driver.in_metro_mode()) | 102 if (metro_driver.in_metro_mode()) |
45 return metro_driver.RunInMetro(instance, &RunChrome); | 103 return metro_driver.RunInMetro(instance, &RunChrome); |
46 // Not in metro mode, proceed as normal. | 104 // Not in metro mode, proceed as normal. |
47 return RunChrome(instance); | 105 return RunChrome(instance); |
48 } | 106 } |
OLD | NEW |