net/base/address_list_unittest.cc - platform/external/chromium - Git at Google

 // 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/base/address_list.h"

 #include "base/memory/scoped_ptr.h"
 #include "base/string_util.h"
 #include "net/base/host_resolver_proc.h"
 #include "net/base/net_util.h"
 #include "net/base/sys_addrinfo.h"
 #if defined(OS_WIN)
 #include "net/base/winsock_init.h"
 #endif
 #include "testing/gtest/include/gtest/gtest.h"

 namespace net {
 namespace {

 // Use getaddrinfo() to allocate an addrinfo structure.
 int CreateAddressList(const std::string& hostname, int port,
                       AddressList* addrlist) {
 #if defined(OS_WIN)
   EnsureWinsockInit();
 #endif
   int rv = SystemHostResolverProc(hostname,
                                   ADDRESS_FAMILY_UNSPECIFIED,
                                   0,
                                   addrlist, NULL);
   if (rv == 0)
     addrlist->SetPort(port);
   return rv;
 }

 void CreateLongAddressList(AddressList* addrlist, int port) {
   EXPECT_EQ(0, CreateAddressList("192.168.1.1", port, addrlist));
   AddressList second_list;
   EXPECT_EQ(0, CreateAddressList("192.168.1.2", port, &second_list));
   addrlist->Append(second_list.head());
 }

 TEST(AddressListTest, GetPort) {
   AddressList addrlist;
   EXPECT_EQ(0, CreateAddressList("192.168.1.1", 81, &addrlist));
   EXPECT_EQ(81, addrlist.GetPort());

   addrlist.SetPort(83);
   EXPECT_EQ(83, addrlist.GetPort());
 }

 TEST(AddressListTest, Assignment) {
   AddressList addrlist1;
   EXPECT_EQ(0, CreateAddressList("192.168.1.1", 85, &addrlist1));
   EXPECT_EQ(85, addrlist1.GetPort());

   // Should reference the same data as addrlist1 -- so when we change addrlist1
   // both are changed.
   AddressList addrlist2 = addrlist1;
   EXPECT_EQ(85, addrlist2.GetPort());

   addrlist1.SetPort(80);
   EXPECT_EQ(80, addrlist1.GetPort());
   EXPECT_EQ(80, addrlist2.GetPort());
 }

 TEST(AddressListTest, CopyRecursive) {
   AddressList addrlist1;
   CreateLongAddressList(&addrlist1, 85);
   EXPECT_EQ(85, addrlist1.GetPort());

   AddressList addrlist2;
   addrlist2.Copy(addrlist1.head(), true);

   ASSERT_TRUE(addrlist2.head()->ai_next != NULL);

   // addrlist1 is the same as addrlist2 at this point.
   EXPECT_EQ(85, addrlist1.GetPort());
   EXPECT_EQ(85, addrlist2.GetPort());

   // Changes to addrlist1 are not reflected in addrlist2.
   addrlist1.SetPort(70);
   addrlist2.SetPort(90);

   EXPECT_EQ(70, addrlist1.GetPort());
   EXPECT_EQ(90, addrlist2.GetPort());
 }

 TEST(AddressListTest, CopyNonRecursive) {
   AddressList addrlist1;
   CreateLongAddressList(&addrlist1, 85);
   EXPECT_EQ(85, addrlist1.GetPort());

   AddressList addrlist2;
   addrlist2.Copy(addrlist1.head(), false);

   ASSERT_TRUE(addrlist2.head()->ai_next == NULL);

   // addrlist1 is the same as addrlist2 at this point.
   EXPECT_EQ(85, addrlist1.GetPort());
   EXPECT_EQ(85, addrlist2.GetPort());

   // Changes to addrlist1 are not reflected in addrlist2.
   addrlist1.SetPort(70);
   addrlist2.SetPort(90);

   EXPECT_EQ(70, addrlist1.GetPort());
   EXPECT_EQ(90, addrlist2.GetPort());
 }

 TEST(AddressListTest, Append) {
   AddressList addrlist1;
   EXPECT_EQ(0, CreateAddressList("192.168.1.1", 11, &addrlist1));
   EXPECT_EQ(11, addrlist1.GetPort());
   AddressList addrlist2;
   EXPECT_EQ(0, CreateAddressList("192.168.1.2", 12, &addrlist2));
   EXPECT_EQ(12, addrlist2.GetPort());

   ASSERT_TRUE(addrlist1.head()->ai_next == NULL);
   addrlist1.Append(addrlist2.head());
   ASSERT_TRUE(addrlist1.head()->ai_next != NULL);

   AddressList addrlist3;
   addrlist3.Copy(addrlist1.head()->ai_next, false);
   EXPECT_EQ(12, addrlist3.GetPort());
 }

 static const char* kCanonicalHostname = "canonical.bar.com";

 TEST(AddressListTest, Canonical) {
   // Create an addrinfo with a canonical name.
   sockaddr_in address;
   // The contents of address do not matter for this test,
   // so just zero-ing them out for consistency.
   memset(&address, 0x0, sizeof(address));
   struct addrinfo ai;
   memset(&ai, 0x0, sizeof(ai));
   ai.ai_family = AF_INET;
   ai.ai_socktype = SOCK_STREAM;
   ai.ai_addrlen = sizeof(address);
   ai.ai_addr = reinterpret_cast<sockaddr*>(&address);
   ai.ai_canonname = const_cast<char *>(kCanonicalHostname);

   // Copy the addrinfo struct into an AddressList object and
   // make sure it seems correct.
   AddressList addrlist1;
   addrlist1.Copy(&ai, true);
   const struct addrinfo* addrinfo1 = addrlist1.head();
   EXPECT_TRUE(addrinfo1 != NULL);
   EXPECT_TRUE(addrinfo1->ai_next == NULL);
   std::string canon_name1;
   EXPECT_TRUE(addrlist1.GetCanonicalName(&canon_name1));
   EXPECT_EQ("canonical.bar.com", canon_name1);

   // Copy the AddressList to another one.
   AddressList addrlist2;
   addrlist2.Copy(addrinfo1, true);
   const struct addrinfo* addrinfo2 = addrlist2.head();
   EXPECT_TRUE(addrinfo2 != NULL);
   EXPECT_TRUE(addrinfo2->ai_next == NULL);
   EXPECT_TRUE(addrinfo2->ai_canonname != NULL);
   EXPECT_NE(addrinfo1, addrinfo2);
   EXPECT_NE(addrinfo1->ai_canonname, addrinfo2->ai_canonname);
   std::string canon_name2;
   EXPECT_TRUE(addrlist2.GetCanonicalName(&canon_name2));
   EXPECT_EQ("canonical.bar.com", canon_name2);

   // Make sure that GetCanonicalName correctly returns false
   // when ai_canonname is NULL.
   ai.ai_canonname = NULL;
   AddressList addrlist_no_canon;
   addrlist_no_canon.Copy(&ai, true);
   std::string canon_name3 = "blah";
   EXPECT_FALSE(addrlist_no_canon.GetCanonicalName(&canon_name3));
   EXPECT_EQ("blah", canon_name3);
 }

 TEST(AddressListTest, IPLiteralConstructor) {
   struct TestData {
     std::string ip_address;
     std::string canonical_ip_address;
     bool is_ipv6;
   } tests[] = {
     { "127.0.00.1", "127.0.0.1", false },
     { "192.168.1.1", "192.168.1.1", false },
     { "::1", "::1", true },
     { "2001:db8:0::42", "2001:db8::42", true },
   };
   for (size_t i = 0; i < ARRAYSIZE_UNSAFE(tests); i++) {
     AddressList expected_list;
     int rv = CreateAddressList(tests[i].canonical_ip_address, 80,
                                &expected_list);
     if (tests[i].is_ipv6 && rv != 0) {
       LOG(WARNING) << "Unable to resolve ip literal '" << tests[i].ip_address
                    << "' test skipped.";
       continue;
     }
     ASSERT_EQ(0, rv);
     const struct addrinfo* good_ai = expected_list.head();

     IPAddressNumber ip_number;
     ParseIPLiteralToNumber(tests[i].ip_address, &ip_number);
     AddressList test_list(ip_number, 80, true);
     const struct addrinfo* test_ai = test_list.head();

     EXPECT_EQ(good_ai->ai_family, test_ai->ai_family);
     EXPECT_EQ(good_ai->ai_socktype, test_ai->ai_socktype);
     EXPECT_EQ(good_ai->ai_addrlen, test_ai->ai_addrlen);
     size_t sockaddr_size =
         good_ai->ai_socktype == AF_INET ? sizeof(struct sockaddr_in) :
         good_ai->ai_socktype == AF_INET6 ? sizeof(struct sockaddr_in6) : 0;
     EXPECT_EQ(memcmp(good_ai->ai_addr, test_ai->ai_addr, sockaddr_size), 0);
     EXPECT_EQ(good_ai->ai_next, test_ai->ai_next);
     EXPECT_EQ(strcmp(tests[i].canonical_ip_address.c_str(),
                      test_ai->ai_canonname), 0);
   }
 }

 TEST(AddressListTest, AddressFromAddrInfo) {
   struct TestData {
     std::string ip_address;
     std::string canonical_ip_address;
     bool is_ipv6;
   } tests[] = {
     { "127.0.00.1", "127.0.0.1", false },
     { "192.168.1.1", "192.168.1.1", false },
     { "::1", "::1", true },
     { "2001:db8:0::42", "2001:db8::42", true },
   };
   for (size_t i = 0; i < ARRAYSIZE_UNSAFE(tests); i++) {
     AddressList expected_list;
     int rv = CreateAddressList(tests[i].canonical_ip_address, 80,
                                &expected_list);
     if (tests[i].is_ipv6 && rv != 0) {
       LOG(WARNING) << "Unable to resolve ip literal '" << tests[i].ip_address
                    << "' test skipped.";
       continue;
     }
     ASSERT_EQ(0, rv);
     const struct addrinfo* good_ai = expected_list.head();

     scoped_ptr<AddressList> test_list(
         AddressList::CreateAddressListFromSockaddr(good_ai->ai_addr,
                                                    good_ai->ai_addrlen,
                                                    SOCK_STREAM,
                                                    IPPROTO_TCP));
     const struct addrinfo* test_ai = test_list->head();

     EXPECT_EQ(good_ai->ai_family, test_ai->ai_family);
     EXPECT_EQ(good_ai->ai_addrlen, test_ai->ai_addrlen);
     size_t sockaddr_size =
         good_ai->ai_socktype == AF_INET ? sizeof(struct sockaddr_in) :
         good_ai->ai_socktype == AF_INET6 ? sizeof(struct sockaddr_in6) : 0;
     EXPECT_EQ(memcmp(good_ai->ai_addr, test_ai->ai_addr, sockaddr_size), 0);
     EXPECT_EQ(good_ai->ai_next, test_ai->ai_next);
   }
 }

 }  // namespace
 }  // namespace net
	// 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/base/address_list.h"

	#include "base/memory/scoped_ptr.h"
	#include "base/string_util.h"
	#include "net/base/host_resolver_proc.h"
	#include "net/base/net_util.h"
	#include "net/base/sys_addrinfo.h"
	#if defined(OS_WIN)
	#include "net/base/winsock_init.h"
	#endif
	#include "testing/gtest/include/gtest/gtest.h"

	namespace net {
	namespace {

	// Use getaddrinfo() to allocate an addrinfo structure.
	int CreateAddressList(const std::string& hostname, int port,
	AddressList* addrlist) {
	#if defined(OS_WIN)
	EnsureWinsockInit();
	#endif
	int rv = SystemHostResolverProc(hostname,
	ADDRESS_FAMILY_UNSPECIFIED,
	0,
	addrlist, NULL);
	if (rv == 0)
	addrlist->SetPort(port);
	return rv;
	}

	void CreateLongAddressList(AddressList* addrlist, int port) {
	EXPECT_EQ(0, CreateAddressList("192.168.1.1", port, addrlist));
	AddressList second_list;
	EXPECT_EQ(0, CreateAddressList("192.168.1.2", port, &second_list));
	addrlist->Append(second_list.head());
	}

	TEST(AddressListTest, GetPort) {
	AddressList addrlist;
	EXPECT_EQ(0, CreateAddressList("192.168.1.1", 81, &addrlist));
	EXPECT_EQ(81, addrlist.GetPort());

	addrlist.SetPort(83);
	EXPECT_EQ(83, addrlist.GetPort());
	}

	TEST(AddressListTest, Assignment) {
	AddressList addrlist1;
	EXPECT_EQ(0, CreateAddressList("192.168.1.1", 85, &addrlist1));
	EXPECT_EQ(85, addrlist1.GetPort());

	// Should reference the same data as addrlist1 -- so when we change addrlist1
	// both are changed.
	AddressList addrlist2 = addrlist1;
	EXPECT_EQ(85, addrlist2.GetPort());

	addrlist1.SetPort(80);
	EXPECT_EQ(80, addrlist1.GetPort());
	EXPECT_EQ(80, addrlist2.GetPort());
	}

	TEST(AddressListTest, CopyRecursive) {
	AddressList addrlist1;
	CreateLongAddressList(&addrlist1, 85);
	EXPECT_EQ(85, addrlist1.GetPort());

	AddressList addrlist2;
	addrlist2.Copy(addrlist1.head(), true);

	ASSERT_TRUE(addrlist2.head()->ai_next != NULL);

	// addrlist1 is the same as addrlist2 at this point.
	EXPECT_EQ(85, addrlist1.GetPort());
	EXPECT_EQ(85, addrlist2.GetPort());

	// Changes to addrlist1 are not reflected in addrlist2.
	addrlist1.SetPort(70);
	addrlist2.SetPort(90);

	EXPECT_EQ(70, addrlist1.GetPort());
	EXPECT_EQ(90, addrlist2.GetPort());
	}

	TEST(AddressListTest, CopyNonRecursive) {
	AddressList addrlist1;
	CreateLongAddressList(&addrlist1, 85);
	EXPECT_EQ(85, addrlist1.GetPort());

	AddressList addrlist2;
	addrlist2.Copy(addrlist1.head(), false);

	ASSERT_TRUE(addrlist2.head()->ai_next == NULL);

	// addrlist1 is the same as addrlist2 at this point.
	EXPECT_EQ(85, addrlist1.GetPort());
	EXPECT_EQ(85, addrlist2.GetPort());

	// Changes to addrlist1 are not reflected in addrlist2.
	addrlist1.SetPort(70);
	addrlist2.SetPort(90);

	EXPECT_EQ(70, addrlist1.GetPort());
	EXPECT_EQ(90, addrlist2.GetPort());
	}

	TEST(AddressListTest, Append) {
	AddressList addrlist1;
	EXPECT_EQ(0, CreateAddressList("192.168.1.1", 11, &addrlist1));
	EXPECT_EQ(11, addrlist1.GetPort());
	AddressList addrlist2;
	EXPECT_EQ(0, CreateAddressList("192.168.1.2", 12, &addrlist2));
	EXPECT_EQ(12, addrlist2.GetPort());

	ASSERT_TRUE(addrlist1.head()->ai_next == NULL);
	addrlist1.Append(addrlist2.head());
	ASSERT_TRUE(addrlist1.head()->ai_next != NULL);

	AddressList addrlist3;
	addrlist3.Copy(addrlist1.head()->ai_next, false);
	EXPECT_EQ(12, addrlist3.GetPort());
	}

	static const char* kCanonicalHostname = "canonical.bar.com";

	TEST(AddressListTest, Canonical) {
	// Create an addrinfo with a canonical name.
	sockaddr_in address;
	// The contents of address do not matter for this test,
	// so just zero-ing them out for consistency.
	memset(&address, 0x0, sizeof(address));
	struct addrinfo ai;
	memset(&ai, 0x0, sizeof(ai));
	ai.ai_family = AF_INET;
	ai.ai_socktype = SOCK_STREAM;
	ai.ai_addrlen = sizeof(address);
	ai.ai_addr = reinterpret_cast<sockaddr*>(&address);
	ai.ai_canonname = const_cast<char *>(kCanonicalHostname);

	// Copy the addrinfo struct into an AddressList object and
	// make sure it seems correct.
	AddressList addrlist1;
	addrlist1.Copy(&ai, true);
	const struct addrinfo* addrinfo1 = addrlist1.head();
	EXPECT_TRUE(addrinfo1 != NULL);
	EXPECT_TRUE(addrinfo1->ai_next == NULL);
	std::string canon_name1;
	EXPECT_TRUE(addrlist1.GetCanonicalName(&canon_name1));
	EXPECT_EQ("canonical.bar.com", canon_name1);

	// Copy the AddressList to another one.
	AddressList addrlist2;
	addrlist2.Copy(addrinfo1, true);
	const struct addrinfo* addrinfo2 = addrlist2.head();
	EXPECT_TRUE(addrinfo2 != NULL);
	EXPECT_TRUE(addrinfo2->ai_next == NULL);
	EXPECT_TRUE(addrinfo2->ai_canonname != NULL);
	EXPECT_NE(addrinfo1, addrinfo2);
	EXPECT_NE(addrinfo1->ai_canonname, addrinfo2->ai_canonname);
	std::string canon_name2;
	EXPECT_TRUE(addrlist2.GetCanonicalName(&canon_name2));
	EXPECT_EQ("canonical.bar.com", canon_name2);

	// Make sure that GetCanonicalName correctly returns false
	// when ai_canonname is NULL.
	ai.ai_canonname = NULL;
	AddressList addrlist_no_canon;
	addrlist_no_canon.Copy(&ai, true);
	std::string canon_name3 = "blah";
	EXPECT_FALSE(addrlist_no_canon.GetCanonicalName(&canon_name3));
	EXPECT_EQ("blah", canon_name3);
	}

	TEST(AddressListTest, IPLiteralConstructor) {
	struct TestData {
	std::string ip_address;
	std::string canonical_ip_address;
	bool is_ipv6;
	} tests[] = {
	{ "127.0.00.1", "127.0.0.1", false },
	{ "192.168.1.1", "192.168.1.1", false },
	{ "::1", "::1", true },
	{ "2001:db8:0::42", "2001:db8::42", true },
	};
	for (size_t i = 0; i < ARRAYSIZE_UNSAFE(tests); i++) {
	AddressList expected_list;
	int rv = CreateAddressList(tests[i].canonical_ip_address, 80,
	&expected_list);
	if (tests[i].is_ipv6 && rv != 0) {
	LOG(WARNING) << "Unable to resolve ip literal '" << tests[i].ip_address
	<< "' test skipped.";
	continue;
	}
	ASSERT_EQ(0, rv);
	const struct addrinfo* good_ai = expected_list.head();

	IPAddressNumber ip_number;
	ParseIPLiteralToNumber(tests[i].ip_address, &ip_number);
	AddressList test_list(ip_number, 80, true);
	const struct addrinfo* test_ai = test_list.head();

	EXPECT_EQ(good_ai->ai_family, test_ai->ai_family);
	EXPECT_EQ(good_ai->ai_socktype, test_ai->ai_socktype);
	EXPECT_EQ(good_ai->ai_addrlen, test_ai->ai_addrlen);
	size_t sockaddr_size =
	good_ai->ai_socktype == AF_INET ? sizeof(struct sockaddr_in) :
	good_ai->ai_socktype == AF_INET6 ? sizeof(struct sockaddr_in6) : 0;
	EXPECT_EQ(memcmp(good_ai->ai_addr, test_ai->ai_addr, sockaddr_size), 0);
	EXPECT_EQ(good_ai->ai_next, test_ai->ai_next);
	EXPECT_EQ(strcmp(tests[i].canonical_ip_address.c_str(),
	test_ai->ai_canonname), 0);
	}
	}

	TEST(AddressListTest, AddressFromAddrInfo) {
	struct TestData {
	std::string ip_address;
	std::string canonical_ip_address;
	bool is_ipv6;
	} tests[] = {
	{ "127.0.00.1", "127.0.0.1", false },
	{ "192.168.1.1", "192.168.1.1", false },
	{ "::1", "::1", true },
	{ "2001:db8:0::42", "2001:db8::42", true },
	};
	for (size_t i = 0; i < ARRAYSIZE_UNSAFE(tests); i++) {
	AddressList expected_list;
	int rv = CreateAddressList(tests[i].canonical_ip_address, 80,
	&expected_list);
	if (tests[i].is_ipv6 && rv != 0) {
	LOG(WARNING) << "Unable to resolve ip literal '" << tests[i].ip_address
	<< "' test skipped.";
	continue;
	}
	ASSERT_EQ(0, rv);
	const struct addrinfo* good_ai = expected_list.head();

	scoped_ptr<AddressList> test_list(
	AddressList::CreateAddressListFromSockaddr(good_ai->ai_addr,
	good_ai->ai_addrlen,
	SOCK_STREAM,
	IPPROTO_TCP));
	const struct addrinfo* test_ai = test_list->head();

	EXPECT_EQ(good_ai->ai_family, test_ai->ai_family);
	EXPECT_EQ(good_ai->ai_addrlen, test_ai->ai_addrlen);
	size_t sockaddr_size =
	good_ai->ai_socktype == AF_INET ? sizeof(struct sockaddr_in) :
	good_ai->ai_socktype == AF_INET6 ? sizeof(struct sockaddr_in6) : 0;
	EXPECT_EQ(memcmp(good_ai->ai_addr, test_ai->ai_addr, sockaddr_size), 0);
	EXPECT_EQ(good_ai->ai_next, test_ai->ai_next);
	}
	}

	} // namespace
	} // namespace net