added get_ref() method to allow setting read/write timeout, see #52 (#58)

Author: matt2xu

src/data_stream.rs

@@ -31,6 +31,17 @@ impl DataStream {
     }
 }
 
+impl DataStream {
+    /// Returns a reference to the underlying TcpStream.
+    pub fn get_ref(&self) -> &TcpStream {
+        match self {
+            &DataStream::Tcp(ref stream) => stream,
+            #[cfg(feature = "secure")]
+            &DataStream::Ssl(ref stream) => stream.get_ref(),
+        }
+    }
+}
+
 impl Read for DataStream {
     fn read(&mut self, buf: &mut [u8]) -> Result<usize> {
         match self {

src/ftp.rs

@@ -164,6 +164,21 @@ impl FtpStream {
             })
     }
 
+    /// Returns a reference to the underlying TcpStream.
+    ///
+    /// Example:
+    /// ```no_run
+    /// use std::net::TcpStream;
+    ///
+    /// let stream = FtpStream::connect("127.0.0.1:21")
+    ///                        .expect("Couldn't connect to the server...");
+    /// stream.get_ref().set_read_timeout(Duration::from_secs(10))
+    ///                 .expect("set_read_timeout call failed");
+    /// ```
+    pub fn get_ref(&self) -> &TcpStream {
+        self.reader.get_ref().get_ref()
+    }
+
     /// Log in to the FTP server.
     pub fn login(&mut self, user: &str, password: &str) -> Result<()> {
         let user_command = format!("USER {}\r\n", user);