Add DatagramClientSocket::BindToDefaultNetwork(),GetBoundNetwork()

net/udp/udp_client_socket.cc

 // 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 "net/udp/udp_client_socket.h"
 
 #include "net/base/net_errors.h"
 #include "net/log/net_log.h"
 
 namespace net {
 
 UDPClientSocket::UDPClientSocket(DatagramSocket::BindType bind_type,
                                  const RandIntCallback& rand_int_cb,
                                  net::NetLog* net_log,
                                  const net::NetLog::Source& source)
-    : socket_(bind_type, rand_int_cb, net_log, source) {
-}
+    : socket_(bind_type, rand_int_cb, net_log, source),
+      network_(NetworkChangeNotifier::kInvalidNetworkHandle) {}
 
 UDPClientSocket::~UDPClientSocket() {
 }
 
 int UDPClientSocket::BindToNetwork(
     NetworkChangeNotifier::NetworkHandle network) {
-  return socket_.BindToNetwork(network);
+  int rv = socket_.BindToNetwork(network);
+  if (rv == OK)
+    network_ = network;
+  return rv;
+}
+
+int UDPClientSocket::BindToDefaultNetwork() {
+  if (!NetworkChangeNotifier::AreNetworkHandlesSupported())
+    return ERR_NOT_IMPLEMENTED;
+  int rv;
+  // Calling connect() will bind a socket to the default network, however there
+  // is no way to determine what network the socket got bound to.  The
+  // alternative is to query what the default network is and bind the socket to
+  // that network explicitly, however this is racy because the default network
+  // can change in between when we query it and when we bind to it.  This is
+  // rare but should be accounted for.  Since changes of the default network
+  // should not come in quick succession, we can simply try again.
+  for (int attempt = 0; attempt < 2; attempt++) {
+    NetworkChangeNotifier::NetworkHandle network =
+        NetworkChangeNotifier::GetDefaultNetwork();
+    if (network == NetworkChangeNotifier::kInvalidNetworkHandle)
+      return ERR_INTERNET_DISCONNECTED;
+    rv = BindToNetwork(network);
+    // |network| may have disconnected between the call to GetDefaultNetwork()
+    // and the call to BindToNetwork(). Loop if this is the case (|rv| will be
+    // ERR_NETWORK_CHANGED).
+    if (rv != ERR_NETWORK_CHANGED)
+      return rv;
+  }
+  return rv;
+}
+
+NetworkChangeNotifier::NetworkHandle UDPClientSocket::GetBoundNetwork() {
+  return network_;
 }
 
 int UDPClientSocket::Connect(const IPEndPoint& address) {
   int rv = socket_.Open(address.GetFamily());
   if (rv != OK)
     return rv;
   return socket_.Connect(address);
 }
 
 int UDPClientSocket::Read(IOBuffer* buf,
@@ skipping 32 matching lines @@
   return socket_.NetLog();
 }
 
 #if defined(OS_WIN)
 void UDPClientSocket::UseNonBlockingIO() {
   socket_.UseNonBlockingIO();
 }
 #endif
 
 }  // namespace net