123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341 |
- use core::{
- mem::size_of,
- slice::{from_raw_parts, from_raw_parts_mut},
- };
- use alloc::{string::String, vec::Vec};
- use crate::mm::{verify_area, VirtAddr};
- use super::SystemError;
- pub unsafe fn clear_user(dest: VirtAddr, len: usize) -> Result<usize, SystemError> {
- verify_area(dest, len).map_err(|_| SystemError::EFAULT)?;
- let p = dest.data() as *mut u8;
-
- p.write_bytes(0, len);
- return Ok(len);
- }
- pub unsafe fn copy_to_user(dest: VirtAddr, src: &[u8]) -> Result<usize, SystemError> {
- verify_area(dest, src.len()).map_err(|_| SystemError::EFAULT)?;
- let p = dest.data() as *mut u8;
-
- p.copy_from_nonoverlapping(src.as_ptr(), src.len());
- return Ok(src.len());
- }
- pub unsafe fn copy_from_user(dst: &mut [u8], src: VirtAddr) -> Result<usize, SystemError> {
- verify_area(src, dst.len()).map_err(|_| SystemError::EFAULT)?;
- let src: &[u8] = core::slice::from_raw_parts(src.data() as *const u8, dst.len());
-
- dst.copy_from_slice(&src);
- return Ok(dst.len());
- }
- pub fn check_and_clone_cstr(
- user: *const u8,
- max_length: Option<usize>,
- ) -> Result<String, SystemError> {
- if user.is_null() {
- return Ok(String::new());
- }
-
- let mut buffer = Vec::new();
- for i in 0.. {
- if max_length.is_some() && max_length.as_ref().unwrap() <= &i {
- break;
- }
- let addr = unsafe { user.add(i) };
- let mut c = [0u8; 1];
- unsafe {
- copy_from_user(&mut c, VirtAddr::new(addr as usize))?;
- }
- if c[0] == 0 {
- break;
- }
- buffer.push(c[0]);
- }
- return Ok(String::from_utf8(buffer).map_err(|_| SystemError::EFAULT)?);
- }
- pub fn check_and_clone_cstr_array(user: *const *const u8) -> Result<Vec<String>, SystemError> {
- if user.is_null() {
- Ok(Vec::new())
- } else {
-
- let mut buffer = Vec::new();
- for i in 0.. {
- let addr = unsafe { user.add(i) };
- let str_ptr: *const u8;
-
- unsafe {
- let dst = [0usize; 1];
- let mut dst = core::mem::transmute::<[usize; 1], [u8; size_of::<usize>()]>(dst);
- copy_from_user(&mut dst, VirtAddr::new(addr as usize))?;
- let dst = core::mem::transmute::<[u8; size_of::<usize>()], [usize; 1]>(dst);
- str_ptr = dst[0] as *const u8;
-
- }
- if str_ptr.is_null() {
- break;
- }
-
- let string = check_and_clone_cstr(str_ptr, None)?;
-
- buffer.push(string);
- }
- return Ok(buffer);
- }
- }
- #[derive(Debug)]
- pub struct UserBufferWriter<'a> {
- buffer: &'a mut [u8],
- }
- #[derive(Debug)]
- pub struct UserBufferReader<'a> {
- buffer: &'a [u8],
- }
- #[allow(dead_code)]
- impl<'a> UserBufferReader<'a> {
-
-
-
-
-
-
-
- pub fn new<U>(addr: *const U, len: usize, from_user: bool) -> Result<Self, SystemError> {
- if from_user && verify_area(VirtAddr::new(addr as usize), len).is_err() {
- return Err(SystemError::EFAULT);
- }
- return Ok(Self {
- buffer: unsafe { core::slice::from_raw_parts(addr as *const u8, len) },
- });
- }
-
-
-
-
-
- pub fn read_from_user<T>(&self, offset: usize) -> Result<&[T], SystemError> {
- return self.convert_with_offset(&self.buffer, offset);
- }
-
-
-
-
-
- pub fn read_one_from_user<T>(&self, offset: usize) -> Result<&T, SystemError> {
- return self.convert_one_with_offset(&self.buffer, offset);
- }
-
-
-
-
-
- pub fn copy_from_user<T: core::marker::Copy>(
- &self,
- dst: &mut [T],
- offset: usize,
- ) -> Result<usize, SystemError> {
- let data = self.convert_with_offset(&self.buffer, offset)?;
- dst.copy_from_slice(data);
- return Ok(dst.len());
- }
-
-
-
-
-
- pub fn copy_one_from_user<T: core::marker::Copy>(
- &self,
- dst: &mut T,
- offset: usize,
- ) -> Result<(), SystemError> {
- let data = self.convert_one_with_offset::<T>(&self.buffer, offset)?;
- dst.clone_from(data);
- return Ok(());
- }
- fn convert_with_offset<T>(&self, src: &[u8], offset: usize) -> Result<&[T], SystemError> {
- if offset >= src.len() {
- return Err(SystemError::EINVAL);
- }
- let byte_buffer: &[u8] = &src[offset..];
- if byte_buffer.len() % core::mem::size_of::<T>() != 0 || byte_buffer.is_empty() {
- return Err(SystemError::EINVAL);
- }
- let chunks = unsafe {
- from_raw_parts(
- byte_buffer.as_ptr() as *const T,
- byte_buffer.len() / core::mem::size_of::<T>(),
- )
- };
- return Ok(chunks);
- }
- fn convert_one_with_offset<T>(&self, src: &[u8], offset: usize) -> Result<&T, SystemError> {
- if offset + core::mem::size_of::<T>() > src.len() {
- return Err(SystemError::EINVAL);
- }
- let byte_buffer: &[u8] = &src[offset..offset + core::mem::size_of::<T>()];
- let chunks = unsafe { from_raw_parts(byte_buffer.as_ptr() as *const T, 1) };
- let data = &chunks[0];
- return Ok(data);
- }
- }
- #[allow(dead_code)]
- impl<'a> UserBufferWriter<'a> {
-
-
-
-
-
-
- pub fn new<U>(addr: *mut U, len: usize, from_user: bool) -> Result<Self, SystemError> {
- if from_user
- && verify_area(
- VirtAddr::new(addr as usize),
- (len * core::mem::size_of::<U>()) as usize,
- )
- .is_err()
- {
- return Err(SystemError::EFAULT);
- }
- return Ok(Self {
- buffer: unsafe {
- core::slice::from_raw_parts_mut(addr as *mut u8, len * core::mem::size_of::<U>())
- },
- });
- }
-
-
-
-
-
-
- pub fn copy_to_user<T: core::marker::Copy>(
- &'a mut self,
- src: &'a [T],
- offset: usize,
- ) -> Result<usize, SystemError> {
- let dst = Self::convert_with_offset(self.buffer, offset)?;
- dst.copy_from_slice(&src);
- return Ok(src.len());
- }
-
-
-
-
-
-
- pub fn copy_one_to_user<T: core::marker::Copy>(
- &'a mut self,
- src: &'a T,
- offset: usize,
- ) -> Result<(), SystemError> {
- let dst = Self::convert_one_with_offset::<T>(self.buffer, offset)?;
- dst.clone_from(src);
- return Ok(());
- }
- pub fn buffer<T>(&'a mut self, offset: usize) -> Result<&mut [T], SystemError> {
- Ok(Self::convert_with_offset::<T>(self.buffer, offset).map_err(|_| SystemError::EINVAL)?)
- }
- fn convert_with_offset<T>(src: &mut [u8], offset: usize) -> Result<&mut [T], SystemError> {
- if offset >= src.len() {
- return Err(SystemError::EINVAL);
- }
- let byte_buffer: &mut [u8] = &mut src[offset..];
- if byte_buffer.len() % core::mem::size_of::<T>() != 0 || byte_buffer.is_empty() {
- return Err(SystemError::EINVAL);
- }
- let chunks = unsafe {
- from_raw_parts_mut(
- byte_buffer.as_mut_ptr() as *mut T,
- byte_buffer.len() / core::mem::size_of::<T>(),
- )
- };
- return Ok(chunks);
- }
- fn convert_one_with_offset<T>(src: &mut [u8], offset: usize) -> Result<&mut T, SystemError> {
- if offset + core::mem::size_of::<T>() > src.len() {
- return Err(SystemError::EINVAL);
- }
- let byte_buffer: &mut [u8] = &mut src[offset..offset + core::mem::size_of::<T>()];
- let chunks = unsafe { from_raw_parts_mut(byte_buffer.as_mut_ptr() as *mut T, 1) };
- let data = &mut chunks[0];
- return Ok(data);
- }
- }
|