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

/*

Copyright 2017 Jinjing Wang

This file is part of mtcp.

mtcp is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

mtcp is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with mtcp.  If not, see <http://www.gnu.org/licenses/>.

*/


extern crate time;

use std::time::{Duration};

use std::result::Result::*;
use std::net::TcpStream;
use core::core::ServerConnection;

pub fn connect_server(
                      stream: TcpStream,
                      timeout: Option<Duration>)
                      -> Option<ServerConnection> {

    let tcp_stream_id = format!("{:#?}", stream);
    let tcp_input_stream = stream;

    match tcp_input_stream.try_clone() {
        Ok(tcp_output_stream) => {
            match tcp_input_stream.set_read_timeout(timeout)
                .and(tcp_output_stream.
                     set_write_timeout(timeout)) {
                    Ok(_) => {
                        let server_connection =
                            ServerConnection {
                                tcp_input_stream: tcp_input_stream,
                                tcp_output_stream: tcp_output_stream,
                            };
                        return Some(server_connection)
                    }
                    Err(err) => {
                        error!("failed to set timeout for: {}, err: {:#?}",
                               tcp_stream_id,
                               err);
                    }
                }

        }
        _ => ()
    }

    None
}