Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(168)

Side by Side Diff: net/tools/quic/quic_simple_server_packet_writer.cc

Issue 1036023002: Move remaining QUIC server files from net/quic/ to net/tools/quic/. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 5 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698