parser/html/nsHtml5StringParser.cpp

   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 "nsHtml5DependentUTF16Buffer.h"
   7 #include "nsHtml5Tokenizer.h"
   8 #include "nsHtml5TreeBuilder.h"
   9 #include "nsHtml5TreeOpExecutor.h"
  10 #include "nsIContent.h"
  11 #include "nsIDocument.h"
  12 #include "mozilla/dom/DocumentFragment.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   mTokenizer->setInterner(&mAtomTable);
  22 }
  23
  24 nsHtml5StringParser::~nsHtml5StringParser() {}
  25
  26 nsresult
  27 nsHtml5StringParser::ParseFragment(const nsAString& aSourceBuffer,
  28                                    nsIContent* aTargetNode,
  29                                    nsAtom* aContextLocalName,
  30                                    int32_t aContextNamespace,
  31                                    bool aQuirks,
  32                                    bool aPreventScriptExecution)
  33 {
  34   NS_ENSURE_TRUE(aSourceBuffer.Length() <= INT32_MAX, NS_ERROR_OUT_OF_MEMORY);
  35
  36   nsIDocument* doc = aTargetNode->OwnerDoc();
  37   nsIURI* uri = doc->GetDocumentURI();
  38   NS_ENSURE_TRUE(uri, NS_ERROR_NOT_AVAILABLE);
  39
  40   mTreeBuilder->setFragmentContext(
  41     aContextLocalName, aContextNamespace, aTargetNode, aQuirks);
  42
  43 #ifdef DEBUG
  44   if (!aPreventScriptExecution) {
  45     NS_ASSERTION(!aTargetNode->IsInUncomposedDoc(),
  46                  "If script execution isn't prevented, "
  47                  "the target node must not be in doc.");
  48     NS_ASSERTION(
  49       aTargetNode->NodeType() == nsINode::DOCUMENT_FRAGMENT_NODE,
  50       "If script execution isn't prevented, must parse to DOM fragment.");
  51   }
  52 #endif
  53
  54   mTreeBuilder->SetPreventScriptExecution(aPreventScriptExecution);
  55
  56   return Tokenize(aSourceBuffer, doc, true);
  57 }
  58
  59 nsresult
  60 nsHtml5StringParser::ParseDocument(const nsAString& aSourceBuffer,
  61                                    nsIDocument* aTargetDoc,
  62                                    bool aScriptingEnabledForNoscriptParsing)
  63 {
  64   MOZ_ASSERT(!aTargetDoc->GetFirstChild());
  65
  66   NS_ENSURE_TRUE(aSourceBuffer.Length() <= INT32_MAX, NS_ERROR_OUT_OF_MEMORY);
  67
  68   mTreeBuilder->setFragmentContext(nullptr, kNameSpaceID_None, nullptr, false);
  69
  70   mTreeBuilder->SetPreventScriptExecution(true);
  71
  72   return Tokenize(
  73     aSourceBuffer, aTargetDoc, aScriptingEnabledForNoscriptParsing);
  74 }
  75
  76 nsresult
  77 nsHtml5StringParser::Tokenize(const nsAString& aSourceBuffer,
  78                               nsIDocument* aDocument,
  79                               bool aScriptingEnabledForNoscriptParsing)
  80 {
  81
  82   nsIURI* uri = aDocument->GetDocumentURI();
  83
  84   mBuilder->Init(aDocument, uri, nullptr, nullptr);
  85
  86   mBuilder->SetParser(this);
  87   mBuilder->SetNodeInfoManager(aDocument->NodeInfoManager());
  88
  89   // Mark the parser as *not* broken by passing NS_OK
  90   nsresult rv = mBuilder->MarkAsBroken(NS_OK);
  91
  92   mTreeBuilder->setScriptingEnabled(aScriptingEnabledForNoscriptParsing);
  93   mTreeBuilder->setIsSrcdocDocument(aDocument->IsSrcdocDocument());
  94   mBuilder->Start();
  95   mTokenizer->start();
  96   if (!aSourceBuffer.IsEmpty()) {
  97     bool lastWasCR = false;
  98     nsHtml5DependentUTF16Buffer buffer(aSourceBuffer);
  99     while (buffer.hasMore()) {
 100       buffer.adjust(lastWasCR);
 101       lastWasCR = false;
 102       if (buffer.hasMore()) {
 103         if (!mTokenizer->EnsureBufferSpace(buffer.getLength())) {
 104           rv = mBuilder->MarkAsBroken(NS_ERROR_OUT_OF_MEMORY);
 105           break;
 106         }
 107         lastWasCR = mTokenizer->tokenizeBuffer(&buffer);
 108         if (NS_FAILED(rv = mBuilder->IsBroken())) {
 109           break;
 110         }
 111       }
 112     }
 113   }
 114   if (NS_SUCCEEDED(rv)) {
 115     mTokenizer->eof();
 116   }
 117   mTokenizer->end();
 118   mBuilder->Finish();
 119   mAtomTable.Clear();
 120   return rv;
 121 }