@@ -17,7 +17,6 @@ use std::fmt::Debug;
1717use std:: mem:: { offset_of, size_of} ;
1818
1919use hyperlight_common:: mem:: { GuestStackData , HyperlightPEB , RunMode , PAGE_SIZE_USIZE } ;
20- use paste:: paste;
2120use rand:: { rng, RngCore } ;
2221use tracing:: { instrument, Span } ;
2322
@@ -737,14 +736,12 @@ impl SandboxMemoryLayout {
737736 ) -> Result < ( ) > {
738737 macro_rules! get_address {
739738 ( $something: ident) => {
740- paste! {
741- if run_inprocess {
742- let offset = self . [ <$something _offset>] ;
743- let calculated_addr = shared_mem. calculate_address( offset) ?;
744- u64 :: try_from( calculated_addr) ?
745- } else {
746- u64 :: try_from( guest_offset + self . [ <$something _offset>] ) ?
747- }
739+ if run_inprocess {
740+ let offset = self . $something;
741+ let calculated_addr = shared_mem. calculate_address( offset) ?;
742+ u64 :: try_from( calculated_addr) ?
743+ } else {
744+ u64 :: try_from( guest_offset + self . $something) ?
748745 }
749746 } ;
750747 }
@@ -789,7 +786,7 @@ impl SandboxMemoryLayout {
789786 . get_input_data_size ( )
790787 . try_into ( ) ?,
791788 ) ?;
792- let addr = get_address ! ( input_data_buffer ) ;
789+ let addr = get_address ! ( input_data_buffer_offset ) ;
793790 shared_mem. write_u64 ( self . get_input_data_pointer_offset ( ) , addr) ?;
794791
795792 // Set up output buffer pointer
@@ -799,11 +796,11 @@ impl SandboxMemoryLayout {
799796 . get_output_data_size ( )
800797 . try_into ( ) ?,
801798 ) ?;
802- let addr = get_address ! ( output_data_buffer ) ;
799+ let addr = get_address ! ( output_data_buffer_offset ) ;
803800 shared_mem. write_u64 ( self . get_output_data_pointer_offset ( ) , addr) ?;
804801
805802 // Set up the guest panic context buffer
806- let addr = get_address ! ( guest_panic_context_buffer ) ;
803+ let addr = get_address ! ( guest_panic_context_buffer_offset ) ;
807804 shared_mem. write_u64 (
808805 self . get_guest_panic_context_size_offset ( ) ,
809806 self . sandbox_memory_config
@@ -813,7 +810,7 @@ impl SandboxMemoryLayout {
813810 shared_mem. write_u64 ( self . get_guest_panic_context_buffer_pointer_offset ( ) , addr) ?;
814811
815812 // Set up heap buffer pointer
816- let addr = get_address ! ( guest_heap_buffer ) ;
813+ let addr = get_address ! ( guest_heap_buffer_offset ) ;
817814 shared_mem. write_u64 ( self . get_heap_size_offset ( ) , self . heap_size . try_into ( ) ?) ?;
818815 shared_mem. write_u64 ( self . get_heap_pointer_offset ( ) , addr) ?;
819816
0 commit comments