#include "config.h"
#include "DOMTimer.h"
#include "HTMLPlugInElement.h"
#include "InspectorInstrumentation.h"
#include "Logging.h"
#include "Page.h"
#include "PluginViewBase.h"
#include "ScheduledAction.h"
#include "ScriptExecutionContext.h"
#include "Settings.h"
#include <wtf/HashMap.h>
#include <wtf/MathExtras.h>
#include <wtf/NeverDestroyed.h>
#include <wtf/RandomNumber.h>
#include <wtf/StdLibExtras.h>
#if PLATFORM(IOS_FAMILY)
#include "ContentChangeObserver.h"
#include "DOMTimerHoldingTank.h"
#endif
namespace WebCore {
static const Seconds minIntervalForNonUserObservableChangeTimers { 1_s }; static const int maxTimerNestingLevel = 5;
class DOMTimerFireState {
public:
DOMTimerFireState(ScriptExecutionContext& context, int nestingLevel)
: m_context(context)
, m_contextIsDocument(is<Document>(m_context))
, m_initialDOMTreeVersion(m_contextIsDocument ? downcast<Document>(m_context).domTreeVersion() : 0)
, m_previous(m_contextIsDocument ? std::exchange(current, this) : nullptr)
{
m_context.setTimerNestingLevel(nestingLevel);
}
~DOMTimerFireState()
{
if (m_contextIsDocument)
current = m_previous;
m_context.setTimerNestingLevel(0);
}
Document* contextDocument() const { return m_contextIsDocument ? &downcast<Document>(m_context) : nullptr; }
void setScriptMadeUserObservableChanges() { m_scriptMadeUserObservableChanges = true; }
void setScriptMadeNonUserObservableChanges() { m_scriptMadeNonUserObservableChanges = true; }
bool scriptMadeNonUserObservableChanges() const { return m_scriptMadeNonUserObservableChanges; }
bool scriptMadeUserObservableChanges() const
{
if (m_scriptMadeUserObservableChanges)
return true;
Document* document = contextDocument();
return document && document->domTreeVersion() != m_initialDOMTreeVersion;
}
static DOMTimerFireState* current;
private:
ScriptExecutionContext& m_context;
bool m_contextIsDocument;
bool m_scriptMadeNonUserObservableChanges { false };
bool m_scriptMadeUserObservableChanges { false };
uint64_t m_initialDOMTreeVersion;
DOMTimerFireState* m_previous;
};
DOMTimerFireState* DOMTimerFireState::current = nullptr;
struct NestedTimersMap {
typedef HashMap<int, Ref<DOMTimer>>::const_iterator const_iterator;
static NestedTimersMap* instanceForContext(ScriptExecutionContext& context)
{
if (is<Document>(context))
return &instance();
return nullptr;
}
void startTracking()
{
nestedTimers.clear();
isTrackingNestedTimers = true;
}
void stopTracking()
{
isTrackingNestedTimers = false;
nestedTimers.clear();
}
void add(int timeoutId, Ref<DOMTimer>&& timer)
{
if (isTrackingNestedTimers)
nestedTimers.add(timeoutId, WTFMove(timer));
}
void remove(int timeoutId)
{
if (isTrackingNestedTimers)
nestedTimers.remove(timeoutId);
}
const_iterator begin() const { return nestedTimers.begin(); }
const_iterator end() const { return nestedTimers.end(); }
private:
static NestedTimersMap& instance()
{
static NeverDestroyed<NestedTimersMap> map;
return map;
}
static bool isTrackingNestedTimers;
HashMap<int , Ref<DOMTimer>> nestedTimers;
};
bool NestedTimersMap::isTrackingNestedTimers = false;
DOMTimer::DOMTimer(ScriptExecutionContext& context, std::unique_ptr<ScheduledAction> action, Seconds interval, bool singleShot)
: SuspendableTimerBase(&context)
, m_nestingLevel(context.timerNestingLevel())
, m_action(WTFMove(action))
, m_originalInterval(interval)
, m_throttleState(Undetermined)
, m_currentTimerInterval(intervalClampedToMinimum())
, m_userGestureTokenToForward(UserGestureIndicator::currentUserGesture())
{
if (singleShot)
startOneShot(m_currentTimerInterval);
else
startRepeating(m_currentTimerInterval);
}
DOMTimer::~DOMTimer() = default;
int DOMTimer::install(ScriptExecutionContext& context, std::unique_ptr<ScheduledAction> action, Seconds timeout, bool singleShot)
{
Ref<DOMTimer> timer = adoptRef(*new DOMTimer(context, WTFMove(action), timeout, singleShot));
timer->suspendIfNeeded();
do {
timer->m_timeoutId = context.circularSequentialID();
} while (!context.addTimeout(timer->m_timeoutId, timer.get()));
InspectorInstrumentation::didInstallTimer(context, timer->m_timeoutId, timeout, singleShot);
if (NestedTimersMap* nestedTimers = NestedTimersMap::instanceForContext(context))
nestedTimers->add(timer->m_timeoutId, timer.get());
#if PLATFORM(IOS_FAMILY)
if (is<Document>(context)) {
auto& document = downcast<Document>(context);
document.contentChangeObserver().didInstallDOMTimer(timer.get(), timeout, singleShot);
if (DeferDOMTimersForScope::isDeferring())
document.domTimerHoldingTank().add(timer.get());
}
#endif
return timer->m_timeoutId;
}
void DOMTimer::removeById(ScriptExecutionContext& context, int timeoutId)
{
if (timeoutId <= 0)
return;
#if PLATFORM(IOS_FAMILY)
if (is<Document>(context)) {
auto& document = downcast<Document>(context);
if (auto* timer = document.findTimeout(timeoutId)) {
document.contentChangeObserver().didRemoveDOMTimer(*timer);
if (auto* holdingTank = document.domTimerHoldingTankIfExists())
holdingTank->remove(*timer);
}
}
#endif
if (NestedTimersMap* nestedTimers = NestedTimersMap::instanceForContext(context))
nestedTimers->remove(timeoutId);
InspectorInstrumentation::didRemoveTimer(context, timeoutId);
context.removeTimeout(timeoutId);
}
inline bool DOMTimer::isDOMTimersThrottlingEnabled(Document& document) const
{
auto* page = document.page();
if (!page)
return true;
return page->settings().domTimersThrottlingEnabled();
}
void DOMTimer::updateThrottlingStateIfNecessary(const DOMTimerFireState& fireState)
{
Document* contextDocument = fireState.contextDocument();
if (!contextDocument)
return;
if (UNLIKELY(!isDOMTimersThrottlingEnabled(*contextDocument))) {
if (m_throttleState == ShouldThrottle) {
LOG(DOMTimers, "%p - Unthrottling DOM timer because throttling was disabled via settings.", this);
m_throttleState = ShouldNotThrottle;
updateTimerIntervalIfNecessary();
}
return;
}
if (fireState.scriptMadeUserObservableChanges()) {
if (m_throttleState != ShouldNotThrottle) {
m_throttleState = ShouldNotThrottle;
updateTimerIntervalIfNecessary();
}
} else if (fireState.scriptMadeNonUserObservableChanges()) {
if (m_throttleState != ShouldThrottle) {
m_throttleState = ShouldThrottle;
updateTimerIntervalIfNecessary();
}
}
}
void DOMTimer::scriptDidInteractWithPlugin(HTMLPlugInElement& pluginElement)
{
if (!DOMTimerFireState::current)
return;
if (pluginElement.isUserObservable())
DOMTimerFireState::current->setScriptMadeUserObservableChanges();
else
DOMTimerFireState::current->setScriptMadeNonUserObservableChanges();
}
void DOMTimer::fired()
{
Ref<DOMTimer> protectedThis(*this);
bool oneShot = !repeatInterval();
ASSERT(scriptExecutionContext());
ScriptExecutionContext& context = *scriptExecutionContext();
#if PLATFORM(IOS_FAMILY)
if (is<Document>(context)) {
auto& document = downcast<Document>(context);
if (auto* holdingTank = document.domTimerHoldingTankIfExists(); holdingTank && holdingTank->contains(*this)) {
if (oneShot)
startOneShot(0_s);
return;
}
}
#endif
DOMTimerFireState fireState(context, std::min(m_nestingLevel + 1, maxTimerNestingLevel));
if (m_userGestureTokenToForward && m_userGestureTokenToForward->hasExpired(UserGestureToken::maximumIntervalForUserGestureForwarding))
m_userGestureTokenToForward = nullptr;
ASSERT(!isSuspended());
ASSERT(!context.activeDOMObjectsAreSuspended());
UserGestureIndicator gestureIndicator(m_userGestureTokenToForward);
m_userGestureTokenToForward = nullptr;
InspectorInstrumentation::willFireTimer(context, m_timeoutId, oneShot);
if (isActive()) {
if (m_nestingLevel < maxTimerNestingLevel) {
m_nestingLevel++;
updateTimerIntervalIfNecessary();
}
m_action->execute(context);
InspectorInstrumentation::didFireTimer(context, m_timeoutId, oneShot);
updateThrottlingStateIfNecessary(fireState);
return;
}
context.removeTimeout(m_timeoutId);
NestedTimersMap* nestedTimers = NestedTimersMap::instanceForContext(context);
if (nestedTimers)
nestedTimers->startTracking();
#if PLATFORM(IOS_FAMILY)
ContentChangeObserver::DOMTimerScope observingScope(is<Document>(context) ? &downcast<Document>(context) : nullptr, *this);
#endif
m_action->execute(context);
InspectorInstrumentation::didFireTimer(context, m_timeoutId, oneShot);
if (nestedTimers) {
for (auto& idAndTimer : *nestedTimers) {
auto& timer = idAndTimer.value;
if (timer->isActive() && !timer->repeatInterval())
timer->updateThrottlingStateIfNecessary(fireState);
}
nestedTimers->stopTracking();
}
}
void DOMTimer::didStop()
{
m_action = nullptr;
}
void DOMTimer::updateTimerIntervalIfNecessary()
{
ASSERT(m_nestingLevel <= maxTimerNestingLevel);
auto previousInterval = m_currentTimerInterval;
m_currentTimerInterval = intervalClampedToMinimum();
if (previousInterval == m_currentTimerInterval)
return;
if (repeatInterval()) {
ASSERT(repeatInterval() == previousInterval);
LOG(DOMTimers, "%p - Updating DOMTimer's repeat interval from %.2f ms to %.2f ms due to throttling.", this, previousInterval.milliseconds(), m_currentTimerInterval.milliseconds());
augmentRepeatInterval(m_currentTimerInterval - previousInterval);
} else {
LOG(DOMTimers, "%p - Updating DOMTimer's fire interval from %.2f ms to %.2f ms due to throttling.", this, previousInterval.milliseconds(), m_currentTimerInterval.milliseconds());
augmentFireInterval(m_currentTimerInterval - previousInterval);
}
}
Seconds DOMTimer::intervalClampedToMinimum() const
{
ASSERT(scriptExecutionContext());
ASSERT(m_nestingLevel <= maxTimerNestingLevel);
Seconds interval = std::max(1_ms, m_originalInterval);
if (m_nestingLevel < maxTimerNestingLevel)
return interval;
interval = std::max(interval, scriptExecutionContext()->minimumDOMTimerInterval());
if (m_throttleState == ShouldThrottle)
interval = std::max(interval, minIntervalForNonUserObservableChangeTimers);
return interval;
}
Optional<MonotonicTime> DOMTimer::alignedFireTime(MonotonicTime fireTime) const
{
Seconds alignmentInterval = scriptExecutionContext()->domTimerAlignmentInterval(m_nestingLevel >= maxTimerNestingLevel);
if (!alignmentInterval)
return WTF::nullopt;
static const double randomizedProportion = randomNumber();
Seconds randomizedOffset = alignmentInterval * randomizedProportion;
MonotonicTime adjustedFireTime = fireTime - randomizedOffset;
return adjustedFireTime - (adjustedFireTime % alignmentInterval) + alignmentInterval + randomizedOffset;
}
const char* DOMTimer::activeDOMObjectName() const
{
return "DOMTimer";
}
}