Add support for default local address in IpcNetworkManager

content/browser/renderer_host/p2p/socket_dispatcher_host.cc

 // 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/browser/renderer_host/p2p/socket_dispatcher_host.h"
 
 #include "base/bind.h"
 #include "base/stl_util.h"
 #include "content/browser/renderer_host/p2p/socket_host.h"
 #include "content/common/p2p_messages.h"
+#include "content/public/browser/browser_thread.h"
 #include "content/public/browser/resource_context.h"
 #include "net/base/address_list.h"
 #include "net/base/completion_callback.h"
 #include "net/base/net_errors.h"
 #include "net/base/network_interfaces.h"
 #include "net/base/sys_addrinfo.h"
 #include "net/dns/single_request_host_resolver.h"
 #include "net/log/net_log.h"
+#include "net/udp/udp_client_socket.h"
 #include "net/url_request/url_request_context_getter.h"
 
 using content::BrowserMessageFilter;
 using content::BrowserThread;
 
 namespace content {
 
+namespace {
+
+// Used by GetDefaultLocalAddress as the target to connect to for getting the
+// default local address. They are public DNS servers on the internet.
+const char kPublicIPv4Host[] = "8.8.8.8";
+const char kPublicIPv6Host[] = "[2001:4860:4860::8888]";
+const int kPublicPort = 53;  // DNS port.
+
+}  // namespace
+
 const size_t kMaximumPacketSize = 32768;
 
 class P2PSocketDispatcherHost::DnsRequest {
  public:
   typedef base::Callback<void(const net::IPAddressList&)> DoneCallback;
 
   DnsRequest(int32 request_id, net::HostResolver* host_resolver)
       : request_id_(request_id),
         resolver_(host_resolver) {
   }
@@ skipping 278 matching lines @@
     delete it->second;
     sockets_.erase(it);
   } else {
     LOG(ERROR) << "Received P2PHostMsg_DestroySocket for invalid socket_id.";
   }
 }
 
 void P2PSocketDispatcherHost::DoGetNetworkList() {
   net::NetworkInterfaceList list;
   net::GetNetworkList(&list, net::EXCLUDE_HOST_SCOPE_VIRTUAL_INTERFACES);
+  default_ipv4_local_address_ = GetDefaultLocalAddress(AF_INET);
+  default_ipv6_local_address_ = GetDefaultLocalAddress(AF_INET6);
   BrowserThread::PostTask(
-      BrowserThread::IO, FROM_HERE, base::Bind(
-          &P2PSocketDispatcherHost::SendNetworkList, this, list));
+      BrowserThread::IO, FROM_HERE,
+      base::Bind(&P2PSocketDispatcherHost::SendNetworkList, this, list,
+                 default_ipv4_local_address_, default_ipv6_local_address_));
 }
 
 void P2PSocketDispatcherHost::SendNetworkList(
-    const net::NetworkInterfaceList& list) {
-  Send(new P2PMsg_NetworkListChanged(list));
+    const net::NetworkInterfaceList& list,
+    const net::IPEndPoint& default_ipv4_local_address,
+    const net::IPEndPoint& default_ipv6_local_address) {
+  Send(new P2PMsg_NetworkListChanged(list, default_ipv4_local_address,
+                                     default_ipv6_local_address));
+}
+
+net::IPEndPoint P2PSocketDispatcherHost::GetDefaultLocalAddress(int family) {
+  DCHECK(family == AF_INET || family == AF_INET6);
+
+  // Please see crbug.com/455942. We want to make sure we're not on these 2

[Sergey Ulanov, 2015/11/10 22:44:17]

This bug reference is confusing here - the bug is about DNS resolution being
janky. Just say in the comments that creation and connection of UDP socket may
be janky. The bug you are referring to is just an example of such jank.

[guoweis_webrtc, 2015/11/11 00:11:51]

Done.

+  // threads.
+  DCHECK(!::content::BrowserThread::CurrentlyOn(BrowserThread::UI));

[Sergey Ulanov, 2015/11/10 22:44:16]

This code is always called on the FILE thread, so maybe just DCHECK it here?

[guoweis_webrtc, 2015/11/11 00:11:51]

Done.

+  DCHECK(!::content::BrowserThread::CurrentlyOn(BrowserThread::IO));
+
+  net::IPEndPoint local_address;

[Sergey Ulanov, 2015/11/10 22:44:16]

nit: Move this just before GetLocalAddress() call below.

[guoweis_webrtc, 2015/11/11 00:11:51]

Done.

+  net::IPAddressNumber ip_address_number;
+
+  scoped_ptr<net::UDPClientSocket> socket(new net::UDPClientSocket(

[Sergey Ulanov, 2015/11/10 22:44:16]

Please use
net::ClientSocketFactory::GetDefaultFactory()->CreateDatagramClientSocket() to
create UDP sockets.

[guoweis_webrtc, 2015/11/11 00:11:51]

Done.

+      net::DatagramSocket::DEFAULT_BIND, net::RandIntCallback(), NULL,
+      net::NetLog::Source()));
+
+  if (!net::ParseURLHostnameToNumber(

[Sergey Ulanov, 2015/11/10 22:44:16]

net::IPAddressNumber is just an std::vector<uint8_t>. So instead of parsing the
strings you can store the target addresses as arrays of integers and initialize
the addresses from these arrays.

[guoweis_webrtc, 2015/11/11 00:11:51]

Done.

+          family == AF_INET ? kPublicIPv4Host : kPublicIPv6Host,
+          &ip_address_number)) {
+    return net::IPEndPoint();

[Sergey Ulanov, 2015/11/10 22:44:16]

this would need to be marked NOTREACHED(), but it doesn't matter if you take my
suggestion above.

[guoweis_webrtc, 2015/11/11 00:11:51]

Done.

+  }
+
+  if (socket->Connect(net::IPEndPoint(ip_address_number, kPublicPort)) !=
+      net::OK) {
+    return net::IPEndPoint();
+  }
+
+  if (socket->GetLocalAddress(&local_address) != net::OK)
+    return net::IPEndPoint();
+
+  return local_address;
 }
 
 void P2PSocketDispatcherHost::OnAddressResolved(
     DnsRequest* request,
     const net::IPAddressList& addresses) {
   Send(new P2PMsg_GetHostAddressResult(request->request_id(), addresses));
 
   dns_requests_.erase(request);
   delete request;
 }
@@ skipping 10 matching lines @@
 
     if (!dump_incoming_rtp_packet_ && !dump_outgoing_rtp_packet_)
       packet_callback_.Reset();
 
     for (SocketsMap::iterator it = sockets_.begin(); it != sockets_.end(); ++it)
       it->second->StopRtpDump(incoming, outgoing);
   }
 }
 
 }  // namespace content