RGBCube/uutils-coreutils
1
Fork 0
mirror of https://github.com/RGBCube/uutils-coreutils synced 2025-07-29 12:07:46 +00:00
Code Issues Activity

uucore::process: add functions to obtain uid/gid, euid/egid

Browse source
This commit is contained in:
Knight 2016-08-20 02:12:19 +08:00
parent 369c2c2d83
commit b6c864b7f3
1 changed files with 39 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

58
src/uucore/process.rs
View file

@ -1,15 +1,14 @@
/* // This file is part of the uutils coreutils package.
* This file is part of the uutils coreutils package. //
* // (c) Maciej Dziardziel <fiedzia@gmail.com>
* (c) Maciej Dziardziel <fiedzia@gmail.com> // (c) Jian Zeng <anonymousknight96 AT gmail.com>
* //
* For the full copyright and license information, please view the LICENSE file // For the full copyright and license information, please view the LICENSE file
* that was distributed with this source code. // that was distributed with this source code.
*/ //
extern crate libc; use super::libc;
use libc::{c_int, pid_t, uid_t, gid_t};
use libc::{c_int, pid_t};
use std::fmt; use std::fmt;
use std::io; use std::io;
use std::process::Child; use std::process::Child;
@ -17,6 +16,30 @@ use std::sync::{Arc, Condvar, Mutex};
use std::thread; use std::thread;
use std::time::{Duration, Instant}; use std::time::{Duration, Instant};
pub fn geteuid() -> uid_t {
unsafe {
libc::geteuid()
}
}
pub fn getegid() -> gid_t {
unsafe {
libc::getegid()
}
}
pub fn getgid() -> gid_t {
unsafe {
libc::getgid()
}
}
pub fn getuid() -> uid_t {
unsafe {
libc::getuid()
}
}
// This is basically sys::unix::process::ExitStatus // This is basically sys::unix::process::ExitStatus
#[derive(PartialEq, Eq, Clone, Copy, Debug)] #[derive(PartialEq, Eq, Clone, Copy, Debug)]
pub enum ExitStatus { pub enum ExitStatus {
@ -26,7 +49,8 @@ pub enum ExitStatus {
impl ExitStatus { impl ExitStatus {
fn from_status(status: c_int) -> ExitStatus { fn from_status(status: c_int) -> ExitStatus {
if status & 0x7F != 0 { // WIFSIGNALED(status) if status & 0x7F != 0 {
// WIFSIGNALED(status)
ExitStatus::Signal(status & 0x7F) ExitStatus::Signal(status & 0x7F)
} else { } else {
ExitStatus::Code(status & 0xFF00 >> 8) ExitStatus::Code(status & 0xFF00 >> 8)
@ -75,8 +99,7 @@ pub trait ChildExt {
impl ChildExt for Child { impl ChildExt for Child {
fn send_signal(&mut self, signal: usize) -> io::Result<()> { fn send_signal(&mut self, signal: usize) -> io::Result<()> {
if unsafe { libc::kill(self.id() as pid_t, if unsafe { libc::kill(self.id() as pid_t, signal as i32) } != 0 {
signal as i32) } != 0 {
Err(io::Error::last_os_error()) Err(io::Error::last_os_error())
} else { } else {
Ok(()) Ok(())
@ -86,10 +109,7 @@ impl ChildExt for Child {
fn wait_or_timeout(&mut self, timeout: Duration) -> io::Result<Option<ExitStatus>> { fn wait_or_timeout(&mut self, timeout: Duration) -> io::Result<Option<ExitStatus>> {
// The result will be written to that Option, protected by a Mutex // The result will be written to that Option, protected by a Mutex
// Then the Condvar will be signaled // Then the Condvar will be signaled
let state = Arc::new(( let state = Arc::new((Mutex::new(Option::None::<io::Result<ExitStatus>>), Condvar::new()));
Mutex::new(Option::None::<io::Result<ExitStatus>>),
Condvar::new(),
));
// Start the waiting thread // Start the waiting thread
let state_th = state.clone(); let state_th = state.clone();
@ -121,7 +141,7 @@ impl ChildExt for Child {
return exitstatus.map(Some); return exitstatus.map(Some);
} }
if start.elapsed() >= timeout { if start.elapsed() >= timeout {
return Ok(None) return Ok(None);
} }
let cvar_timeout = timeout - start.elapsed(); let cvar_timeout = timeout - start.elapsed();
exitstatus = cvar.wait_timeout(exitstatus, cvar_timeout).unwrap().0; exitstatus = cvar.wait_timeout(exitstatus, cvar_timeout).unwrap().0;