#include "config.h"
#include "RuleSet.h"
#include "CSSFontSelector.h"
#include "CSSKeyframesRule.h"
#include "CSSSelector.h"
#include "CSSSelectorList.h"
#include "HTMLNames.h"
#include "MediaQueryEvaluator.h"
#include "SecurityOrigin.h"
#include "SelectorChecker.h"
#include "SelectorFilter.h"
#include "StyleResolver.h"
#include "StyleRule.h"
#include "StyleRuleImport.h"
#include "StyleSheetContents.h"
#include "ViewportStyleResolver.h"
#if ENABLE(VIDEO_TRACK)
#include "TextTrackCue.h"
#endif
namespace WebCore {
using namespace HTMLNames;
static inline MatchBasedOnRuleHash computeMatchBasedOnRuleHash(const CSSSelector& selector)
{
if (selector.tagHistory())
return MatchBasedOnRuleHash::None;
if (selector.match() == CSSSelector::Tag) {
const QualifiedName& tagQualifiedName = selector.tagQName();
const AtomicString& selectorNamespace = tagQualifiedName.namespaceURI();
if (selectorNamespace == starAtom || selectorNamespace == xhtmlNamespaceURI) {
if (tagQualifiedName == anyQName())
return MatchBasedOnRuleHash::Universal;
return MatchBasedOnRuleHash::ClassC;
}
return MatchBasedOnRuleHash::None;
}
if (SelectorChecker::isCommonPseudoClassSelector(&selector))
return MatchBasedOnRuleHash::ClassB;
if (selector.match() == CSSSelector::Id)
return MatchBasedOnRuleHash::ClassA;
if (selector.match() == CSSSelector::Class)
return MatchBasedOnRuleHash::ClassB;
return MatchBasedOnRuleHash::None;
}
static bool selectorCanMatchPseudoElement(const CSSSelector& rootSelector)
{
const CSSSelector* selector = &rootSelector;
do {
if (selector->matchesPseudoElement())
return true;
if (const CSSSelectorList* selectorList = selector->selectorList()) {
for (const CSSSelector* subSelector = selectorList->first(); subSelector; subSelector = CSSSelectorList::next(subSelector)) {
if (selectorCanMatchPseudoElement(*subSelector))
return true;
}
}
selector = selector->tagHistory();
} while (selector);
return false;
}
static inline bool isCommonAttributeSelectorAttribute(const QualifiedName& attribute)
{
return attribute == typeAttr || attribute == readonlyAttr;
}
static bool containsUncommonAttributeSelector(const CSSSelector& rootSelector, bool matchesRightmostElement)
{
const CSSSelector* selector = &rootSelector;
do {
if (selector->isAttributeSelector()) {
if (!isCommonAttributeSelectorAttribute(selector->attribute()) || !matchesRightmostElement)
return true;
}
if (const CSSSelectorList* selectorList = selector->selectorList()) {
for (const CSSSelector* subSelector = selectorList->first(); subSelector; subSelector = CSSSelectorList::next(subSelector)) {
if (containsUncommonAttributeSelector(*subSelector, matchesRightmostElement))
return true;
}
}
if (selector->relation() != CSSSelector::SubSelector)
matchesRightmostElement = false;
selector = selector->tagHistory();
} while (selector);
return false;
}
static inline bool containsUncommonAttributeSelector(const CSSSelector& rootSelector)
{
return containsUncommonAttributeSelector(rootSelector, true);
}
static inline PropertyWhitelistType determinePropertyWhitelistType(const AddRuleFlags addRuleFlags, const CSSSelector* selector)
{
if (addRuleFlags & RuleIsInRegionRule)
return PropertyWhitelistRegion;
#if ENABLE(VIDEO_TRACK)
for (const CSSSelector* component = selector; component; component = component->tagHistory()) {
if (component->match() == CSSSelector::PseudoElement && (component->pseudoElementType() == CSSSelector::PseudoElementCue || component->value() == TextTrackCue::cueShadowPseudoId()))
return PropertyWhitelistCue;
}
#else
UNUSED_PARAM(selector);
#endif
return PropertyWhitelistNone;
}
RuleData::RuleData(StyleRule* rule, unsigned selectorIndex, unsigned position, AddRuleFlags addRuleFlags)
: m_rule(rule)
, m_selectorIndex(selectorIndex)
, m_hasDocumentSecurityOrigin(addRuleFlags & RuleHasDocumentSecurityOrigin)
, m_position(position)
, m_matchBasedOnRuleHash(static_cast<unsigned>(computeMatchBasedOnRuleHash(*selector())))
, m_canMatchPseudoElement(selectorCanMatchPseudoElement(*selector()))
, m_containsUncommonAttributeSelector(WebCore::containsUncommonAttributeSelector(*selector()))
, m_linkMatchType(SelectorChecker::determineLinkMatchType(selector()))
, m_propertyWhitelistType(determinePropertyWhitelistType(addRuleFlags, selector()))
#if ENABLE(CSS_SELECTOR_JIT) && CSS_SELECTOR_JIT_PROFILING
, m_compiledSelectorUseCount(0)
#endif
{
ASSERT(m_position == position);
ASSERT(m_selectorIndex == selectorIndex);
SelectorFilter::collectIdentifierHashes(selector(), m_descendantSelectorIdentifierHashes, maximumIdentifierCount);
}
RuleSet::RuleSet()
{
}
RuleSet::~RuleSet()
{
}
void RuleSet::addToRuleSet(AtomicStringImpl* key, AtomRuleMap& map, const RuleData& ruleData)
{
if (!key)
return;
auto& rules = map.add(key, nullptr).iterator->value;
if (!rules)
rules = std::make_unique<RuleDataVector>();
rules->append(ruleData);
}
static unsigned rulesCountForName(const RuleSet::AtomRuleMap& map, AtomicStringImpl* name)
{
if (const auto* rules = map.get(name))
return rules->size();
return 0;
}
void RuleSet::addRule(StyleRule* rule, unsigned selectorIndex, AddRuleFlags addRuleFlags)
{
RuleData ruleData(rule, selectorIndex, m_ruleCount++, addRuleFlags);
m_features.collectFeatures(ruleData);
unsigned classBucketSize = 0;
const CSSSelector* tagSelector = nullptr;
const CSSSelector* classSelector = nullptr;
const CSSSelector* linkSelector = nullptr;
const CSSSelector* focusSelector = nullptr;
const CSSSelector* selector = ruleData.selector();
do {
if (selector->match() == CSSSelector::Id) {
addToRuleSet(selector->value().impl(), m_idRules, ruleData);
return;
}
#if ENABLE(VIDEO_TRACK)
if (selector->match() == CSSSelector::PseudoElement && selector->pseudoElementType() == CSSSelector::PseudoElementCue) {
m_cuePseudoRules.append(ruleData);
return;
}
#endif
if (selector->isCustomPseudoElement()) {
ruleData.disableSelectorFiltering();
addToRuleSet(selector->value().impl(), m_shadowPseudoElementRules, ruleData);
return;
}
if (selector->match() == CSSSelector::Class) {
AtomicStringImpl* className = selector->value().impl();
if (!classSelector) {
classSelector = selector;
classBucketSize = rulesCountForName(m_classRules, className);
} else if (classBucketSize) {
unsigned newClassBucketSize = rulesCountForName(m_classRules, className);
if (newClassBucketSize < classBucketSize) {
classSelector = selector;
classBucketSize = newClassBucketSize;
}
}
}
if (selector->match() == CSSSelector::Tag && selector->tagQName().localName() != starAtom)
tagSelector = selector;
if (SelectorChecker::isCommonPseudoClassSelector(selector)) {
switch (selector->pseudoClassType()) {
case CSSSelector::PseudoClassLink:
case CSSSelector::PseudoClassVisited:
case CSSSelector::PseudoClassAnyLink:
case CSSSelector::PseudoClassAnyLinkDeprecated:
linkSelector = selector;
break;
case CSSSelector::PseudoClassFocus:
focusSelector = selector;
break;
default:
ASSERT_NOT_REACHED();
}
}
if (selector->match() == CSSSelector::PseudoClass && selector->pseudoClassType() == CSSSelector::PseudoClassHost) {
m_hostPseudoClassRules.append(ruleData);
return;
}
if (selector->match() == CSSSelector::PseudoElement && selector->pseudoElementType() == CSSSelector::PseudoElementSlotted) {
ruleData.disableSelectorFiltering();
m_slottedPseudoElementRules.append(ruleData);
return;
}
if (selector->relation() != CSSSelector::SubSelector)
break;
selector = selector->tagHistory();
} while (selector);
if (classSelector) {
addToRuleSet(classSelector->value().impl(), m_classRules, ruleData);
return;
}
if (linkSelector) {
m_linkPseudoClassRules.append(ruleData);
return;
}
if (focusSelector) {
m_focusPseudoClassRules.append(ruleData);
return;
}
if (tagSelector) {
addToRuleSet(tagSelector->tagQName().localName().impl(), m_tagLocalNameRules, ruleData);
addToRuleSet(tagSelector->tagLowercaseLocalName().impl(), m_tagLowercaseLocalNameRules, ruleData);
return;
}
m_universalRules.append(ruleData);
}
void RuleSet::addPageRule(StyleRulePage* rule)
{
m_pageRules.append(rule);
}
void RuleSet::addRegionRule(StyleRuleRegion* regionRule, bool hasDocumentSecurityOrigin)
{
auto regionRuleSet = std::make_unique<RuleSet>();
regionRuleSet->m_ruleCount = m_ruleCount;
const Vector<RefPtr<StyleRuleBase>>& childRules = regionRule->childRules();
AddRuleFlags addRuleFlags = hasDocumentSecurityOrigin ? RuleHasDocumentSecurityOrigin : RuleHasNoSpecialState;
addRuleFlags = static_cast<AddRuleFlags>(addRuleFlags | RuleIsInRegionRule);
for (auto& childRule : childRules) {
if (is<StyleRule>(*childRule))
regionRuleSet->addStyleRule(downcast<StyleRule>(childRule.get()), addRuleFlags);
}
m_ruleCount = regionRuleSet->m_ruleCount;
m_regionSelectorsAndRuleSets.append(RuleSetSelectorPair(regionRule->selectorList().first(), WTFMove(regionRuleSet)));
}
void RuleSet::addChildRules(const Vector<RefPtr<StyleRuleBase>>& rules, const MediaQueryEvaluator& medium, StyleResolver* resolver, bool hasDocumentSecurityOrigin, bool isInitiatingElementInUserAgentShadowTree, AddRuleFlags addRuleFlags)
{
for (auto& rule : rules) {
if (is<StyleRule>(*rule))
addStyleRule(downcast<StyleRule>(rule.get()), addRuleFlags);
else if (is<StyleRulePage>(*rule))
addPageRule(downcast<StyleRulePage>(rule.get()));
else if (is<StyleRuleMedia>(*rule)) {
auto& mediaRule = downcast<StyleRuleMedia>(*rule);
if ((!mediaRule.mediaQueries() || medium.evaluate(*mediaRule.mediaQueries(), resolver)))
addChildRules(mediaRule.childRules(), medium, resolver, hasDocumentSecurityOrigin, isInitiatingElementInUserAgentShadowTree, addRuleFlags);
} else if (is<StyleRuleFontFace>(*rule) && resolver) {
resolver->document().fontSelector().addFontFaceRule(downcast<StyleRuleFontFace>(*rule.get()), isInitiatingElementInUserAgentShadowTree);
resolver->invalidateMatchedPropertiesCache();
} else if (is<StyleRuleKeyframes>(*rule) && resolver)
resolver->addKeyframeStyle(downcast<StyleRuleKeyframes>(*rule));
else if (is<StyleRuleSupports>(*rule) && downcast<StyleRuleSupports>(*rule).conditionIsSupported())
addChildRules(downcast<StyleRuleSupports>(*rule).childRules(), medium, resolver, hasDocumentSecurityOrigin, isInitiatingElementInUserAgentShadowTree, addRuleFlags);
#if ENABLE(CSS_REGIONS)
else if (is<StyleRuleRegion>(*rule) && resolver) {
addRegionRule(downcast<StyleRuleRegion>(rule.get()), hasDocumentSecurityOrigin);
}
#endif
#if ENABLE(CSS_DEVICE_ADAPTATION)
else if (is<StyleRuleViewport>(*rule) && resolver) {
resolver->viewportStyleResolver()->addViewportRule(downcast<StyleRuleViewport>(rule.get()));
}
#endif
}
}
void RuleSet::addRulesFromSheet(StyleSheetContents& sheet, const MediaQueryEvaluator& medium, StyleResolver* resolver)
{
for (auto& rule : sheet.importRules()) {
if (rule->styleSheet() && (!rule->mediaQueries() || medium.evaluate(*rule->mediaQueries(), resolver)))
addRulesFromSheet(*rule->styleSheet(), medium, resolver);
}
bool hasDocumentSecurityOrigin = resolver && resolver->document().securityOrigin()->canRequest(sheet.baseURL());
AddRuleFlags addRuleFlags = static_cast<AddRuleFlags>((hasDocumentSecurityOrigin ? RuleHasDocumentSecurityOrigin : 0));
bool isInitiatingElementInUserAgentShadowTree = false;
addChildRules(sheet.childRules(), medium, resolver, hasDocumentSecurityOrigin, isInitiatingElementInUserAgentShadowTree, addRuleFlags);
if (m_autoShrinkToFitEnabled)
shrinkToFit();
}
void RuleSet::addStyleRule(StyleRule* rule, AddRuleFlags addRuleFlags)
{
for (size_t selectorIndex = 0; selectorIndex != notFound; selectorIndex = rule->selectorList().indexOfNextSelectorAfter(selectorIndex))
addRule(rule, selectorIndex, addRuleFlags);
}
bool RuleSet::hasShadowPseudoElementRules() const
{
if (!m_shadowPseudoElementRules.isEmpty())
return true;
#if ENABLE(VIDEO_TRACK)
if (!m_cuePseudoRules.isEmpty())
return true;
#endif
return false;
}
void RuleSet::copyShadowPseudoElementRulesFrom(const RuleSet& other)
{
for (auto& keyValuePair : other.m_shadowPseudoElementRules)
m_shadowPseudoElementRules.add(keyValuePair.key, std::make_unique<RuleDataVector>(*keyValuePair.value));
#if ENABLE(VIDEO_TRACK)
for (auto& cue : other.m_cuePseudoRules)
m_cuePseudoRules.append(cue);
#endif
}
static inline void shrinkMapVectorsToFit(RuleSet::AtomRuleMap& map)
{
for (auto& vector : map.values())
vector->shrinkToFit();
}
void RuleSet::shrinkToFit()
{
shrinkMapVectorsToFit(m_idRules);
shrinkMapVectorsToFit(m_classRules);
shrinkMapVectorsToFit(m_tagLocalNameRules);
shrinkMapVectorsToFit(m_tagLowercaseLocalNameRules);
shrinkMapVectorsToFit(m_shadowPseudoElementRules);
m_linkPseudoClassRules.shrinkToFit();
#if ENABLE(VIDEO_TRACK)
m_cuePseudoRules.shrinkToFit();
#endif
m_hostPseudoClassRules.shrinkToFit();
m_slottedPseudoElementRules.shrinkToFit();
m_focusPseudoClassRules.shrinkToFit();
m_universalRules.shrinkToFit();
m_pageRules.shrinkToFit();
m_features.shrinkToFit();
m_regionSelectorsAndRuleSets.shrinkToFit();
}
}