rpc-state-reader: add get_struct_layout method for providing a move layout

Author: bmwill

crates/simulacrum/src/lib.rs

@@ -602,6 +602,14 @@ impl<T: Send + Sync, V: store::SimulatorStore + Send + Sync> RpcStateReader for
     fn indexes(&self) -> Option<&dyn sui_types::storage::RpcIndexes> {
         None
     }
+
+    fn get_struct_layout(
+        &self,
+        _: &move_core_types::language_storage::StructTag,
+    ) -> sui_types::storage::error::Result<Option<move_core_types::annotated_value::MoveTypeLayout>>
+    {
+        Ok(None)
+    }
 }
 
 impl Simulacrum {

crates/sui-core/src/storage.rs

@@ -476,6 +476,21 @@ impl RpcStateReader for RestReadStore {
     fn indexes(&self) -> Option<&dyn RpcIndexes> {
         self.index().ok().map(|index| index as _)
     }
+
+    fn get_struct_layout(
+        &self,
+        struct_tag: &move_core_types::language_storage::StructTag,
+    ) -> Result<Option<move_core_types::annotated_value::MoveTypeLayout>> {
+        self.state
+            .load_epoch_store_one_call_per_task()
+            .executor()
+            // TODO(cache) - must read through cache
+            .type_layout_resolver(Box::new(self.state.get_backing_package_store().as_ref()))
+            .get_annotated_layout(struct_tag)
+            .map(|layout| layout.into_layout())
+            .map(Some)
+            .map_err(StorageError::custom)
+    }
 }
 
 impl RpcIndexes for RpcIndexStore {

crates/sui-transactional-test-runner/src/simulator_persisted_store.rs

@@ -643,6 +643,14 @@ impl RpcStateReader for PersistedStoreInnerReadOnlyWrapper {
     fn indexes(&self) -> Option<&dyn sui_types::storage::RpcIndexes> {
         None
     }
+
+    fn get_struct_layout(
+        &self,
+        _: &move_core_types::language_storage::StructTag,
+    ) -> sui_types::storage::error::Result<Option<move_core_types::annotated_value::MoveTypeLayout>>
+    {
+        Ok(None)
+    }
 }
 
 impl PersistedStoreInnerReadOnlyWrapper {

crates/sui-types/src/storage/read_store.rs

@@ -18,6 +18,7 @@ use crate::messages_checkpoint::{
 use crate::object::Object;
 use crate::storage::{get_transaction_input_objects, get_transaction_output_objects};
 use crate::transaction::{TransactionData, VerifiedTransaction};
+use move_core_types::annotated_value::MoveTypeLayout;
 use move_core_types::language_storage::StructTag;
 use move_core_types::language_storage::TypeTag;
 use serde::Deserialize;
@@ -552,6 +553,25 @@ pub trait RpcStateReader: ObjectStore + ReadStore + Send + Sync {
 
     // Get a handle to an instance of the RpcIndexes
     fn indexes(&self) -> Option<&dyn RpcIndexes>;
+
+    fn get_type_layout(&self, type_tag: &TypeTag) -> Result<Option<MoveTypeLayout>> {
+        match type_tag {
+            TypeTag::Bool => Ok(Some(MoveTypeLayout::Bool)),
+            TypeTag::U8 => Ok(Some(MoveTypeLayout::U8)),
+            TypeTag::U64 => Ok(Some(MoveTypeLayout::U64)),
+            TypeTag::U128 => Ok(Some(MoveTypeLayout::U128)),
+            TypeTag::Address => Ok(Some(MoveTypeLayout::Address)),
+            TypeTag::Signer => Ok(Some(MoveTypeLayout::Signer)),
+            TypeTag::Vector(type_tag) => Ok(self
+                .get_type_layout(type_tag)?
+                .map(|layout| MoveTypeLayout::Vector(Box::new(layout)))),
+            TypeTag::Struct(struct_tag) => self.get_struct_layout(struct_tag),
+            TypeTag::U16 => Ok(Some(MoveTypeLayout::U16)),
+            TypeTag::U32 => Ok(Some(MoveTypeLayout::U32)),
+            TypeTag::U256 => Ok(Some(MoveTypeLayout::U256)),
+        }
+    }
+    fn get_struct_layout(&self, type_tag: &StructTag) -> Result<Option<MoveTypeLayout>>;
 }
 
 pub type DynamicFieldIteratorItem =