Print Preview: change getPreview to cr.sendWithPromise

chrome/browser/ui/webui/print_preview/print_preview_handler.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/ui/webui/print_preview/print_preview_handler.h"
 
 #include <ctype.h>
 #include <stddef.h>
 
 #include <map>
@@ skipping 764 matching lines @@
   }
 
   EnsureExtensionPrinterHandlerSet();
   extension_printer_handler_->StartGetCapability(
       printer_name,
       base::Bind(&PrintPreviewHandler::OnGotExtensionPrinterCapabilities,
                  weak_factory_.GetWeakPtr(), callback_id));
 }
 
 void PrintPreviewHandler::HandleGetPreview(const base::ListValue* args) {
-  DCHECK_EQ(2U, args->GetSize());
+  DCHECK_EQ(3U, args->GetSize());
+  std::string callback_id;
   std::string json_str;
-  if (!args->GetString(0, &json_str))
+  CHECK(args->GetString(0, &callback_id));
+  if (!args->GetString(1, &json_str)) {
+    RejectJavascriptCallback(base::Value(callback_id),
+                             base::Value("INVALID_SETTINGS"));
     return;
+  }
   std::unique_ptr<base::DictionaryValue> settings =
       GetSettingsDictionary(json_str);
-  if (!settings)
+  if (!settings) {
+    RejectJavascriptCallback(base::Value(callback_id),
+                             base::Value("INVALID_SETTINGS"));
     return;
+  }
   int request_id = -1;
-  if (!settings->GetInteger(printing::kPreviewRequestID, &request_id))
+  if (!settings->GetInteger(printing::kPreviewRequestID, &request_id)) {
+    RejectJavascriptCallback(base::Value(callback_id),
+                             base::Value("INVALID_SETTINGS"));
     return;
-
+  }
+  DCHECK(preview_callbacks_.size() == static_cast<size_t>(request_id));
+  preview_callbacks_.push_back(callback_id);
   print_preview_ui()->OnPrintPreviewRequest(request_id);
   // Add an additional key in order to identify |print_preview_ui| later on
   // when calling PrintPreviewUI::GetCurrentPrintPreviewStatus() on the IO
   // thread.
   settings->SetInteger(printing::kPreviewUIID,
                        print_preview_ui()->GetIDForPrintPreviewUI());
 
   // Increment request count.
   ++regenerate_preview_request_count_;
 
@@ skipping 30 matching lines @@
     settings->SetString(printing::kSettingHeaderFooterURL, url);
   }
 
   bool generate_draft_data = false;
   success = settings->GetBoolean(printing::kSettingGenerateDraftData,
                                  &generate_draft_data);
   DCHECK(success);
 
   if (!generate_draft_data) {
     int page_count = -1;
-    success = args->GetInteger(1, &page_count);
+    success = args->GetInteger(2, &page_count);
     DCHECK(success);
 
     if (page_count != -1) {
       bool preview_modifiable = false;
       success = settings->GetBoolean(printing::kSettingPreviewModifiable,
                                      &preview_modifiable);
       DCHECK(success);
 
       if (preview_modifiable &&
           print_preview_ui()->GetAvailableDraftPageCount() != page_count) {
@@ skipping 660 matching lines @@
       ui::SelectFileDialog::SELECT_SAVEAS_FILE, base::string16(), path,
       &file_type_info, 0, base::FilePath::StringType(),
       platform_util::GetTopLevel(preview_web_contents()->GetNativeView()),
       NULL);
 }
 
 void PrintPreviewHandler::OnGotUniqueFileName(const base::FilePath& path) {
   FileSelected(path, 0, nullptr);
 }
 
-void PrintPreviewHandler::OnPrintPreviewFailed() {
-  if (reported_failed_preview_)
+void PrintPreviewHandler::OnPrintPreviewReady(int preview_uid, int request_id) {
+  ResolveJavascriptCallback(base::Value(preview_callbacks_[request_id]),

[Lei Zhang, 2017/06/29 19:25:46]

If |request_id| is negative, or too big, then this goes out of bounds.

In general, any data that comes from the initiator renderer needs to be
sanitized before we use it.

[rbpotter, 2017/06/29 20:44:10]

Done.

+                            base::Value(preview_uid));
+}
+
+void PrintPreviewHandler::OnPrintPreviewFailed(int request_id) {
+  if (!reported_failed_preview_) {
+    reported_failed_preview_ = true;
+    ReportUserActionHistogram(PREVIEW_FAILED);
+  }
+  if (request_id == -1)
     return;
-  reported_failed_preview_ = true;
-  ReportUserActionHistogram(PREVIEW_FAILED);
+  DCHECK(preview_callbacks_.size() > static_cast<size_t>(request_id));
+  RejectJavascriptCallback(base::Value(preview_callbacks_[request_id]),
+                           base::Value("PREVIEW_FAILED"));
+}
+
+void PrintPreviewHandler::OnInvalidPrinterSettings(int request_id) {
+  if (request_id == -1)
+    return;
+  DCHECK(preview_callbacks_.size() > static_cast<size_t>(request_id));
+  RejectJavascriptCallback(base::Value(preview_callbacks_[request_id]),
+                           base::Value("INVALID_SETTINGS"));
 }
 
 #if BUILDFLAG(ENABLE_BASIC_PRINT_DIALOG)
 void PrintPreviewHandler::ShowSystemDialog() {
   HandleShowSystemDialog(NULL);
 }
 #endif
 
 void PrintPreviewHandler::FileSelected(const base::FilePath& path,
                                        int /* index */,
@@ skipping 347 matching lines @@
 
 void PrintPreviewHandler::UnregisterForGaiaCookieChanges() {
   if (gaia_cookie_manager_service_)
     gaia_cookie_manager_service_->RemoveObserver(this);
 }
 
 void PrintPreviewHandler::SetPdfSavedClosureForTesting(
     const base::Closure& closure) {
   pdf_file_saved_closure_ = closure;
 }