transport 0.1.0

Transport abstraction library with nat traversal
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

use std::time::Duration;
use std::io::{Read, Write};

use crossbeam;
use rand;

use stream::Stream;

pub fn timebomb<R, F>(dur: Duration, f: F) -> R
    where R: Send,
          F: Send + FnOnce() -> R
{
    use std::thread;
    use std::sync::mpsc;

    use void::Void;

    crossbeam::scope(|scope| {
        let thread_handle = thread::current();
        let (done_tx, done_rx) = mpsc::channel::<Void>();
        let jh = scope.spawn(move || {
            let ret = f();
            drop(done_tx);
            thread_handle.unpark();
            ret
        });
        thread::park_timeout(dur);
        match done_rx.try_recv() {
            Ok(x) => match x {},
            Err(mpsc::TryRecvError::Empty) => panic!("Timed out!"),
            Err(mpsc::TryRecvError::Disconnected) => jh.join(),
        }
    })
}

pub fn check_stream(stream: &mut Stream) {
    let send_data: [u8; 8] = rand::random();
    unwrap_result!(stream.write_all(&send_data[..]));

    let mut recv_data = [0u8; 8];
    unwrap_result!(stream.read_exact(&mut recv_data[..]));
    assert_eq!(send_data, recv_data);
}

pub fn bounce_stream(stream: &mut Stream) {
    let mut data = [0u8; 8];
    unwrap_result!(stream.read_exact(&mut data[..]));
    unwrap_result!(stream.write_all(&data[..]));
}