#![crate_id(name="users", vers="1.0.0", author="KokaKiwi")] /* * This file is part of the uutils coreutils package. * * (c) KokaKiwi * * 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, non_camel_case_types)] #![feature(macro_rules, globs)] extern crate getopts; extern crate libc; use std::io::print; use std::mem; use std::os; use std::ptr; use std::str; use utmpx::*; #[path = "../common/util.rs"] mod util; #[path = "../common/utmpx.rs"] mod utmpx; 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() { uumain(os::args()); } pub fn uumain(args: Vec) { let program = args.get(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).as_slice()); 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.get(0).as_slice(); } exec(filename); } fn exec(filename: &str) { filename.with_c_str(|filename| { unsafe { utmpxname(filename); } }); let mut users = vec!(); unsafe { setutxent(); loop { let line = getutxent(); if line == ptr::null() { break; } if (*line).ut_type == USER_PROCESS { let user = str::raw::from_c_str(mem::transmute(&(*line).ut_user)); users.push(user); } } endutxent(); } if users.len() > 0 { users.sort(); println!("{}", users.connect(" ")); } }