rust/hedgewars-server/src/main.rs
author alfadur
Tue, 09 Apr 2019 21:08:35 +0300
changeset 14779 f43ab2bd76ae
parent 14691 2071da901c63
child 14783 b3adc030104b
permissions -rw-r--r--
add a thread for internal server IO and implement account checking with it
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
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
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
     4
use log::*;
12853
a9d105dc5c95 Improve this code a bit more
unc0rr
parents: 12852
diff changeset
     5
use mio::net::*;
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
     6
use mio::*;
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
     7
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
     8
mod protocol;
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
     9
mod server;
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    10
mod utils;
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    11
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13529
diff changeset
    12
use crate::server::network::NetworkLayer;
13414
28b314ad566d handle edge polling properly
alfadur
parents: 13119
diff changeset
    13
use std::time::Duration;
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12853
diff changeset
    14
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    15
fn main() {
13797
c5edfcfac68b Bump dependencies
alfadur
parents: 13796
diff changeset
    16
    env_logger::init();
12137
193dfdcb0620 - Use logging facilities instead of plain println!
unc0rr
parents: 12133
diff changeset
    17
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    18
    info!(
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    19
        "Hedgewars game server, protocol {}",
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    20
        utils::PROTOCOL_VERSION
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    21
    );
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    22
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    23
    let address = "0.0.0.0:46631".parse().unwrap();
12126
4348997e502b Refactor code to add more structure to it
unc0rr
parents: 12125
diff changeset
    24
    let listener = TcpListener::bind(&address).unwrap();
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    25
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    26
    let poll = Poll::new().unwrap();
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12853
diff changeset
    27
    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
    28
    hw_network.register_server(&poll).unwrap();
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    29
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    30
    let mut events = Events::with_capacity(1024);
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    31
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    32
    loop {
13414
28b314ad566d handle edge polling properly
alfadur
parents: 13119
diff changeset
    33
        let timeout = if hw_network.has_pending_operations() {
28b314ad566d handle edge polling properly
alfadur
parents: 13119
diff changeset
    34
            Some(Duration::from_millis(1))
28b314ad566d handle edge polling properly
alfadur
parents: 13119
diff changeset
    35
        } else {
28b314ad566d handle edge polling properly
alfadur
parents: 13119
diff changeset
    36
            None
28b314ad566d handle edge polling properly
alfadur
parents: 13119
diff changeset
    37
        };
28b314ad566d handle edge polling properly
alfadur
parents: 13119
diff changeset
    38
        poll.poll(&mut events, timeout).unwrap();
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    39
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    40
        for event in events.iter() {
12853
a9d105dc5c95 Improve this code a bit more
unc0rr
parents: 12852
diff changeset
    41
            if event.readiness() & Ready::readable() == Ready::readable() {
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
    42
                match event.token() {
14779
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14691
diff changeset
    43
                    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: 14691
diff changeset
    44
                    #[cfg(feature = "official-server")]
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14691
diff changeset
    45
                    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
    46
                    Token(tok) => hw_network.client_readable(&poll, tok).unwrap(),
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
    47
                }
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
    48
            }
12853
a9d105dc5c95 Improve this code a bit more
unc0rr
parents: 12852
diff changeset
    49
            if event.readiness() & Ready::writable() == Ready::writable() {
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
    50
                match event.token() {
14779
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14691
diff changeset
    51
                    utils::SERVER_TOKEN => unreachable!(),
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14691
diff changeset
    52
                    utils::IO_TOKEN => unreachable!(),
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12853
diff changeset
    53
                    Token(tok) => hw_network.client_writable(&poll, tok).unwrap(),
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
    54
                }
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    55
            }
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    56
            //            if event.kind().is_hup() || event.kind().is_error() {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    57
            //                match event.token() {
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    58
            //                    utils::SERVER => unreachable!(),
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    59
            //                    Token(tok) => server.client_error(&poll, tok).unwrap(),
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    60
            //                }
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    61
            //            }
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    62
        }
13414
28b314ad566d handle edge polling properly
alfadur
parents: 13119
diff changeset
    63
        hw_network.on_idle(&poll).unwrap();
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    64
    }
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    65
}