Allow doc.written scripts with a matching domain and registry to execute.

third_party/WebKit/Source/platform/network/NetworkUtilsTest.cpp

 // Copyright 2016 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 "platform/network/NetworkUtils.h"
 
 #include "net/base/ip_address.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "wtf/text/WTFString.h"
 
@@ skipping 43 matching lines @@
                 String::fromUTF8(addressString.data(),
                     addressString.length())));
         } else {
             EXPECT_FALSE(NetworkUtils::isReservedIPAddress(
                 String::fromUTF8(addressString.data(),
                     addressString.length())));
         }
     }
 }
 
+TEST(NetworkUtilsTest, GetDomainAndRegistry)
+{
+    EXPECT_EQ("", NetworkUtils::getDomainAndRegistry("", NetworkUtils::IncludePrivateRegistries));
+    EXPECT_EQ("", NetworkUtils::getDomainAndRegistry(".", NetworkUtils::IncludePrivateRegistries));
+    EXPECT_EQ("", NetworkUtils::getDomainAndRegistry("..", NetworkUtils::IncludePrivateRegistries));
+    EXPECT_EQ("", NetworkUtils::getDomainAndRegistry("com", NetworkUtils::IncludePrivateRegistries));
+    EXPECT_EQ("", NetworkUtils::getDomainAndRegistry(".com", NetworkUtils::IncludePrivateRegistries));
+    EXPECT_EQ("", NetworkUtils::getDomainAndRegistry("www.example.com:8000", NetworkUtils::IncludePrivateRegistries));
+
+    EXPECT_EQ("", NetworkUtils::getDomainAndRegistry("localhost", NetworkUtils::IncludePrivateRegistries));
+    EXPECT_EQ("", NetworkUtils::getDomainAndRegistry("127.0.0.1", NetworkUtils::IncludePrivateRegistries));
+
+    EXPECT_EQ("example.com", NetworkUtils::getDomainAndRegistry("example.com", NetworkUtils::IncludePrivateRegistries));
+    EXPECT_EQ("example.com", NetworkUtils::getDomainAndRegistry("www.example.com", NetworkUtils::IncludePrivateRegistries));
+    EXPECT_EQ("example.com", NetworkUtils::getDomainAndRegistry("static.example.com", NetworkUtils::IncludePrivateRegistries));
+    EXPECT_EQ("example.com", NetworkUtils::getDomainAndRegistry("multilevel.www.example.com", NetworkUtils::IncludePrivateRegistries));
+    EXPECT_EQ("example.co.uk", NetworkUtils::getDomainAndRegistry("www.example.co.uk", NetworkUtils::IncludePrivateRegistries));
+
+    // Verify proper handling of 'private registries'.
+    EXPECT_EQ("foo.appspot.com", NetworkUtils::getDomainAndRegistry("www.foo.appspot.com", NetworkUtils::IncludePrivateRegistries));
+    EXPECT_EQ("appspot.com", NetworkUtils::getDomainAndRegistry("www.foo.appspot.com", NetworkUtils::ExcludePrivateRegistries));
+
+    // Verify that unknown registries are included.
+    EXPECT_EQ("example.notarealregistry", NetworkUtils::getDomainAndRegistry("www.example.notarealregistry", NetworkUtils::IncludePrivateRegistries));
+}
+
 } // namespace blink