net/base/registry_controlled_domain_unittest.cc

// Copyright (c) 2006-2008 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 "googleurl/src/gurl.h"
#include "net/base/registry_controlled_domain.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace {

class TestRegistryControlledDomainService :
    public net::RegistryControlledDomainService {
 public:
  // Sets and parses the given data.
  static void UseDomainData(const std::string& data) {
    net::RegistryControlledDomainService::UseDomainData(data);
  }

  // Creates a new dedicated instance to be used for testing, deleting any
  // previously-set one.
  static void UseDedicatedInstance() {
    delete static_cast<TestRegistryControlledDomainService*>(
        SetInstance(new TestRegistryControlledDomainService()));
  }

  // Restores RegistryControlledDomainService to using its default instance,
  // deleting any previously-set test instance.
  static void UseDefaultInstance() {
    delete static_cast<TestRegistryControlledDomainService*>(SetInstance(NULL));
  }
};

class RegistryControlledDomainTest : public testing::Test {
 protected:
  virtual void SetUp() {
    TestRegistryControlledDomainService::UseDedicatedInstance();
  }

  virtual void TearDown() {
    TestRegistryControlledDomainService::UseDefaultInstance();
  }
};

// Convenience functions to shorten the names for repeated use below.
void SetTestData(const std::string& data) {
  TestRegistryControlledDomainService::UseDomainData(data);
}

std::string GetDomainFromURL(const std::string& url) {
  return TestRegistryControlledDomainService::GetDomainAndRegistry(GURL(url));
}

std::string GetDomainFromHost(const std::wstring& host) {
  return TestRegistryControlledDomainService::GetDomainAndRegistry(host);
}

size_t GetRegistryLengthFromURL(const std::string& url,
                                bool allow_unknown_registries) {
  return TestRegistryControlledDomainService::GetRegistryLength(GURL(url),
      allow_unknown_registries);
}

size_t GetRegistryLengthFromHost(const std::wstring& host,
                                 bool allow_unknown_registries) {
  return TestRegistryControlledDomainService::GetRegistryLength(host,
      allow_unknown_registries);
}

bool CompareDomains(const std::string& url1, const std::string& url2) {
  GURL g1 = GURL(url1);
  GURL g2 = GURL(url2);
  return TestRegistryControlledDomainService::SameDomainOrHost(g1, g2);
}

TEST_F(RegistryControlledDomainTest, TestParsing) {
  // Ensure that various simple and pathological cases parse without hanging or
  // crashing.  Testing the correctness of the parsing directly would require
  // opening the singleton class up more.
  SetTestData("com");
  SetTestData("abc.com\n");
  SetTestData("abc.com\ndef.com\n*.abc.com\n!foo.abc.com");
  SetTestData("abc.com.\n");
  SetTestData("");
  SetTestData("*.");
  SetTestData("!");
  SetTestData(".");
}

static const char kTestData[] = "jp\n"            // 1
                                "ac.jp\n"         // 2
                                "*.bar.jp\n"      // 3
                                "*.baz.bar.jp\n"  // 4
                                "*.foo.bar.jp\n"  // 5
                                "!foo.bar.jp\n"   // 6
                                "!pref.bar.jp\n"  // 7
                                "bar.baz.com\n"   // 8
                                "*.c\n"           // 9
                                "!b.c";           // 10

