blob: 2712d1e6d57c56baea2cd4961d31235edeb2d3ea [file] [log] [blame]
/*
* Copyright (C) 2008-2019 Apple Inc. All rights reserved.
* Copyright (C) 2008 David Smith <catfish.man@gmail.com>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public License
* along with this library; see the file COPYING.LIB. If not, write to
* the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
* Boston, MA 02110-1301, USA.
*
*/
#pragma once
#include "ChildNodeList.h"
#include "HTMLCollection.h"
#include "MutationObserverRegistration.h"
#include "QualifiedName.h"
#include "TagCollection.h"
#include <wtf/HashSet.h>
#include <wtf/StdLibExtras.h>
namespace WebCore {
class LabelsNodeList;
class LiveNodeList;
class NameNodeList;
class RadioNodeList;
template<typename ListType> struct NodeListTypeIdentifier;
DECLARE_ALLOCATOR_WITH_HEAP_IDENTIFIER(NodeListsNodeData);
class NodeListsNodeData {
WTF_MAKE_NONCOPYABLE(NodeListsNodeData);
WTF_MAKE_STRUCT_FAST_ALLOCATED_WITH_HEAP_IDENTIFIER(NodeListsNodeData);
public:
NodeListsNodeData() = default;
void clearChildNodeListCache()
{
if (m_childNodeList)
m_childNodeList->invalidateCache();
}
Ref<ChildNodeList> ensureChildNodeList(ContainerNode& node)
{
ASSERT(!m_emptyChildNodeList);
if (m_childNodeList)
return *m_childNodeList;
auto list = ChildNodeList::create(node);
m_childNodeList = list.ptr();
return list;
}
void removeChildNodeList(ChildNodeList* list)
{
ASSERT(m_childNodeList == list);
if (deleteThisAndUpdateNodeRareDataIfAboutToRemoveLastList(list->ownerNode()))
return;
m_childNodeList = nullptr;
}
Ref<EmptyNodeList> ensureEmptyChildNodeList(Node& node)
{
ASSERT(!m_childNodeList);
if (m_emptyChildNodeList)
return *m_emptyChildNodeList;
auto list = EmptyNodeList::create(node);
m_emptyChildNodeList = list.ptr();
return list;
}
void removeEmptyChildNodeList(EmptyNodeList* list)
{
ASSERT(m_emptyChildNodeList == list);
if (deleteThisAndUpdateNodeRareDataIfAboutToRemoveLastList(list->ownerNode()))
return;
m_emptyChildNodeList = nullptr;
}
struct NodeListCacheMapEntryHash {
static unsigned hash(const std::pair<unsigned char, AtomString>& entry)
{
return DefaultHash<AtomString>::Hash::hash(entry.second) + entry.first;
}
static bool equal(const std::pair<unsigned char, AtomString>& a, const std::pair<unsigned char, AtomString>& b) { return a.first == b.first && DefaultHash<AtomString>::Hash::equal(a.second, b.second); }
static const bool safeToCompareToEmptyOrDeleted = DefaultHash<AtomString>::Hash::safeToCompareToEmptyOrDeleted;
};
using NodeListCacheMap = HashMap<std::pair<unsigned char, AtomString>, LiveNodeList*, NodeListCacheMapEntryHash>;
using CollectionCacheMap = HashMap<std::pair<unsigned char, AtomString>, HTMLCollection*, NodeListCacheMapEntryHash>;
using TagCollectionNSCache = HashMap<QualifiedName, TagCollectionNS*>;
template<typename T, typename ContainerType>
ALWAYS_INLINE Ref<T> addCacheWithAtomName(ContainerType& container, const AtomString& name)
{
auto result = m_atomNameCaches.fastAdd(namedNodeListKey<T>(name), nullptr);
if (!result.isNewEntry)
return static_cast<T&>(*result.iterator->value);
auto list = T::create(container, name);
result.iterator->value = &list.get();
return list;
}
ALWAYS_INLINE Ref<TagCollectionNS> addCachedTagCollectionNS(ContainerNode& node, const AtomString& namespaceURI, const AtomString& localName)
{
auto result = m_tagCollectionNSCache.fastAdd(QualifiedName { nullAtom(), localName, namespaceURI }, nullptr);
if (!result.isNewEntry)
return *result.iterator->value;
auto list = TagCollectionNS::create(node, namespaceURI, localName);
result.iterator->value = list.ptr();
return list;
}
template<typename T, typename ContainerType>
ALWAYS_INLINE Ref<T> addCachedCollection(ContainerType& container, CollectionType collectionType, const AtomString& name)
{
auto result = m_cachedCollections.fastAdd(namedCollectionKey(collectionType, name), nullptr);
if (!result.isNewEntry)
return static_cast<T&>(*result.iterator->value);
auto list = T::create(container, collectionType, name);
result.iterator->value = &list.get();
return list;
}
template<typename T, typename ContainerType>
ALWAYS_INLINE Ref<T> addCachedCollection(ContainerType& container, CollectionType collectionType)
{
auto result = m_cachedCollections.fastAdd(namedCollectionKey(collectionType, starAtom()), nullptr);
if (!result.isNewEntry)
return static_cast<T&>(*result.iterator->value);
auto list = T::create(container, collectionType);
result.iterator->value = &list.get();
return list;
}
template<typename T>
T* cachedCollection(CollectionType collectionType)
{
return static_cast<T*>(m_cachedCollections.get(namedCollectionKey(collectionType, starAtom())));
}
template<typename NodeListType>
void removeCacheWithAtomName(NodeListType& list, const AtomString& name)
{
ASSERT(&list == m_atomNameCaches.get(namedNodeListKey<NodeListType>(name)));
if (deleteThisAndUpdateNodeRareDataIfAboutToRemoveLastList(list.ownerNode()))
return;
m_atomNameCaches.remove(namedNodeListKey<NodeListType>(name));
}
void removeCachedTagCollectionNS(HTMLCollection& collection, const AtomString& namespaceURI, const AtomString& localName)
{
QualifiedName name(nullAtom(), localName, namespaceURI);
ASSERT(&collection == m_tagCollectionNSCache.get(name));
if (deleteThisAndUpdateNodeRareDataIfAboutToRemoveLastList(collection.ownerNode()))
return;
m_tagCollectionNSCache.remove(name);
}
void removeCachedCollection(HTMLCollection* collection, const AtomString& name = starAtom())
{
ASSERT(collection == m_cachedCollections.get(namedCollectionKey(collection->type(), name)));
if (deleteThisAndUpdateNodeRareDataIfAboutToRemoveLastList(collection->ownerNode()))
return;
m_cachedCollections.remove(namedCollectionKey(collection->type(), name));
}
void invalidateCaches();
void invalidateCachesForAttribute(const QualifiedName& attrName);
void adoptTreeScope()
{
invalidateCaches();
}
void adoptDocument(Document& oldDocument, Document& newDocument)
{
if (&oldDocument == &newDocument) {
invalidateCaches();
return;
}
for (auto& cache : m_atomNameCaches.values())
cache->invalidateCacheForDocument(oldDocument);
for (auto& list : m_tagCollectionNSCache.values()) {
ASSERT(!list->isRootedAtDocument());
list->invalidateCacheForDocument(oldDocument);
}
for (auto& collection : m_cachedCollections.values())
collection->invalidateCacheForDocument(oldDocument);
}
private:
std::pair<unsigned char, AtomString> namedCollectionKey(CollectionType type, const AtomString& name)
{
return std::pair<unsigned char, AtomString>(type, name);
}
template<typename NodeListType>
std::pair<unsigned char, AtomString> namedNodeListKey(const AtomString& name)
{
return std::pair<unsigned char, AtomString>(NodeListTypeIdentifier<NodeListType>::value(), name);
}
bool deleteThisAndUpdateNodeRareDataIfAboutToRemoveLastList(Node&);
// These two are currently mutually exclusive and could be unioned. Not very important as this class is large anyway.
ChildNodeList* m_childNodeList { nullptr };
EmptyNodeList* m_emptyChildNodeList { nullptr };
NodeListCacheMap m_atomNameCaches;
TagCollectionNSCache m_tagCollectionNSCache;
CollectionCacheMap m_cachedCollections;
};
class NodeMutationObserverData {
WTF_MAKE_NONCOPYABLE(NodeMutationObserverData); WTF_MAKE_FAST_ALLOCATED;
public:
Vector<std::unique_ptr<MutationObserverRegistration>> registry;
HashSet<MutationObserverRegistration*> transientRegistry;
NodeMutationObserverData() { }
};
DECLARE_ALLOCATOR_WITH_HEAP_IDENTIFIER(NodeRareData);
class NodeRareData {
WTF_MAKE_NONCOPYABLE(NodeRareData);
WTF_MAKE_STRUCT_FAST_ALLOCATED_WITH_HEAP_IDENTIFIER(NodeRareData);
public:
#if defined(DUMP_NODE_STATISTICS) && DUMP_NODE_STATISTICS
enum class UseType : uint16_t {
ConnectedFrameCount = 1 << 0,
NodeList = 1 << 1,
MutationObserver = 1 << 2,
TabIndex = 1 << 3,
MinimumSize = 1 << 4,
ScrollingPosition = 1 << 5,
ComputedStyle = 1 << 6,
Dataset = 1 << 7,
ClassList = 1 << 8,
ShadowRoot = 1 << 9,
CustomElementQueue = 1 << 10,
AttributeMap = 1 << 11,
InteractionObserver = 1 << 12,
PseudoElements = 1 << 13,
};
#endif
enum class Type { Element, Node };
NodeRareData(Type type = Type::Node)
: m_connectedFrameCount(0)
, m_isElementRareData(type == Type::Element)
{
}
bool isElementRareData() { return m_isElementRareData; }
void clearNodeLists() { m_nodeLists = nullptr; }
NodeListsNodeData* nodeLists() const { return m_nodeLists.get(); }
NodeListsNodeData& ensureNodeLists()
{
if (!m_nodeLists)
m_nodeLists = makeUnique<NodeListsNodeData>();
return *m_nodeLists;
}
NodeMutationObserverData* mutationObserverData() { return m_mutationObserverData.get(); }
NodeMutationObserverData& ensureMutationObserverData()
{
if (!m_mutationObserverData)
m_mutationObserverData = makeUnique<NodeMutationObserverData>();
return *m_mutationObserverData;
}
unsigned connectedSubframeCount() const { return m_connectedFrameCount; }
void incrementConnectedSubframeCount(unsigned amount)
{
m_connectedFrameCount += amount;
}
void decrementConnectedSubframeCount(unsigned amount)
{
ASSERT(m_connectedFrameCount);
ASSERT(amount <= m_connectedFrameCount);
m_connectedFrameCount -= amount;
}
#if DUMP_NODE_STATISTICS
OptionSet<UseType> useTypes() const
{
OptionSet<UseType> result;
if (m_connectedFrameCount)
result.add(UseType::ConnectedFrameCount);
if (m_nodeLists)
result.add(UseType::NodeList);
if (m_mutationObserverData)
result.add(UseType::MutationObserver);
return result;
}
#endif
private:
unsigned m_connectedFrameCount : 31; // Must fit Page::maxNumberOfFrames.
unsigned m_isElementRareData : 1;
std::unique_ptr<NodeListsNodeData> m_nodeLists;
std::unique_ptr<NodeMutationObserverData> m_mutationObserverData;
};
template<> struct NodeListTypeIdentifier<NameNodeList> {
static constexpr unsigned char value() { return 0; }
};
template<> struct NodeListTypeIdentifier<RadioNodeList> {
static constexpr unsigned char value() { return 1; }
};
template<> struct NodeListTypeIdentifier<LabelsNodeList> {
static constexpr unsigned char value() { return 2; }
};
inline bool NodeListsNodeData::deleteThisAndUpdateNodeRareDataIfAboutToRemoveLastList(Node& ownerNode)
{
ASSERT(ownerNode.nodeLists() == this);
size_t listsCount = (m_childNodeList ? 1 : 0)
+ (m_emptyChildNodeList ? 1 : 0)
+ m_atomNameCaches.size()
+ m_tagCollectionNSCache.size()
+ m_cachedCollections.size();
if (listsCount != 1)
return false;
ownerNode.clearNodeLists();
return true;
}
inline NodeRareData& Node::ensureRareData()
{
if (!hasRareData())
materializeRareData();
return *rareData();
}
} // namespace WebCore