OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 REMOTING_BASE_BUFFERED_SOCKET_WRITER_H_ | 5 #ifndef REMOTING_BASE_BUFFERED_SOCKET_WRITER_H_ |
6 #define REMOTING_BASE_BUFFERED_SOCKET_WRITER_H_ | 6 #define REMOTING_BASE_BUFFERED_SOCKET_WRITER_H_ |
7 | 7 |
8 #include <list> | 8 #include <list> |
9 | 9 |
10 #include "base/callback.h" | 10 #include "base/callback.h" |
(...skipping 18 matching lines...) Expand all Loading... |
29 WriteCallback; | 29 WriteCallback; |
30 typedef base::Callback<void(int)> WriteFailedCallback; | 30 typedef base::Callback<void(int)> WriteFailedCallback; |
31 | 31 |
32 static scoped_ptr<BufferedSocketWriter> CreateForSocket( | 32 static scoped_ptr<BufferedSocketWriter> CreateForSocket( |
33 net::Socket* socket, | 33 net::Socket* socket, |
34 const WriteFailedCallback& write_failed_callback); | 34 const WriteFailedCallback& write_failed_callback); |
35 | 35 |
36 BufferedSocketWriter(); | 36 BufferedSocketWriter(); |
37 virtual ~BufferedSocketWriter(); | 37 virtual ~BufferedSocketWriter(); |
38 | 38 |
39 // Initializes the writer. |write_callback| is called to write data to the | 39 // Starts the writer. |write_callback| is called to write data to the |
40 // socket. |write_failed_callback| is called when write operation fails. | 40 // socket. |write_failed_callback| is called when write operation fails. |
41 // Writing stops after the first failed write. | 41 // Writing stops after the first failed write. |
42 void Init(const WriteCallback& write_callback, | 42 void Start(const WriteCallback& write_callback, |
43 const WriteFailedCallback& write_failed_callback); | 43 const WriteFailedCallback& write_failed_callback); |
44 | 44 |
45 // Puts a new data chunk in the buffer. | 45 // Puts a new data chunk in the buffer. If called before Start() then all data |
| 46 // is buffered until Start(). |
46 void Write(const scoped_refptr<net::IOBufferWithSize>& buffer, | 47 void Write(const scoped_refptr<net::IOBufferWithSize>& buffer, |
47 const base::Closure& done_task); | 48 const base::Closure& done_task); |
48 | 49 |
49 // Returns true when there is data waiting to be written. | 50 // Returns true when there is data waiting to be written. |
50 bool has_data_pending() { return !queue_.empty(); } | 51 bool has_data_pending() { return !queue_.empty(); } |
51 | 52 |
52 private: | 53 private: |
53 struct PendingPacket; | 54 struct PendingPacket; |
54 typedef std::list<PendingPacket*> DataQueue; | 55 typedef std::list<PendingPacket*> DataQueue; |
55 | 56 |
56 // Returns true if the writer is closed due to an error. | |
57 bool is_closed(); | |
58 | |
59 void DoWrite(); | 57 void DoWrite(); |
60 void HandleWriteResult(int result); | 58 void HandleWriteResult(int result); |
61 void OnWritten(int result); | 59 void OnWritten(int result); |
62 | 60 |
63 base::ThreadChecker thread_checker_; | 61 base::ThreadChecker thread_checker_; |
64 | 62 |
65 WriteCallback write_callback_; | 63 WriteCallback write_callback_; |
66 WriteFailedCallback write_failed_callback_; | 64 WriteFailedCallback write_failed_callback_; |
67 | 65 |
| 66 bool closed_ = false; |
| 67 |
68 DataQueue queue_; | 68 DataQueue queue_; |
69 | 69 |
70 bool write_pending_ = false; | 70 bool write_pending_ = false; |
71 | 71 |
72 base::WeakPtrFactory<BufferedSocketWriter> weak_factory_; | 72 base::WeakPtrFactory<BufferedSocketWriter> weak_factory_; |
73 }; | 73 }; |
74 | 74 |
75 } // namespace remoting | 75 } // namespace remoting |
76 | 76 |
77 #endif // REMOTING_BASE_BUFFERED_SOCKET_WRITER_H_ | 77 #endif // REMOTING_BASE_BUFFERED_SOCKET_WRITER_H_ |
OLD | NEW |