| 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/blimp_connection.h" | 5 #include "blimp/net/blimp_connection.h" |
| 6 | 6 |
| 7 #include "base/callback_helpers.h" | 7 #include "base/callback_helpers.h" |
| 8 #include "base/logging.h" | 8 #include "base/logging.h" |
| 9 #include "base/macros.h" | 9 #include "base/macros.h" |
| 10 #include "base/message_loop/message_loop.h" | 10 #include "base/message_loop/message_loop.h" |
| (...skipping 22 matching lines...) Expand all Loading... |
| 33 // BlimpMessageProcessor implementation. | 33 // BlimpMessageProcessor implementation. |
| 34 void ProcessMessage(scoped_ptr<BlimpMessage> message, | 34 void ProcessMessage(scoped_ptr<BlimpMessage> message, |
| 35 const net::CompletionCallback& callback) override; | 35 const net::CompletionCallback& callback) override; |
| 36 | 36 |
| 37 private: | 37 private: |
| 38 void OnWritePacketComplete(int result); | 38 void OnWritePacketComplete(int result); |
| 39 | 39 |
| 40 PacketWriter* writer_; | 40 PacketWriter* writer_; |
| 41 ConnectionErrorObserver* error_observer_; | 41 ConnectionErrorObserver* error_observer_; |
| 42 scoped_refptr<net::DrainableIOBuffer> buffer_; | 42 scoped_refptr<net::DrainableIOBuffer> buffer_; |
| 43 net::CancelableCompletionCallback write_packet_callback_; | |
| 44 net::CompletionCallback pending_process_msg_callback_; | 43 net::CompletionCallback pending_process_msg_callback_; |
| 45 | 44 |
| 46 DISALLOW_COPY_AND_ASSIGN(BlimpMessageSender); | 45 DISALLOW_COPY_AND_ASSIGN(BlimpMessageSender); |
| 47 }; | 46 }; |
| 48 | 47 |
| 49 BlimpMessageSender::BlimpMessageSender(PacketWriter* writer) | 48 BlimpMessageSender::BlimpMessageSender(PacketWriter* writer) |
| 50 : writer_(writer), | 49 : writer_(writer), |
| 51 buffer_(new net::DrainableIOBuffer( | 50 buffer_(new net::DrainableIOBuffer( |
| 52 new net::IOBuffer(kMaxPacketPayloadSizeBytes), | 51 new net::IOBuffer(kMaxPacketPayloadSizeBytes), |
| 53 kMaxPacketPayloadSizeBytes)) { | 52 kMaxPacketPayloadSizeBytes)) { |
| (...skipping 11 matching lines...) Expand all Loading... |
| 65 return; | 64 return; |
| 66 } | 65 } |
| 67 | 66 |
| 68 buffer_->SetOffset(0); | 67 buffer_->SetOffset(0); |
| 69 if (!message->SerializeToArray(buffer_->data(), message->ByteSize())) { | 68 if (!message->SerializeToArray(buffer_->data(), message->ByteSize())) { |
| 70 DLOG(ERROR) << "Failed to serialize message."; | 69 DLOG(ERROR) << "Failed to serialize message."; |
| 71 callback.Run(net::ERR_INVALID_ARGUMENT); | 70 callback.Run(net::ERR_INVALID_ARGUMENT); |
| 72 return; | 71 return; |
| 73 } | 72 } |
| 74 | 73 |
| 75 write_packet_callback_.Reset(base::Bind( | |
| 76 &BlimpMessageSender::OnWritePacketComplete, base::Unretained(this))); | |
| 77 pending_process_msg_callback_ = callback; | 74 pending_process_msg_callback_ = callback; |
| 78 int result = writer_->WritePacket(buffer_, write_packet_callback_.callback()); | 75 writer_->WritePacket(buffer_, |
| 79 if (result != net::ERR_IO_PENDING) { | 76 base::Bind(&BlimpMessageSender::OnWritePacketComplete, |
| 80 base::MessageLoop::current()->PostTask( | 77 base::Unretained(this))); |
| 81 FROM_HERE, base::Bind(write_packet_callback_.callback(), result)); | |
| 82 } | |
| 83 } | 78 } |
| 84 | 79 |
| 85 void BlimpMessageSender::OnWritePacketComplete(int result) { | 80 void BlimpMessageSender::OnWritePacketComplete(int result) { |
| 86 DCHECK_NE(net::ERR_IO_PENDING, result); | 81 DCHECK_NE(net::ERR_IO_PENDING, result); |
| 87 write_packet_callback_.Cancel(); | |
| 88 base::ResetAndReturn(&pending_process_msg_callback_).Run(result); | 82 base::ResetAndReturn(&pending_process_msg_callback_).Run(result); |
| 89 if (result != net::OK) { | 83 if (result != net::OK) { |
| 90 error_observer_->OnConnectionError(result); | 84 error_observer_->OnConnectionError(result); |
| 91 } | 85 } |
| 92 } | 86 } |
| 93 | 87 |
| 94 } // namespace | 88 } // namespace |
| 95 | 89 |
| 96 BlimpConnection::BlimpConnection(scoped_ptr<PacketReader> reader, | 90 BlimpConnection::BlimpConnection(scoped_ptr<PacketReader> reader, |
| 97 scoped_ptr<PacketWriter> writer) | 91 scoped_ptr<PacketWriter> writer) |
| (...skipping 17 matching lines...) Expand all Loading... |
| 115 void BlimpConnection::SetIncomingMessageProcessor( | 109 void BlimpConnection::SetIncomingMessageProcessor( |
| 116 BlimpMessageProcessor* processor) { | 110 BlimpMessageProcessor* processor) { |
| 117 message_pump_->SetMessageProcessor(processor); | 111 message_pump_->SetMessageProcessor(processor); |
| 118 } | 112 } |
| 119 | 113 |
| 120 BlimpMessageProcessor* BlimpConnection::GetOutgoingMessageProcessor() const { | 114 BlimpMessageProcessor* BlimpConnection::GetOutgoingMessageProcessor() const { |
| 121 return outgoing_msg_processor_.get(); | 115 return outgoing_msg_processor_.get(); |
| 122 } | 116 } |
| 123 | 117 |
| 124 } // namespace blimp | 118 } // namespace blimp |
| OLD | NEW |