rust/hedgewars-server/src/handlers/inanteroom.rs
author alfadur
Sat, 22 Feb 2025 19:39:31 +0300
changeset 16120 5febd2bc5372
parent 16019 c40f5e27aaf0
permissions -rw-r--r--
Add server reconnection tokens and anteroom local list of used nicks
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
15554
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
     1
use super::strings::*;
15826
747278149393 Extract network protocol into a separate crate
unc0rr
parents: 15562
diff changeset
     2
use crate::handlers::actions::ToPendingMessage;
13666
09f4a30e50cc Rust 2018 conversion
alfadur
parents: 13450
diff changeset
     3
use crate::{
15095
c5a6e8566425 shuffle server files
alfadur
parents: 14891
diff changeset
     4
    core::{
15542
fd3a20e9d095 move the anteroom out of the server
alfadur <mail@none>
parents: 15124
diff changeset
     5
        anteroom::{HwAnteroom, HwAnteroomClient},
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
     6
        client::HwClient,
15542
fd3a20e9d095 move the anteroom out of the server
alfadur <mail@none>
parents: 15124
diff changeset
     7
        server::HwServer,
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
     8
        types::ClientId,
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
     9
    },
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    10
    utils::is_name_illegal,
13419
81e0ed105f5d implementation of team related messages
alfadur
parents: 13416
diff changeset
    11
};
15826
747278149393 Extract network protocol into a separate crate
unc0rr
parents: 15562
diff changeset
    12
use hedgewars_network_protocol::messages::{
747278149393 Extract network protocol into a separate crate
unc0rr
parents: 15562
diff changeset
    13
    HwProtocolMessage, HwProtocolMessage::LoadRoom, HwServerMessage::*,
747278149393 Extract network protocol into a separate crate
unc0rr
parents: 15562
diff changeset
    14
};
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
    15
use log::*;
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    16
use std::{
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    17
    fmt::{Formatter, LowerHex},
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    18
    num::NonZeroU16,
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    19
};
13774
0118b7412570 use openssl for password message handling
alfadur
parents: 13771
diff changeset
    20
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    21
pub enum LoginResult {
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    22
    Unchanged,
16120
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    23
    Complete(HwAnteroomClient),
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    24
    Exit,
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    25
}
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    26
16120
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    27
fn get_completion_result(
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    28
    anteroom: &mut HwAnteroom,
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    29
    client_id: ClientId,
14802
01f8ab45f806 fix lobby joining
alfadur
parents: 14800
diff changeset
    30
    response: &mut super::Response,
16120
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    31
) -> LoginResult {
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    32
    #[cfg(feature = "official-server")]
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    33
    {
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    34
        let client = anteroom.get_client(client_id);
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    35
        response.request_io(super::IoTask::CheckRegistered {
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    36
            nick: client.nick.as_ref().unwrap().clone(),
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    37
        });
15554
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
    38
        LoginResult::Unchanged
16120
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    39
    }
14802
01f8ab45f806 fix lobby joining
alfadur
parents: 14800
diff changeset
    40
16120
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    41
    #[cfg(not(feature = "official-server"))]
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    42
    {
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    43
        LoginResult::Complete(anteroom.remove_client(client_id).unwrap())
14800
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14717
diff changeset
    44
    }
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14717
diff changeset
    45
}
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14717
diff changeset
    46
14692
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14478
diff changeset
    47
