about summary refs log tree commit diff
path: root/src/main.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/main.rs')
-rw-r--r--src/main.rs42
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,