TEST_F(RegistryControlledDomainTest, TestGetDomainAndRegistry) {
  SetTestData(kTestData);

  // Test GURL version of GetDomainAndRegistry().
  EXPECT_EQ("baz.jp", GetDomainFromURL("http://a.baz.jp/file.html"));   // 1
  EXPECT_EQ("baz.jp.", GetDomainFromURL("http://a.baz.jp./file.html")); // 1
  EXPECT_EQ("", GetDomainFromURL("http://ac.jp"));                      // 2
  EXPECT_EQ("", GetDomainFromURL("http://a.bar.jp"));                   // 3
  EXPECT_EQ("", GetDomainFromURL("http://bar.jp"));                     // 3
  EXPECT_EQ("", GetDomainFromURL("http://baz.bar.jp"));                 // 3 4
  EXPECT_EQ("a.b.baz.bar.jp", GetDomainFromURL("http://a.b.baz.bar.jp"));
                                                                        // 4
  EXPECT_EQ("foo.bar.jp", GetDomainFromURL("http://foo.bar.jp"));       // 3 5 6
  EXPECT_EQ("pref.bar.jp", GetDomainFromURL("http://baz.pref.bar.jp")); // 7
  EXPECT_EQ("b.bar.baz.com.", GetDomainFromURL("http://a.b.bar.baz.com."));
                                                                        // 8
  EXPECT_EQ("a.d.c", GetDomainFromURL("http://a.d.c"));                 // 9
  EXPECT_EQ("a.d.c", GetDomainFromURL("http://.a.d.c"));                // 9
  EXPECT_EQ("a.d.c", GetDomainFromURL("http://..a.d.c"));               // 9
  EXPECT_EQ("b.c", GetDomainFromURL("http://a.b.c"));                   // 9 10
  EXPECT_EQ("baz.com", GetDomainFromURL("http://baz.com"));             // none
  EXPECT_EQ("baz.com.", GetDomainFromURL("http://baz.com."));           // none

  EXPECT_EQ("", GetDomainFromURL(""));
  EXPECT_EQ("", GetDomainFromURL("http://"));
  EXPECT_EQ("", GetDomainFromURL("file:///C:/file.html"));
  EXPECT_EQ("", GetDomainFromURL("http://foo.com.."));
  EXPECT_EQ("", GetDomainFromURL("http://..."));
  EXPECT_EQ("", GetDomainFromURL("http://192.168.0.1"));
  EXPECT_EQ("", GetDomainFromURL("http://localhost"));
  EXPECT_EQ("", GetDomainFromURL("http://localhost."));
  EXPECT_EQ("", GetDomainFromURL("http:////Comment"));

  // Test std::wstring version of GetDomainAndRegistry().  Uses the same
  // underpinnings as the GURL version, so this is really more of a check of
  // CanonicalizeHost().
  EXPECT_EQ("baz.jp", GetDomainFromHost(L"a.baz.jp"));                 // 1
  EXPECT_EQ("baz.jp.", GetDomainFromHost(L"a.baz.jp."));               // 1
  EXPECT_EQ("", GetDomainFromHost(L"ac.jp"));                          // 2
  EXPECT_EQ("", GetDomainFromHost(L"a.bar.jp"));                       // 3
  EXPECT_EQ("", GetDomainFromHost(L"bar.jp"));                         // 3
  EXPECT_EQ("", GetDomainFromHost(L"baz.bar.jp"));                     // 3 4
  EXPECT_EQ("a.b.baz.bar.jp", GetDomainFromHost(L"a.b.baz.bar.jp"));   // 3 4
  EXPECT_EQ("foo.bar.jp", GetDomainFromHost(L"foo.bar.jp"));           // 3 5 6
  EXPECT_EQ("pref.bar.jp", GetDomainFromHost(L"baz.pref.bar.jp"));     // 7
  EXPECT_EQ("b.bar.baz.com.", GetDomainFromHost(L"a.b.bar.baz.com.")); // 8
  EXPECT_EQ("a.d.c", GetDomainFromHost(L"a.d.c"));                     // 9
  EXPECT_EQ("a.d.c", GetDomainFromHost(L".a.d.c"));                    // 9
  EXPECT_EQ("a.d.c", GetDomainFromHost(L"..a.d.c"));                   // 9
  EXPECT_EQ("b.c", GetDomainFromHost(L"a.b.c"));                       // 9 10
  EXPECT_EQ("baz.com", GetDomainFromHost(L"baz.com"));                 // none
  EXPECT_EQ("baz.com.", GetDomainFromHost(L"baz.com."));               // none

  EXPECT_EQ("", GetDomainFromHost(L""));
  EXPECT_EQ("", GetDomainFromHost(L"foo.com.."));
  EXPECT_EQ("", GetDomainFromHost(L"..."));
  EXPECT_EQ("", GetDomainFromHost(L"192.168.0.1"));
  EXPECT_EQ("", GetDomainFromHost(L"localhost."));
  EXPECT_EQ("", GetDomainFromHost(L".localhost."));
}

