Bug 482921 part 3 - Highlight tree builder-level errors in View Source. r=Olli.Pettay.

2011-11-01 13:33:11 +02:00
parent df4f37165e
commit 6bd519ac43
12 changed files with 1368 additions and 399 deletions
--- a/parser/html/nsHtml5StreamParser.cpp
+++ b/parser/html/nsHtml5StreamParser.cpp
@@ -206,7 +206,10 @@ nsHtml5StreamParser::nsHtml5StreamParser(nsHtml5TreeOpExecutor* aExecutor,
  mTokenizer->setEncodingDeclarationHandler(this);

  if (aMode == VIEW_SOURCE_HTML || aMode == VIEW_SOURCE_XML) {
-    mTokenizer->EnableViewSource(new nsHtml5Highlighter(mExecutor->GetStage()));
+    nsHtml5Highlighter* highlighter =
+      new nsHtml5Highlighter(mExecutor->GetStage());
+    mTokenizer->EnableViewSource(highlighter); // takes ownership
+    mTreeBuilder->EnableViewSource(highlighter); // doesn't own
  }

  // Chardet instantiation adapted from nsDOMFile.