esp32-nimble 0.12.0

A wrapper for the ESP32 NimBLE Bluetooth stack.
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use alloc::vec::Vec;
use core::ffi::c_int;
use esp_idf_svc::sys;

#[cfg(not(esp_idf_soc_esp_nimble_controller))]
use sys::os_mbuf_append as _os_mbuf_append;

#[cfg(esp_idf_soc_esp_nimble_controller)]
use sys::r_os_mbuf_append as _os_mbuf_append;

#[derive(Copy, Clone)]
pub(crate) struct OsMBuf(pub *mut sys::os_mbuf);

#[allow(unused)]
impl OsMBuf {
    #[inline]
    pub fn as_slice<'a>(&self) -> &'a [u8] {
        unsafe { core::slice::from_raw_parts((*self.0).om_data, (*self.0).om_len as _) }
    }

    #[inline]
    pub fn entire_len(&self) -> u16 {
        unsafe { (*self.0.add(1).cast::<sys::os_mbuf_pkthdr>()).omp_len }
    }

    /// Append data onto a mbuf
    #[inline]
    pub(crate) fn append(&mut self, data: &[u8]) -> c_int {
        unsafe { _os_mbuf_append(self.0, data.as_ptr() as _, data.len() as _) }
    }

    #[inline]
    pub fn from_flat(buf: &[u8]) -> Self {
        OsMBuf(unsafe { sys::ble_hs_mbuf_from_flat(buf.as_ptr() as _, buf.len() as _) })
    }

    #[inline]
    pub fn iter(&self) -> OsMBufIterator {
        OsMBufIterator(*self)
    }

    #[inline]
    fn has_next(&self) -> bool {
        unsafe { !(*self.0).om_next.sle_next.is_null() }
    }

    #[inline]
    pub fn as_flat(&self) -> FlatData<'_> {
        if (self.has_next()) {
            let mut buf = Vec::with_capacity(self.entire_len() as _);

            for mbuf in self.iter() {
                buf.extend_from_slice(mbuf.as_slice());
            }

            FlatData::Vec(buf)
        } else {
            FlatData::Slice(self.as_slice())
        }
    }
}

pub(crate) struct OsMBufIterator(OsMBuf);

impl Iterator for OsMBufIterator {
    type Item = OsMBuf;

    fn next(&mut self) -> Option<Self::Item> {
        if (self.0).0.is_null() {
            None
        } else {
            let current = self.0;

            self.0 = OsMBuf(unsafe { (*self.0.0).om_next.sle_next });

            Some(current)
        }
    }
}

pub(crate) enum FlatData<'a> {
    Vec(Vec<u8>),
    Slice(&'a [u8]),
}

impl FlatData<'_> {
    pub fn as_slice(&self) -> &[u8] {
        match self {
            FlatData::Vec(v) => v.as_slice(),
            FlatData::Slice(s) => s,
        }
    }
}