vinzenz/redox
0
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Split login into getty and login, like other UNIXen

Browse source
This commit is contained in:
Jeremy Soller 2016-09-29 15:32:32 -06:00
parent 5a99ba6863
commit cd47a60595
9 changed files with 133 additions and 75 deletions
Download patch file Download diff file Expand all files Collapse all files

6
programs/getty/Cargo.toml Normal file
View file

@ -0,0 +1,6 @@
[package]
name = "getty"
version = "0.1.0"
[dependencies]
syscall = { path = "../../syscall/" }

52
programs/getty/src/main.rs Normal file
View file

@ -0,0 +1,52 @@
#![feature(question_mark)]
extern crate syscall;
use std::fs::File;
use std::io::{Read, Write};
use std::process::Command;
use std::{env, io, str, thread};
pub fn main() {
let mut args = env::args().skip(1);
let tty = args.next().expect("getty: no tty provided");
let _ = syscall::close(2);
let _ = syscall::close(1);
let _ = syscall::close(0);
let _ = syscall::open(&tty, syscall::flag::O_RDWR);
let _ = syscall::open(&tty, syscall::flag::O_RDWR);
let _ = syscall::open(&tty, syscall::flag::O_RDWR);
env::set_current_dir("file:").unwrap();
env::set_var("TTY", &tty);
{
let mut path = [0; 4096];
if let Ok(count) = syscall::fpath(0, &mut path) {
let path_str = str::from_utf8(&path[..count]).unwrap_or("");
let reference = path_str.split(':').nth(1).unwrap_or("");
let mut parts = reference.split('/');
env::set_var("COLUMNS", parts.next().unwrap_or("80"));
env::set_var("LINES", parts.next().unwrap_or("30"));
}
}
thread::spawn(move || {
let stdout = io::stdout();
let mut stdout = stdout.lock();
loop {
stdout.write(b"\x1Bc").unwrap();
let _ = stdout.flush();
match Command::new("file:bin/login").spawn() {
Ok(mut child) => match child.wait() {
Ok(_status) => (), //println!("getty: waited for login: {:?}", status.code()),
Err(err) => panic!("getty: failed to wait for login: {}", err)
},
Err(err) => panic!("getty: failed to execute login: {}", err)
}
}
});
}

1
programs/login/Cargo.toml
View file

@ -3,7 +3,6 @@ name = "login"
version = "0.1.0"
[dependencies]
syscall = { path = "../../syscall/" }
termion = "*"
[dependencies.octavo]

133
programs/login/src/main.rs
View file

