From 965941cd1ca3cd038d5cfc6fe83f269098ad9bac Mon Sep 17 00:00:00 2001 From: Jeremy Soller <jackpot51@gmail.com> Date: Wed, 5 Oct 2016 18:01:05 -0600 Subject: [PATCH] Implement unix permissions --- scheme/debug.rs | 2 +- scheme/env.rs | 2 +- scheme/event.rs | 2 +- scheme/initfs.rs | 6 ++++-- scheme/irq.rs | 2 +- scheme/root.rs | 2 +- scheme/user.rs | 10 +++++----- syscall/fs.rs | 28 ++++++++++++++-------------- syscall/mod.rs | 3 +++ syscall/process.rs | 32 ++++++++++++++++++++++++++++++++ 10 files changed, 63 insertions(+), 26 deletions(-) diff --git a/scheme/debug.rs b/scheme/debug.rs index eed44ede..52d9a8bf 100644 --- a/scheme/debug.rs +++ b/scheme/debug.rs @@ -33,7 +33,7 @@ pub extern fn debug_input(b: u8) { pub struct DebugScheme; impl Scheme for DebugScheme { - fn open(&self, _path: &[u8], _flags: usize) -> Result<usize> { + fn open(&self, _path: &[u8], _flags: usize, _uid: u32, _gid: u32) -> Result<usize> { Ok(0) } diff --git a/scheme/env.rs b/scheme/env.rs index 807c2736..4a514850 100644 --- a/scheme/env.rs +++ b/scheme/env.rs @@ -32,7 +32,7 @@ impl EnvScheme { } impl Scheme for EnvScheme { - fn open(&self, path: &[u8], _flags: usize) -> Result<usize> { + fn open(&self, path: &[u8], _flags: usize, _uid: u32, _gid: u32) -> Result<usize> { let path = str::from_utf8(path).map_err(|_err| Error::new(ENOENT))?.trim_matches('/'); let env_lock = { diff --git a/scheme/event.rs b/scheme/event.rs index 2dead8d3..8229559c 100644 --- a/scheme/event.rs +++ b/scheme/event.rs @@ -24,7 +24,7 @@ impl EventScheme { } impl Scheme for EventScheme { - fn open(&self, _path: &[u8], _flags: usize) -> Result<usize> { + fn open(&self, _path: &[u8], _flags: usize, _uid: u32, _gid: u32) -> Result<usize> { let handle = { let contexts = context::contexts(); let context_lock = contexts.current().ok_or(Error::new(ESRCH))?; diff --git a/scheme/initfs.rs b/scheme/initfs.rs index c3a795c8..18696ed0 100644 --- a/scheme/initfs.rs +++ b/scheme/initfs.rs @@ -35,7 +35,7 @@ impl InitFsScheme { } impl Scheme for InitFsScheme { - fn open(&self, path: &[u8], _flags: usize) -> Result<usize> { + fn open(&self, path: &[u8], _flags: usize, _uid: u32, _gid: u32) -> Result<usize> { let path_utf8 = str::from_utf8(path).map_err(|_err| Error::new(ENOENT))?; let path_trimmed = path_utf8.trim_matches('/'); @@ -46,7 +46,7 @@ impl Scheme for InitFsScheme { self.handles.write().insert(id, Handle { path: entry.0, data: (entry.1).0, - mode: if (entry.1).1 { MODE_DIR } else { MODE_FILE }, + mode: if (entry.1).1 { MODE_DIR | 0o755 } else { MODE_FILE | 0o744 }, seek: 0 }); @@ -130,6 +130,8 @@ impl Scheme for InitFsScheme { let handle = handles.get(&id).ok_or(Error::new(EBADF))?; stat.st_mode = handle.mode; + stat.st_uid = 0; + stat.st_gid = 0; stat.st_size = handle.data.len() as u64; Ok(0) diff --git a/scheme/irq.rs b/scheme/irq.rs index 7ec202ab..62fcb1af 100644 --- a/scheme/irq.rs +++ b/scheme/irq.rs @@ -7,7 +7,7 @@ use syscall::scheme::Scheme; pub struct IrqScheme; impl Scheme for IrqScheme { - fn open(&self, path: &[u8], _flags: usize) -> Result<usize> { + fn open(&self, path: &[u8], _flags: usize, _uid: u32, _gid: u32) -> Result<usize> { let path_str = str::from_utf8(path).or(Err(Error::new(ENOENT)))?; let id = path_str.parse::<usize>().or(Err(Error::new(ENOENT)))?; diff --git a/scheme/root.rs b/scheme/root.rs index e982c8f4..c088d6e2 100644 --- a/scheme/root.rs +++ b/scheme/root.rs @@ -25,7 +25,7 @@ impl RootScheme { } impl Scheme for RootScheme { - fn open(&self, path: &[u8], _flags: usize) -> Result<usize> { + fn open(&self, path: &[u8], _flags: usize, _uid: u32, _gid: u32) -> Result<usize> { let context = { let contexts = context::contexts(); let context = contexts.current().ok_or(Error::new(ESRCH))?; diff --git a/scheme/user.rs b/scheme/user.rs index 96ba87e7..542f01e6 100644 --- a/scheme/user.rs +++ b/scheme/user.rs @@ -218,7 +218,7 @@ impl UserScheme { } impl Scheme for UserScheme { - fn open(&self, path: &[u8], flags: usize) -> Result<usize> { + fn open(&self, path: &[u8], flags: usize, _uid: u32, _gid: u32) -> Result<usize> { let inner = self.inner.upgrade().ok_or(Error::new(ENODEV))?; let address = inner.capture(path)?; let result = inner.call(SYS_OPEN, address, path.len(), flags); @@ -226,15 +226,15 @@ impl Scheme for UserScheme { result } - fn mkdir(&self, path: &[u8], mode: usize) -> Result<usize> { + fn mkdir(&self, path: &[u8], mode: u16, _uid: u32, _gid: u32) -> Result<usize> { let inner = self.inner.upgrade().ok_or(Error::new(ENODEV))?; let address = inner.capture(path)?; - let result = inner.call(SYS_MKDIR, address, path.len(), mode); + let result = inner.call(SYS_MKDIR, address, path.len(), mode as usize); let _ = inner.release(address); result } - fn rmdir(&self, path: &[u8]) -> Result<usize> { + fn rmdir(&self, path: &[u8], _uid: u32, _gid: u32) -> Result<usize> { let inner = self.inner.upgrade().ok_or(Error::new(ENODEV))?; let address = inner.capture(path)?; let result = inner.call(SYS_RMDIR, address, path.len(), 0); @@ -242,7 +242,7 @@ impl Scheme for UserScheme { result } - fn unlink(&self, path: &[u8]) -> Result<usize> { + fn unlink(&self, path: &[u8], _uid: u32, _gid: u32) -> Result<usize> { let inner = self.inner.upgrade().ok_or(Error::new(ENODEV))?; let address = inner.capture(path)?; let result = inner.call(SYS_UNLINK, address, path.len(), 0); diff --git a/syscall/fs.rs b/syscall/fs.rs index adb690d4..1c969f67 100644 --- a/syscall/fs.rs +++ b/syscall/fs.rs @@ -2,7 +2,7 @@ use context; use scheme; -use syscall::data::{Packet, Stat}; +use syscall::data::Packet; use syscall::error::*; pub fn file_op(a: usize, fd: usize, c: usize, d: usize) -> Result<usize> { @@ -70,11 +70,11 @@ pub fn getcwd(buf: &mut [u8]) -> Result<usize> { /// Open syscall pub fn open(path: &[u8], flags: usize) -> Result<usize> { - let path_canon = { + let (path_canon, uid, gid) = { let contexts = context::contexts(); let context_lock = contexts.current().ok_or(Error::new(ESRCH))?; let context = context_lock.read(); - context.canonicalize(path) + (context.canonicalize(path), context.uid, context.gid) }; let mut parts = path_canon.splitn(2, |&b| b == b':'); @@ -88,7 +88,7 @@ pub fn open(path: &[u8], flags: usize) -> Result<usize> { let (scheme_id, scheme) = schemes.get_name(namespace).ok_or(Error::new(ENOENT))?; (scheme_id, scheme.clone()) }; - let file_id = scheme.open(reference_opt.unwrap_or(b""), flags)?; + let file_id = scheme.open(reference_opt.unwrap_or(b""), flags, uid, gid)?; (scheme_id, file_id) }; @@ -102,12 +102,12 @@ pub fn open(path: &[u8], flags: usize) -> Result<usize> { } /// mkdir syscall -pub fn mkdir(path: &[u8], mode: usize) -> Result<usize> { - let path_canon = { +pub fn mkdir(path: &[u8], mode: u16) -> Result<usize> { + let (path_canon, uid, gid) = { let contexts = context::contexts(); let context_lock = contexts.current().ok_or(Error::new(ESRCH))?; let context = context_lock.read(); - context.canonicalize(path) + (context.canonicalize(path), context.uid, context.gid) }; let mut parts = path_canon.splitn(2, |&b| b == b':'); @@ -120,16 +120,16 @@ pub fn mkdir(path: &[u8], mode: usize) -> Result<usize> { let (_scheme_id, scheme) = schemes.get_name(namespace).ok_or(Error::new(ENOENT))?; scheme.clone() }; - scheme.mkdir(reference_opt.unwrap_or(b""), mode) + scheme.mkdir(reference_opt.unwrap_or(b""), mode, uid, gid) } /// rmdir syscall pub fn rmdir(path: &[u8]) -> Result<usize> { - let path_canon = { + let (path_canon, uid, gid) = { let contexts = context::contexts(); let context_lock = contexts.current().ok_or(Error::new(ESRCH))?; let context = context_lock.read(); - context.canonicalize(path) + (context.canonicalize(path), context.uid, context.gid) }; let mut parts = path_canon.splitn(2, |&b| b == b':'); @@ -142,16 +142,16 @@ pub fn rmdir(path: &[u8]) -> Result<usize> { let (_scheme_id, scheme) = schemes.get_name(namespace).ok_or(Error::new(ENOENT))?; scheme.clone() }; - scheme.rmdir(reference_opt.unwrap_or(b"")) + scheme.rmdir(reference_opt.unwrap_or(b""), uid, gid) } /// Unlink syscall pub fn unlink(path: &[u8]) -> Result<usize> { - let path_canon = { + let (path_canon, uid, gid) = { let contexts = context::contexts(); let context_lock = contexts.current().ok_or(Error::new(ESRCH))?; let context = context_lock.read(); - context.canonicalize(path) + (context.canonicalize(path), context.uid, context.gid) }; let mut parts = path_canon.splitn(2, |&b| b == b':'); @@ -164,7 +164,7 @@ pub fn unlink(path: &[u8]) -> Result<usize> { let (_scheme_id, scheme) = schemes.get_name(namespace).ok_or(Error::new(ENOENT))?; scheme.clone() }; - scheme.unlink(reference_opt.unwrap_or(b"")) + scheme.unlink(reference_opt.unwrap_or(b""), uid, gid) } /// Close syscall diff --git a/syscall/mod.rs b/syscall/mod.rs index f10469e1..c0d389ef 100644 --- a/syscall/mod.rs +++ b/syscall/mod.rs @@ -37,6 +37,9 @@ pub extern fn syscall(a: usize, b: usize, c: usize, d: usize, e: usize, f: usize }, SYS_CLASS_PATH => match a { SYS_OPEN => open(validate_slice(b as *const u8, c)?, d), + SYS_MKDIR => mkdir(validate_slice(b as *const u8, c)?, d as u16), + SYS_RMDIR => rmdir(validate_slice(b as *const u8, c)?), + SYS_UNLINK => unlink(validate_slice(b as *const u8, c)?), _ => unreachable!() }, _ => match a { diff --git a/syscall/process.rs b/syscall/process.rs index 3821b355..9519ef2b 100644 --- a/syscall/process.rs +++ b/syscall/process.rs @@ -432,9 +432,33 @@ pub fn exec(path: &[u8], arg_ptrs: &[[usize; 2]]) -> Result<usize> { args.push(arg.to_vec()); // Must be moved into kernel space before exec unmaps all memory } + let (uid, gid) = { + let contexts = context::contexts(); + let context_lock = contexts.current().ok_or(Error::new(ESRCH))?; + let context = context_lock.read(); + (context.uid, context.gid) + }; + let file = syscall::open(path, 0)?; let mut stat = Stat::default(); syscall::file_op_mut_slice(syscall::number::SYS_FSTAT, file, &mut stat)?; + + let mut perm = stat.st_mode & 0o7; + if stat.st_uid == uid { + perm |= (stat.st_mode >> 6) & 0o7; + } + if stat.st_gid == gid { + perm |= (stat.st_mode >> 3) & 0o7; + } + if uid == 0 { + perm |= 0o7; + } + + if perm & 0o1 != 0o1 { + let _ = syscall::close(file); + return Err(Error::new(EACCES)); + } + //TODO: Only read elf header, not entire file. Then read required segments let mut data = vec![0; stat.st_size as usize]; syscall::file_op_mut_slice(syscall::number::SYS_READ, file, &mut data)?; @@ -458,6 +482,14 @@ pub fn exec(path: &[u8], arg_ptrs: &[[usize; 2]]) -> Result<usize> { drop(context.stack.take()); context.grants = Arc::new(Mutex::new(Vec::new())); + if stat.st_mode & syscall::flag::MODE_SETUID == syscall::flag::MODE_SETUID { + context.uid = stat.st_uid; + } + + if stat.st_mode & syscall::flag::MODE_SETGID == syscall::flag::MODE_SETGID { + context.gid = stat.st_gid; + } + // Map and copy new segments for segment in elf.segments() { if segment.p_type == program_header::PT_LOAD { -- GitLab