CORS-RFC1918: Introduce the 'addressSpace' IDL attributes.

content/child/web_url_loader_impl_unittest.cc

 // Copyright 2014 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/child/web_url_loader_impl.h"
 
 #include <stdint.h>
 #include <string.h>
 #include <utility>
 #include <vector>
 
 #include "base/command_line.h"
 #include "base/macros.h"
 #include "base/memory/weak_ptr.h"
 #include "base/message_loop/message_loop.h"
 #include "base/single_thread_task_runner.h"
 #include "base/time/default_tick_clock.h"
 #include "base/time/time.h"
 #include "components/scheduler/child/scheduler_tqm_delegate_impl.h"
 #include "components/scheduler/child/web_task_runner_impl.h"
 #include "components/scheduler/child/worker_scheduler.h"
 #include "content/child/request_extra_data.h"
 #include "content/child/request_info.h"
 #include "content/child/resource_dispatcher.h"
 #include "content/public/child/fixed_received_data.h"
 #include "content/public/child/request_peer.h"
 #include "content/public/common/content_switches.h"
 #include "content/public/common/resource_response_info.h"
+#include "net/base/host_port_pair.h"
 #include "net/base/net_errors.h"
 #include "net/http/http_response_headers.h"
 #include "net/http/http_util.h"
 #include "net/url_request/redirect_info.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/WebKit/public/platform/WebString.h"
 #include "third_party/WebKit/public/platform/WebURLError.h"
 #include "third_party/WebKit/public/platform/WebURLLoaderClient.h"
 #include "third_party/WebKit/public/platform/WebURLRequest.h"
 #include "third_party/WebKit/public/platform/WebURLResponse.h"
@@ skipping 664 matching lines @@
   // The response info should have been overriden.
   ASSERT_FALSE(client()->response().isNull());
   EXPECT_EQ(kMimeType, client()->response().mimeType().latin1());
 
   DoReceiveData();
   DoCompleteRequest();
   EXPECT_FALSE(dispatcher()->canceled());
   EXPECT_EQ(kTestData, client()->received_data());
 }
 
+TEST_F(WebURLLoaderImplTest, ResponseIPAddress) {
+  GURL url("http://example.test/");
+
+  struct TestCase {
+    const char* ip;
+    const char* expected;
+  } cases[] = {
+      {"127.0.0.1", "127.0.0.1"},
+      {"123.123.123.123", "123.123.123.123"},
+      {"::1", "[::1]"},
+      {"2001:0db8:85a3:0000:0000:8a2e:0370:7334",
+       "[2001:0db8:85a3:0000:0000:8a2e:0370:7334]"},
+      {"2001:db8:85a3:0:0:8a2e:370:7334", "[2001:db8:85a3:0:0:8a2e:370:7334]"},
+      {"2001:db8:85a3::8a2e:370:7334", "[2001:db8:85a3::8a2e:370:7334]"},
+      {"::ffff:192.0.2.128", "[::ffff:192.0.2.128]"}};
+
+  for (const auto& test : cases) {
+    SCOPED_TRACE(test.ip);
+    content::ResourceResponseInfo info;
+    info.socket_address = net::HostPortPair(test.ip, 443);
+    blink::WebURLResponse response;
+    response.initialize();
+    WebURLLoaderImpl::PopulateURLResponse(url, info, &response, true);
+    EXPECT_EQ(test.expected, response.remoteIPAddress().utf8());
+  };
+}
+
 }  // namespace
 }  // namespace content