1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef REMOTING_BASE_BUFFERED_SOCKET_WRITER_H_
#define REMOTING_BASE_BUFFERED_SOCKET_WRITER_H_
#include <list>
#include <memory>
#include "base/functional/callback.h"
#include "base/memory/weak_ptr.h"
#include "base/synchronization/lock.h"
#include "base/threading/thread_checker.h"
#include "net/base/completion_once_callback.h"
#include "net/base/io_buffer.h"
#include "net/traffic_annotation/network_traffic_annotation.h"
namespace net {
class Socket;
} // namespace net
namespace remoting {
// BufferedSocketWriter implement write data queue for stream sockets.
class BufferedSocketWriter {
public:
typedef base::RepeatingCallback<int(
const scoped_refptr<net::IOBuffer>& buf,
int buf_len,
net::CompletionOnceCallback callback,
const net::NetworkTrafficAnnotationTag& traffic_annotation)>
WriteCallback;
typedef base::OnceCallback<void(int)> WriteFailedCallback;
static std::unique_ptr<BufferedSocketWriter> CreateForSocket(
net::Socket* socket,
WriteFailedCallback write_failed_callback);
BufferedSocketWriter();
virtual ~BufferedSocketWriter();
// Starts the writer. |write_callback| is called to write data to the
// socket. |write_failed_callback| is called when write operation fails.
// Writing stops after the first failed write.
void Start(const WriteCallback& write_callback,
WriteFailedCallback write_failed_callback);
// Puts a new data chunk in the buffer. If called before Start() then all data
// is buffered until Start().
void Write(scoped_refptr<net::IOBufferWithSize> buffer,
base::OnceClosure done_task,
const net::NetworkTrafficAnnotationTag& traffic_annotation);
// Returns true when there is data waiting to be written.
bool has_data_pending() { return !queue_.empty(); }
private:
struct PendingPacket;
void DoWrite();
void HandleWriteResult(int result);
void OnWritten(int result);
base::ThreadChecker thread_checker_;
WriteCallback write_callback_;
WriteFailedCallback write_failed_callback_;
bool closed_ = false;
std::list<std::unique_ptr<PendingPacket>> queue_;
bool write_pending_ = false;
base::WeakPtrFactory<BufferedSocketWriter> weak_factory_{this};
};
} // namespace remoting
#endif // REMOTING_BASE_BUFFERED_SOCKET_WRITER_H_
|
