The Tor Browser: parser/html/nsHtml5StringParser.cpp@925c144e1f1f

     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/. */

     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"

    14 NS_IMPL_ISUPPORTS0(nsHtml5StringParser)

    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 }

    25 nsHtml5StringParser::~nsHtml5StringParser()

    26 {

    27   MOZ_COUNT_DTOR(nsHtml5StringParser);

    28 }

    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);

    41   nsIDocument* doc = aTargetNode->OwnerDoc();

    42   nsIURI* uri = doc->GetDocumentURI();

    43   NS_ENSURE_TRUE(uri, NS_ERROR_NOT_AVAILABLE);

    45   mTreeBuilder->setFragmentContext(aContextLocalName,

    46                                    aContextNamespace,

    47                                    aTargetNode,

    48                                    aQuirks);

    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

    61   mTreeBuilder->SetPreventScriptExecution(aPreventScriptExecution);

    63   return Tokenize(aSourceBuffer, doc, true);

    64 }

    66 nsresult

    67 nsHtml5StringParser::ParseDocument(const nsAString& aSourceBuffer,

    68                                    nsIDocument* aTargetDoc,

    69                                    bool aScriptingEnabledForNoscriptParsing)

    70 {

    71   MOZ_ASSERT(!aTargetDoc->GetFirstChild());

    73   NS_ENSURE_TRUE(aSourceBuffer.Length() <= INT32_MAX,

    74                  NS_ERROR_OUT_OF_MEMORY);

    76   mTreeBuilder->setFragmentContext(nullptr,

    77                                    kNameSpaceID_None,

    78                                    nullptr,

    79                                    false);

    81   mTreeBuilder->SetPreventScriptExecution(true);

    83   return Tokenize(aSourceBuffer, aTargetDoc, aScriptingEnabledForNoscriptParsing);

    84 }

    86 nsresult

    87 nsHtml5StringParser::Tokenize(const nsAString& aSourceBuffer,

    88                               nsIDocument* aDocument,

    89                               bool aScriptingEnabledForNoscriptParsing) {

    91   nsIURI* uri = aDocument->GetDocumentURI();

    93   mBuilder->Init(aDocument, uri, nullptr, nullptr);

    95   mBuilder->SetParser(this);

    96   mBuilder->SetNodeInfoManager(aDocument->NodeInfoManager());

    98   // Mark the parser as *not* broken by passing NS_OK

    99   nsresult rv = mBuilder->MarkAsBroken(NS_OK);

   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 }

The Tor Browser / file revision