mirror of
https://github.com/uutils/coreutils
synced 2024-12-15 07:42:48 +00:00
199 lines
4.9 KiB
Rust
199 lines
4.9 KiB
Rust
#[crate_id(name="users", vers="1.0.0", author="KokaKiwi")];
|
|
|
|
/*
|
|
* This file is part of the uutils coreutils package.
|
|
*
|
|
* (c) KokaKiwi <kokakiwi@kokakiwi.net>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
/* last synced with: whoami (GNU coreutils) 8.22 */
|
|
|
|
// Allow dead code here in order to keep all fields, constants here, for consistency.
|
|
#[allow(dead_code)];
|
|
|
|
#[feature(macro_rules, globs)];
|
|
|
|
extern mod extra;
|
|
extern mod getopts;
|
|
|
|
use std::io::print;
|
|
use std::cast;
|
|
use std::libc;
|
|
use std::os;
|
|
use std::ptr;
|
|
use std::str;
|
|
use utmpx::*;
|
|
|
|
#[path = "../util.rs"]
|
|
mod util;
|
|
|
|
#[cfg(target_os = "linux")]
|
|
mod utmpx {
|
|
use std::libc;
|
|
|
|
pub static DEFAULT_FILE: &'static str = "/var/run/utmp";
|
|
|
|
pub static UT_LINESIZE: uint = 32;
|
|
pub static UT_NAMESIZE: uint = 32;
|
|
pub static UT_IDSIZE: uint = 4;
|
|
pub static UT_HOSTSIZE: uint = 256;
|
|
|
|
pub static EMPTY: libc::c_short = 0;
|
|
pub static RUN_LVL: libc::c_short = 1;
|
|
pub static BOOT_TIME: libc::c_short = 2;
|
|
pub static NEW_TIME: libc::c_short = 3;
|
|
pub static OLD_TIME: libc::c_short = 4;
|
|
pub static INIT_PROCESS: libc::c_short = 5;
|
|
pub static LOGIN_PROCESS: libc::c_short = 6;
|
|
pub static USER_PROCESS: libc::c_short = 7;
|
|
pub static DEAD_PROCESS: libc::c_short = 8;
|
|
pub static ACCOUNTING: libc::c_short = 9;
|
|
|
|
pub struct c_exit_status {
|
|
e_termination: libc::c_short,
|
|
e_exit: libc::c_short,
|
|
}
|
|
|
|
pub struct c_utmp {
|
|
ut_type: libc::c_short,
|
|
ut_pid: libc::pid_t,
|
|
ut_line: [libc::c_char, ..UT_LINESIZE],
|
|
ut_id: [libc::c_char, ..UT_IDSIZE],
|
|
|
|
ut_user: [libc::c_char, ..UT_NAMESIZE],
|
|
ut_host: [libc::c_char, ..UT_HOSTSIZE],
|
|
ut_exit: c_exit_status,
|
|
ut_session: libc::c_long,
|
|
ut_tv: libc::timeval,
|
|
|
|
ut_addr_v6: [libc::int32_t, ..4],
|
|
__unused: [libc::c_char, ..20],
|
|
}
|
|
}
|
|
|
|
#[cfg(target_os = "macos")]
|
|
mod utmpx {
|
|
use std::libc;
|
|
|
|
pub static DEFAULT_FILE: &'static str = "/var/run/utmpx";
|
|
|
|
pub static UT_LINESIZE: uint = 32;
|
|
pub static UT_NAMESIZE: uint = 256;
|
|
pub static UT_IDSIZE: uint = 4;
|
|
pub static UT_HOSTSIZE: uint = 256;
|
|
|
|
pub static EMPTY: libc::c_short = 0;
|
|
pub static RUN_LVL: libc::c_short = 1;
|
|
pub static BOOT_TIME: libc::c_short = 2;
|
|
pub static OLD_TIME: libc::c_short = 3;
|
|
pub static NEW_TIME: libc::c_short = 4;
|
|
pub static INIT_PROCESS: libc::c_short = 5;
|
|
pub static LOGIN_PROCESS: libc::c_short = 6;
|
|
pub static USER_PROCESS: libc::c_short = 7;
|
|
pub static DEAD_PROCESS: libc::c_short = 8;
|
|
pub static ACCOUNTING: libc::c_short = 9;
|
|
|
|
pub struct c_exit_status {
|
|
e_termination: libc::c_short,
|
|
e_exit: libc::c_short,
|
|
}
|
|
|
|
pub struct c_utmp {
|
|
ut_user: [libc::c_char, ..UT_NAMESIZE],
|
|
ut_id: [libc::c_char, ..UT_IDSIZE],
|
|
ut_line: [libc::c_char, ..UT_LINESIZE],
|
|
ut_pid: libc::pid_t,
|
|
ut_type: libc::c_short,
|
|
ut_tv: libc::timeval,
|
|
ut_host: [libc::c_char, ..UT_HOSTSIZE],
|
|
__unused: [libc::c_char, ..16]
|
|
}
|
|
}
|
|
|
|
extern {
|
|
fn getutxent() -> *c_utmp;
|
|
fn getutxid(ut: *c_utmp) -> *c_utmp;
|
|
fn getutxline(ut: *c_utmp) -> *c_utmp;
|
|
|
|
fn pututxline(ut: *c_utmp) -> *c_utmp;
|
|
|
|
fn setutxent();
|
|
fn endutxent();
|
|
|
|
fn utmpxname(file: *libc::c_char) -> libc::c_int;
|
|
}
|
|
|
|
static NAME: &'static str = "users";
|
|
|
|
fn main() {
|
|
let args = os::args();
|
|
let program = args[0].as_slice();
|
|
let opts = ~[
|
|
getopts::optflag("h", "help", "display this help and exit"),
|
|
getopts::optflag("V", "version", "output version information and exit"),
|
|
];
|
|
|
|
let matches = match getopts::getopts(args.tail(), opts) {
|
|
Ok(m) => m,
|
|
Err(f) => fail!(f.to_err_msg()),
|
|
};
|
|
|
|
if matches.opt_present("help") {
|
|
println!("users 1.0.0");
|
|
println!("");
|
|
println!("Usage:");
|
|
println!(" {:s} [OPTION]... [FILE]", program);
|
|
println!("");
|
|
print(getopts::usage("Output who is currently logged in according to FILE.", opts));
|
|
return;
|
|
}
|
|
|
|
if matches.opt_present("version") {
|
|
println!("users 1.0.0");
|
|
return;
|
|
}
|
|
|
|
let mut filename = DEFAULT_FILE;
|
|
if matches.free.len() > 0 {
|
|
filename = matches.free[0].as_slice();
|
|
}
|
|
|
|
exec(filename);
|
|
}
|
|
|
|
fn exec(filename: &str) {
|
|
filename.with_c_str(|filename| {
|
|
unsafe {
|
|
utmpxname(filename);
|
|
}
|
|
});
|
|
|
|
let mut users: ~[~str] = ~[];
|
|
|
|
unsafe {
|
|
setutxent();
|
|
|
|
loop {
|
|
let line = getutxent();
|
|
|
|
if line == ptr::null() {
|
|
break;
|
|
}
|
|
|
|
if (*line).ut_type == USER_PROCESS {
|
|
let user = str::raw::from_c_str(cast::transmute(&(*line).ut_user));
|
|
users.push(user);
|
|
}
|
|
}
|
|
|
|
endutxent();
|
|
}
|
|
|
|
if users.len() > 0 {
|
|
users.sort();
|
|
println!("{}", users.connect(" "));
|
|
}
|
|
}
|