| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 #ifndef CONTENT_CHILD_URL_LOADER_CLIENT_IMPL_H_ | 5 #ifndef CONTENT_CHILD_URL_LOADER_CLIENT_IMPL_H_ |
| 6 #define CONTENT_CHILD_URL_LOADER_CLIENT_IMPL_H_ | 6 #define CONTENT_CHILD_URL_LOADER_CLIENT_IMPL_H_ |
| 7 | 7 |
| 8 #include <stdint.h> | 8 #include <stdint.h> |
| 9 #include <vector> | 9 #include <vector> |
| 10 #include "base/callback_forward.h" | 10 #include "base/callback_forward.h" |
| 11 #include "base/memory/ref_counted.h" | 11 #include "base/memory/ref_counted.h" |
| 12 #include "base/memory/weak_ptr.h" | 12 #include "base/memory/weak_ptr.h" |
| 13 #include "content/common/content_export.h" | 13 #include "content/common/content_export.h" |
| 14 #include "content/common/url_loader.mojom.h" | 14 #include "content/common/url_loader.mojom.h" |
| 15 #include "ipc/ipc_message.h" | 15 #include "ipc/ipc_message.h" |
| 16 #include "mojo/public/cpp/bindings/associated_binding.h" | 16 #include "mojo/public/cpp/bindings/binding.h" |
| 17 #include "mojo/public/cpp/system/data_pipe.h" | 17 #include "mojo/public/cpp/system/data_pipe.h" |
| 18 | 18 |
| 19 namespace base { | 19 namespace base { |
| 20 class SingleThreadTaskRunner; | 20 class SingleThreadTaskRunner; |
| 21 } // namespace base | 21 } // namespace base |
| 22 | 22 |
| 23 namespace net { | 23 namespace net { |
| 24 struct RedirectInfo; | 24 struct RedirectInfo; |
| 25 } // namespace net | 25 } // namespace net |
| 26 | 26 |
| 27 namespace content { | 27 namespace content { |
| 28 class ResourceDispatcher; | 28 class ResourceDispatcher; |
| 29 class URLResponseBodyConsumer; | 29 class URLResponseBodyConsumer; |
| 30 struct ResourceRequestCompletionStatus; | 30 struct ResourceRequestCompletionStatus; |
| 31 struct ResourceResponseHead; | 31 struct ResourceResponseHead; |
| 32 | 32 |
| 33 class CONTENT_EXPORT URLLoaderClientImpl final : public mojom::URLLoaderClient { | 33 class CONTENT_EXPORT URLLoaderClientImpl final : public mojom::URLLoaderClient { |
| 34 public: | 34 public: |
| 35 URLLoaderClientImpl(int request_id, | 35 URLLoaderClientImpl(int request_id, |
| 36 ResourceDispatcher* resource_dispatcher, | 36 ResourceDispatcher* resource_dispatcher, |
| 37 scoped_refptr<base::SingleThreadTaskRunner> task_runner); | 37 scoped_refptr<base::SingleThreadTaskRunner> task_runner); |
| 38 ~URLLoaderClientImpl() override; | 38 ~URLLoaderClientImpl() override; |
| 39 | 39 |
| 40 void Bind(mojom::URLLoaderClientAssociatedPtrInfo* client_ptr_info); | 40 void Bind(mojom::URLLoaderClientPtr* client_ptr); |
| 41 | 41 |
| 42 // Sets |is_deferred_|. From now, the received messages are not dispatched | 42 // Sets |is_deferred_|. From now, the received messages are not dispatched |
| 43 // to clients until UnsetDefersLoading is called. | 43 // to clients until UnsetDefersLoading is called. |
| 44 void SetDefersLoading(); | 44 void SetDefersLoading(); |
| 45 | 45 |
| 46 // Unsets |is_deferred_|. | 46 // Unsets |is_deferred_|. |
| 47 void UnsetDefersLoading(); | 47 void UnsetDefersLoading(); |
| 48 | 48 |
| 49 // Disaptches the messages received after SetDefersLoading is called. | 49 // Disaptches the messages received after SetDefersLoading is called. |
| 50 void FlushDeferredMessages(); | 50 void FlushDeferredMessages(); |
| 51 | 51 |
| 52 // mojom::URLLoaderClient implementation | 52 // mojom::URLLoaderClient implementation |
| 53 void OnReceiveResponse( | 53 void OnReceiveResponse(const ResourceResponseHead& response_head, |
| 54 const ResourceResponseHead& response_head, | 54 mojom::DownloadedTempFilePtr downloaded_file) override; |
| 55 mojom::DownloadedTempFileAssociatedPtrInfo downloaded_file) override; | |
| 56 void OnReceiveRedirect(const net::RedirectInfo& redirect_info, | 55 void OnReceiveRedirect(const net::RedirectInfo& redirect_info, |
| 57 const ResourceResponseHead& response_head) override; | 56 const ResourceResponseHead& response_head) override; |
| 58 void OnDataDownloaded(int64_t data_len, int64_t encoded_data_len) override; | 57 void OnDataDownloaded(int64_t data_len, int64_t encoded_data_len) override; |
| 59 void OnUploadProgress(int64_t current_position, | 58 void OnUploadProgress(int64_t current_position, |
| 60 int64_t total_size, | 59 int64_t total_size, |
| 61 const base::Closure& ack_callback) override; | 60 const base::Closure& ack_callback) override; |
| 62 void OnReceiveCachedMetadata(const std::vector<uint8_t>& data) override; | 61 void OnReceiveCachedMetadata(const std::vector<uint8_t>& data) override; |
| 63 void OnTransferSizeUpdated(int32_t transfer_size_diff) override; | 62 void OnTransferSizeUpdated(int32_t transfer_size_diff) override; |
| 64 void OnStartLoadingResponseBody( | 63 void OnStartLoadingResponseBody( |
| 65 mojo::ScopedDataPipeConsumerHandle body) override; | 64 mojo::ScopedDataPipeConsumerHandle body) override; |
| 66 void OnComplete(const ResourceRequestCompletionStatus& status) override; | 65 void OnComplete(const ResourceRequestCompletionStatus& status) override; |
| 67 | 66 |
| 68 private: | 67 private: |
| 69 void Dispatch(const IPC::Message& message); | 68 void Dispatch(const IPC::Message& message); |
| 70 | 69 |
| 71 mojo::AssociatedBinding<mojom::URLLoaderClient> binding_; | 70 mojo::Binding<mojom::URLLoaderClient> binding_; |
| 72 scoped_refptr<URLResponseBodyConsumer> body_consumer_; | 71 scoped_refptr<URLResponseBodyConsumer> body_consumer_; |
| 73 mojom::DownloadedTempFileAssociatedPtr downloaded_file_; | 72 mojom::DownloadedTempFilePtr downloaded_file_; |
| 74 std::vector<IPC::Message> deferred_messages_; | 73 std::vector<IPC::Message> deferred_messages_; |
| 75 const int request_id_; | 74 const int request_id_; |
| 76 bool has_received_response_ = false; | 75 bool has_received_response_ = false; |
| 77 bool is_deferred_ = false; | 76 bool is_deferred_ = false; |
| 78 int32_t accumulated_transfer_size_diff_during_deferred_ = 0; | 77 int32_t accumulated_transfer_size_diff_during_deferred_ = 0; |
| 79 ResourceDispatcher* const resource_dispatcher_; | 78 ResourceDispatcher* const resource_dispatcher_; |
| 80 scoped_refptr<base::SingleThreadTaskRunner> task_runner_; | 79 scoped_refptr<base::SingleThreadTaskRunner> task_runner_; |
| 81 base::WeakPtrFactory<URLLoaderClientImpl> weak_factory_; | 80 base::WeakPtrFactory<URLLoaderClientImpl> weak_factory_; |
| 82 }; | 81 }; |
| 83 | 82 |
| 84 } // namespace content | 83 } // namespace content |
| 85 | 84 |
| 86 #endif // CONTENT_CHILD_URL_LOADER_CLIENT_IMPL_H_ | 85 #endif // CONTENT_CHILD_URL_LOADER_CLIENT_IMPL_H_ |
| OLD | NEW |