src/third_party/WebKit/Source/core/html/parser/HTMLParserScheduler.h

   1 /*
   2  * Copyright (C) 2010 Google, Inc. All Rights Reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
  14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
  17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #ifndef HTMLParserScheduler_h
  27 #define HTMLParserScheduler_h
  28
  29 #include "core/html/parser/NestingLevelIncrementer.h"
  30 #include "platform/Timer.h"
  31 #include "wtf/CurrentTime.h"
  32 #include "wtf/PassOwnPtr.h"
  33 #include "wtf/RefPtr.h"
  34
  35 namespace WebCore {
  36
  37 class Document;
  38 class HTMLDocumentParser;
  39
  40 class ActiveParserSession {
  41 public:
  42     explicit ActiveParserSession(Document*);
  43     ~ActiveParserSession();
  44
  45 private:
  46     RefPtr<Document> m_document;
  47 };
  48
  49 class PumpSession : public NestingLevelIncrementer, public ActiveParserSession {
  50 public:
  51     PumpSession(unsigned& nestingLevel, Document*);
  52     ~PumpSession();
  53
  54     int processedTokens;
  55     double startTime;
  56     bool needsYield;
  57     bool didSeeScript;
  58 };
  59
  60 class HTMLParserScheduler {
  61     WTF_MAKE_NONCOPYABLE(HTMLParserScheduler); WTF_MAKE_FAST_ALLOCATED;
  62 public:
  63     static PassOwnPtr<HTMLParserScheduler> create(HTMLDocumentParser* parser)
  64     {
  65         return adoptPtr(new HTMLParserScheduler(parser));
  66     }
  67     ~HTMLParserScheduler();
  68
  69     // Inline as this is called after every token in the parser.
  70     void checkForYieldBeforeToken(PumpSession& session)
  71     {
  72         if (session.processedTokens > parserChunkSize || session.didSeeScript) {
  73             // currentTime() can be expensive.  By delaying, we avoided calling
  74             // currentTime() when constructing non-yielding PumpSessions.
  75             if (!session.startTime)
  76                 session.startTime = currentTime();
  77
  78             session.processedTokens = 0;
  79             session.didSeeScript = false;
  80
  81             double elapsedTime = currentTime() - session.startTime;
  82             if (elapsedTime > parserTimeLimit)
  83                 session.needsYield = true;
  84         }
  85         ++session.processedTokens;
  86     }
  87     void checkForYieldBeforeScript(PumpSession&);
  88
  89     void scheduleForResume();
  90     bool isScheduledForResume() const { return m_isSuspendedWithActiveTimer || m_continueNextChunkTimer.isActive(); }
  91
  92     void suspend();
  93     void resume();
  94
  95 private:
  96     static const double parserTimeLimit;
  97     static const int parserChunkSize;
  98
  99     HTMLParserScheduler(HTMLDocumentParser*);
 100
 101     void continueNextChunkTimerFired(Timer<HTMLParserScheduler>*);
 102
 103     HTMLDocumentParser* m_parser;
 104
 105     Timer<HTMLParserScheduler> m_continueNextChunkTimer;
 106     bool m_isSuspendedWithActiveTimer;
 107 };
 108
 109 }
 110
 111 #endif