TEST_F(RegistryControlledDomainTest, TestGetRegistryLength) {
  SetTestData(kTestData);

  // Test GURL version of GetRegistryLength().
  EXPECT_EQ(2U, GetRegistryLengthFromURL("http://a.baz.jp/file.html", false));
                                                                        // 1
  EXPECT_EQ(3U, GetRegistryLengthFromURL("http://a.baz.jp./file.html", false));
                                                                        // 1
  EXPECT_EQ(0U, GetRegistryLengthFromURL("http://ac.jp", false));       // 2
  EXPECT_EQ(0U, GetRegistryLengthFromURL("http://a.bar.jp", false));    // 3
  EXPECT_EQ(0U, GetRegistryLengthFromURL("http://bar.jp", false));      // 3
  EXPECT_EQ(0U, GetRegistryLengthFromURL("http://baz.bar.jp", false));  // 3 4
  EXPECT_EQ(12U, GetRegistryLengthFromURL("http://a.b.baz.bar.jp", false));
                                                                        // 4
  EXPECT_EQ(6U, GetRegistryLengthFromURL("http://foo.bar.jp", false));  // 3 5 6
  EXPECT_EQ(6U, GetRegistryLengthFromURL("http://baz.pref.bar.jp", false));
                                                                        // 7
  EXPECT_EQ(11U, GetRegistryLengthFromURL("http://a.b.bar.baz.com", false));
                                                                        // 8
  EXPECT_EQ(3U, GetRegistryLengthFromURL("http://a.d.c", false));       // 9
  EXPECT_EQ(3U, GetRegistryLengthFromURL("http://.a.d.c", false));      // 9
  EXPECT_EQ(3U, GetRegistryLengthFromURL("http://..a.d.c", false));     // 9
  EXPECT_EQ(1U, GetRegistryLengthFromURL("http://a.b.c", false));       // 9 10
  EXPECT_EQ(0U, GetRegistryLengthFromURL("http://baz.com", false));     // none
  EXPECT_EQ(0U, GetRegistryLengthFromURL("http://baz.com.", false));    // none
  EXPECT_EQ(3U, GetRegistryLengthFromURL("http://baz.com", true));      // none
  EXPECT_EQ(4U, GetRegistryLengthFromURL("http://baz.com.", true));     // none

  EXPECT_EQ(std::string::npos, GetRegistryLengthFromURL("", false));
  EXPECT_EQ(std::string::npos, GetRegistryLengthFromURL("http://", false));
  EXPECT_EQ(std::string::npos,
            GetRegistryLengthFromURL("file:///C:/file.html", false));
  EXPECT_EQ(0U, GetRegistryLengthFromURL("http://foo.com..", false));
  EXPECT_EQ(0U, GetRegistryLengthFromURL("http://...", false));
  EXPECT_EQ(0U, GetRegistryLengthFromURL("http://192.168.0.1", false));
  EXPECT_EQ(0U, GetRegistryLengthFromURL("http://localhost", false));
  EXPECT_EQ(0U, GetRegistryLengthFromURL("http://localhost", true));
  EXPECT_EQ(0U, GetRegistryLengthFromURL("http://localhost.", false));
  EXPECT_EQ(0U, GetRegistryLengthFromURL("http://localhost.", true));
  EXPECT_EQ(0U, GetRegistryLengthFromURL("http:////Comment", false));

  // Test std::wstring version of GetRegistryLength().  Uses the same
  // underpinnings as the GURL version, so this is really more of a check of
  // CanonicalizeHost().
  EXPECT_EQ(2U, GetRegistryLengthFromHost(L"a.baz.jp", false));         // 1
  EXPECT_EQ(3U, GetRegistryLengthFromHost(L"a.baz.jp.", false));        // 1
  EXPECT_EQ(0U, GetRegistryLengthFromHost(L"ac.jp", false));            // 2
  EXPECT_EQ(0U, GetRegistryLengthFromHost(L"a.bar.jp", false));         // 3
  EXPECT_EQ(0U, GetRegistryLengthFromHost(L"bar.jp", false));           // 3
  EXPECT_EQ(0U, GetRegistryLengthFromHost(L"baz.bar.jp", false));       // 3 4
  EXPECT_EQ(12U, GetRegistryLengthFromHost(L"a.b.baz.bar.jp", false));  // 4
  EXPECT_EQ(6U, GetRegistryLengthFromHost(L"foo.bar.jp", false));       // 3 5 6
  EXPECT_EQ(6U, GetRegistryLengthFromHost(L"baz.pref.bar.jp", false));  // 7
  EXPECT_EQ(11U, GetRegistryLengthFromHost(L"a.b.bar.baz.com", false));
                                                                        // 8
  EXPECT_EQ(3U, GetRegistryLengthFromHost(L"a.d.c", false));            // 9
  EXPECT_EQ(3U, GetRegistryLengthFromHost(L".a.d.c", false));           // 9
  EXPECT_EQ(3U, GetRegistryLengthFromHost(L"..a.d.c", false));          // 9
  EXPECT_EQ(1U, GetRegistryLengthFromHost(L"a.b.c", false));            // 9 10
  EXPECT_EQ(0U, GetRegistryLengthFromHost(L"baz.com", false));          // none
  EXPECT_EQ(0U, GetRegistryLengthFromHost(L"baz.com.", false));         // none
  EXPECT_EQ(3U, GetRegistryLengthFromHost(L"baz.com", true));           // none
  EXPECT_EQ(4U, GetRegistryLengthFromHost(L"baz.com.", true));          // none

  EXPECT_EQ(std::string::npos, GetRegistryLengthFromHost(L"", false));
  EXPECT_EQ(0U, GetRegistryLengthFromHost(L"foo.com..", false));
  EXPECT_EQ(0U, GetRegistryLengthFromHost(L"..", false));
  EXPECT_EQ(0U, GetRegistryLengthFromHost(L"192.168.0.1", false));
  EXPECT_EQ(0U, GetRegistryLengthFromHost(L"localhost", false));
  EXPECT_EQ(0U, GetRegistryLengthFromHost(L"localhost", true));
  EXPECT_EQ(0U, GetRegistryLengthFromHost(L"localhost.", false));
  EXPECT_EQ(0U, GetRegistryLengthFromHost(L"localhost.", true));
}

