| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (C) 2010 Google, Inc. All Rights Reserved. | 2 * Copyright (C) 2010 Google, Inc. All Rights Reserved. |
| 3 * | 3 * |
| 4 * Redistribution and use in source and binary forms, with or without | 4 * Redistribution and use in source and binary forms, with or without |
| 5 * modification, are permitted provided that the following conditions | 5 * modification, are permitted provided that the following conditions |
| 6 * are met: | 6 * are met: |
| 7 * 1. Redistributions of source code must retain the above copyright | 7 * 1. Redistributions of source code must retain the above copyright |
| 8 * notice, this list of conditions and the following disclaimer. | 8 * notice, this list of conditions and the following disclaimer. |
| 9 * 2. Redistributions in binary form must reproduce the above copyright | 9 * 2. Redistributions in binary form must reproduce the above copyright |
| 10 * notice, this list of conditions and the following disclaimer in the | 10 * notice, this list of conditions and the following disclaimer in the |
| 11 * documentation and/or other materials provided with the distribution. | 11 * documentation and/or other materials provided with the distribution. |
| 12 * | 12 * |
| 13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY | 13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY |
| 14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | 14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| 15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR | 15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR |
| 16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR | 16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR |
| 17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, | 17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
| 18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, | 18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
| 19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR | 19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR |
| 20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY | 20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY |
| 21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | 21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| 22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | 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. | 23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 24 */ | 24 */ |
| 25 | 25 |
| 26 #include "config.h" | 26 #include "config.h" |
| 27 #include "core/html/parser/HTMLViewSourceParser.h" | 27 #include "core/html/parser/HTMLViewSourceParser.h" |
| 28 | 28 |
| 29 #include "core/dom/DOMImplementation.h" | 29 #include "core/dom/DOMImplementation.h" |
| 30 #include "core/html/parser/HTMLParserIdioms.h" | |
| 31 #include "core/html/parser/HTMLParserOptions.h" | 30 #include "core/html/parser/HTMLParserOptions.h" |
| 32 #include "core/html/parser/HTMLToken.h" | 31 #include "core/html/parser/HTMLToken.h" |
| 33 | 32 |
| 34 namespace WebCore { | 33 namespace WebCore { |
| 35 | 34 |
| 36 HTMLViewSourceParser::HTMLViewSourceParser(HTMLViewSourceDocument* document, con
st String& mimeType) | 35 HTMLViewSourceParser::HTMLViewSourceParser(HTMLViewSourceDocument* document, con
st String& mimeType) |
| 37 : DecodedDataDocumentParser(document) | 36 : DecodedDataDocumentParser(document) |
| 38 , m_tokenizer(HTMLTokenizer::create(HTMLParserOptions(document))) | 37 , m_tokenizer(HTMLTokenizer::create(HTMLParserOptions(document))) |
| 39 { | 38 { |
| 40 if (mimeType != "text/html" && !DOMImplementation::isXMLMIMEType(mimeType)) | 39 if (mimeType != "text/html" && !DOMImplementation::isXMLMIMEType(mimeType)) |
| 41 m_tokenizer->setState(HTMLTokenizer::PLAINTEXTState); | 40 m_tokenizer->setState(HTMLTokenizer::PLAINTEXTState); |
| 42 } | 41 } |
| 43 | 42 |
| 44 void HTMLViewSourceParser::pumpTokenizer() | 43 void HTMLViewSourceParser::pumpTokenizer() |
| 45 { | 44 { |
| 46 while (true) { | 45 while (true) { |
| 47 m_sourceTracker.start(m_input.current(), m_tokenizer.get(), m_token); | 46 m_sourceTracker.start(m_input.current(), m_tokenizer.get(), m_token); |
| 48 if (!m_tokenizer->nextToken(m_input.current(), m_token)) | 47 if (!m_tokenizer->nextToken(m_input.current(), m_token)) |
| 49 return; | 48 return; |
| 50 m_sourceTracker.end(m_input.current(), m_tokenizer.get(), m_token); | 49 m_sourceTracker.end(m_input.current(), m_tokenizer.get(), m_token); |
| 51 | 50 |
| 52 document()->addSource(m_sourceTracker.sourceForToken(m_token), m_token); | 51 document()->addSource(m_sourceTracker.sourceForToken(m_token), m_token); |
| 53 | 52 |
| 54 // FIXME: The tokenizer should do this work for us. | 53 // FIXME: The tokenizer should do this work for us. |
| 55 if (m_token.type() == HTMLToken::StartTag) | 54 if (m_token.type() == HTMLToken::StartTag) |
| 56 m_tokenizer->updateStateFor(attemptStaticStringCreation(m_token.name
(), Likely8Bit)); | 55 m_tokenizer->updateStateFor(AtomicString(m_token.name())); |
| 57 m_token.clear(); | 56 m_token.clear(); |
| 58 } | 57 } |
| 59 } | 58 } |
| 60 | 59 |
| 61 void HTMLViewSourceParser::append(PassRefPtr<StringImpl> input) | 60 void HTMLViewSourceParser::append(PassRefPtr<StringImpl> input) |
| 62 { | 61 { |
| 63 m_input.appendToEnd(String(input)); | 62 m_input.appendToEnd(String(input)); |
| 64 pumpTokenizer(); | 63 pumpTokenizer(); |
| 65 } | 64 } |
| 66 | 65 |
| 67 void HTMLViewSourceParser::finish() | 66 void HTMLViewSourceParser::finish() |
| 68 { | 67 { |
| 69 if (!m_input.haveSeenEndOfFile()) | 68 if (!m_input.haveSeenEndOfFile()) |
| 70 m_input.markEndOfFile(); | 69 m_input.markEndOfFile(); |
| 71 pumpTokenizer(); | 70 pumpTokenizer(); |
| 72 document()->finishedParsing(); | 71 document()->finishedParsing(); |
| 73 } | 72 } |
| 74 | 73 |
| 75 } | 74 } |
| OLD | NEW |