82c97c34d37b1feb3504b3bd2aa8ad4d621fda3d
[WebKit-https.git] / Source / WebCore / 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 "NestingLevelIncrementer.h"
30 #include "Timer.h"
31 #include <wtf/CurrentTime.h>
32 #include <wtf/PassOwnPtr.h>
33 #include <wtf/RefPtr.h>
34
35 #if PLATFORM(IOS)
36 #include "WebCoreThread.h"
37 #endif
38
39 namespace WebCore {
40
41 class Document;
42 class HTMLDocumentParser;
43
44 class ActiveParserSession {
45 public:
46     explicit ActiveParserSession(Document*);
47     ~ActiveParserSession();
48
49 private:
50     RefPtr<Document> m_document;
51 };
52
53 class PumpSession : public NestingLevelIncrementer, public ActiveParserSession {
54 public:
55     PumpSession(unsigned& nestingLevel, Document*);
56     ~PumpSession();
57
58     int processedTokens;
59     double startTime;
60     bool needsYield;
61     bool didSeeScript;
62 };
63
64 class HTMLParserScheduler {
65     WTF_MAKE_NONCOPYABLE(HTMLParserScheduler); WTF_MAKE_FAST_ALLOCATED;
66 public:
67     static OwnPtr<HTMLParserScheduler> create(HTMLDocumentParser& parser)
68     {
69         return adoptPtr(new HTMLParserScheduler(parser));
70     }
71     ~HTMLParserScheduler();
72
73     // Inline as this is called after every token in the parser.
74     void checkForYieldBeforeToken(PumpSession& session)
75     {
76 #if PLATFORM(IOS)
77         if (WebThreadShouldYield())
78             session.needsYield = true;
79 #endif
80         if (session.processedTokens > m_parserChunkSize || session.didSeeScript) {
81             // monotonicallyIncreasingTime() can be expensive. By delaying, we avoided calling
82             // monotonicallyIncreasingTime() when constructing non-yielding PumpSessions.
83             if (!session.startTime)
84                 session.startTime = monotonicallyIncreasingTime();
85
86             session.processedTokens = 0;
87             session.didSeeScript = false;
88
89             double elapsedTime = monotonicallyIncreasingTime() - session.startTime;
90             if (elapsedTime > m_parserTimeLimit)
91                 session.needsYield = true;
92         }
93         ++session.processedTokens;
94     }
95     void checkForYieldBeforeScript(PumpSession&);
96
97     void scheduleForResume();
98     bool isScheduledForResume() const { return m_isSuspendedWithActiveTimer || m_continueNextChunkTimer.isActive(); }
99
100     void suspend();
101     void resume();
102
103 private:
104     HTMLParserScheduler(HTMLDocumentParser&);
105
106     void continueNextChunkTimerFired(Timer<HTMLParserScheduler>*);
107
108     HTMLDocumentParser& m_parser;
109
110     double m_parserTimeLimit;
111     int m_parserChunkSize;
112     Timer<HTMLParserScheduler> m_continueNextChunkTimer;
113     bool m_isSuspendedWithActiveTimer;
114 #if !ASSERT_DISABLED
115     bool m_suspended;
116 #endif
117 };
118
119 }
120
121 #endif