nvme/ds/identify/
ns_desc_list.rs

1#![allow(dead_code)]
2use modular_bitfield::prelude::*;
3
4pub struct NamespaceIdentBuffer {
5    data: [u8; 4096],
6}
7
8#[bitfield(bits = 32)]
9struct NamespaceIdentDescriptorHead {
10    #[skip(setters)]
11    ident_type: u8,
12    #[skip(setters)]
13    ident_len: u8,
14    #[skip]
15    res: u16,
16}
17
18pub enum NamespaceDescriptor<'a> {
19    IeeeExtended(&'a [u8]),
20    NamespaceGUID(&'a [u8]),
21    NamespaceUUID(&'a [u8]),
22    CommandSetId(&'a [u8]),
23}
24
25pub struct NamespaceIdentBufferIterator<'a> {
26    buf: &'a NamespaceIdentBuffer,
27    pos: usize,
28}
29
30impl NamespaceIdentBuffer {
31    pub fn iter(&self) -> NamespaceIdentBufferIterator<'_> {
32        NamespaceIdentBufferIterator { buf: self, pos: 0 }
33    }
34}
35
36impl<'a> Iterator for NamespaceIdentBufferIterator<'a> {
37    type Item = NamespaceDescriptor<'a>;
38
39    fn next(&mut self) -> Option<Self::Item> {
40        if self.pos >= 4096 {
41            return None;
42        }
43        let head =
44            NamespaceIdentDescriptorHead::from_bytes(self.buf.data[0..4].try_into().unwrap());
45        let desc_len = head.ident_len() as usize + 4;
46        let loc = self.pos;
47        self.pos += desc_len;
48        if self.pos > 4096 {
49            return None;
50        }
51
52        let slice = &self.buf.data[(loc + 4)..(loc + 4 + head.ident_len() as usize)];
53
54        Some(match head.ident_type() {
55            1 => NamespaceDescriptor::IeeeExtended(slice),
56            2 => NamespaceDescriptor::NamespaceGUID(slice),
57            3 => NamespaceDescriptor::NamespaceUUID(slice),
58            4 => NamespaceDescriptor::CommandSetId(slice),
59            _ => return self.next(),
60        })
61    }
62}
nvme/ds/identify/ns_desc_list.rs

nvme/ds/identify/
ns_desc_list.rs
