2014-03-31 16:40:21 +00:00
|
|
|
#![crate_id(name="whoami", version="1.0.0", author="KokaKiwi")]
|
2013-10-18 12:00:38 +00:00
|
|
|
|
|
|
|
/*
|
|
|
|
* This file is part of the uutils coreutils package.
|
|
|
|
*
|
|
|
|
* (c) Jordi Boggiano <j.boggiano@seld.be>
|
|
|
|
*
|
|
|
|
* 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.21 */
|
|
|
|
|
2014-03-31 16:40:21 +00:00
|
|
|
#![allow(non_camel_case_types)]
|
2014-02-23 21:31:51 +00:00
|
|
|
|
2014-03-31 16:40:21 +00:00
|
|
|
#![feature(macro_rules)]
|
2014-02-07 06:39:07 +00:00
|
|
|
|
2014-02-16 21:29:31 +00:00
|
|
|
extern crate getopts;
|
2014-04-26 05:03:08 +00:00
|
|
|
extern crate libc;
|
2013-10-18 12:00:38 +00:00
|
|
|
|
2014-01-13 09:05:02 +00:00
|
|
|
use std::io::print;
|
2013-10-18 12:00:38 +00:00
|
|
|
use std::os;
|
2014-02-07 06:39:07 +00:00
|
|
|
|
2014-02-24 05:55:01 +00:00
|
|
|
#[path = "../common/util.rs"] mod util;
|
2013-10-18 12:00:38 +00:00
|
|
|
|
2014-06-15 20:37:58 +00:00
|
|
|
#[cfg(unix)]
|
|
|
|
mod platform {
|
|
|
|
use super::libc;
|
|
|
|
use std::str;
|
|
|
|
use self::c_types::{c_passwd, getpwuid};
|
|
|
|
|
|
|
|
#[path = "../../common/c_types.rs"] mod c_types;
|
|
|
|
|
|
|
|
extern {
|
2014-06-16 20:12:36 +00:00
|
|
|
pub fn geteuid() -> libc::uid_t;
|
2014-06-15 20:37:58 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
pub unsafe fn getusername() -> String {
|
|
|
|
let passwd: *c_passwd = getpwuid(geteuid());
|
|
|
|
|
|
|
|
let pw_name: *libc::c_char = (*passwd).pw_name;
|
|
|
|
let name = str::raw::from_c_str(pw_name);
|
|
|
|
|
|
|
|
name
|
|
|
|
}
|
2013-10-18 12:00:38 +00:00
|
|
|
}
|
|
|
|
|
2014-06-15 20:37:58 +00:00
|
|
|
#[cfg(windows)]
|
|
|
|
mod platform {
|
|
|
|
pub use super::libc;
|
|
|
|
use std::mem;
|
|
|
|
use std::str;
|
2013-10-18 12:00:38 +00:00
|
|
|
|
2014-06-15 20:37:58 +00:00
|
|
|
extern "system" {
|
2014-06-15 21:42:58 +00:00
|
|
|
pub fn GetUserNameA(out: *libc::c_char, len: *libc::uint32_t) -> libc::uint8_t;
|
2014-06-15 20:37:58 +00:00
|
|
|
}
|
2013-10-18 12:00:38 +00:00
|
|
|
|
2014-06-15 20:37:58 +00:00
|
|
|
#[allow(unused_unsafe)]
|
|
|
|
pub unsafe fn getusername() -> String {
|
|
|
|
let buffer: [libc::c_char, ..2048] = mem::uninitialized(); // XXX: it may be possible that this isn't long enough. I don't know
|
2014-06-15 21:42:58 +00:00
|
|
|
if !GetUserNameA(buffer.as_ptr(), &(buffer.len() as libc::uint32_t)) == 0 {
|
2014-06-15 20:37:58 +00:00
|
|
|
crash!(1, "username is too long");
|
|
|
|
}
|
|
|
|
str::raw::from_c_str(buffer.as_ptr())
|
|
|
|
}
|
2013-10-18 12:00:38 +00:00
|
|
|
}
|
|
|
|
|
2014-02-07 06:39:07 +00:00
|
|
|
static NAME: &'static str = "whoami";
|
|
|
|
|
2014-05-28 12:01:30 +00:00
|
|
|
#[allow(dead_code)]
|
2014-06-08 07:56:37 +00:00
|
|
|
fn main() { os::set_exit_status(uumain(os::args())); }
|
2014-05-28 11:43:37 +00:00
|
|
|
|
2014-06-08 07:56:37 +00:00
|
|
|
pub fn uumain(args: Vec<String>) -> int {
|
2014-05-16 08:32:58 +00:00
|
|
|
let program = args.get(0).as_slice();
|
2014-05-30 08:35:54 +00:00
|
|
|
let opts = [
|
2014-02-07 06:39:07 +00:00
|
|
|
getopts::optflag("h", "help", "display this help and exit"),
|
|
|
|
getopts::optflag("V", "version", "output version information and exit"),
|
2013-10-18 12:00:38 +00:00
|
|
|
];
|
2014-02-07 06:39:07 +00:00
|
|
|
let matches = match getopts::getopts(args.tail(), opts) {
|
2013-10-18 12:00:38 +00:00
|
|
|
Ok(m) => m,
|
2014-06-15 10:50:40 +00:00
|
|
|
Err(f) => crash!(1, "{}", f),
|
2013-10-18 12:00:38 +00:00
|
|
|
};
|
2013-10-22 12:22:10 +00:00
|
|
|
if matches.opt_present("help") {
|
2014-01-13 09:05:02 +00:00
|
|
|
println!("whoami 1.0.0");
|
|
|
|
println!("");
|
|
|
|
println!("Usage:");
|
2013-10-18 12:00:38 +00:00
|
|
|
println!(" {:s}", program);
|
2014-01-13 09:05:02 +00:00
|
|
|
println!("");
|
2014-05-17 10:32:14 +00:00
|
|
|
print(getopts::usage("print effective userid", opts).as_slice());
|
2014-06-08 07:56:37 +00:00
|
|
|
return 0;
|
2013-10-18 12:00:38 +00:00
|
|
|
}
|
2013-10-22 12:22:10 +00:00
|
|
|
if matches.opt_present("version") {
|
2014-01-13 09:05:02 +00:00
|
|
|
println!("whoami 1.0.0");
|
2014-06-08 07:56:37 +00:00
|
|
|
return 0;
|
2013-10-18 12:00:38 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
exec();
|
2014-06-08 07:56:37 +00:00
|
|
|
|
2014-06-12 04:41:53 +00:00
|
|
|
0
|
2013-10-18 12:00:38 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
pub fn exec() {
|
|
|
|
unsafe {
|
2014-06-15 20:37:58 +00:00
|
|
|
let username = platform::getusername();
|
2013-10-18 12:00:38 +00:00
|
|
|
println!("{:s}", username);
|
|
|
|
}
|
|
|
|
}
|