08d60d7521
This also fixes a bug where task IDs were being deallocated from the wrong IDAllocator. I don't know if it was actually possible to cause any real trouble with that mistake, nor do I know how to write a test for it, but this makes the bug go away.
134 lines
4.6 KiB
C++
134 lines
4.6 KiB
C++
/*
|
||
* Copyright (c) 2022, Linus Groh <linusg@serenityos.org>
|
||
*
|
||
* SPDX-License-Identifier: BSD-2-Clause
|
||
*/
|
||
|
||
#include <LibJS/Heap/Heap.h>
|
||
#include <LibJS/Runtime/VM.h>
|
||
#include <LibWeb/Fetch/Infrastructure/FetchAlgorithms.h>
|
||
#include <LibWeb/Fetch/Infrastructure/FetchController.h>
|
||
#include <LibWeb/Fetch/Infrastructure/FetchParams.h>
|
||
#include <LibWeb/HTML/EventLoop/EventLoop.h>
|
||
#include <LibWeb/WebIDL/DOMException.h>
|
||
|
||
namespace Web::Fetch::Infrastructure {
|
||
|
||
JS_DEFINE_ALLOCATOR(FetchController);
|
||
|
||
FetchController::FetchController() = default;
|
||
|
||
JS::NonnullGCPtr<FetchController> FetchController::create(JS::VM& vm)
|
||
{
|
||
return vm.heap().allocate_without_realm<FetchController>();
|
||
}
|
||
|
||
void FetchController::visit_edges(JS::Cell::Visitor& visitor)
|
||
{
|
||
Base::visit_edges(visitor);
|
||
visitor.visit(m_full_timing_info);
|
||
visitor.visit(m_report_timing_steps);
|
||
visitor.visit(m_next_manual_redirect_steps);
|
||
visitor.visit(m_fetch_params);
|
||
}
|
||
|
||
void FetchController::set_report_timing_steps(Function<void(JS::Object const&)> report_timing_steps)
|
||
{
|
||
m_report_timing_steps = JS::create_heap_function(vm().heap(), move(report_timing_steps));
|
||
}
|
||
|
||
void FetchController::set_next_manual_redirect_steps(Function<void()> next_manual_redirect_steps)
|
||
{
|
||
m_next_manual_redirect_steps = JS::create_heap_function(vm().heap(), move(next_manual_redirect_steps));
|
||
}
|
||
|
||
// https://fetch.spec.whatwg.org/#finalize-and-report-timing
|
||
void FetchController::report_timing(JS::Object const& global) const
|
||
{
|
||
// 1. Assert: this’s report timing steps is not null.
|
||
VERIFY(m_report_timing_steps);
|
||
|
||
// 2. Call this’s report timing steps with global.
|
||
m_report_timing_steps->function()(global);
|
||
}
|
||
|
||
// https://fetch.spec.whatwg.org/#fetch-controller-process-the-next-manual-redirect
|
||
void FetchController::process_next_manual_redirect() const
|
||
{
|
||
// 1. Assert: controller’s next manual redirect steps are not null.
|
||
VERIFY(m_next_manual_redirect_steps);
|
||
|
||
// 2. Call controller’s next manual redirect steps.
|
||
m_next_manual_redirect_steps->function()();
|
||
}
|
||
|
||
// https://fetch.spec.whatwg.org/#extract-full-timing-info
|
||
JS::NonnullGCPtr<FetchTimingInfo> FetchController::extract_full_timing_info() const
|
||
{
|
||
// 1. Assert: this’s full timing info is not null.
|
||
VERIFY(m_full_timing_info);
|
||
|
||
// 2. Return this’s full timing info.
|
||
return *m_full_timing_info;
|
||
}
|
||
|
||
// https://fetch.spec.whatwg.org/#fetch-controller-abort
|
||
void FetchController::abort(JS::Realm& realm, Optional<JS::Value> error)
|
||
{
|
||
// 1. Set controller’s state to "aborted".
|
||
m_state = State::Aborted;
|
||
|
||
// 2. Let fallbackError be an "AbortError" DOMException.
|
||
auto fallback_error = WebIDL::AbortError::create(realm, "Fetch was aborted"_fly_string);
|
||
|
||
// 3. Set error to fallbackError if it is not given.
|
||
if (!error.has_value())
|
||
error = fallback_error;
|
||
|
||
// FIXME: 4. Let serializedError be StructuredSerialize(error). If that threw an exception, catch it, and let serializedError be StructuredSerialize(fallbackError).
|
||
// FIXME: 5. Set controller’s serialized abort reason to serializedError.
|
||
(void)error;
|
||
}
|
||
|
||
// FIXME: https://fetch.spec.whatwg.org/#deserialize-a-serialized-abort-reason
|
||
|
||
// https://fetch.spec.whatwg.org/#fetch-controller-terminate
|
||
void FetchController::terminate()
|
||
{
|
||
// To terminate a fetch controller controller, set controller’s state to "terminated".
|
||
m_state = State::Terminated;
|
||
}
|
||
|
||
void FetchController::stop_fetch()
|
||
{
|
||
auto& vm = this->vm();
|
||
|
||
// AD-HOC: Some HTML elements need to stop an ongoing fetching process without causing any network error to be raised
|
||
// (which abort() and terminate() will both do). This is tricky because the fetch process runs across several
|
||
// nested Platform::EventLoopPlugin::deferred_invoke() invocations. For now, we "stop" the fetch process by
|
||
// cancelling any queued fetch tasks and then ignoring any callbacks.
|
||
auto ongoing_fetch_tasks = move(m_ongoing_fetch_tasks);
|
||
|
||
HTML::main_thread_event_loop().task_queue().remove_tasks_matching([&](auto const& task) {
|
||
return ongoing_fetch_tasks.remove_all_matching([&](u64, HTML::TaskID task_id) {
|
||
return task.id() == task_id;
|
||
});
|
||
});
|
||
|
||
if (m_fetch_params) {
|
||
auto fetch_algorithms = FetchAlgorithms::create(vm, {});
|
||
m_fetch_params->set_algorithms(fetch_algorithms);
|
||
}
|
||
}
|
||
|
||
void FetchController::fetch_task_queued(u64 fetch_task_id, HTML::TaskID event_id)
|
||
{
|
||
m_ongoing_fetch_tasks.set(fetch_task_id, event_id);
|
||
}
|
||
|
||
void FetchController::fetch_task_complete(u64 fetch_task_id)
|
||
{
|
||
m_ongoing_fetch_tasks.remove(fetch_task_id);
|
||
}
|
||
|
||
}
|