| // Copyright 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 <utility> |
| |
| #include "base/bind.h" |
| #include "base/memory/shared_memory_handle.h" |
| #include "components/printing/common/print_messages.h" |
| #include "content/public/browser/browser_thread.h" |
| #include "content/public/browser/render_frame_host.h" |
| #include "content/public/browser/render_process_host.h" |
| #include "content/public/common/service_manager_connection.h" |
| #include "mojo/public/cpp/system/platform_handle.h" |
| #include "printing/printing_utils.h" |
| #include "services/service_manager/public/cpp/connector.h" |
| |
| DEFINE_WEB_CONTENTS_USER_DATA_KEY(printing::PrintCompositeClient); |
| |
| namespace { |
| |
| uint64_t GenerateFrameGuid(int process_id, int frame_id) { |
| return static_cast<uint64_t>(process_id) << 32 | frame_id; |
| } |
| |
| } // namespace |
| |
| namespace printing { |
| |
| PrintCompositeClient::PrintCompositeClient(content::WebContents* web_contents) |
| : content::WebContentsObserver(web_contents), for_preview_(false) {} |
| |
| PrintCompositeClient::~PrintCompositeClient() {} |
| |
| bool PrintCompositeClient::OnMessageReceived( |
| const IPC::Message& message, |
| content::RenderFrameHost* render_frame_host) { |
| bool handled = true; |
| IPC_BEGIN_MESSAGE_MAP_WITH_PARAM(PrintCompositeClient, message, |
| render_frame_host) |
| IPC_MESSAGE_HANDLER(PrintHostMsg_DidPrintFrameContent, |
| OnDidPrintFrameContent) |
| IPC_MESSAGE_UNHANDLED(handled = false) |
| IPC_END_MESSAGE_MAP() |
| return handled; |
| } |
| |
| void PrintCompositeClient::OnDidPrintFrameContent( |
| content::RenderFrameHost* render_frame_host, |
| int document_cookie, |
| const PrintHostMsg_DidPrintContent_Params& params) { |
| // Content in |params| is sent from untrusted source; only minimal processing |
| // is done here. Most of it will be directly forwarded to pdf compositor |
| // service. |
| auto& compositor = GetCompositeRequest(document_cookie); |
| DCHECK(compositor.is_bound()); |
| |
| mojo::ScopedSharedBufferHandle buffer_handle = mojo::WrapSharedMemoryHandle( |
| params.metafile_data_handle, params.data_size, |
| mojo::UnwrappedSharedMemoryHandleProtection::kReadOnly); |
| auto frame_guid = GenerateFrameGuid(render_frame_host->GetProcess()->GetID(), |
| render_frame_host->GetRoutingID()); |
| compositor->AddSubframeContent( |
| frame_guid, std::move(buffer_handle), |
| ConvertContentInfoMap(web_contents(), render_frame_host, |
| params.subframe_content_info)); |
| } |
| |
| void PrintCompositeClient::PrintCrossProcessSubframe( |
| const gfx::Rect& rect, |
| int document_cookie, |
| content::RenderFrameHost* subframe_host) { |
| PrintMsg_PrintFrame_Params params; |
| params.printable_area = rect; |
| params.document_cookie = document_cookie; |
| // Send the request to the destination frame. |
| subframe_host->Send( |
| new PrintMsg_PrintFrameContent(subframe_host->GetRoutingID(), params)); |
| } |
| |
| void PrintCompositeClient::DoCompositePageToPdf( |
| int document_cookie, |
| content::RenderFrameHost* render_frame_host, |
| int page_num, |
| base::SharedMemoryHandle handle, |
| uint32_t data_size, |
| const ContentToProxyIdMap& subframe_content_info, |
| mojom::PdfCompositor::CompositePageToPdfCallback callback) { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| auto& compositor = GetCompositeRequest(document_cookie); |
| |
| DCHECK(data_size); |
| mojo::ScopedSharedBufferHandle buffer_handle = mojo::WrapSharedMemoryHandle( |
| handle, data_size, |
| mojo::UnwrappedSharedMemoryHandleProtection::kReadOnly); |
| // Since this class owns compositor, compositor will be gone when this class |
| // is destructed. Mojo won't call its callback in that case so it is safe to |
| // use unretained |this| pointer here. |
| compositor->CompositePageToPdf( |
| GenerateFrameGuid(render_frame_host->GetProcess()->GetID(), |
| render_frame_host->GetRoutingID()), |
| page_num, std::move(buffer_handle), |
| ConvertContentInfoMap(web_contents(), render_frame_host, |
| subframe_content_info), |
| base::BindOnce(&PrintCompositeClient::OnDidCompositePageToPdf, |
| base::Unretained(this), std::move(callback))); |
| } |
| |
| void PrintCompositeClient::DoCompositeDocumentToPdf( |
| int document_cookie, |
| content::RenderFrameHost* render_frame_host, |
| base::SharedMemoryHandle handle, |
| uint32_t data_size, |
| const ContentToProxyIdMap& subframe_content_info, |
| mojom::PdfCompositor::CompositeDocumentToPdfCallback callback) { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| auto& compositor = GetCompositeRequest(document_cookie); |
| |
| DCHECK(data_size); |
| mojo::ScopedSharedBufferHandle buffer_handle = mojo::WrapSharedMemoryHandle( |
| handle, data_size, |
| mojo::UnwrappedSharedMemoryHandleProtection::kReadOnly); |
| // Since this class owns compositor, compositor will be gone when this class |
| // is destructed. Mojo won't call its callback in that case so it is safe to |
| // use unretained |this| pointer here. |
| compositor->CompositeDocumentToPdf( |
| GenerateFrameGuid(render_frame_host->GetProcess()->GetID(), |
| render_frame_host->GetRoutingID()), |
| std::move(buffer_handle), |
| ConvertContentInfoMap(web_contents(), render_frame_host, |
| subframe_content_info), |
| base::BindOnce(&PrintCompositeClient::OnDidCompositeDocumentToPdf, |
| base::Unretained(this), document_cookie, |
| std::move(callback))); |
| } |
| |
| void PrintCompositeClient::OnDidCompositePageToPdf( |
| printing::mojom::PdfCompositor::CompositePageToPdfCallback callback, |
| printing::mojom::PdfCompositor::Status status, |
| mojo::ScopedSharedBufferHandle handle) { |
| std::move(callback).Run(status, std::move(handle)); |
| } |
| |
| void PrintCompositeClient::OnDidCompositeDocumentToPdf( |
| int document_cookie, |
| printing::mojom::PdfCompositor::CompositeDocumentToPdfCallback callback, |
| printing::mojom::PdfCompositor::Status status, |
| mojo::ScopedSharedBufferHandle handle) { |
| RemoveCompositeRequest(document_cookie); |
| std::move(callback).Run(status, std::move(handle)); |
| } |
| |
| ContentToFrameMap PrintCompositeClient::ConvertContentInfoMap( |
| content::WebContents* web_contents, |
| content::RenderFrameHost* render_frame_host, |
| const ContentToProxyIdMap& content_proxy_map) { |
| ContentToFrameMap content_frame_map; |
| int process_id = render_frame_host->GetProcess()->GetID(); |
| for (auto& entry : content_proxy_map) { |
| auto content_id = entry.first; |
| auto proxy_id = entry.second; |
| // Find the RenderFrameHost that the proxy id corresponds to. |
| content::RenderFrameHost* rfh = |
| content::RenderFrameHost::FromPlaceholderId(process_id, proxy_id); |
| if (!rfh) { |
| // If we could not find the corresponding RenderFrameHost, |
| // just skip it. |
| continue; |
| } |
| |
| // Store this frame's global unique id into the map. |
| content_frame_map[content_id] = |
| GenerateFrameGuid(rfh->GetProcess()->GetID(), rfh->GetRoutingID()); |
| } |
| return content_frame_map; |
| } |
| |
| mojom::PdfCompositorPtr& PrintCompositeClient::GetCompositeRequest(int cookie) { |
| auto iter = compositor_map_.find(cookie); |
| if (iter != compositor_map_.end()) |
| return iter->second; |
| |
| auto iterator = |
| compositor_map_.emplace(cookie, CreateCompositeRequest()).first; |
| return iterator->second; |
| } |
| |
| void PrintCompositeClient::RemoveCompositeRequest(int cookie) { |
| size_t erased = compositor_map_.erase(cookie); |
| DCHECK_EQ(erased, 1u); |
| } |
| |
| mojom::PdfCompositorPtr PrintCompositeClient::CreateCompositeRequest() { |
| if (!connector_) { |
| service_manager::mojom::ConnectorRequest connector_request; |
| connector_ = service_manager::Connector::Create(&connector_request); |
| content::ServiceManagerConnection::GetForProcess() |
| ->GetConnector() |
| ->BindConnectorRequest(std::move(connector_request)); |
| } |
| mojom::PdfCompositorPtr compositor; |
| connector_->BindInterface(mojom::kServiceName, &compositor); |
| return compositor; |
| } |
| |
| } // namespace printing |