OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 "blimp/net/stream_packet_writer.h" | 5 #include "blimp/net/stream_packet_writer.h" |
6 | 6 |
7 #include <iostream> | 7 #include <iostream> |
8 | 8 |
9 #include "base/callback_helpers.h" | 9 #include "base/callback_helpers.h" |
10 #include "base/logging.h" | 10 #include "base/logging.h" |
11 #include "base/memory/ref_counted.h" | 11 #include "base/memory/ref_counted.h" |
12 #include "base/message_loop/message_loop.h" | 12 #include "base/message_loop/message_loop.h" |
13 #include "base/sys_byteorder.h" | 13 #include "base/sys_byteorder.h" |
14 #include "blimp/common/proto/blimp_message.pb.h" | 14 #include "blimp/common/proto/blimp_message.pb.h" |
15 #include "blimp/net/blimp_connection_statistics.h" | |
16 #include "blimp/net/common.h" | 15 #include "blimp/net/common.h" |
17 #include "net/base/io_buffer.h" | 16 #include "net/base/io_buffer.h" |
18 #include "net/base/net_errors.h" | 17 #include "net/base/net_errors.h" |
19 #include "net/socket/stream_socket.h" | 18 #include "net/socket/stream_socket.h" |
20 | 19 |
21 namespace blimp { | 20 namespace blimp { |
22 | 21 |
23 std::ostream& operator<<(std::ostream& out, | 22 std::ostream& operator<<(std::ostream& out, |
24 const StreamPacketWriter::WriteState state) { | 23 const StreamPacketWriter::WriteState state) { |
25 switch (state) { | 24 switch (state) { |
26 case StreamPacketWriter::WriteState::IDLE: | 25 case StreamPacketWriter::WriteState::IDLE: |
27 out << "IDLE"; | 26 out << "IDLE"; |
28 break; | 27 break; |
29 case StreamPacketWriter::WriteState::HEADER: | 28 case StreamPacketWriter::WriteState::HEADER: |
30 out << "HEADER"; | 29 out << "HEADER"; |
31 break; | 30 break; |
32 case StreamPacketWriter::WriteState::PAYLOAD: | 31 case StreamPacketWriter::WriteState::PAYLOAD: |
33 out << "PAYLOAD"; | 32 out << "PAYLOAD"; |
34 break; | 33 break; |
35 } | 34 } |
36 return out; | 35 return out; |
37 } | 36 } |
38 | 37 |
39 StreamPacketWriter::StreamPacketWriter(net::StreamSocket* socket, | 38 StreamPacketWriter::StreamPacketWriter(net::StreamSocket* socket) |
40 BlimpConnectionStatistics* statistics) | |
41 : write_state_(WriteState::IDLE), | 39 : write_state_(WriteState::IDLE), |
42 socket_(socket), | 40 socket_(socket), |
43 header_buffer_( | 41 header_buffer_( |
44 new net::DrainableIOBuffer(new net::IOBuffer(kPacketHeaderSizeBytes), | 42 new net::DrainableIOBuffer(new net::IOBuffer(kPacketHeaderSizeBytes), |
45 kPacketHeaderSizeBytes)), | 43 kPacketHeaderSizeBytes)), |
46 statistics_(statistics), | |
47 weak_factory_(this) { | 44 weak_factory_(this) { |
48 DCHECK(socket_); | 45 DCHECK(socket_); |
49 DCHECK(statistics_); | |
50 } | 46 } |
51 | 47 |
52 StreamPacketWriter::~StreamPacketWriter() {} | 48 StreamPacketWriter::~StreamPacketWriter() {} |
53 | 49 |
54 void StreamPacketWriter::WritePacket( | 50 void StreamPacketWriter::WritePacket( |
55 const scoped_refptr<net::DrainableIOBuffer>& data, | 51 const scoped_refptr<net::DrainableIOBuffer>& data, |
56 const net::CompletionCallback& callback) { | 52 const net::CompletionCallback& callback) { |
57 DCHECK_EQ(WriteState::IDLE, write_state_); | 53 DCHECK_EQ(WriteState::IDLE, write_state_); |
58 DCHECK(data); | 54 DCHECK(data); |
59 CHECK(data->BytesRemaining()); | 55 CHECK(data->BytesRemaining()); |
60 | 56 |
61 write_state_ = WriteState::HEADER; | 57 write_state_ = WriteState::HEADER; |
62 header_buffer_->SetOffset(0); | 58 header_buffer_->SetOffset(0); |
63 *reinterpret_cast<uint32_t*>(header_buffer_->data()) = | 59 *reinterpret_cast<uint32_t*>(header_buffer_->data()) = |
64 base::HostToNet32(data->BytesRemaining()); | 60 base::HostToNet32(data->BytesRemaining()); |
65 payload_buffer_ = data; | 61 payload_buffer_ = data; |
66 | 62 |
67 statistics_->Add(BlimpConnectionStatistics::BYTES_SENT, | |
68 payload_buffer_->BytesRemaining()); | |
69 int result = DoWriteLoop(net::OK); | 63 int result = DoWriteLoop(net::OK); |
70 if (result != net::ERR_IO_PENDING) { | 64 if (result != net::ERR_IO_PENDING) { |
71 // Release the payload buffer, since the write operation has completed | 65 // Release the payload buffer, since the write operation has completed |
72 // synchronously. | 66 // synchronously. |
73 payload_buffer_ = nullptr; | 67 payload_buffer_ = nullptr; |
74 | 68 |
75 // Adapt synchronous completion to an asynchronous style. | 69 // Adapt synchronous completion to an asynchronous style. |
76 base::MessageLoop::current()->PostTask(FROM_HERE, | 70 base::MessageLoop::current()->PostTask(FROM_HERE, |
77 base::Bind(callback, result)); | 71 base::Bind(callback, result)); |
78 } else { | 72 } else { |
(...skipping 69 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
148 | 142 |
149 // If the write finished, either successfully or by error, inform the | 143 // If the write finished, either successfully or by error, inform the |
150 // caller. | 144 // caller. |
151 if (result != net::ERR_IO_PENDING) { | 145 if (result != net::ERR_IO_PENDING) { |
152 payload_buffer_ = nullptr; | 146 payload_buffer_ = nullptr; |
153 base::ResetAndReturn(&callback_).Run(result); | 147 base::ResetAndReturn(&callback_).Run(result); |
154 } | 148 } |
155 } | 149 } |
156 | 150 |
157 } // namespace blimp | 151 } // namespace blimp |
OLD | NEW |