1 /* 2 * Copyright (C) 2006, 2008, 2009 Apple 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 6 * are met: 7 * 1. Redistributions of source code must retain the above copyright 8 * notice, this list of conditions and the following disclaimer. 9 * 2. Redistributions in binary form must reproduce the above copyright 10 * notice, this list of conditions and the following disclaimer in the 11 * documentation and/or other materials provided with the distribution. 12 * 13 * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY 14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR 17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 19 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY 20 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 21 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 22 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 23 */ 24 25 #ifndef HTMLViewSourceDocument_h 26 #define HTMLViewSourceDocument_h 27 28 #include "core/html/HTMLDocument.h" 29 30 namespace WebCore { 31 32 class HTMLTableCellElement; 33 class HTMLTableSectionElement; 34 class HTMLToken; 35 36 class HTMLViewSourceDocument FINAL : public HTMLDocument { 37 public: 38 enum SourceAnnotation { 39 AnnotateSourceAsSafe, 40 AnnotateSourceAsXSS 41 }; 42 43 static PassRefPtrWillBeRawPtr<HTMLViewSourceDocument> create(const DocumentInit& initializer, const String& mimeType) 44 { 45 return adoptRefWillBeNoop(new HTMLViewSourceDocument(initializer, mimeType)); 46 } 47 48 void addSource(const String&, HTMLToken&, SourceAnnotation); 49 50 virtual void trace(Visitor*) OVERRIDE; 51 52 private: 53 HTMLViewSourceDocument(const DocumentInit&, const String& mimeType); 54 55 virtual PassRefPtrWillBeRawPtr<DocumentParser> createParser() OVERRIDE; 56 57 void processDoctypeToken(const String& source, HTMLToken&); 58 void processEndOfFileToken(const String& source, HTMLToken&); 59 void processTagToken(const String& source, HTMLToken&, SourceAnnotation); 60 void processCommentToken(const String& source, HTMLToken&); 61 void processCharacterToken(const String& source, HTMLToken&, SourceAnnotation); 62 63 void createContainingTable(); 64 PassRefPtrWillBeRawPtr<Element> addSpanWithClassName(const AtomicString&); 65 void addLine(const AtomicString& className); 66 void finishLine(); 67 void addText(const String& text, const AtomicString& className, SourceAnnotation = AnnotateSourceAsSafe); 68 int addRange(const String& source, int start, int end, const AtomicString& className, bool isLink = false, bool isAnchor = false, const AtomicString& link = nullAtom); 69 void maybeAddSpanForAnnotation(SourceAnnotation); 70 71 PassRefPtrWillBeRawPtr<Element> addLink(const AtomicString& url, bool isAnchor); 72 PassRefPtrWillBeRawPtr<Element> addBase(const AtomicString& href); 73 74 String m_type; 75 RefPtrWillBeMember<Element> m_current; 76 RefPtrWillBeMember<HTMLTableSectionElement> m_tbody; 77 RefPtrWillBeMember<HTMLTableCellElement> m_td; 78 int m_lineNumber; 79 }; 80 81 } 82 83 #endif // HTMLViewSourceDocument_h 84
