Depend initial parser state in fragment case with scripting flag of context element

Signed-off-by: Simon Wülker <[email protected]>

Author: simonwuelker

html5ever/src/driver.rs

@@ -61,12 +61,13 @@ pub fn parse_fragment<Sink>(
     opts: ParseOpts,
     context_name: QualName,
     context_attrs: Vec<Attribute>,
+    context_element_allows_scripting: bool,
 ) -> Parser<Sink>
 where
     Sink: TreeSink,
 {
     let context_elem = create_element(&sink, context_name, context_attrs);
-    parse_fragment_for_element(sink, opts, context_elem, None)
+    parse_fragment_for_element(sink, opts, context_elem, context_element_allows_scripting, None)
 }
 
 /// Like `parse_fragment`, but with an existing context element
@@ -75,19 +76,20 @@ pub fn parse_fragment_for_element<Sink>(
     sink: Sink,
     opts: ParseOpts,
     context_element: Sink::Handle,
+    context_element_allows_scripting: bool,
     form_element: Option<Sink::Handle>,
 ) -> Parser<Sink>
 where
     Sink: TreeSink,
 {
-    let tb = TreeBuilder::new_for_fragment(sink, context_element, form_element, opts.tree_builder);
-    let tok_opts = TokenizerOpts {
-        initial_state: Some(tb.tokenizer_state_for_context_elem()),
+    let tree_builder = TreeBuilder::new_for_fragment(sink, context_element, form_element, opts.tree_builder);
+    let tokenizer_options = TokenizerOpts {
+        initial_state: Some(tree_builder.tokenizer_state_for_context_elem(context_element_allows_scripting)),
         ..opts.tokenizer
     };
-    let tok = Tokenizer::new(tb, tok_opts);
+    let tokenizer = Tokenizer::new(tree_builder, tokenizer_options);
     Parser {
-        tokenizer: tok,
+        tokenizer,
         input_buffer: BufferQueue::default(),
     }
 }

html5ever/src/tree_builder/mod.rs

@@ -223,7 +223,7 @@ where
 
     // https://html.spec.whatwg.org/multipage/#concept-frag-parse-context
     // Step 4. Set the state of the HTML parser's tokenization stage as follows:
-    pub fn tokenizer_state_for_context_elem(&self) -> tok_state::State {
+    pub fn tokenizer_state_for_context_elem(&self, context_element_allows_scripting: bool) -> tok_state::State {
         let context_elem = self.context_elem.borrow();
         let elem = context_elem.as_ref().expect("no context element");
         let elem_name = self.sink.elem_name(elem);
@@ -246,7 +246,7 @@ where
             local_name!("script") => tok_state::RawData(tok_state::ScriptData),
 
             local_name!("noscript") => {
-                if self.opts.scripting_enabled {
+                if context_element_allows_scripting {
                     tok_state::RawData(tok_state::Rawtext)
                 } else {
                     tok_state::Data

markup5ever/Cargo.toml

@@ -16,4 +16,4 @@ path = "lib.rs"
 [dependencies]
 web_atoms = { version = "0.1", path = "../web_atoms" }
 tendril = "0.4"
-log = "0.4"
+log = "0.4"

rcdom/tests/html-serializer.rs

@@ -94,6 +94,7 @@ fn parse_and_serialize(input: StrTendril) -> StrTendril {
         ParseOpts::default(),
         QualName::new(None, ns!(html), local_name!("body")),
         vec![],
+        true,
     )
     .one(input);
     let inner: SerializableHandle = dom.document.children.borrow()[0].clone().into();
@@ -252,6 +253,7 @@ fn deep_tree() {
         ParseOpts::default(),
         QualName::new(None, ns!(html), local_name!("div")),
         vec![],
+        true
     );
     let src = "<b>".repeat(60_000);
     let dom = parser.one(src);

rcdom/tests/html-tree-builder.rs

@@ -225,7 +225,7 @@ fn make_test_desc_with_scripting_flag(
                     }
                 },
                 Some(ref context) => {
-                    let dom = parse_fragment(RcDom::default(), opts, context.clone(), vec![])
+                    let dom = parse_fragment(RcDom::default(), opts, context.clone(), vec![], true)
                         .one(data.clone());
                     // fragment case: serialize children of the html element
                     // rather than children of the document