/* This Source Code Form is subject to the terms of the Mozilla Public * License, v. 2.0. If a copy of the MPL was not distributed with this * file, You can obtain one at https://mozilla.org/MPL/2.0/. */ use std::mem::replace; use std::sync::atomic::AtomicBool; use std::sync::Arc; use std::thread::{self, JoinHandle}; use base::id::{BrowsingContextId, PipelineId, TopLevelBrowsingContextId}; use crossbeam_channel::{unbounded, Receiver, Sender}; use devtools_traits::DevtoolScriptControlMsg; use dom_struct::dom_struct; use ipc_channel::ipc::IpcReceiver; use ipc_channel::router::ROUTER; use js::jsapi::{Heap, JSContext, JSObject, JS_AddInterruptCallback}; use js::jsval::UndefinedValue; use js::rust::{CustomAutoRooter, CustomAutoRooterGuard, HandleValue}; use net_traits::image_cache::ImageCache; use net_traits::request::{ CredentialsMode, Destination, ParserMetadata, Referrer, RequestBuilder, RequestMode, }; use net_traits::IpcSend; use script_traits::{WorkerGlobalScopeInit, WorkerScriptLoadOrigin}; use servo_rand::random; use servo_url::{ImmutableOrigin, ServoUrl}; use style::thread_state::{self, ThreadState}; use crate::devtools; use crate::dom::abstractworker::{SimpleWorkerErrorHandler, WorkerScriptMsg}; use crate::dom::abstractworkerglobalscope::{ run_worker_event_loop, SendableWorkerScriptChan, WorkerEventLoopMethods, WorkerThreadWorkerChan, }; use crate::dom::bindings::cell::DomRefCell; use crate::dom::bindings::codegen::Bindings::DedicatedWorkerGlobalScopeBinding; use crate::dom::bindings::codegen::Bindings::DedicatedWorkerGlobalScopeBinding::DedicatedWorkerGlobalScopeMethods; use crate::dom::bindings::codegen::Bindings::MessagePortBinding::StructuredSerializeOptions; use crate::dom::bindings::codegen::Bindings::WorkerBinding::WorkerType; use crate::dom::bindings::error::{ErrorInfo, ErrorResult}; use crate::dom::bindings::inheritance::Castable; use crate::dom::bindings::reflector::DomObject; use crate::dom::bindings::root::{DomRoot, RootCollection, ThreadLocalStackRoots}; use crate::dom::bindings::str::DOMString; use crate::dom::bindings::structuredclone; use crate::dom::bindings::trace::RootedTraceableBox; use crate::dom::errorevent::ErrorEvent; use crate::dom::event::{Event, EventBubbles, EventCancelable, EventStatus}; use crate::dom::eventtarget::EventTarget; use crate::dom::globalscope::GlobalScope; use crate::dom::identityhub::Identities; use crate::dom::messageevent::MessageEvent; use crate::dom::worker::{TrustedWorkerAddress, Worker}; use crate::dom::workerglobalscope::WorkerGlobalScope; use crate::fetch::load_whole_resource; use crate::realms::{enter_realm, AlreadyInRealm, InRealm}; use crate::script_runtime::ScriptThreadEventCategory::WorkerEvent; use crate::script_runtime::{ new_child_runtime, CommonScriptMsg, ContextForRequestInterrupt, JSContext as SafeJSContext, Runtime, ScriptChan, ScriptPort, }; use crate::task_queue::{QueuedTask, QueuedTaskConversion, TaskQueue}; use crate::task_source::networking::NetworkingTaskSource; use crate::task_source::TaskSourceName; /// Set the `worker` field of a related DedicatedWorkerGlobalScope object to a particular /// value for the duration of this object's lifetime. This ensures that the related Worker /// object only lives as long as necessary (ie. while events are being executed), while /// providing a reference that can be cloned freely. pub struct AutoWorkerReset<'a> { workerscope: &'a DedicatedWorkerGlobalScope, old_worker: Option, } impl<'a> AutoWorkerReset<'a> { fn new( workerscope: &'a DedicatedWorkerGlobalScope, worker: TrustedWorkerAddress, ) -> AutoWorkerReset<'a> { AutoWorkerReset { workerscope, old_worker: replace(&mut *workerscope.worker.borrow_mut(), Some(worker)), } } } impl<'a> Drop for AutoWorkerReset<'a> { fn drop(&mut self) { self.workerscope .worker .borrow_mut() .clone_from(&self.old_worker) } } /// Messages sent from the owning global. pub enum DedicatedWorkerControlMsg { /// Shutdown the worker. Exit, } pub enum DedicatedWorkerScriptMsg { /// Standard message from a worker. CommonWorker(TrustedWorkerAddress, WorkerScriptMsg), /// Wake-up call from the task queue. WakeUp, } pub enum MixedMessage { Worker(DedicatedWorkerScriptMsg), Devtools(DevtoolScriptControlMsg), Control(DedicatedWorkerControlMsg), } impl QueuedTaskConversion for DedicatedWorkerScriptMsg { fn task_source_name(&self) -> Option<&TaskSourceName> { let common_worker_msg = match self { DedicatedWorkerScriptMsg::CommonWorker(_, common_worker_msg) => common_worker_msg, _ => return None, }; let script_msg = match common_worker_msg { WorkerScriptMsg::Common(ref script_msg) => script_msg, _ => return None, }; match script_msg { CommonScriptMsg::Task(_category, _boxed, _pipeline_id, source_name) => { Some(source_name) }, _ => None, } } fn pipeline_id(&self) -> Option { // Workers always return None, since the pipeline_id is only used to check for document activity, // and this check does not apply to worker event-loops. None } fn into_queued_task(self) -> Option { let (worker, common_worker_msg) = match self { DedicatedWorkerScriptMsg::CommonWorker(worker, common_worker_msg) => { (worker, common_worker_msg) }, _ => return None, }; let script_msg = match common_worker_msg { WorkerScriptMsg::Common(script_msg) => script_msg, _ => return None, }; let (category, boxed, pipeline_id, task_source) = match script_msg { CommonScriptMsg::Task(category, boxed, pipeline_id, task_source) => { (category, boxed, pipeline_id, task_source) }, _ => return None, }; Some((Some(worker), category, boxed, pipeline_id, task_source)) } fn from_queued_task(queued_task: QueuedTask) -> Self { let (worker, category, boxed, pipeline_id, task_source) = queued_task; let script_msg = CommonScriptMsg::Task(category, boxed, pipeline_id, task_source); DedicatedWorkerScriptMsg::CommonWorker(worker.unwrap(), WorkerScriptMsg::Common(script_msg)) } fn inactive_msg() -> Self { // Inactive is only relevant in the context of a browsing-context event-loop. panic!("Workers should never receive messages marked as inactive"); } fn wake_up_msg() -> Self { DedicatedWorkerScriptMsg::WakeUp } fn is_wake_up(&self) -> bool { matches!(self, DedicatedWorkerScriptMsg::WakeUp) } } unsafe_no_jsmanaged_fields!(TaskQueue); // https://html.spec.whatwg.org/multipage/#dedicatedworkerglobalscope #[dom_struct] pub struct DedicatedWorkerGlobalScope { workerglobalscope: WorkerGlobalScope, #[ignore_malloc_size_of = "Defined in std"] task_queue: TaskQueue, #[ignore_malloc_size_of = "Defined in std"] #[no_trace] own_sender: Sender, #[ignore_malloc_size_of = "Trusted has unclear ownership like Dom"] worker: DomRefCell>, #[ignore_malloc_size_of = "Can't measure trait objects"] /// Sender to the parent thread. parent_sender: Box, #[ignore_malloc_size_of = "Arc"] #[no_trace] image_cache: Arc, #[no_trace] browsing_context: Option, /// A receiver of control messages, /// currently only used to signal shutdown. #[ignore_malloc_size_of = "Channels are hard"] #[no_trace] control_receiver: Receiver, } impl WorkerEventLoopMethods for DedicatedWorkerGlobalScope { type WorkerMsg = DedicatedWorkerScriptMsg; type ControlMsg = DedicatedWorkerControlMsg; type Event = MixedMessage; fn task_queue(&self) -> &TaskQueue { &self.task_queue } fn handle_event(&self, event: MixedMessage) -> bool { self.handle_mixed_message(event) } fn handle_worker_post_event(&self, worker: &TrustedWorkerAddress) -> Option { let ar = AutoWorkerReset::new(self, worker.clone()); Some(ar) } fn from_control_msg(msg: DedicatedWorkerControlMsg) -> MixedMessage { MixedMessage::Control(msg) } fn from_worker_msg(msg: DedicatedWorkerScriptMsg) -> MixedMessage { MixedMessage::Worker(msg) } fn from_devtools_msg(msg: DevtoolScriptControlMsg) -> MixedMessage { MixedMessage::Devtools(msg) } fn control_receiver(&self) -> &Receiver { &self.control_receiver } } impl DedicatedWorkerGlobalScope { #[allow(clippy::too_many_arguments)] fn new_inherited( init: WorkerGlobalScopeInit, worker_name: DOMString, worker_type: WorkerType, worker_url: ServoUrl, from_devtools_receiver: Receiver, runtime: Runtime, parent_sender: Box, own_sender: Sender, receiver: Receiver, closing: Arc, image_cache: Arc, browsing_context: Option, gpu_id_hub: Arc, control_receiver: Receiver, ) -> DedicatedWorkerGlobalScope { DedicatedWorkerGlobalScope { workerglobalscope: WorkerGlobalScope::new_inherited( init, worker_name, worker_type, worker_url, runtime, from_devtools_receiver, closing, gpu_id_hub, ), task_queue: TaskQueue::new(receiver, own_sender.clone()), own_sender, parent_sender, worker: DomRefCell::new(None), image_cache, browsing_context, control_receiver, } } #[allow(unsafe_code, clippy::too_many_arguments)] pub fn new( init: WorkerGlobalScopeInit, worker_name: DOMString, worker_type: WorkerType, worker_url: ServoUrl, from_devtools_receiver: Receiver, runtime: Runtime, parent_sender: Box, own_sender: Sender, receiver: Receiver, closing: Arc, image_cache: Arc, browsing_context: Option, gpu_id_hub: Arc, control_receiver: Receiver, ) -> DomRoot { let cx = runtime.cx(); let scope = Box::new(DedicatedWorkerGlobalScope::new_inherited( init, worker_name, worker_type, worker_url, from_devtools_receiver, runtime, parent_sender, own_sender, receiver, closing, image_cache, browsing_context, gpu_id_hub, control_receiver, )); unsafe { DedicatedWorkerGlobalScopeBinding::Wrap(SafeJSContext::from_ptr(cx), scope) } } /// #[allow(unsafe_code, clippy::too_many_arguments)] pub fn run_worker_scope( mut init: WorkerGlobalScopeInit, worker_url: ServoUrl, from_devtools_receiver: IpcReceiver, worker: TrustedWorkerAddress, parent_sender: Box, own_sender: Sender, receiver: Receiver, worker_load_origin: WorkerScriptLoadOrigin, worker_name: String, worker_type: WorkerType, closing: Arc, image_cache: Arc, browsing_context: Option, gpu_id_hub: Arc, control_receiver: Receiver, context_sender: Sender, ) -> JoinHandle<()> { let serialized_worker_url = worker_url.to_string(); let top_level_browsing_context_id = TopLevelBrowsingContextId::installed(); let current_global = GlobalScope::current().expect("No current global object"); let origin = current_global.origin().immutable().clone(); let referrer = current_global.get_referrer(); let parent = current_global.runtime_handle(); let current_global_https_state = current_global.get_https_state(); thread::Builder::new() .name(format!("WW:{}", worker_url.debug_compact())) .spawn(move || { thread_state::initialize(ThreadState::SCRIPT | ThreadState::IN_WORKER); if let Some(top_level_browsing_context_id) = top_level_browsing_context_id { TopLevelBrowsingContextId::install(top_level_browsing_context_id); } let roots = RootCollection::new(); let _stack_roots = ThreadLocalStackRoots::new(&roots); let WorkerScriptLoadOrigin { referrer_url, referrer_policy, pipeline_id, } = worker_load_origin; let referrer = referrer_url.map(Referrer::ReferrerUrl).unwrap_or(referrer); let request = RequestBuilder::new(worker_url.clone(), referrer) .destination(Destination::Worker) .mode(RequestMode::SameOrigin) .credentials_mode(CredentialsMode::CredentialsSameOrigin) .parser_metadata(ParserMetadata::NotParserInserted) .use_url_credentials(true) .pipeline_id(Some(pipeline_id)) .referrer_policy(referrer_policy) .origin(origin); let runtime = unsafe { let task_source = NetworkingTaskSource( Box::new(WorkerThreadWorkerChan { sender: own_sender.clone(), worker: worker.clone(), }), pipeline_id, ); new_child_runtime(parent, Some(task_source)) }; let context_for_interrupt = ContextForRequestInterrupt::new(runtime.cx()); let _ = context_sender.send(context_for_interrupt.clone()); let (devtools_mpsc_chan, devtools_mpsc_port) = unbounded(); ROUTER.route_ipc_receiver_to_crossbeam_sender( from_devtools_receiver, devtools_mpsc_chan, ); // Step 8 "Set up a worker environment settings object [...]" // // // // > The origin: Return a unique opaque origin if `worker global // > scope`'s url's scheme is "data", and `inherited origin` // > otherwise. if worker_url.scheme() == "data" { init.origin = ImmutableOrigin::new_opaque(); } let global = DedicatedWorkerGlobalScope::new( init, DOMString::from_string(worker_name), worker_type, worker_url, devtools_mpsc_port, runtime, parent_sender.clone(), own_sender, receiver, closing, image_cache, browsing_context, gpu_id_hub, control_receiver, ); // FIXME(njn): workers currently don't have a unique ID suitable for using in reporter // registration (#6631), so we instead use a random number and cross our fingers. let scope = global.upcast::(); let global_scope = global.upcast::(); global_scope.set_https_state(current_global_https_state); let (metadata, bytes) = match load_whole_resource( request, &global_scope.resource_threads().sender(), global_scope, ) { Err(_) => { println!("error loading script {}", serialized_worker_url); parent_sender .send(CommonScriptMsg::Task( WorkerEvent, Box::new(SimpleWorkerErrorHandler::new(worker)), Some(pipeline_id), TaskSourceName::DOMManipulation, )) .unwrap(); scope.clear_js_runtime(context_for_interrupt); return; }, Ok((metadata, bytes)) => (metadata, bytes), }; scope.set_url(metadata.final_url); global_scope.set_https_state(metadata.https_state); let source = String::from_utf8_lossy(&bytes); unsafe { // Handle interrupt requests JS_AddInterruptCallback(*scope.get_cx(), Some(interrupt_callback)); } if scope.is_closing() { scope.clear_js_runtime(context_for_interrupt); return; } { let _ar = AutoWorkerReset::new(&global, worker.clone()); let _ac = enter_realm(scope); scope.execute_script(DOMString::from(source)); } let reporter_name = format!("dedicated-worker-reporter-{}", random::()); scope .upcast::() .mem_profiler_chan() .run_with_memory_reporting( || { // Step 27, Run the responsible event loop specified // by inside settings until it is destroyed. // The worker processing model remains on this step // until the event loop is destroyed, // which happens after the closing flag is set to true. while !scope.is_closing() { run_worker_event_loop(&*global, Some(&worker)); } }, reporter_name, parent_sender, CommonScriptMsg::CollectReports, ); scope.clear_js_runtime(context_for_interrupt); }) .expect("Thread spawning failed") } pub fn image_cache(&self) -> Arc { self.image_cache.clone() } pub fn script_chan(&self) -> Box { Box::new(WorkerThreadWorkerChan { sender: self.own_sender.clone(), worker: self.worker.borrow().as_ref().unwrap().clone(), }) } pub fn new_script_pair(&self) -> (Box, Box) { let (tx, rx) = unbounded(); let chan = Box::new(SendableWorkerScriptChan { sender: tx, worker: self.worker.borrow().as_ref().unwrap().clone(), }); (chan, Box::new(rx)) } fn handle_script_event(&self, msg: WorkerScriptMsg) { match msg { WorkerScriptMsg::DOMMessage { origin, data } => { let scope = self.upcast::(); let target = self.upcast(); let _ac = enter_realm(self); rooted!(in(*scope.get_cx()) let mut message = UndefinedValue()); if let Ok(ports) = structuredclone::read(scope.upcast(), data, message.handle_mut()) { MessageEvent::dispatch_jsval( target, scope.upcast(), message.handle(), Some(&origin.ascii_serialization()), None, ports, ); } else { MessageEvent::dispatch_error(target, scope.upcast()); } }, WorkerScriptMsg::Common(msg) => { self.upcast::().process_event(msg); }, } } fn handle_mixed_message(&self, msg: MixedMessage) -> bool { // FIXME(#26324): `self.worker` is None in devtools messages. match msg { MixedMessage::Devtools(msg) => match msg { DevtoolScriptControlMsg::EvaluateJS(_pipe_id, string, sender) => { devtools::handle_evaluate_js(self.upcast(), string, sender) }, DevtoolScriptControlMsg::WantsLiveNotifications(_pipe_id, bool_val) => { devtools::handle_wants_live_notifications(self.upcast(), bool_val) }, _ => debug!("got an unusable devtools control message inside the worker!"), }, MixedMessage::Worker(DedicatedWorkerScriptMsg::CommonWorker(linked_worker, msg)) => { let _ar = AutoWorkerReset::new(self, linked_worker); self.handle_script_event(msg); }, MixedMessage::Worker(DedicatedWorkerScriptMsg::WakeUp) => {}, MixedMessage::Control(DedicatedWorkerControlMsg::Exit) => { return false; }, } true } // https://html.spec.whatwg.org/multipage/#runtime-script-errors-2 #[allow(unsafe_code)] pub fn forward_error_to_worker_object(&self, error_info: ErrorInfo) { let worker = self.worker.borrow().as_ref().unwrap().clone(); let pipeline_id = self.upcast::().pipeline_id(); let task = Box::new(task!(forward_error_to_worker_object: move || { let worker = worker.root(); let global = worker.global(); // Step 1. let event = ErrorEvent::new( &global, atom!("error"), EventBubbles::DoesNotBubble, EventCancelable::Cancelable, error_info.message.as_str().into(), error_info.filename.as_str().into(), error_info.lineno, error_info.column, HandleValue::null(), ); let event_status = event.upcast::().fire(worker.upcast::()); // Step 2. if event_status == EventStatus::NotCanceled { global.report_an_error(error_info, HandleValue::null()); } })); self.parent_sender .send(CommonScriptMsg::Task( WorkerEvent, task, Some(pipeline_id), TaskSourceName::DOMManipulation, )) .unwrap(); } // https://html.spec.whatwg.org/multipage/#dom-dedicatedworkerglobalscope-postmessage fn post_message_impl( &self, cx: SafeJSContext, message: HandleValue, transfer: CustomAutoRooterGuard>, ) -> ErrorResult { let data = structuredclone::write(cx, message, Some(transfer))?; let worker = self.worker.borrow().as_ref().unwrap().clone(); let global_scope = self.upcast::(); let pipeline_id = global_scope.pipeline_id(); let task = Box::new(task!(post_worker_message: move || { Worker::handle_message(worker, data); })); self.parent_sender .send(CommonScriptMsg::Task( WorkerEvent, task, Some(pipeline_id), TaskSourceName::DOMManipulation, )) .expect("Sending to parent failed"); Ok(()) } pub(crate) fn browsing_context(&self) -> Option { self.browsing_context } } #[allow(unsafe_code)] unsafe extern "C" fn interrupt_callback(cx: *mut JSContext) -> bool { let in_realm_proof = AlreadyInRealm::assert_for_cx(SafeJSContext::from_ptr(cx)); let global = GlobalScope::from_context(cx, InRealm::Already(&in_realm_proof)); let worker = DomRoot::downcast::(global).expect("global is not a worker scope"); assert!(worker.is::()); // A false response causes the script to terminate !worker.is_closing() } impl DedicatedWorkerGlobalScopeMethods for DedicatedWorkerGlobalScope { /// fn PostMessage( &self, cx: SafeJSContext, message: HandleValue, transfer: CustomAutoRooterGuard>, ) -> ErrorResult { self.post_message_impl(cx, message, transfer) } /// fn PostMessage_( &self, cx: SafeJSContext, message: HandleValue, options: RootedTraceableBox, ) -> ErrorResult { let mut rooted = CustomAutoRooter::new( options .transfer .iter() .map(|js: &RootedTraceableBox>| js.get()) .collect(), ); let guard = CustomAutoRooterGuard::new(*cx, &mut rooted); self.post_message_impl(cx, message, guard) } // https://html.spec.whatwg.org/multipage/#dom-dedicatedworkerglobalscope-close fn Close(&self) { // Step 2 self.upcast::().close(); } // https://html.spec.whatwg.org/multipage/#handler-dedicatedworkerglobalscope-onmessage event_handler!(message, GetOnmessage, SetOnmessage); }