TEST_F(RegistryControlledDomainTest, TestSameDomainOrHost) {
  SetTestData("jp\nbar.jp");

  EXPECT_EQ(true, CompareDomains("http://a.b.bar.jp/file.html",
                                 "http://a.b.bar.jp/file.html")); // b.bar.jp
  EXPECT_EQ(true, CompareDomains("http://a.b.bar.jp/file.html",
                                 "http://b.b.bar.jp/file.html")); // b.bar.jp
  EXPECT_EQ(false, CompareDomains("http://a.foo.jp/file.html",    // foo.jp
                                  "http://a.not.jp/file.html"));  // not.jp
  EXPECT_EQ(false, CompareDomains("http://a.foo.jp/file.html",    // foo.jp
                                  "http://a.foo.jp./file.html")); // foo.jp.
  EXPECT_EQ(false, CompareDomains("http://a.com/file.html",       // a.com
                                  "http://b.com/file.html"));     // b.com
  EXPECT_EQ(true, CompareDomains("http://a.x.com/file.html",
                                 "http://b.x.com/file.html"));    // x.com
  EXPECT_EQ(true, CompareDomains("http://a.x.com/file.html",
                                 "http://.x.com/file.html"));     // x.com
  EXPECT_EQ(true, CompareDomains("http://a.x.com/file.html",
                                 "http://..b.x.com/file.html"));  // x.com
  EXPECT_EQ(true, CompareDomains("http://intranet/file.html",
                                 "http://intranet/file.html"));   // intranet
  EXPECT_EQ(true, CompareDomains("http://127.0.0.1/file.html",
                                 "http://127.0.0.1/file.html"));  // 127.0.0.1
  EXPECT_EQ(false, CompareDomains("http://192.168.0.1/file.html", // 192.168.0.1
                                  "http://127.0.0.1/file.html")); // 127.0.0.1
  EXPECT_EQ(false, CompareDomains("file:///C:/file.html",
                                  "file:///C:/file.html"));       // no host
}

TEST_F(RegistryControlledDomainTest, TestDefaultData) {
  TestRegistryControlledDomainService::UseDefaultInstance();

  // Note that no data is set: we're using the default rules.
  EXPECT_EQ(3U, GetRegistryLengthFromURL("http://google.com", false));
  EXPECT_EQ(3U, GetRegistryLengthFromURL("http://stanford.edu", false));
  EXPECT_EQ(3U, GetRegistryLengthFromURL("http://ustreas.gov", false));
  EXPECT_EQ(3U, GetRegistryLengthFromURL("http://icann.net", false));
  EXPECT_EQ(3U, GetRegistryLengthFromURL("http://ferretcentral.org", false));
  EXPECT_EQ(0U, GetRegistryLengthFromURL("http://nowhere.foo", false));
  EXPECT_EQ(3U, GetRegistryLengthFromURL("http://nowhere.foo", true));
}

} // namespace