builds

OffchainLabs · Dec 27, 2024 · d02336b · d02336b
1 parent 280ff2e
commit d02336b
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/stylus-proc/src/macros/entrypoint.rs b/stylus-proc/src/macros/entrypoint.rs
@@ -127,7 +127,7 @@ fn top_level_storage_impl(item: &syn::ItemStruct) -> syn::ItemImpl {
 fn struct_entrypoint_fn(name: &Ident) -> syn::ItemFn {
     parse_quote! {
         fn #STRUCT_ENTRYPOINT_FN<H: Host>(input: alloc::vec::Vec<u8>, host: Rc<H>) -> stylus_sdk::ArbResult {
-            stylus_sdk::abi::router_entrypoint::<#name, #name, H>(input, host)
+            stylus_sdk::abi::router_entrypoint::<#name<H>, #name<H>, H>(input, host)
         }
     }
 }

diff --git a/stylus-proc/src/macros/storage.rs b/stylus-proc/src/macros/storage.rs
@@ -166,6 +166,9 @@ impl StorageField {
             return None;
         };
         let ty = &self.ty;
+        if ty.to_token_stream().to_string() == "Rc < H >".to_string() {
+            return None;
+        }
         Some(parse_quote! {
             #ident: {
                 let bytes = <#ty as storage::StorageType<H>>::SLOT_BYTES;
@@ -177,7 +180,7 @@ impl StorageField {
                 space -= bytes;
 
                 let root = root + alloy_primitives::U256::from(slot);
-                let field = <#ty as storage::StorageType<H>>::new(root, space as u8, Rc::clone(&self.host));
+                let field = <#ty as storage::StorageType<H>>::new(root, space as u8, Rc::clone(&host));
                 if words > 0 {
                     slot += words;
                     space = 32;
@@ -189,6 +192,9 @@ impl StorageField {
 
     fn size(&self) -> TokenStream {
         let ty = &self.ty;
+        if ty.to_token_stream().to_string() == "Rc < H >".to_string() {
+            return quote! {};
+        }
         quote! {
             let bytes = <#ty as storage::StorageType<H>>::SLOT_BYTES;
             let words = <#ty as storage::StorageType<H>>::REQUIRED_SLOTS;

diff --git a/stylus-sdk/src/storage/traits.rs b/stylus-sdk/src/storage/traits.rs
@@ -319,7 +319,7 @@ pub trait GlobalStorage<H: Host> {
 
         if B == 256 {
             return Self::set_word(
-                host,
+                Rc::clone(&host),
                 key,
                 FixedBytes::from_slice(&value.to_be_bytes::<32>()),
             );