/* 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/. */ //! Utilities for tracing JS-managed values. //! //! The lifetime of DOM objects is managed by the SpiderMonkey Garbage //! Collector. A rooted DOM object implementing the interface `Foo` is traced //! as follows: //! //! 1. The GC calls `_trace` defined in `FooBinding` during the marking //! phase. (This happens through `JSClass.trace` for non-proxy bindings, and //! through `ProxyTraps.trace` otherwise.) //! 2. `_trace` calls `Foo::trace()` (an implementation of `JSTraceable`). //! This is typically derived via a `#[dom_struct]` //! (implies `#[derive(JSTraceable)]`) annotation. //! Non-JS-managed types have an empty inline `trace()` method, //! achieved via `unsafe_no_jsmanaged_fields!` or similar. //! 3. For all fields, `Foo::trace()` //! calls `trace()` on the field. //! For example, for fields of type `Dom`, `Dom::trace()` calls //! `trace_reflector()`. //! 4. `trace_reflector()` calls `Dom::TraceEdge()` with a //! pointer to the `JSObject` for the reflector. This notifies the GC, which //! will add the object to the graph, and will trace that object as well. //! 5. When the GC finishes tracing, it [`finalizes`](../index.html#destruction) //! any reflectors that were not reachable. //! //! The `unsafe_no_jsmanaged_fields!()` macro adds an empty implementation of //! `JSTraceable` to a datatype. use crate::dom::bindings::cell::DomRefCell; use crate::dom::bindings::error::Error; use crate::dom::bindings::refcounted::{Trusted, TrustedPromise}; use crate::dom::bindings::reflector::{DomObject, Reflector}; use crate::dom::bindings::root::{Dom, DomRoot}; use crate::dom::bindings::str::{DOMString, USVString}; use crate::dom::bindings::utils::WindowProxyHandler; use crate::dom::gpubuffer::GPUBufferState; use crate::dom::gpucanvascontext::WebGPUContextId; use crate::dom::gpucommandencoder::GPUCommandEncoderState; use crate::dom::htmlimageelement::SourceSet; use crate::dom::htmlmediaelement::HTMLMediaElementFetchContext; use crate::script_runtime::{ContextForRequestInterrupt, StreamConsumer}; use crate::script_thread::IncompleteParserContexts; use crate::task::TaskBox; use indexmap::IndexMap; use js::glue::{CallObjectTracer, CallScriptTracer, CallStringTracer, CallValueTracer}; use js::jsapi::{ GCTraceKindToAscii, Heap, JSObject, JSScript, JSString, JSTracer, JobQueue, TraceKind, }; use js::jsval::JSVal; use js::rust::{GCMethods, Handle, Runtime, Stencil}; use js::typedarray::TypedArray; use js::typedarray::TypedArrayElement; use malloc_size_of::{MallocSizeOf, MallocSizeOfOps}; use parking_lot::RwLock; use servo_arc::Arc as ServoArc; use smallvec::SmallVec; use std::borrow::Cow; use std::cell::{Cell, RefCell, UnsafeCell}; use std::collections::hash_map::RandomState; use std::collections::{BTreeMap, HashMap, HashSet, VecDeque}; use std::fmt::Display; use std::hash::{BuildHasher, Hash}; use std::mem; use std::num::NonZeroU64; use std::ops::{Deref, DerefMut, Range}; use std::path::PathBuf; use std::rc::Rc; use std::sync::atomic::{AtomicBool, AtomicUsize}; use std::sync::Arc; use std::thread::JoinHandle; use std::time::{Instant, SystemTime}; use style::author_styles::AuthorStyles; use style::stylesheet_set::{AuthorStylesheetSet, DocumentStylesheetSet}; use tendril::fmt::UTF8; use tendril::stream::LossyDecoder; use tendril::TendrilSink; use webxr_api::{Finger, Hand}; unsafe_no_jsmanaged_fields!(JoinHandle<()>); /// A trait to allow tracing (only) DOM objects. pub unsafe trait JSTraceable { /// Trace `self`. unsafe fn trace(&self, trc: *mut JSTracer); } /// Wrapper type for nop traceble /// /// SAFETY: Inner type must not impl JSTraceable #[derive(Clone, Copy, Debug, Default, Eq, Hash, Ord, PartialEq, PartialOrd)] #[trace_in_no_trace_lint::must_not_have_traceable] pub struct NoTrace(pub T); impl Display for NoTrace { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { self.0.fmt(f) } } impl From for NoTrace { fn from(item: T) -> Self { Self(item) } } #[allow(unsafe_code)] unsafe impl JSTraceable for NoTrace { #[inline] unsafe fn trace(&self, _: *mut ::js::jsapi::JSTracer) {} } impl MallocSizeOf for NoTrace { fn size_of(&self, ops: &mut MallocSizeOfOps) -> usize { self.0.size_of(ops) } } /// HashMap wrapper, that has non-jsmanaged keys /// /// Not all methods are reexposed, but you can access inner type via .0 #[trace_in_no_trace_lint::must_not_have_traceable(0)] #[derive(Clone, Debug)] pub struct HashMapTracedValues(pub HashMap); impl Default for HashMapTracedValues { fn default() -> Self { Self(Default::default()) } } impl HashMapTracedValues { /// Wrapper for HashMap::new() #[inline] #[must_use] pub fn new() -> HashMapTracedValues { Self(HashMap::new()) } } impl HashMapTracedValues { #[inline] pub fn iter(&self) -> std::collections::hash_map::Iter<'_, K, V> { self.0.iter() } #[inline] pub fn drain(&mut self) -> std::collections::hash_map::Drain<'_, K, V> { self.0.drain() } #[inline] pub fn is_empty(&self) -> bool { self.0.is_empty() } } impl HashMapTracedValues where K: Eq + Hash, S: BuildHasher, { #[inline] pub fn insert(&mut self, k: K, v: V) -> Option { self.0.insert(k, v) } #[inline] pub fn get(&self, k: &Q) -> Option<&V> where K: std::borrow::Borrow, Q: Hash + Eq, { self.0.get(k) } #[inline] pub fn get_mut(&mut self, k: &Q) -> Option<&mut V> where K: std::borrow::Borrow, Q: Hash + Eq, { self.0.get_mut(k) } #[inline] pub fn contains_key(&self, k: &Q) -> bool where K: std::borrow::Borrow, Q: Hash + Eq, { self.0.contains_key(k) } #[inline] pub fn remove(&mut self, k: &Q) -> Option where K: std::borrow::Borrow, Q: Hash + Eq, { self.0.remove(k) } #[inline] pub fn entry(&mut self, key: K) -> std::collections::hash_map::Entry<'_, K, V> { self.0.entry(key) } } impl MallocSizeOf for HashMapTracedValues where K: Eq + Hash + MallocSizeOf, V: MallocSizeOf, S: BuildHasher, { fn size_of(&self, ops: &mut MallocSizeOfOps) -> usize { self.0.size_of(ops) } } #[allow(unsafe_code)] unsafe impl JSTraceable for HashMapTracedValues { #[inline] unsafe fn trace(&self, trc: *mut ::js::jsapi::JSTracer) { for v in self.0.values() { v.trace(trc); } } } unsafe_no_jsmanaged_fields!(Box); unsafe_no_jsmanaged_fields!(IncompleteParserContexts); unsafe_no_jsmanaged_fields!(Reflector); unsafe_no_jsmanaged_fields!(*mut JobQueue); unsafe_no_jsmanaged_fields!(Cow<'static, str>); /// Trace a `JSScript`. pub fn trace_script(tracer: *mut JSTracer, description: &str, script: &Heap<*mut JSScript>) { unsafe { trace!("tracing {}", description); CallScriptTracer( tracer, script.ptr.get() as *mut _, GCTraceKindToAscii(TraceKind::Script), ); } } /// Trace a `JSVal`. pub fn trace_jsval(tracer: *mut JSTracer, description: &str, val: &Heap) { unsafe { if !val.get().is_markable() { return; } trace!("tracing value {}", description); CallValueTracer( tracer, val.ptr.get() as *mut _, GCTraceKindToAscii(val.get().trace_kind()), ); } } /// Trace the `JSObject` held by `reflector`. #[allow(unrooted_must_root)] pub fn trace_reflector(tracer: *mut JSTracer, description: &str, reflector: &Reflector) { trace!("tracing reflector {}", description); trace_object(tracer, description, reflector.rootable()) } /// Trace a `JSObject`. pub fn trace_object(tracer: *mut JSTracer, description: &str, obj: &Heap<*mut JSObject>) { unsafe { trace!("tracing {}", description); CallObjectTracer( tracer, obj.ptr.get() as *mut _, GCTraceKindToAscii(TraceKind::Object), ); } } /// Trace a `JSString`. pub fn trace_string(tracer: *mut JSTracer, description: &str, s: &Heap<*mut JSString>) { unsafe { trace!("tracing {}", description); CallStringTracer( tracer, s.ptr.get() as *mut _, GCTraceKindToAscii(TraceKind::String), ); } } unsafe impl JSTraceable for Rc { unsafe fn trace(&self, trc: *mut JSTracer) { (**self).trace(trc) } } unsafe impl JSTraceable for Arc { unsafe fn trace(&self, trc: *mut JSTracer) { (**self).trace(trc) } } unsafe impl JSTraceable for ServoArc { unsafe fn trace(&self, trc: *mut JSTracer) { (**self).trace(trc) } } unsafe impl JSTraceable for RwLock { unsafe fn trace(&self, trc: *mut JSTracer) { self.read().trace(trc) } } unsafe impl JSTraceable for Box { unsafe fn trace(&self, trc: *mut JSTracer) { (**self).trace(trc) } } unsafe impl JSTraceable for [T] { unsafe fn trace(&self, trc: *mut JSTracer) { for e in self.iter() { e.trace(trc); } } } unsafe impl JSTraceable for Cell { unsafe fn trace(&self, trc: *mut JSTracer) { self.get().trace(trc) } } unsafe impl JSTraceable for UnsafeCell { unsafe fn trace(&self, trc: *mut JSTracer) { (*self.get()).trace(trc) } } unsafe impl JSTraceable for DomRefCell { unsafe fn trace(&self, trc: *mut JSTracer) { (*self).borrow().trace(trc) } } unsafe impl JSTraceable for RefCell { unsafe fn trace(&self, trc: *mut JSTracer) { (*self).borrow().trace(trc) } } unsafe impl JSTraceable for Heap<*mut JSScript> { unsafe fn trace(&self, trc: *mut JSTracer) { if self.get().is_null() { return; } trace_script(trc, "heap script", self); } } unsafe impl JSTraceable for Heap<*mut JSObject> { unsafe fn trace(&self, trc: *mut JSTracer) { if self.get().is_null() { return; } trace_object(trc, "heap object", self); } } unsafe impl JSTraceable for Heap<*mut JSString> { unsafe fn trace(&self, trc: *mut JSTracer) { if self.get().is_null() { return; } trace_string(trc, "heap string", self); } } unsafe impl JSTraceable for Heap { unsafe fn trace(&self, trc: *mut JSTracer) { trace_jsval(trc, "heap value", self); } } // XXXManishearth Check if the following three are optimized to no-ops // if e.trace() is a no-op (e.g it is an unsafe_no_jsmanaged_fields type) unsafe impl JSTraceable for Vec { #[inline] unsafe fn trace(&self, trc: *mut JSTracer) { for e in &*self { e.trace(trc); } } } unsafe impl JSTraceable for VecDeque { #[inline] unsafe fn trace(&self, trc: *mut JSTracer) { for e in &*self { e.trace(trc); } } } unsafe impl JSTraceable for indexmap::IndexSet { #[inline] unsafe fn trace(&self, trc: *mut JSTracer) { for e in self.iter() { e.trace(trc); } } } unsafe impl JSTraceable for (A, B, C, D) where A: JSTraceable, B: JSTraceable, C: JSTraceable, D: JSTraceable, { unsafe fn trace(&self, trc: *mut JSTracer) { self.0.trace(trc); self.1.trace(trc); self.2.trace(trc); self.3.trace(trc); } } // XXXManishearth Check if the following three are optimized to no-ops // if e.trace() is a no-op (e.g it is an unsafe_no_jsmanaged_fields type) unsafe impl JSTraceable for SmallVec<[T; 1]> { #[inline] unsafe fn trace(&self, trc: *mut JSTracer) { for e in self.iter() { e.trace(trc); } } } unsafe impl JSTraceable for Option { #[inline] unsafe fn trace(&self, trc: *mut JSTracer) { self.as_ref().map(|e| e.trace(trc)); } } unsafe impl JSTraceable for Result { #[inline] unsafe fn trace(&self, trc: *mut JSTracer) { match *self { Ok(ref inner) => inner.trace(trc), Err(ref inner) => inner.trace(trc), } } } unsafe impl JSTraceable for HashMap where K: Hash + Eq + JSTraceable, V: JSTraceable, S: BuildHasher, { #[inline] unsafe fn trace(&self, trc: *mut JSTracer) { for (k, v) in &*self { k.trace(trc); v.trace(trc); } } } unsafe impl JSTraceable for HashSet where T: Hash + Eq + JSTraceable, S: BuildHasher, { #[inline] unsafe fn trace(&self, trc: *mut JSTracer) { for v in &*self { v.trace(trc); } } } unsafe impl JSTraceable for BTreeMap { #[inline] unsafe fn trace(&self, trc: *mut JSTracer) { for (k, v) in self { k.trace(trc); v.trace(trc); } } } unsafe impl JSTraceable for IndexMap where K: Hash + Eq + JSTraceable, V: JSTraceable, S: BuildHasher, { #[inline] unsafe fn trace(&self, trc: *mut JSTracer) { for (k, v) in &*self { k.trace(trc); v.trace(trc); } } } unsafe impl JSTraceable for (A, B) { #[inline] unsafe fn trace(&self, trc: *mut JSTracer) { let (ref a, ref b) = *self; a.trace(trc); b.trace(trc); } } unsafe impl JSTraceable for (A, B, C) { #[inline] unsafe fn trace(&self, trc: *mut JSTracer) { let (ref a, ref b, ref c) = *self; a.trace(trc); b.trace(trc); c.trace(trc); } } unsafe_no_jsmanaged_fields!(bool, f32, f64, String, AtomicBool, AtomicUsize, char); unsafe_no_jsmanaged_fields!(usize, u8, u16, u32, u64); unsafe_no_jsmanaged_fields!(isize, i8, i16, i32, i64); unsafe_no_jsmanaged_fields!(NonZeroU64); unsafe_no_jsmanaged_fields!(Error); unsafe_no_jsmanaged_fields!(TrustedPromise); unsafe_no_jsmanaged_fields!(Runtime); unsafe_no_jsmanaged_fields!(ContextForRequestInterrupt); unsafe_no_jsmanaged_fields!(WindowProxyHandler); unsafe_no_jsmanaged_fields!(DOMString); unsafe_no_jsmanaged_fields!(USVString); unsafe_no_jsmanaged_fields!(SystemTime); unsafe_no_jsmanaged_fields!(Instant); unsafe_no_jsmanaged_fields!(PathBuf); unsafe_no_jsmanaged_fields!(WebGPUContextId); unsafe_no_jsmanaged_fields!(GPUBufferState); unsafe_no_jsmanaged_fields!(GPUCommandEncoderState); unsafe_no_jsmanaged_fields!(Range); unsafe_no_jsmanaged_fields!(SourceSet); unsafe_no_jsmanaged_fields!(HTMLMediaElementFetchContext); unsafe_no_jsmanaged_fields!(StreamConsumer); unsafe_no_jsmanaged_fields!(Stencil); unsafe impl<'a> JSTraceable for &'a str { #[inline] unsafe fn trace(&self, _: *mut JSTracer) { // Do nothing } } unsafe impl JSTraceable for fn(A) -> B { #[inline] unsafe fn trace(&self, _: *mut JSTracer) { // Do nothing } } unsafe impl JSTraceable for () { #[inline] unsafe fn trace(&self, _: *mut JSTracer) { // Do nothing } } unsafe impl JSTraceable for Trusted { #[inline] unsafe fn trace(&self, _: *mut JSTracer) { // Do nothing } } unsafe impl JSTraceable for TypedArray>> where T: TypedArrayElement, { unsafe fn trace(&self, trc: *mut JSTracer) { self.underlying_object().trace(trc); } } unsafe impl JSTraceable for DocumentStylesheetSet where S: JSTraceable + ::style::stylesheets::StylesheetInDocument + PartialEq + 'static, { unsafe fn trace(&self, tracer: *mut JSTracer) { for (s, _origin) in self.iter() { s.trace(tracer) } } } unsafe impl JSTraceable for AuthorStylesheetSet where S: JSTraceable + ::style::stylesheets::StylesheetInDocument + PartialEq + 'static, { unsafe fn trace(&self, tracer: *mut JSTracer) { for s in self.iter() { s.trace(tracer) } } } unsafe impl JSTraceable for AuthorStyles where S: JSTraceable + ::style::stylesheets::StylesheetInDocument + PartialEq + 'static, { unsafe fn trace(&self, tracer: *mut JSTracer) { self.stylesheets.trace(tracer) } } unsafe impl JSTraceable for LossyDecoder where Sink: JSTraceable + TendrilSink, { unsafe fn trace(&self, tracer: *mut JSTracer) { self.inner_sink().trace(tracer); } } unsafe impl JSTraceable for Hand where J: JSTraceable, { #[inline] unsafe fn trace(&self, trc: *mut JSTracer) { // exhaustive match so we don't miss new fields let Hand { ref wrist, ref thumb_metacarpal, ref thumb_phalanx_proximal, ref thumb_phalanx_distal, ref thumb_phalanx_tip, ref index, ref middle, ref ring, ref little, } = *self; wrist.trace(trc); thumb_metacarpal.trace(trc); thumb_phalanx_proximal.trace(trc); thumb_phalanx_distal.trace(trc); thumb_phalanx_tip.trace(trc); index.trace(trc); middle.trace(trc); ring.trace(trc); little.trace(trc); } } unsafe impl JSTraceable for Finger where J: JSTraceable, { #[inline] unsafe fn trace(&self, trc: *mut JSTracer) { // exhaustive match so we don't miss new fields let Finger { ref metacarpal, ref phalanx_proximal, ref phalanx_intermediate, ref phalanx_distal, ref phalanx_tip, } = *self; metacarpal.trace(trc); phalanx_proximal.trace(trc); phalanx_intermediate.trace(trc); phalanx_distal.trace(trc); phalanx_tip.trace(trc); } } /// Holds a set of JSTraceables that need to be rooted struct RootedTraceableSet { set: Vec<*const dyn JSTraceable>, } thread_local!( /// TLV Holds a set of JSTraceables that need to be rooted static ROOTED_TRACEABLES: RefCell = RefCell::new(RootedTraceableSet::new()); ); impl RootedTraceableSet { fn new() -> RootedTraceableSet { RootedTraceableSet { set: vec![] } } unsafe fn remove(traceable: *const dyn JSTraceable) { ROOTED_TRACEABLES.with(|ref traceables| { let mut traceables = traceables.borrow_mut(); let idx = match traceables .set .iter() .rposition(|x| *x as *const () == traceable as *const ()) { Some(idx) => idx, None => unreachable!(), }; traceables.set.remove(idx); }); } unsafe fn add(traceable: *const dyn JSTraceable) { ROOTED_TRACEABLES.with(|ref traceables| { traceables.borrow_mut().set.push(traceable); }) } unsafe fn trace(&self, tracer: *mut JSTracer) { for traceable in &self.set { (**traceable).trace(tracer); } } } /// Roots any JSTraceable thing /// /// If you have a valid DomObject, use DomRoot. /// If you have GC things like *mut JSObject or JSVal, use rooted!. /// If you have an arbitrary number of DomObjects to root, use rooted_vec!. /// If you know what you're doing, use this. #[unrooted_must_root_lint::allow_unrooted_interior] pub struct RootedTraceableBox { ptr: *mut T, } unsafe impl JSTraceable for RootedTraceableBox { unsafe fn trace(&self, tracer: *mut JSTracer) { (*self.ptr).trace(tracer); } } impl RootedTraceableBox { /// DomRoot a JSTraceable thing for the life of this RootedTraceableBox pub fn new(traceable: T) -> RootedTraceableBox { Self::from_box(Box::new(traceable)) } /// Consumes a boxed JSTraceable and roots it for the life of this RootedTraceableBox. pub fn from_box(boxed_traceable: Box) -> RootedTraceableBox { let traceable = Box::into_raw(boxed_traceable); unsafe { RootedTraceableSet::add(traceable); } RootedTraceableBox { ptr: traceable } } } impl RootedTraceableBox> where Heap: JSTraceable + 'static, T: GCMethods + Copy, { pub fn handle(&self) -> Handle { unsafe { Handle::from_raw((*self.ptr).handle()) } } } impl MallocSizeOf for RootedTraceableBox { fn size_of(&self, ops: &mut MallocSizeOfOps) -> usize { // Briefly resurrect the real Box value so we can rely on the existing calculations. // Then immediately forget about it again to avoid dropping the box. let inner = unsafe { Box::from_raw(self.ptr) }; let size = inner.size_of(ops); mem::forget(inner); size } } impl Default for RootedTraceableBox { fn default() -> RootedTraceableBox { RootedTraceableBox::new(T::default()) } } impl Deref for RootedTraceableBox { type Target = T; fn deref(&self) -> &T { unsafe { &*self.ptr } } } impl DerefMut for RootedTraceableBox { fn deref_mut(&mut self) -> &mut T { unsafe { &mut *self.ptr } } } impl Drop for RootedTraceableBox { fn drop(&mut self) { unsafe { RootedTraceableSet::remove(self.ptr); let _ = Box::from_raw(self.ptr); } } } /// A vector of items to be rooted with `RootedVec`. /// Guaranteed to be empty when not rooted. /// Usage: `rooted_vec!(let mut v);` or if you have an /// iterator of `DomRoot`s, `rooted_vec!(let v <- iterator);`. #[allow(unrooted_must_root)] #[derive(JSTraceable)] #[unrooted_must_root_lint::allow_unrooted_interior] pub struct RootableVec { v: Vec, } impl RootableVec { /// Create a vector of items of type T that can be rooted later. pub fn new_unrooted() -> RootableVec { RootableVec { v: vec![] } } } /// A vector of items that are rooted for the lifetime 'a. #[unrooted_must_root_lint::allow_unrooted_interior] pub struct RootedVec<'a, T: 'static + JSTraceable> { root: &'a mut RootableVec, } impl<'a, T: 'static + JSTraceable> RootedVec<'a, T> { /// Create a vector of items of type T that is rooted for /// the lifetime of this struct pub fn new(root: &'a mut RootableVec) -> Self { unsafe { RootedTraceableSet::add(root); } RootedVec { root: root } } } impl<'a, T: 'static + JSTraceable + DomObject> RootedVec<'a, Dom> { /// Create a vector of items of type Dom that is rooted for /// the lifetime of this struct pub fn from_iter(root: &'a mut RootableVec>, iter: I) -> Self where I: Iterator>, { unsafe { RootedTraceableSet::add(root); } root.v.extend(iter.map(|item| Dom::from_ref(&*item))); RootedVec { root: root } } } impl<'a, T: JSTraceable + 'static> Drop for RootedVec<'a, T> { fn drop(&mut self) { self.clear(); unsafe { RootedTraceableSet::remove(self.root); } } } impl<'a, T: JSTraceable> Deref for RootedVec<'a, T> { type Target = Vec; fn deref(&self) -> &Vec { &self.root.v } } impl<'a, T: JSTraceable> DerefMut for RootedVec<'a, T> { fn deref_mut(&mut self) -> &mut Vec { &mut self.root.v } } /// SM Callback that traces the rooted traceables pub unsafe fn trace_traceables(tracer: *mut JSTracer) { trace!("tracing stack-rooted traceables"); ROOTED_TRACEABLES.with(|ref traceables| { let traceables = traceables.borrow(); traceables.trace(tracer); }); }