OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 #ifndef CHROME_BROWSER_SAFE_BROWSING_SRT_FETCHER_WIN_H_ | 5 #ifndef CHROME_BROWSER_SAFE_BROWSING_SRT_FETCHER_WIN_H_ |
6 #define CHROME_BROWSER_SAFE_BROWSING_SRT_FETCHER_WIN_H_ | 6 #define CHROME_BROWSER_SAFE_BROWSING_SRT_FETCHER_WIN_H_ |
7 | 7 |
8 #include <limits.h> | 8 #include <limits.h> |
9 #include <stdint.h> | 9 #include <stdint.h> |
10 | 10 |
11 #include <queue> | 11 #include <queue> |
12 #include <string> | 12 #include <string> |
13 | 13 |
14 #include "base/command_line.h" | 14 #include "base/command_line.h" |
15 #include "base/feature_list.h" | 15 #include "base/feature_list.h" |
16 #include "base/process/launch.h" | 16 #include "base/process/launch.h" |
17 #include "base/process/process.h" | 17 #include "base/process/process.h" |
18 #include "base/time/time.h" | 18 #include "base/time/time.h" |
19 #include "components/chrome_cleaner/public/interfaces/chrome_prompt.mojom.h" | |
20 | 19 |
21 namespace base { | 20 namespace base { |
22 class FilePath; | 21 class FilePath; |
23 class TaskRunner; | 22 class TaskRunner; |
24 class Version; | 23 class Version; |
25 } | 24 } |
26 | 25 |
27 class Browser; | 26 class Browser; |
28 | 27 |
29 namespace safe_browsing { | 28 namespace safe_browsing { |
30 | 29 |
31 class ChromePromptImpl; | |
32 | |
33 // A special exit code identifying a failure to run the reporter. | 30 // A special exit code identifying a failure to run the reporter. |
34 const int kReporterNotLaunchedExitCode = INT_MAX; | 31 const int kReporterFailureExitCode = INT_MAX; |
35 | 32 |
36 // The number of days to wait before triggering another reporter run. | 33 // The number of days to wait before triggering another reporter run. |
37 const int kDaysBetweenSuccessfulSwReporterRuns = 7; | 34 const int kDaysBetweenSuccessfulSwReporterRuns = 7; |
38 const int kDaysBetweenSwReporterRunsForPendingPrompt = 1; | 35 const int kDaysBetweenSwReporterRunsForPendingPrompt = 1; |
39 // The number of days to wait before sending out reporter logs. | 36 // The number of days to wait before sending out reporter logs. |
40 const int kDaysBetweenReporterLogsSent = 7; | 37 const int kDaysBetweenReporterLogsSent = 7; |
41 | 38 |
42 // When enabled, moves all user interaction with the Software Reporter and the | 39 // When enabled, moves all user interaction with the Software Reporter and the |
43 // Chrome Cleanup tool to Chrome. | 40 // Chrome Cleanup tool to Chrome. |
44 extern const base::Feature kInBrowserCleanerUIFeature; | 41 extern const base::Feature kInBrowserCleanerUIFeature; |
(...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
99 // recent Reporter run terminated with an exit code indicating the presence of | 96 // recent Reporter run terminated with an exit code indicating the presence of |
100 // UwS. | 97 // UwS. |
101 bool ReporterFoundUws(); | 98 bool ReporterFoundUws(); |
102 | 99 |
103 // Returns true iff a valid registry key for the SRT Cleaner exists, and that | 100 // Returns true iff a valid registry key for the SRT Cleaner exists, and that |
104 // key is nonempty. | 101 // key is nonempty. |
105 // TODO(tmartino): Consider changing to check whether the user has recently | 102 // TODO(tmartino): Consider changing to check whether the user has recently |
106 // run the cleaner, rather than checking if they've run it at all. | 103 // run the cleaner, rather than checking if they've run it at all. |
107 bool UserHasRunCleaner(); | 104 bool UserHasRunCleaner(); |
108 | 105 |
109 // A delegate used by tests to implement test doubles (e.g., stubs, fakes, or | 106 // Mocks and callbacks for the unit tests. |
110 // mocks). | |
111 class SwReporterTestingDelegate { | 107 class SwReporterTestingDelegate { |
112 public: | 108 public: |
113 virtual ~SwReporterTestingDelegate() {} | 109 virtual ~SwReporterTestingDelegate() {} |
114 | 110 |
115 // Invoked by tests in places of base::LaunchProcess. | 111 // Test mock for launching the reporter. |
116 // See chrome_cleaner::mojom::ChromePromptRequest(). | |
117 virtual base::Process LaunchReporter( | 112 virtual base::Process LaunchReporter( |
118 const SwReporterInvocation& invocation, | 113 const SwReporterInvocation& invocation, |
119 const base::LaunchOptions& launch_options) = 0; | 114 const base::LaunchOptions& launch_options) = 0; |
120 | 115 |
121 // Invoked by tests in place of the actual prompting logic. | 116 // Test mock for showing the prompt. |
122 // See MaybeFetchSRT(). | |
123 virtual void TriggerPrompt(Browser* browser, | 117 virtual void TriggerPrompt(Browser* browser, |
124 const std::string& reporter_version) = 0; | 118 const std::string& reporter_version) = 0; |
125 | 119 |
126 // Invoked by tests to override the current time. | 120 // Returns the test's idea of the current time. |
127 // See Now() in srt_fetcher_win.cc. | |
128 virtual base::Time Now() const = 0; | 121 virtual base::Time Now() const = 0; |
129 | 122 |
130 // A task runner used to spawn the reporter process (which blocks). | 123 // A task runner used to spawn the reporter process (which blocks). |
131 // See ReporterRunner::ScheduleNextInvocation(). | |
132 virtual base::TaskRunner* BlockingTaskRunner() const = 0; | 124 virtual base::TaskRunner* BlockingTaskRunner() const = 0; |
133 | |
134 // Returns a ChromePromptImpl object that keeps track of specific | |
135 // actions during tests. Replaces the object returned by | |
136 // SwReporterProcess::CreateChromePromptImpl(). | |
137 // See SwReporterProcess::LaunchConnectedReporterProcess(). | |
138 virtual std::unique_ptr<ChromePromptImpl> CreateChromePromptImpl( | |
139 chrome_cleaner::mojom::ChromePromptRequest request) = 0; | |
140 | |
141 // Connection closed callback defined by tests in place of the default | |
142 // error handler. See SRTFetcherTest::CreateChromePromptImpl(). | |
143 virtual void OnConnectionClosed() = 0; | |
144 | |
145 // Bad message handler callback defined by tests in place of the default | |
146 // error handler. See SwReporterProcess::LaunchConnectedReporterProcess(). | |
147 virtual void OnConnectionError(const std::string& message) = 0; | |
148 }; | 125 }; |
149 | 126 |
150 // Set a delegate for testing. The implementation will not take ownership of | 127 // Set a delegate for testing. The implementation will not take ownership of |
151 // |delegate| - it must remain valid until this function is called again to | 128 // |delegate| - it must remain valid until this function is called again to |
152 // reset the delegate. If |delegate| is nullptr, any previous delegate is | 129 // reset the delegate. If |delegate| is nullptr, any previous delegate is |
153 // cleared. | 130 // cleared. |
154 void SetSwReporterTestingDelegate(SwReporterTestingDelegate* delegate); | 131 void SetSwReporterTestingDelegate(SwReporterTestingDelegate* delegate); |
155 | 132 |
156 void DisplaySRTPromptForTesting(const base::FilePath& download_path); | 133 void DisplaySRTPromptForTesting(const base::FilePath& download_path); |
157 | 134 |
158 } // namespace safe_browsing | 135 } // namespace safe_browsing |
159 | 136 |
160 #endif // CHROME_BROWSER_SAFE_BROWSING_SRT_FETCHER_WIN_H_ | 137 #endif // CHROME_BROWSER_SAFE_BROWSING_SRT_FETCHER_WIN_H_ |
OLD | NEW |