OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 DEVICE_SERIAL_DATA_SENDER_H_ | 5 #ifndef DEVICE_SERIAL_DATA_SENDER_H_ |
6 #define DEVICE_SERIAL_DATA_SENDER_H_ | 6 #define DEVICE_SERIAL_DATA_SENDER_H_ |
7 | 7 |
8 #include <queue> | 8 #include <queue> |
9 | 9 |
10 #include "base/callback.h" | 10 #include "base/callback.h" |
11 #include "base/memory/linked_ptr.h" | 11 #include "base/memory/linked_ptr.h" |
12 #include "base/strings/string_piece.h" | 12 #include "base/strings/string_piece.h" |
13 #include "device/serial/buffer.h" | 13 #include "device/serial/buffer.h" |
14 #include "device/serial/data_stream.mojom.h" | 14 #include "device/serial/data_stream.mojom.h" |
15 #include "third_party/mojo/src/mojo/public/cpp/system/data_pipe.h" | 15 #include "third_party/mojo/src/mojo/public/cpp/system/data_pipe.h" |
16 | 16 |
17 namespace device { | 17 namespace device { |
18 | 18 |
19 // A DataSender sends data to a DataSink. | 19 // A DataSender sends data to a DataSink. |
20 class DataSender : public serial::DataSinkClient, public mojo::ErrorHandler { | 20 class DataSender : public mojo::ErrorHandler { |
21 public: | 21 public: |
22 typedef base::Callback<void(uint32_t bytes_sent)> DataSentCallback; | 22 typedef base::Callback<void(uint32_t bytes_sent)> DataSentCallback; |
23 typedef base::Callback<void(uint32_t bytes_sent, int32_t error)> | 23 typedef base::Callback<void(uint32_t bytes_sent, int32_t error)> |
24 SendErrorCallback; | 24 SendErrorCallback; |
25 typedef base::Callback<void()> CancelCallback; | 25 typedef base::Callback<void()> CancelCallback; |
26 | 26 |
27 // Constructs a DataSender to send data to |sink|, using a buffer size of | 27 // Constructs a DataSender to send data to |sink|, using a buffer size of |
28 // |buffer_size|, with connection errors reported as |fatal_error_value|. | 28 // |buffer_size|, with connection errors reported as |fatal_error_value|. |
29 DataSender(mojo::InterfacePtr<serial::DataSink> sink, | 29 DataSender(mojo::InterfacePtr<serial::DataSink> sink, |
30 uint32_t buffer_size, | 30 uint32_t buffer_size, |
(...skipping 11 matching lines...) Expand all Loading... |
42 const DataSentCallback& callback, | 42 const DataSentCallback& callback, |
43 const SendErrorCallback& error_callback); | 43 const SendErrorCallback& error_callback); |
44 | 44 |
45 // Requests the cancellation of any in-progress sends. Calls to Send() will | 45 // Requests the cancellation of any in-progress sends. Calls to Send() will |
46 // fail until |callback| is called. | 46 // fail until |callback| is called. |
47 bool Cancel(int32_t error, const CancelCallback& callback); | 47 bool Cancel(int32_t error, const CancelCallback& callback); |
48 | 48 |
49 private: | 49 private: |
50 class PendingSend; | 50 class PendingSend; |
51 | 51 |
52 // serial::DataSinkClient overrides. | 52 // Invoked when a PendingSend completes. |
53 void ReportBytesSent(uint32_t bytes_sent) override; | 53 void SendComplete(); |
54 void ReportBytesSentAndError(uint32_t bytes_sent, | 54 |
55 int32_t error, | 55 // Invoked when a PendingSend fails with |error|. |
56 const mojo::Callback<void()>& callback) override; | 56 void SendFailed(int32_t error); |
57 | 57 |
58 // mojo::ErrorHandler override. | 58 // mojo::ErrorHandler override. |
59 void OnConnectionError() override; | 59 void OnConnectionError() override; |
60 | 60 |
61 // Sends up to |available_buffer_capacity_| bytes of data from | |
62 // |pending_sends_| to |sink_|. When a PendingSend in |pending_sends_| has | |
63 // been fully copied transmitted to |sink_|, it moves to | |
64 // |sends_awaiting_ack_|. | |
65 void SendInternal(); | |
66 | |
67 // Dispatches a cancel callback if one is pending. | 61 // Dispatches a cancel callback if one is pending. |
68 void RunCancelCallback(); | 62 void RunCancelCallback(); |
69 | 63 |
70 // Shuts down this DataSender and dispatches fatal errors to all pending | 64 // Shuts down this DataSender and dispatches fatal errors to all pending |
71 // operations. | 65 // operations. |
72 void ShutDown(); | 66 void ShutDown(); |
73 | 67 |
74 // The control connection to the data sink. | 68 // The control connection to the data sink. |
75 mojo::InterfacePtr<serial::DataSink> sink_; | 69 mojo::InterfacePtr<serial::DataSink> sink_; |
76 | 70 |
77 // The error value to report in the event of a fatal error. | 71 // The error value to report in the event of a fatal error. |
78 const int32_t fatal_error_value_; | 72 const int32_t fatal_error_value_; |
79 | 73 |
80 // A queue of PendingSend that have not yet been fully sent to |sink_|. | |
81 std::queue<linked_ptr<PendingSend> > pending_sends_; | |
82 | |
83 // A queue of PendingSend that have been sent to |sink_|, but have not yet | 74 // A queue of PendingSend that have been sent to |sink_|, but have not yet |
84 // been acked by the DataSink. | 75 // been acked by the DataSink. |
85 std::queue<linked_ptr<PendingSend> > sends_awaiting_ack_; | 76 std::queue<linked_ptr<PendingSend>> sends_awaiting_ack_; |
86 | 77 |
87 // The callback to report cancel completion if a cancel operation is in | 78 // The callback to report cancel completion if a cancel operation is in |
88 // progress. | 79 // progress. |
89 CancelCallback pending_cancel_; | 80 CancelCallback pending_cancel_; |
90 | 81 |
91 // The number of bytes available for buffering in the DataSink. | |
92 uint32_t available_buffer_capacity_; | |
93 | |
94 // Whether we have encountered a fatal error and shut down. | 82 // Whether we have encountered a fatal error and shut down. |
95 bool shut_down_; | 83 bool shut_down_; |
96 | 84 |
97 DISALLOW_COPY_AND_ASSIGN(DataSender); | 85 DISALLOW_COPY_AND_ASSIGN(DataSender); |
98 }; | 86 }; |
99 | 87 |
100 } // namespace device | 88 } // namespace device |
101 | 89 |
102 #endif // DEVICE_SERIAL_DATA_SENDER_H_ | 90 #endif // DEVICE_SERIAL_DATA_SENDER_H_ |
OLD | NEW |