author | alfadur |
Fri, 12 Apr 2019 19:26:44 +0300 | |
changeset 14799 | fc2cfec95d86 |
parent 14798 | 9de13d9a6312 |
child 14808 | 92225a708bda |
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; |
14462 | 6 |
use log::*; |
12858 | 7 |
use mio::net::*; |
12130 | 8 |
use mio::*; |
14799 | 9 |
use std::env; |
12130 | 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 |
|
14799 | 18 |
const PROGRAM_NAME: &'_ str = "Hedgewars Game Server"; |
19 |
||
12130 | 20 |
fn main() { |
13802 | 21 |
env_logger::init(); |
12142
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12138
diff
changeset
|
22 |
|
14797
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
23 |
let args: Vec<String> = env::args().collect(); |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
24 |
let mut opts = Options::new(); |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
25 |
|
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
26 |
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
|
27 |
opts.optflag("h", "help", "help"); |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
28 |
let matches = match opts.parse(&args[1..]) { |
14799 | 29 |
Ok(m) => m, |
30 |
Err(e) => { |
|
31 |
println!("{}\n{}", e, opts.short_usage("")); |
|
32 |
return; |
|
33 |
} |
|
14797
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
34 |
}; |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
35 |
if matches.opt_present("h") { |
14799 | 36 |
println!("{}", opts.usage(PROGRAM_NAME)); |
14797
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
37 |
return; |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
38 |
} |
14788 | 39 |
info!("Hedgewars game server, protocol {}", utils::SERVER_VERSION); |
12130 | 40 |
|
14797
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
41 |
let address; |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
42 |
if matches.opt_present("p") { |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
43 |
match matches.opt_str("p") { |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
44 |
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
|
45 |
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
|
46 |
} |
14799 | 47 |
} else { |
14797
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
48 |
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
|
49 |
} |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14788
diff
changeset
|
50 |
|
12131 | 51 |
let listener = TcpListener::bind(&address).unwrap(); |
12130 | 52 |
|
53 |
let poll = Poll::new().unwrap(); |
|
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12858
diff
changeset
|
54 |
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
|
55 |
hw_network.register_server(&poll).unwrap(); |
12130 | 56 |
|
57 |
let mut events = Events::with_capacity(1024); |
|
58 |
||
59 |
loop { |
|
13419 | 60 |
let timeout = if hw_network.has_pending_operations() { |
61 |
Some(Duration::from_millis(1)) |
|
62 |
} else { |
|
63 |
None |
|
64 |
}; |
|
65 |
poll.poll(&mut events, timeout).unwrap(); |
|
12130 | 66 |
|
67 |
for event in events.iter() { |
|
12858 | 68 |
if event.readiness() & Ready::readable() == Ready::readable() { |
12132 | 69 |
match event.token() { |
14784
f43ab2bd76ae
add a thread for internal server IO and implement account checking with it
alfadur
parents:
14696
diff
changeset
|
70 |
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
|
71 |
#[cfg(feature = "official-server")] |
f43ab2bd76ae
add a thread for internal server IO and implement account checking with it
alfadur
parents:
14696
diff
changeset
|
72 |
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
|
73 |
Token(tok) => hw_network.client_readable(&poll, tok).unwrap(), |
12132 | 74 |
} |
75 |
} |
|
12858 | 76 |
if event.readiness() & Ready::writable() == Ready::writable() { |
12132 | 77 |
match event.token() { |
14784
f43ab2bd76ae
add a thread for internal server IO and implement account checking with it
alfadur
parents:
14696
diff
changeset
|
78 |
utils::SERVER_TOKEN => unreachable!(), |
f43ab2bd76ae
add a thread for internal server IO and implement account checking with it
alfadur
parents:
14696
diff
changeset
|
79 |
utils::IO_TOKEN => unreachable!(), |
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12858
diff
changeset
|
80 |
Token(tok) => hw_network.client_writable(&poll, tok).unwrap(), |
12132 | 81 |
} |
12130 | 82 |
} |
14462 | 83 |
// if event.kind().is_hup() || event.kind().is_error() { |
84 |
// match event.token() { |
|
85 |
// utils::SERVER => unreachable!(), |
|
86 |
// Token(tok) => server.client_error(&poll, tok).unwrap(), |
|
87 |
// } |
|
88 |
// } |
|
12130 | 89 |
} |
13419 | 90 |
hw_network.on_idle(&poll).unwrap(); |
12130 | 91 |
} |
92 |
} |