Enable PDF plugin for iframes within the print preview dialog.

chrome/browser/printing/print_view_manager.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/printing/print_view_manager.h"
 
 #include <map>
 
 #include "base/bind.h"
 #include "base/lazy_instance.h"
 #include "base/metrics/histogram.h"
 #include "chrome/browser/browser_process.h"
+#include "chrome/browser/plugins/chrome_plugin_service_filter.h"
 #include "chrome/browser/printing/print_job_manager.h"
 #include "chrome/browser/printing/print_preview_dialog_controller.h"
 #include "chrome/browser/printing/print_view_manager_observer.h"
 #include "chrome/browser/ui/webui/print_preview/print_preview_ui.h"
+#include "chrome/common/chrome_content_client.h"
 #include "components/printing/common/print_messages.h"
 #include "content/public/browser/browser_thread.h"
+#include "content/public/browser/plugin_service.h"
+#include "content/public/browser/render_frame_host.h"
+#include "content/public/browser/render_process_host.h"
 #include "content/public/browser/web_contents.h"
+#include "content/public/common/webplugininfo.h"
 
 using content::BrowserThread;
 
 DEFINE_WEB_CONTENTS_USER_DATA_KEY(printing::PrintViewManager);
 
 namespace {
 
 // Keeps track of pending scripted print preview closures.
 // No locking, only access on the UI thread.
 typedef std::map<content::RenderProcessHost*, base::Closure>
     ScriptedPrintPreviewClosureMap;
 static base::LazyInstance<ScriptedPrintPreviewClosureMap>
     g_scripted_print_preview_closure_map = LAZY_INSTANCE_INITIALIZER;
 
+void EnableInternalPDFPluginForContents(int render_process_id,
+                                        int render_frame_id) {
+  // Always enable the internal PDF plugin for the print preview page.
+  base::FilePath pdf_plugin_path = base::FilePath::FromUTF8Unsafe(
+      ChromeContentClient::kPDFPluginPath);
+
+  content::WebPluginInfo pdf_plugin;
+  if (!content::PluginService::GetInstance()->GetPluginInfoByPath(
+      pdf_plugin_path, &pdf_plugin))

[raymes, 2015/03/19 02:14:33]

nit: add {}

[Lei Zhang, 2015/03/19 02:20:08]

Done.

+    return;
+
+  ChromePluginServiceFilter::GetInstance()->OverridePluginForFrame(
+      render_process_id, render_frame_id, GURL(), pdf_plugin);
+}
+
 }  // namespace
 
 namespace printing {
 
 PrintViewManager::PrintViewManager(content::WebContents* web_contents)
     : PrintViewManagerBase(web_contents),
       observer_(NULL),
       print_preview_state_(NOT_PREVIEWING),
       scripted_print_preview_rph_(NULL) {
+  if (PrintPreviewDialogController::IsPrintPreviewDialog(web_contents)) {
+    EnableInternalPDFPluginForContents(
+        web_contents->GetRenderProcessHost()->GetID(),
+        web_contents->GetMainFrame()->GetRoutingID());
+  }
 }
 
 PrintViewManager::~PrintViewManager() {
   DCHECK_EQ(NOT_PREVIEWING, print_preview_state_);
 }
 
 #if defined(ENABLE_BASIC_PRINTING)
 bool PrintViewManager::PrintForSystemDialogNow() {
   return PrintNowInternal(new PrintMsg_PrintForSystemDialog(routing_id()));
 }
@@ skipping 10 matching lines @@
       return false;
     PrintPreviewUI* print_preview_ui = static_cast<PrintPreviewUI*>(
         print_preview_dialog->GetWebUI()->GetController());
     print_preview_ui->OnShowSystemDialog();
     return true;
   } else {
     return PrintNow();
   }
 }
 #endif  // ENABLE_BASIC_PRINTING
+
 bool PrintViewManager::PrintPreviewNow(bool selection_only) {
   // Users can send print commands all they want and it is beyond
   // PrintViewManager's control. Just ignore the extra commands.
   // See http://crbug.com/136842 for example.
   if (print_preview_state_ != NOT_PREVIEWING)
     return false;
 
   if (!PrintNowInternal(new PrintMsg_InitiatePrintPreview(routing_id(),
                                                           selection_only))) {
     return false;
@@ skipping 24 matching lines @@
     scripted_print_preview_rph_ = NULL;
   }
   print_preview_state_ = NOT_PREVIEWING;
 }
 
 void PrintViewManager::set_observer(PrintViewManagerObserver* observer) {
   DCHECK(!observer || !observer_);
   observer_ = observer;
 }
 
+void PrintViewManager::RenderFrameCreated(
+    content::RenderFrameHost* render_frame_host) {
+  if (PrintPreviewDialogController::IsPrintPreviewDialog(web_contents())) {
+    EnableInternalPDFPluginForContents(render_frame_host->GetProcess()->GetID(),
+                                       render_frame_host->GetRoutingID());
+  }
+}
+
 void PrintViewManager::RenderProcessGone(base::TerminationStatus status) {
   print_preview_state_ = NOT_PREVIEWING;
   PrintViewManagerBase::RenderProcessGone(status);
 }
 
 void PrintViewManager::OnDidShowPrintDialog() {
   if (observer_)
     observer_->OnPrintDialogShown();
 }
 
@@ skipping 60 matching lines @@
                                     OnSetupScriptedPrintPreview)
     IPC_MESSAGE_HANDLER(PrintHostMsg_ShowScriptedPrintPreview,
                         OnShowScriptedPrintPreview)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
 
   return handled ? true : PrintViewManagerBase::OnMessageReceived(message);
 }
 
 }  // namespace printing