| // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "content/renderer/p2p/host_address_request.h" |
| |
| #include "base/bind.h" |
| #include "base/message_loop/message_loop_proxy.h" |
| #include "content/common/p2p_messages.h" |
| #include "content/renderer/p2p/socket_dispatcher.h" |
| |
| namespace content { |
| |
| P2PHostAddressRequest::P2PHostAddressRequest(P2PSocketDispatcher* dispatcher) |
| : dispatcher_(dispatcher), |
| ipc_message_loop_(dispatcher->message_loop()), |
| delegate_message_loop_(base::MessageLoopProxy::current()), |
| state_(STATE_CREATED), |
| request_id_(0), |
| registered_(false) { |
| } |
| |
| P2PHostAddressRequest::~P2PHostAddressRequest() { |
| DCHECK(state_ == STATE_CREATED || state_ == STATE_FINISHED); |
| DCHECK(!registered_); |
| } |
| |
| void P2PHostAddressRequest::Request(const std::string& host_name, |
| const DoneCallback& done_callback) { |
| DCHECK(delegate_message_loop_->BelongsToCurrentThread()); |
| DCHECK_EQ(STATE_CREATED, state_); |
| |
| state_ = STATE_SENT; |
| ipc_message_loop_->PostTask(FROM_HERE, base::Bind( |
| &P2PHostAddressRequest::DoSendRequest, this, host_name, done_callback)); |
| } |
| |
| void P2PHostAddressRequest::Cancel() { |
| DCHECK(delegate_message_loop_->BelongsToCurrentThread()); |
| |
| if (state_ != STATE_FINISHED) { |
| state_ = STATE_FINISHED; |
| ipc_message_loop_->PostTask(FROM_HERE, base::Bind( |
| &P2PHostAddressRequest::DoUnregister, this)); |
| } |
| } |
| |
| void P2PHostAddressRequest::DoSendRequest(const std::string& host_name, |
| const DoneCallback& done_callback) { |
| DCHECK(ipc_message_loop_->BelongsToCurrentThread()); |
| |
| done_callback_ = done_callback; |
| request_id_ = dispatcher_->RegisterHostAddressRequest(this); |
| registered_ = true; |
| dispatcher_->SendP2PMessage( |
| new P2PHostMsg_GetHostAddress(host_name, request_id_)); |
| } |
| |
| void P2PHostAddressRequest::DoUnregister() { |
| DCHECK(ipc_message_loop_->BelongsToCurrentThread()); |
| if (registered_) { |
| dispatcher_->UnregisterHostAddressRequest(request_id_); |
| registered_ = false; |
| } |
| } |
| |
| void P2PHostAddressRequest::OnResponse(const net::IPAddressNumber& address) { |
| DCHECK(ipc_message_loop_->BelongsToCurrentThread()); |
| DCHECK(registered_); |
| |
| dispatcher_->UnregisterHostAddressRequest(request_id_); |
| registered_ = false; |
| |
| delegate_message_loop_->PostTask(FROM_HERE, base::Bind( |
| &P2PHostAddressRequest::DeliverResponse, this, address)); |
| } |
| |
| void P2PHostAddressRequest::DeliverResponse( |
| const net::IPAddressNumber& address) { |
| DCHECK(delegate_message_loop_->BelongsToCurrentThread()); |
| if (state_ == STATE_SENT) { |
| done_callback_.Run(address); |
| state_ = STATE_FINISHED; |
| } |
| } |
| |
| } // namespace content |