avm2: Implement SharedObject.size

core/src/avm2/globals/flash/net/SharedObject.as

@@ -19,6 +19,8 @@ package flash.net
 
       native public static function getLocal(name:String, localPath:String = null, secure:Boolean = false): SharedObject;
 
+      native public function get size() : uint;
+
       native public function flush(minDiskSpace:int = 0) : String;
       native public function close() : void;
       native public function clear() : void;
@@ -29,9 +31,6 @@ package flash.net
 
       // note: this is supposed to be a read-only property
       public var data: Object;
-
-      // note: this is supposed to be a read-only property
-      public var size: uint;
 
       ruffle var _ruffleName: String;
    }

core/src/avm2/globals/flash/net/shared_object.rs

@@ -10,6 +10,23 @@ use crate::string::AvmString;
 use flash_lso::types::{AMFVersion, Lso};
 use std::borrow::Cow;
 
+fn new_lso<'gc>(
+    activation: &mut Activation<'_, 'gc>,
+    name: &str,
+    data: Object<'gc>,
+) -> Result<Lso, Error<'gc>> {
+    let mut elements = Vec::new();
+    crate::avm2::amf::recursive_serialize(activation, data, &mut elements, AMFVersion::AMF3)?;
+    Ok(Lso::new(
+        elements,
+        name.split('/')
+            .last()
+            .map(|e| e.to_string())
+            .unwrap_or_else(|| "<unknown>".to_string()),
+        AMFVersion::AMF3,
+    ))
+}
+
 pub fn get_local<'gc>(
     activation: &mut Activation<'_, 'gc>,
     this: Object<'gc>,
@@ -193,20 +210,42 @@ pub fn flush<'gc>(
         .coerce_to_string(activation)?;
     let name = name.to_utf8_lossy();
 
-    let mut elements = Vec::new();
-    crate::avm2::amf::recursive_serialize(activation, data, &mut elements, AMFVersion::AMF3)?;
-    let mut lso = Lso::new(
-        elements,
-        name.split('/')
-            .last()
-            .map(|e| e.to_string())
-            .unwrap_or_else(|| "<unknown>".to_string()),
-        AMFVersion::AMF3,
-    );
+    let mut lso = new_lso(activation, &name, data)?;
+    // Flash does not write empty LSOs to disk
+    if lso.body.is_empty() {
+        Ok(true.into())
+    } else {
+        let bytes = flash_lso::write::write_to_bytes(&mut lso).unwrap_or_default();
+        Ok(activation.context.storage.put(&name, &bytes).into())
+    }
+}
+
+pub fn get_size<'gc>(
+    activation: &mut Activation<'_, 'gc>,
+    this: Object<'gc>,
+    _args: &[Value<'gc>],
+) -> Result<Value<'gc>, Error<'gc>> {
+    let data = this
+        .get_public_property("data", activation)?
+        .coerce_to_object(activation)?;
 
-    let bytes = flash_lso::write::write_to_bytes(&mut lso).unwrap_or_default();
+    let ruffle_name = Multiname::new(
+        Namespace::package("__ruffle__", &mut activation.borrow_gc()),
+        "_ruffleName",
+    );
+    let name = this
+        .get_property(&ruffle_name, activation)?
+        .coerce_to_string(activation)?;
+    let name = name.to_utf8_lossy();
 
-    Ok(activation.context.storage.put(&name, &bytes).into())
+    let mut lso = new_lso(activation, &name, data)?;
+    // Flash returns 0 for empty LSOs, but the actual number of bytes (including the header) otherwise
+    if lso.body.is_empty() {
+        Ok(0.into())
+    } else {
+        let bytes = flash_lso::write::write_to_bytes(&mut lso).unwrap_or_default();
+        Ok(bytes.len().into())
+    }
 }
 
 pub fn close<'gc>(

tests/tests/swfs/avm2/shared_object/Test.as

@@ -63,6 +63,7 @@
 
 			if(obj.data.props === undefined) {
 				trace("No data found. Initializing...");
+				trace("sizes: obj = " + obj.size + ", otherObj = " + otherObj.size);
 
 				storeData(obj.data)
 				storeData(otherObj.data)
@@ -74,8 +75,10 @@
 				trace("otherObj.data.props:")
 				dump(otherObj.data.props);
 
+				trace("sizes: obj = " + obj.size + ", otherObj = " + otherObj.size);
 				obj.flush();
 				otherObj.flush();
+				trace("sizes: obj = " + obj.size + ", otherObj = " + otherObj.size);
 			} else {		
 				trace("obj.data.hidden: " + obj.data.hidden);
 				trace("otherObj.data.hidden: " + otherObj.data.hidden);
@@ -87,6 +90,9 @@
 				trace()
 				trace("otherObj dump:")
 				dump(otherObj.data.props)
+
+				trace();
+				trace("sizes: obj = " + obj.size + ", otherObj = " + otherObj.size);
 
 				// Test SharedObject.clear()
 				trace();
@@ -98,6 +104,9 @@
 				trace();
 				trace("obj dump:");
 				dump(obj.data);
+
+				trace();
+				trace("sizes: obj = " + obj.size + ", otherObj = " + otherObj.size);
 
 				trace();
 				trace("obj dump again:");
@@ -132,5 +141,3 @@ function dump(obj:Object) {
 
 	}
 }
-
-test.main();

tests/tests/swfs/avm2/shared_object/output1.txt

@@ -1,6 +1,7 @@
 typeof obj =object
 typeof otherObj =object
 No data found. Initializing...
+sizes: obj = 0, otherObj = 0
 otherObj.data.props:
 0 = true type boolean
 1 = hello type string
@@ -13,3 +14,5 @@ otherObj.data.props:
 8 =  type object
 val.aPropertyThatIsNotANumber: strange property
 9 = (UTC)  Sun Jan 25 20:31:23 1970 UTC
+sizes: obj = 299, otherObj = 330
+sizes: obj = 299, otherObj = 330

tests/tests/swfs/avm2/shared_object/output2.txt

@@ -29,9 +29,13 @@ otherObj dump:
 val.aPropertyThatIsNotANumber: strange property
 9 = (UTC)  Sun Jan 25 20:31:23 1970 UTC
 
+sizes: obj = 299, otherObj = 330
+
 obj.clear()
 obj == previousData: false
 
 obj dump:
 
+sizes: obj = 0, otherObj = 330
+
 obj dump again: