| Index: net/base/registry_controlled_domains/registry_controlled_domain.cc
|
| diff --git a/net/base/registry_controlled_domains/registry_controlled_domain.cc b/net/base/registry_controlled_domains/registry_controlled_domain.cc
|
| index 645a4f2132571886bd5bb25c26a2fcccfe692f08..607d735f3080be6b37354b5963064df6e78bd7fa 100644
|
| --- a/net/base/registry_controlled_domains/registry_controlled_domain.cc
|
| +++ b/net/base/registry_controlled_domains/registry_controlled_domain.cc
|
| @@ -66,19 +66,10 @@
|
| const unsigned char* g_graph = kDafsa;
|
| size_t g_graph_length = sizeof(kDafsa);
|
|
|
| -struct MappedHostComponent {
|
| - size_t original_begin;
|
| - size_t original_end;
|
| -
|
| - size_t canonical_begin;
|
| - size_t canonical_end;
|
| -};
|
| -
|
| size_t GetRegistryLengthImpl(base::StringPiece host,
|
| UnknownRegistryFilter unknown_filter,
|
| PrivateRegistryFilter private_filter) {
|
| - if (host.empty())
|
| - return std::string::npos;
|
| + DCHECK(!host.empty());
|
|
|
| // Skip leading dots.
|
| const size_t host_check_begin = host.find_first_not_of('.');
|
| @@ -198,117 +189,6 @@
|
| return GetDomainAndRegistryImpl(host, filter);
|
| }
|
|
|
| -// These two functions append the given string as-is to the given output,
|
| -// converting to UTF-8 if necessary.
|
| -void AppendInvalidString(base::StringPiece str, url::CanonOutput* output) {
|
| - output->Append(str.data(), static_cast<int>(str.length()));
|
| -}
|
| -void AppendInvalidString(base::StringPiece16 str, url::CanonOutput* output) {
|
| - std::string utf8 = base::UTF16ToUTF8(str);
|
| - output->Append(utf8.data(), static_cast<int>(utf8.length()));
|
| -}
|
| -
|
| -// Backend for PermissiveGetHostRegistryLength that handles both UTF-8 and
|
| -// UTF-16 input. The template type is the std::string type to use (it makes the
|
| -// typedefs easier than using the character type).
|
| -template <typename Str>
|
| -size_t DoPermissiveGetHostRegistryLength(base::BasicStringPiece<Str> host,
|
| - UnknownRegistryFilter unknown_filter,
|
| - PrivateRegistryFilter private_filter) {
|
| - std::string canonical_host; // Do not modify outside of canon_output.
|
| - canonical_host.reserve(host.length());
|
| - url::StdStringCanonOutput canon_output(&canonical_host);
|
| -
|
| - std::vector<MappedHostComponent> components;
|
| -
|
| - for (size_t current = 0; current < host.length(); current++) {
|
| - size_t begin = current;
|
| -
|
| - // Advance to next "." or end.
|
| - current = host.find('.', begin);
|
| - if (current == std::string::npos)
|
| - current = host.length();
|
| -
|
| - MappedHostComponent mapping;
|
| - mapping.original_begin = begin;
|
| - mapping.original_end = current;
|
| - mapping.canonical_begin = static_cast<size_t>(canon_output.length());
|
| -
|
| - // Try to append the canonicalized version of this component.
|
| - int current_len = static_cast<int>(current - begin);
|
| - if (!url::CanonicalizeHostSubstring(
|
| - host.data(), url::Component(static_cast<int>(begin), current_len),
|
| - &canon_output)) {
|
| - // Failed to canonicalize this component; append as-is.
|
| - AppendInvalidString(host.substr(begin, current_len), &canon_output);
|
| - }
|
| -
|
| - mapping.canonical_end = static_cast<size_t>(canon_output.length());
|
| - components.push_back(mapping);
|
| -
|
| - if (current < host.length())
|
| - canon_output.push_back('.');
|
| - }
|
| - canon_output.Complete();
|
| -
|
| - size_t canonical_rcd_len =
|
| - GetRegistryLengthImpl(canonical_host, unknown_filter, private_filter);
|
| - if (canonical_rcd_len == 0 || canonical_rcd_len == std::string::npos)
|
| - return canonical_rcd_len; // Error or no registry controlled domain.
|
| -
|
| - // Find which host component the result started in.
|
| - size_t canonical_rcd_begin = canonical_host.length() - canonical_rcd_len;
|
| - for (const auto& mapping : components) {
|
| - // In the common case, GetRegistryLengthImpl will identify the beginning
|
| - // of a component and we can just return where that component was in the
|
| - // original string.
|
| - if (canonical_rcd_begin == mapping.canonical_begin)
|
| - return host.length() - mapping.original_begin;
|
| -
|
| - if (canonical_rcd_begin >= mapping.canonical_end)
|
| - continue;
|
| -
|
| - // The registry controlled domain begin was identified as being in the
|
| - // middle of this dot-separated domain component in the non-canonical
|
| - // input. This indicates some form of escaped dot, or a non-ASCII
|
| - // character that was canonicalized to a dot.
|
| - //
|
| - // Brute-force search from the end by repeatedly canonicalizing longer
|
| - // substrings until we get a match for the canonicalized version. This
|
| - // can't be done with binary search because canonicalization might increase
|
| - // or decrease the length of the produced string depending on where it's
|
| - // split. This depends on the canonicalization process not changing the
|
| - // order of the characters. Punycode can change the order of characters,
|
| - // but it doesn't work across dots so this is safe.
|
| -
|
| - // Expected canonical registry controlled domain.
|
| - base::StringPiece canonical_rcd(&canonical_host[canonical_rcd_begin],
|
| - canonical_rcd_len);
|
| -
|
| - for (int current_try = static_cast<int>(mapping.original_end) - 1;
|
| - current_try >= static_cast<int>(mapping.original_begin);
|
| - current_try--) {
|
| - std::string try_string;
|
| - url::StdStringCanonOutput try_output(&try_string);
|
| -
|
| - if (!url::CanonicalizeHostSubstring(
|
| - host.data(),
|
| - url::Component(
|
| - current_try,
|
| - static_cast<int>(mapping.original_end) - current_try),
|
| - &try_output))
|
| - continue; // Invalid substring, skip.
|
| -
|
| - try_output.Complete();
|
| - if (try_string == canonical_rcd)
|
| - return host.length() - current_try;
|
| - }
|
| - }
|
| -
|
| - NOTREACHED();
|
| - return canonical_rcd_len;
|
| -}
|
| -
|
| } // namespace
|
|
|
| std::string GetDomainAndRegistry(const GURL& gurl,
|
| @@ -358,63 +238,24 @@
|
| const GURL& gurl,
|
| UnknownRegistryFilter unknown_filter,
|
| PrivateRegistryFilter private_filter) {
|
| - return GetRegistryLengthImpl(gurl.host_piece(), unknown_filter,
|
| - private_filter);
|
| -}
|
| -
|
| -bool HostHasRegistryControlledDomain(base::StringPiece host,
|
| - UnknownRegistryFilter unknown_filter,
|
| - PrivateRegistryFilter private_filter) {
|
| + base::StringPiece host = gurl.host_piece();
|
| + if (host.empty())
|
| + return std::string::npos;
|
| + if (gurl.HostIsIPAddress())
|
| + return 0;
|
| + return GetRegistryLengthImpl(host, unknown_filter, private_filter);
|
| +}
|
| +
|
| +size_t GetRegistryLength(base::StringPiece host,
|
| + UnknownRegistryFilter unknown_filter,
|
| + PrivateRegistryFilter private_filter) {
|
| url::CanonHostInfo host_info;
|
| const std::string canon_host(CanonicalizeHost(host, &host_info));
|
| -
|
| - size_t rcd_length;
|
| - switch (host_info.family) {
|
| - case url::CanonHostInfo::IPV4:
|
| - case url::CanonHostInfo::IPV6:
|
| - // IP addresses don't have R.C.D.'s.
|
| - return false;
|
| - case url::CanonHostInfo::BROKEN:
|
| - // Host is not canonicalizable. Fall back to the slower "permissive"
|
| - // version.
|
| - rcd_length =
|
| - PermissiveGetHostRegistryLength(host, unknown_filter, private_filter);
|
| - break;
|
| - case url::CanonHostInfo::NEUTRAL:
|
| - rcd_length =
|
| - GetRegistryLengthImpl(canon_host, unknown_filter, private_filter);
|
| - break;
|
| - default:
|
| - NOTREACHED();
|
| - return false;
|
| - }
|
| - return (rcd_length != 0) && (rcd_length != std::string::npos);
|
| -}
|
| -
|
| -size_t GetCanonicalHostRegistryLength(base::StringPiece canon_host,
|
| - UnknownRegistryFilter unknown_filter,
|
| - PrivateRegistryFilter private_filter) {
|
| -#ifndef NDEBUG
|
| - // Ensure passed-in host name is canonical.
|
| - url::CanonHostInfo host_info;
|
| - DCHECK_EQ(net::CanonicalizeHost(canon_host, &host_info), canon_host);
|
| -#endif
|
| -
|
| + if (canon_host.empty())
|
| + return std::string::npos;
|
| + if (host_info.IsIPAddress())
|
| + return 0;
|
| return GetRegistryLengthImpl(canon_host, unknown_filter, private_filter);
|
| -}
|
| -
|
| -size_t PermissiveGetHostRegistryLength(base::StringPiece host,
|
| - UnknownRegistryFilter unknown_filter,
|
| - PrivateRegistryFilter private_filter) {
|
| - return DoPermissiveGetHostRegistryLength<std::string>(host, unknown_filter,
|
| - private_filter);
|
| -}
|
| -
|
| -size_t PermissiveGetHostRegistryLength(base::StringPiece16 host,
|
| - UnknownRegistryFilter unknown_filter,
|
| - PrivateRegistryFilter private_filter) {
|
| - return DoPermissiveGetHostRegistryLength<base::string16>(host, unknown_filter,
|
| - private_filter);
|
| }
|
|
|
| void SetFindDomainGraph() {
|
|
|