rust/hedgewars-server/src/main.rs
author alfadur
Fri, 12 Apr 2019 19:26:44 +0300
changeset 14794 fc2cfec95d86
parent 14793 9de13d9a6312
child 14803 92225a708bda
permissions -rw-r--r--
a bit more option error handling

#![allow(unused_imports)]
#![deny(bare_trait_objects)]

extern crate getopts;
use getopts::Options;
use log::*;
use mio::net::*;
use mio::*;
use std::env;

mod protocol;
mod server;
mod utils;

use crate::server::network::NetworkLayer;
use std::time::Duration;

const PROGRAM_NAME: &'_ str = "Hedgewars Game Server";

fn main() {
    env_logger::init();

    let args: Vec<String> = env::args().collect();
    let mut opts = Options::new();

    opts.optopt("p", "port", "port - defaults to 46631", "PORT");
    opts.optflag("h", "help", "help");
    let matches = match opts.parse(&args[1..]) {
        Ok(m) => m,
        Err(e) => {
            println!("{}\n{}", e, opts.short_usage(""));
            return;
        }
    };
    if matches.opt_present("h") {
        println!("{}", opts.usage(PROGRAM_NAME));
        return;
    }
    info!("Hedgewars game server, protocol {}", utils::SERVER_VERSION);

    let address;
    if matches.opt_present("p") {
        match matches.opt_str("p") {
            Some(x) => address = format!("0.0.0.0:{}", x).parse().unwrap(),
            None => address = "0.0.0.0:46631".parse().unwrap(),
        }
    } else {
        address = "0.0.0.0:46631".parse().unwrap();
    }

    let listener = TcpListener::bind(&address).unwrap();

    let poll = Poll::new().unwrap();
    let mut hw_network = NetworkLayer::new(listener, 1024, 512);
    hw_network.register_server(&poll).unwrap();

    let mut events = Events::with_capacity(1024);

    loop {
        let timeout = if hw_network.has_pending_operations() {
            Some(Duration::from_millis(1))
        } else {
            None
        };
        poll.poll(&mut events, timeout).unwrap();

        for event in events.iter() {
            if event.readiness() & Ready::readable() == Ready::readable() {
                match event.token() {
                    utils::SERVER_TOKEN => hw_network.accept_client(&poll).unwrap(),
                    #[cfg(feature = "official-server")]
                    utils::IO_TOKEN => hw_network.handle_io_result(),
                    Token(tok) => hw_network.client_readable(&poll, tok).unwrap(),
                }
            }
            if event.readiness() & Ready::writable() == Ready::writable() {
                match event.token() {
                    utils::SERVER_TOKEN => unreachable!(),
                    utils::IO_TOKEN => unreachable!(),
                    Token(tok) => hw_network.client_writable(&poll, tok).unwrap(),
                }
            }
            //            if event.kind().is_hup() || event.kind().is_error() {
            //                match event.token() {
            //                    utils::SERVER => unreachable!(),
            //                    Token(tok) => server.client_error(&poll, tok).unwrap(),
            //                }
            //            }
        }
        hw_network.on_idle(&poll).unwrap();
    }
}