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 #include "net/quic/quic_server_packet_writer.h" | 5 #include "net/tools/quic/quic_simple_server_packet_writer.h" |
6 | 6 |
7 #include "base/callback_helpers.h" | 7 #include "base/callback_helpers.h" |
8 #include "base/location.h" | 8 #include "base/location.h" |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "base/metrics/sparse_histogram.h" | 10 #include "base/metrics/sparse_histogram.h" |
11 #include "net/base/io_buffer.h" | 11 #include "net/base/io_buffer.h" |
12 #include "net/base/net_errors.h" | 12 #include "net/base/net_errors.h" |
13 #include "net/udp/udp_server_socket.h" | 13 #include "net/udp/udp_server_socket.h" |
14 | 14 |
15 namespace net { | 15 namespace net { |
| 16 namespace tools { |
16 | 17 |
17 QuicServerPacketWriter::QuicServerPacketWriter( | 18 QuicSimpleServerPacketWriter::QuicSimpleServerPacketWriter( |
18 UDPServerSocket* socket, | 19 UDPServerSocket* socket, |
19 QuicBlockedWriterInterface* blocked_writer) | 20 QuicBlockedWriterInterface* blocked_writer) |
20 : socket_(socket), | 21 : socket_(socket), |
21 blocked_writer_(blocked_writer), | 22 blocked_writer_(blocked_writer), |
22 write_blocked_(false), | 23 write_blocked_(false), |
23 weak_factory_(this) { | 24 weak_factory_(this) { |
24 } | 25 } |
25 | 26 |
26 QuicServerPacketWriter::~QuicServerPacketWriter() { | 27 QuicSimpleServerPacketWriter::~QuicSimpleServerPacketWriter() { |
27 } | 28 } |
28 | 29 |
29 WriteResult QuicServerPacketWriter::WritePacketWithCallback( | 30 WriteResult QuicSimpleServerPacketWriter::WritePacketWithCallback( |
30 const char* buffer, | 31 const char* buffer, |
31 size_t buf_len, | 32 size_t buf_len, |
32 const IPAddressNumber& self_address, | 33 const IPAddressNumber& self_address, |
33 const IPEndPoint& peer_address, | 34 const IPEndPoint& peer_address, |
34 WriteCallback callback) { | 35 WriteCallback callback) { |
35 DCHECK(callback_.is_null()); | 36 DCHECK(callback_.is_null()); |
36 callback_ = callback; | 37 callback_ = callback; |
37 WriteResult result = WritePacket(buffer, buf_len, self_address, peer_address); | 38 WriteResult result = WritePacket(buffer, buf_len, self_address, peer_address); |
38 if (result.status != WRITE_STATUS_BLOCKED) { | 39 if (result.status != WRITE_STATUS_BLOCKED) { |
39 callback_.Reset(); | 40 callback_.Reset(); |
40 } | 41 } |
41 return result; | 42 return result; |
42 } | 43 } |
43 | 44 |
44 void QuicServerPacketWriter::OnWriteComplete(int rv) { | 45 void QuicSimpleServerPacketWriter::OnWriteComplete(int rv) { |
45 DCHECK_NE(rv, ERR_IO_PENDING); | 46 DCHECK_NE(rv, ERR_IO_PENDING); |
46 write_blocked_ = false; | 47 write_blocked_ = false; |
47 WriteResult result(rv < 0 ? WRITE_STATUS_ERROR : WRITE_STATUS_OK, rv); | 48 WriteResult result(rv < 0 ? WRITE_STATUS_ERROR : WRITE_STATUS_OK, rv); |
48 base::ResetAndReturn(&callback_).Run(result); | 49 base::ResetAndReturn(&callback_).Run(result); |
49 blocked_writer_->OnCanWrite(); | 50 blocked_writer_->OnCanWrite(); |
50 } | 51 } |
51 | 52 |
52 bool QuicServerPacketWriter::IsWriteBlockedDataBuffered() const { | 53 bool QuicSimpleServerPacketWriter::IsWriteBlockedDataBuffered() const { |
53 // UDPServerSocket::SendTo buffers the data until the Write is permitted. | 54 // UDPServerSocket::SendTo buffers the data until the Write is permitted. |
54 return true; | 55 return true; |
55 } | 56 } |
56 | 57 |
57 bool QuicServerPacketWriter::IsWriteBlocked() const { | 58 bool QuicSimpleServerPacketWriter::IsWriteBlocked() const { |
58 return write_blocked_; | 59 return write_blocked_; |
59 } | 60 } |
60 | 61 |
61 void QuicServerPacketWriter::SetWritable() { | 62 void QuicSimpleServerPacketWriter::SetWritable() { |
62 write_blocked_ = false; | 63 write_blocked_ = false; |
63 } | 64 } |
64 | 65 |
65 WriteResult QuicServerPacketWriter::WritePacket( | 66 WriteResult QuicSimpleServerPacketWriter::WritePacket( |
66 const char* buffer, | 67 const char* buffer, |
67 size_t buf_len, | 68 size_t buf_len, |
68 const IPAddressNumber& self_address, | 69 const IPAddressNumber& self_address, |
69 const IPEndPoint& peer_address) { | 70 const IPEndPoint& peer_address) { |
70 scoped_refptr<StringIOBuffer> buf( | 71 scoped_refptr<StringIOBuffer> buf( |
71 new StringIOBuffer(std::string(buffer, buf_len))); | 72 new StringIOBuffer(std::string(buffer, buf_len))); |
72 DCHECK(!IsWriteBlocked()); | 73 DCHECK(!IsWriteBlocked()); |
73 DCHECK(!callback_.is_null()); | 74 DCHECK(!callback_.is_null()); |
74 int rv; | 75 int rv; |
75 if (buf_len <= static_cast<size_t>(std::numeric_limits<int>::max())) { | 76 if (buf_len <= static_cast<size_t>(std::numeric_limits<int>::max())) { |
76 rv = socket_->SendTo(buf.get(), | 77 rv = socket_->SendTo( |
77 static_cast<int>(buf_len), | 78 buf.get(), |
78 peer_address, | 79 static_cast<int>(buf_len), |
79 base::Bind(&QuicServerPacketWriter::OnWriteComplete, | 80 peer_address, |
80 weak_factory_.GetWeakPtr())); | 81 base::Bind( |
| 82 &QuicSimpleServerPacketWriter::OnWriteComplete, |
| 83 weak_factory_.GetWeakPtr())); |
81 } else { | 84 } else { |
82 rv = ERR_MSG_TOO_BIG; | 85 rv = ERR_MSG_TOO_BIG; |
83 } | 86 } |
84 WriteStatus status = WRITE_STATUS_OK; | 87 WriteStatus status = WRITE_STATUS_OK; |
85 if (rv < 0) { | 88 if (rv < 0) { |
86 if (rv != ERR_IO_PENDING) { | 89 if (rv != ERR_IO_PENDING) { |
87 UMA_HISTOGRAM_SPARSE_SLOWLY("Net.QuicSession.WriteError", -rv); | 90 UMA_HISTOGRAM_SPARSE_SLOWLY("Net.QuicSession.WriteError", -rv); |
88 status = WRITE_STATUS_ERROR; | 91 status = WRITE_STATUS_ERROR; |
89 } else { | 92 } else { |
90 status = WRITE_STATUS_BLOCKED; | 93 status = WRITE_STATUS_BLOCKED; |
91 write_blocked_ = true; | 94 write_blocked_ = true; |
92 } | 95 } |
93 } | 96 } |
94 return WriteResult(status, rv); | 97 return WriteResult(status, rv); |
95 } | 98 } |
96 | 99 |
| 100 } // namespace tools |
97 } // namespace net | 101 } // namespace net |
OLD | NEW |