| // Copyright 2015 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef CONTENT_CHILD_BLOB_STORAGE_BLOB_TRANSPORT_CONTROLLER_H_ |
| #define CONTENT_CHILD_BLOB_STORAGE_BLOB_TRANSPORT_CONTROLLER_H_ |
| |
| #include <stddef.h> |
| |
| #include <map> |
| #include <string> |
| #include <vector> |
| |
| #include "base/gtest_prod_util.h" |
| #include "base/macros.h" |
| #include "base/memory/ref_counted.h" |
| #include "base/memory/scoped_ptr.h" |
| #include "base/memory/shared_memory_handle.h" |
| #include "content/common/content_export.h" |
| #include "ipc/ipc_platform_file.h" |
| #include "storage/common/blob_storage/blob_storage_constants.h" |
| |
| namespace base { |
| template <typename T> |
| struct DefaultLazyInstanceTraits; |
| class SingleThreadTaskRunner; |
| } |
| |
| namespace storage { |
| class DataElement; |
| struct BlobItemBytesRequest; |
| struct BlobItemBytesResponse; |
| } |
| |
| namespace IPC { |
| class Sender; |
| } |
| |
| namespace content { |
| |
| class BlobConsolidation; |
| class ThreadSafeSender; |
| |
| // This class is used to manage all the asynchronous transporation of blobs from |
| // the Renderer to the Browser process, where it's handling the Renderer side. |
| // The function of this class is to: |
| // * Be a lazy singleton, |
| // * hold all of the blob data that is being transported to the Browser process, |
| // * create the blob item 'descriptions' for the browser, |
| // * include shortcut data in the descriptions, |
| // * generate responses to blob memory requests, and |
| // * send IPC responses. |
| // Must be used on the IO thread. |
| class CONTENT_EXPORT BlobTransportController { |
| public: |
| static BlobTransportController* GetInstance(); |
| |
| // This kicks off a blob transfer to the browser thread, which involves |
| // sending an IPC message and storing the blob consolidation object. Designed |
| // to be called by the main thread or a worker thread. |
| // This also calls ChildProcess::AddRefProcess to keep our process around |
| // while we transfer. |
| static void InitiateBlobTransfer( |
| const std::string& uuid, |
| const std::string& content_type, |
| scoped_ptr<BlobConsolidation> consolidation, |
| scoped_refptr<ThreadSafeSender> sender, |
| base::SingleThreadTaskRunner* io_runner, |
| scoped_refptr<base::SingleThreadTaskRunner> main_runner); |
| |
| // This responds to the request using the sender. |
| void OnMemoryRequest( |
| const std::string& uuid, |
| const std::vector<storage::BlobItemBytesRequest>& requests, |
| std::vector<base::SharedMemoryHandle>* memory_handles, |
| const std::vector<IPC::PlatformFileForTransit>& file_handles, |
| IPC::Sender* sender); |
| |
| void OnCancel(const std::string& uuid, |
| storage::IPCBlobCreationCancelCode code); |
| |
| void OnDone(const std::string& uuid); |
| |
| bool IsTransporting(const std::string& uuid) { |
| return blob_storage_.find(uuid) != blob_storage_.end(); |
| } |
| |
| private: |
| friend class BlobTransportControllerTest; |
| FRIEND_TEST_ALL_PREFIXES(BlobTransportControllerTest, Descriptions); |
| FRIEND_TEST_ALL_PREFIXES(BlobTransportControllerTest, Responses); |
| FRIEND_TEST_ALL_PREFIXES(BlobTransportControllerTest, SharedMemory); |
| FRIEND_TEST_ALL_PREFIXES(BlobTransportControllerTest, ResponsesErrors); |
| |
| static void GetDescriptions(BlobConsolidation* consolidation, |
| size_t max_data_population, |
| std::vector<storage::DataElement>* out); |
| |
| BlobTransportController(); |
| ~BlobTransportController(); |
| |
| // Clears all internal state. If our map wasn't previously empty, then we call |
| // ChildProcess::ReleaseProcess to release our previous reference. |
| void ClearForTesting(); |
| |
| enum class ResponsesStatus { |
| BLOB_NOT_FOUND, |
| SHARED_MEMORY_MAP_FAILED, |
| SUCCESS |
| }; |
| friend struct base::DefaultLazyInstanceTraits<BlobTransportController>; |
| |
| void StoreBlobDataForRequests( |
| const std::string& uuid, |
| scoped_ptr<BlobConsolidation> consolidation, |
| scoped_refptr<base::SingleThreadTaskRunner> main_runner); |
| |
| ResponsesStatus GetResponses( |
| const std::string& uuid, |
| const std::vector<storage::BlobItemBytesRequest>& requests, |
| std::vector<base::SharedMemoryHandle>* memory_handles, |
| const std::vector<IPC::PlatformFileForTransit>& file_handles, |
| std::vector<storage::BlobItemBytesResponse>* output); |
| |
| // Deletes the consolidation, and if we removed the last consolidation from |
| // our map, we call ChildProcess::ReleaseProcess to release our previous |
| // reference. |
| void ReleaseBlobConsolidation(const std::string& uuid); |
| |
| scoped_refptr<base::SingleThreadTaskRunner> main_thread_runner_; |
| std::map<std::string, scoped_ptr<BlobConsolidation>> blob_storage_; |
| |
| DISALLOW_COPY_AND_ASSIGN(BlobTransportController); |
| }; |
| |
| } // namespace content |
| #endif // CONTENT_CHILD_BLOB_STORAGE_BLOB_TRANSPORT_CONTROLLER_H_ |