47 lines
1.6 KiB
Rust
47 lines
1.6 KiB
Rust
// Do not remove inline: will result in relocation failure
|
|
#[inline(always)]
|
|
pub(crate) unsafe fn rel_ptr<T>(offset: u64) -> *const T {
|
|
(image_base() + offset) as *const T
|
|
}
|
|
|
|
// Do not remove inline: will result in relocation failure
|
|
#[inline(always)]
|
|
pub(crate) unsafe fn rel_ptr_mut<T>(offset: u64) -> *mut T {
|
|
(image_base() + offset) as *mut T
|
|
}
|
|
|
|
extern "C" {
|
|
static ENCLAVE_SIZE: usize;
|
|
}
|
|
|
|
// Do not remove inline: will result in relocation failure
|
|
// For the same reason we use inline ASM here instead of an extern static to
|
|
// locate the base
|
|
/// Returns address at which current enclave is loaded.
|
|
#[inline(always)]
|
|
#[unstable(feature = "sgx_platform", issue = "56975")]
|
|
pub fn image_base() -> u64 {
|
|
let base;
|
|
unsafe { llvm_asm!("lea IMAGE_BASE(%rip),$0":"=r"(base)) };
|
|
base
|
|
}
|
|
|
|
/// Returns `true` if the specified memory range is in the enclave.
|
|
///
|
|
/// `p + len` must not overflow.
|
|
#[unstable(feature = "sgx_platform", issue = "56975")]
|
|
pub fn is_enclave_range(p: *const u8, len: usize) -> bool {
|
|
let start = p as u64;
|
|
let end = start + (len as u64);
|
|
start >= image_base() && end <= image_base() + (unsafe { ENCLAVE_SIZE } as u64) // unsafe ok: link-time constant
|
|
}
|
|
|
|
/// Returns `true` if the specified memory range is in userspace.
|
|
///
|
|
/// `p + len` must not overflow.
|
|
#[unstable(feature = "sgx_platform", issue = "56975")]
|
|
pub fn is_user_range(p: *const u8, len: usize) -> bool {
|
|
let start = p as u64;
|
|
let end = start + (len as u64);
|
|
end <= image_base() || start >= image_base() + (unsafe { ENCLAVE_SIZE } as u64) // unsafe ok: link-time constant
|
|
}
|