1 /*
2 * Copyright (C) 2010 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
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 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 INC. OR
17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26 #include "config.h"
27 #include "HTMLViewSourceParser.h"
28
29 #include "HTMLDocumentParser.h"
30 #include "HTMLNames.h"
31 #include "HTMLViewSourceDocument.h"
32
33 namespace WebCore {
34
HTMLViewSourceParser(HTMLViewSourceDocument * document)35 HTMLViewSourceParser::HTMLViewSourceParser(HTMLViewSourceDocument* document)
36 : DecodedDataDocumentParser(document)
37 , m_tokenizer(HTMLTokenizer::create(HTMLDocumentParser::usePreHTML5ParserQuirks(document)))
38 {
39 }
40
~HTMLViewSourceParser()41 HTMLViewSourceParser::~HTMLViewSourceParser()
42 {
43 }
44
insert(const SegmentedString &)45 void HTMLViewSourceParser::insert(const SegmentedString&)
46 {
47 ASSERT_NOT_REACHED();
48 }
49
pumpTokenizer()50 void HTMLViewSourceParser::pumpTokenizer()
51 {
52 while (true) {
53 m_sourceTracker.start(m_input, m_token);
54 if (!m_tokenizer->nextToken(m_input.current(), m_token))
55 break;
56 m_sourceTracker.end(m_input, m_token);
57
58 document()->addSource(sourceForToken(), m_token);
59 updateTokenizerState();
60 m_token.clear();
61 }
62 }
63
append(const SegmentedString & input)64 void HTMLViewSourceParser::append(const SegmentedString& input)
65 {
66 m_input.appendToEnd(input);
67 pumpTokenizer();
68 }
69
sourceForToken()70 String HTMLViewSourceParser::sourceForToken()
71 {
72 return m_sourceTracker.sourceForToken(m_token);
73 }
74
updateTokenizerState()75 void HTMLViewSourceParser::updateTokenizerState()
76 {
77 // FIXME: The tokenizer should do this work for us.
78 if (m_token.type() != HTMLToken::StartTag)
79 return;
80
81 AtomicString tagName(m_token.name().data(), m_token.name().size());
82 m_tokenizer->updateStateFor(tagName, document()->frame());
83 }
84
finish()85 void HTMLViewSourceParser::finish()
86 {
87 if (!m_input.haveSeenEndOfFile())
88 m_input.markEndOfFile();
89 pumpTokenizer();
90 document()->finishedParsing();
91 }
92
finishWasCalled()93 bool HTMLViewSourceParser::finishWasCalled()
94 {
95 return m_input.haveSeenEndOfFile();
96 }
97
98 }
99