mirror of
https://github.com/librespot-org/librespot.git
synced 2024-12-18 17:11:53 +00:00
try reading password from env variable first, so it doesn't appear in process listing
This commit is contained in:
parent
525b27df98
commit
7f8e85f90b
1 changed files with 21 additions and 5 deletions
22
src/main.rs
22
src/main.rs
|
@ -8,6 +8,7 @@ use std::io::{stdout, Read, Write};
|
||||||
use std::path::Path;
|
use std::path::Path;
|
||||||
use std::thread;
|
use std::thread;
|
||||||
use std::path::PathBuf;
|
use std::path::PathBuf;
|
||||||
|
use std::env;
|
||||||
|
|
||||||
use getopts::Options;
|
use getopts::Options;
|
||||||
use rpassword::read_password;
|
use rpassword::read_password;
|
||||||
|
@ -17,6 +18,8 @@ use librespot::util::version::version_string;
|
||||||
use librespot::player::Player;
|
use librespot::player::Player;
|
||||||
use librespot::spirc::SpircManager;
|
use librespot::spirc::SpircManager;
|
||||||
|
|
||||||
|
static PASSWORD_ENV_NAME: &'static str = "SPOTIFY_PASSWORD";
|
||||||
|
|
||||||
fn usage(program: &str, opts: &Options) -> String {
|
fn usage(program: &str, opts: &Options) -> String {
|
||||||
let brief = format!("Usage: {} [options]", program);
|
let brief = format!("Usage: {} [options]", program);
|
||||||
format!("{}", opts.usage(&brief))
|
format!("{}", opts.usage(&brief))
|
||||||
|
@ -48,11 +51,24 @@ fn main() {
|
||||||
let cache_location = matches.opt_str("c").unwrap();
|
let cache_location = matches.opt_str("c").unwrap();
|
||||||
let name = matches.opt_str("n").unwrap();
|
let name = matches.opt_str("n").unwrap();
|
||||||
|
|
||||||
let password = matches.opt_str("p").unwrap_or_else(|| {
|
let password: String = match env::var(PASSWORD_ENV_NAME) {
|
||||||
print!("Password: ");
|
Ok(val) => {
|
||||||
|
// unset password so e.g. child process can't leak it; but still appears in /proc/$PID/environ
|
||||||
|
env::remove_var(PASSWORD_ENV_NAME);
|
||||||
|
//assert!(env::var(PASSWORD_ENV_NAME).is_err());
|
||||||
|
val
|
||||||
|
},
|
||||||
|
Err(_) => {
|
||||||
|
match matches.opt_str("p") {
|
||||||
|
Some(val) => val,
|
||||||
|
None => {
|
||||||
|
print!("Password not found in env var {} or param `-p`, please enter: ", PASSWORD_ENV_NAME);
|
||||||
stdout().flush().unwrap();
|
stdout().flush().unwrap();
|
||||||
read_password().unwrap()
|
read_password().unwrap()
|
||||||
});
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
let mut appkey = Vec::new();
|
let mut appkey = Vec::new();
|
||||||
appkey_file.read_to_end(&mut appkey).unwrap();
|
appkey_file.read_to_end(&mut appkey).unwrap();
|
||||||
|
|
Loading…
Reference in a new issue