@ -1,7 +1,6 @@
#![feature(question_mark)]
extern crate octavo;
extern crate syscall;
extern crate termion;
use octavo::octavo_digest::Digest;
@ -47,48 +46,43 @@ impl<'a> Passwd<'a> {
}
pub fn main() {
let mut args = env::args().skip(1);
let stdin = io::stdin();
let mut stdin = stdin.lock();
let stdout = io::stdout();
let mut stdout = stdout.lock();
let tty = args.next().expect("login: no tty provided");
let _ = syscall::close(2);
let _ = syscall::close(1);
let _ = syscall::close(0);
let _ = syscall::open(&tty, syscall::flag::O_RDWR);
let _ = syscall::open(&tty, syscall::flag::O_RDWR);
let _ = syscall::open(&tty, syscall::flag::O_RDWR);
env::set_current_dir("file:").unwrap();
env::set_var("TTY", &tty);
{
let mut path = [0; 4096];
if let Ok(count) = syscall::fpath(0, &mut path) {
let path_str = str::from_utf8(&path[..count]).unwrap_or("");
let reference = path_str.split(':').nth(1).unwrap_or("");
let mut parts = reference.split('/');
env::set_var("COLUMNS", parts.next().unwrap_or("80"));
env::set_var("LINES", parts.next().unwrap_or("30"));
}
if let Ok(mut issue) = File::open("/etc/issue") {
io::copy(&mut issue, &mut stdout).unwrap();
let _ = stdout.flush();
}
thread::spawn(move || {
let stdin = io::stdin();
let mut stdin = stdin.lock();
let stdout = io::stdout();
let mut stdout = stdout.lock();
loop {
stdout.write_all(b"\x1B[1mredox login:\x1B[0m ").unwrap();
let _ = stdout.flush();
loop {
stdout.write_all(b"\x1B[1mredox login:\x1B[0m ").unwrap();
let _ = stdout.flush();
let user = (&mut stdin as &mut Read).read_line().unwrap().unwrap_or(String::new());
if ! user.is_empty() {
let mut passwd_string = String::new();
File::open("file:etc/passwd").unwrap().read_to_string(&mut passwd_string).unwrap();
let user = (&mut stdin as &mut Read).read_line().unwrap().unwrap_or(String::new());
if ! user.is_empty() {
let mut passwd_option = None;
for line in passwd_string.lines() {
if let Ok(passwd) = Passwd::parse(line) {
if user == passwd.user && "" == passwd.hash {
passwd_option = Some(passwd);
break;
}
}
}
if passwd_option.is_none() {
stdout.write_all(b"\x1B[1mpassword:\x1B[0m ").unwrap();
let _ = stdout.flush();
if let Some(password) = stdin.read_passwd(&mut stdout).unwrap() {
stdout.write(b"\n").unwrap();
let _ = stdout.flush();
let password_hash = {
let mut output = vec![0; Sha512::output_bytes()];
let mut hash = Sha512::default();
@ -101,15 +95,10 @@ pub fn main() {
encoded
};
{
let mut debug = File::open("debug:").unwrap();
write!(debug, "{};{}\n", user, password_hash).unwrap();
if let Ok(mut debug) = File::open("debug:") {
let _ = write!(debug, "{};{}\n", user, password_hash);
}
let mut passwd_string = String::new();
File::open("file:etc/passwd").unwrap().read_to_string(&mut passwd_string).unwrap();
let mut passwd_option = None;
for line in passwd_string.lines() {
if let Ok(passwd) = Passwd::parse(line) {
if user == passwd.user && password_hash == passwd.hash {
@ -118,37 +107,39 @@ pub fn main() {
}
}
}
if let Some(passwd) = passwd_option {
stdout.write(b"\n").unwrap();
let _ = stdout.flush();
let mut command = Command::new(passwd.shell);
env::set_current_dir(passwd.home).unwrap();
command.env("USER", &user);
command.env("HOME", passwd.home);
command.env("PATH", "file:bin");
match command.spawn() {
Ok(mut child) => match child.wait() {
Ok(_status) => (), //println!("login: waited for {}: {:?}", sh, status.code()),
Err(err) => panic!("login: failed to wait for '{}': {}", passwd.shell, err)
},
Err(err) => panic!("login: failed to execute '{}': {}", passwd.shell, err)
}
env::set_current_dir("file:").unwrap();
stdout.write(b"\x1Bc").unwrap();
let _ = stdout.flush();
} else {
stdout.write(b"\nLogin failed\n").unwrap();
let _ = stdout.flush();
}
}
}
if let Some(passwd) = passwd_option {
if let Ok(mut motd) = File::open("/etc/motd") {
io::copy(&mut motd, &mut stdout).unwrap();
let _ = stdout.flush();
}
let mut command = Command::new(passwd.shell);
env::set_current_dir(passwd.home).unwrap();
command.env("USER", &user);
command.env("HOME", passwd.home);
command.env("PATH", "file:bin");
match command.spawn() {
Ok(mut child) => match child.wait() {
Ok(_status) => (), //println!("login: waited for {}: {:?}", sh, status.code()),
Err(err) => panic!("login: failed to wait for '{}': {}", passwd.shell, err)
},
Err(err) => panic!("login: failed to execute '{}': {}", passwd.shell, err)
}
break;
} else {
stdout.write(b"\nLogin failed\n").unwrap();
let _ = stdout.flush();
}
} else {
stdout.write(b"\n").unwrap();
let _ = stdout.flush();
}
});
}
}