diff options
Diffstat (limited to 'src/main.rs')
-rw-r--r-- | src/main.rs | 42 |
1 files changed, 7 insertions, 35 deletions
diff --git a/src/main.rs b/src/main.rs index e45a4cb..09c97e2 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,57 +1,29 @@ +mod util; + use clap::clap_app; use failure::{err_msg, format_err, Error}; use isahc::prelude::*; use serde::Deserialize; use std::ffi::CString; use std::ffi::{OsStr, OsString}; -use std::io::{self, stderr, stdin, BufRead, BufReader, ErrorKind, Read, Write}; +use std::io::{stderr, stdin, BufRead, BufReader, Read, Write}; use std::mem::MaybeUninit; use std::os::unix::prelude::*; use std::path::Path; use std::process::{exit, Command, ExitStatus, Stdio}; - use libc::{ - gethostname, posix_spawn_file_actions_adddup2, posix_spawn_file_actions_init, posix_spawnp, - waitpid, _SC_HOST_NAME_MAX, + posix_spawn_file_actions_adddup2, posix_spawn_file_actions_init, posix_spawnp, + waitpid }; +use self::util::*; + extern "C" { static mut environ: *mut *mut libc::c_char; } type Result<T> = std::result::Result<T, Error>; -fn copy_stream(src: &mut dyn Read, dst: &mut dyn Write) -> io::Result<()> { - let mut buf = vec![0; 65536]; - loop { - match src.read(&mut buf) { - Ok(0) => return Ok(()), - Ok(len) => dst.write_all(&buf[..len])?, - Err(e) if e.kind() == ErrorKind::Interrupted => continue, - Err(e) => return Err(e), - } - } -} - -fn hostname() -> OsString { - let mut bytes = vec![0; _SC_HOST_NAME_MAX as usize + 2]; - - if unsafe { gethostname(bytes.as_mut_ptr(), bytes.len()) } == -1 { - panic!("gethostname failed: {}", std::io::Error::last_os_error()); - } - - if bytes[_SC_HOST_NAME_MAX as usize + 1] != 0 { - panic!("hostname longer than HOST_NAME_MAX"); - } - - let bytes: Vec<_> = bytes - .into_iter() - .take_while(|b| *b != 0) - .map(|b| b as u8) - .collect(); - OsString::from_vec(bytes) -} - #[derive(Clone, Debug)] struct Config<'a> { from: &'a OsStr, |