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
68
69
70
71
72
73
74
75
|
//! An UDP echo server that just sends back everything that it receives.
extern crate futures;
#[macro_use]
extern crate tokio_core;
use std::{env, io};
use std::net::SocketAddr;
use futures::{Future, Poll};
use tokio_core::net::UdpSocket;
use tokio_core::reactor::Core;
/// UDP echo server
struct Server {
socket : UdpSocket,
buf : Vec<u8>,
to_send : Option<(usize, SocketAddr)>,
}
impl Server {
fn new(s : UdpSocket) -> Self {
Server {
socket: s,
to_send: None,
buf: vec![0u8; 1600],
}
}
}
impl Future for Server {
type Item = ();
type Error = io::Error;
fn poll(&mut self) -> Poll<(), io::Error> {
loop {
if let Some((size, peer)) = self.to_send.take() {
match self.socket.send_to(&self.buf[..size], &peer) {
Err(e) => {
self.to_send = Some((size, peer));
return try_nb!(Err(e));
},
Ok(_) => {
println!("Echoed {} bytes", size);
}
}
}
self.to_send = Some(
try_nb!(self.socket.recv_from(&mut self.buf))
);
}
}
}
fn main() {
let addr = env::args().nth(1).unwrap_or("127.0.0.1:8080".to_string());
let addr = addr.parse::<SocketAddr>().unwrap();
// Create the event loop that will drive this server
let mut l = Core::new().unwrap();
let handle = l.handle();
// Create and bind an UDP socket
let socket = UdpSocket::bind(&addr, &handle).unwrap();
// Inform that we have are listening
println!("Listening on: {}", addr);
// Create a server future
let server = Server::new(socket);
// Start event loop with the initial future: UDP echo server
l.run(server).unwrap();
}
|
