author | nemo |
Fri, 12 Apr 2019 11:47:35 -0400 | |
changeset 14798 | 9de13d9a6312 |
parent 14797 | 38e66519e585 |
child 14799 | fc2cfec95d86 |
permissions | -rw-r--r-- |
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12858
diff
changeset
|
1 |
#![allow(unused_imports)] |
13426
d1368c776a4f
Enable all lints from the rust-2018-idioms suite.
marmistrz
parents:
13419
diff
changeset
|
2 |
#![deny(bare_trait_objects)] |
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12858
diff
changeset
|
3 |
|
14797
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
4 |
extern crate getopts; |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
5 |
use getopts::Options; |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
6 |
use std::env; |
14462 | 7 |
use log::*; |
12858 | 8 |
use mio::net::*; |
12130 | 9 |
use mio::*; |
10 |
||
14462 | 11 |
mod protocol; |
12 |
mod server; |
|
12130 | 13 |
mod utils; |
14 |
||
13671 | 15 |
use crate::server::network::NetworkLayer; |
13419 | 16 |
use std::time::Duration; |
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12858
diff
changeset
|
17 |
|
12130 | 18 |
fn main() { |
13802 | 19 |
env_logger::init(); |
12142
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12138
diff
changeset
|
20 |
|
14797
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
21 |
let args: Vec<String> = env::args().collect(); |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
22 |
let mut opts = Options::new(); |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
23 |
|
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
24 |
opts.optopt("p", "port", "port - defaults to 46631", "PORT"); |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
25 |
opts.optflag("h", "help", "help"); |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
26 |
let matches = match opts.parse(&args[1..]) { |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
27 |
Ok(m) => { m } |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
28 |
Err(f) => { panic!(f.to_string()) } |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
29 |
}; |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
30 |
if matches.opt_present("h") { |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
31 |
println!("-p/--port - defaults to 46631"); |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
32 |
return; |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
33 |
} |
14788 | 34 |
info!("Hedgewars game server, protocol {}", utils::SERVER_VERSION); |
12130 | 35 |
|
14797
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
36 |
let address; |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
37 |
if matches.opt_present("p") { |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
38 |
match matches.opt_str("p") { |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
39 |
Some(x) => address = format!("0.0.0.0:{}", x).parse().unwrap(), |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
40 |
None => address = "0.0.0.0:46631".parse().unwrap(), |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
41 |
} |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
42 |
} |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
43 |
else { |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
44 |
address = "0.0.0.0:46631".parse().unwrap(); |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
45 |
} |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
46 |
|
12131 | 47 |
let listener = TcpListener::bind(&address).unwrap(); |
12130 | 48 |
|
49 |
let poll = Poll::new().unwrap(); |
|
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12858
diff
changeset
|
50 |
let mut hw_network = NetworkLayer::new(listener, 1024, 512); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12858
diff
changeset
|
51 |
hw_network.register_server(&poll).unwrap(); |
12130 | 52 |
|
53 |
let mut events = Events::with_capacity(1024); |
|
54 |
||
55 |
loop { |
|
13419 | 56 |
let timeout = if hw_network.has_pending_operations() { |
57 |
Some(Duration::from_millis(1)) |
|
58 |
} else { |
|
59 |
None |
|
60 |
}; |
|
61 |
poll.poll(&mut events, timeout).unwrap(); |
|
12130 | 62 |
|
63 |
for event in events.iter() { |
|
12858 | 64 |
if event.readiness() & Ready::readable() == Ready::readable() { |
12132 | 65 |
match event.token() { |
14784
f43ab2bd76ae
add a thread for internal server IO and implement account checking with it
alfadur
parents:
14696
diff
changeset
|
66 |
utils::SERVER_TOKEN => hw_network.accept_client(&poll).unwrap(), |
f43ab2bd76ae
add a thread for internal server IO and implement account checking with it
alfadur
parents:
14696
diff
changeset
|
67 |
#[cfg(feature = "official-server")] |
f43ab2bd76ae
add a thread for internal server IO and implement account checking with it
alfadur
parents:
14696
diff
changeset
|
68 |
utils::IO_TOKEN => hw_network.handle_io_result(), |
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12858
diff
changeset
|
69 |
Token(tok) => hw_network.client_readable(&poll, tok).unwrap(), |
12132 | 70 |
} |
71 |
} |
|
12858 | 72 |
if event.readiness() & Ready::writable() == Ready::writable() { |
12132 | 73 |
match event.token() { |
14784
f43ab2bd76ae
add a thread for internal server IO and implement account checking with it
alfadur
parents:
14696
diff
changeset
|
74 |
utils::SERVER_TOKEN => unreachable!(), |
f43ab2bd76ae
add a thread for internal server IO and implement account checking with it
alfadur
parents:
14696
diff
changeset
|
75 |
utils::IO_TOKEN => unreachable!(), |
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12858
diff
changeset
|
76 |
Token(tok) => hw_network.client_writable(&poll, tok).unwrap(), |
12132 | 77 |
} |
12130 | 78 |
} |
14462 | 79 |
// if event.kind().is_hup() || event.kind().is_error() { |
80 |
// match event.token() { |
|
81 |
// utils::SERVER => unreachable!(), |
|
82 |
// Token(tok) => server.client_error(&poll, tok).unwrap(), |
|
83 |
// } |
|
84 |
// } |
|
12130 | 85 |
} |
13419 | 86 |
hw_network.on_idle(&poll).unwrap(); |
12130 | 87 |
} |
88 |
} |