Fix #1 '"bind_any" doesn't respect '--addr' argument

Author: 2dav

src/handler.rs

@@ -27,10 +27,10 @@ fn init_txc(dll_path: PathBuf, log_dir: PathBuf) -> ah::Result<libtxc::Module> {
     Ok(lib)
 }
 
-fn init_upstream(con: &mut net::TcpStream) -> ah::Result<()> {
+fn init_upstream(con: &mut net::TcpStream, local_addr: net::IpAddr) -> ah::Result<()> {
     // open 'data' server
-    let (data_server, data_port) =
-        ws2::bind_any().context("Не удалось зарегистрировать сервер трансляции данных")?;
+    let (data_server, data_port) = ws2::bind_any(local_addr)
+        .context("Не удалось зарегистрировать сервер трансляции данных")?;
 
     // send 'data' server port to client and await for connection
     con.write_all(&data_port.to_ne_bytes())?;
@@ -42,8 +42,10 @@ fn init_upstream(con: &mut net::TcpStream) -> ah::Result<()> {
     Ok(())
 }
 
-pub fn handler(mut con: net::TcpStream, dll_path: PathBuf, log_dir: PathBuf) -> ah::Result<()> {
-    init_upstream(&mut con)?;
+pub fn handler(
+    (mut con, dll_path, log_dir, local_addr): (net::TcpStream, PathBuf, PathBuf, net::IpAddr),
+) -> ah::Result<()> {
+    init_upstream(&mut con, local_addr)?;
     let lib = init_txc(dll_path, log_dir)?;
 
     // loop

src/lib.rs

@@ -6,7 +6,7 @@ pub use handler::handler;
 
 use std::{
     env, io,
-    net::{SocketAddr, TcpListener, TcpStream},
+    net::{IpAddr, SocketAddr, TcpListener, TcpStream},
     path::PathBuf,
     process::{Command, Stdio},
 };
@@ -64,8 +64,8 @@ pub fn txc_log_level() -> i32 {
     }
 }
 
-pub fn read_handler_params() -> ah::Result<(TcpStream, PathBuf, PathBuf)> {
-    // 'master' process ensures that the handler 'fork' is passed the 'dll_path' and 'log_dir'
+pub fn read_handler_params() -> ah::Result<(TcpStream, PathBuf, PathBuf, IpAddr)> {
+    // 'master' process ensures that the handler 'fork' is run with the 'dll_path', 'log_dir' and 'addr'
     // as positional arguments, and client socket handle is written to 'stdin'
 
     // init winsock2
@@ -75,7 +75,9 @@ pub fn read_handler_params() -> ah::Result<(TcpStream, PathBuf, PathBuf)> {
         .context("Не удалось получить дескриптор подключения через 'stdin'")?;
 
     match env::args().collect::<Vec<String>>().as_slice() {
-        [.., log_dir, dll_path] => Ok((con, log_dir.into(), dll_path.into())),
+        [.., log_dir, dll_path, addr] => {
+            Ok((con, log_dir.into(), dll_path.into(), addr.parse().expect("infallible")))
+        }
         _ => unreachable!("unexpected number of arguments"),
     }
 }
@@ -100,6 +102,7 @@ pub fn master<A: Into<SocketAddr>>(addr: A, log_dir: PathBuf, dll_path: PathBuf)
             .stdout(Stdio::null())
             .arg(dll_path.to_string_lossy().to_string())
             .arg(log_dir.to_string_lossy().to_string())
+            .arg(addr.ip().to_string())
             .spawn()
             .context("Ошибка запуска обработчика клиентcкого подключения")?;
 

src/main.rs

@@ -44,8 +44,7 @@ fn main() -> ah::Result<()> {
         }
         Role::Handler => read_handler_params()
             .context("Ошибка инициализации обработчика подключения")
-            .and_then(|(con, dll_path, log_dir)| {
-                handler(con, dll_path, log_dir).context("Ошибка обработки подключения")
-            }),
+            .and_then(handler)
+            .context("Ошибка обработки подключения"),
     }
 }

src/ws2.rs

@@ -53,11 +53,10 @@ pub fn write_pinfo<W: Write>(mut writer: W, info: WSAPROTOCOL_INFOW) -> ah::Resu
     }
 }
 
-pub fn bind_any() -> io::Result<(net::TcpListener, u16)> {
+pub fn bind_any<A: Into<Option<net::IpAddr>>>(addr: A) -> io::Result<(net::TcpListener, u16)> {
+    let addr = addr.into().unwrap_or([127, 0, 0, 1].into());
     for port in 4243..65535 {
-        if let Ok(listener) =
-            net::TcpListener::bind(net::SocketAddrV4::new([127, 0, 0, 1].into(), port))
-        {
+        if let Ok(listener) = net::TcpListener::bind(net::SocketAddr::new(addr, port)) {
             return Ok((listener, port));
         }
     }