PrintPreview: Print Preview is not staying associated with initiator renderer.

chrome/browser/printing/print_preview_message_handler.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/browser/printing/print_preview_message_handler.h"
 
+#include "base/memory/ref_counted.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/printing/print_job_manager.h"
 #include "chrome/browser/printing/print_preview_tab_controller.h"
 #include "chrome/browser/printing/print_view_manager.h"
 #include "chrome/browser/printing/printer_query.h"
 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
 #include "chrome/browser/ui/webui/print_preview_handler.h"
 #include "chrome/browser/ui/webui/print_preview_ui.h"
-#include "chrome/browser/ui/webui/print_preview_ui_html_source.h"
 #include "chrome/common/print_messages.h"
 #include "content/browser/browser_thread.h"
 #include "content/browser/renderer_host/render_view_host.h"
 #include "content/browser/tab_contents/tab_contents.h"
 #include "content/common/content_restriction.h"
 
 namespace {
 
 void StopWorker(int document_cookie) {
   printing::PrintJobManager* print_job_manager =
@@ skipping 51 matching lines @@
     delete shared_buf;
     return;
   }
 
   TabContentsWrapper* wrapper =
       TabContentsWrapper::GetCurrentWrapperForContents(print_preview_tab);
   wrapper->print_view_manager()->OverrideTitle(tab_contents());
 
   PrintPreviewUI* print_preview_ui =
       static_cast<PrintPreviewUI*>(print_preview_tab->web_ui());
-  PrintPreviewUIHTMLSource* html_source = print_preview_ui->html_source();
-  html_source->SetPrintPreviewData(
-      std::make_pair(shared_buf, params.data_size));
+
+  char* preview_data = reinterpret_cast<char*>(shared_buf->memory());

[Lei Zhang, 2011/05/27 17:26:53]

nit: static_cast

[kmadhusu, 2011/05/27 23:44:42]

Done.

+  uint32 preview_data_size = params.data_size;
+
+  scoped_refptr<RefCountedBytes> html_bytes(new RefCountedBytes);
+  html_bytes->data.resize(preview_data_size);
+  std::vector<unsigned char>::iterator it = html_bytes->data.begin();
+  for (uint32 i = 0; i < preview_data_size; ++i, ++it)
+    *it = *(preview_data + i);
+
+  print_preview_ui->SetPrintPreviewData(html_bytes.get());
   print_preview_ui->OnPreviewDataIsAvailable(
       params.expected_pages_count,
       wrapper->print_view_manager()->RenderSourceName(),
       params.modifiable);
 }
 
 void PrintPreviewMessageHandler::OnPrintPreviewNodeUnderContextMenu() {
   PrintPreviewTabController::PrintPreview(tab_contents());
 }
 
@@ skipping 31 matching lines @@
 }
 
 void PrintPreviewMessageHandler::DidStartLoading() {
   if (tab_contents()->delegate() &&
       printing::PrintPreviewTabController::IsPrintPreviewTab(tab_contents())) {
     tab_contents()->SetContentRestrictions(CONTENT_RESTRICTION_PRINT);
   }
 }
 
 }  // namespace printing