pub fn handle(
16120
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    48
    anteroom: &mut HwAnteroom,
14692
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14478
diff changeset
    49
    client_id: ClientId,
455865ccd36c Server action refactoring part 2 of N
alfadur <mail@none>
parents: 14478
diff changeset
    50
    response: &mut super::Response,
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    51
    message: HwProtocolMessage,
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    52
) -> LoginResult {
16019
c40f5e27aaf0 fix connection errors carrying over to new clients
alfadur
parents: 15967
diff changeset
    53
    //todo!("Handle parsing of empty nicks")
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
    54
    match message {
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    55
        HwProtocolMessage::Quit(_) => {
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    56
            response.add(Bye("User quit".to_string()).send_self());
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    57
            LoginResult::Exit
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    58
        }
16120
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    59
        HwProtocolMessage::Nick(nick, token) => {
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    60
            if anteroom.nick_taken(&nick) {
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    61
                response.add(Notice("NickAlreadyInUse".to_string()).send_self());
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    62
                return LoginResult::Unchanged;
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    63
            }
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    64
            let reconnect = token
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    65
                .map(|t| anteroom.get_nick_token(&nick) == Some(&t[..]))
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    66
                .unwrap_or(false);
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    67
            let client = anteroom.get_client_mut(client_id);
14891
ce98c37826a7 fix accepting on wrong server socket
alfadur
parents: 14802
diff changeset
    68
14802
01f8ab45f806 fix lobby joining
alfadur
parents: 14800
diff changeset
    69
            if client.nick.is_some() {
15554
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
    70
                response.error(NICKNAME_PROVIDED);
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    71
                LoginResult::Unchanged
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
    72
            } else if is_name_illegal(&nick) {
15554
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
    73
                response.add(Bye(ILLEGAL_CLIENT_NAME.to_string()).send_self());
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    74
                LoginResult::Exit
14478
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14436
diff changeset
    75
            } else {
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    76
                client.nick = Some(nick.clone());
16120
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    77
                let protocol_number = client.protocol_number;
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    78
                if reconnect {
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    79
                    client.is_registered = reconnect;
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    80
                } else if let Some(token) = anteroom.register_nick_token(&nick) {
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    81
                    response.add(Token(token.to_string()).send_self());
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    82
                }
14693
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14692
diff changeset
    83
                response.add(Nick(nick).send_self());
14694
08a8605bafaf Server action refactoring part 4 of N
alfadur <mail@none>
parents: 14693
diff changeset
    84
16120
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    85
                if protocol_number.is_some() {
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    86
                    get_completion_result(anteroom, client_id, response)
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    87
                } else {
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    88
                    LoginResult::Unchanged
14694
08a8605bafaf Server action refactoring part 4 of N
alfadur <mail@none>
parents: 14693
diff changeset
    89
                }
08a8605bafaf Server action refactoring part 4 of N
alfadur <mail@none>
parents: 14693
diff changeset
    90
            }
08a8605bafaf Server action refactoring part 4 of N
alfadur <mail@none>
parents: 14693
diff changeset
    91
        }
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
    92
        HwProtocolMessage::Proto(proto) => {
16120
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
    93
            let client = anteroom.get_client_mut(client_id);
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    94
            if client.protocol_number.is_some() {
15554
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
    95
                response.error(PROTOCOL_PROVIDED);
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
    96
                LoginResult::Unchanged
15562
479911540e17 backout a798e6441a36
alfadur <mail@none>
parents: 15558
diff changeset
    97
            } else if proto < 48 {
15554
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
    98
                response.add(Bye(PROTOCOL_TOO_OLD.to_string()).send_self());
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
    99
                LoginResult::Exit
14694
08a8605bafaf Server action refactoring part 4 of N
alfadur <mail@none>
parents: 14693
diff changeset
   100
            } else {
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
   101
                client.protocol_number = NonZeroU16::new(proto);
14694
08a8605bafaf Server action refactoring part 4 of N
alfadur <mail@none>
parents: 14693
diff changeset
   102
                response.add(Proto(proto).send_self());
08a8605bafaf Server action refactoring part 4 of N
alfadur <mail@none>
parents: 14693
diff changeset
   103
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
   104
                if client.nick.is_some() {
16120
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
   105
                    get_completion_result(anteroom, client_id, response)
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
   106
                } else {
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
   107
                    LoginResult::Unchanged
14694
08a8605bafaf Server action refactoring part 4 of N
alfadur <mail@none>
parents: 14693
diff changeset
   108
                }
14693
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14692
diff changeset
   109
            }
13771
4664da990556 Add official server feature to cargo
alfadur
parents: 13666
diff changeset
   110
        }
4664da990556 Add official server feature to cargo
alfadur
parents: 13666
diff changeset
   111
        #[cfg(feature = "official-server")]
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
   112
        HwProtocolMessage::Password(hash, salt) => {
16120
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
   113
            let client = anteroom.get_client(client_id);
13774
0118b7412570 use openssl for password message handling
alfadur
parents: 13771
diff changeset
   114
14800
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14717
diff changeset
   115
            if let (Some(nick), Some(protocol)) = (client.nick.as_ref(), client.protocol_number) {
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14717
diff changeset
   116
                response.request_io(super::IoTask::GetAccount {
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14717
diff changeset
   117
                    nick: nick.clone(),
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14717
diff changeset
   118
                    protocol: protocol.get(),
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14717
diff changeset
   119
                    server_salt: client.server_salt.clone(),
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14717
diff changeset
   120
                    client_salt: salt,
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14717
diff changeset
   121
                    password_hash: hash,
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14717
diff changeset
   122
                });
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14717
diff changeset
   123
            };
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14717
diff changeset
   124
f43ab2bd76ae add a thread for internal server IO and implement account checking with it
alfadur
parents: 14717
diff changeset
   125
            LoginResult::Unchanged
14693
6e6632068a33 Server action refactoring part 3 of N
alfadur <mail@none>
parents: 14692
diff changeset
   126
        }
13774
0118b7412570 use openssl for password message handling
alfadur
parents: 13771
diff changeset
   127
        #[cfg(feature = "official-server")]
15096
e935b1ad23f3 normalize type names
alfadur
parents: 15095
diff changeset
   128
        HwProtocolMessage::Checker(protocol, nick, password) => {
16120
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
   129
            let client = anteroom.get_client_mut(client_id);
14802
01f8ab45f806 fix lobby joining
alfadur
parents: 14800
diff changeset
   130
            if protocol == 0 {
15554
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
   131
                response.error("Bad number.");
14802
01f8ab45f806 fix lobby joining
alfadur
parents: 14800
diff changeset
   132
                LoginResult::Unchanged
01f8ab45f806 fix lobby joining
alfadur
parents: 14800
diff changeset
   133
            } else {
01f8ab45f806 fix lobby joining
alfadur
parents: 14800
diff changeset
   134
                client.protocol_number = NonZeroU16::new(protocol);
01f8ab45f806 fix lobby joining
alfadur
parents: 14800
diff changeset
   135
                client.is_checker = true;
15554
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
   136
                #[cfg(not(feature = "official-server"))]
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
   137
                {
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
   138
                    response.request_io(super::IoTask::GetCheckerAccount {
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
   139
                        nick: nick,
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
   140
                        password: password,
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
   141
                    });
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
   142
                    LoginResult::Unchanged
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
   143
                }
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
   144
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
   145
                #[cfg(feature = "official-server")]
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
   146
                {
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
   147
                    response.add(LogonPassed.send_self());
16120
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
   148
                    anteroom.remember_nick(nick);
5febd2bc5372 Add server reconnection tokens and anteroom local list of used nicks
alfadur
parents: 16019
diff changeset
   149
                    LoginResult::Complete(anteroom.remove_client(client_id).unwrap())
15554
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
   150
                }
14802
01f8ab45f806 fix lobby joining
alfadur
parents: 14800
diff changeset
   151
            }
13771
4664da990556 Add official server feature to cargo
alfadur
parents: 13666
diff changeset
   152
        }
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
   153
        _ => {
15554
f1205f33bf5b complete checker login handling
alfadur <mail@none>
parents: 15542
diff changeset
   154
            warn!("Incorrect command in anteroom");
14714
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
   155
            LoginResult::Unchanged
6a2e13e36b7f add server anteroom
alfadur <mail@none>
parents: 14711
diff changeset
   156
        }
12147
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
   157
    }
03ccb89820f3 Room creation halfplemented
unc0rr
parents:
diff changeset
   158
}