use crate::avm1::function::{Executable, NativeFunction}; use crate::avm1::property::{Attribute, Property}; use crate::avm1::return_value::ReturnValue; use crate::avm1::{Avm1, Error, Object, ObjectPtr, TObject, UpdateContext, Value}; use crate::display_object::DisplayObject; use core::fmt; use enumset::EnumSet; use gc_arena::{Collect, GcCell, MutationContext}; use std::collections::hash_map::Entry; use std::collections::{HashMap, HashSet}; pub const TYPE_OF_OBJECT: &str = "object"; pub const TYPE_OF_FUNCTION: &str = "function"; #[derive(Debug, Clone, Collect)] #[collect(no_drop)] pub enum ArrayStorage<'gc> { Vector(Vec>), Properties { length: usize }, } #[derive(Debug, Copy, Clone, Collect)] #[collect(no_drop)] pub struct ScriptObject<'gc>(GcCell<'gc, ScriptObjectData<'gc>>); pub struct ScriptObjectData<'gc> { prototype: Option>, values: HashMap>, function: Option>, interfaces: Vec>, type_of: &'static str, array: ArrayStorage<'gc>, } unsafe impl<'gc> Collect for ScriptObjectData<'gc> { fn trace(&self, cc: gc_arena::CollectionContext) { self.prototype.trace(cc); self.values.trace(cc); self.function.trace(cc); self.array.trace(cc); self.interfaces.trace(cc); } } impl fmt::Debug for ScriptObjectData<'_> { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { f.debug_struct("Object") .field("prototype", &self.prototype) .field("values", &self.values) .field("function", &self.function.is_some()) .field("array", &self.array) .finish() } } impl<'gc> ScriptObject<'gc> { pub fn object( gc_context: MutationContext<'gc, '_>, proto: Option>, ) -> ScriptObject<'gc> { ScriptObject(GcCell::allocate( gc_context, ScriptObjectData { prototype: proto, type_of: TYPE_OF_OBJECT, values: HashMap::new(), function: None, array: ArrayStorage::Properties { length: 0 }, interfaces: vec![], }, )) } pub fn array( gc_context: MutationContext<'gc, '_>, proto: Option>, ) -> ScriptObject<'gc> { let object = ScriptObject(GcCell::allocate( gc_context, ScriptObjectData { prototype: proto, type_of: TYPE_OF_OBJECT, values: HashMap::new(), function: None, array: ArrayStorage::Vector(Vec::new()), interfaces: vec![], }, )); object.sync_native_property("length", gc_context, Some(0.into())); object } /// Constructs and allocates an empty but normal object in one go. pub fn object_cell( gc_context: MutationContext<'gc, '_>, proto: Option>, ) -> Object<'gc> { ScriptObject(GcCell::allocate( gc_context, ScriptObjectData { prototype: proto, type_of: TYPE_OF_OBJECT, values: HashMap::new(), function: None, array: ArrayStorage::Properties { length: 0 }, interfaces: vec![], }, )) .into() } /// Constructs an object with no values, not even builtins. /// /// Intended for constructing scope chains, since they exclusively use the /// object values, but can't just have a hashmap because of `with` and /// friends. pub fn bare_object(gc_context: MutationContext<'gc, '_>) -> Self { ScriptObject(GcCell::allocate( gc_context, ScriptObjectData { prototype: None, type_of: TYPE_OF_OBJECT, values: HashMap::new(), function: None, array: ArrayStorage::Properties { length: 0 }, interfaces: vec![], }, )) } /// Construct a function sans prototype. pub fn bare_function( gc_context: MutationContext<'gc, '_>, function: impl Into>, fn_proto: Option>, ) -> Self { ScriptObject(GcCell::allocate( gc_context, ScriptObjectData { prototype: fn_proto, type_of: TYPE_OF_FUNCTION, function: Some(function.into()), values: HashMap::new(), array: ArrayStorage::Properties { length: 0 }, interfaces: vec![], }, )) } /// Construct a function from an executable and associated protos. /// /// Since prototypes need to link back to themselves, this function builds /// both objects itself and returns the function to you, fully allocated. /// /// `fn_proto` refers to the implicit proto of the function object, and the /// `prototype` refers to the explicit prototype of the function. If /// provided, the function and it's prototype will be linked to each other. pub fn function( gc_context: MutationContext<'gc, '_>, function: impl Into>, fn_proto: Option>, prototype: Option>, ) -> Object<'gc> { let function = Self::bare_function(gc_context, function, fn_proto).into(); //TODO: Can we make these proper sets or no? if let Some(p) = prototype { p.define_value( gc_context, "constructor", Value::Object(function), Attribute::DontEnum.into(), ); function.define_value(gc_context, "prototype", p.into(), EnumSet::empty()); } function } /// Declare a native function on the current object. /// /// This is intended for use with defining host object prototypes. Notably, /// this creates a function object without an explicit `prototype`, which /// is only possible when defining host functions. User-defined functions /// always get a fresh explicit prototype, so you should never force set a /// user-defined function. pub fn force_set_function( &mut self, name: &str, function: NativeFunction<'gc>, gc_context: MutationContext<'gc, '_>, attributes: A, fn_proto: Option>, ) where A: Into>, { self.define_value( gc_context, name, Value::Object(ScriptObject::function(gc_context, function, fn_proto, None)), attributes.into(), ) } pub fn set_type_of(&mut self, gc_context: MutationContext<'gc, '_>, type_of: &'static str) { self.0.write(gc_context).type_of = type_of; } #[allow(clippy::trivially_copy_pass_by_ref)] pub fn sync_native_property( &self, name: &str, gc_context: MutationContext<'gc, '_>, native_value: Option>, ) { match self.0.write(gc_context).values.entry(name.to_string()) { Entry::Occupied(mut entry) => { if let Property::Stored { value, .. } = entry.get_mut() { match native_value { None => { entry.remove_entry(); } Some(native_value) => { *value = native_value; } } } } Entry::Vacant(entry) => { if let Some(native_value) = native_value { entry.insert(Property::Stored { value: native_value, attributes: Attribute::DontEnum.into(), }); } } } } #[allow(clippy::trivially_copy_pass_by_ref)] pub(crate) fn internal_set( &self, name: &str, value: Value<'gc>, avm: &mut Avm1<'gc>, context: &mut UpdateContext<'_, 'gc, '_>, this: Object<'gc>, ) -> Result<(), Error> { if name == "__proto__" { self.0.write(context.gc_context).prototype = value.as_object().ok(); } else if let Ok(index) = name.parse::() { self.set_array_element(index, value.to_owned(), context.gc_context); } else { if name == "length" { let length = value .as_number(avm, context) .map(|v| v.abs() as i32) .unwrap_or(0); if length > 0 { self.set_length(context.gc_context, length as usize); } else { self.set_length(context.gc_context, 0); } } match self .0 .write(context.gc_context) .values .entry(name.to_owned()) { Entry::Occupied(mut entry) => { entry.get_mut().set(avm, context, this, value)?; } Entry::Vacant(entry) => { entry.insert(Property::Stored { value, attributes: Default::default(), }); } } } Ok(()) } } impl<'gc> TObject<'gc> for ScriptObject<'gc> { /// Get the value of a particular property on this object. /// /// The `avm`, `context`, and `this` parameters exist so that this object /// can call virtual properties. Furthermore, since some virtual properties /// may resolve on the AVM stack, this function may return `None` instead /// of a `Value`. *This is not equivalent to `undefined`.* Instead, it is a /// signal that your value will be returned on the ActionScript stack, and /// that you should register a stack continuation in order to get it. fn get_local( &self, name: &str, avm: &mut Avm1<'gc>, context: &mut UpdateContext<'_, 'gc, '_>, this: Object<'gc>, ) -> Result, Error> { if name == "__proto__" { return Ok(self.proto().map_or(Value::Undefined, Value::Object).into()); } if let Some(value) = self.0.read().values.get(name) { return value.get(avm, context, this); } Ok(Value::Undefined.into()) } /// Set a named property on the object. /// /// This function takes a redundant `this` parameter which should be /// the object's own `GcCell`, so that it can pass it to user-defined /// overrides that may need to interact with the underlying object. fn set( &self, name: &str, value: Value<'gc>, avm: &mut Avm1<'gc>, context: &mut UpdateContext<'_, 'gc, '_>, ) -> Result<(), Error> { self.internal_set(name, value, avm, context, (*self).into()) } /// Call the underlying object. /// /// This function takes a redundant `this` parameter which should be /// the object's own `GcCell`, so that it can pass it to user-defined /// overrides that may need to interact with the underlying object. fn call( &self, avm: &mut Avm1<'gc>, context: &mut UpdateContext<'_, 'gc, '_>, this: Object<'gc>, args: &[Value<'gc>], ) -> Result, Error> { if let Some(function) = &self.0.read().function { function.exec(avm, context, this, args) } else { Ok(Value::Undefined.into()) } } #[allow(clippy::new_ret_no_self)] fn new( &self, _avm: &mut Avm1<'gc>, context: &mut UpdateContext<'_, 'gc, '_>, this: Object<'gc>, _args: &[Value<'gc>], ) -> Result, Error> { match self.0.read().array { ArrayStorage::Vector(_) => { Ok(ScriptObject::array(context.gc_context, Some(this)).into()) } ArrayStorage::Properties { .. } => { Ok(ScriptObject::object(context.gc_context, Some(this)).into()) } } } /// Delete a named property from the object. /// /// Returns false if the property cannot be deleted. fn delete(&self, gc_context: MutationContext<'gc, '_>, name: &str) -> bool { let mut object = self.0.write(gc_context); if let Some(prop) = object.values.get(name) { if prop.can_delete() { object.values.remove(name); return true; } } false } fn add_property( &self, gc_context: MutationContext<'gc, '_>, name: &str, get: Executable<'gc>, set: Option>, attributes: EnumSet, ) { self.0.write(gc_context).values.insert( name.to_owned(), Property::Virtual { get, set, attributes, }, ); } fn define_value( &self, gc_context: MutationContext<'gc, '_>, name: &str, value: Value<'gc>, attributes: EnumSet, ) { self.0 .write(gc_context) .values .insert(name.to_string(), Property::Stored { value, attributes }); } fn set_attributes( &mut self, gc_context: MutationContext<'gc, '_>, name: Option<&str>, set_attributes: EnumSet, clear_attributes: EnumSet, ) { match name { None => { // Change *all* attributes. for (_name, prop) in self.0.write(gc_context).values.iter_mut() { let new_atts = (prop.attributes() - clear_attributes) | set_attributes; prop.set_attributes(new_atts); } } Some(name) => { if let Some(prop) = self.0.write(gc_context).values.get_mut(name) { let new_atts = (prop.attributes() - clear_attributes) | set_attributes; prop.set_attributes(new_atts); } } } } fn proto(&self) -> Option> { self.0.read().prototype } /// Checks if the object has a given named property. fn has_property(&self, name: &str) -> bool { self.has_own_property(name) || self .proto() .as_ref() .map_or(false, |p| p.has_property(name)) } /// Checks if the object has a given named property on itself (and not, /// say, the object's prototype or superclass) fn has_own_property(&self, name: &str) -> bool { if name == "__proto__" { return true; } self.0.read().values.contains_key(name) } fn is_property_overwritable(&self, name: &str) -> bool { self.0 .read() .values .get(name) .map(|p| p.is_overwritable()) .unwrap_or(false) } /// Checks if a named property appears when enumerating the object. fn is_property_enumerable(&self, name: &str) -> bool { if let Some(prop) = self.0.read().values.get(name) { prop.is_enumerable() } else { false } } /// Enumerate the object. fn get_keys(&self) -> HashSet { let mut result = self.proto().map_or_else(HashSet::new, |p| p.get_keys()); self.0 .read() .values .iter() .filter_map(|(k, p)| { if p.is_enumerable() { Some(k.to_string()) } else { None } }) .for_each(|k| { result.insert(k); }); result } fn as_string(&self) -> String { if self.0.read().function.is_some() { "[type Function]".to_string() } else { "[object Object]".to_string() } } fn type_of(&self) -> &'static str { self.0.read().type_of } fn interfaces(&self) -> Vec> { self.0.read().interfaces.clone() } fn set_interfaces(&mut self, context: MutationContext<'gc, '_>, iface_list: Vec>) { self.0.write(context).interfaces = iface_list; } fn as_script_object(&self) -> Option> { Some(*self) } /// Get the underlying display node for this object, if it exists. fn as_display_object(&self) -> Option> { None } /// Returns a copy of a given function. /// /// TODO: We have to clone here because of how executables are stored on /// objects directly. This might not be a good idea for performance. fn as_executable(&self) -> Option> { self.0.read().function.clone() } fn as_ptr(&self) -> *const ObjectPtr { self.0.as_ptr() as *const ObjectPtr } fn length(&self) -> usize { match &self.0.read().array { ArrayStorage::Vector(vector) => vector.len(), ArrayStorage::Properties { length } => *length, } } fn set_length(&self, gc_context: MutationContext<'gc, '_>, new_length: usize) { let mut to_remove = None; match &mut self.0.write(gc_context).array { ArrayStorage::Vector(vector) => { let old_length = vector.len(); vector.resize(new_length, Value::Undefined); if new_length < old_length { to_remove = Some(new_length..old_length); } } ArrayStorage::Properties { length } => { *length = new_length; } } if let Some(to_remove) = to_remove { for i in to_remove { self.sync_native_property(&i.to_string(), gc_context, None); } } self.sync_native_property("length", gc_context, Some(new_length.into())); } fn array(&self) -> Vec> { match &self.0.read().array { ArrayStorage::Vector(vector) => vector.to_owned(), ArrayStorage::Properties { length } => { let mut values = Vec::new(); for i in 0..*length { values.push(self.array_element(i)); } values } } } fn array_element(&self, index: usize) -> Value<'gc> { match &self.0.read().array { ArrayStorage::Vector(vector) => { if let Some(value) = vector.get(index) { value.to_owned() } else { Value::Undefined } } ArrayStorage::Properties { length } => { if index < *length { if let Some(Property::Stored { value, .. }) = self.0.read().values.get(&index.to_string()) { return value.to_owned(); } } Value::Undefined } } } fn set_array_element( &self, index: usize, value: Value<'gc>, gc_context: MutationContext<'gc, '_>, ) -> usize { self.sync_native_property(&index.to_string(), gc_context, Some(value.clone())); let mut adjust_length = false; let length = match &mut self.0.write(gc_context).array { ArrayStorage::Vector(vector) => { if index >= vector.len() { vector.resize(index + 1, Value::Undefined); } vector[index] = value.clone(); adjust_length = true; vector.len() } ArrayStorage::Properties { length } => *length, }; if adjust_length { self.sync_native_property("length", gc_context, Some(length.into())); } length } fn delete_array_element(&self, index: usize, gc_context: MutationContext<'gc, '_>) { if let ArrayStorage::Vector(vector) = &mut self.0.write(gc_context).array { if index < vector.len() { vector[index] = Value::Undefined; } } } } #[cfg(test)] mod tests { use super::*; use crate::avm1::activation::Activation; use crate::avm1::property::Attribute::*; use crate::backend::audio::NullAudioBackend; use crate::backend::input::NullInputBackend; use crate::backend::navigator::NullNavigatorBackend; use crate::backend::render::NullRenderer; use crate::display_object::MovieClip; use crate::library::Library; use crate::prelude::*; use gc_arena::rootless_arena; use rand::{rngs::SmallRng, SeedableRng}; use std::sync::Arc; fn with_object(swf_version: u8, test: F) -> R where F: for<'a, 'gc> FnOnce(&mut Avm1<'gc>, &mut UpdateContext<'a, 'gc, '_>, Object<'gc>) -> R, { rootless_arena(|gc_context| { let mut avm = Avm1::new(gc_context, swf_version); let mut root: DisplayObject<'_> = MovieClip::new(swf_version, gc_context).into(); root.post_instantiation(gc_context, root, avm.prototypes().movie_clip); let mut context = UpdateContext { gc_context, global_time: 0, player_version: 32, swf_version, root, rng: &mut SmallRng::from_seed([0u8; 16]), action_queue: &mut crate::context::ActionQueue::new(), audio: &mut NullAudioBackend::new(), input: &mut NullInputBackend::new(), background_color: &mut Color { r: 0, g: 0, b: 0, a: 0, }, library: &mut Library::new(), navigator: &mut NullNavigatorBackend::new(), renderer: &mut NullRenderer::new(), swf_data: &mut Arc::new(vec![]), system_prototypes: avm.prototypes().clone(), mouse_hovered_object: None, mouse_position: &(Twips::new(0), Twips::new(0)), drag_object: &mut None, stage_size: (Twips::from_pixels(550.0), Twips::from_pixels(400.0)), }; let object = ScriptObject::object(gc_context, Some(avm.prototypes().object)).into(); let globals = avm.global_object_cell(); avm.insert_stack_frame(GcCell::allocate( gc_context, Activation::from_nothing(swf_version, globals, gc_context, root), )); test(&mut avm, &mut context, object) }) } #[test] fn test_get_undefined() { with_object(0, |avm, context, object| { assert_eq!( object.get("not_defined", avm, context).unwrap(), ReturnValue::Immediate(Value::Undefined) ); }) } #[test] fn test_set_get() { with_object(0, |avm, context, object| { object.as_script_object().unwrap().define_value( context.gc_context, "forced", "forced".into(), EnumSet::empty(), ); object .set("natural", "natural".into(), avm, context) .unwrap(); assert_eq!( object.get("forced", avm, context).unwrap(), ReturnValue::Immediate("forced".into()) ); assert_eq!( object.get("natural", avm, context).unwrap(), ReturnValue::Immediate("natural".into()) ); }) } #[test] fn test_set_readonly() { with_object(0, |avm, context, object| { object.as_script_object().unwrap().define_value( context.gc_context, "normal", "initial".into(), EnumSet::empty(), ); object.as_script_object().unwrap().define_value( context.gc_context, "readonly", "initial".into(), ReadOnly.into(), ); object .set("normal", "replaced".into(), avm, context) .unwrap(); object .set("readonly", "replaced".into(), avm, context) .unwrap(); assert_eq!( object.get("normal", avm, context).unwrap(), ReturnValue::Immediate("replaced".into()) ); assert_eq!( object.get("readonly", avm, context).unwrap(), ReturnValue::Immediate("initial".into()) ); }) } #[test] fn test_deletable_not_readonly() { with_object(0, |avm, context, object| { object.as_script_object().unwrap().define_value( context.gc_context, "test", "initial".into(), DontDelete.into(), ); assert_eq!(object.delete(context.gc_context, "test"), false); assert_eq!( object.get("test", avm, context).unwrap(), ReturnValue::Immediate("initial".into()) ); object .as_script_object() .unwrap() .set("test", "replaced".into(), avm, context) .unwrap(); assert_eq!(object.delete(context.gc_context, "test"), false); assert_eq!( object.get("test", avm, context).unwrap(), ReturnValue::Immediate("replaced".into()) ); }) } #[test] fn test_virtual_get() { with_object(0, |avm, context, object| { let getter = Executable::Native(|_avm, _context, _this, _args| { Ok(ReturnValue::Immediate("Virtual!".into())) }); object.as_script_object().unwrap().add_property( context.gc_context, "test", getter, None, EnumSet::empty(), ); assert_eq!( object.get("test", avm, context).unwrap(), ReturnValue::Immediate("Virtual!".into()) ); // This set should do nothing object.set("test", "Ignored!".into(), avm, context).unwrap(); assert_eq!( object.get("test", avm, context).unwrap(), ReturnValue::Immediate("Virtual!".into()) ); }) } #[test] fn test_delete() { with_object(0, |avm, context, object| { let getter = Executable::Native(|_avm, _context, _this, _args| { Ok(ReturnValue::Immediate("Virtual!".into())) }); object.as_script_object().unwrap().add_property( context.gc_context, "virtual", getter.clone(), None, EnumSet::empty(), ); object.as_script_object().unwrap().add_property( context.gc_context, "virtual_un", getter, None, DontDelete.into(), ); object.as_script_object().unwrap().define_value( context.gc_context, "stored", "Stored!".into(), EnumSet::empty(), ); object.as_script_object().unwrap().define_value( context.gc_context, "stored_un", "Stored!".into(), DontDelete.into(), ); assert_eq!(object.delete(context.gc_context, "virtual"), true); assert_eq!(object.delete(context.gc_context, "virtual_un"), false); assert_eq!(object.delete(context.gc_context, "stored"), true); assert_eq!(object.delete(context.gc_context, "stored_un"), false); assert_eq!(object.delete(context.gc_context, "non_existent"), false); assert_eq!( object.get("virtual", avm, context).unwrap(), ReturnValue::Immediate(Value::Undefined) ); assert_eq!( object.get("virtual_un", avm, context).unwrap(), ReturnValue::Immediate("Virtual!".into()) ); assert_eq!( object.get("stored", avm, context).unwrap(), ReturnValue::Immediate(Value::Undefined) ); assert_eq!( object.get("stored_un", avm, context).unwrap(), ReturnValue::Immediate("Stored!".into()) ); }) } #[test] fn test_iter_values() { with_object(0, |_avm, context, object| { let getter = Executable::Native(|_avm, _context, _this, _args| { Ok(ReturnValue::Immediate(Value::Null)) }); object.as_script_object().unwrap().define_value( context.gc_context, "stored", Value::Null, EnumSet::empty(), ); object.as_script_object().unwrap().define_value( context.gc_context, "stored_hidden", Value::Null, DontEnum.into(), ); object.as_script_object().unwrap().add_property( context.gc_context, "virtual", getter.clone(), None, EnumSet::empty(), ); object.as_script_object().unwrap().add_property( context.gc_context, "virtual_hidden", getter, None, DontEnum.into(), ); let keys = object.get_keys(); assert_eq!(keys.len(), 2); assert_eq!(keys.contains(&"stored".to_string()), true); assert_eq!(keys.contains(&"stored_hidden".to_string()), false); assert_eq!(keys.contains(&"virtual".to_string()), true); assert_eq!(keys.contains(&"virtual_hidden".to_string()), false); }) } }