gameServer2/src/protocol/mod.rs
author alfadur
Wed, 04 Jul 2018 00:01:25 +0300
changeset 13438 da71e0d88a1c
parent 13416 cdf69667593b
child 13796 59ea2403f62d
permissions -rw-r--r--
update nom
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
12129
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
     1
use netbuf;
13438
da71e0d88a1c update nom
alfadur
parents: 13416
diff changeset
     2
use std::{
da71e0d88a1c update nom
alfadur
parents: 13416
diff changeset
     3
    io::{Read, Result}
da71e0d88a1c update nom
alfadur
parents: 13416
diff changeset
     4
};
da71e0d88a1c update nom
alfadur
parents: 13416
diff changeset
     5
use nom::{
da71e0d88a1c update nom
alfadur
parents: 13416
diff changeset
     6
    IResult, Err
da71e0d88a1c update nom
alfadur
parents: 13416
diff changeset
     7
};
12129
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
     8
12136
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
     9
pub mod messages;
13416
cdf69667593b partial room implementation
alfadur
parents: 12136
diff changeset
    10
pub mod test;
12133
81df2e1f9ae9 Some parsing using nom
unc0rr
parents: 12132
diff changeset
    11
mod parser;
12129
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    12
12136
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    13
pub struct ProtocolDecoder {
12129
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    14
    buf: netbuf::Buf,
12136
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    15
    consumed: usize,
12129
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    16
}
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    17
12136
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    18
impl ProtocolDecoder {
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    19
    pub fn new() -> ProtocolDecoder {
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    20
        ProtocolDecoder {
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    21
            buf: netbuf::Buf::new(),
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    22
            consumed: 0,
12129
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    23
        }
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    24
    }
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    25
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    26
    pub fn read_from<R: Read>(&mut self, stream: &mut R) -> Result<usize> {
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    27
        self.buf.read_from(stream)
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    28
    }
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    29
12136
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    30
    pub fn extract_messages(&mut self) -> Vec<messages::HWProtocolMessage> {
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    31
        let parse_result = parser::extract_messages(&self.buf[..]);
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    32
        match parse_result {
13438
da71e0d88a1c update nom
alfadur
parents: 13416
diff changeset
    33
            Ok((tail, msgs)) => {
12136
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    34
                self.consumed = self.buf.len() - self.consumed - tail.len();
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    35
                msgs
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    36
            },
13438
da71e0d88a1c update nom
alfadur
parents: 13416
diff changeset
    37
            Err(Err::Incomplete(_)) => unreachable!(),
da71e0d88a1c update nom
alfadur
parents: 13416
diff changeset
    38
            Err(Err::Error(_)) | Err(Err::Failure(_)) => unreachable!(),
12136
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    39
        }
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    40
    }
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    41
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    42
    pub fn sweep(&mut self) {
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    43
        self.buf.consume(self.consumed);
e25a82ce2374 - Render messages to string
unc0rr
parents: 12133
diff changeset
    44
        self.consumed = 0;
12129
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    45
    }
07972a8c2433 - Start protocol parser implementation
unc0rr
parents:
diff changeset
    46
}