rust/hedgewars-server/src/protocol.rs
author Wuzzy <Wuzzy2@mail.ru>
Sun, 07 Apr 2019 19:26:16 +0200
changeset 14755 ab7bf5036314
parent 14462 98ef2913ec73
child 14780 09d46ab83361
permissions -rw-r--r--
Render mine timer if gear info is active
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
12134
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
     1
use netbuf;
14462
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14420
diff changeset
     2
use nom::{Err, IResult};
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14420
diff changeset
     3
use std::io::{Read, Result};
12134
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
     4
12141
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
     5
pub mod messages;
14462
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14420
diff changeset
     6
mod parser;
13801
59ea2403f62d move everything test related into test cfg
alfadur
parents: 13443
diff changeset
     7
#[cfg(test)]
13421
cdf69667593b partial room implementation
alfadur
parents: 12141
diff changeset
     8
pub mod test;
12134
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
     9
12141
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    10
pub struct ProtocolDecoder {
12134
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    11
    buf: netbuf::Buf,
12141
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    12
    consumed: usize,
12134
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    13
}
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    14
12141
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    15
impl ProtocolDecoder {
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    16
    pub fn new() -> ProtocolDecoder {
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    17
        ProtocolDecoder {
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    18
            buf: netbuf::Buf::new(),
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    19
            consumed: 0,
12134
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    20
        }
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
    pub fn read_from<R: Read>(&mut self, stream: &mut R) -> Result<usize> {
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    24
        self.buf.read_from(stream)
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    25
    }
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    26
12141
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    27
    pub fn extract_messages(&mut self) -> Vec<messages::HWProtocolMessage> {
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    28
        let parse_result = parser::extract_messages(&self.buf[..]);
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    29
        match parse_result {
13443
da71e0d88a1c update nom
alfadur
parents: 13421
diff changeset
    30
            Ok((tail, msgs)) => {
12141
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    31
                self.consumed = self.buf.len() - self.consumed - tail.len();
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    32
                msgs
14462
98ef2913ec73 Apply rustfmt to all files
unc0rr
parents: 14420
diff changeset
    33
            }
13443
da71e0d88a1c update nom
alfadur
parents: 13421
diff changeset
    34
            Err(Err::Incomplete(_)) => unreachable!(),
da71e0d88a1c update nom
alfadur
parents: 13421
diff changeset
    35
            Err(Err::Error(_)) | Err(Err::Failure(_)) => unreachable!(),
12141
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    36
        }
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    37
    }
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    38
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    39
    pub fn sweep(&mut self) {
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    40
        self.buf.consume(self.consumed);
e25a82ce2374 - Render messages to string
unc0rr
parents: 12138
diff changeset
    41
        self.consumed = 0;
12134
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    42
    }
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    43
}