mcai_ftp 3.0.1

FTP client for Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

#[cfg(feature = "secure")]
use openssl::ssl::SslStream;
use std::io::{Read, Result, Write};
use std::net::TcpStream;

/// Data Stream used for communications
#[derive(Debug)]
pub enum DataStream {
  Tcp(TcpStream),
  #[cfg(feature = "secure")]
  Ssl(SslStream<TcpStream>),
}

#[cfg(feature = "secure")]
impl DataStream {
  /// Unwrap the stream into TcpStream. This method is only used in secure connection.
  pub fn into_tcp_stream(self) -> TcpStream {
    match self {
      DataStream::Tcp(stream) => stream,
      DataStream::Ssl(stream) => stream.get_ref().try_clone().unwrap(),
    }
  }

  /// Test if the stream is secured
  pub fn is_ssl(&self) -> bool {
    matches!(*self, DataStream::Ssl(_))
  }
}

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 {
      DataStream::Tcp(ref mut stream) => stream.read(buf),
      #[cfg(feature = "secure")]
      DataStream::Ssl(ref mut stream) => stream.read(buf),
    }
  }
}

impl Write for DataStream {
  fn write(&mut self, buf: &[u8]) -> Result<usize> {
    match *self {
      DataStream::Tcp(ref mut stream) => stream.write(buf),
      #[cfg(feature = "secure")]
      DataStream::Ssl(ref mut stream) => stream.write(buf),
    }
  }

  fn flush(&mut self) -> Result<()> {
    match *self {
      DataStream::Tcp(ref mut stream) => stream.flush(),
      #[cfg(feature = "secure")]
      DataStream::Ssl(ref mut stream) => stream.flush(),
    }
  }
}