@ -17,8 +17,16 @@ use std::{
use libc ::{
mmap , munmap , madvise , MAP_FAILED ,
} ;
use mapped_file ::{
MappedFile ,
Perm ,
Flags ,
file ::memory ::{
MemoryFile ,
} ,
} ;
use openssl ::symm ::Crypter ;
/*
#[ derive(Debug) ]
struct MapInner
{
@ -35,7 +43,7 @@ impl ops::Drop for MapInner
munmap ( self . mem . as_ptr ( ) as * mut _ , self . size ) ;
}
}
}
} * /
#[ inline ]
pub fn raw_file_size ( fd : & ( impl AsRawFd + ? Sized ) ) -> io ::Result < u64 >
@ -51,6 +59,17 @@ pub fn raw_file_size(fd: &(impl AsRawFd + ?Sized)) -> io::Result<u64>
}
}
#[ inline ]
fn try_truncate ( fd : & ( impl AsRawFd + ? Sized ) , to : u64 ) -> io ::Result < ( ) >
{
//use libc::ftruncate;
let to = to . try_into ( ) . map_err ( | e | io ::Error ::new ( io ::ErrorKind ::InvalidInput , e ) ) ? ;
match unsafe { ftruncate ( fd . as_raw_fd ( ) , to ) } {
0 = > Ok ( ( ) ) ,
_ = > Err ( io ::Error ::last_os_error ( ) ) ,
}
}
/*
#[ derive(Debug) ]
pub struct Mapped < T : ? Sized >
{
@ -79,7 +98,7 @@ impl<T: AsRawFd> Mapped<T>
let size = raw_file_size ( & file ) ? ;
Self ::try_new_sized ( file , size , rw )
}
}
} * /
/*
impl < T : AsRawFd > TryFrom < T > for Mapped < T >
{
@ -92,19 +111,178 @@ impl<T: AsRawFd> TryFrom<T> for Mapped<T>
} * /
fn process_mapped_files < T , U > ( mode : & mut Crypter , input : & mut Mapped < T > , output : & mut Mapped < U > ) -> io ::Result < ( ) >
where T : AsRawFd + ? Sized ,
U : AsRawFd + ? Sized
fn process_mapped_files < T , U > ( mode : & mut Crypter , input : & mut MappedFile < T > , output : & mut MappedFile < U > ) -> io ::Result < ( ) >
where T : AsRawFd ,
U : AsRawFd ,
{
mode . update ( & input [ .. ] , & mut output [ .. ] ) ? ;
Ok ( ( ) )
}
fn try_map_sized < T : AsRawFd > ( file : T , perm : mapped_file ::Perm , flags : impl mapped_file ::MapFlags ) -> Result < MappedFile < T > , T >
{
macro_rules! unwrap {
( $res :expr ) = > {
match $res {
Ok ( v ) = > v ,
_ = > return Err ( file )
}
} ;
}
if let Ok ( sz ) = raw_file_size ( & file ) {
let sz = unwrap ! ( sz . try_into ( ) ) ;
MappedFile ::try_new ( file , sz , perm , flags ) . map_err ( | err | err . into_inner ( ) )
} else {
Err ( file )
}
}
#[ inline ]
fn try_map_to < T : AsRawFd + ? Sized > ( file : & T , file_size : usize ) -> bool
{
todo! ( "mode.update(input.slice(), output.slice()); /* unneeded: mode.finalize(input.slice(), output.slice()) */ " ) ;
let file_size = match file_size . try_into ( ) {
Ok ( v ) = > v ,
_ = > return false
} ;
if let Ok ( stdout_size ) = raw_file_size ( file ) {
if stdout_size > = file_size {
// Size already ok
return true ;
}
}
// Grow file
unsafe {
libc ::ftruncate ( file . as_raw_fd ( ) , file_size ) = = 0
}
}
#[ inline ]
fn translate_hugetlb_size ( size : usize ) -> mapped_file ::hugetlb ::HugePage
{
#[ inline ]
fn check_func ( sizes_kb : & [ usize ] ) -> Option < & usize >
{
sizes_kb . iter ( ) . skip_while ( | & & kb | kb < 1024 * 1024 ) . next ( )
. or_else ( | | sizes_kb . iter ( ) . nth ( 1 )
. or_else ( | | sizes_kb . first ( ) ) )
}
match size {
0 .. ( 1024 * 1024 ) = > mapped_file ::hugetlb ::HugePage ::Smallest ,
( 1024 * 1024 ) .. ( 1024 * 1024 * 1024 ) = > mapped_file ::hugetlb ::HugePage ::Selected ( check_func ) ,
very_high = > mapped_file ::hugetlb ::HugePage ::Largest ,
}
}
/// Create and map a temporary memory file of this size.
///
/// This function may optionally choose to huge hugepages if `size` is large enough
fn create_sized_temp_mapping ( size : usize ) -> io ::Result < ( MappedFile < MemoryFile > , bool ) >
{
const HUGETLB_THRESH : usize = 1024 * 1024 ; // 1MB
let file = match size {
0 = > MemoryFile ::new ( ) ,
0 .. HUGETLB_THRESH = > MemoryFile ::with_size ( size ) ,
size = > {
let hugetlb = translate_hugetlb_size ( size ) ;
let file = MemoryFile ::with_size_hugetlb ( size , hugetlb ) ? ;
return MappedFile ::new ( file , size , Perm ::ReadWrite , Flags ::Shared . with_hugetlb ( hugetlb ) ) . map ( | x | ( x , true ) ) ;
} ,
} ? ;
MappedFile ::new ( file , size , Perm ::ReadWrite , Flags ::Shared ) . map ( | x | ( x , false ) )
}
pub fn try_process ( mut mode : impl BorrowMut < Crypter > ) -> io ::Result < io ::Result < ( ) > >
{
let mode = mode . borrow_mut ( ) ;
let stdin = io ::stdin ( ) . lock ( ) ;
let stdout = io ::stdout ( ) . lock ( ) ;
let file_size = raw_file_size ( & stdin ) ? ;
macro_rules! attempt_mapping {
( $file :expr , $perm :expr , $flags :expr ) = > {
{
let file = $file ;
if try_map_to ( & file , file_size ) {
MappedFile ::try_new ( file , file_size , $perm , $flags ) . map_err ( | e | e . into_inner ( ) )
} else {
return Err ( io ::Error ::new ( io ::ErrorKind ::InvalidData , concat! ( "Failed to truncate " , stringify! ( $file ) , " to size" ) ) ) ;
}
}
} ;
( $file :expr , $perm :expr ) = > {
attempt_mapping ( $file , $perm , Flags ::default ( ) )
} ;
}
// Try map stdout
Ok ( match attempt_mapping ! ( stdout , Perm ::ReadWrite , Flags ::Shared ) {
Ok ( mut mstdout ) = > {
let res = match try_map_sized ( stdin , mapped_file ::Perm ::Readonly , mapped_file ::Flags ::Private )
{
Ok ( mut mstdin ) = > {
// Stdin and stdout are mapped. (3)
process_mapped_files ( & mut mode , & mut mstdin , & mut mstdout )
} ,
Err ( _ ) = > {
// Only stdout is mapped. (1)
let size = file_size as usize ;
let is_huge = size > = 1024 * 1024 ;
if is_huge {
let ( mapped_memfd , _ ) = create_sized_temp_mapping ( size ) ? ;
io ::copy ( & mut stdin , & mut & mapped_memfd [ .. ] ) . and_then ( move | _ | mapped_memfd )
} else {
MemoryFile ::with_size ( size ) . or_else ( | _ | MemoryFile ::new ( ) ) . and_then ( | mut memfd | {
let size = io ::copy ( & mut stdin , & mut memfd ) ? ;
MappedFile ::new ( memfd , size as usize , Perm ::ReadWrite , Flags ::Shared )
} )
} . and_then ( move | mut mapped_memfd | {
process_mapped_files ( mode , & mut mapped_memfd , & mut mstdout )
} )
//todo!("Copy stdin into a memory-file, and then map that and process it to stdout (XXX: What if the file is too large, but we cannot tell the size of stdin?)")
}
} ;
if res . is_ok ( ) {
mstdout . flush ( mapped_file ::Flush ::Wait )
} else {
res
}
} ,
Err ( mut stdout ) = > {
match try_map_sized ( stdin , mapped_file ::Perm ::Readonly , mapped_file ::Flags ::Private )
{
Ok ( mut mstdin ) = > {
// Only stdin is mapped. (2)
if cfg! ( feature = "sodiumoxide" ) {
todo! ( "XXX: When we switch to `sodiumoxide`, we won't *need* this, we can mutate the *private* stdin mapping directly." )
} else {
//TODO: XXX: When we switch to `sodiumoxide`, we won't *need* this, we can mutate the *private* stdin mapping directly.
//todo!("Create a memory file (possibly with hugetlb, depending on `file_size`), map that, process_mapped_files(...) into that memory file, then `io::copy(&mut &memfile[..], &stdout)`")
let ( mapped_memfd , is_huge ) = create_sized_temp_mapping ( file_size as usize ) ? ;
process_mapped_files ( & mut mode , & mut mstdin , & mut mapped_memfd ) . and_then ( move | _ | {
if is_huge {
// Cannot use fd syscalls on `MFD_HUGELB`, copy into stdout from the mapped buffer itself.
io ::copy ( & mut & mapped_file [ .. ] , & mut stdout )
} else {
// Sync the contents into the memory file then use fd syscalls to copy into stdout.
mapped_memfd . flush ( mapped_file ::Flush ::Wait ) . and_then ( move | _ | {
let mut memfd = mapped_memfd . into_inner ( ) ;
io ::copy ( & mut memfd , & mut stdout )
} )
}
} ) . map ( | _ | ( ) )
}
} ,
Err ( _ ) = > {
// Neither are mapped. (0)
return Err ( io ::Error ::new ( io ::ErrorKind ::InvalidInput , "cannot map stdin or stdout" ) ) ;
}
}
} ,
} )
/* let mstdin = Mapped::try_new(input)?;
let mstdout = Mapped ::try_new ( output ) ? ; * / //TODO: if failed to map output, but input is successful (TODO: XXX: implement other way around), we can fall back to wrapping output in `Sink`.
todo! ( "Try to map the stdin and stdout streams, if that fails, return Err(last_os_err)." ) ;
todo! ( "return Ok(process_mapped_files(mode, mstdin, mstdout, key, iv))" )
//todo!(
//todo!(
}
