From 557159a2e00508820cb7bad66822cc97c61b6de5 Mon Sep 17 00:00:00 2001
From: bnewbold <bnewbold@robocracy.org>
Date: Sun, 29 May 2016 20:31:09 -0400
Subject: begin implementation of actual file transfer

---
 src/server.rs | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'src/server.rs')

diff --git a/src/server.rs b/src/server.rs
index e60c054..4c590be 100644
--- a/src/server.rs
+++ b/src/server.rs
@@ -47,13 +47,14 @@ fn run_server(path: &str, is_recv: bool, recursive: bool) {
 
     let (mut socket, _src) = listener.accept().unwrap();
     println!("Got connection from {}", socket.peer_addr().unwrap());
+    let mut stream = socket.into();
 
     if is_recv {
-        common::receive_files(&mut socket, path, recursive);
+        common::sink_files(&mut stream, path, recursive);
     } else {
-        common::send_files(&mut socket, path, recursive);
+        common::source_files(&mut stream, path, recursive);
     }
-    socket.close().unwrap();
+    stream.close().unwrap();
 }
 
 fn usage_server(opts: Options) {
-- 
cgit v1.2.3