@@ -878,14 +878,14 @@ mod tests {
878
878
assert_eq ! ( packet. header, PacketHeader :: default ( ) ) ;
879
879
let header = packet. header_slice ( ) ;
880
880
assert_eq ! (
881
- header. as_ptr( ) ,
881
+ header. ptr_guard ( ) . as_ptr( ) ,
882
882
mem. get_host_address( GuestAddress ( 0x5_0000 ) ) . unwrap( )
883
883
) ;
884
884
assert_eq ! ( header. len( ) , PKT_HEADER_SIZE ) ;
885
885
886
886
let data = packet. data_slice ( ) . unwrap ( ) ;
887
887
assert_eq ! (
888
- data. as_ptr( ) ,
888
+ data. ptr_guard ( ) . as_ptr( ) ,
889
889
mem. get_host_address( GuestAddress ( 0x8_0000 ) ) . unwrap( )
890
890
) ;
891
891
assert_eq ! ( data. len( ) , 0x100 ) ;
@@ -911,14 +911,14 @@ mod tests {
911
911
assert_eq ! ( packet. header, PacketHeader :: default ( ) ) ;
912
912
let header = packet. header_slice ( ) ;
913
913
assert_eq ! (
914
- header. as_ptr( ) ,
914
+ header. ptr_guard ( ) . as_ptr( ) ,
915
915
mem. get_host_address( GuestAddress ( 0x5_0000 ) ) . unwrap( )
916
916
) ;
917
917
assert_eq ! ( header. len( ) , PKT_HEADER_SIZE ) ;
918
918
919
919
let data = packet. data_slice ( ) . unwrap ( ) ;
920
920
assert_eq ! (
921
- data. as_ptr( ) ,
921
+ data. ptr_guard ( ) . as_ptr( ) ,
922
922
mem. get_host_address( GuestAddress ( 0x5_0000 + PKT_HEADER_SIZE as u64 ) )
923
923
. unwrap( )
924
924
) ;
@@ -976,7 +976,7 @@ mod tests {
976
976
assert_eq ! ( packet. header, header) ;
977
977
let header_slice = packet. header_slice ( ) ;
978
978
assert_eq ! (
979
- header_slice. as_ptr( ) ,
979
+ header_slice. ptr_guard ( ) . as_ptr( ) ,
980
980
mem. get_host_address( GuestAddress ( 0x10_0000 ) ) . unwrap( )
981
981
) ;
982
982
assert_eq ! ( header_slice. len( ) , PKT_HEADER_SIZE ) ;
@@ -1116,7 +1116,7 @@ mod tests {
1116
1116
assert_eq ! ( packet. header, header) ;
1117
1117
let header_slice = packet. header_slice ( ) ;
1118
1118
assert_eq ! (
1119
- header_slice. as_ptr( ) ,
1119
+ header_slice. ptr_guard ( ) . as_ptr( ) ,
1120
1120
mem. get_host_address( GuestAddress ( 0x5_0000 ) ) . unwrap( )
1121
1121
) ;
1122
1122
assert_eq ! ( header_slice. len( ) , PKT_HEADER_SIZE ) ;
@@ -1125,7 +1125,7 @@ mod tests {
1125
1125
1126
1126
let data = packet. data_slice ( ) . unwrap ( ) ;
1127
1127
assert_eq ! (
1128
- data. as_ptr( ) ,
1128
+ data. ptr_guard ( ) . as_ptr( ) ,
1129
1129
mem. get_host_address( GuestAddress ( 0x8_0000 ) ) . unwrap( )
1130
1130
) ;
1131
1131
assert_eq ! ( data. len( ) , LEN as usize ) ;
@@ -1151,7 +1151,7 @@ mod tests {
1151
1151
assert_eq ! ( packet. header, header) ;
1152
1152
let header_slice = packet. header_slice ( ) ;
1153
1153
assert_eq ! (
1154
- header_slice. as_ptr( ) ,
1154
+ header_slice. ptr_guard ( ) . as_ptr( ) ,
1155
1155
mem. get_host_address( GuestAddress ( 0x5_0000 ) ) . unwrap( )
1156
1156
) ;
1157
1157
assert_eq ! ( header_slice. len( ) , PKT_HEADER_SIZE ) ;
@@ -1160,7 +1160,7 @@ mod tests {
1160
1160
1161
1161
let data = packet. data_slice ( ) . unwrap ( ) ;
1162
1162
assert_eq ! (
1163
- data. as_ptr( ) ,
1163
+ data. ptr_guard ( ) . as_ptr( ) ,
1164
1164
mem. get_host_address( GuestAddress ( 0x5_0000 + PKT_HEADER_SIZE as u64 ) )
1165
1165
. unwrap( )
1166
1166
) ;
@@ -1346,16 +1346,25 @@ mod tests {
1346
1346
// SAFETY: safe because ``hdr_raw` and `data_raw` live for as long as
1347
1347
// the scope of the current test.
1348
1348
let packet = unsafe { VsockPacket :: new ( hdr_raw, Some ( data_raw) ) . unwrap ( ) } ;
1349
- assert_eq ! ( packet. header_slice. as_ptr( ) , hdr_raw. as_mut_ptr( ) ) ;
1349
+ assert_eq ! (
1350
+ packet. header_slice. ptr_guard( ) . as_ptr( ) ,
1351
+ hdr_raw. as_mut_ptr( ) ,
1352
+ ) ;
1350
1353
assert_eq ! ( packet. header_slice. len( ) , PKT_HEADER_SIZE ) ;
1351
1354
assert_eq ! ( packet. header, PacketHeader :: default ( ) ) ;
1352
- assert_eq ! ( packet. data_slice. unwrap( ) . as_ptr( ) , data_raw. as_mut_ptr( ) ) ;
1355
+ assert_eq ! (
1356
+ packet. data_slice. unwrap( ) . ptr_guard( ) . as_ptr( ) ,
1357
+ data_raw. as_mut_ptr( ) ,
1358
+ ) ;
1353
1359
assert_eq ! ( packet. data_slice. unwrap( ) . len( ) , LEN as usize ) ;
1354
1360
1355
1361
// SAFETY: Safe because ``hdr_raw` and `data_raw` live as long as the
1356
1362
// scope of the current test.
1357
1363
let packet = unsafe { VsockPacket :: new ( hdr_raw, None ) . unwrap ( ) } ;
1358
- assert_eq ! ( packet. header_slice. as_ptr( ) , hdr_raw. as_mut_ptr( ) ) ;
1364
+ assert_eq ! (
1365
+ packet. header_slice. ptr_guard( ) . as_ptr( ) ,
1366
+ hdr_raw. as_mut_ptr( ) ,
1367
+ ) ;
1359
1368
assert_eq ! ( packet. header, PacketHeader :: default ( ) ) ;
1360
1369
assert ! ( packet. data_slice. is_none( ) ) ;
1361
1370
0 commit comments