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 |