rust/hedgewars-server/src/main.rs
author alfadur
Tue, 22 Jun 2021 01:41:33 +0300
changeset 15800 6af892a0a4b8
parent 15795 40929af15167
child 15831 7d0f747afcb8
permissions -rw-r--r--
update mio
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
14792
38e66519e585 ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents: 14783
diff changeset
     4
use getopts::Options;
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
     5
use log::*;
14830
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
     6
use mio::{net::*, *};
15800
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
     7
use std::{
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
     8
    env,
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
     9
    str::FromStr as _,
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
    10
    time::{Duration, Instant},
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
    11
};
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    12
15074
c5a6e8566425 shuffle server files
alfadur
parents: 14916
diff changeset
    13
mod core;
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
    14
mod handlers;
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    15
mod protocol;
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    16
mod server;
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    17
mod utils;
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    18
14830
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    19
use crate::server::network::{NetworkLayer, NetworkLayerBuilder};
13119
1e39b8749072 separated the server logic from all the async io mess.
alfadur
parents: 12853
diff changeset
    20
14794
fc2cfec95d86 a bit more option error handling
alfadur
parents: 14793
diff changeset
    21
const PROGRAM_NAME: &'_ str = "Hedgewars Game Server";
fc2cfec95d86 a bit more option error handling
alfadur
parents: 14793
diff changeset
    22
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    23
fn main() {
13797
c5edfcfac68b Bump dependencies
alfadur
parents: 13796
diff changeset
    24
    env_logger::init();
12137
193dfdcb0620 - Use logging facilities instead of plain println!
unc0rr
parents: 12133
diff changeset
    25
14830
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    26
    info!("Hedgewars game server, protocol {}", utils::SERVER_VERSION);
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    27
14792
38e66519e585 ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents: 14783
diff changeset
    28
    let args: Vec<String> = env::args().collect();
38e66519e585 ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents: 14783
diff changeset
    29
    let mut opts = Options::new();
38e66519e585 ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents: 14783
diff changeset
    30
38e66519e585 ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents: 14783
diff changeset
    31
    opts.optopt("p", "port", "port - defaults to 46631", "PORT");
38e66519e585 ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents: 14783
diff changeset
    32
    opts.optflag("h", "help", "help");
38e66519e585 ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents: 14783
diff changeset
    33
    let matches = match opts.parse(&args[1..]) {
14794
fc2cfec95d86 a bit more option error handling
alfadur
parents: 14793
diff changeset
    34
        Ok(m) => m,
fc2cfec95d86 a bit more option error handling
alfadur
parents: 14793
diff changeset
    35
        Err(e) => {
fc2cfec95d86 a bit more option error handling
alfadur
parents: 14793
diff changeset
    36
            println!("{}\n{}", e, opts.short_usage(""));
fc2cfec95d86 a bit more option error handling
alfadur
parents: 14793
diff changeset
    37
            return;
fc2cfec95d86 a bit more option error handling
alfadur
parents: 14793
diff changeset
    38
        }
14792
38e66519e585 ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents: 14783
diff changeset
    39
    };
38e66519e585 ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents: 14783
diff changeset
    40
    if matches.opt_present("h") {
14794
fc2cfec95d86 a bit more option error handling
alfadur
parents: 14793
diff changeset
    41
        println!("{}", opts.usage(PROGRAM_NAME));
14792
38e66519e585 ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents: 14783
diff changeset
    42
        return;
38e66519e585 ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents: 14783
diff changeset
    43
    }
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    44
14830
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    45
    let port = matches
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    46
        .opt_str("p")
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    47
        .and_then(|s| u16::from_str(&s).ok())
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    48
        .unwrap_or(46631);
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    49
    let address = format!("0.0.0.0:{}", port).parse().unwrap();
14792
38e66519e585 ugly hacked in argument for port to remove unc0rr's excuse
nemo
parents: 14783
diff changeset
    50
15800
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
    51
    let listener = TcpListener::bind(address).unwrap();
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    52
15800
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
    53
    let mut poll = Poll::new().unwrap();
14830
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    54
    let mut hw_builder = NetworkLayerBuilder::default().with_listener(listener);
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    55
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    56
    #[cfg(feature = "tls-connections")]
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    57
    {
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    58
        let address = format!("0.0.0.0:{}", port + 1).parse().unwrap();
15800
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
    59
        hw_builder = hw_builder.with_secure_listener(TcpListener::bind(address).unwrap());
14830
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    60
    }
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    61
15800
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
    62
    let mut hw_network = hw_builder.build(&poll);
14830
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    63
    hw_network.register(&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
    let mut events = Events::with_capacity(1024);
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    66
15800
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
    67
    let mut time = Instant::now();
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
    68
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    69
    loop {
13414
28b314ad566d handle edge polling properly
alfadur
parents: 13119
diff changeset
    70
        let timeout = if hw_network.has_pending_operations() {
28b314ad566d handle edge polling properly
alfadur
parents: 13119
diff changeset
    71
            Some(Duration::from_millis(1))
28b314ad566d handle edge polling properly
alfadur
parents: 13119
diff changeset
    72
        } else {
28b314ad566d handle edge polling properly
alfadur
parents: 13119
diff changeset
    73
            None
28b314ad566d handle edge polling properly
alfadur
parents: 13119
diff changeset
    74
        };
15800
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
    75
13414
28b314ad566d handle edge polling properly
alfadur
parents: 13119
diff changeset
    76
        poll.poll(&mut events, timeout).unwrap();
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    77
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
    78
        for event in events.iter() {
15800
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
    79
            if event.is_readable() {
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
    80
                match event.token() {
15795
40929af15167 find excuses to use shiny new 🦀 features
alfadur
parents: 15102
diff changeset
    81
                    token @ (utils::SERVER_TOKEN | utils::SECURE_SERVER_TOKEN) => {
14916
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
    82
                        match hw_network.accept_client(&poll, token) {
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
    83
                            Ok(()) => (),
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
    84
                            Err(e) => debug!("Error accepting client: {}", e),
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
    85
                        }
14830
8ddb5842fe0b allow running plaintext and tls servers in parallel
alfadur
parents: 14803
diff changeset
    86
                    }
14779
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14691
diff changeset
    87
                    #[cfg(feature = "official-server")]
15102
80ff12edf5e6 handle response from IO result handler
alfadur
parents: 15075
diff changeset
    88
                    utils::IO_TOKEN => match hw_network.handle_io_result(&poll) {
14916
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
    89
                        Ok(()) => (),
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
    90
                        Err(e) => debug!("Error in IO task: {}", e),
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
    91
                    },
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
    92
                    Token(token) => match hw_network.client_readable(&poll, token) {
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
    93
                        Ok(()) => (),
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
    94
                        Err(e) => debug!("Error reading from client socket {}: {}", token, e),
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
    95
                    },
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
    96
                }
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
    97
            }
15800
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
    98
            if event.is_writable() {
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
    99
                match event.token() {
15800
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
   100
                    utils::SERVER_TOKEN | utils::SECURE_SERVER_TOKEN | utils::IO_TOKEN => {
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
   101
                        unreachable!()
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
   102
                    }
14916
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
   103
                    Token(token) => match hw_network.client_writable(&poll, token) {
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
   104
                        Ok(()) => (),
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
   105
                        Err(e) => debug!("Error writing to client socket {}: {}", token, e),
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
   106
                    },
12127
36ac9c075d0d - Use netbuf buffers for client connection stream
unc0rr
parents: 12126
diff changeset
   107
                }
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
   108
            }
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
   109
        }
14916
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
   110
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
   111
        match hw_network.on_idle(&poll) {
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
   112
            Ok(()) => (),
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
   113
            Err(e) => debug!("Error in idle handler: {}", e),
8750530bf7e7 avoid crashing server in the event loop
alfadur
parents: 14830
diff changeset
   114
        };
15800
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
   115
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
   116
        if time.elapsed() > Duration::from_secs(1) {
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
   117
            time = Instant::now();
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
   118
            match hw_network.handle_timeout(&mut poll) {
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
   119
                Ok(()) => (),
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
   120
                Err(e) => debug!("Error in timer event: {}", e),
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
   121
            }
6af892a0a4b8 update mio
alfadur
parents: 15795
diff changeset
   122
        }
12125
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
   123
    }
858bf4d04c54 Start server implementation in rust
unc0rr
parents:
diff changeset
   124
}