Initial support for cloudprint in print preview

chrome/renderer/print_web_view_helper.cc

 // Copyright (c) 2011 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/renderer/print_web_view_helper.h"
 
 #include <string>
 
 #include "base/command_line.h"
 #include "base/logging.h"
 #include "base/metrics/histogram.h"
 #include "base/process_util.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/print_messages.h"
 #include "chrome/common/render_messages.h"
 #include "chrome/renderer/prerender/prerender_helper.h"
 #include "content/renderer/render_view.h"
 #include "grit/generated_resources.h"
 #include "printing/metafile.h"
+#include "printing/print_job_constants.h"
 #include "printing/units.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebConsoleMessage.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebDataSource.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebDocument.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebElement.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebFrame.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebNode.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/WebSecurityPolicy.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/WebSize.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/WebString.h"
+#include "third_party/WebKit/Source/WebKit/chromium/public/WebURLRequest.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebURLResponse.h"
-#include "third_party/WebKit/Source/WebKit/chromium/public/WebSize.h"
-#include "third_party/WebKit/Source/WebKit/chromium/public/WebURLRequest.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebView.h"
 #include "ui/base/l10n/l10n_util.h"
 
 #if defined(OS_POSIX)
 #include "content/common/view_messages.h"
 #endif
 
 using printing::ConvertPixelsToPoint;
 using printing::ConvertPixelsToPointDouble;
 using printing::ConvertUnit;
 using printing::ConvertUnitDouble;
 using WebKit::WebConsoleMessage;
 using WebKit::WebDocument;
 using WebKit::WebElement;
 using WebKit::WebFrame;
 using WebKit::WebNode;
 using WebKit::WebSize;
 using WebKit::WebString;
 using WebKit::WebURLRequest;
 using WebKit::WebView;
 
 namespace {
 
+const double kMinDpi = 1.0;
+
 int GetDPI(const PrintMsg_Print_Params* print_params) {
 #if defined(OS_MACOSX)
   // On the Mac, the printable area is in points, don't do any scaling based
   // on dpi.
   return printing::kPointsPerInch;
 #else
   return static_cast<int>(print_params->dpi);
 #endif  // defined(OS_MACOSX)
 }
 
@@ skipping 95 matching lines @@
     context_menu_preview_node_.reset();
     Send(new PrintHostMsg_RequestPrintPreview(routing_id()));
   } else {
     Print(frame, NULL);
   }
 }
 
 bool PrintWebViewHelper::OnMessageReceived(const IPC::Message& message) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(PrintWebViewHelper, message)
+    IPC_MESSAGE_HANDLER(PrintMsg_PrintSetupWhitelist,
+                        OnPrintSetupWhitelist)
+    IPC_MESSAGE_HANDLER(PrintMsg_PrintTeardownWhitelist,
+                        OnPrintTeardownWhitelist)
+    IPC_MESSAGE_HANDLER(PrintMsg_PrintForPrintPreview,
+                        OnPrintForPrintPreview)
     IPC_MESSAGE_HANDLER(PrintMsg_PrintPages, OnPrintPages)
     IPC_MESSAGE_HANDLER(PrintMsg_InitiatePrintPreview,
                         OnInitiatePrintPreview)
     IPC_MESSAGE_HANDLER(PrintMsg_PrintNodeUnderContextMenu,
                         OnPrintNodeUnderContextMenu)
     IPC_MESSAGE_HANDLER(PrintMsg_PrintPreview, OnPrintPreview)
-    IPC_MESSAGE_HANDLER(PrintMsg_PrintForPrintPreview,
-                        OnPrintForPrintPreview)
     IPC_MESSAGE_HANDLER(PrintMsg_PrintingDone, OnPrintingDone)
     IPC_MESSAGE_HANDLER(PrintMsg_ResetScriptedPrintCount,
                         ResetScriptedPrintCount)
     IPC_MESSAGE_UNHANDLED(handled = false)
     IPC_END_MESSAGE_MAP()
   return handled;
 }
 
