1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127
|
/*
* Copyright (C) 2024 Apple Inc. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
* THE POSSIBILITY OF SUCH DAMAGE.
*/
#include "config.h"
#include "PublicSuffixStore.h"
#include <wtf/CrossThreadCopier.h>
#include <wtf/NeverDestroyed.h>
namespace WebCore {
PublicSuffixStore& PublicSuffixStore::singleton()
{
static LazyNeverDestroyed<PublicSuffixStore> store;
static std::once_flag flag;
std::call_once(flag, [&] {
store.construct();
});
return store.get();
}
void PublicSuffixStore::clearHostTopPrivatelyControlledDomainCache()
{
Locker locker { m_HostTopPrivatelyControlledDomainCacheLock };
m_hostTopPrivatelyControlledDomainCache.clear();
}
bool PublicSuffixStore::isPublicSuffix(StringView domain) const
{
return platformIsPublicSuffix(domain);
}
PublicSuffix PublicSuffixStore::publicSuffix(const URL& url) const
{
if (!url.isValid())
return { };
auto host = url.host();
if (URL::hostIsIPAddress(host))
return { };
size_t separatorPosition;
for (unsigned labelStart = 0; (separatorPosition = host.find('.', labelStart)) != notFound; labelStart = separatorPosition + 1) {
auto candidate = host.substring(separatorPosition + 1);
if (isPublicSuffix(candidate))
return PublicSuffix::fromRawString(candidate.toString());
}
return { };
}
String PublicSuffixStore::topPrivatelyControlledDomain(StringView host) const
{
// FIXME: if host is a URL, we could drop these checks.
if (host.isEmpty())
return { };
if (!host.containsOnlyASCII())
return host.toString();
Locker locker { m_HostTopPrivatelyControlledDomainCacheLock };
auto result = m_hostTopPrivatelyControlledDomainCache.ensure<ASCIICaseInsensitiveStringViewHashTranslator>(host, [&] {
const auto lowercaseHost = host.convertToASCIILowercase();
if (lowercaseHost == "localhost"_s || URL::hostIsIPAddress(lowercaseHost))
return lowercaseHost;
return platformTopPrivatelyControlledDomain(lowercaseHost);
}).iterator->value.isolatedCopy();
constexpr auto maxHostTopPrivatelyControlledDomainCache = 128;
if (m_hostTopPrivatelyControlledDomainCache.size() > maxHostTopPrivatelyControlledDomainCache)
m_hostTopPrivatelyControlledDomainCache.remove(m_hostTopPrivatelyControlledDomainCache.random());
return result;
}
String PublicSuffixStore::topPrivatelyControlledDomainWithoutPublicSuffix(StringView host) const
{
auto topPrivatelyControlledDomain = this->topPrivatelyControlledDomain(host);
if (topPrivatelyControlledDomain.isEmpty())
return { };
return domainWithoutPublicSuffix(topPrivatelyControlledDomain);
}
String PublicSuffixStore::domainWithoutPublicSuffix(StringView domain) const
{
if (URL::hostIsIPAddress(domain))
return domain.toString();
size_t separatorPosition;
for (unsigned labelStart = 0; (separatorPosition = domain.find('.', labelStart)) != notFound; labelStart = separatorPosition + 1) {
auto candidate = domain.substring(separatorPosition + 1);
if (isPublicSuffix(candidate)) {
if (separatorPosition > 0 && domain.length() > separatorPosition) {
auto domainWithoutSuffix = domain.substring(0, separatorPosition);
return domainWithoutSuffix.toString();
}
}
}
return domain.toString();
}
} // namespace WebCore
|