Add IPC for DNS host address resolution.

content/renderer/p2p/host_address_request.cc

Index: content/renderer/p2p/host_address_request.cc
diff --git a/content/renderer/p2p/host_address_request.cc b/content/renderer/p2p/host_address_request.cc
new file mode 100644
index 0000000000000000000000000000000000000000..1db935a594b7be2f93b66c716aff8705e83b08b0
--- /dev/null
+++ b/content/renderer/p2p/host_address_request.cc
@@ -0,0 +1,78 @@
+// Copyright (c) 2011 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_proxy.h"
+#include "content/common/p2p_messages.h"
+#include "content/renderer/p2p/socket_dispatcher.h"
+
+P2PHostAddressRequest::P2PHostAddressRequest(P2PSocketDispatcher* dispatcher)
+    : dispatcher_(dispatcher),
+      ipc_message_loop_(dispatcher->message_loop()),
+      delegate_message_loop_(base::MessageLoopProxy::CreateForCurrentThread()),
+      state_(STATE_CREATED),
+      request_id_(0),
+      registered_(false) {
+}
+
+P2PHostAddressRequest::~P2PHostAddressRequest() {

[Wez, 2011/08/08 21:47:18]

nit: Check that you're being torn-down on the right thread?

[Sergey Ulanov, 2011/08/09 00:41:35]

Because this object is refcounted, there are not guarantees about thread that
destroys this object.

+  DCHECK(state_ == STATE_CREATED || state_ == STATE_FINISHED);
+}
+
+void P2PHostAddressRequest::Request(const std::string& host_name,
+                                    const DoneCallback& done_callback) {
+  DCHECK(delegate_message_loop_->BelongsToCurrentThread());
+
+  state_ = STATE_SENT;
+  ipc_message_loop_->PostTask(FROM_HERE, base::Bind(
+      &P2PHostAddressRequest::DoSendRequest, this, host_name, done_callback));

[Wez, 2011/08/08 21:47:18]

If the IPC message-loop has been torn down, this callback will never arrive, so
this request will stay in the SENT state and we'll DCHECK() when we tear it
down?

[Sergey Ulanov, 2011/08/09 00:41:35]

The DCHECK in the destructor is supposed to check that ether the request was
finished (callback called) or Cancel() was called. In the case when IPC thread
is destroyed, the request may never be sent, so the code that uses this class
must call Cancel() before dropping reference to this object.

+}
+
+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(0, 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());
+  delegate_message_loop_->PostTask(FROM_HERE, base::Bind(
+      &P2PHostAddressRequest::DeliverResponse, this, address));
+  dispatcher_->UnregisterHostAddressRequest(request_id_);
+  registered_ = false;
+}
+
+void P2PHostAddressRequest::DeliverResponse(
+    const net::IPAddressNumber& address) {
+  DCHECK(delegate_message_loop_->BelongsToCurrentThread());
+  if (state_ == STATE_SENT) {
+    done_callback_.Run(address);
+    state_ = STATE_FINISHED;
+  }
+}