@@ -17,7 +17,6 @@ use std::fmt::Debug;
17
17
use std:: mem:: { offset_of, size_of} ;
18
18
19
19
use hyperlight_common:: mem:: { GuestStackData , HyperlightPEB , RunMode , PAGE_SIZE_USIZE } ;
20
- use paste:: paste;
21
20
use rand:: { rng, RngCore } ;
22
21
use tracing:: { instrument, Span } ;
23
22
@@ -737,14 +736,12 @@ impl SandboxMemoryLayout {
737
736
) -> Result < ( ) > {
738
737
macro_rules! get_address {
739
738
( $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) ?
748
745
}
749
746
} ;
750
747
}
@@ -789,7 +786,7 @@ impl SandboxMemoryLayout {
789
786
. get_input_data_size ( )
790
787
. try_into ( ) ?,
791
788
) ?;
792
- let addr = get_address ! ( input_data_buffer ) ;
789
+ let addr = get_address ! ( input_data_buffer_offset ) ;
793
790
shared_mem. write_u64 ( self . get_input_data_pointer_offset ( ) , addr) ?;
794
791
795
792
// Set up output buffer pointer
@@ -799,11 +796,11 @@ impl SandboxMemoryLayout {
799
796
. get_output_data_size ( )
800
797
. try_into ( ) ?,
801
798
) ?;
802
- let addr = get_address ! ( output_data_buffer ) ;
799
+ let addr = get_address ! ( output_data_buffer_offset ) ;
803
800
shared_mem. write_u64 ( self . get_output_data_pointer_offset ( ) , addr) ?;
804
801
805
802
// 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 ) ;
807
804
shared_mem. write_u64 (
808
805
self . get_guest_panic_context_size_offset ( ) ,
809
806
self . sandbox_memory_config
@@ -813,7 +810,7 @@ impl SandboxMemoryLayout {
813
810
shared_mem. write_u64 ( self . get_guest_panic_context_buffer_pointer_offset ( ) , addr) ?;
814
811
815
812
// Set up heap buffer pointer
816
- let addr = get_address ! ( guest_heap_buffer ) ;
813
+ let addr = get_address ! ( guest_heap_buffer_offset ) ;
817
814
shared_mem. write_u64 ( self . get_heap_size_offset ( ) , self . heap_size . try_into ( ) ?) ?;
818
815
shared_mem. write_u64 ( self . get_heap_pointer_offset ( ) , addr) ?;
819
816
0 commit comments