1/* 2 * Copyright (C) 2014 Google Inc. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions are 6 * met: 7 * 8 * * Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * * Neither the name of Google Inc. nor the names of its 11 * contributors may be used to endorse or promote products derived from 12 * this software without specific prior written permission. 13 * 14 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 15 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 16 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 17 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 18 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 19 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 20 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 21 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 22 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 24 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 25 */ 26 27#ifndef DocumentStyleSheetCollector_h 28#define DocumentStyleSheetCollector_h 29 30#include "platform/heap/Handle.h" 31#include "wtf/HashSet.h" 32#include "wtf/RefPtr.h" 33#include "wtf/Vector.h" 34 35namespace blink { 36 37class CSSStyleSheet; 38class Document; 39class StyleSheet; 40class StyleSheetCollection; 41 42class DocumentStyleSheetCollector { 43 // This class contains references to two on-heap collections, therefore 44 // it's unhealthy to have it anywhere but on the stack, where stack 45 // scanning will keep them alive. 46 STACK_ALLOCATED(); 47public: 48 friend class ImportedDocumentStyleSheetCollector; 49 50 DocumentStyleSheetCollector(WillBeHeapVector<RefPtrWillBeMember<StyleSheet> >& sheetsForList, WillBeHeapVector<RefPtrWillBeMember<CSSStyleSheet> >& activeList, WillBeHeapHashSet<RawPtrWillBeMember<Document> >&); 51 ~DocumentStyleSheetCollector(); 52 53 void appendActiveStyleSheets(const WillBeHeapVector<RefPtrWillBeMember<CSSStyleSheet> >&); 54 void appendActiveStyleSheet(CSSStyleSheet*); 55 void appendSheetForList(StyleSheet*); 56 57 bool hasVisited(Document* document) const { return m_visitedDocuments.contains(document); } 58 void willVisit(Document* document) { m_visitedDocuments.add(document); } 59 60private: 61 WillBeHeapVector<RefPtrWillBeMember<StyleSheet> >& m_styleSheetsForStyleSheetList; 62 WillBeHeapVector<RefPtrWillBeMember<CSSStyleSheet> >& m_activeAuthorStyleSheets; 63 WillBeHeapHashSet<RawPtrWillBeMember<Document> >& m_visitedDocuments; 64}; 65 66class ActiveDocumentStyleSheetCollector FINAL : public DocumentStyleSheetCollector { 67public: 68 ActiveDocumentStyleSheetCollector(StyleSheetCollection&); 69private: 70 WillBeHeapHashSet<RawPtrWillBeMember<Document> > m_visitedDocuments; 71}; 72 73class ImportedDocumentStyleSheetCollector FINAL : public DocumentStyleSheetCollector { 74public: 75 ImportedDocumentStyleSheetCollector(DocumentStyleSheetCollector&, WillBeHeapVector<RefPtrWillBeMember<StyleSheet> >&); 76}; 77 78} // namespace blink 79 80#endif // DocumentStyleSheetCollector_h 81