| // Copyright 2018 the V8 project authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include <memory> |
| #include <unordered_map> |
| |
| #include "include/v8-platform.h" |
| #include "src/base/logging.h" |
| #include "src/base/macros.h" |
| #include "src/base/platform/mutex.h" |
| #include "src/base/platform/platform.h" |
| #include "src/base/platform/time.h" |
| #include "src/base/utils/random-number-generator.h" |
| #include "src/d8/d8-platforms.h" |
| |
| namespace v8 { |
| |
| class PredictablePlatform final : public Platform { |
| public: |
| explicit PredictablePlatform(std::unique_ptr<Platform> platform) |
| : platform_(std::move(platform)) { |
| DCHECK_NOT_NULL(platform_); |
| } |
| |
| PageAllocator* GetPageAllocator() override { |
| return platform_->GetPageAllocator(); |
| } |
| |
| void OnCriticalMemoryPressure() override { |
| platform_->OnCriticalMemoryPressure(); |
| } |
| |
| bool OnCriticalMemoryPressure(size_t length) override { |
| return platform_->OnCriticalMemoryPressure(length); |
| } |
| |
| std::shared_ptr<TaskRunner> GetForegroundTaskRunner( |
| v8::Isolate* isolate) override { |
| return platform_->GetForegroundTaskRunner(isolate); |
| } |
| |
| int NumberOfWorkerThreads() override { return 0; } |
| |
| void CallOnWorkerThread(std::unique_ptr<Task> task) override { |
| // We post worker tasks on the foreground task runner of the |
| // {kProcessGlobalPredictablePlatformWorkerTaskQueue} isolate. The task |
| // queue of the {kProcessGlobalPredictablePlatformWorkerTaskQueue} isolate |
| // is then executed on the main thread to achieve predictable behavior. |
| // |
| // In this context here it is okay to call {GetForegroundTaskRunner} from a |
| // background thread. The reason is that code is executed sequentially with |
| // the PredictablePlatform, and that the {DefaultPlatform} does not access |
| // the isolate but only uses it as the key in a HashMap. |
| GetForegroundTaskRunner(kProcessGlobalPredictablePlatformWorkerTaskQueue) |
| ->PostTask(std::move(task)); |
| } |
| |
| void CallDelayedOnWorkerThread(std::unique_ptr<Task> task, |
| double delay_in_seconds) override { |
| // Never run delayed tasks. |
| } |
| |
| bool IdleTasksEnabled(Isolate* isolate) override { return false; } |
| |
| std::unique_ptr<JobHandle> PostJob( |
| TaskPriority priority, std::unique_ptr<JobTask> job_task) override { |
| return platform_->PostJob(priority, std::move(job_task)); |
| } |
| |
| double MonotonicallyIncreasingTime() override { |
| return synthetic_time_in_sec_ += 0.00001; |
| } |
| |
| double CurrentClockTimeMillis() override { |
| return MonotonicallyIncreasingTime() * base::Time::kMillisecondsPerSecond; |
| } |
| |
| v8::TracingController* GetTracingController() override { |
| return platform_->GetTracingController(); |
| } |
| |
| Platform* platform() const { return platform_.get(); } |
| |
| private: |
| double synthetic_time_in_sec_ = 0.0; |
| std::unique_ptr<Platform> platform_; |
| |
| DISALLOW_COPY_AND_ASSIGN(PredictablePlatform); |
| }; |
| |
| std::unique_ptr<Platform> MakePredictablePlatform( |
| std::unique_ptr<Platform> platform) { |
| return std::make_unique<PredictablePlatform>(std::move(platform)); |
| } |
| |
| class DelayedTasksPlatform final : public Platform { |
| public: |
| explicit DelayedTasksPlatform(std::unique_ptr<Platform> platform) |
| : platform_(std::move(platform)) { |
| DCHECK_NOT_NULL(platform_); |
| } |
| |
| explicit DelayedTasksPlatform(std::unique_ptr<Platform> platform, |
| int64_t random_seed) |
| : platform_(std::move(platform)), rng_(random_seed) { |
| DCHECK_NOT_NULL(platform_); |
| } |
| |
| ~DelayedTasksPlatform() override { |
| // When the platform shuts down, all task runners must be freed. |
| DCHECK_EQ(0, delayed_task_runners_.size()); |
| } |
| |
| PageAllocator* GetPageAllocator() override { |
| return platform_->GetPageAllocator(); |
| } |
| |
| void OnCriticalMemoryPressure() override { |
| platform_->OnCriticalMemoryPressure(); |
| } |
| |
| bool OnCriticalMemoryPressure(size_t length) override { |
| return platform_->OnCriticalMemoryPressure(length); |
| } |
| |
| std::shared_ptr<TaskRunner> GetForegroundTaskRunner( |
| v8::Isolate* isolate) override { |
| std::shared_ptr<TaskRunner> runner = |
| platform_->GetForegroundTaskRunner(isolate); |
| |
| base::MutexGuard lock_guard(&mutex_); |
| // Check if we can re-materialize the weak ptr in our map. |
| std::weak_ptr<DelayedTaskRunner>& weak_delayed_runner = |
| delayed_task_runners_[runner.get()]; |
| std::shared_ptr<DelayedTaskRunner> delayed_runner = |
| weak_delayed_runner.lock(); |
| |
| if (!delayed_runner) { |
| // Create a new {DelayedTaskRunner} and keep a weak reference in our map. |
| delayed_runner.reset(new DelayedTaskRunner(runner, this), |
| DelayedTaskRunnerDeleter{}); |
| weak_delayed_runner = delayed_runner; |
| } |
| |
| return std::move(delayed_runner); |
| } |
| |
| int NumberOfWorkerThreads() override { |
| return platform_->NumberOfWorkerThreads(); |
| } |
| |
| void CallOnWorkerThread(std::unique_ptr<Task> task) override { |
| platform_->CallOnWorkerThread(MakeDelayedTask(std::move(task))); |
| } |
| |
| void CallDelayedOnWorkerThread(std::unique_ptr<Task> task, |
| double delay_in_seconds) override { |
| platform_->CallDelayedOnWorkerThread(MakeDelayedTask(std::move(task)), |
| delay_in_seconds); |
| } |
| |
| bool IdleTasksEnabled(Isolate* isolate) override { |
| return platform_->IdleTasksEnabled(isolate); |
| } |
| |
| std::unique_ptr<JobHandle> PostJob( |
| TaskPriority priority, std::unique_ptr<JobTask> job_task) override { |
| return platform_->PostJob(priority, MakeDelayedJob(std::move(job_task))); |
| } |
| |
| double MonotonicallyIncreasingTime() override { |
| return platform_->MonotonicallyIncreasingTime(); |
| } |
| |
| double CurrentClockTimeMillis() override { |
| return platform_->CurrentClockTimeMillis(); |
| } |
| |
| v8::TracingController* GetTracingController() override { |
| return platform_->GetTracingController(); |
| } |
| |
| private: |
| class DelayedTaskRunnerDeleter; |
| class DelayedTaskRunner final : public TaskRunner { |
| public: |
| DelayedTaskRunner(std::shared_ptr<TaskRunner> task_runner, |
| DelayedTasksPlatform* platform) |
| : task_runner_(task_runner), platform_(platform) {} |
| |
| void PostTask(std::unique_ptr<Task> task) final { |
| task_runner_->PostTask(platform_->MakeDelayedTask(std::move(task))); |
| } |
| |
| void PostNonNestableTask(std::unique_ptr<Task> task) final { |
| task_runner_->PostNonNestableTask( |
| platform_->MakeDelayedTask(std::move(task))); |
| } |
| |
| void PostDelayedTask(std::unique_ptr<Task> task, |
| double delay_in_seconds) final { |
| task_runner_->PostDelayedTask(platform_->MakeDelayedTask(std::move(task)), |
| delay_in_seconds); |
| } |
| |
| void PostIdleTask(std::unique_ptr<IdleTask> task) final { |
| task_runner_->PostIdleTask( |
| platform_->MakeDelayedIdleTask(std::move(task))); |
| } |
| |
| bool IdleTasksEnabled() final { return task_runner_->IdleTasksEnabled(); } |
| |
| bool NonNestableTasksEnabled() const final { |
| return task_runner_->NonNestableTasksEnabled(); |
| } |
| |
| private: |
| friend class DelayedTaskRunnerDeleter; |
| std::shared_ptr<TaskRunner> task_runner_; |
| DelayedTasksPlatform* platform_; |
| }; |
| |
| class DelayedTaskRunnerDeleter { |
| public: |
| void operator()(DelayedTaskRunner* runner) const { |
| TaskRunner* original_runner = runner->task_runner_.get(); |
| base::MutexGuard lock_guard(&runner->platform_->mutex_); |
| auto& delayed_task_runners = runner->platform_->delayed_task_runners_; |
| DCHECK_EQ(1, delayed_task_runners.count(original_runner)); |
| delayed_task_runners.erase(original_runner); |
| } |
| }; |
| |
| class DelayedTask final : public Task { |
| public: |
| DelayedTask(std::unique_ptr<Task> task, int32_t delay_ms) |
| : task_(std::move(task)), delay_ms_(delay_ms) {} |
| |
| void Run() override { |
| base::OS::Sleep(base::TimeDelta::FromMicroseconds(delay_ms_)); |
| task_->Run(); |
| } |
| |
| private: |
| std::unique_ptr<Task> task_; |
| int32_t delay_ms_; |
| }; |
| |
| class DelayedIdleTask final : public IdleTask { |
| public: |
| DelayedIdleTask(std::unique_ptr<IdleTask> task, int32_t delay_ms) |
| : task_(std::move(task)), delay_ms_(delay_ms) {} |
| |
| void Run(double deadline_in_seconds) override { |
| base::OS::Sleep(base::TimeDelta::FromMicroseconds(delay_ms_)); |
| task_->Run(deadline_in_seconds); |
| } |
| |
| private: |
| std::unique_ptr<IdleTask> task_; |
| int32_t delay_ms_; |
| }; |
| |
| class DelayedJob final : public JobTask { |
| public: |
| DelayedJob(std::unique_ptr<JobTask> job_task, int32_t delay_ms) |
| : job_task_(std::move(job_task)), delay_ms_(delay_ms) {} |
| |
| void Run(JobDelegate* delegate) override { |
| // If this job is being executed via worker tasks (as e.g. the |
| // {DefaultJobHandle} implementation does it), the worker task would |
| // already include a delay. In order to not depend on that, we add our own |
| // delay here anyway. |
| base::OS::Sleep(base::TimeDelta::FromMicroseconds(delay_ms_)); |
| job_task_->Run(delegate); |
| } |
| |
| size_t GetMaxConcurrency(size_t worker_count) const override { |
| return job_task_->GetMaxConcurrency(worker_count); |
| } |
| |
| private: |
| std::unique_ptr<JobTask> job_task_; |
| int32_t delay_ms_; |
| }; |
| |
| std::unique_ptr<Platform> platform_; |
| |
| // The Mutex protects the RNG, which is used by foreground and background |
| // threads, and the {delayed_task_runners_} map might be accessed concurrently |
| // by the shared_ptr destructor. |
| base::Mutex mutex_; |
| base::RandomNumberGenerator rng_; |
| std::unordered_map<TaskRunner*, std::weak_ptr<DelayedTaskRunner>> |
| delayed_task_runners_; |
| |
| int32_t GetRandomDelayInMilliseconds() { |
| base::MutexGuard lock_guard(&mutex_); |
| double delay_fraction = rng_.NextDouble(); |
| // Sleep up to 100ms (100000us). Square {delay_fraction} to shift |
| // distribution towards shorter sleeps. |
| return 1e5 * (delay_fraction * delay_fraction); |
| } |
| |
| std::unique_ptr<Task> MakeDelayedTask(std::unique_ptr<Task> task) { |
| return std::make_unique<DelayedTask>(std::move(task), |
| GetRandomDelayInMilliseconds()); |
| } |
| |
| std::unique_ptr<IdleTask> MakeDelayedIdleTask( |
| std::unique_ptr<IdleTask> task) { |
| return std::make_unique<DelayedIdleTask>(std::move(task), |
| GetRandomDelayInMilliseconds()); |
| } |
| |
| std::unique_ptr<JobTask> MakeDelayedJob(std::unique_ptr<JobTask> task) { |
| return std::make_unique<DelayedJob>(std::move(task), |
| GetRandomDelayInMilliseconds()); |
| } |
| |
| DISALLOW_COPY_AND_ASSIGN(DelayedTasksPlatform); |
| }; |
| |
| std::unique_ptr<Platform> MakeDelayedTasksPlatform( |
| std::unique_ptr<Platform> platform, int64_t random_seed) { |
| if (random_seed) { |
| return std::make_unique<DelayedTasksPlatform>(std::move(platform), |
| random_seed); |
| } |
| return std::make_unique<DelayedTasksPlatform>(std::move(platform)); |
| } |
| |
| } // namespace v8 |