rust/hedgewars-server/src/protocol.rs
author sheepluva
Sat, 04 Jul 2020 03:11:41 +0200
changeset 15665 63e2b7b2ec47
parent 15439 a158ff8f84ef
child 15795 40929af15167
permissions -rw-r--r--
Merge chat-scaling (= current state of branch "ui-scaling") into default branch
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
15074
c5a6e8566425 shuffle server files
alfadur
parents: 14796
diff changeset
     1
use self::parser::message;
14795
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
     2
use log::*;
12129
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
     3
use netbuf;
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
     4
use std::io::{Read, Result};
12129
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
     5
12136
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
     6
pub mod messages;
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
     7
mod parser;
13796
59ea2403f62d move everything test related into test cfg
alfadur
parents: 13438
diff changeset
     8
#[cfg(test)]
13416
cdf69667593b partial room implementation
alfadur
parents: 12136
diff changeset
     9
pub mod test;
12129
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    10
12136
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    11
pub struct ProtocolDecoder {
12129
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    12
    buf: netbuf::Buf,
14795
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    13
    is_recovering: bool,
12129
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    14
}
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    15
12136
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    16
impl ProtocolDecoder {
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    17
    pub fn new() -> ProtocolDecoder {
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    18
        ProtocolDecoder {
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    19
            buf: netbuf::Buf::new(),
14795
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    20
            is_recovering: false,
12129
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    21
        }
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    22
    }
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    23
14795
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    24
    fn recover(&mut self) -> bool {
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    25
        self.is_recovering = match parser::malformed_message(&self.buf[..]) {
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    26
            Ok((tail, ())) => {
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15123
diff changeset
    27
                let length = tail.len();
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15123
diff changeset
    28
                self.buf.consume(self.buf.len() - length);
14795
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    29
                false
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    30
            }
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    31
            _ => {
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    32
                self.buf.consume(self.buf.len());
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    33
                true
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    34
            }
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    35
        };
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    36
        !self.is_recovering
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    37
    }
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    38
12129
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    39
    pub fn read_from<R: Read>(&mut self, stream: &mut R) -> Result<usize> {
14795
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    40
        let count = self.buf.read_from(stream)?;
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    41
        if count > 0 && self.is_recovering {
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    42
            self.recover();
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    43
        }
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    44
        Ok(count)
12129
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    45
    }
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    46
15075
e935b1ad23f3 normalize type names
alfadur
parents: 15074
diff changeset
    47
    pub fn extract_messages(&mut self) -> Vec<messages::HwProtocolMessage> {
14795
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    48
        let mut messages = vec![];
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    49
        if !self.is_recovering {
15123
1aa3b44c0441 fix parser getting stuck
alfadur
parents: 15114
diff changeset
    50
            while !self.buf.is_empty() {
14796
f5d43f007970 start by actually handling client messages
alfadur
parents: 14795
diff changeset
    51
                match parser::message(&self.buf[..]) {
14795
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    52
                    Ok((tail, message)) => {
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    53
                        messages.push(message);
15439
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15123
diff changeset
    54
                        let length = tail.len();
a158ff8f84ef refactor the lobby handler
alfadur
parents: 15123
diff changeset
    55
                        self.buf.consume(self.buf.len() - length);
14795
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    56
                    }
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    57
                    Err(nom::Err::Incomplete(_)) => break,
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    58
                    Err(nom::Err::Failure(e)) | Err(nom::Err::Error(e)) => {
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    59
                        debug!("Invalid message: {:?}", e);
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    60
                        if !self.recover() || self.buf.is_empty() {
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    61
                            break;
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    62
                        }
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    63
                    }
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    64
                }
14457
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14415
diff changeset
    65
            }
12136
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    66
        }
14795
add191d825f4 add parser error handling
alfadur
parents: 14775
diff changeset
    67
        messages
12129
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    68
    }
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    69
}