From 16348fc8956dcee0712863bc9bb2b866532740ea Mon Sep 17 00:00:00 2001
From: Haojian Wu <hokein.wu@gmail.com>
Date: Tue, 16 Jun 2015 19:42:17 +0800
Subject: [PATCH] Copy pdf data on IO thread to avoid causing main process
 hangs.

---
 .../printing/print_preview_message_handler.cc | 39 ++++++++++++-------
 .../printing/print_preview_message_handler.h  |  3 +-
 2 files changed, 27 insertions(+), 15 deletions(-)

diff --git a/chromium_src/chrome/browser/printing/print_preview_message_handler.cc b/chromium_src/chrome/browser/printing/print_preview_message_handler.cc
index 51bfffce9cb1..ca70ac2c58ef 100644
--- a/chromium_src/chrome/browser/printing/print_preview_message_handler.cc
+++ b/chromium_src/chrome/browser/printing/print_preview_message_handler.cc
@@ -40,6 +40,19 @@ void StopWorker(int document_cookie) {
   }
 }
 
+char* CopyPDFDataOnIOThread(
+    const PrintHostMsg_DidPreviewDocument_Params& params) {
+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
+  scoped_ptr<base::SharedMemory> shared_buf(
+      new base::SharedMemory(params.metafile_data_handle, true));
+  if (!shared_buf->Map(params.data_size))
+    return nullptr;
+  char* memory_pdf_data = static_cast<char*>(shared_buf->memory());
+  char* pdf_data = new char[params.data_size];
+  memcpy(pdf_data, memory_pdf_data, params.data_size);
+  return pdf_data;
+}
+
 }  // namespace
 
 namespace printing {
@@ -64,22 +77,20 @@ void PrintPreviewMessageHandler::OnMetafileReadyForPrinting(
     return;
   }
 
-  scoped_ptr<base::SharedMemory> shared_buf(
-      new base::SharedMemory(params.metafile_data_handle, true));
-  if (!shared_buf->Map(params.data_size)) {
-    RunPrintToPDFCallback(params.preview_request_id, nullptr, 0);
-    return;
-  }
-
-  RunPrintToPDFCallback(params.preview_request_id,
-      static_cast<char*>(shared_buf->memory()),
-      params.data_size);
+  BrowserThread::PostTaskAndReplyWithResult(
+      BrowserThread::IO,
+      FROM_HERE,
+      base::Bind(&CopyPDFDataOnIOThread, params),
+      base::Bind(&PrintPreviewMessageHandler::RunPrintToPDFCallback,
+                 base::Unretained(this),
+                 params.preview_request_id,
+                 params.data_size));
 }
 
 void PrintPreviewMessageHandler::OnPrintPreviewFailed(int document_cookie,
                                                       int request_id) {
   StopWorker(document_cookie);
-  RunPrintToPDFCallback(request_id, nullptr, 0);
+  RunPrintToPDFCallback(request_id, 0, nullptr);
 }
 
 bool PrintPreviewMessageHandler::OnMessageReceived(
@@ -107,12 +118,14 @@ void PrintPreviewMessageHandler::PrintToPDF(
 }
 
 void PrintPreviewMessageHandler::RunPrintToPDFCallback(
-     int request_id, char* data, uint32 data_size) {
+     int request_id, uint32 data_size, char* data) {
+  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+
   v8::Isolate* isolate = v8::Isolate::GetCurrent();
   v8::Locker locker(isolate);
   v8::HandleScope handle_scope(isolate);
   if (data) {
-    v8::Local<v8::Value> buffer = node::Buffer::New(isolate,
+    v8::Local<v8::Value> buffer = node::Buffer::Use(isolate,
         data, static_cast<size_t>(data_size));
     print_to_pdf_callback_map_[request_id].Run(v8::Null(isolate), buffer);
   } else {
diff --git a/chromium_src/chrome/browser/printing/print_preview_message_handler.h b/chromium_src/chrome/browser/printing/print_preview_message_handler.h
index a16591dd9f66..453d78761bbe 100644
--- a/chromium_src/chrome/browser/printing/print_preview_message_handler.h
+++ b/chromium_src/chrome/browser/printing/print_preview_message_handler.h
@@ -47,8 +47,7 @@ class PrintPreviewMessageHandler
       const PrintHostMsg_DidPreviewDocument_Params& params);
   void OnPrintPreviewFailed(int document_cookie, int request_id);
 
-  void RunPrintToPDFCallback(
-     int request_id, char* data, uint32 data_size);
+  void RunPrintToPDFCallback(int request_id, uint32 data_size, char* data);
 
   PrintToPDFCallbackMap print_to_pdf_callback_map_;