Add token field to ClientSafeBrowsingReportReqeust

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"
 
 #include "base/compiler_specific.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_context.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/download_danger_type.h"
 #include "content/public/browser/download_item.h"
 #include "grit/components_strings.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/base/resource/resource_bundle.h"
 #include "ui/views/controls/button/label_button.h"
 #include "ui/views/controls/label.h"
 #include "ui/views/layout/grid_layout.h"
 #include "ui/views/view.h"
 #include "ui/views/widget/widget.h"
 #include "ui/views/window/dialog_client_view.h"
 #include "ui/views/window/dialog_delegate.h"
 #include "url/gurl.h"
 
 using extensions::ExperienceSamplingEvent;
+using safe_browsing::ClientSafeBrowsingReportRequest;
 
 namespace {
 
 const int kMessageWidth = 320;
 const int kParagraphPadding = 15;
 
 // Views-specific implementation of download danger prompt dialog. We use this
 // class rather than a TabModalConfirmDialog so that we can use custom
 // formatting on the text in the body of the dialog.
 class DownloadDangerPromptViews : public DownloadDangerPrompt,
@@ skipping 24 matching lines @@
 
  private:
   base::string16 GetAcceptButtonTitle() const;
   base::string16 GetCancelButtonTitle() const;
   // The message lead is separated from the main text and is bolded.
   base::string16 GetMessageLead() const;
   base::string16 GetMessageBody() const;
   void RunDone(Action action);
 
   content::DownloadItem* download_;
+  // If show_context_ is true, this is a download confirmation dialog by
+  // download API, otherwise it is download recovery dialog from a regular
+  // download.
   bool show_context_;
   OnDone done_;
 
   std::unique_ptr<ExperienceSamplingEvent> sampling_event_;
 
   views::View* contents_view_;
 };
 
 DownloadDangerPromptViews::DownloadDangerPromptViews(
     content::DownloadItem* item,
@@ skipping 239 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 this download is no longer dangerous, or is already canceled or
+    // If this download is no longer dangerous, is already canceled or
     // completed, don't send any report.
     if (download_->IsDangerous() && !download_->IsDone()) {
       const bool accept = action == DownloadDangerPrompt::ACCEPT;
       RecordDownloadDangerPrompt(accept, *download_);
       if (!download_->GetURL().is_empty() &&
           !download_->GetBrowserContext()->IsOffTheRecord()) {
-        SendSafeBrowsingDownloadRecoveryReport(accept, *download_);
+        ClientSafeBrowsingReportRequest::ReportType report_type
+            = show_context_ ?
+                ClientSafeBrowsingReportRequest::DANGEROUS_DOWNLOAD_BY_API :
+                ClientSafeBrowsingReportRequest::DANGEROUS_DOWNLOAD_RECOVERY;
+        SendSafeBrowsingDownloadReport(report_type, accept, *download_);
       }
     }
     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;
 }