components/autofill_assistant/browser/url_utils.cc

// Copyright 2020 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 "components/autofill_assistant/browser/url_utils.h"

#include <algorithm>

#include "base/strings/string_piece.h"
#include "base/strings/string_util.h"
#include "net/base/registry_controlled_domains/registry_controlled_domain.h"

namespace {

bool IsInSubDomain(const GURL& url, const std::string& domain) {
  return base::EndsWith(base::StringPiece(url.host()),
                        base::StringPiece("." + domain),
                        base::CompareCase::INSENSITIVE_ASCII);
}

std::string GetRegistryControlledDomain(const GURL& signon_realm) {
  return net::registry_controlled_domains::GetDomainAndRegistry(
      signon_realm,
      net::registry_controlled_domains::INCLUDE_PRIVATE_REGISTRIES);
}

}  // namespace

namespace autofill_assistant {
namespace url_utils {

bool IsInDomainOrSubDomain(const GURL& url, const GURL& domain) {
  if (url.host() == domain.host()) {
    return true;
  }

  return IsInSubDomain(url, domain.host());
}

bool IsInDomainOrSubDomain(const GURL& url,
                           const std::vector<std::string>& allowed_domains) {
  return std::find_if(allowed_domains.begin(), allowed_domains.end(),
                      [url](const std::string& allowed_domain) {
                        return url.host() == allowed_domain ||
                               IsInSubDomain(url, allowed_domain);
                      }) != allowed_domains.end();
}

bool IsSamePublicSuffixDomain(const GURL& url1, const GURL& url2) {
  if (!url1.is_valid() || !url2.is_valid()) {
    return false;
  }

  if (url1.GetOrigin() == url2.GetOrigin()) {
    return true;
  }

  auto domain1 = GetRegistryControlledDomain(url1);
  auto domain2 = GetRegistryControlledDomain(url2);

  if (domain1.empty() || domain2.empty()) {
    return false;
  }

  return url1.scheme() == url2.scheme() && domain1 == domain2;
}

}  // namespace url_utils
}  // namespace autofill_assistant