Allow the localhost IP address even if it does not match the tcp port address

webrtc/p2p/base/tcpport_unittest.cc

+/*
+ *  Copyright 2016 The WebRTC Project Authors. All rights reserved.
+ *
+ *  Use of this source code is governed by a BSD-style license
+ *  that can be found in the LICENSE file in the root of the source
+ *  tree. An additional intellectual property rights grant can be found
+ *  in the file PATENTS.  All contributing project authors may
+ *  be found in the AUTHORS file in the root of the source tree.
+ */
+
+#include "webrtc/base/gunit.h"
+#include "webrtc/base/physicalsocketserver.h"
+#include "webrtc/base/thread.h"
+#include "webrtc/base/virtualsocketserver.h"
+#include "webrtc/p2p/base/basicpacketsocketfactory.h"
+#include "webrtc/p2p/base/tcpport.h"
+
+using rtc::SocketAddress;
+using cricket::Connection;
+using cricket::Port;
+using cricket::TCPPort;
+using cricket::ICE_UFRAG_LENGTH;
+using cricket::ICE_PWD_LENGTH;
+
+static int kTimeout = 1000;
+static const SocketAddress kLocalAddr("11.11.11.11", 1);
+static const SocketAddress kRemoteAddr("22.22.22.22", 2);
+
+class TCPPortTest : public testing::Test, public sigslot::has_slots<> {
+ public:
+  TCPPortTest()
+      : main_(rtc::Thread::Current()),
+        pss_(new rtc::PhysicalSocketServer),
+        ss_(new rtc::VirtualSocketServer(pss_.get())),
+        ss_scope_(ss_.get()),
+        network_("unittest", "unittest", rtc::IPAddress(INADDR_ANY), 32),
+        socket_factory_(rtc::Thread::Current()),
+        username_(rtc::CreateRandomString(ICE_UFRAG_LENGTH)),
+        password_(rtc::CreateRandomString(ICE_PWD_LENGTH)) {
+    network_.AddIP(rtc::IPAddress(INADDR_ANY));
+  }
+
+  void ConnectSignalSocketCreated() {
+    ss_->SignalSocketCreated.connect(this, &TCPPortTest::OnSocketCreated);
+  }
+
+  void OnSocketCreated(rtc::VirtualSocket* socket) {
+    LOG(LS_INFO) << "socket created ";
+    socket->SignalAddressReady.connect(
+        this, &TCPPortTest::SetLocalhostAsAlternativeLocalAddress);
+  }
+
+  void SetLocalhostAsAlternativeLocalAddress(rtc::VirtualSocket* socket,
+                                             const SocketAddress& address) {
+    SocketAddress local_address("127.0.0.1", 2000);
+    socket->SetAlternativeLocalAddress(local_address);
+  }
+
+  TCPPort* CreateTCPPort(const SocketAddress& addr) {
+    return TCPPort::Create(main_, &socket_factory_, &network_, addr.ipaddr(), 0,
+                           0, username_, password_, true);
+  }
+
+ protected:
+  rtc::Thread* main_;
+  rtc::scoped_ptr<rtc::PhysicalSocketServer> pss_;
+  rtc::scoped_ptr<rtc::VirtualSocketServer> ss_;
+  rtc::SocketServerScope ss_scope_;
+  rtc::Network network_;
+  rtc::BasicPacketSocketFactory socket_factory_;
+  std::string username_;
+  std::string password_;
+};
+
+TEST_F(TCPPortTest, TestTCPPortWithLocalhostAddress) {
+  rtc::scoped_ptr<TCPPort> lport(CreateTCPPort(kLocalAddr));
+  rtc::scoped_ptr<TCPPort> rport(CreateTCPPort(kRemoteAddr));
+  lport->PrepareAddress();
+  rport->PrepareAddress();
+  // Start to listen to new socket creation event.
+  ConnectSignalSocketCreated();
+  Connection* conn =
+      lport->CreateConnection(rport->Candidates()[0], Port::ORIGIN_MESSAGE);
+  EXPECT_TRUE_WAIT(conn->connected(), kTimeout);
+}