1 // Copyright 2013 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.
5 #include "net/quic/quic_default_packet_writer.h"
7 #include "base/location.h"
8 #include "base/logging.h"
9 #include "base/metrics/sparse_histogram.h"
10 #include "net/base/io_buffer.h"
11 #include "net/base/net_errors.h"
15 QuicDefaultPacketWriter::QuicDefaultPacketWriter() : weak_factory_(this) {
18 QuicDefaultPacketWriter::QuicDefaultPacketWriter(DatagramClientSocket* socket)
20 write_blocked_(false),
24 QuicDefaultPacketWriter::~QuicDefaultPacketWriter() {}
26 WriteResult QuicDefaultPacketWriter::WritePacket(
27 const char* buffer, size_t buf_len,
28 const net::IPAddressNumber& self_address,
29 const net::IPEndPoint& peer_address) {
30 scoped_refptr<StringIOBuffer> buf(
31 new StringIOBuffer(std::string(buffer, buf_len)));
32 DCHECK(!IsWriteBlocked());
33 int rv = socket_->Write(buf.get(),
35 base::Bind(&QuicDefaultPacketWriter::OnWriteComplete,
36 weak_factory_.GetWeakPtr()));
37 WriteStatus status = WRITE_STATUS_OK;
39 if (rv != ERR_IO_PENDING) {
40 UMA_HISTOGRAM_SPARSE_SLOWLY("Net.QuicSession.WriteError", -rv);
41 status = WRITE_STATUS_ERROR;
43 status = WRITE_STATUS_BLOCKED;
44 write_blocked_ = true;
48 return WriteResult(status, rv);
51 bool QuicDefaultPacketWriter::IsWriteBlockedDataBuffered() const {
52 // Chrome sockets' Write() methods buffer the data until the Write is
57 bool QuicDefaultPacketWriter::IsWriteBlocked() const {
58 return write_blocked_;
61 void QuicDefaultPacketWriter::SetWritable() {
62 write_blocked_ = false;
65 void QuicDefaultPacketWriter::OnWriteComplete(int rv) {
66 DCHECK_NE(rv, ERR_IO_PENDING);
67 write_blocked_ = false;
69 connection_->OnWriteError(rv);
71 connection_->OnCanWrite();