| 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 |
| (...skipping 12 matching lines...) Expand all Loading... |
| 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 "core/html/parser/HTMLViewSourceParser.h" | 26 #include "core/html/parser/HTMLViewSourceParser.h" |
| 27 | 27 |
| 28 #include "core/dom/DOMImplementation.h" | 28 #include "core/dom/DOMImplementation.h" |
| 29 #include "core/html/parser/HTMLParserIdioms.h" | 29 #include "core/html/parser/HTMLParserIdioms.h" |
| 30 #include "core/html/parser/HTMLParserOptions.h" | 30 #include "core/html/parser/HTMLParserOptions.h" |
| 31 #include "core/html/parser/HTMLToken.h" | 31 #include "core/html/parser/HTMLToken.h" |
| 32 #include "core/html/parser/XSSAuditorDelegate.h" | 32 #include "core/html/parser/XSSAuditorDelegate.h" |
| 33 #include <memory> |
| 33 | 34 |
| 34 namespace blink { | 35 namespace blink { |
| 35 | 36 |
| 36 HTMLViewSourceParser::HTMLViewSourceParser(HTMLViewSourceDocument& document, con
st String& mimeType) | 37 HTMLViewSourceParser::HTMLViewSourceParser(HTMLViewSourceDocument& document, con
st String& mimeType) |
| 37 : DecodedDataDocumentParser(document) | 38 : DecodedDataDocumentParser(document) |
| 38 , m_tokenizer(HTMLTokenizer::create(HTMLParserOptions(&document))) | 39 , m_tokenizer(HTMLTokenizer::create(HTMLParserOptions(&document))) |
| 39 { | 40 { |
| 40 if (mimeType != "text/html" && !DOMImplementation::isXMLMIMEType(mimeType)) | 41 if (mimeType != "text/html" && !DOMImplementation::isXMLMIMEType(mimeType)) |
| 41 m_tokenizer->setState(HTMLTokenizer::PLAINTEXTState); | 42 m_tokenizer->setState(HTMLTokenizer::PLAINTEXTState); |
| 42 } | 43 } |
| 43 | 44 |
| 44 void HTMLViewSourceParser::pumpTokenizer() | 45 void HTMLViewSourceParser::pumpTokenizer() |
| 45 { | 46 { |
| 46 m_xssAuditor.init(document(), 0); | 47 m_xssAuditor.init(document(), 0); |
| 47 | 48 |
| 48 while (true) { | 49 while (true) { |
| 49 m_sourceTracker.start(m_input.current(), m_tokenizer.get(), m_token); | 50 m_sourceTracker.start(m_input.current(), m_tokenizer.get(), m_token); |
| 50 if (!m_tokenizer->nextToken(m_input.current(), m_token)) | 51 if (!m_tokenizer->nextToken(m_input.current(), m_token)) |
| 51 return; | 52 return; |
| 52 m_sourceTracker.end(m_input.current(), m_tokenizer.get(), m_token); | 53 m_sourceTracker.end(m_input.current(), m_tokenizer.get(), m_token); |
| 53 | 54 |
| 54 OwnPtr<XSSInfo> xssInfo = m_xssAuditor.filterToken(FilterTokenRequest(m_
token, m_sourceTracker, m_tokenizer->shouldAllowCDATA())); | 55 std::unique_ptr<XSSInfo> xssInfo = m_xssAuditor.filterToken(FilterTokenR
equest(m_token, m_sourceTracker, m_tokenizer->shouldAllowCDATA())); |
| 55 HTMLViewSourceDocument::SourceAnnotation annotation = xssInfo ? HTMLView
SourceDocument::AnnotateSourceAsXSS : HTMLViewSourceDocument::AnnotateSourceAsSa
fe; | 56 HTMLViewSourceDocument::SourceAnnotation annotation = xssInfo ? HTMLView
SourceDocument::AnnotateSourceAsXSS : HTMLViewSourceDocument::AnnotateSourceAsSa
fe; |
| 56 document()->addSource(m_sourceTracker.sourceForToken(m_token), m_token,
annotation); | 57 document()->addSource(m_sourceTracker.sourceForToken(m_token), m_token,
annotation); |
| 57 | 58 |
| 58 // FIXME: The tokenizer should do this work for us. | 59 // FIXME: The tokenizer should do this work for us. |
| 59 if (m_token.type() == HTMLToken::StartTag) | 60 if (m_token.type() == HTMLToken::StartTag) |
| 60 m_tokenizer->updateStateFor(attemptStaticStringCreation(m_token.name
(), Likely8Bit)); | 61 m_tokenizer->updateStateFor(attemptStaticStringCreation(m_token.name
(), Likely8Bit)); |
| 61 m_token.clear(); | 62 m_token.clear(); |
| 62 } | 63 } |
| 63 } | 64 } |
| 64 | 65 |
| 65 void HTMLViewSourceParser::append(const String& input) | 66 void HTMLViewSourceParser::append(const String& input) |
| 66 { | 67 { |
| 67 m_input.appendToEnd(input); | 68 m_input.appendToEnd(input); |
| 68 pumpTokenizer(); | 69 pumpTokenizer(); |
| 69 } | 70 } |
| 70 | 71 |
| 71 void HTMLViewSourceParser::finish() | 72 void HTMLViewSourceParser::finish() |
| 72 { | 73 { |
| 73 if (!m_input.haveSeenEndOfFile()) | 74 if (!m_input.haveSeenEndOfFile()) |
| 74 m_input.markEndOfFile(); | 75 m_input.markEndOfFile(); |
| 75 pumpTokenizer(); | 76 pumpTokenizer(); |
| 76 document()->finishedParsing(); | 77 document()->finishedParsing(); |
| 77 } | 78 } |
| 78 | 79 |
| 79 } // namespace blink | 80 } // namespace blink |
| OLD | NEW |