1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
#[macro_export]
macro_rules! parse_match {
( $matchee:expr, $pos:expr, $map:expr, $bytes:expr, $name:expr, $(($pattern:pat, $type:ty, $variant:expr)),* ) => {
match $matchee {
$(
$pattern => {
let value: $type;
$pos = $pos + value.parse(&$bytes[($pos)..]);
$map.insert($name, $variant(value));
},
)*
};
};
}
#[macro_export]
macro_rules! match_variant {
( $values:expr, $x:path, $field:expr ) => {
match &$values[$field] {
$x(x) => { Ok(x.clone()) },
_ => { Err("") }
}.unwrap();
}
}
use crate::protocol::primitive::{Variant};
use crate::protocol::error::ProtocolError;
use failure::Error;
pub fn get_msg_type(val: &Variant) -> Result<&str, Error> {
match val {
Variant::String(x) => return Ok(x),
Variant::StringUTF8(x) => return Ok(x),
_ => bail!(ProtocolError::WrongVariant)
};
}
pub fn prepend_byte_len(buf: &mut Vec<u8>) {
use std::convert::TryInto;
let len: i32 = buf.len().try_into().unwrap();
let ulen: &[u8] = &len.to_be_bytes();
buf.insert(0, ulen[3]);
buf.insert(0, ulen[2]);
buf.insert(0, ulen[1]);
buf.insert(0, ulen[0]);
}
pub fn insert_bytes(pos: usize, buf: &mut Vec<u8>, input: &mut [u8]) {
input.reverse();
for i in input {
buf.insert(pos, *i)
}
}
|
