author | alfadur |
Mon, 15 Apr 2019 21:22:51 +0300 | |
changeset 14824 | 92225a708bda |
parent 14815 | fc2cfec95d86 |
child 14851 | 8ddb5842fe0b |
permissions | -rw-r--r-- |
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12853
diff
changeset
|
1 |
#![allow(unused_imports)] |
13421
d1368c776a4f
Enable all lints from the rust-2018-idioms suite.
marmistrz
parents:
13414
diff
changeset
|
2 |
#![deny(bare_trait_objects)] |
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12853
diff
changeset
|
3 |
|
14813
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
diff
changeset
|
4 |
extern crate getopts; |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
diff
changeset
|
5 |
use getopts::Options; |
14478 | 6 |
use log::*; |
12853 | 7 |
use mio::net::*; |
12125 | 8 |
use mio::*; |
14815 | 9 |
use std::env; |
12125 | 10 |
|
14478 | 11 |
mod protocol; |
12 |
mod server; |
|
12125 | 13 |
mod utils; |
14 |
||
13666 | 15 |
use crate::server::network::NetworkLayer; |
13414 | 16 |
use std::time::Duration; |
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12853
diff
changeset
|
17 |
|
14815 | 18 |
const PROGRAM_NAME: &'_ str = "Hedgewars Game Server"; |
19 |
||
12125 | 20 |
fn main() { |
13769 | 21 |
env_logger::init(); |
12137
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12133
diff
changeset
|
22 |
|
14813
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
diff
changeset
|
23 |
let args: Vec<String> = env::args().collect(); |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
diff
changeset
|
24 |
let mut opts = Options::new(); |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
diff
changeset
|
25 |
|
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
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:
14804
diff
changeset
|
27 |
opts.optflag("h", "help", "help"); |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
diff
changeset
|
28 |
let matches = match opts.parse(&args[1..]) { |
14815 | 29 |
Ok(m) => m, |
30 |
Err(e) => { |
|
31 |
println!("{}\n{}", e, opts.short_usage("")); |
|
32 |
return; |
|
33 |
} |
|
14813
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
diff
changeset
|
34 |
}; |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
diff
changeset
|
35 |
if matches.opt_present("h") { |
14815 | 36 |
println!("{}", opts.usage(PROGRAM_NAME)); |
14813
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
diff
changeset
|
37 |
return; |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
diff
changeset
|
38 |
} |
14804 | 39 |
info!("Hedgewars game server, protocol {}", utils::SERVER_VERSION); |
12125 | 40 |
|
14813
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
diff
changeset
|
41 |
let address; |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
diff
changeset
|
42 |
if matches.opt_present("p") { |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
diff
changeset
|
43 |
match matches.opt_str("p") { |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
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:
14804
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:
14804
diff
changeset
|
46 |
} |
14815 | 47 |
} else { |
14813
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
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:
14804
diff
changeset
|
49 |
} |
38e66519e585
ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents:
14804
diff
changeset
|
50 |
|
12126 | 51 |
let listener = TcpListener::bind(&address).unwrap(); |
12125 | 52 |
|
53 |
let poll = Poll::new().unwrap(); |
|
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12853
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:
12853
diff
changeset
|
55 |
hw_network.register_server(&poll).unwrap(); |
12125 | 56 |
|
57 |
let mut events = Events::with_capacity(1024); |
|
58 |
||
59 |
loop { |
|
13414 | 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(); |
|
12125 | 66 |
|
67 |
for event in events.iter() { |
|
12853 | 68 |
if event.readiness() & Ready::readable() == Ready::readable() { |
12127 | 69 |
match event.token() { |
14800
f43ab2bd76ae
add a thread for internal server IO and implement account checking with it
alfadur
parents:
14712
diff
changeset
|
70 |
utils::SERVER_TOKEN => hw_network.accept_client(&poll).unwrap(), |
14824 | 71 |
utils::TIMER_TOKEN => hw_network.handle_timeout(&poll).unwrap(), |
14800
f43ab2bd76ae
add a thread for internal server IO and implement account checking with it
alfadur
parents:
14712
diff
changeset
|
72 |
#[cfg(feature = "official-server")] |
f43ab2bd76ae
add a thread for internal server IO and implement account checking with it
alfadur
parents:
14712
diff
changeset
|
73 |
utils::IO_TOKEN => hw_network.handle_io_result(), |
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12853
diff
changeset
|
74 |
Token(tok) => hw_network.client_readable(&poll, tok).unwrap(), |
12127 | 75 |
} |
76 |
} |
|
12853 | 77 |
if event.readiness() & Ready::writable() == Ready::writable() { |
12127 | 78 |
match event.token() { |
14824 | 79 |
utils::SERVER_TOKEN | utils::TIMER_TOKEN | utils::IO_TOKEN => unreachable!(), |
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12853
diff
changeset
|
80 |
Token(tok) => hw_network.client_writable(&poll, tok).unwrap(), |
12127 | 81 |
} |
12125 | 82 |
} |
14478 | 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 |
// } |
|
12125 | 89 |
} |
13414 | 90 |
hw_network.on_idle(&poll).unwrap(); |
12125 | 91 |
} |
92 |
} |