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

chrome/browser/ui/views/download/download_danger_prompt_views.cc

 // Copyright 2013 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"

[msw, 2015/11/17 01:17:38]

nit: add a blank line above this.

[Jialiu Lin, 2015/11/17 02:05:50]

Done.

 
 #include "base/compiler_specific.h"
-#include "chrome/browser/download/download_danger_prompt.h"
 #include "chrome/browser/download/download_stats.h"
 #include "chrome/browser/extensions/api/experience_sampling_private/experience_sampling.h"
 #include "chrome/grit/chromium_strings.h"
 #include "chrome/grit/generated_resources.h"
 #include "components/constrained_window/constrained_window_views.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/download_danger_type.h"
 #include "content/public/browser/download_item.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/base/resource/resource_bundle.h"
@@ skipping 17 matching lines @@
 // formatting on the text in the body of the dialog.
 class DownloadDangerPromptViews : public DownloadDangerPrompt,
                                   public content::DownloadItem::Observer,
                                   public views::DialogDelegate {
  public:
   DownloadDangerPromptViews(content::DownloadItem* item,
                             bool show_context,
                             const OnDone& done);
 
   // DownloadDangerPrompt methods:
-  void InvokeActionForTesting(Action action) override;
+  std::string InvokeActionForTesting(Action action, const GURL& url) override;
 
   // views::DialogDelegate methods:
   base::string16 GetDialogButtonLabel(ui::DialogButton button) const override;
   base::string16 GetWindowTitle() const override;
   void DeleteDelegate() override;
   ui::ModalType GetModalType() const override;
   bool Cancel() override;
   bool Accept() override;
   bool Close() override;
   views::View* GetInitiallyFocusedView() override;
@@ skipping 70 matching lines @@
   // ExperienceSampling: A malicious download warning is being shown to the
   // user, so we start a new SamplingEvent and track it.
   sampling_event_.reset(new ExperienceSamplingEvent(
       ExperienceSamplingEvent::kDownloadDangerPrompt,
       item->GetURL(),
       item->GetReferrerUrl(),
       item->GetBrowserContext()));
 }
 
 // DownloadDangerPrompt methods:
-void DownloadDangerPromptViews::InvokeActionForTesting(Action action) {
+std::string DownloadDangerPromptViews::InvokeActionForTesting(Action action,
+                                                              const GURL& url) {
   switch (action) {
     case ACCEPT:
       Accept();
       break;
 
     case CANCEL:
     case DISMISS:
       Cancel();
       break;
 
     default:
       NOTREACHED();
       break;
   }
+
+  return CreateSafeBrowsingDownloadRecoveryReport(action == ACCEPT, url);
 }
 
 // views::DialogDelegate methods:
 base::string16 DownloadDangerPromptViews::GetDialogButtonLabel(
     ui::DialogButton button) const {
   switch (button) {
     case ui::DIALOG_BUTTON_OK:
       return GetAcceptButtonTitle();
 
     case ui::DIALOG_BUTTON_CANCEL:
@@ skipping 166 matching lines @@
   NOTREACHED();
   return base::string16();
 }
 
 void DownloadDangerPromptViews::RunDone(Action action) {
   // Invoking the callback can cause the download item state to change or cause
   // the 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
 
 DownloadDangerPrompt* DownloadDangerPrompt::Create(
     content::DownloadItem* item,
     content::WebContents* web_contents,
     bool show_context,
     const OnDone& done) {
   DownloadDangerPromptViews* download_danger_prompt =
       new DownloadDangerPromptViews(item, show_context, done);
   constrained_window::ShowWebModalDialogViews(download_danger_prompt,
                                               web_contents);
   return download_danger_prompt;
 }