Merge pull request #11 from woshiluo/chosen

feat: add chosen, chosen_stdout_path, chosen_stdin_path for Node

Author: luojia65

.gitignore

@@ -1,5 +1,6 @@
 /target
 Cargo.lock
+*.dts
 
 # Visual Studio Code configuration files
 .vscode/

examples/bl808.dtb

@@ -47,9 +47,9 @@ impl<'de> Deserialize<'de> for StrSeq<'_> {
     }
 }
 
-impl StrSeq<'_> {
+impl<'de> StrSeq<'de> {
     /// 构造一个可访问每个字符串的迭代器。
-    pub fn iter(&self) -> StrSeqIter {
+    pub fn iter<'b>(&'b self) -> StrSeqIter<'de> {
         StrSeqIter {
             data: self.0.cursor.data_on(self.0.dtb),
         }

examples/cv1812cp_milkv_duo256m_sd.dts

@@ -0,0 +1,63 @@
+use crate::buildin::{Node, StrSeq};
+
+impl<'de> Node<'de> {
+    /// Get node /chosen
+    #[inline]
+    pub fn chosen<'b>(&'b self) -> Option<Node<'de>> {
+        self.find("/chosen")
+    }
+    /// Get /chosen/stdin-path
+    pub fn chosen_stdin_path(&self) -> Option<&'de str> {
+        let result = self
+            .chosen()?
+            .get_prop("stdin-path")?
+            .deserialize::<StrSeq>()
+            .iter()
+            .next()?;
+        if let Some(pos) = result.find(':') {
+            Some(result.split_at(pos).0)
+        } else {
+            Some(result)
+        }
+    }
+    /// Get /chosen/stdout-path
+    pub fn chosen_stdout_path(&self) -> Option<&'de str> {
+        let result = self
+            .chosen()?
+            .get_prop("stdout-path")?
+            .deserialize::<StrSeq>()
+            .iter()
+            .next()?;
+        if let Some(pos) = result.find(':') {
+            Some(result.split_at(pos).0)
+        } else {
+            Some(result)
+        }
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::{buildin::Node, from_raw_mut, Dtb, DtbPtr};
+
+    const RAW_DEVICE_TREE: &[u8] = include_bytes!("../../examples/bl808.dtb");
+    const BUFFER_SIZE: usize = RAW_DEVICE_TREE.len();
+    #[test]
+    fn test_chosen_stdout() {
+        #[repr(align(8))]
+        struct AlignedBuffer {
+            pub data: [u8; RAW_DEVICE_TREE.len()],
+        }
+        let mut aligned_data: Box<AlignedBuffer> = Box::new(AlignedBuffer {
+            data: [0; BUFFER_SIZE],
+        });
+        aligned_data.data[..BUFFER_SIZE].clone_from_slice(RAW_DEVICE_TREE);
+        let mut slice = aligned_data.data.to_vec();
+        let ptr = DtbPtr::from_raw(slice.as_mut_ptr()).unwrap();
+        let dtb = Dtb::from(ptr).share();
+
+        let node: Node = from_raw_mut(&dtb).unwrap();
+        assert!(node.chosen().is_some());
+        assert_eq!(node.chosen_stdout_path(), Some("serial3"));
+    }
+}

examples/hifive-unmatched-a00.dts

@@ -1,8 +1,10 @@
+pub mod chosen;
+
 use crate::buildin::{Node, StrSeq};
 
-impl Node<'_> {
+impl<'de> Node<'de> {
     /// Try to get a node by a full-path.
-    fn raw_find(&self, path: &str) -> Option<Node> {
+    fn raw_find<'b>(&'b self, path: &str) -> Option<Node<'de>> {
         // Direct return root node
         let mut current_node = Some(self.clone());
         if path == "/" {
@@ -30,7 +32,7 @@ impl Node<'_> {
         current_node
     }
     /// Try to get a node by path.
-    pub fn find(&self, path: &str) -> Option<Node> {
+    pub fn find<'b>(&'b self, path: &str) -> Option<Node<'de>> {
         // Direct return root node
         let current_node = Some(self.clone());
         if path == "/" {