rust/hedgewars-server/src/protocol.rs
author unc0rr
Mon, 10 Dec 2018 22:44:46 +0100
changeset 14420 06672690d71b
parent 14380 gameServer2/src/protocol.rs@cc99f7c673c7
child 14462 98ef2913ec73
permissions -rw-r--r--
Move rust server into rust folder

use netbuf;
use std::{
    io::{Read, Result}
};
use nom::{
    IResult, Err
};

pub mod messages;
#[cfg(test)]
pub mod test;
mod parser;

pub struct ProtocolDecoder {
    buf: netbuf::Buf,
    consumed: usize,
}

impl ProtocolDecoder {
    pub fn new() -> ProtocolDecoder {
        ProtocolDecoder {
            buf: netbuf::Buf::new(),
            consumed: 0,
        }
    }

    pub fn read_from<R: Read>(&mut self, stream: &mut R) -> Result<usize> {
        self.buf.read_from(stream)
    }

    pub fn extract_messages(&mut self) -> Vec<messages::HWProtocolMessage> {
        let parse_result = parser::extract_messages(&self.buf[..]);
        match parse_result {
            Ok((tail, msgs)) => {
                self.consumed = self.buf.len() - self.consumed - tail.len();
                msgs
            },
            Err(Err::Incomplete(_)) => unreachable!(),
            Err(Err::Error(_)) | Err(Err::Failure(_)) => unreachable!(),
        }
    }

    pub fn sweep(&mut self) {
        self.buf.consume(self.consumed);
        self.consumed = 0;
    }
}