Use pdf compositor service for printing when OOPIF is enabled

components/printing/browser/print_composite_client.cc

Index: components/printing/browser/print_composite_client.cc
diff --git a/components/printing/browser/print_composite_client.cc b/components/printing/browser/print_composite_client.cc
new file mode 100644
index 0000000000000000000000000000000000000000..64d21679d264bcb97f2bb3e051ca06d17117630b
--- /dev/null
+++ b/components/printing/browser/print_composite_client.cc
@@ -0,0 +1,87 @@
+// Copyright (c) 2017 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 "components/printing/browser/print_composite_client.h"
+
+#include <memory>
+#include <utility>
+
+#include "base/threading/thread_task_runner_handle.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/common/service_manager_connection.h"
+#include "mojo/public/cpp/system/platform_handle.h"
+#include "services/service_manager/public/cpp/connector.h"
+
+DEFINE_WEB_CONTENTS_USER_DATA_KEY(printing::PrintCompositeClient);
+
+namespace printing {
+
+PrintCompositeClient::PrintCompositeClient(content::WebContents* web_contents)
+    : weak_factory_(this) {}
+
+PrintCompositeClient::~PrintCompositeClient() {}
+
+void PrintCompositeClient::CreateConnectorRequest() {
+  connector_ = service_manager::Connector::Create(&connector_request_);
+  content::ServiceManagerConnection::GetForProcess()
+      ->GetConnector()
+      ->BindConnectorRequest(std::move(connector_request_));
+}
+
+void PrintCompositeClient::DoComposite(
+    base::SharedMemoryHandle handle,
+    uint32_t data_size,
+    mojom::PdfCompositor::CompositePdfCallback callback,
+    scoped_refptr<base::SequencedTaskRunner> callback_task_runner) {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
+
+  if (!data_size)
+    return;
+
+  // if (connector_request_.is_pending())
+  if (!connector_)
+    CreateConnectorRequest();
+  Composite(connector_.get(), handle, data_size, std::move(callback),
+            callback_task_runner);
+}
+
+std::unique_ptr<base::SharedMemory> PrintCompositeClient::GetShmFromMojoHandle(
+    mojo::ScopedSharedBufferHandle handle) {
+  if (!handle.is_valid()) {
+    DLOG(ERROR) << "Invalid mojo handle.";
+    return nullptr;
+  }
+
+  base::SharedMemoryHandle memory_handle;
+  size_t memory_size = 0;
+  bool read_only_flag = false;
+
+  const MojoResult result = mojo::UnwrapSharedMemoryHandle(
+      std::move(handle), &memory_handle, &memory_size, &read_only_flag);
+  DCHECK_EQ(MOJO_RESULT_OK, result);
+  DCHECK_GT(memory_size, 0u);
+
+  std::unique_ptr<base::SharedMemory> shm(
+      new base::SharedMemory(memory_handle, true /* read_only */));

[Lei Zhang, 2017/08/28 23:01:12]

MakeUnique

[Wei Li, 2017/08/30 00:24:02]

Done.

+  if (!shm->Map(memory_size)) {
+    DLOG(ERROR) << "Map shared memory failed.";
+    return nullptr;
+  }
+  return shm;
+}
+
+scoped_refptr<base::RefCountedBytes>
+PrintCompositeClient::GetDataFromMojoHandle(
+    mojo::ScopedSharedBufferHandle handle) {
+  std::unique_ptr<base::SharedMemory> shm =
+      GetShmFromMojoHandle(std::move(handle));
+  if (!shm)
+    return nullptr;
+
+  return scoped_refptr<base::RefCountedBytes>(new base::RefCountedBytes(

[Lei Zhang, 2017/08/28 23:01:12]

MakeRefCounted

[Wei Li, 2017/08/30 00:24:02]

RefCountedBytes doesn't have similar function defined.

[Lei Zhang, 2017/08/30 02:00:35]

return base::MakeRefCounted<base::RefCountedBytes>(
    reinterpret_cast<const unsigned char*>(shm->memory()),
    shm->mapped_size());

[Wei Li, 2017/08/31 21:00:25]

Thanks, done.

+      reinterpret_cast<const unsigned char*>(shm->memory()),
+      shm->mapped_size()));
+}
+
+}  // namespace printing