Update align pointer macro

Author: febo

sdk/pinocchio/src/entrypoint/mod.rs

@@ -13,6 +13,7 @@ pub use alloc::BumpAllocator;
 use core::{
     cmp::min,
     mem::{size_of, MaybeUninit},
+    ptr::with_exposed_provenance_mut,
     slice::from_raw_parts,
 };
 
@@ -195,9 +196,14 @@ macro_rules! program_entrypoint {
 /// Align a pointer to the BPF alignment of [`u128`].
 macro_rules! align_pointer {
     ($ptr:ident) => {
-        // integer-to-pointer cast: the resulting pointer will have the same provenance as
-        // the original pointer and it follows the alignment requirement for the input.
-        $ptr.with_addr((($ptr.addr() + (BPF_ALIGN_OF_U128 - 1)) & !(BPF_ALIGN_OF_U128 - 1)))
+        // integer-to-pointer cast: calculate the aligned address as a usize first
+        // since it is more CU efficient than `ptr::align_offset()` or strict Provenance
+        // API (e.g., `ptr::with_addr()`); and then cast it back to a pointer. The runtime
+        // guarantees that the resulting pointer is valid, since it follows the layout
+        // serialized by the SVM loader.
+        with_exposed_provenance_mut(
+            ($ptr.expose_provenance() + (BPF_ALIGN_OF_U128 - 1)) & !(BPF_ALIGN_OF_U128 - 1),
+        )
     };
 }