fcplug-rclib 0.2.1

Foreign Clang Plugin solution
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

use std::fmt::Debug;

use protobuf::{Message, ProtobufError};

use crate::{abi, ABIResult, Buffer, FFIResult};
use crate::abi::{ABIMessage, ABIRequest, ABIResponse, LeakBuffer, OriginType};

#[inline]
pub fn callback<'a, A, R, F>(_ffi_method_name: &'static str, f: F, args: &'a Buffer) -> FFIResult
    where A: Message,
          R: Message,
          F: Fn(A) -> ABIResult<R> {
    abi::callback::<'a, PbMessage<A>, PbMessage<R>, _>(_ffi_method_name, |req| {
        f(req.0).map(PbMessage)
    }, args)
}

#[derive(Debug)]
pub struct PbMessage<T: Message>(T);

impl<T: Message> PbMessage<T> {
    pub fn new(v: T) -> Self {
        Self(v)
    }
}

impl<'a, T: Message> ABIRequest<'a> for PbMessage<T> {
    type DecodeError = ProtobufError;

    fn try_from_bytes(buf: &'a [u8]) -> Result<Self, Self::DecodeError> where Self: Sized {
        T::parse_from_bytes(buf).map(PbMessage)
    }
}

impl<T: Message> ABIResponse for PbMessage<T> {
    type EncodeError = ProtobufError;
    const ORIGIN_TYPE_FOR_FREE: OriginType = OriginType::Vec;

    fn try_into_buffer(self) -> Result<LeakBuffer, Self::EncodeError> {
        self.0.write_to_bytes().map(|v| LeakBuffer::from_vec(v))
    }
}

impl<'a, T: Message> ABIMessage<'a> for PbMessage<T> {}