+void PrintWebViewHelper::OnPrintSetupWhitelist(std::string host) {
+  WebKit::WebSecurityPolicy::addOriginAccessWhitelistEntry(

[Lei Zhang, 2011/06/09 21:23:30]

I'm not a web security expert. Is this ok to do?

[Albert Bodenhamer, 2011/06/10 01:29:36]

I'm still trying to figure out the best person to ask.  I'll be sure to get an
expert opinion before submit.

Extensions use the same construct much more aggressively.  I expect this will be
fine.

+      GURL("chrome://print"),

[Lei Zhang, 2011/06/09 21:23:30]

chrome://print -> chrome::kChromeUIPrintHost

+      WebKit::WebString::fromUTF8("HTTPS"),

[Lei Zhang, 2011/06/09 21:23:30]

HTTPS -> chrome::kHttpsScheme

+      WebKit::WebString::fromUTF8(host),
+      true);
+}
+
+void PrintWebViewHelper::OnPrintTeardownWhitelist(std::string host) {
+  WebKit::WebSecurityPolicy::removeOriginAccessWhitelistEntry(
+      GURL("chrome://print"),
+      WebKit::WebString::fromUTF8("HTTPS"),
+      WebKit::WebString::fromUTF8(host),
+      true);
+}
+
 void PrintWebViewHelper::OnPrintForPrintPreview(
     const DictionaryValue& job_settings) {
   DCHECK(is_preview_);
   // If still not finished with earlier print request simply ignore.
   if (print_web_view_)
     return;
 
   if (!render_view()->webview())
     return;
   WebFrame* main_frame = render_view()->webview()->mainFrame();
   if (!main_frame)
     return;
 
   WebDocument document = main_frame->document();
   // <object> with id="pdf-viewer" is created in
   // chrome/browser/resources/print_preview.js
   WebElement pdf_element = document.getElementById("pdf-viewer");
   if (pdf_element.isNull()) {
     NOTREACHED();
     return;
   }
 
   WebFrame* pdf_frame = pdf_element.document().frame();
   if (!InitPrintSettings(pdf_frame, &pdf_element)) {
     NOTREACHED() << "Failed to initialize print page settings";
     return;
   }
-
   if (!UpdatePrintSettings(job_settings)) {
     DidFinishPrinting(FAIL_PRINT);
     return;
   }
 
   // Render Pages for printing.
   if (!RenderPagesForPrint(pdf_frame, &pdf_element))
     DidFinishPrinting(FAIL_PRINT);
 }
 
@@ skipping 359 matching lines @@
                                                 &settings.params));
   // Check if the printer returned any settings, if the settings is empty, we
   // can safely assume there are no printer drivers configured. So we safely
   // terminate.
   if (PrintMsg_Print_Params_IsEmpty(settings.params)) {
     render_view()->runModalAlertDialog(
         frame,
         l10n_util::GetStringUTF16(IDS_DEFAULT_PRINTER_NOT_FOUND_WARNING));
     return false;
   }
-  if (!settings.params.dpi || !settings.params.document_cookie) {
+  if (settings.params.dpi < kMinDpi || settings.params.document_cookie == 0) {
     // Invalid print page settings.
     NOTREACHED();
     return false;
   }
   UpdatePrintableSizeInPrintParameters(frame, node, &settings.params);
   settings.pages.clear();
   print_pages_params_.reset(new PrintMsg_PrintPages_Params(settings));
   return true;
 }
 
 bool PrintWebViewHelper::UpdatePrintSettings(
     const DictionaryValue& job_settings) {
-  PrintMsg_PrintPages_Params settings;
-  Send(new PrintHostMsg_UpdatePrintSettings(routing_id(),
-      print_pages_params_->params.document_cookie, job_settings, &settings));
+  bool print_to_cloud = job_settings.HasKey(printing::kSettingCloudPrintId);
+  if (print_to_cloud) {
+    // TODO(abodenha@chromium.org) Really update the settings for a cloud
+    // printer.
+    return true;
+  } else {
+    PrintMsg_PrintPages_Params settings;
+    Send(new PrintHostMsg_UpdatePrintSettings(routing_id(),
+        print_pages_params_->params.document_cookie, job_settings, &settings));
 
-  if (!settings.params.dpi)
-    return false;
+    if (settings.params.dpi < kMinDpi)
+      return false;
 
-  print_pages_params_.reset(new PrintMsg_PrintPages_Params(settings));
-  return true;
+    print_pages_params_.reset(new PrintMsg_PrintPages_Params(settings));
+    return true;
+  }
 }
 
 bool PrintWebViewHelper::GetPrintSettingsFromUser(WebKit::WebFrame* frame,
                                                   int expected_pages_count,
                                                   bool use_browser_overlays) {
   PrintHostMsg_ScriptedPrint_Params params;
   PrintMsg_PrintPages_Params print_settings;
 
   // The routing id is sent across as it is needed to look up the
   // corresponding RenderViewHost instance to signal and reset the
@@ skipping 120 matching lines @@
 
 void PrintWebViewHelper::ResetScriptedPrintCount() {
   // Reset cancel counter on successful print.
   user_cancelled_scripted_print_count_ = 0;
 }
 
 void PrintWebViewHelper::IncrementScriptedPrintCount() {
   ++user_cancelled_scripted_print_count_;
   last_cancelled_script_print_ = base::Time::Now();
 }