be326f0fc5e6ae12c3c550c408b0dd1922504376
[WebKit-https.git] / Source / WebCore / dom / Document.cpp
1 /*
2  * Copyright (C) 1999 Lars Knoll (knoll@kde.org)
3  *           (C) 1999 Antti Koivisto (koivisto@kde.org)
4  *           (C) 2001 Dirk Mueller (mueller@kde.org)
5  *           (C) 2006 Alexey Proskuryakov (ap@webkit.org)
6  * Copyright (C) 2004-2015 Apple Inc. All rights reserved.
7  * Copyright (C) 2008, 2009 Torch Mobile Inc. All rights reserved. (http://www.torchmobile.com/)
8  * Copyright (C) 2008, 2009, 2011, 2012 Google Inc. All rights reserved.
9  * Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies)
10  * Copyright (C) Research In Motion Limited 2010-2011. All rights reserved.
11  *
12  * This library is free software; you can redistribute it and/or
13  * modify it under the terms of the GNU Library General Public
14  * License as published by the Free Software Foundation; either
15  * version 2 of the License, or (at your option) any later version.
16  *
17  * This library is distributed in the hope that it will be useful,
18  * but WITHOUT ANY WARRANTY; without even the implied warranty of
19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
20  * Library General Public License for more details.
21  *
22  * You should have received a copy of the GNU Library General Public License
23  * along with this library; see the file COPYING.LIB.  If not, write to
24  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
25  * Boston, MA 02110-1301, USA.
26  */
27
28 #include "config.h"
29 #include "Document.h"
30
31 #include "AXObjectCache.h"
32 #include "AnimationController.h"
33 #include "Attr.h"
34 #include "CDATASection.h"
35 #include "CSSFontSelector.h"
36 #include "CSSStyleDeclaration.h"
37 #include "CSSStyleSheet.h"
38 #include "CachedCSSStyleSheet.h"
39 #include "CachedResourceLoader.h"
40 #include "Chrome.h"
41 #include "ChromeClient.h"
42 #include "Comment.h"
43 #include "ContentSecurityPolicy.h"
44 #include "CookieJar.h"
45 #include "DOMImplementation.h"
46 #include "DOMNamedFlowCollection.h"
47 #include "DOMWindow.h"
48 #include "DateComponents.h"
49 #include "DebugPageOverlays.h"
50 #include "Dictionary.h"
51 #include "DocumentLoader.h"
52 #include "DocumentMarkerController.h"
53 #include "DocumentSharedObjectPool.h"
54 #include "DocumentType.h"
55 #include "Editor.h"
56 #include "ElementIterator.h"
57 #include "EntityReference.h"
58 #include "EventFactory.h"
59 #include "EventHandler.h"
60 #include "FocusController.h"
61 #include "FontLoader.h"
62 #include "FormController.h"
63 #include "FrameLoader.h"
64 #include "FrameLoaderClient.h"
65 #include "FrameView.h"
66 #include "GenericCachedHTMLCollection.h"
67 #include "HTMLAllCollection.h"
68 #include "HTMLAnchorElement.h"
69 #include "HTMLBaseElement.h"
70 #include "HTMLBodyElement.h"
71 #include "HTMLCanvasElement.h"
72 #include "HTMLDocument.h"
73 #include "HTMLElementFactory.h"
74 #include "HTMLFormControlElement.h"
75 #include "HTMLFrameOwnerElement.h"
76 #include "HTMLFrameSetElement.h"
77 #include "HTMLHeadElement.h"
78 #include "HTMLIFrameElement.h"
79 #include "HTMLImageElement.h"
80 #include "HTMLLinkElement.h"
81 #include "HTMLMediaElement.h"
82 #include "HTMLNameCollection.h"
83 #include "HTMLParserIdioms.h"
84 #include "HTMLPlugInElement.h"
85 #include "HTMLScriptElement.h"
86 #include "HTMLStyleElement.h"
87 #include "HTMLTitleElement.h"
88 #include "HTTPHeaderNames.h"
89 #include "HTTPParsers.h"
90 #include "HashChangeEvent.h"
91 #include "History.h"
92 #include "HitTestResult.h"
93 #include "IconController.h"
94 #include "ImageLoader.h"
95 #include "InspectorInstrumentation.h"
96 #include "JSLazyEventListener.h"
97 #include "Language.h"
98 #include "LoaderStrategy.h"
99 #include "Logging.h"
100 #include "MainFrame.h"
101 #include "MediaCanStartListener.h"
102 #include "MediaProducer.h"
103 #include "MediaQueryList.h"
104 #include "MediaQueryMatcher.h"
105 #include "MouseEventWithHitTestResults.h"
106 #include "NameNodeList.h"
107 #include "NestingLevelIncrementer.h"
108 #include "NodeIterator.h"
109 #include "NodeRareData.h"
110 #include "NodeWithIndex.h"
111 #include "PageConsoleClient.h"
112 #include "PageGroup.h"
113 #include "PageTransitionEvent.h"
114 #include "PlatformLocale.h"
115 #include "PlatformMediaSessionManager.h"
116 #include "PlatformStrategies.h"
117 #include "PlugInsResources.h"
118 #include "PluginDocument.h"
119 #include "PointerLockController.h"
120 #include "PopStateEvent.h"
121 #include "ProcessingInstruction.h"
122 #include "RenderChildIterator.h"
123 #include "RenderLayerCompositor.h"
124 #include "RenderView.h"
125 #include "RenderWidget.h"
126 #include "ResourceLoadScheduler.h"
127 #include "RuntimeEnabledFeatures.h"
128 #include "SVGDocumentExtensions.h"
129 #include "SVGElement.h"
130 #include "SVGElementFactory.h"
131 #include "SVGNames.h"
132 #include "SVGTitleElement.h"
133 #include "SchemeRegistry.h"
134 #include "ScopedEventQueue.h"
135 #include "ScriptController.h"
136 #include "ScriptRunner.h"
137 #include "ScriptSourceCode.h"
138 #include "ScrollingCoordinator.h"
139 #include "SecurityOrigin.h"
140 #include "SecurityOriginPolicy.h"
141 #include "SecurityPolicy.h"
142 #include "SegmentedString.h"
143 #include "SelectorQuery.h"
144 #include "Settings.h"
145 #include "ShadowRoot.h"
146 #include "StyleProperties.h"
147 #include "StyleResolver.h"
148 #include "StyleSheetContents.h"
149 #include "StyleSheetList.h"
150 #include "SubresourceLoader.h"
151 #include "TextNodeTraversal.h"
152 #include "TextResourceDecoder.h"
153 #include "TransformSource.h"
154 #include "TreeWalker.h"
155 #include "VisitedLinkState.h"
156 #include "XMLDocumentParser.h"
157 #include "XMLNSNames.h"
158 #include "XMLNames.h"
159 #include "XPathEvaluator.h"
160 #include "XPathExpression.h"
161 #include "XPathNSResolver.h"
162 #include "XPathResult.h"
163 #include "htmlediting.h"
164 #include <JavaScriptCore/Profile.h>
165 #include <ctime>
166 #include <inspector/ScriptCallStack.h>
167 #include <wtf/CurrentTime.h>
168 #include <wtf/TemporaryChange.h>
169 #include <wtf/text/StringBuffer.h>
170 #include <yarr/RegularExpression.h>
171
172 #if ENABLE(XSLT)
173 #include "XSLTProcessor.h"
174 #endif
175
176
177 #if ENABLE(TOUCH_EVENTS)
178 #include "TouchList.h"
179 #endif
180
181 #if PLATFORM(IOS)
182 #include "CSSFontSelector.h"
183 #include "DeviceMotionClientIOS.h"
184 #include "DeviceMotionController.h"
185 #include "DeviceOrientationClientIOS.h"
186 #include "DeviceOrientationController.h"
187 #include "Geolocation.h"
188 #include "Navigator.h"
189 #include "NavigatorGeolocation.h"
190 #include "WKContentObservation.h"
191 #include "WebCoreSystemInterface.h"
192 #endif
193
194 #if ENABLE(IOS_GESTURE_EVENTS)
195 #include "GestureEvent.h"
196 #endif
197
198 #if ENABLE(MATHML)
199 #include "MathMLElement.h"
200 #include "MathMLElementFactory.h"
201 #include "MathMLNames.h"
202 #endif
203
204 #if ENABLE(FULLSCREEN_API)
205 #include "RenderFullScreen.h"
206 #endif
207
208 #if ENABLE(REQUEST_ANIMATION_FRAME)
209 #include "RequestAnimationFrameCallback.h"
210 #include "ScriptedAnimationController.h"
211 #endif
212
213 #if ENABLE(IOS_TEXT_AUTOSIZING)
214 #include "TextAutoSizing.h"
215 #endif
216
217 #if ENABLE(TEXT_AUTOSIZING)
218 #include "TextAutosizer.h"
219 #endif
220
221 #if ENABLE(CSP_NEXT)
222 #include "DOMSecurityPolicy.h"
223 #endif
224
225 #if ENABLE(VIDEO_TRACK)
226 #include "CaptionUserPreferences.h"
227 #endif
228
229 #if ENABLE(WEB_REPLAY)
230 #include "WebReplayInputs.h"
231 #include <replay/EmptyInputCursor.h>
232 #include <replay/InputCursor.h>
233 #endif
234
235 #if ENABLE(WIRELESS_PLAYBACK_TARGET)
236 #include "MediaPlaybackTargetClient.h"
237 #endif
238
239 #if ENABLE(MEDIA_SESSION)
240 #include "MediaSession.h"
241 #endif
242
243 using namespace WTF;
244 using namespace Unicode;
245
246 namespace WebCore {
247
248 using namespace HTMLNames;
249
250 // #define INSTRUMENT_LAYOUT_SCHEDULING 1
251
252 static const unsigned cMaxWriteRecursionDepth = 21;
253
254 // DOM Level 2 says (letters added):
255 //
256 // a) Name start characters must have one of the categories Ll, Lu, Lo, Lt, Nl.
257 // b) Name characters other than Name-start characters must have one of the categories Mc, Me, Mn, Lm, or Nd.
258 // c) Characters in the compatibility area (i.e. with character code greater than #xF900 and less than #xFFFE) are not allowed in XML names.
259 // d) Characters which have a font or compatibility decomposition (i.e. those with a "compatibility formatting tag" in field 5 of the database -- marked by field 5 beginning with a "<") are not allowed.
260 // e) The following characters are treated as name-start characters rather than name characters, because the property file classifies them as Alphabetic: [#x02BB-#x02C1], #x0559, #x06E5, #x06E6.
261 // f) Characters #x20DD-#x20E0 are excluded (in accordance with Unicode, section 5.14).
262 // g) Character #x00B7 is classified as an extender, because the property list so identifies it.
263 // h) Character #x0387 is added as a name character, because #x00B7 is its canonical equivalent.
264 // i) Characters ':' and '_' are allowed as name-start characters.
265 // j) Characters '-' and '.' are allowed as name characters.
266 //
267 // It also contains complete tables. If we decide it's better, we could include those instead of the following code.
268
269 static inline bool isValidNameStart(UChar32 c)
270 {
271     // rule (e) above
272     if ((c >= 0x02BB && c <= 0x02C1) || c == 0x559 || c == 0x6E5 || c == 0x6E6)
273         return true;
274
275     // rule (i) above
276     if (c == ':' || c == '_')
277         return true;
278
279     // rules (a) and (f) above
280     if (!(U_GET_GC_MASK(c) & (U_GC_LL_MASK | U_GC_LU_MASK | U_GC_LO_MASK | U_GC_LT_MASK | U_GC_NL_MASK)))
281         return false;
282
283     // rule (c) above
284     if (c >= 0xF900 && c < 0xFFFE)
285         return false;
286
287     // rule (d) above
288     int type = u_getIntPropertyValue(c, UCHAR_DECOMPOSITION_TYPE);
289     if (type == U_DT_FONT || type == U_DT_COMPAT)
290         return false;
291
292     return true;
293 }
294
295 static inline bool isValidNamePart(UChar32 c)
296 {
297     // rules (a), (e), and (i) above
298     if (isValidNameStart(c))
299         return true;
300
301     // rules (g) and (h) above
302     if (c == 0x00B7 || c == 0x0387)
303         return true;
304
305     // rule (j) above
306     if (c == '-' || c == '.')
307         return true;
308
309     // rules (b) and (f) above
310     if (!(U_GET_GC_MASK(c) & (U_GC_M_MASK | U_GC_LM_MASK | U_GC_ND_MASK)))
311         return false;
312
313     // rule (c) above
314     if (c >= 0xF900 && c < 0xFFFE)
315         return false;
316
317     // rule (d) above
318     int type = u_getIntPropertyValue(c, UCHAR_DECOMPOSITION_TYPE);
319     if (type == U_DT_FONT || type == U_DT_COMPAT)
320         return false;
321
322     return true;
323 }
324
325 static bool shouldInheritSecurityOriginFromOwner(const URL& url)
326 {
327     // http://www.whatwg.org/specs/web-apps/current-work/#origin-0
328     //
329     // If a Document has the address "about:blank"
330     //     The origin of the Document is the origin it was assigned when its browsing context was created.
331     //
332     // Note: We generalize this to all "blank" URLs and invalid URLs because we
333     // treat all of these URLs as about:blank.
334     //
335     return url.isEmpty() || url.isBlankURL();
336 }
337
338 static Widget* widgetForElement(Element* focusedElement)
339 {
340     if (!focusedElement)
341         return nullptr;
342     auto* renderer = focusedElement->renderer();
343     if (!is<RenderWidget>(renderer))
344         return nullptr;
345     return downcast<RenderWidget>(*renderer).widget();
346 }
347
348 static bool acceptsEditingFocus(Node* node)
349 {
350     ASSERT(node);
351     ASSERT(node->hasEditableStyle());
352
353     Node* root = node->rootEditableElement();
354     Frame* frame = node->document().frame();
355     if (!frame || !root)
356         return false;
357
358     return frame->editor().shouldBeginEditing(rangeOfContents(*root).ptr());
359 }
360
361 static bool canAccessAncestor(const SecurityOrigin* activeSecurityOrigin, Frame* targetFrame)
362 {
363     // targetFrame can be 0 when we're trying to navigate a top-level frame
364     // that has a 0 opener.
365     if (!targetFrame)
366         return false;
367
368     const bool isLocalActiveOrigin = activeSecurityOrigin->isLocal();
369     for (Frame* ancestorFrame = targetFrame; ancestorFrame; ancestorFrame = ancestorFrame->tree().parent()) {
370         Document* ancestorDocument = ancestorFrame->document();
371         // FIXME: Should be an ASSERT? Frames should alway have documents.
372         if (!ancestorDocument)
373             return true;
374
375         const SecurityOrigin* ancestorSecurityOrigin = ancestorDocument->securityOrigin();
376         if (activeSecurityOrigin->canAccess(ancestorSecurityOrigin))
377             return true;
378         
379         // Allow file URL descendant navigation even when allowFileAccessFromFileURLs is false.
380         // FIXME: It's a bit strange to special-case local origins here. Should we be doing
381         // something more general instead?
382         if (isLocalActiveOrigin && ancestorSecurityOrigin->isLocal())
383             return true;
384     }
385
386     return false;
387 }
388
389 static void printNavigationErrorMessage(Frame* frame, const URL& activeURL, const char* reason)
390 {
391     String message = "Unsafe JavaScript attempt to initiate navigation for frame with URL '" + frame->document()->url().string() + "' from frame with URL '" + activeURL.string() + "'. " + reason + "\n";
392
393     // FIXME: should we print to the console of the document performing the navigation instead?
394     frame->document()->domWindow()->printErrorMessage(message);
395 }
396
397 uint64_t Document::s_globalTreeVersion = 0;
398
399 #if ENABLE(IOS_TEXT_AUTOSIZING)
400 void TextAutoSizingTraits::constructDeletedValue(TextAutoSizingKey& slot)
401 {
402     new (&slot) TextAutoSizingKey(TextAutoSizingKey::deletedKeyStyle(), TextAutoSizingKey::deletedKeyDoc());
403 }
404
405 bool TextAutoSizingTraits::isDeletedValue(const TextAutoSizingKey& value)
406 {
407     return value.style() == TextAutoSizingKey::deletedKeyStyle() && value.doc() == TextAutoSizingKey::deletedKeyDoc();
408 }
409 #endif
410
411 HashSet<Document*>& Document::allDocuments()
412 {
413     static NeverDestroyed<HashSet<Document*>> documents;
414     return documents;
415 }
416
417 Document::Document(Frame* frame, const URL& url, unsigned documentClasses, unsigned constructionFlags)
418     : ContainerNode(*this, CreateDocument)
419     , TreeScope(*this)
420 #if ENABLE(IOS_TOUCH_EVENTS)
421     , m_handlingTouchEvent(false)
422     , m_touchEventRegionsDirty(false)
423     , m_touchEventsChangedTimer(*this, &Document::touchEventsChangedTimerFired)
424 #endif
425     , m_referencingNodeCount(0)
426     , m_didCalculateStyleResolver(false)
427     , m_hasNodesWithPlaceholderStyle(false)
428     , m_needsNotifyRemoveAllPendingStylesheet(false)
429     , m_ignorePendingStylesheets(false)
430     , m_pendingSheetLayout(NoLayoutWithPendingSheets)
431     , m_frame(frame)
432     , m_cachedResourceLoader(m_frame ? Ref<CachedResourceLoader>(m_frame->loader().activeDocumentLoader()->cachedResourceLoader()) : CachedResourceLoader::create(nullptr))
433     , m_activeParserCount(0)
434     , m_wellFormed(false)
435     , m_printing(false)
436     , m_paginatedForScreen(false)
437     , m_compatibilityMode(DocumentCompatibilityMode::NoQuirksMode)
438     , m_compatibilityModeLocked(false)
439     , m_textColor(Color::black)
440     , m_domTreeVersion(++s_globalTreeVersion)
441     , m_listenerTypes(0)
442     , m_mutationObserverTypes(0)
443     , m_styleSheetCollection(*this)
444     , m_visitedLinkState(std::make_unique<VisitedLinkState>(*this))
445     , m_visuallyOrdered(false)
446     , m_readyState(Complete)
447     , m_bParsing(false)
448     , m_optimizedStyleSheetUpdateTimer(*this, &Document::optimizedStyleSheetUpdateTimerFired)
449     , m_styleRecalcTimer(*this, &Document::updateStyleIfNeeded)
450     , m_pendingStyleRecalcShouldForce(false)
451     , m_inStyleRecalc(false)
452     , m_closeAfterStyleRecalc(false)
453     , m_gotoAnchorNeededAfterStylesheetsLoad(false)
454     , m_frameElementsShouldIgnoreScrolling(false)
455     , m_updateFocusAppearanceRestoresSelection(false)
456     , m_ignoreDestructiveWriteCount(0)
457     , m_markers(std::make_unique<DocumentMarkerController>())
458     , m_updateFocusAppearanceTimer(*this, &Document::updateFocusAppearanceTimerFired)
459     , m_cssTarget(nullptr)
460     , m_processingLoadEvent(false)
461     , m_loadEventFinished(false)
462     , m_startTime(std::chrono::steady_clock::now())
463     , m_overMinimumLayoutThreshold(false)
464     , m_scriptRunner(std::make_unique<ScriptRunner>(*this))
465     , m_xmlVersion(ASCIILiteral("1.0"))
466     , m_xmlStandalone(StandaloneUnspecified)
467     , m_hasXMLDeclaration(false)
468     , m_designMode(inherit)
469 #if !ASSERT_DISABLED
470     , m_inInvalidateNodeListAndCollectionCaches(false)
471 #endif
472 #if ENABLE(DASHBOARD_SUPPORT)
473     , m_hasAnnotatedRegions(false)
474     , m_annotatedRegionsDirty(false)
475 #endif
476     , m_createRenderers(true)
477     , m_inPageCache(false)
478     , m_accessKeyMapValid(false)
479     , m_documentClasses(documentClasses)
480     , m_isSynthesized(constructionFlags & Synthesized)
481     , m_isNonRenderedPlaceholder(constructionFlags & NonRenderedPlaceholder)
482     , m_sawElementsInKnownNamespaces(false)
483     , m_isSrcdocDocument(false)
484     , m_eventQueue(*this)
485     , m_weakFactory(this)
486 #if ENABLE(FULLSCREEN_API)
487     , m_areKeysEnabledInFullScreen(0)
488     , m_fullScreenRenderer(nullptr)
489     , m_fullScreenChangeDelayTimer(*this, &Document::fullScreenChangeDelayTimerFired)
490     , m_isAnimatingFullScreen(false)
491 #endif
492     , m_loadEventDelayCount(0)
493     , m_loadEventDelayTimer(*this, &Document::loadEventDelayTimerFired)
494     , m_referrerPolicy(ReferrerPolicyDefault)
495     , m_directionSetOnDocumentElement(false)
496     , m_writingModeSetOnDocumentElement(false)
497     , m_writeRecursionIsTooDeep(false)
498     , m_writeRecursionDepth(0)
499     , m_lastHandledUserGestureTimestamp(0)
500 #if PLATFORM(IOS)
501 #if ENABLE(DEVICE_ORIENTATION)
502     , m_deviceMotionClient(std::make_unique<DeviceMotionClientIOS>())
503     , m_deviceMotionController(std::make_unique<DeviceMotionController>(m_deviceMotionClient.get()))
504     , m_deviceOrientationClient(std::make_unique<DeviceOrientationClientIOS>())
505     , m_deviceOrientationController(std::make_unique<DeviceOrientationController>(m_deviceOrientationClient.get()))
506 #endif
507 #endif
508 #if ENABLE(TELEPHONE_NUMBER_DETECTION)
509     , m_isTelephoneNumberParsingAllowed(true)
510 #endif
511     , m_pendingTasksTimer(*this, &Document::pendingTasksTimerFired)
512     , m_scheduledTasksAreSuspended(false)
513     , m_visualUpdatesAllowed(true)
514     , m_visualUpdatesSuppressionTimer(*this, &Document::visualUpdatesSuppressionTimerFired)
515     , m_sharedObjectPoolClearTimer(*this, &Document::sharedObjectPoolClearTimerFired)
516 #ifndef NDEBUG
517     , m_didDispatchViewportPropertiesChanged(false)
518 #endif
519 #if ENABLE(TEMPLATE_ELEMENT)
520     , m_templateDocumentHost(nullptr)
521 #endif
522 #if ENABLE(WEB_REPLAY)
523     , m_inputCursor(EmptyInputCursor::create())
524 #endif
525     , m_didAssociateFormControlsTimer(*this, &Document::didAssociateFormControlsTimerFired)
526     , m_cookieCacheExpiryTimer(*this, &Document::invalidateDOMCookieCache)
527     , m_disabledFieldsetElementsCount(0)
528     , m_hasInjectedPlugInsScript(false)
529     , m_renderTreeBeingDestroyed(false)
530     , m_hasPreparedForDestruction(false)
531     , m_hasStyleWithViewportUnits(false)
532 {
533     allDocuments().add(this);
534
535     // We depend on the url getting immediately set in subframes, but we
536     // also depend on the url NOT getting immediately set in opened windows.
537     // See fast/dom/early-frame-url.html
538     // and fast/dom/location-new-window-no-crash.html, respectively.
539     // FIXME: Can/should we unify this behavior?
540     if ((frame && frame->ownerElement()) || !url.isEmpty())
541         setURL(url);
542
543     m_cachedResourceLoader->setDocument(this);
544
545 #if ENABLE(TEXT_AUTOSIZING)
546     m_textAutosizer = std::make_unique<TextAutosizer>(this);
547 #endif
548
549     resetLinkColor();
550     resetVisitedLinkColor();
551     resetActiveLinkColor();
552
553     initSecurityContext();
554     initDNSPrefetch();
555
556     for (unsigned i = 0; i < WTF_ARRAY_LENGTH(m_nodeListAndCollectionCounts); ++i)
557         m_nodeListAndCollectionCounts[i] = 0;
558 }
559
560 #if ENABLE(FULLSCREEN_API)
561 static bool isAttributeOnAllOwners(const WebCore::QualifiedName& attribute, const WebCore::QualifiedName& prefixedAttribute, const HTMLFrameOwnerElement* owner)
562 {
563     if (!owner)
564         return true;
565     do {
566         if (!(owner->hasAttribute(attribute) || owner->hasAttribute(prefixedAttribute)))
567             return false;
568     } while ((owner = owner->document().ownerElement()));
569     return true;
570 }
571 #endif
572
573 Ref<Document> Document::create(ScriptExecutionContext& context)
574 {
575     Ref<Document> document = adoptRef(*new Document(nullptr, URL()));
576     document->setSecurityOriginPolicy(context.securityOriginPolicy());
577
578     return document;
579 }
580
581 Document::~Document()
582 {
583     allDocuments().remove(this);
584
585     ASSERT(!renderView());
586     ASSERT(!m_inPageCache);
587     ASSERT(m_ranges.isEmpty());
588     ASSERT(!m_parentTreeScope);
589     ASSERT(!m_disabledFieldsetElementsCount);
590
591 #if ENABLE(DEVICE_ORIENTATION) && PLATFORM(IOS)
592     m_deviceMotionClient->deviceMotionControllerDestroyed();
593     m_deviceOrientationClient->deviceOrientationControllerDestroyed();
594 #endif
595     
596 #if ENABLE(TEMPLATE_ELEMENT)
597     if (m_templateDocument)
598         m_templateDocument->setTemplateDocumentHost(nullptr); // balanced in templateDocument().
599 #endif
600
601     // FIXME: Should we reset m_domWindow when we detach from the Frame?
602     if (m_domWindow)
603         m_domWindow->resetUnlessSuspendedForPageCache();
604
605     m_scriptRunner = nullptr;
606
607     removeAllEventListeners();
608
609     // Currently we believe that Document can never outlive the parser.
610     // Although the Document may be replaced synchronously, DocumentParsers
611     // generally keep at least one reference to an Element which would in turn
612     // has a reference to the Document.  If you hit this ASSERT, then that
613     // assumption is wrong.  DocumentParser::detach() should ensure that even
614     // if the DocumentParser outlives the Document it won't cause badness.
615     ASSERT(!m_parser || m_parser->refCount() == 1);
616     detachParser();
617
618     if (this == &topDocument())
619         clearAXObjectCache();
620
621     m_decoder = nullptr;
622
623     if (m_styleSheetList)
624         m_styleSheetList->detachFromDocument();
625
626     if (m_elementSheet)
627         m_elementSheet->detachFromDocument();
628     m_styleSheetCollection.detachFromDocument();
629
630     clearStyleResolver(); // We need to destroy CSSFontSelector before destroying m_cachedResourceLoader.
631
632     // It's possible for multiple Documents to end up referencing the same CachedResourceLoader (e.g., SVGImages
633     // load the initial empty document and the SVGDocument with the same DocumentLoader).
634     if (m_cachedResourceLoader->document() == this)
635         m_cachedResourceLoader->setDocument(nullptr);
636
637 #if ENABLE(VIDEO)
638     if (auto* platformMediaSessionManager = PlatformMediaSessionManager::sharedManagerIfExists())
639         platformMediaSessionManager->stopAllMediaPlaybackForDocument(this);
640 #endif
641     
642     // We must call clearRareData() here since a Document class inherits TreeScope
643     // as well as Node. See a comment on TreeScope.h for the reason.
644     if (hasRareData())
645         clearRareData();
646
647     ASSERT(!m_listsInvalidatedAtDocument.size());
648     ASSERT(!m_collectionsInvalidatedAtDocument.size());
649
650     for (unsigned i = 0; i < WTF_ARRAY_LENGTH(m_nodeListAndCollectionCounts); ++i)
651         ASSERT(!m_nodeListAndCollectionCounts[i]);
652 }
653
654 void Document::removedLastRef()
655 {
656     ASSERT(!m_deletionHasBegun);
657     if (m_referencingNodeCount) {
658         // If removing a child removes the last node reference, we don't want the scope to be destroyed
659         // until after removeDetachedChildren returns, so we protect ourselves.
660         incrementReferencingNodeCount();
661
662         // We must make sure not to be retaining any of our children through
663         // these extra pointers or we will create a reference cycle.
664         m_focusedElement = nullptr;
665         m_hoveredElement = nullptr;
666         m_activeElement = nullptr;
667         m_titleElement = nullptr;
668         m_documentElement = nullptr;
669         m_userActionElements.documentDidRemoveLastRef();
670 #if ENABLE(FULLSCREEN_API)
671         m_fullScreenElement = nullptr;
672         m_fullScreenElementStack.clear();
673 #endif
674
675         detachParser();
676
677         // removeDetachedChildren() doesn't always unregister IDs,
678         // so tear down scope information up front to avoid having
679         // stale references in the map.
680
681         destroyTreeScopeData();
682         removeDetachedChildren();
683         m_formController = nullptr;
684         
685         m_markers->detach();
686         
687         m_cssCanvasElements.clear();
688         
689         commonTeardown();
690
691 #ifndef NDEBUG
692         // We need to do this right now since selfOnlyDeref() can delete this.
693         m_inRemovedLastRefFunction = false;
694 #endif
695         decrementReferencingNodeCount();
696     } else {
697 #ifndef NDEBUG
698         m_inRemovedLastRefFunction = false;
699         m_deletionHasBegun = true;
700 #endif
701         delete this;
702     }
703 }
704
705 void Document::commonTeardown()
706 {
707     if (svgExtensions())
708         accessSVGExtensions().pauseAnimations();
709
710 #if ENABLE(REQUEST_ANIMATION_FRAME)
711     clearScriptedAnimationController();
712 #endif
713 }
714
715 Element* Document::getElementByAccessKey(const String& key)
716 {
717     if (key.isEmpty())
718         return nullptr;
719     if (!m_accessKeyMapValid) {
720         buildAccessKeyMap(this);
721         m_accessKeyMapValid = true;
722     }
723     return m_elementsByAccessKey.get(key.impl());
724 }
725
726 void Document::buildAccessKeyMap(TreeScope* scope)
727 {
728     ASSERT(scope);
729     for (auto& element : descendantsOfType<Element>(scope->rootNode())) {
730         const AtomicString& accessKey = element.fastGetAttribute(accesskeyAttr);
731         if (!accessKey.isEmpty())
732             m_elementsByAccessKey.set(accessKey.impl(), &element);
733
734         if (ShadowRoot* root = element.shadowRoot())
735             buildAccessKeyMap(root);
736     }
737 }
738
739 void Document::invalidateAccessKeyMap()
740 {
741     m_accessKeyMapValid = false;
742     m_elementsByAccessKey.clear();
743 }
744
745 void Document::addImageElementByLowercasedUsemap(const AtomicStringImpl& name, HTMLImageElement& element)
746 {
747     return m_imagesByUsemap.add(name, element, *this);
748 }
749
750 void Document::removeImageElementByLowercasedUsemap(const AtomicStringImpl& name, HTMLImageElement& element)
751 {
752     return m_imagesByUsemap.remove(name, element);
753 }
754
755 HTMLImageElement* Document::imageElementByLowercasedUsemap(const AtomicStringImpl& name) const
756 {
757     return m_imagesByUsemap.getElementByLowercasedUsemap(name, *this);
758 }
759
760 SelectorQuery* Document::selectorQueryForString(const String& selectorString, ExceptionCode& ec)
761 {
762     if (selectorString.isEmpty()) {
763         ec = SYNTAX_ERR;
764         return nullptr;
765     }
766
767     if (!m_selectorQueryCache)
768         m_selectorQueryCache = std::make_unique<SelectorQueryCache>();
769     return m_selectorQueryCache->add(selectorString, *this, ec);
770 }
771
772 void Document::clearSelectorQueryCache()
773 {
774     m_selectorQueryCache = nullptr;
775 }
776
777 MediaQueryMatcher& Document::mediaQueryMatcher()
778 {
779     if (!m_mediaQueryMatcher)
780         m_mediaQueryMatcher = MediaQueryMatcher::create(this);
781     return *m_mediaQueryMatcher;
782 }
783
784 void Document::setCompatibilityMode(DocumentCompatibilityMode mode)
785 {
786     if (m_compatibilityModeLocked || mode == m_compatibilityMode)
787         return;
788     bool wasInQuirksMode = inQuirksMode();
789     m_compatibilityMode = mode;
790
791     clearSelectorQueryCache();
792
793     if (inQuirksMode() != wasInQuirksMode) {
794         // All user stylesheets have to reparse using the different mode.
795         m_styleSheetCollection.clearPageUserSheet();
796         m_styleSheetCollection.invalidateInjectedStyleSheetCache();
797     }
798 }
799
800 String Document::compatMode() const
801 {
802     return inQuirksMode() ? "BackCompat" : "CSS1Compat";
803 }
804
805 void Document::resetLinkColor()
806 {
807     m_linkColor = Color(0, 0, 238);
808 }
809
810 void Document::resetVisitedLinkColor()
811 {
812     m_visitedLinkColor = Color(85, 26, 139);    
813 }
814
815 void Document::resetActiveLinkColor()
816 {
817     m_activeLinkColor.setNamedColor("red");
818 }
819
820 DOMImplementation& Document::implementation()
821 {
822     if (!m_implementation)
823         m_implementation = std::make_unique<DOMImplementation>(*this);
824     return *m_implementation;
825 }
826
827 bool Document::hasManifest() const
828 {
829     return documentElement() && documentElement()->hasTagName(htmlTag) && documentElement()->fastHasAttribute(manifestAttr);
830 }
831
832 DocumentType* Document::doctype() const
833 {
834     for (Node* node = firstChild(); node; node = node->nextSibling()) {
835         if (node->isDocumentTypeNode())
836             return static_cast<DocumentType*>(node);
837     }
838     return nullptr;
839 }
840
841 void Document::childrenChanged(const ChildChange& change)
842 {
843     ContainerNode::childrenChanged(change);
844
845 #if PLATFORM(IOS)
846     // FIXME: Chrome::didReceiveDocType() used to be called only when the doctype changed. We need to check the
847     // impact of calling this systematically. If the overhead is negligible, we need to rename didReceiveDocType,
848     // otherwise, we need to detect the doc type changes before updating the viewport.
849     if (Page* page = this->page())
850         page->chrome().didReceiveDocType(frame());
851 #endif
852
853     Element* newDocumentElement = childrenOfType<Element>(*this).first();
854     if (newDocumentElement == m_documentElement)
855         return;
856     m_documentElement = newDocumentElement;
857     // The root style used for media query matching depends on the document element.
858     clearStyleResolver();
859 }
860
861 RefPtr<Element> Document::createElement(const AtomicString& name, ExceptionCode& ec)
862 {
863     if (!isValidName(name)) {
864         ec = INVALID_CHARACTER_ERR;
865         return nullptr;
866     }
867
868     if (isXHTMLDocument())
869         return HTMLElementFactory::createElement(QualifiedName(nullAtom, name, xhtmlNamespaceURI), *this);
870
871     return createElement(QualifiedName(nullAtom, name, nullAtom), false);
872 }
873
874 Ref<DocumentFragment> Document::createDocumentFragment()
875 {
876     return DocumentFragment::create(document());
877 }
878
879 Ref<Text> Document::createTextNode(const String& data)
880 {
881     return Text::create(*this, data);
882 }
883
884 Ref<Comment> Document::createComment(const String& data)
885 {
886     return Comment::create(*this, data);
887 }
888
889 RefPtr<CDATASection> Document::createCDATASection(const String& data, ExceptionCode& ec)
890 {
891     if (isHTMLDocument()) {
892         ec = NOT_SUPPORTED_ERR;
893         return nullptr;
894     }
895     return CDATASection::create(*this, data);
896 }
897
898 RefPtr<ProcessingInstruction> Document::createProcessingInstruction(const String& target, const String& data, ExceptionCode& ec)
899 {
900     if (!isValidName(target)) {
901         ec = INVALID_CHARACTER_ERR;
902         return nullptr;
903     }
904
905     if (data.contains("?>")) {
906         ec = INVALID_CHARACTER_ERR;
907         return nullptr;
908     }
909
910     return ProcessingInstruction::create(*this, target, data);
911 }
912
913 RefPtr<EntityReference> Document::createEntityReference(const String& name, ExceptionCode& ec)
914 {
915     if (!isValidName(name)) {
916         ec = INVALID_CHARACTER_ERR;
917         return nullptr;
918     }
919     if (isHTMLDocument()) {
920         ec = NOT_SUPPORTED_ERR;
921         return nullptr;
922     }
923     return EntityReference::create(*this, name);
924 }
925
926 Ref<Text> Document::createEditingTextNode(const String& text)
927 {
928     return Text::createEditingText(*this, text);
929 }
930
931 Ref<CSSStyleDeclaration> Document::createCSSStyleDeclaration()
932 {
933     Ref<MutableStyleProperties> propertySet(MutableStyleProperties::create());
934     return *propertySet->ensureCSSStyleDeclaration();
935 }
936
937 RefPtr<Node> Document::importNode(Node* importedNode, bool deep, ExceptionCode& ec)
938 {
939     if (!importedNode) {
940         ec = NOT_SUPPORTED_ERR;
941         return nullptr;
942     }
943
944     switch (importedNode->nodeType()) {
945     case ELEMENT_NODE:
946     case TEXT_NODE:
947     case CDATA_SECTION_NODE:
948     case ENTITY_REFERENCE_NODE:
949     case PROCESSING_INSTRUCTION_NODE:
950     case COMMENT_NODE:
951     case DOCUMENT_FRAGMENT_NODE:
952         return importedNode->cloneNodeInternal(document(), deep ? CloningOperation::Everything : CloningOperation::OnlySelf);
953
954     case ATTRIBUTE_NODE:
955         // FIXME: This will "Attr::normalize" child nodes of Attr.
956         return Attr::create(*this, QualifiedName(nullAtom, downcast<Attr>(*importedNode).name(), nullAtom), downcast<Attr>(*importedNode).value());
957
958     case DOCUMENT_NODE: // Can't import a document into another document.
959     case DOCUMENT_TYPE_NODE: // FIXME: Support cloning a DocumentType node per DOM4.
960         break;
961
962     case ENTITY_NODE:
963     case XPATH_NAMESPACE_NODE:
964         ASSERT_NOT_REACHED(); // These two types of DOM nodes are not implemented.
965         break;
966     }
967     ec = NOT_SUPPORTED_ERR;
968     return nullptr;
969 }
970
971
972 RefPtr<Node> Document::adoptNode(PassRefPtr<Node> source, ExceptionCode& ec)
973 {
974     if (!source) {
975         ec = NOT_SUPPORTED_ERR;
976         return nullptr;
977     }
978
979     if (source->isReadOnlyNode()) {
980         ec = NO_MODIFICATION_ALLOWED_ERR;
981         return nullptr;
982     }
983
984     EventQueueScope scope;
985
986     switch (source->nodeType()) {
987     case ENTITY_NODE:
988     case DOCUMENT_NODE:
989     case XPATH_NAMESPACE_NODE:
990         ec = NOT_SUPPORTED_ERR;
991         return nullptr;
992     case ATTRIBUTE_NODE: {                   
993         Attr& attr = downcast<Attr>(*source);
994         if (attr.ownerElement())
995             attr.ownerElement()->removeAttributeNode(&attr, ec);
996         break;
997     }       
998     default:
999         if (source->isShadowRoot()) {
1000             // ShadowRoot cannot disconnect itself from the host node.
1001             ec = HIERARCHY_REQUEST_ERR;
1002             return nullptr;
1003         }
1004         if (is<HTMLFrameOwnerElement>(*source)) {
1005             HTMLFrameOwnerElement& frameOwnerElement = downcast<HTMLFrameOwnerElement>(*source);
1006             if (frame() && frame()->tree().isDescendantOf(frameOwnerElement.contentFrame())) {
1007                 ec = HIERARCHY_REQUEST_ERR;
1008                 return nullptr;
1009             }
1010         }
1011         if (source->parentNode()) {
1012             source->parentNode()->removeChild(*source, ec);
1013             if (ec)
1014                 return nullptr;
1015         }
1016     }
1017
1018     adoptIfNeeded(source.get());
1019
1020     return source;
1021 }
1022
1023 bool Document::hasValidNamespaceForElements(const QualifiedName& qName)
1024 {
1025     // These checks are from DOM Core Level 2, createElementNS
1026     // http://www.w3.org/TR/DOM-Level-2-Core/core.html#ID-DocCrElNS
1027     if (!qName.prefix().isEmpty() && qName.namespaceURI().isNull()) // createElementNS(null, "html:div")
1028         return false;
1029     if (qName.prefix() == xmlAtom && qName.namespaceURI() != XMLNames::xmlNamespaceURI) // createElementNS("http://www.example.com", "xml:lang")
1030         return false;
1031
1032     // Required by DOM Level 3 Core and unspecified by DOM Level 2 Core:
1033     // http://www.w3.org/TR/2004/REC-DOM-Level-3-Core-20040407/core.html#ID-DocCrElNS
1034     // createElementNS("http://www.w3.org/2000/xmlns/", "foo:bar"), createElementNS(null, "xmlns:bar"), createElementNS(null, "xmlns")
1035     if (qName.prefix() == xmlnsAtom || (qName.prefix().isEmpty() && qName.localName() == xmlnsAtom))
1036         return qName.namespaceURI() == XMLNSNames::xmlnsNamespaceURI;
1037     return qName.namespaceURI() != XMLNSNames::xmlnsNamespaceURI;
1038 }
1039
1040 bool Document::hasValidNamespaceForAttributes(const QualifiedName& qName)
1041 {
1042     return hasValidNamespaceForElements(qName);
1043 }
1044
1045 // FIXME: This should really be in a possible ElementFactory class.
1046 Ref<Element> Document::createElement(const QualifiedName& name, bool createdByParser)
1047 {
1048     RefPtr<Element> element;
1049
1050     // FIXME: Use registered namespaces and look up in a hash to find the right factory.
1051     if (name.namespaceURI() == xhtmlNamespaceURI)
1052         element = HTMLElementFactory::createElement(name, *this, nullptr, createdByParser);
1053     else if (name.namespaceURI() == SVGNames::svgNamespaceURI)
1054         element = SVGElementFactory::createElement(name, *this, createdByParser);
1055 #if ENABLE(MATHML)
1056     else if (name.namespaceURI() == MathMLNames::mathmlNamespaceURI)
1057         element = MathMLElementFactory::createElement(name, *this, createdByParser);
1058 #endif
1059
1060     if (element)
1061         m_sawElementsInKnownNamespaces = true;
1062     else
1063         element = Element::create(name, document());
1064
1065     // <image> uses imgTag so we need a special rule.
1066     ASSERT((name.matches(imageTag) && element->tagQName().matches(imgTag) && element->tagQName().prefix() == name.prefix()) || name == element->tagQName());
1067
1068     return element.releaseNonNull();
1069 }
1070
1071 bool Document::cssRegionsEnabled() const
1072 {
1073     return RuntimeEnabledFeatures::sharedFeatures().cssRegionsEnabled(); 
1074 }
1075
1076 bool Document::cssCompositingEnabled() const
1077 {
1078     return RuntimeEnabledFeatures::sharedFeatures().cssCompositingEnabled();
1079 }
1080
1081 #if ENABLE(CSS_REGIONS)
1082
1083 RefPtr<DOMNamedFlowCollection> Document::webkitGetNamedFlows()
1084 {
1085     if (!cssRegionsEnabled() || !renderView())
1086         return nullptr;
1087
1088     updateStyleIfNeeded();
1089
1090     return namedFlows().createCSSOMSnapshot();
1091 }
1092
1093 #endif
1094
1095 NamedFlowCollection& Document::namedFlows()
1096 {
1097     if (!m_namedFlows)
1098         m_namedFlows = NamedFlowCollection::create(this);
1099
1100     return *m_namedFlows;
1101 }
1102
1103 RefPtr<Element> Document::createElementNS(const String& namespaceURI, const String& qualifiedName, ExceptionCode& ec)
1104 {
1105     String prefix, localName;
1106     if (!parseQualifiedName(qualifiedName, prefix, localName, ec))
1107         return nullptr;
1108
1109     QualifiedName qName(prefix, localName, namespaceURI);
1110     if (!hasValidNamespaceForElements(qName)) {
1111         ec = NAMESPACE_ERR;
1112         return nullptr;
1113     }
1114
1115     return createElement(qName, false);
1116 }
1117
1118 String Document::readyState() const
1119 {
1120     DEPRECATED_DEFINE_STATIC_LOCAL(const String, loading, (ASCIILiteral("loading")));
1121     DEPRECATED_DEFINE_STATIC_LOCAL(const String, interactive, (ASCIILiteral("interactive")));
1122     DEPRECATED_DEFINE_STATIC_LOCAL(const String, complete, (ASCIILiteral("complete")));
1123
1124     switch (m_readyState) {
1125     case Loading:
1126         return loading;
1127     case Interactive:
1128         return interactive;
1129     case Complete:
1130         return complete;
1131     }
1132
1133     ASSERT_NOT_REACHED();
1134     return String();
1135 }
1136
1137 void Document::setReadyState(ReadyState readyState)
1138 {
1139     if (readyState == m_readyState)
1140         return;
1141
1142 #if ENABLE(WEB_TIMING)
1143     switch (readyState) {
1144     case Loading:
1145         if (!m_documentTiming.domLoading)
1146             m_documentTiming.domLoading = monotonicallyIncreasingTime();
1147         break;
1148     case Interactive:
1149         if (!m_documentTiming.domInteractive)
1150             m_documentTiming.domInteractive = monotonicallyIncreasingTime();
1151         break;
1152     case Complete:
1153         if (!m_documentTiming.domComplete)
1154             m_documentTiming.domComplete = monotonicallyIncreasingTime();
1155         break;
1156     }
1157 #endif
1158
1159     m_readyState = readyState;
1160     dispatchEvent(Event::create(eventNames().readystatechangeEvent, false, false));
1161     
1162     if (settings() && settings()->suppressesIncrementalRendering())
1163         setVisualUpdatesAllowed(readyState);
1164 }
1165
1166 void Document::setVisualUpdatesAllowed(ReadyState readyState)
1167 {
1168     ASSERT(settings() && settings()->suppressesIncrementalRendering());
1169     switch (readyState) {
1170     case Loading:
1171         ASSERT(!m_visualUpdatesSuppressionTimer.isActive());
1172         ASSERT(m_visualUpdatesAllowed);
1173         setVisualUpdatesAllowed(false);
1174         break;
1175     case Interactive:
1176         ASSERT(m_visualUpdatesSuppressionTimer.isActive() || m_visualUpdatesAllowed);
1177         break;
1178     case Complete:
1179         if (m_visualUpdatesSuppressionTimer.isActive()) {
1180             ASSERT(!m_visualUpdatesAllowed);
1181
1182             if (!view()->visualUpdatesAllowedByClient())
1183                 return;
1184
1185             setVisualUpdatesAllowed(true);
1186         } else
1187             ASSERT(m_visualUpdatesAllowed);
1188         break;
1189     }
1190 }
1191     
1192 void Document::setVisualUpdatesAllowed(bool visualUpdatesAllowed)
1193 {
1194     if (m_visualUpdatesAllowed == visualUpdatesAllowed)
1195         return;
1196
1197     m_visualUpdatesAllowed = visualUpdatesAllowed;
1198
1199     if (visualUpdatesAllowed)
1200         m_visualUpdatesSuppressionTimer.stop();
1201     else
1202         m_visualUpdatesSuppressionTimer.startOneShot(settings()->incrementalRenderingSuppressionTimeoutInSeconds());
1203
1204     if (!visualUpdatesAllowed)
1205         return;
1206
1207     FrameView* frameView = view();
1208     bool needsLayout = frameView && renderView() && (frameView->layoutPending() || renderView()->needsLayout());
1209     if (needsLayout)
1210         updateLayout();
1211
1212     if (Page* page = this->page()) {
1213         if (frame()->isMainFrame()) {
1214             frameView->addPaintPendingMilestones(DidFirstPaintAfterSuppressedIncrementalRendering);
1215             if (page->requestedLayoutMilestones() & DidFirstLayoutAfterSuppressedIncrementalRendering)
1216                 frame()->loader().didLayout(DidFirstLayoutAfterSuppressedIncrementalRendering);
1217         }
1218     }
1219
1220     if (view())
1221         view()->updateCompositingLayersAfterLayout();
1222
1223     if (RenderView* renderView = this->renderView())
1224         renderView->repaintViewAndCompositedLayers();
1225
1226     if (Frame* frame = this->frame())
1227         frame->loader().forcePageTransitionIfNeeded();
1228 }
1229
1230 void Document::visualUpdatesSuppressionTimerFired()
1231 {
1232     ASSERT(!m_visualUpdatesAllowed);
1233
1234     // If the client is extending the visual update suppression period explicitly, the
1235     // watchdog should not re-enable visual updates itself, but should wait for the client.
1236     if (!view()->visualUpdatesAllowedByClient())
1237         return;
1238
1239     setVisualUpdatesAllowed(true);
1240 }
1241
1242 void Document::setVisualUpdatesAllowedByClient(bool visualUpdatesAllowedByClient)
1243 {
1244     // We should only re-enable visual updates if ReadyState is Completed or the watchdog timer has fired,
1245     // both of which we can determine by looking at the timer.
1246
1247     if (visualUpdatesAllowedByClient && !m_visualUpdatesSuppressionTimer.isActive() && !visualUpdatesAllowed())
1248         setVisualUpdatesAllowed(true);
1249 }
1250
1251 AtomicString Document::encoding() const
1252 {
1253     if (TextResourceDecoder* d = decoder())
1254         return d->encoding().domName();
1255     return nullAtom;
1256 }
1257
1258 String Document::characterSetForBindings() const
1259 {
1260     AtomicString name = encoding();
1261     if (!name.isNull())
1262         return name;
1263     return UTF8Encoding().domName();
1264 }
1265
1266 String Document::defaultCharset() const
1267 {
1268     if (Settings* settings = this->settings())
1269         return settings->defaultTextEncodingName();
1270     return String();
1271 }
1272
1273 void Document::setCharset(const String& charset)
1274 {
1275     if (!decoder())
1276         return;
1277     decoder()->setEncoding(charset, TextResourceDecoder::UserChosenEncoding);
1278 }
1279
1280 void Document::setContentLanguage(const String& language)
1281 {
1282     if (m_contentLanguage == language)
1283         return;
1284     m_contentLanguage = language;
1285
1286     // Recalculate style so language is used when selecting the initial font.
1287     styleResolverChanged(DeferRecalcStyle);
1288 }
1289
1290 void Document::setXMLVersion(const String& version, ExceptionCode& ec)
1291 {
1292     if (!implementation().hasFeature("XML", String())) {
1293         ec = NOT_SUPPORTED_ERR;
1294         return;
1295     }
1296
1297     if (!XMLDocumentParser::supportsXMLVersion(version)) {
1298         ec = NOT_SUPPORTED_ERR;
1299         return;
1300     }
1301
1302     m_xmlVersion = version;
1303 }
1304
1305 void Document::setXMLStandalone(bool standalone, ExceptionCode& ec)
1306 {
1307     if (!implementation().hasFeature("XML", String())) {
1308         ec = NOT_SUPPORTED_ERR;
1309         return;
1310     }
1311
1312     m_xmlStandalone = standalone ? Standalone : NotStandalone;
1313 }
1314
1315 void Document::setDocumentURI(const String& uri)
1316 {
1317     // This property is read-only from JavaScript, but writable from Objective-C.
1318     m_documentURI = uri;
1319     updateBaseURL();
1320 }
1321
1322 void Document::setContent(const String& content)
1323 {
1324     open();
1325     // FIXME: This should probably use insert(), but that's (intentionally)
1326     // not implemented for the XML parser as it's normally synonymous with
1327     // document.write(). append() will end up yielding, but close() will
1328     // pump the tokenizer syncrhonously and finish the parse.
1329     m_parser->append(content.impl());
1330     close();
1331 }
1332
1333 String Document::suggestedMIMEType() const
1334 {
1335     if (isXHTMLDocument())
1336         return ASCIILiteral("application/xhtml+xml");
1337     if (isSVGDocument())
1338         return ASCIILiteral("image/svg+xml");
1339     if (xmlStandalone())
1340         return ASCIILiteral("text/xml");
1341     if (isHTMLDocument())
1342         return ASCIILiteral("text/html");
1343     if (DocumentLoader* loader = this->loader())
1344         return loader->responseMIMEType();
1345     return String();
1346 }
1347
1348 void Document::overrideMIMEType(const String& mimeType)
1349 {
1350     m_overriddenMIMEType = mimeType;
1351 }
1352
1353 String Document::contentType() const
1354 {
1355     if (!m_overriddenMIMEType.isNull())
1356         return m_overriddenMIMEType;
1357
1358     if (DocumentLoader* documentLoader = loader())
1359         return documentLoader->currentContentType();
1360
1361     String mimeType = suggestedMIMEType();
1362     if (!mimeType.isNull())
1363         return mimeType;
1364
1365     return ASCIILiteral("application/xml");
1366 }
1367
1368 Node* Document::nodeFromPoint(const LayoutPoint& clientPoint, LayoutPoint* localPoint)
1369 {
1370     if (!frame() || !view())
1371         return nullptr;
1372     
1373     Frame& frame = *this->frame();
1374     
1375     float scaleFactor = frame.pageZoomFactor() * frame.frameScaleFactor();
1376
1377     LayoutPoint contentsPoint = clientPoint;
1378     contentsPoint.scale(scaleFactor, scaleFactor);
1379     contentsPoint.moveBy(view()->contentsScrollPosition());
1380
1381     LayoutRect visibleRect;
1382 #if PLATFORM(IOS)
1383     visibleRect = view()->unobscuredContentRect();
1384 #else
1385     visibleRect = view()->visibleContentRect();
1386 #endif
1387     if (!visibleRect.contains(contentsPoint))
1388         return nullptr;
1389
1390     HitTestResult result(contentsPoint);
1391     renderView()->hitTest(HitTestRequest(), result);
1392
1393     if (localPoint)
1394         *localPoint = result.localPoint();
1395
1396     return result.innerNode();
1397 }
1398
1399 Element* Document::elementFromPoint(const LayoutPoint& clientPoint)
1400 {
1401     if (!hasLivingRenderTree())
1402         return nullptr;
1403
1404     Node* node = nodeFromPoint(clientPoint);
1405     while (node && !is<Element>(*node))
1406         node = node->parentNode();
1407
1408     if (node)
1409         node = ancestorInThisScope(node);
1410
1411     return downcast<Element>(node);
1412 }
1413
1414 RefPtr<Range> Document::caretRangeFromPoint(int x, int y)
1415 {
1416     return caretRangeFromPoint(LayoutPoint(x, y));
1417 }
1418
1419 RefPtr<Range> Document::caretRangeFromPoint(const LayoutPoint& clientPoint)
1420 {
1421     if (!hasLivingRenderTree())
1422         return nullptr;
1423
1424     LayoutPoint localPoint;
1425     Node* node = nodeFromPoint(clientPoint, &localPoint);
1426     if (!node)
1427         return nullptr;
1428
1429     Node* shadowAncestorNode = ancestorInThisScope(node);
1430     if (shadowAncestorNode != node) {
1431         unsigned offset = shadowAncestorNode->computeNodeIndex();
1432         ContainerNode* container = shadowAncestorNode->parentNode();
1433         return Range::create(*this, container, offset, container, offset);
1434     }
1435
1436     RenderObject* renderer = node->renderer();
1437     if (!renderer)
1438         return nullptr;
1439     VisiblePosition visiblePosition = renderer->positionForPoint(localPoint, nullptr);
1440     if (visiblePosition.isNull())
1441         return nullptr;
1442
1443     Position rangeCompliantPosition = visiblePosition.deepEquivalent().parentAnchoredEquivalent();
1444     return Range::create(*this, rangeCompliantPosition, rangeCompliantPosition);
1445 }
1446
1447 Element* Document::scrollingElement()
1448 {
1449     // FIXME: When we fix https://bugs.webkit.org/show_bug.cgi?id=106133, this should be replaced with the full implementation
1450     // of Document.scrollingElement() as specified at http://dev.w3.org/csswg/cssom-view/#dom-document-scrollingelement.
1451
1452     return body();
1453 }
1454
1455 /*
1456  * Performs three operations:
1457  *  1. Convert control characters to spaces
1458  *  2. Trim leading and trailing spaces
1459  *  3. Collapse internal whitespace.
1460  */
1461 template <typename CharacterType>
1462 static inline StringWithDirection canonicalizedTitle(Document* document, const StringWithDirection& titleWithDirection)
1463 {
1464     const String& title = titleWithDirection.string();
1465     const CharacterType* characters = title.characters<CharacterType>();
1466     unsigned length = title.length();
1467     unsigned i;
1468
1469     StringBuffer<CharacterType> buffer(length);
1470     unsigned builderIndex = 0;
1471
1472     // Skip leading spaces and leading characters that would convert to spaces
1473     for (i = 0; i < length; ++i) {
1474         CharacterType c = characters[i];
1475         if (!(c <= 0x20 || c == 0x7F))
1476             break;
1477     }
1478
1479     if (i == length)
1480         return StringWithDirection();
1481
1482     // Replace control characters with spaces, and backslashes with currency symbols, and collapse whitespace.
1483     bool previousCharWasWS = false;
1484     for (; i < length; ++i) {
1485         CharacterType c = characters[i];
1486         if (c <= 0x20 || c == 0x7F || (U_GET_GC_MASK(c) & (U_GC_ZL_MASK | U_GC_ZP_MASK))) {
1487             if (previousCharWasWS)
1488                 continue;
1489             buffer[builderIndex++] = ' ';
1490             previousCharWasWS = true;
1491         } else {
1492             buffer[builderIndex++] = c;
1493             previousCharWasWS = false;
1494         }
1495     }
1496
1497     // Strip trailing spaces
1498     while (builderIndex > 0) {
1499         --builderIndex;
1500         if (buffer[builderIndex] != ' ')
1501             break;
1502     }
1503
1504     if (!builderIndex && buffer[builderIndex] == ' ')
1505         return StringWithDirection();
1506
1507     buffer.shrink(builderIndex + 1);
1508
1509     // Replace the backslashes with currency symbols if the encoding requires it.
1510     document->displayBufferModifiedByEncoding(buffer.characters(), buffer.length());
1511     
1512     return StringWithDirection(String::adopt(buffer), titleWithDirection.direction());
1513 }
1514
1515 void Document::updateTitle(const StringWithDirection& title)
1516 {
1517     if (m_rawTitle == title)
1518         return;
1519
1520     m_rawTitle = title;
1521
1522     if (m_rawTitle.string().isEmpty())
1523         m_title = StringWithDirection();
1524     else {
1525         if (m_rawTitle.string().is8Bit())
1526             m_title = canonicalizedTitle<LChar>(this, m_rawTitle);
1527         else
1528             m_title = canonicalizedTitle<UChar>(this, m_rawTitle);
1529     }
1530     if (DocumentLoader* loader = this->loader())
1531         loader->setTitle(m_title);
1532 }
1533
1534 void Document::updateTitleFromTitleElement()
1535 {
1536     if (!m_titleElement) {
1537         updateTitle(StringWithDirection());
1538         return;
1539     }
1540
1541     if (is<HTMLTitleElement>(*m_titleElement))
1542         updateTitle(downcast<HTMLTitleElement>(*m_titleElement).textWithDirection());
1543     else if (is<SVGTitleElement>(*m_titleElement)) {
1544         // FIXME: does SVG have a title text direction?
1545         updateTitle(StringWithDirection(downcast<SVGTitleElement>(*m_titleElement).textContent(), LTR));
1546     }
1547 }
1548
1549 void Document::setTitle(const String& title)
1550 {
1551     if (!isHTMLDocument() && !isXHTMLDocument())
1552         m_titleElement = nullptr;
1553     else if (!m_titleElement) {
1554         auto* headElement = head();
1555         if (!headElement)
1556             return;
1557         m_titleElement = createElement(titleTag, false);
1558         headElement->appendChild(*m_titleElement, ASSERT_NO_EXCEPTION);
1559     }
1560
1561     // The DOM API has no method of specifying direction, so assume LTR.
1562     updateTitle(StringWithDirection(title, LTR));
1563
1564     if (is<HTMLTitleElement>(m_titleElement.get()))
1565         downcast<HTMLTitleElement>(*m_titleElement).setText(title);
1566 }
1567
1568 void Document::updateTitleElement(Element* newTitleElement)
1569 {
1570     // Only allow the first title element in tree order to change the title -- others have no effect.
1571     if (m_titleElement) {
1572         if (isHTMLDocument() || isXHTMLDocument())
1573             m_titleElement = descendantsOfType<HTMLTitleElement>(*this).first();
1574         else if (isSVGDocument())
1575             m_titleElement = descendantsOfType<SVGTitleElement>(*this).first();
1576     } else
1577         m_titleElement = newTitleElement;
1578
1579     updateTitleFromTitleElement();
1580 }
1581
1582 void Document::titleElementAdded(Element& titleElement)
1583 {
1584     if (m_titleElement == &titleElement)
1585         return;
1586
1587     updateTitleElement(&titleElement);
1588 }
1589
1590 void Document::titleElementRemoved(Element& titleElement)
1591 {
1592     if (m_titleElement != &titleElement)
1593         return;
1594
1595     updateTitleElement(nullptr);
1596 }
1597
1598 void Document::titleElementTextChanged(Element& titleElement)
1599 {
1600     if (m_titleElement != &titleElement)
1601         return;
1602
1603     updateTitleFromTitleElement();
1604 }
1605
1606 void Document::registerForVisibilityStateChangedCallbacks(Element* element)
1607 {
1608     m_visibilityStateCallbackElements.add(element);
1609 }
1610
1611 void Document::unregisterForVisibilityStateChangedCallbacks(Element* element)
1612 {
1613     m_visibilityStateCallbackElements.remove(element);
1614 }
1615
1616 void Document::visibilityStateChanged()
1617 {
1618     dispatchEvent(Event::create(eventNames().visibilitychangeEvent, false, false));
1619     for (auto* element : m_visibilityStateCallbackElements)
1620         element->visibilityStateChanged();
1621 }
1622
1623 PageVisibilityState Document::pageVisibilityState() const
1624 {
1625     // The visibility of the document is inherited from the visibility of the
1626     // page. If there is no page associated with the document, we will assume
1627     // that the page is hidden, as specified by the spec:
1628     // http://dvcs.w3.org/hg/webperf/raw-file/tip/specs/PageVisibility/Overview.html#dom-document-hidden
1629     if (!m_frame || !m_frame->page())
1630         return PageVisibilityStateHidden;
1631     return m_frame->page()->visibilityState();
1632 }
1633
1634 String Document::visibilityState() const
1635 {
1636     return pageVisibilityStateString(pageVisibilityState());
1637 }
1638
1639 bool Document::hidden() const
1640 {
1641     return pageVisibilityState() != PageVisibilityStateVisible;
1642 }
1643
1644
1645 #if ENABLE(VIDEO)
1646 void Document::registerForAllowsMediaDocumentInlinePlaybackChangedCallbacks(HTMLMediaElement& element)
1647 {
1648     m_allowsMediaDocumentInlinePlaybackElements.add(&element);
1649 }
1650
1651 void Document::unregisterForAllowsMediaDocumentInlinePlaybackChangedCallbacks(HTMLMediaElement& element)
1652 {
1653     m_allowsMediaDocumentInlinePlaybackElements.remove(&element);
1654 }
1655
1656 void Document::allowsMediaDocumentInlinePlaybackChanged()
1657 {
1658     for (auto* element : m_allowsMediaDocumentInlinePlaybackElements)
1659         element->allowsMediaDocumentInlinePlaybackChanged();
1660 }
1661 #endif
1662
1663 #if ENABLE(CSP_NEXT)
1664 DOMSecurityPolicy& Document::securityPolicy()
1665 {
1666     if (!m_domSecurityPolicy)
1667         m_domSecurityPolicy = DOMSecurityPolicy::create(this);
1668     return *m_domSecurityPolicy;
1669 }
1670 #endif
1671
1672 String Document::nodeName() const
1673 {
1674     return "#document";
1675 }
1676
1677 Node::NodeType Document::nodeType() const
1678 {
1679     return DOCUMENT_NODE;
1680 }
1681
1682 FormController& Document::formController()
1683 {
1684     if (!m_formController)
1685         m_formController = std::make_unique<FormController>();
1686     return *m_formController;
1687 }
1688
1689 Vector<String> Document::formElementsState() const
1690 {
1691     if (!m_formController)
1692         return Vector<String>();
1693     return m_formController->formElementsState();
1694 }
1695
1696 void Document::setStateForNewFormElements(const Vector<String>& stateVector)
1697 {
1698     if (!stateVector.size() && !m_formController)
1699         return;
1700     formController().setStateForNewFormElements(stateVector);
1701 }
1702
1703 FrameView* Document::view() const
1704 {
1705     return m_frame ? m_frame->view() : nullptr;
1706 }
1707
1708 Page* Document::page() const
1709 {
1710     return m_frame ? m_frame->page() : nullptr;
1711 }
1712
1713 Settings* Document::settings() const
1714 {
1715     return m_frame ? &m_frame->settings() : nullptr;
1716 }
1717
1718 Ref<Range> Document::createRange()
1719 {
1720     return Range::create(*this);
1721 }
1722
1723 RefPtr<NodeIterator> Document::createNodeIterator(Node* root, unsigned long whatToShow, RefPtr<NodeFilter>&& filter, bool expandEntityReferences, ExceptionCode& ec)
1724 {
1725     if (!root) {
1726         ec = TypeError;
1727         return nullptr;
1728     }
1729     return NodeIterator::create(root, whatToShow, WTF::move(filter), expandEntityReferences);
1730 }
1731
1732 RefPtr<NodeIterator> Document::createNodeIterator(Node* root, unsigned long whatToShow, RefPtr<NodeFilter>&& filter, ExceptionCode& ec)
1733 {
1734     return createNodeIterator(root, whatToShow, WTF::move(filter), false, ec);
1735 }
1736
1737 RefPtr<NodeIterator> Document::createNodeIterator(Node* root, unsigned long whatToShow, ExceptionCode& ec)
1738 {
1739     return createNodeIterator(root, whatToShow, nullptr, false, ec);
1740 }
1741
1742 RefPtr<NodeIterator> Document::createNodeIterator(Node* root, ExceptionCode& ec)
1743 {
1744     return createNodeIterator(root, 0xFFFFFFFF, nullptr, false, ec);
1745 }
1746
1747 RefPtr<TreeWalker> Document::createTreeWalker(Node* root, unsigned long whatToShow, RefPtr<NodeFilter>&& filter, bool expandEntityReferences, ExceptionCode& ec)
1748 {
1749     if (!root) {
1750         ec = TypeError;
1751         return nullptr;
1752     }
1753     return TreeWalker::create(root, whatToShow, WTF::move(filter), expandEntityReferences);
1754 }
1755
1756 RefPtr<TreeWalker> Document::createTreeWalker(Node* root, unsigned long whatToShow, RefPtr<NodeFilter>&& filter, ExceptionCode& ec)
1757 {
1758     return createTreeWalker(root, whatToShow, WTF::move(filter), false, ec);
1759 }
1760
1761 RefPtr<TreeWalker> Document::createTreeWalker(Node* root, unsigned long whatToShow, ExceptionCode& ec)
1762 {
1763     return createTreeWalker(root, whatToShow, nullptr, false, ec);
1764 }
1765
1766 RefPtr<TreeWalker> Document::createTreeWalker(Node* root, ExceptionCode& ec)
1767 {
1768     return createTreeWalker(root, 0xFFFFFFFF, nullptr, false, ec);
1769 }
1770
1771 void Document::scheduleForcedStyleRecalc()
1772 {
1773     m_pendingStyleRecalcShouldForce = true;
1774     scheduleStyleRecalc();
1775 }
1776
1777 void Document::scheduleStyleRecalc()
1778 {
1779     if (m_styleRecalcTimer.isActive() || inPageCache())
1780         return;
1781
1782     ASSERT(childNeedsStyleRecalc() || m_pendingStyleRecalcShouldForce);
1783
1784     // FIXME: Why on earth is this here? This is clearly misplaced.
1785     invalidateAccessKeyMap();
1786     
1787     m_styleRecalcTimer.startOneShot(0);
1788
1789     InspectorInstrumentation::didScheduleStyleRecalculation(*this);
1790 }
1791
1792 void Document::unscheduleStyleRecalc()
1793 {
1794     ASSERT(!childNeedsStyleRecalc());
1795
1796     m_styleRecalcTimer.stop();
1797     m_pendingStyleRecalcShouldForce = false;
1798 }
1799
1800 bool Document::hasPendingStyleRecalc() const
1801 {
1802     return m_styleRecalcTimer.isActive() && !m_inStyleRecalc;
1803 }
1804
1805 bool Document::hasPendingForcedStyleRecalc() const
1806 {
1807     return m_styleRecalcTimer.isActive() && m_pendingStyleRecalcShouldForce;
1808 }
1809
1810 void Document::recalcStyle(Style::Change change)
1811 {
1812     ASSERT(!view() || !view()->isPainting());
1813
1814     // NOTE: XSL code seems to be the only client stumbling in here without a RenderView.
1815     if (!m_renderView)
1816         return;
1817
1818     FrameView& frameView = m_renderView->frameView();
1819     Ref<FrameView> protect(frameView);
1820     if (frameView.isPainting())
1821         return;
1822     
1823     if (m_inStyleRecalc)
1824         return; // Guard against re-entrancy. -dwh
1825
1826     RenderView::RepaintRegionAccumulator repaintRegionAccumulator(renderView());
1827     AnimationUpdateBlock animationUpdateBlock(&m_frame->animation());
1828
1829     // FIXME: We should update style on our ancestor chain before proceeding (especially for seamless),
1830     // however doing so currently causes several tests to crash, as Frame::setDocument calls Document::attach
1831     // before setting the DOMWindow on the Frame, or the SecurityOrigin on the document. The attach, in turn
1832     // resolves style (here) and then when we resolve style on the parent chain, we may end up
1833     // re-attaching our containing iframe, which when asked HTMLFrameElementBase::isURLAllowed
1834     // hits a null-dereference due to security code always assuming the document has a SecurityOrigin.
1835
1836     m_styleSheetCollection.flushPendingUpdates();
1837
1838     frameView.willRecalcStyle();
1839
1840     InspectorInstrumentationCookie cookie = InspectorInstrumentation::willRecalculateStyle(*this);
1841
1842     // FIXME: We never reset this flags.
1843     if (m_elementSheet && m_elementSheet->contents().usesRemUnits())
1844         m_styleSheetCollection.setUsesRemUnit(true);
1845     // We don't call setUsesStyleBasedEditability here because the whole point of the flag is to avoid style recalc.
1846     // i.e. updating the flag here would be too late.
1847
1848     m_inStyleRecalc = true;
1849     bool updatedCompositingLayers = false;
1850     {
1851         Style::PostResolutionCallbackDisabler disabler(*this);
1852         WidgetHierarchyUpdatesSuspensionScope suspendWidgetHierarchyUpdates;
1853
1854         if (m_pendingStyleRecalcShouldForce)
1855             change = Style::Force;
1856
1857         if (change == Style::Force) {
1858             // This may get set again during style resolve.
1859             m_hasNodesWithPlaceholderStyle = false;
1860         }
1861
1862         Style::resolveTree(*this, change);
1863
1864         updatedCompositingLayers = frameView.updateCompositingLayersAfterStyleChange();
1865
1866         clearNeedsStyleRecalc();
1867         clearChildNeedsStyleRecalc();
1868         unscheduleStyleRecalc();
1869
1870         m_inStyleRecalc = false;
1871
1872         // Pseudo element removal and similar may only work with these flags still set. Reset them after the style recalc.
1873         if (m_styleResolver)
1874             m_styleSheetCollection.resetCSSFeatureFlags();
1875     }
1876
1877     // If we wanted to call implicitClose() during recalcStyle, do so now that we're finished.
1878     if (m_closeAfterStyleRecalc) {
1879         m_closeAfterStyleRecalc = false;
1880         implicitClose();
1881     }
1882     
1883     ++m_styleRecalcCount;
1884
1885     InspectorInstrumentation::didRecalculateStyle(cookie);
1886
1887     // Some animated images may now be inside the viewport due to style recalc,
1888     // resume them if necessary if there is no layout pending. Otherwise, we'll
1889     // check if they need to be resumed after layout.
1890     if (updatedCompositingLayers && !frameView.needsLayout())
1891         frameView.viewportContentsChanged();
1892
1893     if (!frameView.needsLayout())
1894         frameView.frame().selection().updateAppearanceAfterLayout();
1895
1896     // As a result of the style recalculation, the currently hovered element might have been
1897     // detached (for example, by setting display:none in the :hover style), schedule another mouseMove event
1898     // to check if any other elements ended up under the mouse pointer due to re-layout.
1899     if (m_hoveredElement && !m_hoveredElement->renderer())
1900         frameView.frame().mainFrame().eventHandler().dispatchFakeMouseMoveEventSoon();
1901 }
1902
1903 void Document::updateStyleIfNeeded()
1904 {
1905     ASSERT(isMainThread());
1906     ASSERT(!view() || !view()->isPainting());
1907
1908     if (!view() || view()->isInLayout())
1909         return;
1910
1911     if (m_optimizedStyleSheetUpdateTimer.isActive())
1912         styleResolverChanged(RecalcStyleIfNeeded);
1913
1914     if (!needsStyleRecalc())
1915         return;
1916
1917     recalcStyle(Style::NoChange);
1918 }
1919
1920 void Document::updateLayout()
1921 {
1922     ASSERT(isMainThread());
1923
1924     FrameView* frameView = view();
1925     if (frameView && frameView->isInLayout()) {
1926         // View layout should not be re-entrant.
1927         ASSERT_NOT_REACHED();
1928         return;
1929     }
1930
1931     RenderView::RepaintRegionAccumulator repaintRegionAccumulator(renderView());
1932
1933     if (HTMLFrameOwnerElement* owner = ownerElement())
1934         owner->document().updateLayout();
1935
1936     updateStyleIfNeeded();
1937
1938     StackStats::LayoutCheckPoint layoutCheckPoint;
1939
1940     // Only do a layout if changes have occurred that make it necessary.      
1941     if (frameView && renderView() && (frameView->layoutPending() || renderView()->needsLayout()))
1942         frameView->layout();
1943 }
1944
1945 // FIXME: This is a bad idea and needs to be removed eventually.
1946 // Other browsers load stylesheets before they continue parsing the web page.
1947 // Since we don't, we can run JavaScript code that needs answers before the
1948 // stylesheets are loaded. Doing a layout ignoring the pending stylesheets
1949 // lets us get reasonable answers. The long term solution to this problem is
1950 // to instead suspend JavaScript execution.
1951 void Document::updateLayoutIgnorePendingStylesheets(Document::RunPostLayoutTasks runPostLayoutTasks)
1952 {
1953     bool oldIgnore = m_ignorePendingStylesheets;
1954
1955     if (!haveStylesheetsLoaded()) {
1956         m_ignorePendingStylesheets = true;
1957         // FIXME: We are willing to attempt to suppress painting with outdated style info only once.  Our assumption is that it would be
1958         // dangerous to try to stop it a second time, after page content has already been loaded and displayed
1959         // with accurate style information.  (Our suppression involves blanking the whole page at the
1960         // moment.  If it were more refined, we might be able to do something better.)
1961         // It's worth noting though that this entire method is a hack, since what we really want to do is
1962         // suspend JS instead of doing a layout with inaccurate information.
1963         HTMLElement* bodyElement = bodyOrFrameset();
1964         if (bodyElement && !bodyElement->renderer() && m_pendingSheetLayout == NoLayoutWithPendingSheets) {
1965             m_pendingSheetLayout = DidLayoutWithPendingSheets;
1966             styleResolverChanged(RecalcStyleImmediately);
1967         } else if (m_hasNodesWithPlaceholderStyle)
1968             // If new nodes have been added or style recalc has been done with style sheets still pending, some nodes 
1969             // may not have had their real style calculated yet. Normally this gets cleaned when style sheets arrive 
1970             // but here we need up-to-date style immediately.
1971             recalcStyle(Style::Force);
1972     }
1973
1974     updateLayout();
1975
1976     if (runPostLayoutTasks == RunPostLayoutTasks::Synchronously && view())
1977         view()->flushAnyPendingPostLayoutTasks();
1978
1979     m_ignorePendingStylesheets = oldIgnore;
1980 }
1981
1982 Ref<RenderStyle> Document::styleForElementIgnoringPendingStylesheets(Element* element)
1983 {
1984     ASSERT_ARG(element, &element->document() == this);
1985
1986     // On iOS request delegates called during styleForElement may result in re-entering WebKit and killing the style resolver.
1987     ResourceLoadScheduler::Suspender suspender(*platformStrategies()->loaderStrategy()->resourceLoadScheduler());
1988
1989     TemporaryChange<bool> change(m_ignorePendingStylesheets, true);
1990     return ensureStyleResolver().styleForElement(element, element->parentNode() ? element->parentNode()->computedStyle() : nullptr);
1991 }
1992
1993 bool Document::updateLayoutIfDimensionsOutOfDate(Element& element, DimensionsCheck dimensionsCheck)
1994 {
1995     ASSERT(isMainThread());
1996     
1997     // If the stylesheets haven't loaded, just give up and do a full layout ignoring pending stylesheets.
1998     if (!haveStylesheetsLoaded()) {
1999         updateLayoutIgnorePendingStylesheets();
2000         return true;
2001     }
2002     
2003     // Check for re-entrancy and assert (same code that is in updateLayout()).
2004     FrameView* frameView = view();
2005     if (frameView && frameView->isInLayout()) {
2006         // View layout should not be re-entrant.
2007         ASSERT_NOT_REACHED();
2008         return true;
2009     }
2010     
2011     RenderView::RepaintRegionAccumulator repaintRegionAccumulator(renderView());
2012     
2013     // Mimic the structure of updateLayout(), but at each step, see if we have been forced into doing a full
2014     // layout.
2015     bool requireFullLayout = false;
2016     if (HTMLFrameOwnerElement* owner = ownerElement()) {
2017         if (owner->document().updateLayoutIfDimensionsOutOfDate(*owner))
2018             requireFullLayout = true;
2019     }
2020     
2021     updateStyleIfNeeded();
2022
2023     RenderObject* renderer = element.renderer();
2024     if (!renderer || renderer->needsLayout() || element.renderNamedFlowFragment()) {
2025         // If we don't have a renderer or if the renderer needs layout for any reason, give up.
2026         // Named flows can have auto height, so don't try to enforce the optimization in this case.
2027         // The 2-pass nature of auto height named flow layout means the region may not be dirty yet.
2028         requireFullLayout = true;
2029     }
2030
2031     bool isVertical = renderer && !renderer->isHorizontalWritingMode();
2032     bool checkingLogicalWidth = ((dimensionsCheck & WidthDimensionsCheck) && !isVertical) || ((dimensionsCheck & HeightDimensionsCheck) && isVertical);
2033     bool checkingLogicalHeight = ((dimensionsCheck & HeightDimensionsCheck) && !isVertical) || ((dimensionsCheck & WidthDimensionsCheck) && isVertical);
2034     bool hasSpecifiedLogicalHeight = renderer && renderer->style().logicalMinHeight() == Length(0, Fixed) && renderer->style().logicalHeight().isFixed() && renderer->style().logicalMaxHeight().isAuto();
2035     
2036     if (!requireFullLayout) {
2037         RenderBox* previousBox = nullptr;
2038         RenderBox* currentBox = nullptr;
2039         
2040         // Check our containing block chain. If anything in the chain needs a layout, then require a full layout.
2041         for (RenderObject* currRenderer = element.renderer(); currRenderer && !currRenderer->isRenderView(); currRenderer = currRenderer->container()) {
2042             
2043             // Require the entire container chain to be boxes.
2044             if (!is<RenderBox>(currRenderer)) {
2045                 requireFullLayout = true;
2046                 break;
2047             }
2048             
2049             previousBox = currentBox;
2050             currentBox = downcast<RenderBox>(currRenderer);
2051             
2052             // If a box needs layout for itself or if a box has changed children and sizes its width to
2053             // its content, then require a full layout.
2054             if (currentBox->selfNeedsLayout() ||
2055                 (checkingLogicalWidth && currRenderer->needsLayout() && currentBox->sizesLogicalWidthToFitContent(MainOrPreferredSize))) {
2056                 requireFullLayout = true;
2057                 break;
2058             }
2059             
2060             // If a block contains floats and the child's height isn't specified, then
2061             // give up also, since our height could end up being influenced by the floats.
2062             if (checkingLogicalHeight && !hasSpecifiedLogicalHeight && currentBox->isRenderBlockFlow()) {
2063                 RenderBlockFlow* currentBlockFlow = downcast<RenderBlockFlow>(currentBox);
2064                 if (currentBlockFlow->containsFloats() && previousBox && !previousBox->isFloatingOrOutOfFlowPositioned()) {
2065                     requireFullLayout = true;
2066                     break;
2067                 }
2068             }
2069             
2070             if (!currentBox->isRenderBlockFlow() || currentBox->flowThreadContainingBlock() || currentBox->isWritingModeRoot()) {
2071                 // FIXME: For now require only block flows all the way back to the root. This limits the optimization
2072                 // for now, and we'll expand it in future patches to apply to more and more scenarios.
2073                 // Disallow regions/columns from having the optimization.
2074                 // Give up if the writing mode changes at all in the containing block chain.
2075                 requireFullLayout = true;
2076                 break;
2077             }
2078             
2079             if (currRenderer == frameView->layoutRoot())
2080                 break;
2081         }
2082     }
2083     
2084     StackStats::LayoutCheckPoint layoutCheckPoint;
2085
2086     // Only do a layout if changes have occurred that make it necessary.      
2087     if (requireFullLayout && frameView && renderView() && (frameView->layoutPending() || renderView()->needsLayout()))
2088         frameView->layout();
2089     
2090     return requireFullLayout;
2091 }
2092
2093 bool Document::isPageBoxVisible(int pageIndex)
2094 {
2095     Ref<RenderStyle> pageStyle(ensureStyleResolver().styleForPage(pageIndex));
2096     return pageStyle->visibility() != HIDDEN; // display property doesn't apply to @page.
2097 }
2098
2099 void Document::pageSizeAndMarginsInPixels(int pageIndex, IntSize& pageSize, int& marginTop, int& marginRight, int& marginBottom, int& marginLeft)
2100 {
2101     RefPtr<RenderStyle> style = ensureStyleResolver().styleForPage(pageIndex);
2102
2103     int width = pageSize.width();
2104     int height = pageSize.height();
2105     switch (style->pageSizeType()) {
2106     case PAGE_SIZE_AUTO:
2107         break;
2108     case PAGE_SIZE_AUTO_LANDSCAPE:
2109         if (width < height)
2110             std::swap(width, height);
2111         break;
2112     case PAGE_SIZE_AUTO_PORTRAIT:
2113         if (width > height)
2114             std::swap(width, height);
2115         break;
2116     case PAGE_SIZE_RESOLVED: {
2117         LengthSize size = style->pageSize();
2118         ASSERT(size.width().isFixed());
2119         ASSERT(size.height().isFixed());
2120         width = valueForLength(size.width(), 0);
2121         height = valueForLength(size.height(), 0);
2122         break;
2123     }
2124     default:
2125         ASSERT_NOT_REACHED();
2126     }
2127     pageSize = IntSize(width, height);
2128
2129     // The percentage is calculated with respect to the width even for margin top and bottom.
2130     // http://www.w3.org/TR/CSS2/box.html#margin-properties
2131     marginTop = style->marginTop().isAuto() ? marginTop : intValueForLength(style->marginTop(), width);
2132     marginRight = style->marginRight().isAuto() ? marginRight : intValueForLength(style->marginRight(), width);
2133     marginBottom = style->marginBottom().isAuto() ? marginBottom : intValueForLength(style->marginBottom(), width);
2134     marginLeft = style->marginLeft().isAuto() ? marginLeft : intValueForLength(style->marginLeft(), width);
2135 }
2136
2137 void Document::createStyleResolver()
2138 {
2139     bool matchAuthorAndUserStyles = true;
2140     if (Settings* settings = this->settings())
2141         matchAuthorAndUserStyles = settings->authorAndUserStylesEnabled();
2142     m_styleResolver = std::make_unique<StyleResolver>(*this, matchAuthorAndUserStyles);
2143     m_styleSheetCollection.combineCSSFeatureFlags();
2144 }
2145
2146 void Document::fontsNeedUpdate(FontSelector&)
2147 {
2148     if (m_styleResolver)
2149         m_styleResolver->invalidateMatchedPropertiesCache();
2150     if (inPageCache() || !renderView())
2151         return;
2152     scheduleForcedStyleRecalc();
2153 }
2154
2155 CSSFontSelector& Document::fontSelector()
2156 {
2157     if (!m_fontSelector) {
2158         m_fontSelector = CSSFontSelector::create(*this);
2159         m_fontSelector->registerForInvalidationCallbacks(*this);
2160     }
2161     return *m_fontSelector;
2162 }
2163
2164 void Document::clearStyleResolver()
2165 {
2166     m_styleResolver = nullptr;
2167
2168     // FIXME: It would be better if the FontSelector could survive this operation.
2169     if (m_fontSelector) {
2170         m_fontSelector->clearDocument();
2171         m_fontSelector->unregisterForInvalidationCallbacks(*this);
2172         m_fontSelector = nullptr;
2173     }
2174 }
2175
2176 void Document::createRenderTree()
2177 {
2178     ASSERT(!renderView());
2179     ASSERT(!m_inPageCache);
2180     ASSERT(!m_axObjectCache || this != &topDocument());
2181
2182     if (m_isNonRenderedPlaceholder)
2183         return;
2184
2185     // FIXME: It would be better if we could pass the resolved document style directly here.
2186     m_renderView = createRenderer<RenderView>(*this, RenderStyle::create());
2187     Node::setRenderer(m_renderView.get());
2188
2189     renderView()->setIsInWindow(true);
2190
2191     recalcStyle(Style::Force);
2192 }
2193
2194 void Document::didBecomeCurrentDocumentInFrame()
2195 {
2196     // FIXME: Are there cases where the document can be dislodged from the frame during the event handling below?
2197     // If so, then m_frame could become 0, and we need to do something about that.
2198
2199     m_frame->script().updateDocument();
2200
2201     if (!hasLivingRenderTree())
2202         createRenderTree();
2203
2204     updateViewportArguments();
2205
2206     // FIXME: Doing this only for the main frame is insufficient.
2207     // Changing a subframe can also change the wheel event handler count.
2208     // FIXME: Doing this only when a document goes into the frame is insufficient.
2209     // Removing a document can also change the wheel event handler count.
2210     // FIXME: Doing this every time is a waste. If the current document and its
2211     // subframes' documents have no wheel event handlers, then the count did not change,
2212     // unless the documents they are replacing had wheel event handlers.
2213     if (page() && m_frame->isMainFrame())
2214         wheelEventHandlersChanged();
2215
2216 #if ENABLE(TOUCH_EVENTS)
2217     // FIXME: Doing this only for the main frame is insufficient.
2218     // A subframe could have touch event handlers.
2219     if (hasTouchEventHandlers() && page() && m_frame->isMainFrame())
2220         page()->chrome().client().needTouchEvents(true);
2221 #endif
2222
2223 #if PLATFORM(IOS)
2224     // Ensure that document scheduled task state matches frame timer state. It can be out of sync
2225     // if timers state changed while the document was not in the frame (possibly in page cache,
2226     // or simply newly created).
2227     // FIXME: How does this interact with cross-platform code below?
2228     if (m_frame->timersPaused())
2229         suspendScheduledTasks(ActiveDOMObject::DocumentWillBePaused);
2230     else
2231         resumeScheduledTasks(ActiveDOMObject::DocumentWillBePaused);
2232 #endif
2233
2234     if (m_frame->activeDOMObjectsAndAnimationsSuspended()) {
2235         suspendScriptedAnimationControllerCallbacks();
2236         m_frame->animation().suspendAnimationsForDocument(this);
2237         suspendActiveDOMObjects(ActiveDOMObject::PageWillBeSuspended);
2238     }
2239 }
2240
2241 void Document::disconnectFromFrame()
2242 {
2243     m_frame = nullptr;
2244 }
2245
2246 void Document::destroyRenderTree()
2247 {
2248     ASSERT(hasLivingRenderTree());
2249     ASSERT(!m_inPageCache);
2250
2251     TemporaryChange<bool> change(m_renderTreeBeingDestroyed, true);
2252
2253     if (this == &topDocument())
2254         clearAXObjectCache();
2255
2256     documentWillBecomeInactive();
2257
2258     if (FrameView* frameView = view())
2259         frameView->detachCustomScrollbars();
2260
2261 #if ENABLE(FULLSCREEN_API)
2262     if (m_fullScreenRenderer)
2263         setFullScreenRenderer(nullptr);
2264 #endif
2265
2266     m_hoveredElement = nullptr;
2267     m_focusedElement = nullptr;
2268     m_activeElement = nullptr;
2269
2270     if (m_documentElement)
2271         Style::detachRenderTree(*m_documentElement);
2272
2273     clearChildNeedsStyleRecalc();
2274
2275     unscheduleStyleRecalc();
2276
2277     m_renderView = nullptr;
2278     Node::setRenderer(nullptr);
2279
2280 #if ENABLE(IOS_TEXT_AUTOSIZING)
2281     // Do this before the arena is cleared, which is needed to deref the RenderStyle on TextAutoSizingKey.
2282     m_textAutoSizedNodes.clear();
2283 #endif
2284 }
2285
2286 void Document::prepareForDestruction()
2287 {
2288     if (m_hasPreparedForDestruction)
2289         return;
2290
2291 #if ENABLE(IOS_TOUCH_EVENTS)
2292     clearTouchEventListeners();
2293 #endif
2294
2295 #if HAVE(ACCESSIBILITY)
2296     // Sub-frames need to cleanup Nodes in the text marker cache when the Document disappears.
2297     if (this != &topDocument()) {
2298         if (AXObjectCache* cache = existingAXObjectCache())
2299             cache->clearTextMarkerNodesInUse(this);
2300     }
2301 #endif
2302     
2303     disconnectDescendantFrames();
2304     if (m_domWindow && m_frame)
2305         m_domWindow->willDetachDocumentFromFrame();
2306
2307     if (hasLivingRenderTree())
2308         destroyRenderTree();
2309
2310     if (is<PluginDocument>(*this))
2311         downcast<PluginDocument>(*this).detachFromPluginElement();
2312
2313 #if ENABLE(POINTER_LOCK)
2314     if (page())
2315         page()->pointerLockController().documentDetached(this);
2316 #endif
2317
2318     InspectorInstrumentation::documentDetached(*this);
2319
2320     stopActiveDOMObjects();
2321     m_eventQueue.close();
2322 #if ENABLE(FULLSCREEN_API)
2323     m_fullScreenChangeEventTargetQueue.clear();
2324     m_fullScreenErrorEventTargetQueue.clear();
2325 #endif
2326
2327     commonTeardown();
2328
2329 #if ENABLE(TOUCH_EVENTS)
2330     if (m_touchEventTargets && m_touchEventTargets->size() && parentDocument())
2331         parentDocument()->didRemoveEventTargetNode(*this);
2332 #endif
2333
2334     if (m_wheelEventTargets && m_wheelEventTargets->size() && parentDocument())
2335         parentDocument()->didRemoveEventTargetNode(*this);
2336
2337     if (m_mediaQueryMatcher)
2338         m_mediaQueryMatcher->documentDestroyed();
2339
2340 #if ENABLE(WIRELESS_PLAYBACK_TARGET)
2341     if (!m_clientToIDMap.isEmpty() && page()) {
2342         Vector<WebCore::MediaPlaybackTargetClient*> clients;
2343         copyKeysToVector(m_clientToIDMap, clients);
2344         for (auto client : clients)
2345             removePlaybackTargetPickerClient(*client);
2346     }
2347 #endif
2348
2349     disconnectFromFrame();
2350
2351     m_hasPreparedForDestruction = true;
2352 }
2353
2354 void Document::removeAllEventListeners()
2355 {
2356     EventTarget::removeAllEventListeners();
2357
2358     if (m_domWindow)
2359         m_domWindow->removeAllEventListeners();
2360 #if ENABLE(IOS_TOUCH_EVENTS)
2361     clearTouchEventListeners();
2362 #endif
2363     for (Node* node = firstChild(); node; node = NodeTraversal::next(*node))
2364         node->removeAllEventListeners();
2365 }
2366
2367 void Document::platformSuspendOrStopActiveDOMObjects()
2368 {
2369 #if PLATFORM(IOS)
2370 #if ENABLE(DEVICE_ORIENTATION)
2371     if (m_deviceMotionController)
2372         m_deviceMotionController->suspendUpdates();
2373     if (m_deviceOrientationController)
2374         m_deviceOrientationController->suspendUpdates();
2375 #endif
2376
2377     if (WebThreadCountOfObservedContentModifiers() > 0) {
2378         Frame* frame = this->frame();
2379         if (Page* page = frame ? frame->page() : nullptr)
2380             page->chrome().client().clearContentChangeObservers(frame);
2381     }
2382 #endif
2383 }
2384
2385 void Document::suspendActiveDOMObjects(ActiveDOMObject::ReasonForSuspension why)
2386 {
2387     ScriptExecutionContext::suspendActiveDOMObjects(why);
2388     platformSuspendOrStopActiveDOMObjects();
2389 }
2390
2391 void Document::resumeActiveDOMObjects(ActiveDOMObject::ReasonForSuspension why)
2392 {
2393     ScriptExecutionContext::resumeActiveDOMObjects(why);
2394
2395 #if ENABLE(DEVICE_ORIENTATION) && PLATFORM(IOS)
2396     if (m_deviceMotionController)
2397         m_deviceMotionController->resumeUpdates();
2398     if (m_deviceOrientationController)
2399         m_deviceOrientationController->resumeUpdates();
2400 #endif
2401     // FIXME: For iOS, do we need to add content change observers that were removed in Document::suspendActiveDOMObjects()?
2402 }
2403
2404 void Document::stopActiveDOMObjects()
2405 {
2406     ScriptExecutionContext::stopActiveDOMObjects();
2407     platformSuspendOrStopActiveDOMObjects();
2408 }
2409
2410 void Document::clearAXObjectCache()
2411 {
2412     ASSERT(&topDocument() == this);
2413     // Clear the cache member variable before calling delete because attempts
2414     // are made to access it during destruction.
2415     m_axObjectCache = nullptr;
2416 }
2417
2418 AXObjectCache* Document::existingAXObjectCache() const
2419 {
2420     Document& topDocument = this->topDocument();
2421     if (!topDocument.hasLivingRenderTree())
2422         return nullptr;
2423     return topDocument.m_axObjectCache.get();
2424 }
2425
2426 AXObjectCache* Document::axObjectCache() const
2427 {
2428     if (!AXObjectCache::accessibilityEnabled())
2429         return nullptr;
2430     
2431     // The only document that actually has a AXObjectCache is the top-level
2432     // document.  This is because we need to be able to get from any WebCoreAXObject
2433     // to any other WebCoreAXObject on the same page.  Using a single cache allows
2434     // lookups across nested webareas (i.e. multiple documents).
2435     Document& topDocument = this->topDocument();
2436
2437     // If the document has already been detached, do not make a new axObjectCache.
2438     if (!topDocument.hasLivingRenderTree())
2439         return nullptr;
2440
2441     ASSERT(&topDocument == this || !m_axObjectCache);
2442     if (!topDocument.m_axObjectCache)
2443         topDocument.m_axObjectCache = std::make_unique<AXObjectCache>(topDocument);
2444     return topDocument.m_axObjectCache.get();
2445 }
2446
2447 void Document::setVisuallyOrdered()
2448 {
2449     m_visuallyOrdered = true;
2450     if (renderView())
2451         renderView()->style().setRTLOrdering(VisualOrder);
2452 }
2453
2454 Ref<DocumentParser> Document::createParser()
2455 {
2456     // FIXME: this should probably pass the frame instead
2457     return XMLDocumentParser::create(*this, view());
2458 }
2459
2460 ScriptableDocumentParser* Document::scriptableDocumentParser() const
2461 {
2462     return parser() ? parser()->asScriptableDocumentParser() : nullptr;
2463 }
2464
2465 void Document::open(Document* ownerDocument)
2466 {
2467     if (ownerDocument) {
2468         setURL(ownerDocument->url());
2469         setCookieURL(ownerDocument->cookieURL());
2470         setSecurityOriginPolicy(ownerDocument->securityOriginPolicy());
2471     }
2472
2473     if (m_frame) {
2474         if (ScriptableDocumentParser* parser = scriptableDocumentParser()) {
2475             if (parser->isParsing()) {
2476                 // FIXME: HTML5 doesn't tell us to check this, it might not be correct.
2477                 if (parser->isExecutingScript())
2478                     return;
2479
2480                 if (!parser->wasCreatedByScript() && parser->hasInsertionPoint())
2481                     return;
2482             }
2483         }
2484
2485         if (m_frame->loader().state() == FrameStateProvisional)
2486             m_frame->loader().stopAllLoaders();
2487     }
2488
2489     removeAllEventListeners();
2490     implicitOpen();
2491     if (ScriptableDocumentParser* parser = scriptableDocumentParser())
2492         parser->setWasCreatedByScript(true);
2493
2494     if (m_frame)
2495         m_frame->loader().didExplicitOpen();
2496 }
2497
2498 void Document::detachParser()
2499 {
2500     if (!m_parser)
2501         return;
2502     m_parser->detach();
2503     m_parser = nullptr;
2504 }
2505
2506 void Document::cancelParsing()
2507 {
2508     if (!m_parser)
2509         return;
2510
2511     // We have to clear the parser to avoid possibly triggering
2512     // the onload handler when closing as a side effect of a cancel-style
2513     // change, such as opening a new document or closing the window while
2514     // still parsing
2515     detachParser();
2516     explicitClose();
2517 }
2518
2519 void Document::implicitOpen()
2520 {
2521     cancelParsing();
2522
2523     removeChildren();
2524
2525     setCompatibilityMode(DocumentCompatibilityMode::NoQuirksMode);
2526
2527     m_parser = createParser();
2528     setParsing(true);
2529     setReadyState(Loading);
2530 }
2531
2532 HTMLBodyElement* Document::body() const
2533 {
2534     auto* element = documentElement();
2535     if (!element)
2536         return nullptr;
2537     return childrenOfType<HTMLBodyElement>(*element).first();
2538 }
2539
2540 HTMLElement* Document::bodyOrFrameset() const
2541 {
2542     // Return the first body or frameset child of the html element.
2543     auto* element = documentElement();
2544     if (!element)
2545         return nullptr;
2546     for (auto& child : childrenOfType<HTMLElement>(*element)) {
2547         if (is<HTMLBodyElement>(child) || is<HTMLFrameSetElement>(child))
2548             return &child;
2549     }
2550     return nullptr;
2551 }
2552
2553 void Document::setBodyOrFrameset(PassRefPtr<HTMLElement> prpNewBody, ExceptionCode& ec)
2554 {
2555     RefPtr<HTMLElement> newBody = prpNewBody;
2556
2557     // FIXME: This does not support setting a <frameset> Element, only a <body>. This does
2558     // not match the HTML specification:
2559     // https://html.spec.whatwg.org/multipage/dom.html#dom-document-body
2560     if (!newBody || !documentElement() || !newBody->hasTagName(bodyTag)) { 
2561         ec = HIERARCHY_REQUEST_ERR;
2562         return;
2563     }
2564
2565     if (&newBody->document() != this) {
2566         ec = 0;
2567         RefPtr<Node> node = importNode(newBody.get(), true, ec);
2568         if (ec)
2569             return;
2570         
2571         newBody = downcast<HTMLElement>(node.get());
2572     }
2573
2574     if (auto* body = bodyOrFrameset())
2575         documentElement()->replaceChild(newBody.releaseNonNull(), *body, ec);
2576     else
2577         documentElement()->appendChild(newBody.releaseNonNull(), ec);
2578 }
2579
2580 HTMLHeadElement* Document::head()
2581 {
2582     if (auto element = documentElement())
2583         return childrenOfType<HTMLHeadElement>(*element).first();
2584     return nullptr;
2585 }
2586
2587 void Document::close()
2588 {
2589     // FIXME: We should follow the specification more closely:
2590     //        http://www.whatwg.org/specs/web-apps/current-work/#dom-document-close
2591
2592     if (!scriptableDocumentParser() || !scriptableDocumentParser()->wasCreatedByScript() || !scriptableDocumentParser()->isParsing())
2593         return;
2594
2595     explicitClose();
2596 }
2597
2598 void Document::explicitClose()
2599 {
2600     if (RefPtr<DocumentParser> parser = m_parser)
2601         parser->finish();
2602
2603     if (!m_frame) {
2604         // Because we have no frame, we don't know if all loading has completed,
2605         // so we just call implicitClose() immediately. FIXME: This might fire
2606         // the load event prematurely <http://bugs.webkit.org/show_bug.cgi?id=14568>.
2607         implicitClose();
2608         return;
2609     }
2610
2611     m_frame->loader().checkCompleted();
2612 }
2613
2614 void Document::implicitClose()
2615 {
2616     // If we're in the middle of recalcStyle, we need to defer the close until the style information is accurate and all elements are re-attached.
2617     if (m_inStyleRecalc) {
2618         m_closeAfterStyleRecalc = true;
2619         return;
2620     }
2621
2622     bool wasLocationChangePending = frame() && frame()->navigationScheduler().locationChangePending();
2623     bool doload = !parsing() && m_parser && !m_processingLoadEvent && !wasLocationChangePending;
2624     
2625     if (!doload)
2626         return;
2627
2628     // Call to dispatchWindowLoadEvent can blow us from underneath.
2629     Ref<Document> protect(*this);
2630
2631     m_processingLoadEvent = true;
2632
2633     ScriptableDocumentParser* parser = scriptableDocumentParser();
2634     m_wellFormed = parser && parser->wellFormed();
2635
2636     // We have to clear the parser, in case someone document.write()s from the
2637     // onLoad event handler, as in Radar 3206524.
2638     detachParser();
2639
2640     // FIXME: We kick off the icon loader when the Document is done parsing.
2641     // There are earlier opportunities we could start it:
2642     //  -When the <head> finishes parsing
2643     //  -When any new HTMLLinkElement is inserted into the document
2644     // But those add a dynamic component to the favicon that has UI 
2645     // ramifications, and we need to decide what is the Right Thing To Do(tm)
2646     Frame* f = frame();
2647     if (f) {
2648         f->loader().icon().startLoader();
2649         f->animation().startAnimationsIfNotSuspended(this);
2650
2651         // FIXME: We shouldn't be dispatching pending events globally on all Documents here.
2652         // For now, only do this when there is a Frame, otherwise this could cause JS reentrancy
2653         // below SVG font parsing, for example. <https://webkit.org/b/136269>
2654         ImageLoader::dispatchPendingBeforeLoadEvents();
2655         ImageLoader::dispatchPendingLoadEvents();
2656         ImageLoader::dispatchPendingErrorEvents();
2657         HTMLLinkElement::dispatchPendingLoadEvents();
2658         HTMLStyleElement::dispatchPendingLoadEvents();
2659     }
2660
2661     // To align the HTML load event and the SVGLoad event for the outermost <svg> element, fire it from
2662     // here, instead of doing it from SVGElement::finishedParsingChildren (if externalResourcesRequired="false",
2663     // which is the default, for ='true' its fired at a later time, once all external resources finished loading).
2664     if (svgExtensions())
2665         accessSVGExtensions().dispatchSVGLoadEventToOutermostSVGElements();
2666
2667     dispatchWindowLoadEvent();
2668     enqueuePageshowEvent(PageshowEventNotPersisted);
2669     enqueuePopstateEvent(m_pendingStateObject ? m_pendingStateObject.release() : SerializedScriptValue::nullValue());
2670     
2671     if (f)
2672         f->loader().handledOnloadEvents();
2673 #ifdef INSTRUMENT_LAYOUT_SCHEDULING
2674     if (!ownerElement())
2675         printf("onload fired at %lld\n", elapsedTime().count());
2676 #endif
2677
2678     // An event handler may have removed the frame
2679     if (!frame()) {
2680         m_processingLoadEvent = false;
2681         return;
2682     }
2683
2684     // Make sure both the initial layout and reflow happen after the onload
2685     // fires. This will improve onload scores, and other browsers do it.
2686     // If they wanna cheat, we can too. -dwh
2687
2688     if (frame()->navigationScheduler().locationChangePending() && elapsedTime() < settings()->layoutInterval()) {
2689         // Just bail out. Before or during the onload we were shifted to another page.
2690         // The old i-Bench suite does this. When this happens don't bother painting or laying out.        
2691         m_processingLoadEvent = false;
2692         view()->unscheduleRelayout();
2693         return;
2694     }
2695
2696     frame()->loader().checkCallImplicitClose();
2697     
2698     // We used to force a synchronous display and flush here.  This really isn't
2699     // necessary and can in fact be actively harmful if pages are loading at a rate of > 60fps
2700     // (if your platform is syncing flushes and limiting them to 60fps).
2701     m_overMinimumLayoutThreshold = true;
2702     if (!ownerElement() || (ownerElement()->renderer() && !ownerElement()->renderer()->needsLayout())) {
2703         updateStyleIfNeeded();
2704         
2705         // Always do a layout after loading if needed.
2706         if (view() && renderView() && (!renderView()->firstChild() || renderView()->needsLayout()))
2707             view()->layout();
2708     }
2709
2710     m_processingLoadEvent = false;
2711
2712 #if PLATFORM(COCOA) || PLATFORM(WIN) || PLATFORM(GTK) || PLATFORM(EFL)
2713     if (f && hasLivingRenderTree() && AXObjectCache::accessibilityEnabled()) {
2714         // The AX cache may have been cleared at this point, but we need to make sure it contains an
2715         // AX object to send the notification to. getOrCreate will make sure that an valid AX object
2716         // exists in the cache (we ignore the return value because we don't need it here). This is 
2717         // only safe to call when a layout is not in progress, so it can not be used in postNotification.
2718         //
2719         // This notification is now called AXNewDocumentLoadComplete because there are other handlers that will
2720         // catch new AND page history loads, and that uses AXLoadComplete
2721         
2722         axObjectCache()->getOrCreate(renderView());
2723         if (this == &topDocument())
2724             axObjectCache()->postNotification(renderView(), AXObjectCache::AXNewDocumentLoadComplete);
2725         else {
2726             // AXLoadComplete can only be posted on the top document, so if it's a document
2727             // in an iframe that just finished loading, post AXLayoutComplete instead.
2728             axObjectCache()->postNotification(renderView(), AXObjectCache::AXLayoutComplete);
2729         }
2730     }
2731 #endif
2732
2733     if (svgExtensions())
2734         accessSVGExtensions().startAnimations();
2735 }
2736
2737 void Document::setParsing(bool b)
2738 {
2739     m_bParsing = b;
2740
2741     if (m_bParsing && !m_sharedObjectPool)
2742         m_sharedObjectPool = std::make_unique<DocumentSharedObjectPool>();
2743
2744     if (!m_bParsing && view() && !view()->needsLayout())
2745         view()->fireLayoutRelatedMilestonesIfNeeded();
2746
2747 #ifdef INSTRUMENT_LAYOUT_SCHEDULING
2748     if (!ownerElement() && !m_bParsing)
2749         printf("Parsing finished at %lld\n", elapsedTime().count());
2750 #endif
2751 }
2752
2753 bool Document::shouldScheduleLayout()
2754 {
2755     // This function will only be called when FrameView thinks a layout is needed.
2756     // This enforces a couple extra rules.
2757     //
2758     //    (a) Only schedule a layout once the stylesheets are loaded.
2759     //    (b) Only schedule layout once we have a body element.
2760
2761     return (haveStylesheetsLoaded() && bodyOrFrameset())
2762         || (documentElement() && !is<HTMLHtmlElement>(*documentElement()));
2763 }
2764     
2765 bool Document::isLayoutTimerActive()
2766 {
2767     return view() && view()->layoutPending() && !minimumLayoutDelay().count();
2768 }
2769
2770 std::chrono::milliseconds Document::minimumLayoutDelay()
2771 {
2772     if (m_overMinimumLayoutThreshold)
2773         return std::chrono::milliseconds(0);
2774     
2775     std::chrono::milliseconds elapsed = elapsedTime();
2776     m_overMinimumLayoutThreshold = elapsed > settings()->layoutInterval();
2777
2778     // We'll want to schedule the timer to fire at the minimum layout threshold.
2779     return std::max(std::chrono::milliseconds(0), settings()->layoutInterval() - elapsed);
2780 }
2781
2782 std::chrono::milliseconds Document::elapsedTime() const
2783 {
2784     auto elapsedTime = std::chrono::steady_clock::now() - m_startTime;
2785
2786     return std::chrono::duration_cast<std::chrono::milliseconds>(elapsedTime);
2787 }
2788
2789 void Document::write(const SegmentedString& text, Document* ownerDocument)
2790 {
2791     NestingLevelIncrementer nestingLevelIncrementer(m_writeRecursionDepth);
2792
2793     m_writeRecursionIsTooDeep = (m_writeRecursionDepth > 1) && m_writeRecursionIsTooDeep;
2794     m_writeRecursionIsTooDeep = (m_writeRecursionDepth > cMaxWriteRecursionDepth) || m_writeRecursionIsTooDeep;
2795
2796     if (m_writeRecursionIsTooDeep)
2797        return;
2798
2799 #ifdef INSTRUMENT_LAYOUT_SCHEDULING
2800     if (!ownerElement())
2801         printf("Beginning a document.write at %lld\n", elapsedTime().count());
2802 #endif
2803
2804     bool hasInsertionPoint = m_parser && m_parser->hasInsertionPoint();
2805     if (!hasInsertionPoint && m_ignoreDestructiveWriteCount)
2806         return;
2807
2808     if (!hasInsertionPoint)
2809         open(ownerDocument);
2810
2811     ASSERT(m_parser);
2812     m_parser->insert(text);
2813
2814 #ifdef INSTRUMENT_LAYOUT_SCHEDULING
2815     if (!ownerElement())
2816         printf("Ending a document.write at %lld\n", elapsedTime().count());
2817 #endif    
2818 }
2819
2820 void Document::write(const String& text, Document* ownerDocument)
2821 {
2822     write(SegmentedString(text), ownerDocument);
2823 }
2824
2825 void Document::writeln(const String& text, Document* ownerDocument)
2826 {
2827     write(text, ownerDocument);
2828     write("\n", ownerDocument);
2829 }
2830
2831 double Document::minimumTimerInterval() const
2832 {
2833     Page* page = this->page();
2834     if (!page)
2835         return ScriptExecutionContext::minimumTimerInterval();
2836     return page->settings().minimumDOMTimerInterval();
2837 }
2838
2839 void Document::setTimerThrottlingEnabled(bool shouldThrottle)
2840 {
2841     if (m_isTimerThrottlingEnabled == shouldThrottle)
2842         return;
2843
2844     m_isTimerThrottlingEnabled = shouldThrottle;
2845     didChangeTimerAlignmentInterval();
2846 }
2847
2848 double Document::timerAlignmentInterval(bool hasReachedMaxNestingLevel) const
2849 {
2850     // Apply Document-level DOMTimer throttling only if timers have reached their maximum nesting level as the Page may still be visible.
2851     if (m_isTimerThrottlingEnabled && hasReachedMaxNestingLevel)
2852         return DOMTimer::hiddenPageAlignmentInterval();
2853
2854     Page* page = this->page();
2855     if (!page)
2856         return ScriptExecutionContext::timerAlignmentInterval(hasReachedMaxNestingLevel);
2857     return page->settings().domTimerAlignmentInterval();
2858 }
2859
2860 EventTarget* Document::errorEventTarget()
2861 {
2862     return m_domWindow.get();
2863 }
2864
2865 void Document::logExceptionToConsole(const String& errorMessage, const String& sourceURL, int lineNumber, int columnNumber, RefPtr<Inspector::ScriptCallStack>&& callStack)
2866 {
2867     addMessage(MessageSource::JS, MessageLevel::Error, errorMessage, sourceURL, lineNumber, columnNumber, WTF::move(callStack));
2868 }
2869
2870 void Document::setURL(const URL& url)
2871 {
2872     const URL& newURL = url.isEmpty() ? blankURL() : url;
2873     if (newURL == m_url)
2874         return;
2875
2876     m_url = newURL;
2877     m_documentURI = m_url.string();
2878     updateBaseURL();
2879 }
2880
2881 void Document::updateBaseURL()
2882 {
2883     URL oldBaseURL = m_baseURL;
2884     // DOM 3 Core: When the Document supports the feature "HTML" [DOM Level 2 HTML], the base URI is computed using
2885     // first the value of the href attribute of the HTML BASE element if any, and the value of the documentURI attribute
2886     // from the Document interface otherwise.
2887     if (!m_baseElementURL.isEmpty())
2888         m_baseURL = m_baseElementURL;
2889     else if (!m_baseURLOverride.isEmpty())
2890         m_baseURL = m_baseURLOverride;
2891     else {
2892         // The documentURI attribute is read-only from JavaScript, but writable from Objective C, so we need to retain
2893         // this fallback behavior. We use a null base URL, since the documentURI attribute is an arbitrary string
2894         // and DOM 3 Core does not specify how it should be resolved.
2895         m_baseURL = URL(ParsedURLString, documentURI());
2896     }
2897
2898     clearSelectorQueryCache();
2899
2900     if (!m_baseURL.isValid())
2901         m_baseURL = URL();
2902
2903     if (m_elementSheet) {
2904         // Element sheet is silly. It never contains anything.
2905         ASSERT(!m_elementSheet->contents().ruleCount());
2906         bool usesRemUnits = m_elementSheet->contents().usesRemUnits();
2907         bool usesStyleBasedEditability = m_elementSheet->contents().usesStyleBasedEditability();
2908         m_elementSheet = CSSStyleSheet::createInline(*this, m_baseURL);
2909         // FIXME: So we are not really the parser. The right fix is to eliminate the element sheet completely.
2910         if (usesRemUnits)
2911             m_elementSheet->contents().parserSetUsesRemUnits();
2912         if (usesStyleBasedEditability)
2913             m_elementSheet->contents().parserSetUsesStyleBasedEditability();
2914     }
2915
2916     if (!equalIgnoringFragmentIdentifier(oldBaseURL, m_baseURL)) {
2917         // Base URL change changes any relative visited links.
2918         // FIXME: There are other URLs in the tree that would need to be re-evaluated on dynamic base URL change. Style should be invalidated too.
2919         for (auto& anchor : descendantsOfType<HTMLAnchorElement>(*this))
2920             anchor.invalidateCachedVisitedLinkHash();
2921     }
2922 }
2923
2924 void Document::setBaseURLOverride(const URL& url)
2925 {
2926     m_baseURLOverride = url;
2927     updateBaseURL();
2928 }
2929
2930 void Document::processBaseElement()
2931 {
2932     // Find the first href attribute in a base element and the first target attribute in a base element.
2933     const AtomicString* href = nullptr;
2934     const AtomicString* target = nullptr;
2935     auto baseDescendants = descendantsOfType<HTMLBaseElement>(*this);
2936     for (auto& base : baseDescendants) {
2937         if (!href) {
2938             const AtomicString& value = base.fastGetAttribute(hrefAttr);
2939             if (!value.isNull()) {
2940                 href = &value;
2941                 if (target)
2942                     break;
2943             }
2944         }
2945         if (!target) {
2946             const AtomicString& value = base.fastGetAttribute(targetAttr);
2947             if (!value.isNull()) {
2948                 target = &value;
2949                 if (href)
2950                     break;
2951             }
2952         }
2953     }
2954
2955     // FIXME: Since this doesn't share code with completeURL it may not handle encodings correctly.
2956     URL baseElementURL;
2957     if (href) {
2958         String strippedHref = stripLeadingAndTrailingHTMLSpaces(*href);
2959         if (!strippedHref.isEmpty())
2960             baseElementURL = URL(url(), strippedHref);
2961     }
2962     if (m_baseElementURL != baseElementURL && contentSecurityPolicy()->allowBaseURI(baseElementURL)) {
2963         m_baseElementURL = baseElementURL;
2964         updateBaseURL();
2965     }
2966
2967     m_baseTarget = target ? *target : nullAtom;
2968 }
2969
2970 String Document::userAgent(const URL& url) const
2971 {
2972     return frame() ? frame()->loader().userAgent(url) : String();
2973 }
2974
2975 void Document::disableEval(const String& errorMessage)
2976 {
2977     if (!frame())
2978         return;
2979
2980     frame()->script().disableEval(errorMessage);
2981 }
2982
2983 bool Document::canNavigate(Frame* targetFrame)
2984 {
2985     if (!m_frame)
2986         return false;
2987
2988     // FIXME: We shouldn't call this function without a target frame, but
2989     // fast/forms/submit-to-blank-multiple-times.html depends on this function
2990     // returning true when supplied with a 0 targetFrame.
2991     if (!targetFrame)
2992         return true;
2993
2994     // Frame-busting is generally allowed, but blocked for sandboxed frames lacking the 'allow-top-navigation' flag.
2995     if (!isSandboxed(SandboxTopNavigation) && targetFrame == &m_frame->tree().top())
2996         return true;
2997
2998     if (isSandboxed(SandboxNavigation)) {
2999         if (targetFrame->tree().isDescendantOf(m_frame))
3000             return true;
3001
3002         const char* reason = "The frame attempting navigation is sandboxed, and is therefore disallowed from navigating its ancestors.";
3003         if (isSandboxed(SandboxTopNavigation) && targetFrame == &m_frame->tree().top())
3004             reason = "The frame attempting navigation of the top-level window is sandboxed, but the 'allow-top-navigation' flag is not set.";
3005
3006         printNavigationErrorMessage(targetFrame, url(), reason);
3007         return false;
3008     }
3009
3010     // This is the normal case. A document can navigate its decendant frames,
3011     // or, more generally, a document can navigate a frame if the document is
3012     // in the same origin as any of that frame's ancestors (in the frame
3013     // hierarchy).
3014     //
3015     // See http://www.adambarth.com/papers/2008/barth-jackson-mitchell.pdf for
3016     // historical information about this security check.
3017     if (canAccessAncestor(securityOrigin(), targetFrame))
3018         return true;
3019
3020     // Top-level frames are easier to navigate than other frames because they
3021     // display their URLs in the address bar (in most browsers). However, there
3022     // are still some restrictions on navigation to avoid nuisance attacks.
3023     // Specifically, a document can navigate a top-level frame if that frame
3024     // opened the document or if the document is the same-origin with any of
3025     // the top-level frame's opener's ancestors (in the frame hierarchy).
3026     //
3027     // In both of these cases, the document performing the navigation is in
3028     // some way related to the frame being navigate (e.g., by the "opener"
3029     // and/or "parent" relation). Requiring some sort of relation prevents a
3030     // document from navigating arbitrary, unrelated top-level frames.
3031     if (!targetFrame->tree().parent()) {
3032         if (targetFrame == m_frame->loader().opener())
3033             return true;
3034
3035         if (canAccessAncestor(securityOrigin(), targetFrame->loader().opener()))
3036             return true;
3037     }
3038
3039     printNavigationErrorMessage(targetFrame, url(), "The frame attempting navigation is neither same-origin with the target, nor is it the target's parent or opener.");
3040     return false;
3041 }
3042
3043 Frame* Document::findUnsafeParentScrollPropagationBoundary()
3044 {
3045     Frame* currentFrame = m_frame;
3046     if (!currentFrame)
3047         return nullptr;
3048
3049     Frame* ancestorFrame = currentFrame->tree().parent();
3050
3051     while (ancestorFrame) {
3052         if (!ancestorFrame->document()->securityOrigin()->canAccess(securityOrigin()))
3053             return currentFrame;
3054         currentFrame = ancestorFrame;
3055         ancestorFrame = ancestorFrame->tree().parent();
3056     }
3057     return nullptr;
3058 }
3059
3060 void Document::didRemoveAllPendingStylesheet()
3061 {
3062     m_needsNotifyRemoveAllPendingStylesheet = false;
3063
3064     styleResolverChanged(DeferRecalcStyleIfNeeded);
3065
3066     if (m_pendingSheetLayout == DidLayoutWithPendingSheets) {
3067         m_pendingSheetLayout = IgnoreLayoutWithPendingSheets;
3068         if (renderView())
3069             renderView()->repaintViewAndCompositedLayers();
3070     }
3071
3072     if (ScriptableDocumentParser* parser = scriptableDocumentParser())
3073         parser->executeScriptsWaitingForStylesheets();
3074
3075     if (m_gotoAnchorNeededAfterStylesheetsLoad && view())
3076         view()->scrollToFragment(m_url);
3077 }
3078
3079 CSSStyleSheet& Document::elementSheet()
3080 {
3081     if (!m_elementSheet)
3082         m_elementSheet = CSSStyleSheet::createInline(*this, m_baseURL);
3083     return *m_elementSheet;
3084 }
3085
3086 bool Document::usesStyleBasedEditability() const
3087 {
3088     if (m_elementSheet && m_elementSheet->contents().usesStyleBasedEditability())
3089         return true;
3090
3091     ASSERT(!m_renderView || !m_renderView->frameView().isPainting());
3092     ASSERT(!m_inStyleRecalc);
3093
3094     auto& collection = document().styleSheetCollection();
3095     collection.flushPendingUpdates();
3096     return collection.usesStyleBasedEditability();
3097 }
3098
3099 void Document::processHttpEquiv(const String& equiv, const String& content)
3100 {
3101     ASSERT(!equiv.isNull() && !content.isNull());
3102
3103     HttpEquivPolicy policy = httpEquivPolicy();
3104     if (policy != HttpEquivPolicy::Enabled) {
3105         String reason;
3106         switch (policy) {
3107         case HttpEquivPolicy::Enabled:
3108             ASSERT_NOT_REACHED();
3109             break;
3110         case HttpEquivPolicy::DisabledBySettings:
3111             reason = "by the embedder.";
3112             break;
3113         case HttpEquivPolicy::DisabledByContentDispositionAttachmentSandbox:
3114             reason = "for documents with Content-Disposition: attachment.";
3115             break;
3116         }
3117         String message = "http-equiv '" + equiv + "' is disabled " + reason;
3118         addConsoleMessage(MessageSource::Security, MessageLevel::Error, message);
3119         return;
3120     }
3121
3122     Frame* frame = this->frame();
3123
3124     HTTPHeaderName headerName;
3125     if (!findHTTPHeaderName(equiv, headerName))
3126         return;
3127
3128     switch (headerName) {
3129     case HTTPHeaderName::DefaultStyle:
3130         // The preferred style set has been overridden as per section
3131         // 14.3.2 of the HTML4.0 specification.  We need to update the
3132         // sheet used variable and then update our style selector.
3133         // For more info, see the test at:
3134         // http://www.hixie.ch/tests/evil/css/import/main/preferred.html
3135         // -dwh
3136         m_styleSheetCollection.setSelectedStylesheetSetName(content);
3137         m_styleSheetCollection.setPreferredStylesheetSetName(content);
3138         styleResolverChanged(DeferRecalcStyle);
3139         break;
3140
3141     case HTTPHeaderName::Refresh: {
3142         double delay;
3143         String urlString;
3144         if (frame && parseHTTPRefresh(content, true, delay, urlString)) {
3145             URL completedURL;
3146             if (urlString.isEmpty())
3147                 completedURL = m_url;
3148             else
3149                 completedURL = completeURL(urlString);
3150             if (!protocolIsJavaScript(completedURL))
3151                 frame->navigationScheduler().scheduleRedirect(this, delay, completedURL);
3152             else {
3153                 String message = "Refused to refresh " + m_url.stringCenterEllipsizedToLength() + " to a javascript: URL";
3154                 addConsoleMessage(MessageSource::Security, MessageLevel::Error, message);
3155             }
3156         }
3157
3158         break;
3159     }
3160
3161     case HTTPHeaderName::SetCookie:
3162         // FIXME: make setCookie work on XML documents too; e.g. in case of <html:meta .....>
3163         if (is<HTMLDocument>(*this)) {
3164             // Exception (for sandboxed documents) ignored.
3165             downcast<HTMLDocument>(*this).setCookie(content, IGNORE_EXCEPTION);
3166         }
3167         break;
3168
3169     case HTTPHeaderName::ContentLanguage:
3170         setContentLanguage(content);
3171         break;
3172
3173     case HTTPHeaderName::XDNSPrefetchControl:
3174         parseDNSPrefetchControlHeader(content);
3175         break;
3176
3177     case HTTPHeaderName::XFrameOptions:
3178         if (frame) {
3179             FrameLoader& frameLoader = frame->loader();
3180             unsigned long requestIdentifier = 0;
3181             if (frameLoader.activeDocumentLoader() && frameLoader.activeDocumentLoader()->mainResourceLoader())
3182                 requestIdentifier = frameLoader.activeDocumentLoader()->mainResourceLoader()->identifier();
3183             if (frameLoader.shouldInterruptLoadForXFrameOptions(content, url(), requestIdentifier)) {
3184                 String message = "Refused to display '" + url().stringCenterEllipsizedToLength() + "' in a frame because it set 'X-Frame-Options' to '" + content + "'.";
3185                 frameLoader.stopAllLoaders();
3186                 // Stopping the loader isn't enough, as we're already parsing the document; to honor the header's
3187                 // intent, we must navigate away from the possibly partially-rendered document to a location that
3188                 // doesn't inherit the parent's SecurityOrigin.
3189                 frame->navigationScheduler().scheduleLocationChange(this, securityOrigin(), SecurityOrigin::urlWithUniqueSecurityOrigin(), String());
3190                 addConsoleMessage(MessageSource::Security, MessageLevel::Error, message, requestIdentifier);
3191             }
3192         }
3193         break;
3194
3195     case HTTPHeaderName::ContentSecurityPolicy:
3196         contentSecurityPolicy()->didReceiveHeader(content, ContentSecurityPolicy::Enforce);
3197         break;
3198
3199     case HTTPHeaderName::ContentSecurityPolicyReportOnly:
3200         contentSecurityPolicy()->didReceiveHeader(content, ContentSecurityPolicy::Report);
3201         break;
3202
3203     case HTTPHeaderName::XWebKitCSP:
3204         contentSecurityPolicy()->didReceiveHeader(content, ContentSecurityPolicy::PrefixedEnforce);
3205         break;
3206
3207     case HTTPHeaderName::XWebKitCSPReportOnly:
3208         contentSecurityPolicy()->didReceiveHeader(content, ContentSecurityPolicy::PrefixedReport);
3209         break;
3210
3211     default:
3212         break;
3213     }
3214 }
3215
3216 // Though isspace() considers \t and \v to be whitespace, Win IE doesn't.
3217 static bool isSeparator(UChar c)
3218 {
3219     return c == ' ' || c == '\t' || c == '\n' || c == '\r' || c == '=' || c == ',' || c == '\0';
3220 }
3221
3222 void Document::processArguments(const String& features, void* data, ArgumentsCallback callback)
3223 {
3224     // Tread lightly in this code -- it was specifically designed to mimic Win IE's parsing behavior.
3225     unsigned keyBegin, keyEnd;
3226     unsigned valueBegin, valueEnd;
3227
3228     String buffer = features.lower();
3229     unsigned length = buffer.length();
3230     for (unsigned i = 0; i < length; ) {
3231         // skip to first non-separator, but don't skip past the end of the string
3232         while (isSeparator(buffer[i])) {
3233             if (i >= length)
3234                 break;
3235             i++;
3236         }
3237         keyBegin = i;
3238
3239         // skip to first separator
3240         while (!isSeparator(buffer[i]))
3241             i++;
3242         keyEnd = i;
3243
3244         // skip to first '=', but don't skip past a ',' or the end of the string
3245         while (buffer[i] != '=') {
3246             if (buffer[i] == ',' || i >= length)
3247                 break;
3248             i++;
3249         }
3250
3251         // skip to first non-separator, but don't skip past a ',' or the end of the string
3252         while (isSeparator(buffer[i])) {
3253             if (buffer[i] == ',' || i >= length)
3254                 break;
3255             i++;
3256         }
3257         valueBegin = i;
3258
3259         // skip to first separator
3260         while (!isSeparator(buffer[i]))
3261             i++;
3262         valueEnd = i;
3263
3264         ASSERT_WITH_SECURITY_IMPLICATION(i <= length);
3265
3266         String keyString = buffer.substring(keyBegin, keyEnd - keyBegin);
3267         String valueString = buffer.substring(valueBegin, valueEnd - valueBegin);
3268         callback(keyString, valueString, this, data);
3269     }
3270 }
3271
3272 void Document::processViewport(const String& features, ViewportArguments::Type origin)
3273 {
3274     ASSERT(!features.isNull());
3275
3276     if (origin < m_viewportArguments.type)
3277         return;
3278
3279     m_viewportArguments = ViewportArguments(origin);
3280     processArguments(features, (void*)&m_viewportArguments, &setViewportFeature);
3281
3282     updateViewportArguments();
3283 }
3284
3285 void Document::updateViewportArguments()
3286 {
3287     if (page() && frame()->isMainFrame()) {
3288 #ifndef NDEBUG
3289         m_didDispatchViewportPropertiesChanged = true;
3290 #endif
3291         page()->chrome().dispatchViewportPropertiesDidChange(m_viewportArguments);
3292 #if PLATFORM(IOS)
3293         page()->chrome().didReceiveDocType(frame());
3294 #endif
3295     }
3296 }
3297
3298 #if PLATFORM(IOS)
3299 // FIXME: Find a better place for this functionality.
3300 void setParserFeature(const String& key, const String& value, Document* document, void*)
3301 {
3302     if (key == "telephone" && equalIgnoringCase(value, "no"))
3303         document->setIsTelephoneNumberParsingAllowed(false);
3304 }
3305
3306 void Document::processFormatDetection(const String& features)
3307 {
3308     ASSERT(!features.isNull());
3309     processArguments(features, nullptr, &setParserFeature);
3310 }
3311
3312 void Document::processWebAppOrientations()
3313 {
3314     if (Page* page = this->page())
3315         page->chrome().client().webAppOrientationsUpdated();
3316 }
3317 #endif
3318
3319 void Document::processReferrerPolicy(const String& policy)
3320 {
3321     ASSERT(!policy.isNull());
3322
3323     // Note that we're supporting both the standard and legacy keywords for referrer
3324     // policies, as defined by http://www.w3.org/TR/referrer-policy/#referrer-policy-delivery-meta
3325     if (equalIgnoringCase(policy, "no-referrer") || equalIgnoringCase(policy, "never"))
3326         setReferrerPolicy(ReferrerPolicyNever);
3327     else if (equalIgnoringCase(policy, "unsafe-url") || equalIgnoringCase(policy, "always"))
3328         setReferrerPolicy(ReferrerPolicyAlways);
3329     else if (equalIgnoringCase(policy, "origin"))
3330         setReferrerPolicy(ReferrerPolicyOrigin);
3331     else if (equalIgnoringCase(policy, "no-referrer-when-downgrade") || equalIgnoringCase(policy, "default"))
3332         setReferrerPolicy(ReferrerPolicyDefault);
3333     else {
3334         addConsoleMessage(MessageSource::Rendering, MessageLevel::Error, "Failed to set referrer policy: The value '" + policy + "' is not one of 'no-referrer', 'origin', 'no-referrer-when-downgrade', or 'unsafe-url'. Defaulting to 'no-referrer'.");
3335         setReferrerPolicy(ReferrerPolicyNever);
3336     }
3337 }
3338
3339 MouseEventWithHitTestResults Document::prepareMouseEvent(const HitTestRequest& request, const LayoutPoint& documentPoint, const PlatformMouseEvent& event)
3340 {
3341     if (!hasLivingRenderTree())
3342         return MouseEventWithHitTestResults(event, HitTestResult(LayoutPoint()));
3343
3344     HitTestResult result(documentPoint);
3345     renderView()->hitTest(request, result);
3346
3347     if (!request.readOnly())
3348         updateHoverActiveState(request, result.innerElement());
3349
3350     return MouseEventWithHitTestResults(event, result);
3351 }
3352
3353 // DOM Section 1.1.1
3354 bool Document::childTypeAllowed(NodeType type) const
3355 {
3356     switch (type) {
3357     case ATTRIBUTE_NODE:
3358     case CDATA_SECTION_NODE:
3359     case DOCUMENT_FRAGMENT_NODE:
3360     case DOCUMENT_NODE:
3361     case ENTITY_NODE:
3362     case ENTITY_REFERENCE_NODE:
3363     case TEXT_NODE:
3364     case XPATH_NAMESPACE_NODE:
3365         return false;
3366     case COMMENT_NODE:
3367     case PROCESSING_INSTRUCTION_NODE:
3368         return true;
3369     case DOCUMENT_TYPE_NODE:
3370     case ELEMENT_NODE:
3371         // Documents may contain no more than one of each of these.
3372         // (One Element and one DocumentType.)
3373         for (Node* c = firstChild(); c; c = c->nextSibling())
3374             if (c->nodeType() == type)
3375                 return false;
3376         return true;
3377     }
3378     return false;
3379 }
3380
3381 bool Document::canAcceptChild(const Node& newChild, const Node* refChild, AcceptChildOperation operation) const
3382 {
3383     if (operation == AcceptChildOperation::Replace && refChild->nodeType() == newChild.nodeType())
3384         return true;
3385
3386     switch (newChild.nodeType()) {
3387     case ATTRIBUTE_NODE:
3388     case CDATA_SECTION_NODE:
3389     case DOCUMENT_NODE:
3390     case ENTITY_NODE:
3391     case ENTITY_REFERENCE_NODE:
3392     case TEXT_NODE:
3393     case XPATH_NAMESPACE_NODE:
3394         return false;
3395     case COMMENT_NODE:
3396     case PROCESSING_INSTRUCTION_NODE:
3397         return true;
3398     case DOCUMENT_FRAGMENT_NODE: {
3399         bool hasSeenElementChild = false;
3400         for (auto* node = downcast<DocumentFragment>(newChild).firstChild(); node; node = node->nextSibling()) {
3401             if (is<Element>(*node)) {
3402                 if (hasSeenElementChild)
3403                     return false;
3404                 hasSeenElementChild = true;
3405             }
3406             if (!canAcceptChild(*node, refChild, operation))
3407                 return false;
3408         }
3409         break;
3410     }
3411     case DOCUMENT_TYPE_NODE: {
3412         auto* existingDocType = childrenOfType<DocumentType>(*this).first();
3413         if (operation == AcceptChildOperation::Replace) {
3414             //  parent has a doctype child that is not child, or an element is preceding child.
3415             if (existingDocType && existingDocType != refChild)
3416                 return false;
3417             if (refChild->previousElementSibling())
3418                 return false;
3419         } else {
3420             ASSERT(operation == AcceptChildOperation::InsertOrAdd);
3421             if (existingDocType)
3422                 return false;
3423             if ((refChild && refChild->previousElementSibling()) || (!refChild && firstElementChild()))
3424                 return false;
3425         }
3426         break;
3427     }
3428     case ELEMENT_NODE: {
3429         auto* existingElementChild = firstElementChild();
3430         if (operation == AcceptChildOperation::Replace) {
3431             if (existingElementChild && existingElementChild != refChild)
3432                 return false;
3433             for (auto* child = refChild->nextSibling(); child; child = child->nextSibling()) {
3434                 if (is<DocumentType>(*child))
3435                     return false;
3436             }
3437         } else {
3438             ASSERT(operation == AcceptChildOperation::InsertOrAdd);
3439             if (existingElementChild)
3440                 return false;
3441             for (auto* child = refChild; child; child = child->nextSibling()) {
3442                 if (is<DocumentType>(*child))
3443                     return false;
3444             }
3445         }
3446         break;
3447     }
3448     }
3449     return true;
3450 }
3451
3452 Ref<Node> Document::cloneNodeInternal(Document&, CloningOperation type)
3453 {
3454     Ref<Document> clone = cloneDocumentWithoutChildren();
3455     clone->cloneDataFromDocument(*this);
3456     switch (type) {
3457     case CloningOperation::OnlySelf:
3458     case CloningOperation::SelfWithTemplateContent:
3459         break;
3460     case CloningOperation::Everything:
3461         cloneChildNodes(clone);
3462         break;
3463     }
3464     return WTF::move(clone);
3465 }
3466
3467 Ref<Document> Document::cloneDocumentWithoutChildren() const
3468 {
3469     return isXHTMLDocument() ? createXHTML(nullptr, url()) : create(nullptr, url());
3470 }
3471
3472 void Document::cloneDataFromDocument(const Document& other)
3473 {
3474     ASSERT(m_url == other.url());
3475     m_baseURL = other.baseURL();
3476     m_baseURLOverride = other.baseURLOverride();
3477     m_documentURI = other.documentURI();
3478
3479     setCompatibilityMode(other.m_compatibilityMode);
3480     setSecurityOriginPolicy(other.securityOriginPolicy());
3481     overrideMIMEType(other.contentType());
3482     setDecoder(other.decoder());
3483 }
3484
3485 StyleSheetList& Document::styleSheets()
3486 {
3487     if (!m_styleSheetList)
3488         m_styleSheetList = StyleSheetList::create(this);
3489     return *m_styleSheetList;
3490 }
3491
3492 String Document::preferredStylesheetSet() const
3493 {
3494     return m_styleSheetCollection.preferredStylesheetSetName();
3495 }
3496
3497 String Document::selectedStylesheetSet() const
3498 {
3499     return m_styleSheetCollection.selectedStylesheetSetName();
3500 }
3501
3502 void Document::setSelectedStylesheetSet(const String& aString)
3503 {
3504     m_styleSheetCollection.setSelectedStylesheetSetName(aString);
3505     styleResolverChanged(DeferRecalcStyle);
3506 }
3507
3508 void Document::evaluateMediaQueryList()
3509 {
3510     if (m_mediaQueryMatcher)
3511         m_mediaQueryMatcher->styleResolverChanged();
3512 }
3513
3514 void Document::optimizedStyleSheetUpdateTimerFired()
3515 {
3516     styleResolverChanged(RecalcStyleIfNeeded);
3517 }
3518
3519 void Document::scheduleOptimizedStyleSheetUpdate()
3520 {
3521     if (m_optimizedStyleSheetUpdateTimer.isActive())
3522         return;
3523     m_styleSheetCollection.setPendingUpdateType(DocumentStyleSheetCollection::OptimizedUpdate);
3524     m_optimizedStyleSheetUpdateTimer.startOneShot(0);
3525 }
3526
3527 void Document::updateViewportUnitsOnResize()
3528 {
3529     if (!hasStyleWithViewportUnits())
3530         return;
3531
3532     ensureStyleResolver().clearCachedPropertiesAffectedByViewportUnits();
3533
3534     // FIXME: Ideally, we should save the list of elements that have viewport units and only iterate over those.
3535     for (Element* element = ElementTraversal::firstWithin(rootNode()); element; element = ElementTraversal::nextIncludingPseudo(*element)) {
3536         auto* renderer = element->renderer();
3537         if (renderer && renderer->style().hasViewportUnits())
3538             element->setNeedsStyleRecalc(InlineStyleChange);
3539     }
3540 }
3541
3542 void Document::addAudioProducer(MediaProducer* audioProducer)
3543 {
3544     m_audioProducers.add(audioProducer);
3545     updateIsPlayingMedia();
3546 }
3547
3548 void Document::removeAudioProducer(MediaProducer* audioProducer)
3549 {
3550     m_audioProducers.remove(audioProducer);
3551     updateIsPlayingMedia();
3552 }
3553
3554 void Document::updateIsPlayingMedia(uint64_t sourceElementID)
3555 {
3556     MediaProducer::MediaStateFlags state = MediaProducer::IsNotPlaying;
3557     for (auto audioProducer : m_audioProducers)
3558         state |= audioProducer->mediaState();
3559
3560 #if ENABLE(MEDIA_SESSION)
3561     if (HTMLMediaElement* sourceElement = HTMLMediaElement::elementWithID(sourceElementID)) {
3562         if (sourceElement->isPlaying())
3563             state |= MediaProducer::IsSourceElementPlaying;
3564
3565         if (MediaSession* session = sourceElement->session()) {
3566             bool isNull;
3567             if (MediaRemoteControls* controls = session->controls(isNull)) {
3568                 if (controls->previousTrackEnabled())
3569                     state |= MediaProducer::IsPreviousTrackControlEnabled;
3570                 if (controls->nextTrackEnabled())
3571                     state |= MediaProducer::IsNextTrackControlEnabled;
3572             }
3573         }
3574     }
3575 #endif
3576
3577     if (state == m_mediaState)
3578         return;
3579
3580     m_mediaState = state;
3581
3582     if (page())
3583         page()->updateIsPlayingMedia(sourceElementID);
3584 }
3585
3586 void Document::pageMutedStateDidChange()
3587 {
3588     for (auto audioProducer : m_audioProducers)
3589         audioProducer->pageMutedStateDidChange();
3590 }
3591
3592 void Document::styleResolverChanged(StyleResolverUpdateFlag updateFlag)
3593 {
3594     if (m_optimizedStyleSheetUpdateTimer.isActive())
3595         m_optimizedStyleSheetUpdateTimer.stop();
3596
3597     // Don't bother updating, since we haven't loaded all our style info yet
3598     // and haven't calculated the style selector for the first time.
3599     if (!hasLivingRenderTree() || (!m_didCalculateStyleResolver && !haveStylesheetsLoaded())) {
3600         m_styleResolver = nullptr;
3601         return;
3602     }
3603     m_didCalculateStyleResolver = true;
3604
3605 #ifdef INSTRUMENT_LAYOUT_SCHEDULING
3606     if (!ownerElement())
3607         printf("Beginning update of style selector at time %lld.\n", elapsedTime().count());
3608 #endif
3609
3610     DocumentStyleSheetCollection::UpdateFlag styleSheetUpdate = (updateFlag == RecalcStyleIfNeeded || updateFlag == DeferRecalcStyleIfNeeded)
3611         ? DocumentStyleSheetCollection::OptimizedUpdate
3612         : DocumentStyleSheetCollection::FullUpdate;
3613     bool stylesheetChangeRequiresStyleRecalc = m_styleSheetCollection.updateActiveStyleSheets(styleSheetUpdate);
3614
3615     if (updateFlag == DeferRecalcStyle) {
3616         scheduleForcedStyleRecalc();
3617         return;
3618     }
3619
3620     if (updateFlag == DeferRecalcStyleIfNeeded) {
3621         if (stylesheetChangeRequiresStyleRecalc)
3622             scheduleForcedStyleRecalc();
3623         return;
3624     }
3625
3626     if (!stylesheetChangeRequiresStyleRecalc)
3627         return;
3628
3629     // This recalcStyle initiates a new recalc cycle. We need to bracket it to
3630     // make sure animations get the correct update time
3631     {
3632         AnimationUpdateBlock animationUpdateBlock(m_frame ? &m_frame->animation() : nullptr);
3633         recalcStyle(Style::Force);
3634     }
3635
3636 #ifdef INSTRUMENT_LAYOUT_SCHEDULING
3637     if (!ownerElement())
3638         printf("Finished update of style selector at time %lld\n", elapsedTime().count());
3639 #endif
3640
3641     if (renderView()) {
3642         renderView()->setNeedsLayoutAndPrefWidthsRecalc();
3643         if (view())
3644             view()->scheduleRelayout();
3645     }
3646
3647     evaluateMediaQueryList();
3648 }
3649
3650 void Document::removeFocusedNodeOfSubtree(Node* node, bool amongChildrenOnly)
3651 {
3652     if (!m_focusedElement || this->inPageCache()) // If the document is in the page cache, then we don't need to clear out the focused node.
3653         return;
3654
3655     Element* focusedElement = node->treeScope().focusedElement();
3656     if (!focusedElement)
3657         return;
3658
3659     bool nodeInSubtree = false;
3660     if (amongChildrenOnly)
3661         nodeInSubtree = focusedElement->isDescendantOf(node);
3662     else
3663         nodeInSubtree = (focusedElement == node) || focusedElement->isDescendantOf(node);
3664     
3665     if (nodeInSubtree)
3666         setFocusedElement(nullptr);
3667 }
3668
3669 void Document::hoveredElementDidDetach(Element* element)
3670 {
3671     if (!m_hoveredElement || element != m_hoveredElement)
3672         return;
3673
3674     m_hoveredElement = element->parentElement();
3675     while (m_hoveredElement && !m_hoveredElement->renderer())
3676         m_hoveredElement = m_hoveredElement->parentElement();
3677     if (frame())
3678         frame()->eventHandler().scheduleHoverStateUpdate();
3679 }
3680
3681 void Document::elementInActiveChainDidDetach(Element* element)
3682 {
3683     if (!m_activeElement || element != m_activeElement)
3684         return;
3685
3686     m_activeElement = element->parentElement();
3687     while (m_activeElement && !m_activeElement->renderer())
3688         m_activeElement = m_activeElement->parentElement();
3689 }
3690
3691 #if ENABLE(DASHBOARD_SUPPORT)
3692 const Vector<AnnotatedRegionValue>& Document::annotatedRegions() const
3693 {
3694     return m_annotatedRegions;
3695 }
3696
3697 void Document::setAnnotatedRegions(const Vector<AnnotatedRegionValue>& regions)
3698 {
3699     m_annotatedRegions = regions;
3700     setAnnotatedRegionsDirty(false);
3701 }
3702 #endif
3703
3704 bool Document::setFocusedElement(PassRefPtr<Element> prpNewFocusedElement, FocusDirection direction)
3705 {
3706     RefPtr<Element> newFocusedElement = prpNewFocusedElement;
3707
3708     // Make sure newFocusedElement is actually in this document
3709     if (newFocusedElement && (&newFocusedElement->document() != this))
3710         return true;
3711
3712     if (m_focusedElement == newFocusedElement)
3713         return true;
3714
3715     if (m_inPageCache)
3716         return false;
3717
3718     bool focusChangeBlocked = false;
3719     RefPtr<Element> oldFocusedElement = m_focusedElement.release();
3720
3721     // Remove focus from the existing focus node (if any)
3722     if (oldFocusedElement) {
3723         if (oldFocusedElement->active())
3724             oldFocusedElement->setActive(false);
3725
3726         oldFocusedElement->setFocus(false);
3727
3728         // Dispatch a change event for form control elements that have been edited.
3729         if (is<HTMLFormControlElement>(*oldFocusedElement)) {
3730             HTMLFormControlElement& formControlElement = downcast<HTMLFormControlElement>(*oldFocusedElement);
3731             if (formControlElement.wasChangedSinceLastFormControlChangeEvent())
3732                 formControlElement.dispatchFormControlChangeEvent();
3733         }
3734
3735         // Dispatch the blur event and let the node do any other blur related activities (important for text fields)
3736         oldFocusedElement->dispatchBlurEvent(newFocusedElement.copyRef());
3737
3738         if (m_focusedElement) {
3739             // handler shifted focus
3740             focusChangeBlocked = true;
3741             newFocusedElement = nullptr;
3742         }
3743         
3744         oldFocusedElement->dispatchFocusOutEvent(eventNames().focusoutEvent, newFocusedElement.copyRef()); // DOM level 3 name for the bubbling blur event.
3745         // FIXME: We should remove firing DOMFocusOutEvent event when we are sure no content depends
3746         // on it, probably when <rdar://problem/8503958> is resolved.
3747         oldFocusedElement->dispatchFocusOutEvent(eventNames().DOMFocusOutEvent, newFocusedElement.copyRef()); // DOM level 2 name for compatibility.
3748
3749         if (m_focusedElement) {
3750             // handler shifted focus
3751             focusChangeBlocked = true;
3752             newFocusedElement = nullptr;
3753         }
3754             
3755         if (oldFocusedElement->isRootEditableElement())
3756             frame()->editor().didEndEditing();
3757
3758         if (view()) {
3759             if (Widget* oldWidget = widgetForElement(oldFocusedElement.get()))
3760                 oldWidget->setFocus(false);
3761             else
3762                 view()->setFocus(false);
3763         }
3764     }
3765
3766     if (newFocusedElement && newFocusedElement->isFocusable()) {
3767         if (newFocusedElement->isRootEditableElement() && !acceptsEditingFocus(newFocusedElement.get())) {
3768             // delegate blocks focus change
3769             focusChangeBlocked = true;
3770             goto SetFocusedNodeDone;
3771         }
3772         // Set focus on the new node
3773         m_focusedElement = newFocusedElement;
3774
3775         // Dispatch the focus event and let the node do any other focus related activities (important for text fields)
3776         m_focusedElement->dispatchFocusEvent(oldFocusedElement.copyRef(), direction);
3777
3778         if (m_focusedElement != newFocusedElement) {
3779             // handler shifted focus
3780             focusChangeBlocked = true;
3781             goto SetFocusedNodeDone;
3782         }
3783
3784         m_focusedElement->dispatchFocusInEvent(eventNames().focusinEvent, oldFocusedElement.copyRef()); // DOM level 3 bubbling focus event.
3785
3786         if (m_focusedElement != newFocusedElement) {
3787             // handler shifted focus
3788             focusChangeBlocked = true;
3789             goto SetFocusedNodeDone;
3790         }
3791
3792         // FIXME: We should remove firing DOMFocusInEvent event when we are sure no content depends
3793         // on it, probably when <rdar://problem/8503958> is m.
3794         m_focusedElement->dispatchFocusInEvent(eventNames().DOMFocusInEvent, oldFocusedElement.copyRef()); // DOM level 2 for compatibility.
3795
3796         if (m_focusedElement != newFocusedElement) {
3797             // handler shifted focus
3798             focusChangeBlocked = true;
3799             goto SetFocusedNodeDone;
3800         }
3801
3802         m_focusedElement->setFocus(true);
3803
3804         if (m_focusedElement->isRootEditableElement())
3805             frame()->editor().didBeginEditing();
3806
3807         // eww, I suck. set the qt focus correctly
3808         // ### find a better place in the code for this
3809         if (view()) {
3810             Widget* focusWidget = widgetForElement(m_focusedElement.get());
3811             if (focusWidget) {
3812                 // Make sure a widget has the right size before giving it focus.
3813                 // Otherwise, we are testing edge cases of the Widget code.
3814                 // Specifically, in WebCore this does not work well for text fields.
3815                 updateLayout();
3816                 // Re-get the widget in case updating the layout changed things.
3817                 focusWidget = widgetForElement(m_focusedElement.get());
3818             }
3819             if (focusWidget)
3820                 focusWidget->setFocus(true);
3821             else
3822                 view()->setFocus(true);
3823         }
3824     }
3825
3826     if (!focusChangeBlocked && m_focusedElement) {
3827         // Create the AXObject cache in a focus change because GTK relies on it.
3828         if (AXObjectCache* cache = axObjectCache())
3829             cache->handleFocusedUIElementChanged(oldFocusedElement.get(), newFocusedElement.get());
3830     }
3831
3832     if (!focusChangeBlocked && page())
3833         page()->chrome().focusedElementChanged(m_focusedElement.get());
3834
3835 SetFocusedNodeDone:
3836     updateStyleIfNeeded();
3837     return !focusChangeBlocked;
3838 }
3839
3840 void Document::setCSSTarget(Element* n)
3841 {
3842     if (m_cssTarget)
3843         m_cssTarget->setNeedsStyleRecalc();
3844     m_cssTarget = n;
3845     if (n)
3846         n->setNeedsStyleRecalc();
3847 }
3848
3849 void Document::registerNodeListForInvalidation(LiveNodeList& list)
3850 {
3851     m_nodeListAndCollectionCounts[list.invalidationType()]++;
3852     if (!list.isRootedAtDocument())
3853         return;
3854     ASSERT(!list.isRegisteredForInvalidationAtDocument());
3855     list.setRegisteredForInvalidationAtDocument(true);
3856     m_listsInvalidatedAtDocument.add(&list);
3857 }
3858
3859 void Document::unregisterNodeListForInvalidation(LiveNodeList& list)
3860 {
3861     m_nodeListAndCollectionCounts[list.invalidationType()]--;
3862     if (!list.isRegisteredForInvalidationAtDocument())
3863         return;
3864
3865     list.setRegisteredForInvalidationAtDocument(false);
3866     ASSERT(m_inInvalidateNodeListAndCollectionCaches
3867         ? m_listsInvalidatedAtDocument.isEmpty()
3868         : m_listsInvalidatedAtDocument.contains(&list));
3869     m_listsInvalidatedAtDocument.remove(&list);
3870 }
3871
3872 void Document::registerCollection(HTMLCollection& collection)
3873 {
3874     m_nodeListAndCollectionCounts[collection.invalidationType()]++;
3875     if (collection.isRootedAtDocument())
3876         m_collectionsInvalidatedAtDocument.add(&collection);
3877 }
3878
3879 void Document::unregisterCollection(HTMLCollection& collection)
3880 {
3881     ASSERT(m_nodeListAndCollectionCounts[collection.invalidationType()]);
3882     m_nodeListAndCollectionCounts[collection.invalidationType()]--;
3883     if (!collection.isRootedAtDocument())
3884         return;
3885
3886     m_collectionsInvalidatedAtDocument.remove(&collection);
3887 }
3888
3889 void Document::collectionCachedIdNameMap(const HTMLCollection& collection)
3890 {
3891     ASSERT_UNUSED(collection, collection.hasNamedElementCache());
3892     m_nodeListAndCollectionCounts[InvalidateOnIdNameAttrChange]++;
3893 }
3894
3895 void Document::collectionWillClearIdNameMap(const HTMLCollection& collection)
3896 {
3897     ASSERT_UNUSED(collection, collection.hasNamedElementCache());
3898     ASSERT(m_nodeListAndCollectionCounts[InvalidateOnIdNameAttrChange]);
3899     m_nodeListAndCollectionCounts[InvalidateOnIdNameAttrChange]--;
3900 }
3901
3902 void Document::attachNodeIterator(NodeIterator* ni)
3903 {
3904     m_nodeIterators.add(ni);
3905 }
3906
3907 void Document::detachNodeIterator(NodeIterator* ni)
3908 {
3909     // The node iterator can be detached without having been attached if its root node didn't have a document
3910     // when the iterator was created, but has it now.
3911     m_nodeIterators.remove(ni);
3912 }
3913
3914 void Document::moveNodeIteratorsToNewDocument(Node* node, Document* newDocument)
3915 {
3916     Vector<NodeIterator*> nodeIterators;
3917     copyToVector(m_nodeIterators, nodeIterators);
3918     for (auto* it : nodeIterators) {
3919         if (it->root() == node) {
3920             detachNodeIterator(it);
3921             newDocument->attachNodeIterator(it);
3922         }
3923     }
3924 }
3925
3926 void Document::updateRangesAfterChildrenChanged(ContainerNode& container)
3927 {
3928     for (auto* range : m_ranges)
3929         range->nodeChildrenChanged(container);
3930 }
3931
3932 void Document::nodeChildrenWillBeRemoved(ContainerNode& container)
3933 {
3934     for (auto* range : m_ranges)
3935         range->nodeChildrenWillBeRemoved(container);
3936
3937     for (auto* it : m_nodeIterators) {
3938         for (Node* n = container.firstChild(); n; n = n->nextSibling())
3939             it->nodeWillBeRemoved(*n);
3940     }
3941
3942     if (Frame* frame = this->frame()) {
3943         for (Node* n = container.firstChild(); n; n = n->nextSibling()) {
3944             frame->eventHandler().nodeWillBeRemoved(*n);
3945             frame->selection().nodeWillBeRemoved(*n);
3946             frame->page()->dragCaretController().nodeWillBeRemoved(*n);
3947         }
3948     }
3949
3950     if (m_markers->hasMarkers()) {
3951         for (Text* textNode = TextNodeTraversal::firstChild(container); textNode; textNode = TextNodeTraversal::nextSibling(*textNode))
3952             m_markers->removeMarkers(textNode);
3953     }
3954 }
3955
3956 void Document::nodeWillBeRemoved(Node& n)
3957 {
3958     for (auto* it : m_nodeIterators)
3959         it->nodeWillBeRemoved(n);
3960
3961     for (auto* range : m_ranges)
3962         range->nodeWillBeRemoved(n);
3963
3964     if (Frame* frame = this->frame()) {
3965         frame->eventHandler().nodeWillBeRemoved(n);
3966         frame->selection().nodeWillBeRemoved(n);
3967         frame->page()->dragCaretController().nodeWillBeRemoved(n);
3968     }
3969
3970     if (is<Text>(n))
3971         m_markers->removeMarkers(&n);
3972 }
3973
3974 void Document::textInserted(Node* text, unsigned offset, unsigned length)
3975 {
3976     if (!m_ranges.isEmpty()) {
3977         for (auto* range : m_ranges)
3978             range->textInserted(text, offset, length);
3979     }
3980
3981     // Update the markers for spelling and grammar checking.
3982     m_markers->shiftMarkers(text, offset, length);
3983 }
3984
3985 void Document::textRemoved(Node* text, unsigned offset, unsigned length)
3986 {
3987     if (!m_ranges.isEmpty()) {
3988         for (auto* range : m_ranges)
3989             range->textRemoved(text, offset, length);
3990     }
3991
3992     // Update the markers for spelling and grammar checking.
3993     m_markers->removeMarkers(text, offset, length);
3994     m_markers->shiftMarkers(text, offset + length, 0 - length);
3995 }
3996
3997 void Document::textNodesMerged(Text* oldNode, unsigned offset)
3998 {
3999     if (!m_ranges.isEmpty()) {
4000         NodeWithIndex oldNodeWithIndex(oldNode);
4001         for (auto* range : m_ranges)
4002             range->textNodesMerged(oldNodeWithIndex, offset);
4003     }
4004
4005     // FIXME: This should update markers for spelling and grammar checking.
4006 }
4007
4008 void Document::textNodeSplit(Text* oldNode)
4009 {
4010     for (auto* range : m_ranges)
4011         range->textNodeSplit(oldNode);
4012
4013     // FIXME: This should update markers for spelling and grammar checking.
4014 }
4015
4016 void Document::createDOMWindow()
4017 {
4018     ASSERT(m_frame);
4019     ASSERT(!m_domWindow);
4020
4021     m_domWindow = DOMWindow::create(this);
4022
4023     ASSERT(m_domWindow->document() == this);
4024     ASSERT(m_domWindow->frame() == m_frame);
4025 }
4026
4027 void Document::takeDOMWindowFrom(Document* document)
4028 {
4029     ASSERT(m_frame);
4030     ASSERT(!m_domWindow);
4031     ASSERT(document->m_domWindow);
4032     // A valid DOMWindow is needed by CachedFrame for its documents.
4033     ASSERT(!document->inPageCache());
4034
4035     m_domWindow = document->m_domWindow.release();
4036     m_domWindow->didSecureTransitionTo(this);
4037
4038     ASSERT(m_domWindow->document() == this);
4039     ASSERT(m_domWindow->frame() == m_frame);
4040 }
4041
4042 void Document::setWindowAttributeEventListener(const AtomicString& eventType, PassRefPtr<EventListener> listener)
4043 {
4044     if (!m_domWindow)
4045         return;
4046     m_domWindow->setAttributeEventListener(eventType, listener);
4047 }
4048
4049 void Document::setWindowAttributeEventListener(const AtomicString& eventType, const QualifiedName& attributeName, const AtomicString& attributeValue)
4050 {
4051     if (!m_frame)
4052         return;
4053     setWindowAttributeEventListener(eventType, JSLazyEventListener::createForDOMWindow(*m_frame, attributeName, attributeValue));
4054 }
4055
4056 EventListener* Document::getWindowAttributeEventListener(const AtomicString& eventType)
4057 {
4058     if (!m_domWindow)
4059         return nullptr;
4060     return m_domWindow->getAttributeEventListener(eventType);
4061 }
4062
4063 void Document::dispatchWindowEvent(PassRefPtr<Event> event,  PassRefPtr<EventTarget> target)
4064 {
4065     ASSERT_WITH_SECURITY_IMPLICATION(!NoEventDispatchAssertion::isEventDispatchForbidden());
4066     if (!m_domWindow)
4067         return;
4068     m_domWindow->dispatchEvent(event, target);
4069 }
4070
4071 void Document::dispatchWindowLoadEvent()
4072 {
4073     ASSERT_WITH_SECURITY_IMPLICATION(!NoEventDispatchAssertion::isEventDispatchForbidden());
4074     if (!m_domWindow)
4075         return;
4076     m_domWindow->dispatchLoadEvent();
4077     m_loadEventFinished = true;
4078     m_cachedResourceLoader->documentDidFinishLoadEvent();
4079 }
4080
4081 void Document::enqueueWindowEvent(PassRefPtr<Event> event)
4082 {
4083     event->setTarget(m_domWindow.get());
4084     m_eventQueue.enqueueEvent(event);
4085 }
4086
4087 void Document::enqueueDocumentEvent(PassRefPtr<Event> event)
4088 {
4089     event->setTarget(this);
4090     m_eventQueue.enqueueEvent(event);
4091 }
4092
4093 void Document::enqueueOverflowEvent(PassRefPtr<Event> event)
4094 {
4095     m_eventQueue.enqueueEvent(event);
4096 }
4097
4098 RefPtr<Event> Document::createEvent(const String& eventType, ExceptionCode& ec)
4099 {
4100     RefPtr<Event> event = EventFactory::create(eventType);
4101     if (event)
4102         return event.release();
4103
4104     ec = NOT_SUPPORTED_ERR;
4105     return nullptr;
4106 }
4107
4108 bool Document::hasListenerTypeForEventType(PlatformEvent::Type eventType) const
4109 {
4110     switch (eventType) {
4111     case PlatformEvent::MouseForceChanged:
4112         return m_listenerTypes & Document::FORCECHANGED_LISTENER;
4113     case PlatformEvent::MouseForceDown:
4114         return m_listenerTypes & Document::FORCEDOWN_LISTENER;
4115     case PlatformEvent::MouseForceUp:
4116         return m_listenerTypes & Document::FORCEUP_LISTENER;
4117     case PlatformEvent::MouseScroll:
4118         return m_listenerTypes & Document::SCROLL_LISTENER;
4119     default:
4120         return false;
4121     }
4122 }
4123
4124 void Document::addListenerTypeIfNeeded(const AtomicString& eventType)
4125 {
4126     if (eventType == eventNames().DOMSubtreeModifiedEvent)
4127         addListenerType(DOMSUBTREEMODIFIED_LISTENER);
4128     else if (eventType == eventNames().DOMNodeInsertedEvent)
4129         addListenerType(DOMNODEINSERTED_LISTENER);
4130     else if (eventType == eventNames().DOMNodeRemovedEvent)
4131         addListenerType(DOMNODEREMOVED_LISTENER);
4132     else if (eventType == eventNames().DOMNodeRemovedFromDocumentEvent)
4133         addListenerType(DOMNODEREMOVEDFROMDOCUMENT_LISTENER);
4134     else if (eventType == eventNames().DOMNodeInsertedIntoDocumentEvent)
4135         addListenerType(DOMNODEINSERTEDINTODOCUMENT_LISTENER);
4136     else if (eventType == eventNames().DOMCharacterDataModifiedEvent)
4137         addListenerType(DOMCHARACTERDATAMODIFIED_LISTENER);
4138     else if (eventType == eventNames().overflowchangedEvent)
4139         addListenerType(OVERFLOWCHANGED_LISTENER);
4140     else if (eventType == eventNames().webkitAnimationStartEvent || eventType == eventNames().animationstartEvent)
4141         addListenerType(ANIMATIONSTART_LISTENER);
4142     else if (eventType == eventNames().webkitAnimationEndEvent || eventType == eventNames().animationendEvent)
4143         addListenerType(ANIMATIONEND_LISTENER);
4144     else if (eventType == eventNames().webkitAnimationIterationEvent || eventType == eventNames().animationiterationEvent)
4145         addListenerType(ANIMATIONITERATION_LISTENER);
4146     else if (eventType == eventNames().webkitTransitionEndEvent || eventType == eventNames().transitionendEvent)
4147         addListenerType(TRANSITIONEND_LISTENER);
4148     else if (eventType == eventNames().beforeloadEvent)
4149         addListenerType(BEFORELOAD_LISTENER);
4150     else if (eventType == eventNames().scrollEvent)
4151         addListenerType(SCROLL_LISTENER);
4152     else if (eventType == eventNames().webkitmouseforcewillbeginEvent)
4153         addListenerType(FORCEWILLBEGIN_LISTENER);
4154     else if (eventType == eventNames().webkitmouseforcechangedEvent)
4155         addListenerType(FORCECHANGED_LISTENER);
4156     else if (eventType == eventNames().webkitmouseforcedownEvent)
4157         addListenerType(FORCEDOWN_LISTENER);
4158     else if (eventType == eventNames().webkitmouseforceupEvent)
4159         addListenerType(FORCEUP_LISTENER);
4160 }
4161
4162 CSSStyleDeclaration* Document::getOverrideStyle(Element*, const String&)
4163 {
4164     return nullptr;
4165 }
4166
4167 HTMLFrameOwnerElement* Document::ownerElement() const
4168 {
4169     if (!frame())
4170         return nullptr;
4171     return frame()->ownerElement();
4172 }
4173
4174 String Document::cookie(ExceptionCode& ec)
4175 {
4176     if (page() && !page()->settings().cookieEnabled())
4177         return String();
4178
4179     // FIXME: The HTML5 DOM spec states that this attribute can raise an
4180     // INVALID_STATE_ERR exception on getting if the Document has no
4181     // browsing context.
4182
4183     if (!securityOrigin()->canAccessCookies()) {
4184         ec = SECURITY_ERR;
4185         return String();
4186     }
4187
4188     URL cookieURL = this->cookieURL();
4189     if (cookieURL.isEmpty())
4190         return String();
4191
4192     if (!isDOMCookieCacheValid())
4193         setCachedDOMCookies(cookies(this, cookieURL));
4194
4195     return cachedDOMCookies();
4196 }
4197
4198 void Document::setCookie(const String& value, ExceptionCode& ec)
4199 {
4200     if (page() && !page()->settings().cookieEnabled())
4201         return;
4202
4203     // FIXME: The HTML5 DOM spec states that this attribute can raise an
4204     // INVALID_STATE_ERR exception on setting if the Document has no
4205     // browsing context.
4206
4207     if (!securityOrigin()->canAccessCookies()) {
4208         ec = SECURITY_ERR;
4209         return;
4210     }
4211
4212     URL cookieURL = this->cookieURL();
4213     if (cookieURL.isEmpty())
4214         return;
4215
4216     invalidateDOMCookieCache();
4217     setCookies(this, cookieURL, value);
4218 }
4219
4220 String Document::referrer() const
4221 {
4222     if (frame())
4223         return frame()->loader().referrer();
4224     return String();
4225 }
4226
4227 String Document::origin() const
4228 {
4229     return securityOrigin()->databaseIdentifier();
4230 }
4231
4232 String Document::domain() const
4233 {
4234     return securityOrigin()->domain();
4235 }
4236
4237 void Document::setDomain(const String& newDomain, ExceptionCode& ec)
4238 {
4239     if (SchemeRegistry::isDomainRelaxationForbiddenForURLScheme(securityOrigin()->protocol())) {
4240         ec = SECURITY_ERR;
4241         return;
4242     }
4243
4244     // Both NS and IE specify that changing the domain is only allowed when
4245     // the new domain is a suffix of the old domain.
4246
4247     // FIXME: We should add logging indicating why a domain was not allowed.
4248
4249     // If the new domain is the same as the old domain, still call
4250     // securityOrigin()->setDomainForDOM. This will change the
4251     // security check behavior. For example, if a page loaded on port 8000
4252     // assigns its current domain using document.domain, the page will
4253     // allow other pages loaded on different ports in the same domain that
4254     // have also assigned to access this page.
4255     if (equalIgnoringCase(domain(), newDomain)) {
4256         securityOrigin()->setDomainFromDOM(newDomain);
4257         return;
4258     }
4259
4260     int oldLength = domain().length();
4261     int newLength = newDomain.length();
4262     // e.g. newDomain = webkit.org (10) and domain() = www.webkit.org (14)
4263     if (newLength >= oldLength) {
4264         ec = SECURITY_ERR;
4265         return;
4266     }
4267
4268     String test = domain();
4269     // Check that it's a subdomain, not e.g. "ebkit.org"
4270     if (test[oldLength - newLength - 1] != '.') {
4271         ec = SECURITY_ERR;
4272         return;
4273     }
4274
4275     // Now test is "webkit.org" from domain()
4276     // and we check that it's the same thing as newDomain
4277     test.remove(0, oldLength - newLength);
4278     if (test != newDomain) {
4279         ec = SECURITY_ERR;
4280         return;
4281     }
4282
4283     securityOrigin()->setDomainFromDOM(newDomain);
4284 }
4285
4286 // http://www.whatwg.org/specs/web-apps/current-work/#dom-document-lastmodified
4287 String Document::lastModified() const
4288 {
4289     using namespace std::chrono;
4290     Optional<system_clock::time_point> dateTime;
4291     if (m_frame && loader())
4292         dateTime = loader()->response().lastModified();
4293
4294     // FIXME: If this document came from the file system, the HTML5
4295     // specification tells us to read the last modification date from the file
4296     // system.
4297     if (!dateTime) {
4298         dateTime = system_clock::now();
4299 #if ENABLE(WEB_REPLAY)
4300         InputCursor& cursor = inputCursor();
4301         if (cursor.isCapturing())
4302             cursor.appendInput<DocumentLastModifiedDate>(duration_cast<milliseconds>(dateTime.value().time_since_epoch()).count());
4303         else if (cursor.isReplaying()) {
4304             if (DocumentLastModifiedDate* input = cursor.fetchInput<DocumentLastModifiedDate>())
4305                 dateTime = system_clock::time_point(milliseconds(static_cast<long long>(input->fallbackValue())));
4306         }
4307 #endif
4308     }
4309
4310     auto ctime = system_clock::to_time_t(dateTime.value());
4311     auto localDateTime = std::localtime(&ctime);
4312     return String::format("%02d/%02d/%04d %02d:%02d:%02d", localDateTime->tm_mon + 1, localDateTime->tm_mday, 1900 + localDateTime->tm_year, localDateTime->tm_hour, localDateTime->tm_min, localDateTime->tm_sec);
4313 }
4314
4315 void Document::setCookieURL(const URL& url)
4316 {
4317     if (m_cookieURL == url)
4318         return;
4319     m_cookieURL = url;