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