|
1 /* This Source Code Form is subject to the terms of the Mozilla Public |
|
2 * License, v. 2.0. If a copy of the MPL was not distributed with this |
|
3 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ |
|
4 |
|
5 #include "nsHtml5StringParser.h" |
|
6 #include "nsHtml5TreeOpExecutor.h" |
|
7 #include "nsHtml5TreeBuilder.h" |
|
8 #include "nsHtml5Tokenizer.h" |
|
9 #include "nsIContent.h" |
|
10 #include "nsIDocument.h" |
|
11 #include "nsIDOMDocumentFragment.h" |
|
12 #include "nsHtml5DependentUTF16Buffer.h" |
|
13 |
|
14 NS_IMPL_ISUPPORTS0(nsHtml5StringParser) |
|
15 |
|
16 nsHtml5StringParser::nsHtml5StringParser() |
|
17 : mBuilder(new nsHtml5OplessBuilder()) |
|
18 , mTreeBuilder(new nsHtml5TreeBuilder(mBuilder)) |
|
19 , mTokenizer(new nsHtml5Tokenizer(mTreeBuilder, false)) |
|
20 { |
|
21 MOZ_COUNT_CTOR(nsHtml5StringParser); |
|
22 mTokenizer->setInterner(&mAtomTable); |
|
23 } |
|
24 |
|
25 nsHtml5StringParser::~nsHtml5StringParser() |
|
26 { |
|
27 MOZ_COUNT_DTOR(nsHtml5StringParser); |
|
28 } |
|
29 |
|
30 nsresult |
|
31 nsHtml5StringParser::ParseFragment(const nsAString& aSourceBuffer, |
|
32 nsIContent* aTargetNode, |
|
33 nsIAtom* aContextLocalName, |
|
34 int32_t aContextNamespace, |
|
35 bool aQuirks, |
|
36 bool aPreventScriptExecution) |
|
37 { |
|
38 NS_ENSURE_TRUE(aSourceBuffer.Length() <= INT32_MAX, |
|
39 NS_ERROR_OUT_OF_MEMORY); |
|
40 |
|
41 nsIDocument* doc = aTargetNode->OwnerDoc(); |
|
42 nsIURI* uri = doc->GetDocumentURI(); |
|
43 NS_ENSURE_TRUE(uri, NS_ERROR_NOT_AVAILABLE); |
|
44 |
|
45 mTreeBuilder->setFragmentContext(aContextLocalName, |
|
46 aContextNamespace, |
|
47 aTargetNode, |
|
48 aQuirks); |
|
49 |
|
50 #ifdef DEBUG |
|
51 if (!aPreventScriptExecution) { |
|
52 NS_ASSERTION(!aTargetNode->IsInDoc(), |
|
53 "If script execution isn't prevented, " |
|
54 "the target node must not be in doc."); |
|
55 nsCOMPtr<nsIDOMDocumentFragment> domFrag = do_QueryInterface(aTargetNode); |
|
56 NS_ASSERTION(domFrag, |
|
57 "If script execution isn't prevented, must parse to DOM fragment."); |
|
58 } |
|
59 #endif |
|
60 |
|
61 mTreeBuilder->SetPreventScriptExecution(aPreventScriptExecution); |
|
62 |
|
63 return Tokenize(aSourceBuffer, doc, true); |
|
64 } |
|
65 |
|
66 nsresult |
|
67 nsHtml5StringParser::ParseDocument(const nsAString& aSourceBuffer, |
|
68 nsIDocument* aTargetDoc, |
|
69 bool aScriptingEnabledForNoscriptParsing) |
|
70 { |
|
71 MOZ_ASSERT(!aTargetDoc->GetFirstChild()); |
|
72 |
|
73 NS_ENSURE_TRUE(aSourceBuffer.Length() <= INT32_MAX, |
|
74 NS_ERROR_OUT_OF_MEMORY); |
|
75 |
|
76 mTreeBuilder->setFragmentContext(nullptr, |
|
77 kNameSpaceID_None, |
|
78 nullptr, |
|
79 false); |
|
80 |
|
81 mTreeBuilder->SetPreventScriptExecution(true); |
|
82 |
|
83 return Tokenize(aSourceBuffer, aTargetDoc, aScriptingEnabledForNoscriptParsing); |
|
84 } |
|
85 |
|
86 nsresult |
|
87 nsHtml5StringParser::Tokenize(const nsAString& aSourceBuffer, |
|
88 nsIDocument* aDocument, |
|
89 bool aScriptingEnabledForNoscriptParsing) { |
|
90 |
|
91 nsIURI* uri = aDocument->GetDocumentURI(); |
|
92 |
|
93 mBuilder->Init(aDocument, uri, nullptr, nullptr); |
|
94 |
|
95 mBuilder->SetParser(this); |
|
96 mBuilder->SetNodeInfoManager(aDocument->NodeInfoManager()); |
|
97 |
|
98 // Mark the parser as *not* broken by passing NS_OK |
|
99 nsresult rv = mBuilder->MarkAsBroken(NS_OK); |
|
100 |
|
101 mTreeBuilder->setScriptingEnabled(aScriptingEnabledForNoscriptParsing); |
|
102 mTreeBuilder->setIsSrcdocDocument(aDocument->IsSrcdocDocument()); |
|
103 mBuilder->Start(); |
|
104 mTokenizer->start(); |
|
105 if (!aSourceBuffer.IsEmpty()) { |
|
106 bool lastWasCR = false; |
|
107 nsHtml5DependentUTF16Buffer buffer(aSourceBuffer); |
|
108 while (buffer.hasMore()) { |
|
109 buffer.adjust(lastWasCR); |
|
110 lastWasCR = false; |
|
111 if (buffer.hasMore()) { |
|
112 lastWasCR = mTokenizer->tokenizeBuffer(&buffer); |
|
113 if (NS_FAILED(rv = mBuilder->IsBroken())) { |
|
114 break; |
|
115 } |
|
116 } |
|
117 } |
|
118 } |
|
119 mTokenizer->eof(); |
|
120 mTokenizer->end(); |
|
121 mBuilder->Finish(); |
|
122 mAtomTable.Clear(); |
|
123 return rv; |
|
124 } |