rpc-core 0.3.2

a tiny rpc library for rust
Documentation 
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

use crate::detail::msg_wrapper::{MsgType, MsgWrapper};
use crate::type_def::SeqType;

const PAYLOAD_MIN_LEN: usize = 4 /*seq*/ + 2 /*cmdLen*/ + 1 /*type*/;

pub fn serialize(msg: &MsgWrapper) -> Vec<u8> {
    let mut payload: Vec<u8> = vec![];
    payload.reserve(PAYLOAD_MIN_LEN);
    payload.extend_from_slice(&msg.seq.to_le_bytes());
    let cmd_len: u16 = msg.cmd.len() as u16;
    payload.extend_from_slice(&cmd_len.to_le_bytes());
    payload.extend(msg.cmd.bytes());
    let type_: u8 = msg.type_.bits();
    payload.extend_from_slice(&type_.to_le_bytes());
    if let Some(request_payload) = &msg.request_payload {
        payload.extend(request_payload);
    } else {
        payload.extend(&msg.data);
    }
    payload
}

pub fn deserialize(payload: &Vec<u8>) -> Option<MsgWrapper> {
    if payload.len() < PAYLOAD_MIN_LEN {
        return None;
    }

    let mut msg = MsgWrapper::new();
    unsafe {
        let mut p = payload.as_ptr();
        let pend = p.add(payload.len());

        msg.seq = *(p as *const SeqType);
        p = p.add(4);

        let cmd_len: u16;
        cmd_len = *(p as *const u16);
        p = p.add(2);

        msg.cmd = String::from_utf8_unchecked(Vec::from(std::slice::from_raw_parts(
            p as *mut u8,
            cmd_len as usize,
        )));
        p = p.add(cmd_len as usize);

        msg.type_ = MsgType::from_bits(*p).unwrap();
        p = p.add(1);

        msg.data = Vec::from(std::slice::from_raw_parts(p, pend.offset_from(p) as usize));
    }

    Some(msg)
}