Adds error handling support for the SwReporter launcher.

chrome/browser/safe_browsing/srt_chrome_prompt_impl.cc

 // Copyright 2017 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/safe_browsing/srt_chrome_prompt_impl.h"
 
 #include "base/logging.h"
 
 namespace safe_browsing {
 
 using chrome_cleaner::mojom::ChromePrompt;
 using chrome_cleaner::mojom::ChromePromptRequest;
 using chrome_cleaner::mojom::ElevationStatus;
 using chrome_cleaner::mojom::PromptAcceptance;
 using chrome_cleaner::mojom::UwSPtr;
 
 ChromePromptImpl::ChromePromptImpl(ChromePromptRequest request)
-    : binding_(this, std::move(request)) {}
+    : binding_(this, std::move(request)) {
+  binding_.set_connection_error_handler(base::Bind(
+      &ChromePromptImpl::OnConnectionClosed, base::Unretained(this)));

[grt (UTC plus 2), 2017/04/21 11:10:58]

please document why Unretained is safe

[ftirelo, 2017/04/24 15:47:45]

Obsolete.

+}
 
 ChromePromptImpl::~ChromePromptImpl() {}
 
 void ChromePromptImpl::PromptUser(
     std::vector<UwSPtr> removable_uws_found,
     ElevationStatus elevation_status,
     const ChromePrompt::PromptUserCallback& callback) {
+  if (test_delegate_)
+    test_delegate_->OnPromptUser();
+
   // Placeholder. The actual implementation will show the prompt dialog to the
   // user and invoke this callback depending on the user's response.
   callback.Run(PromptAcceptance::DENIED);
 }
 
+void ChromePromptImpl::OnConnectionClosed() {
+  // Placeholder. This should handle cases when the reporter process is
+  // disconnected (e.g. due to a crash) and the prompt dialog is being shown
+  // to the user.
+}
+
+void ChromePromptImpl::OnConnectionError(const std::string& message) {
+  if (test_delegate_)
+    test_delegate_->OnConnectionError();
+
+  // Placeholder. This should handle cases when the reporter process sends
+  // a bad message and the prompt dialog is being shown to the user.
+}
+
+// static
+void ChromePromptImpl::SetTestDelegate(
+    ChromePromptImpl::TestDelegate* test_delegate) {
+  test_delegate_ = test_delegate;
+}
+
+// static
+ChromePromptImpl::TestDelegate* ChromePromptImpl::test_delegate_ = nullptr;
+
 }  // namespace safe_browsing