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