| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/child/blob_storage/blob_message_filter.h" | 5 #include "content/child/blob_storage/blob_message_filter.h" |
| 6 | 6 |
| 7 #include "base/bind.h" |
| 8 #include "base/task_runner.h" |
| 7 #include "content/child/blob_storage/blob_transport_controller.h" | 9 #include "content/child/blob_storage/blob_transport_controller.h" |
| 8 #include "content/child/thread_safe_sender.h" | 10 #include "content/child/thread_safe_sender.h" |
| 9 #include "content/common/fileapi/webblob_messages.h" | 11 #include "content/common/fileapi/webblob_messages.h" |
| 10 #include "ipc/ipc_message.h" | 12 #include "ipc/ipc_message.h" |
| 11 #include "ipc/ipc_sender.h" | 13 #include "ipc/ipc_sender.h" |
| 12 #include "storage/common/blob_storage/blob_item_bytes_request.h" | 14 #include "storage/common/blob_storage/blob_item_bytes_request.h" |
| 15 #include "storage/common/blob_storage/blob_item_bytes_response.h" |
| 13 | 16 |
| 14 namespace content { | 17 namespace content { |
| 15 | 18 |
| 16 BlobMessageFilter::BlobMessageFilter() | 19 BlobMessageFilter::BlobMessageFilter( |
| 17 : IPC::MessageFilter(), sender_(nullptr) {} | 20 scoped_refptr<base::TaskRunner> io_runner, |
| 21 scoped_refptr<base::TaskRunner> file_runner) |
| 22 : sender_(nullptr), |
| 23 io_runner_(std::move(io_runner)), |
| 24 file_runner_(std::move(file_runner)) {} |
| 18 | 25 |
| 19 BlobMessageFilter::~BlobMessageFilter() {} | 26 BlobMessageFilter::~BlobMessageFilter() {} |
| 20 | 27 |
| 21 void BlobMessageFilter::OnFilterAdded(IPC::Sender* sender) { | 28 void BlobMessageFilter::OnFilterAdded(IPC::Sender* sender) { |
| 22 sender_ = sender; | 29 sender_ = sender; |
| 23 } | 30 } |
| 24 | 31 |
| 32 void BlobMessageFilter::OnChannelClosing() { |
| 33 BlobTransportController::GetInstance()->CancelAllBlobTransfers(); |
| 34 sender_ = nullptr; |
| 35 } |
| 36 |
| 25 bool BlobMessageFilter::OnMessageReceived(const IPC::Message& message) { | 37 bool BlobMessageFilter::OnMessageReceived(const IPC::Message& message) { |
| 26 bool handled = true; | 38 bool handled = true; |
| 27 IPC_BEGIN_MESSAGE_MAP(BlobMessageFilter, message) | 39 IPC_BEGIN_MESSAGE_MAP(BlobMessageFilter, message) |
| 28 IPC_MESSAGE_HANDLER(BlobStorageMsg_RequestMemoryItem, OnRequestMemoryItem) | 40 IPC_MESSAGE_HANDLER(BlobStorageMsg_RequestMemoryItem, OnRequestMemoryItem) |
| 29 IPC_MESSAGE_HANDLER(BlobStorageMsg_CancelBuildingBlob, OnCancelBuildingBlob) | 41 IPC_MESSAGE_HANDLER(BlobStorageMsg_CancelBuildingBlob, OnCancelBuildingBlob) |
| 30 IPC_MESSAGE_HANDLER(BlobStorageMsg_DoneBuildingBlob, OnDoneBuildingBlob) | 42 IPC_MESSAGE_HANDLER(BlobStorageMsg_DoneBuildingBlob, OnDoneBuildingBlob) |
| 31 IPC_MESSAGE_UNHANDLED(handled = false) | 43 IPC_MESSAGE_UNHANDLED(handled = false) |
| 32 IPC_END_MESSAGE_MAP() | 44 IPC_END_MESSAGE_MAP() |
| 33 return handled; | 45 return handled; |
| 34 } | 46 } |
| 35 | 47 |
| 36 bool BlobMessageFilter::GetSupportedMessageClasses( | 48 bool BlobMessageFilter::GetSupportedMessageClasses( |
| 37 std::vector<uint32_t>* supported_message_classes) const { | 49 std::vector<uint32_t>* supported_message_classes) const { |
| 38 supported_message_classes->push_back(BlobMsgStart); | 50 supported_message_classes->push_back(BlobMsgStart); |
| 39 return true; | 51 return true; |
| 40 } | 52 } |
| 41 | 53 |
| 42 void BlobMessageFilter::OnRequestMemoryItem( | 54 void BlobMessageFilter::OnRequestMemoryItem( |
| 43 const std::string& uuid, | 55 const std::string& uuid, |
| 44 const std::vector<storage::BlobItemBytesRequest>& requests, | 56 const std::vector<storage::BlobItemBytesRequest>& requests, |
| 45 std::vector<base::SharedMemoryHandle> memory_handles, | 57 std::vector<base::SharedMemoryHandle> memory_handles, |
| 46 const std::vector<IPC::PlatformFileForTransit>& file_handles) { | 58 const std::vector<IPC::PlatformFileForTransit>& file_handles) { |
| 47 BlobTransportController::GetInstance()->OnMemoryRequest( | 59 BlobTransportController::GetInstance()->OnMemoryRequest( |
| 48 uuid, requests, &memory_handles, file_handles, sender_); | 60 uuid, requests, &memory_handles, file_handles, file_runner_.get(), |
| 61 sender_); |
| 49 } | 62 } |
| 50 | 63 |
| 51 void BlobMessageFilter::OnCancelBuildingBlob( | 64 void BlobMessageFilter::OnCancelBuildingBlob( |
| 52 const std::string& uuid, | 65 const std::string& uuid, |
| 53 storage::IPCBlobCreationCancelCode code) { | 66 storage::IPCBlobCreationCancelCode code) { |
| 54 BlobTransportController::GetInstance()->OnCancel(uuid, code); | 67 BlobTransportController::GetInstance()->OnCancel(uuid, code); |
| 55 } | 68 } |
| 56 | 69 |
| 57 void BlobMessageFilter::OnDoneBuildingBlob(const std::string& uuid) { | 70 void BlobMessageFilter::OnDoneBuildingBlob(const std::string& uuid) { |
| 58 BlobTransportController::GetInstance()->OnDone(uuid); | 71 BlobTransportController::GetInstance()->OnDone(uuid); |
| 59 } | 72 } |
| 60 | 73 |
| 61 } // namespace content | 74 } // namespace content |
| OLD | NEW |