| // Copyright 2018 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_BROWSER_LOADER_SOURCE_STREAM_TO_DATA_PIPE_H_ |
| #define CONTENT_BROWSER_LOADER_SOURCE_STREAM_TO_DATA_PIPE_H_ |
| |
| #include "base/callback_forward.h" |
| #include "base/memory/weak_ptr.h" |
| #include "content/common/content_export.h" |
| #include "mojo/public/cpp/system/data_pipe.h" |
| #include "mojo/public/cpp/system/simple_watcher.h" |
| #include "services/network/public/cpp/net_adapters.h" |
| |
| namespace net { |
| class SourceStream; |
| } |
| |
| namespace content { |
| |
| // A convenient adapter class to read out data from net::SourceStream |
| // and write them into a data pipe. |
| class CONTENT_EXPORT SourceStreamToDataPipe { |
| public: |
| // Reads out the data from |source| and write into |dest|. |
| SourceStreamToDataPipe(std::unique_ptr<net::SourceStream> source, |
| mojo::ScopedDataPipeProducerHandle dest, |
| base::OnceCallback<void(int)> completion_callback); |
| ~SourceStreamToDataPipe(); |
| |
| // Start reading the source. |
| void Start(); |
| int64_t TransferredBytes() const { return transferred_bytes_; } |
| |
| private: |
| void ReadMore(); |
| void DidRead(int result); |
| |
| void OnDataPipeWritable(MojoResult result); |
| void OnDataPipeClosed(MojoResult result); |
| void OnComplete(int result); |
| |
| std::unique_ptr<net::SourceStream> source_; |
| mojo::ScopedDataPipeProducerHandle dest_; |
| base::OnceCallback<void(int)> completion_callback_; |
| int64_t transferred_bytes_ = 0; |
| |
| scoped_refptr<network::NetToMojoPendingBuffer> pending_write_; |
| mojo::SimpleWatcher writable_handle_watcher_; |
| |
| base::WeakPtrFactory<SourceStreamToDataPipe> weak_factory_{this}; |
| }; |
| |
| } // namespace content |
| |
| #endif // CONTENT_BROWSER_LOADER_SOURCE_STREAM_TO_DATA_PIPE_H_ |