Validate memory pointers

kernel/scheme/user.rs

@@ -65,6 +65,9 @@ impl UserInner {
     }
 
     fn capture_inner(&self, address: usize, size: usize, writable: bool) -> Result<usize> {
+        if size == 0 {
+            Ok(0)
+        } else {
             let context_lock = self.context.upgrade().ok_or(Error::new(ESRCH))?;
             let context = context_lock.read();
 
@@ -112,10 +115,14 @@ impl UserInner {
                 &mut temporary_page
             ));
 
-        return Ok(to_address + offset);
+            Ok(to_address + offset)
+        }
     }
 
     pub fn release(&self, address: usize) -> Result<()> {
+        if address == 0 {
+            Ok(())
+        } else {
             let context_lock = self.context.upgrade().ok_or(Error::new(ESRCH))?;
             let context = context_lock.read();
 
@@ -136,6 +143,7 @@ impl UserInner {
 
             Err(Error::new(EFAULT))
         }
+    }
 
     pub fn read(&self, buf: &mut [u8]) -> Result<usize> {
         let packet_size = mem::size_of::<Packet>();

kernel/syscall/validate.rs

@@ -1,15 +1,39 @@
-use core::slice;
+use core::{mem, slice};
 
+use arch::paging::{ActivePageTable, Page, VirtualAddress, entry};
 use syscall::error::*;
 
-/// Convert a pointer and length to slice, if valid
-/// TODO: Check validity
-pub fn validate_slice<T>(ptr: *const T, len: usize) -> Result<&'static [T]> {
-    Ok(unsafe { slice::from_raw_parts(ptr, len) })
+fn validate(address: usize, size: usize, flags: entry::EntryFlags) -> Result<()> {
+    let active_table = unsafe { ActivePageTable::new() };
+
+    let start_page = Page::containing_address(VirtualAddress::new(address));
+    let end_page = Page::containing_address(VirtualAddress::new(address + size - 1));
+    for page in Page::range_inclusive(start_page, end_page) {
+        let page_flags = active_table.translate_page_flags(page).ok_or(Error::new(EFAULT))?;
+        if ! page_flags.contains(flags) {
+            return Err(Error::new(EFAULT));
+        }
+    }
+
+    Ok(())
 }
 
 /// Convert a pointer and length to slice, if valid
-/// TODO: Check validity
-pub fn validate_slice_mut<T>(ptr: *mut T, len: usize) -> Result<&'static mut [T]> {
-    Ok(unsafe { slice::from_raw_parts_mut(ptr, len) })
+pub fn validate_slice<T>(ptr: *const T, len: usize) -> Result<&'static [T]> {
+    if len == 0 {
+        Ok(&[])
+    } else {
+        validate(ptr as usize, len * mem::size_of::<T>(), entry::PRESENT /* TODO | entry::USER_ACCESSIBLE */)?;
+        Ok(unsafe { slice::from_raw_parts(ptr, len) })
+    }
+}
+
+/// Convert a pointer and length to slice, if valid
+pub fn validate_slice_mut<T>(ptr: *mut T, len: usize) -> Result<&'static mut [T]> {
+    if len == 0 {
+        Ok(&mut [])
+    } else {
+        validate(ptr as usize, len * mem::size_of::<T>(), entry::PRESENT | entry::WRITABLE /* TODO | entry::USER_ACCESSIBLE */)?;
+        Ok(unsafe { slice::from_raw_parts_mut(ptr, len) })
+    }
 }

schemes/example/src/main.rs

@@ -31,6 +31,7 @@ fn main(){
         loop {
             let mut packet = Packet::default();
             socket.read(&mut packet).expect("example: failed to read events from example scheme");
+            println!("{:?}", packet);
             scheme.handle(&mut packet);
             socket.write(&packet).expect("example: failed to write responses to example scheme");
         }