fcplug_rclib/
flatbuf.rs

1use std::convert::Infallible;
2use std::fmt::{Debug, Formatter};
3use std::marker::PhantomData;
4use std::ops::{Deref, DerefMut};
5
6pub use flatbuffers::{FlatBufferBuilder, WIPOffset};
7use flatbuffers::{Follow, InvalidFlatbuffer, Verifiable};
8#[cfg(debug_assertions)]
9use tracing::error;
10
11use crate::{ABIResult, Buffer, ResultCode};
12use crate::abi::{ABIRequest, ABIResponse, LeakBuffer, OriginType};
13
14pub struct FbRequest<'a, T> {
15    request: T,
16    _phantom: PhantomData<&'a T>,
17}
18
19impl<'a, T> FbRequest<'a, T> where T: 'a {
20    pub fn new(req: T) -> Self {
21        FbRequest { request: req, _phantom: Default::default() }
22    }
23    pub fn new_response_writer<U>(&self) -> FbResponseWriter<U> {
24        FbResponseWriter { fbb: Box::new(FlatBufferBuilder::new()), _phantom: Default::default() }
25    }
26    pub fn try_from_buffer(req: &'a Buffer) -> ABIResult<Self>
27        where T: 'a + Follow<'a, Inner=T> + Verifiable,
28    {
29        Self::try_from_bytes(req.read().unwrap_or_default())
30            .map_err(|_err| {
31                #[cfg(debug_assertions)]{
32                    error!("{:?}", _err);
33                }
34                ResultCode::Decode
35            })
36    }
37}
38
39impl<'a, T> Deref for FbRequest<'a, T> {
40    type Target = T;
41
42    fn deref(&self) -> &Self::Target {
43        &self.request
44    }
45}
46
47impl<'a, T> Debug for FbRequest<'a, T> {
48    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
49        write!(f, "{}", std::any::type_name::<T>())
50    }
51}
52
53impl<'a, T> ABIRequest<'a> for FbRequest<'a, T> where
54    T: 'a + Follow<'a, Inner=T> + Verifiable {
55    type DecodeError = InvalidFlatbuffer;
56
57    fn try_from_bytes(buf: &'a [u8]) -> Result<Self, Self::DecodeError> where Self: Sized {
58        flatbuffers::root::<T>(buf).map(|v| Self::new(v))
59    }
60}
61
62pub struct FbResponseWriter<T> {
63    fbb: Box<FlatBufferBuilder<'static>>,
64    _phantom: PhantomData<T>,
65}
66
67impl<T> FbResponseWriter<T> {
68    pub fn mut_fbb(&mut self) -> &mut FlatBufferBuilder<'static> {
69        &mut self.fbb
70    }
71    fn into_raw(self) -> *mut FlatBufferBuilder<'static> {
72        Box::into_raw(self.fbb)
73    }
74}
75
76impl<T> Deref for FbResponseWriter<T> {
77    type Target = FlatBufferBuilder<'static>;
78
79    fn deref(&self) -> &Self::Target {
80        &self.fbb
81    }
82}
83
84impl<T> DerefMut for FbResponseWriter<T> {
85    fn deref_mut(&mut self) -> &mut Self::Target {
86        &mut self.fbb
87    }
88}
89
90impl<T> Debug for FbResponseWriter<T> {
91    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
92        write!(f, "{}", std::any::type_name::<T>())
93    }
94}
95
96impl<T> ABIResponse for FbResponseWriter<T> {
97    type EncodeError = Infallible;
98    const ORIGIN_TYPE_FOR_FREE: OriginType = OriginType::FlatBuffer;
99
100    fn try_into_buffer(mut self) -> Result<LeakBuffer, Self::EncodeError> {
101        let b = self.mut_finished_buffer();
102        let mut buf = LeakBuffer {
103            free_type: OriginType::FlatBuffer,
104            free_ptr: 0,
105            buffer: Buffer {
106                ptr: b.0[b.1..].as_mut_ptr(),
107                len: b.0.len() - b.1,
108                cap: b.0.len(),
109            },
110        };
111        buf.free_ptr = self.into_raw() as usize;
112        Ok(buf)
113    }
114}
fcplug_rclib/flatbuf.rs

fcplug_rclib/
flatbuf.rs
