cleaned up prune

ffi
Avril 3 years ago
parent 1621ca45b6
commit f42cedafc3
Signed by: flanchan
GPG Key ID: 284488987C31F630

@ -0,0 +1,29 @@
use std::ffi::c_void;
use std::ptr;
#[cfg(nightly)]
#[inline(never)]
pub fn explicit_prune(buffer: &mut[u8]) {
unsafe {
ptr::write_bytes(buffer.as_mut_ptr() as *mut c_void, 0, buffer.len());
if cfg!(target_arch = "x86_64") || cfg !(target_arch = "x86") {
asm!("clflush [{}]", in(reg)buffer.as_mut_ptr());
} else {
asm!("")
}
}
}
#[cfg(not(nightly))]
#[inline]
pub fn explicit_prune(buffer: &mut[u8]) {
extern "C" {
fn explicit_bzero(_: *mut c_void, _:usize);
}
unsafe {
explicit_bzero(buffer.as_mut_ptr() as *mut c_void, buffer.len());
}
}

@ -57,7 +57,7 @@ pub trait HexStringSliceIterExt
} }
impl<S> HexStringSliceIterExt for S impl<S> HexStringSliceIterExt for S
where S: AsRef<[u8]> where S: AsRef<[u8]>
{ {
fn hex(&self) -> HexStringSliceIter<'_> fn hex(&self) -> HexStringSliceIter<'_>
{ {

@ -9,6 +9,7 @@
pub mod key; pub mod key;
mod cha; mod cha;
mod stream; mod stream;
mod bytes;
pub use stream::Sink; pub use stream::Sink;
pub use key::{ pub use key::{

@ -118,36 +118,12 @@ where W: Write
/// Clear the internal buffer while keeping it allocated for further use. /// Clear the internal buffer while keeping it allocated for further use.
/// ///
/// This does not affect operations at all, all it does is 0 out the left-over temporary buffer from the last operation(s). /// This does not affect operations at all, all it does is 0 out the left-over temporary buffer from the last operation(s).
#[cfg_attr(all(nightly, feature="explicit_clear"), inline(never))] // We don't want this asm! being inlined and preventing other optimisations on caller functions. #[inline]
pub fn prune(&mut self) pub fn prune(&mut self)
{ {
#[cfg(feature="explicit_clear")] #[cfg(feature="explicit_clear")]
{ {
use std::ffi::c_void; bytes::explicit_prune(&mut self.buffer[..]);
#[cfg(nightly)]
#[inline(never)] unsafe fn explicit_bzero(p: *mut c_void, s: usize)
{
std::ptr::write_bytes(p, 0, s);
asm!("");
}
#[cfg(not(nightly))]
extern "C" {
fn explicit_bzero(_: *mut c_void, _:usize);
}
unsafe {
explicit_bzero(self.buffer.as_mut_ptr() as *mut c_void, self.buffer.len());
#[cfg(nightly)]
if cfg!(target_arch = "x86_64") || cfg!(target_arch = "x86"){
asm!(
"clflush [{}]",
in(reg) self.buffer.as_mut_ptr()
);
}
}
return; return;
} }
#[cfg(not(feature="explicit_clear"))] #[cfg(not(feature="explicit_clear"))]

Loading…
Cancel
Save