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 |