rust/hedgewars-server/src/server/handlers/loggingin.rs
author unc0rr
Sun, 16 Dec 2018 00:12:29 +0100
changeset 14457 98ef2913ec73
parent 14415 06672690d71b
child 14671 455865ccd36c
permissions -rw-r--r--
Apply rustfmt to all files
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
     1
use mio;
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
     2
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
     3
use crate::{
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
     4
    protocol::messages::{HWProtocolMessage, HWServerMessage::*},
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
     5
    server::{
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
     6
        actions::{Action, Action::*},
13800
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
     7
        client::HWClient,
14374
e5db279308d7 dispose of server mods
alfadur
parents: 13805
diff changeset
     8
        core::HWServer,
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
     9
        coretypes::ClientId,
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
    10
    },
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    11
    utils::is_name_illegal,
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    12
};
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    13
use log::*;
13800
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    14
#[cfg(feature = "official-server")]
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    15
use openssl::sha::sha1;
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    16
use std::fmt::{Formatter, LowerHex};
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    17
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    18
#[derive(PartialEq)]
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    19
struct Sha1Digest([u8; 20]);
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    20
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    21
impl LowerHex for Sha1Digest {
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    22
    fn fmt(&self, f: &mut Formatter) -> Result<(), std::fmt::Error> {
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    23
        for byte in &self.0 {
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    24
            write!(f, "{:02x}", byte)?;
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    25
        }
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    26
        Ok(())
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    27
    }
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    28
}
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    29
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    30
#[cfg(feature = "official-server")]
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    31
fn get_hash(client: &HWClient, salt1: &str, salt2: &str) -> Sha1Digest {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    32
    let s = format!(
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    33
        "{}{}{}{}{}",
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    34
        salt1, salt2, client.web_password, client.protocol_number, "!hedgewars"
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    35
    );
13800
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    36
    Sha1Digest(sha1(s.as_bytes()))
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    37
}
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
    38
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    39
pub fn handle(server: &mut HWServer, client_id: ClientId, message: HWProtocolMessage) {
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
    40
    match message {
13416
cdf69667593b partial room implementation
alfadur
parents: 13119
diff changeset
    41
        HWProtocolMessage::Nick(nick) => {
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
    42
            let client = &mut server.clients[client_id];
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
    43
            debug!("{} {}", nick, is_name_illegal(&nick));
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
    44
            let actions = if client.room_id != None {
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
    45
                unreachable!()
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    46
            } else if !client.nick.is_empty() {
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
    47
                vec![ProtocolError("Nickname already provided.".to_string())]
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    48
            } else if is_name_illegal(&nick) {
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
    49
                vec![ByeClient("Illegal nickname! Nicknames must be between 1-40 characters long, must not have a trailing or leading space and must not have any of these characters: $()*+?[]^{|}".to_string())]
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    50
            } else {
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
    51
                client.nick = nick.clone();
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    52
                vec![Nick(nick).send_self().action(), CheckRegistered]
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
    53
            };
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
    54
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    55
            server.react(client_id, actions);
13798
4664da990556 Add official server feature to cargo
alfadur
parents: 13666
diff changeset
    56
        }
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
    57
        HWProtocolMessage::Proto(proto) => {
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
    58
            let client = &mut server.clients[client_id];
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
    59
            let actions = if client.protocol_number != 0 {
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
    60
                vec![ProtocolError("Protocol already known.".to_string())]
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    61
            } else if proto == 0 {
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
    62
                vec![ProtocolError("Bad number.".to_string())]
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    63
            } else {
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
    64
                client.protocol_number = proto;
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    65
                vec![Proto(proto).send_self().action(), CheckRegistered]
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13478
diff changeset
    66
            };
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    67
            server.react(client_id, actions);
13798
4664da990556 Add official server feature to cargo
alfadur
parents: 13666
diff changeset
    68
        }
4664da990556 Add official server feature to cargo
alfadur
parents: 13666
diff changeset
    69
        #[cfg(feature = "official-server")]
13800
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    70
        HWProtocolMessage::Password(hash, salt) => {
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    71
            let c = &server.clients[client_id];
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    72
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    73
            let client_hash = get_hash(c, &salt, &c.server_salt);
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    74
            let server_hash = get_hash(c, &c.server_salt, &salt);
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    75
            let actions = if client_hash == server_hash {
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    76
                vec![
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    77
                    ServerAuth(format!("{:x}", server_hash))
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    78
                        .send_self()
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    79
                        .action(),
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    80
                    JoinLobby,
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    81
                ]
13800
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    82
            } else {
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    83
                vec![ByeClient("Authentication failed".to_string())]
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    84
            };
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    85
            server.react(client_id, actions);
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    86
        }
0118b7412570 use openssl for password message handling
alfadur
parents: 13798
diff changeset
    87
        #[cfg(feature = "official-server")]
13798
4664da990556 Add official server feature to cargo
alfadur
parents: 13666
diff changeset
    88
        HWProtocolMessage::Checker(protocol, nick, password) => {
4664da990556 Add official server feature to cargo
alfadur
parents: 13666
diff changeset
    89
            let c = &mut server.clients[client_id];
4664da990556 Add official server feature to cargo
alfadur
parents: 13666
diff changeset
    90
            c.nick = nick;
4664da990556 Add official server feature to cargo
alfadur
parents: 13666
diff changeset
    91
            c.web_password = password;
4664da990556 Add official server feature to cargo
alfadur
parents: 13666
diff changeset
    92
            c.set_is_checker(true);
4664da990556 Add official server feature to cargo
alfadur
parents: 13666
diff changeset
    93
        }
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
    94
        _ => warn!("Incorrect command in logging-in state"),
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
    95
    }
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
    96
}