Send safe browsing ThreatDetails to track download CTR when user tries to recover blocked downloads

chrome/browser/download/download_danger_prompt.cc

 // Copyright (c) 2012 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/download/download_danger_prompt.h"
 
 #include "base/bind.h"
 #include "base/strings/utf_string_conversions.h"
+#include "chrome/browser/browser_process.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/download/chrome_download_manager_delegate.h"
 #include "chrome/browser/download/download_stats.h"
 #include "chrome/browser/extensions/api/experience_sampling_private/experience_sampling.h"
+#include "chrome/browser/safe_browsing/ping_manager.h"
+#include "chrome/browser/safe_browsing/safe_browsing_service.h"
 #include "chrome/browser/ui/tab_modal_confirm_dialog.h"
 #include "chrome/browser/ui/tab_modal_confirm_dialog_delegate.h"
+#include "chrome/common/safe_browsing/csd.pb.h"
 #include "chrome/grit/chromium_strings.h"
 #include "chrome/grit/generated_resources.h"
 #include "content/public/browser/download_danger_type.h"
 #include "content/public/browser/download_item.h"
 #include "ui/base/l10n/l10n_util.h"
 
 using extensions::ExperienceSamplingEvent;
+using safe_browsing::ClientSafeBrowsingReportRequest;
 
 namespace {
 
 // TODO(wittman): Create a native web contents modal dialog implementation of
 // this dialog for non-Views platforms, to support bold formatting of the
 // message lead.
 
 // Implements DownloadDangerPrompt using a TabModalConfirmDialog.
 class DownloadDangerPromptImpl : public DownloadDangerPrompt,
                                  public content::DownloadItem::Observer,
                                  public TabModalConfirmDialogDelegate {
  public:
   DownloadDangerPromptImpl(content::DownloadItem* item,
                            content::WebContents* web_contents,
                            bool show_context,
                            const OnDone& done);
   ~DownloadDangerPromptImpl() override;
 
   // DownloadDangerPrompt:
-  void InvokeActionForTesting(Action action) override;
+  std::string InvokeActionForTesting(Action action, const GURL& url) override;
 
  private:
   // content::DownloadItem::Observer:
   void OnDownloadUpdated(content::DownloadItem* download) override;
 
   // TabModalConfirmDialogDelegate:
   base::string16 GetTitle() override;
   base::string16 GetDialogMessage() override;
   base::string16 GetAcceptButtonTitle() override;
   base::string16 GetCancelButtonTitle() override;
@@ skipping 33 matching lines @@
       download->GetReferrerUrl(),
       download->GetBrowserContext()));
 }
 
 DownloadDangerPromptImpl::~DownloadDangerPromptImpl() {
   // |this| might be deleted without invoking any callbacks. E.g. pressing Esc
   // on GTK or if the user navigates away from the page showing the prompt.
   RunDone(DISMISS);
 }
 
-void DownloadDangerPromptImpl::InvokeActionForTesting(Action action) {
+std::string DownloadDangerPromptImpl::InvokeActionForTesting(
+    Action action, const GURL& url) {
   switch (action) {
     case ACCEPT: Accept(); break;
     case CANCEL: Cancel(); break;
     case DISMISS:
       RunDone(DISMISS);
       Cancel();
       break;
   }
+  return CreateSafeBrowsingDownloadRecoveryReport(action == ACCEPT, url);
 }
 
 void DownloadDangerPromptImpl::OnDownloadUpdated(
     content::DownloadItem* download) {
   // If the download is nolonger dangerous (accepted externally) or the download
   // is in a terminal state, then the download danger prompt is no longer
   // necessary.
   if (!download->IsDangerous() || download->IsDone()) {
     RunDone(DISMISS);
     Cancel();
@@ skipping 119 matching lines @@
   RunDone(DISMISS);
 }
 
 void DownloadDangerPromptImpl::RunDone(Action action) {
   // Invoking the callback can cause the download item state to change or cause
   // the constrained window to close, and |callback| refers to a member
   // variable.
   OnDone done = done_;
   done_.Reset();
   if (download_ != NULL) {
+    if (!download_->GetURL().is_empty()) {
+      std::string report = CreateSafeBrowsingDownloadRecoveryReport(
+          action == DownloadDangerPrompt::ACCEPT, download_->GetURL());
+      if (!report.empty())
+        SendSerializedReport(report);
+    }
     download_->RemoveObserver(this);
     download_ = NULL;
   }
   if (!done.is_null())
     done.Run(action);
 }
 
 }  // namespace
 
 #if !defined(USE_AURA)
 // static
 DownloadDangerPrompt* DownloadDangerPrompt::Create(
     content::DownloadItem* item,
     content::WebContents* web_contents,
     bool show_context,
     const OnDone& done) {
   DownloadDangerPromptImpl* prompt = new DownloadDangerPromptImpl(
       item, web_contents, show_context, done);
   // |prompt| will be deleted when the dialog is done.
   TabModalConfirmDialog::Create(prompt, web_contents);
   return prompt;
 }
 #endif
+
+std::string DownloadDangerPrompt::CreateSafeBrowsingDownloadRecoveryReport(
+    bool did_proceed,
+    const GURL& url) {
+  ClientSafeBrowsingReportRequest report;
+  report.set_type(ClientSafeBrowsingReportRequest::MALICIOUS_DOWNLOAD_RECOVERY);
+  report.set_url(url.spec());
+  report.set_did_proceed(did_proceed);
+
+  std::string serialized_report;
+  if (!report.SerializeToString(&serialized_report)) {
+    DLOG(ERROR) << "Unable to serialize the threat report.";
+    return "";
+  }
+
+  return serialized_report;
+}
+
+void DownloadDangerPrompt::SendSerializedReport(
+    const std::string& serialized_report) {
+  SafeBrowsingService* sb_service = g_browser_process->safe_browsing_service();
+  if (sb_service) {
+    sb_service->SendDownloadRecoveryReport(serialized_report);
+  }
+}