OLD | NEW |
(Empty) | |
| 1 // Copyright 2017 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 #ifndef CHROME_BROWSER_SAFE_BROWSING_CHROME_CLEANER_MOCK_CHROME_CLEANER_PROCESS_
WIN_H_ |
| 6 #define CHROME_BROWSER_SAFE_BROWSING_CHROME_CLEANER_MOCK_CHROME_CLEANER_PROCESS_
WIN_H_ |
| 7 |
| 8 #include <set> |
| 9 #include <string> |
| 10 #include <vector> |
| 11 |
| 12 #include "base/callback.h" |
| 13 #include "base/command_line.h" |
| 14 #include "base/files/file_path.h" |
| 15 #include "components/chrome_cleaner/public/interfaces/chrome_prompt.mojom.h" |
| 16 |
| 17 namespace safe_browsing { |
| 18 |
| 19 // Class that mocks the behaviour of the Chrome Cleaner process. Intended to be |
| 20 // used in multi process tests. Example Usage: |
| 21 class MockChromeCleanerProcess { |
| 22 public: |
| 23 enum class CrashPoint { |
| 24 kNone, |
| 25 kOnStartup, |
| 26 kAfterConnection, |
| 27 kAfterRequestSent, |
| 28 kAfterResponseReceived, |
| 29 kNumCrashPoints, |
| 30 }; |
| 31 |
| 32 static constexpr int kInternalTestFailureExitCode = -1; |
| 33 static constexpr int kDeliberateCrashExitCode = -2; |
| 34 static constexpr int kNothingFoundExitCode = 2; |
| 35 static constexpr int kCanceledExitCode = 3; |
| 36 static constexpr int kRebootRequiredExitCode = 15; |
| 37 static constexpr int kRebootNotRequiredExitCode = 0; |
| 38 |
| 39 class Options { |
| 40 public: |
| 41 static bool FromCommandLine(const base::CommandLine& command_line, |
| 42 Options* options); |
| 43 |
| 44 Options(bool do_find_uws = false, |
| 45 bool reboot_required = false, |
| 46 CrashPoint crash_point = CrashPoint::kNone); |
| 47 Options(const Options& other); |
| 48 |
| 49 void AddSwitchesToCommandLine(base::CommandLine* command_line) const; |
| 50 |
| 51 void SetDoFindUws(bool do_find_uws); |
| 52 const std::vector<chrome_cleaner::mojom::UwSPtr>& found_uws() const { |
| 53 return found_uws_; |
| 54 } |
| 55 const std::set<base::FilePath>& files_to_delete() const { |
| 56 return files_to_delete_; |
| 57 } |
| 58 |
| 59 void set_reboot_required(bool reboot_required) { |
| 60 reboot_required_ = reboot_required; |
| 61 } |
| 62 bool reboot_required() const { return reboot_required_; } |
| 63 |
| 64 void set_crash_point(CrashPoint crash_point) { crash_point_ = crash_point; } |
| 65 CrashPoint crash_point() const { return crash_point_; } |
| 66 |
| 67 int ExpectedExitCode(chrome_cleaner::mojom::PromptAcceptance |
| 68 received_prompt_acceptance) const; |
| 69 |
| 70 private: |
| 71 // No assignment operator. |
| 72 Options& operator=(const Options&) = delete; |
| 73 |
| 74 std::vector<chrome_cleaner::mojom::UwSPtr> found_uws_; |
| 75 std::set<base::FilePath> files_to_delete_; |
| 76 bool reboot_required_; |
| 77 CrashPoint crash_point_; |
| 78 }; |
| 79 |
| 80 MockChromeCleanerProcess(const Options& options, |
| 81 const std::string& chrome_mojo_pipe_token); |
| 82 |
| 83 // Call this in the main function of the mock Chrome Cleaner process. Returns |
| 84 // the exit code that should be used when the process exits. |
| 85 // |
| 86 // If a crash point has been specified in the options passed to the |
| 87 // constructor, the process will exit with a kDeliberateCrashExitCode exit |
| 88 // code, and this function will not return. |
| 89 int Run(); |
| 90 |
| 91 private: |
| 92 // Function that recieves the Mojo response to the PromptUser message. |
| 93 void SendScanResults( |
| 94 chrome_cleaner::mojom::ChromePromptPtrInfo prompt_ptr_info, |
| 95 base::OnceClosure quit_closure); |
| 96 void PromptUserCallback( |
| 97 base::OnceClosure quit_closure, |
| 98 chrome_cleaner::mojom::PromptAcceptance prompt_acceptance); |
| 99 |
| 100 Options options_; |
| 101 std::string chrome_mojo_pipe_token_; |
| 102 // The PromptAcceptance received in PromptUserCallback(). |
| 103 chrome_cleaner::mojom::PromptAcceptance received_prompt_acceptance_ = |
| 104 chrome_cleaner::mojom::PromptAcceptance::UNSPECIFIED; |
| 105 chrome_cleaner::mojom::ChromePromptPtr* chrome_prompt_ptr_ = nullptr; |
| 106 }; |
| 107 |
| 108 } // namespace safe_browsing |
| 109 |
| 110 #endif // CHROME_BROWSER_SAFE_BROWSING_CHROME_CLEANER_MOCK_CHROME_CLEANER_PROCE
SS_WIN_H_ |
OLD | NEW |