use crate::{Error, Result, page};
use core::ptr;
pub fn round_to_page_boundaries<T>(address: *const T, size: usize) -> Result<(*const T, usize)> {
if size == 0 {
return Err(Error::InvalidParameter("size"));
}
let size = address.addr() % page::size() + size;
let size = page::ceil(ptr::with_exposed_provenance::<T>(size)).addr();
Ok((page::floor(address), size))
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn round_to_page_boundaries_works() -> Result<()> {
let pz = page::size();
let values = &[
((1, pz), (0, pz * 2)),
((0, pz - 1), (0, pz)),
((0, pz + 1), (0, pz * 2)),
((pz - 1, 1), (0, pz)),
((pz + 1, pz), (pz, pz * 2)),
((pz, pz), (pz, pz)),
];
for ((before_address, before_size), (after_address, after_size)) in values {
let (address, size) = round_to_page_boundaries(
ptr::with_exposed_provenance::<()>(*before_address),
*before_size,
)?;
assert_eq!(
(address, size),
(
ptr::with_exposed_provenance::<()>(*after_address),
*after_size
)
);
}
Ok(())
}
}