| 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> file_runner) |
| 21 : sender_(nullptr), file_runner_(std::move(file_runner)) {} |
| 18 | 22 |
| 19 BlobMessageFilter::~BlobMessageFilter() {} | 23 BlobMessageFilter::~BlobMessageFilter() {} |
| 20 | 24 |
| 21 void BlobMessageFilter::OnFilterAdded(IPC::Sender* sender) { | 25 void BlobMessageFilter::OnFilterAdded(IPC::Sender* sender) { |
| 22 sender_ = sender; | 26 sender_ = sender; |
| 23 } | 27 } |
| 24 | 28 |
| 29 void BlobMessageFilter::OnChannelClosing() { |
| 30 BlobTransportController::GetInstance()->CancelAllBlobTransfers(); |
| 31 sender_ = nullptr; |
| 32 } |
| 33 |
| 25 bool BlobMessageFilter::OnMessageReceived(const IPC::Message& message) { | 34 bool BlobMessageFilter::OnMessageReceived(const IPC::Message& message) { |
| 26 bool handled = true; | 35 bool handled = true; |
| 27 IPC_BEGIN_MESSAGE_MAP(BlobMessageFilter, message) | 36 IPC_BEGIN_MESSAGE_MAP(BlobMessageFilter, message) |
| 28 IPC_MESSAGE_HANDLER(BlobStorageMsg_RequestMemoryItem, OnRequestMemoryItem) | 37 IPC_MESSAGE_HANDLER(BlobStorageMsg_RequestMemoryItem, OnRequestMemoryItem) |
| 29 IPC_MESSAGE_HANDLER(BlobStorageMsg_CancelBuildingBlob, OnCancelBuildingBlob) | 38 IPC_MESSAGE_HANDLER(BlobStorageMsg_CancelBuildingBlob, OnCancelBuildingBlob) |
| 30 IPC_MESSAGE_HANDLER(BlobStorageMsg_DoneBuildingBlob, OnDoneBuildingBlob) | 39 IPC_MESSAGE_HANDLER(BlobStorageMsg_DoneBuildingBlob, OnDoneBuildingBlob) |
| 31 IPC_MESSAGE_UNHANDLED(handled = false) | 40 IPC_MESSAGE_UNHANDLED(handled = false) |
| 32 IPC_END_MESSAGE_MAP() | 41 IPC_END_MESSAGE_MAP() |
| 33 return handled; | 42 return handled; |
| 34 } | 43 } |
| 35 | 44 |
| 36 bool BlobMessageFilter::GetSupportedMessageClasses( | 45 bool BlobMessageFilter::GetSupportedMessageClasses( |
| 37 std::vector<uint32_t>* supported_message_classes) const { | 46 std::vector<uint32_t>* supported_message_classes) const { |
| 38 supported_message_classes->push_back(BlobMsgStart); | 47 supported_message_classes->push_back(BlobMsgStart); |
| 39 return true; | 48 return true; |
| 40 } | 49 } |
| 41 | 50 |
| 42 void BlobMessageFilter::OnRequestMemoryItem( | 51 void BlobMessageFilter::OnRequestMemoryItem( |
| 43 const std::string& uuid, | 52 const std::string& uuid, |
| 44 const std::vector<storage::BlobItemBytesRequest>& requests, | 53 const std::vector<storage::BlobItemBytesRequest>& requests, |
| 45 std::vector<base::SharedMemoryHandle> memory_handles, | 54 std::vector<base::SharedMemoryHandle> memory_handles, |
| 46 const std::vector<IPC::PlatformFileForTransit>& file_handles) { | 55 const std::vector<IPC::PlatformFileForTransit>& file_handles) { |
| 47 BlobTransportController::GetInstance()->OnMemoryRequest( | 56 BlobTransportController::GetInstance()->OnMemoryRequest( |
| 48 uuid, requests, &memory_handles, file_handles, sender_); | 57 uuid, requests, &memory_handles, file_handles, file_runner_.get(), |
| 58 sender_); |
| 49 } | 59 } |
| 50 | 60 |
| 51 void BlobMessageFilter::OnCancelBuildingBlob( | 61 void BlobMessageFilter::OnCancelBuildingBlob( |
| 52 const std::string& uuid, | 62 const std::string& uuid, |
| 53 storage::IPCBlobCreationCancelCode code) { | 63 storage::IPCBlobCreationCancelCode code) { |
| 54 BlobTransportController::GetInstance()->OnCancel(uuid, code); | 64 BlobTransportController::GetInstance()->OnCancel(uuid, code); |
| 55 } | 65 } |
| 56 | 66 |
| 57 void BlobMessageFilter::OnDoneBuildingBlob(const std::string& uuid) { | 67 void BlobMessageFilter::OnDoneBuildingBlob(const std::string& uuid) { |
| 58 BlobTransportController::GetInstance()->OnDone(uuid); | 68 BlobTransportController::GetInstance()->OnDone(uuid); |
| 59 } | 69 } |
| 60 | 70 |
| 61 } // namespace content | 71 } // namespace content |
| OLD | NEW |