blob: 58b4463619ed63b0860a0f08b3b730fa03baff8b [file] [log] [blame]
David Brazdil1baa9a92022-06-28 14:47:50 +01001// Copyright 2022, The Android Open Source Project
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15//! Wrapper around libfdt library. Provides parsing/generating functionality
16//! to a bare-metal environment.
17
18#![no_std]
19
Pierre-Clément Tosiecd5bbc2023-12-21 15:12:45 +000020mod ctypes;
Andrew Walbran55ad01b2022-12-05 17:00:40 +000021mod iterators;
Pierre-Clément Tosifbb5ee22023-12-21 13:49:59 +000022mod libfdt;
Pierre-Clément Tosi99a76902023-12-21 10:30:37 +000023mod result;
Andrew Walbran55ad01b2022-12-05 17:00:40 +000024
Pierre-Clément Tosiecd5bbc2023-12-21 15:12:45 +000025pub use ctypes::Phandle;
Jaewan Kimfe06c852023-10-05 23:40:06 +090026pub use iterators::{
Jaewan Kimc9e14112023-12-04 17:05:27 +090027 AddressRange, CellIterator, CompatibleIterator, DescendantsIterator, MemRegIterator,
28 PropertyIterator, RangesIterator, Reg, RegIterator, SubnodeIterator,
Jaewan Kimfe06c852023-10-05 23:40:06 +090029};
Pierre-Clément Tosi99a76902023-12-21 10:30:37 +000030pub use result::{FdtError, Result};
Andrew Walbran55ad01b2022-12-05 17:00:40 +000031
David Brazdil1baa9a92022-06-28 14:47:50 +010032use core::ffi::{c_int, c_void, CStr};
Alice Wang2422bdc2023-06-12 08:37:55 +000033use core::ops::Range;
Pierre-Clément Tosi1bf532b2023-11-13 11:06:20 +000034use cstr::cstr;
Pierre-Clément Tosi60282ae2023-12-21 16:00:02 +000035use libfdt::get_slice_at_ptr;
Pierre-Clément Tosi99a76902023-12-21 10:30:37 +000036use result::{fdt_err, fdt_err_expect_zero, fdt_err_or_option};
Pierre-Clément Tosic27c4272023-05-19 15:46:26 +000037use zerocopy::AsBytes as _;
David Brazdil1baa9a92022-06-28 14:47:50 +010038
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +000039use crate::libfdt::{Libfdt, LibfdtMut};
40
David Brazdil1baa9a92022-06-28 14:47:50 +010041/// Value of a #address-cells property.
Andrew Walbranb39e6922022-12-05 17:01:20 +000042#[derive(Copy, Clone, Debug, Eq, PartialEq)]
David Brazdil1baa9a92022-06-28 14:47:50 +010043enum AddrCells {
44 Single = 1,
45 Double = 2,
Andrew Walbranb39e6922022-12-05 17:01:20 +000046 Triple = 3,
David Brazdil1baa9a92022-06-28 14:47:50 +010047}
48
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +000049impl TryFrom<usize> for AddrCells {
David Brazdil1baa9a92022-06-28 14:47:50 +010050 type Error = FdtError;
51
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +000052 fn try_from(value: usize) -> Result<Self> {
53 match value {
54 x if x == Self::Single as _ => Ok(Self::Single),
55 x if x == Self::Double as _ => Ok(Self::Double),
56 x if x == Self::Triple as _ => Ok(Self::Triple),
David Brazdil1baa9a92022-06-28 14:47:50 +010057 _ => Err(FdtError::BadNCells),
58 }
59 }
60}
61
62/// Value of a #size-cells property.
Andrew Walbranb39e6922022-12-05 17:01:20 +000063#[derive(Copy, Clone, Debug, Eq, PartialEq)]
David Brazdil1baa9a92022-06-28 14:47:50 +010064enum SizeCells {
65 None = 0,
66 Single = 1,
67 Double = 2,
68}
69
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +000070impl TryFrom<usize> for SizeCells {
David Brazdil1baa9a92022-06-28 14:47:50 +010071 type Error = FdtError;
72
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +000073 fn try_from(value: usize) -> Result<Self> {
74 match value {
75 x if x == Self::None as _ => Ok(Self::None),
76 x if x == Self::Single as _ => Ok(Self::Single),
77 x if x == Self::Double as _ => Ok(Self::Double),
David Brazdil1baa9a92022-06-28 14:47:50 +010078 _ => Err(FdtError::BadNCells),
79 }
80 }
81}
82
Jaewan Kim72d10902023-10-12 21:59:26 +090083/// DT property wrapper to abstract endianess changes
84#[repr(transparent)]
85#[derive(Debug)]
86struct FdtPropertyStruct(libfdt_bindgen::fdt_property);
87
Pierre-Clément Tosidf3037f2024-01-22 15:41:43 +000088impl AsRef<FdtPropertyStruct> for libfdt_bindgen::fdt_property {
89 fn as_ref(&self) -> &FdtPropertyStruct {
90 let ptr = self as *const _ as *const _;
91 // SAFETY: Types have the same layout (transparent) so the valid reference remains valid.
92 unsafe { &*ptr }
93 }
94}
95
Jaewan Kim72d10902023-10-12 21:59:26 +090096impl FdtPropertyStruct {
97 fn from_offset(fdt: &Fdt, offset: c_int) -> Result<&Self> {
Pierre-Clément Tosi410e46a2023-12-24 11:33:11 +000098 Ok(fdt.get_property_by_offset(offset)?.as_ref())
Jaewan Kim72d10902023-10-12 21:59:26 +090099 }
100
101 fn name_offset(&self) -> c_int {
102 u32::from_be(self.0.nameoff).try_into().unwrap()
103 }
104
105 fn data_len(&self) -> usize {
106 u32::from_be(self.0.len).try_into().unwrap()
107 }
108
109 fn data_ptr(&self) -> *const c_void {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000110 self.0.data.as_ptr().cast()
Jaewan Kim72d10902023-10-12 21:59:26 +0900111 }
112}
113
114/// DT property.
115#[derive(Clone, Copy, Debug)]
116pub struct FdtProperty<'a> {
117 fdt: &'a Fdt,
118 offset: c_int,
119 property: &'a FdtPropertyStruct,
120}
121
122impl<'a> FdtProperty<'a> {
123 fn new(fdt: &'a Fdt, offset: c_int) -> Result<Self> {
124 let property = FdtPropertyStruct::from_offset(fdt, offset)?;
125 Ok(Self { fdt, offset, property })
126 }
127
128 /// Returns the property name
129 pub fn name(&self) -> Result<&'a CStr> {
130 self.fdt.string(self.property.name_offset())
131 }
132
133 /// Returns the property value
134 pub fn value(&self) -> Result<&'a [u8]> {
135 self.fdt.get_from_ptr(self.property.data_ptr(), self.property.data_len())
136 }
137
138 fn next_property(&self) -> Result<Option<Self>> {
Pierre-Clément Tosi410e46a2023-12-24 11:33:11 +0000139 if let Some(offset) = self.fdt.next_property_offset(self.offset)? {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000140 Ok(Some(Self::new(self.fdt, offset)?))
141 } else {
142 Ok(None)
143 }
Jaewan Kim72d10902023-10-12 21:59:26 +0900144 }
145}
146
David Brazdil1baa9a92022-06-28 14:47:50 +0100147/// DT node.
Alice Wang9d4df702023-05-25 14:14:12 +0000148#[derive(Clone, Copy, Debug)]
David Brazdil1baa9a92022-06-28 14:47:50 +0100149pub struct FdtNode<'a> {
150 fdt: &'a Fdt,
151 offset: c_int,
152}
153
154impl<'a> FdtNode<'a> {
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900155 /// Returns parent node.
David Brazdil1baa9a92022-06-28 14:47:50 +0100156 pub fn parent(&self) -> Result<Self> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000157 let offset = self.fdt.parent_offset(self.offset)?;
David Brazdil1baa9a92022-06-28 14:47:50 +0100158
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000159 Ok(Self { fdt: self.fdt, offset })
David Brazdil1baa9a92022-06-28 14:47:50 +0100160 }
161
Jaewan Kim5b057772023-10-19 01:02:17 +0900162 /// Returns supernode with depth. Note that root is at depth 0.
163 pub fn supernode_at_depth(&self, depth: usize) -> Result<Self> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000164 let offset = self.fdt.supernode_atdepth_offset(self.offset, depth)?;
Jaewan Kim5b057772023-10-19 01:02:17 +0900165
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000166 Ok(Self { fdt: self.fdt, offset })
Jaewan Kim5b057772023-10-19 01:02:17 +0900167 }
168
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900169 /// Returns the standard (deprecated) device_type <string> property.
Pierre-Clément Tosib244d932022-11-24 16:45:53 +0000170 pub fn device_type(&self) -> Result<Option<&CStr>> {
Jaewan Kimb635bb02023-11-01 13:00:34 +0900171 self.getprop_str(cstr!("device_type"))
David Brazdil1baa9a92022-06-28 14:47:50 +0100172 }
173
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900174 /// Returns the standard reg <prop-encoded-array> property.
Pierre-Clément Tosib244d932022-11-24 16:45:53 +0000175 pub fn reg(&self) -> Result<Option<RegIterator<'a>>> {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000176 if let Some(cells) = self.getprop_cells(cstr!("reg"))? {
Pierre-Clément Tosib244d932022-11-24 16:45:53 +0000177 let parent = self.parent()?;
David Brazdil1baa9a92022-06-28 14:47:50 +0100178
Pierre-Clément Tosib244d932022-11-24 16:45:53 +0000179 let addr_cells = parent.address_cells()?;
180 let size_cells = parent.size_cells()?;
181
182 Ok(Some(RegIterator::new(cells, addr_cells, size_cells)))
183 } else {
184 Ok(None)
185 }
David Brazdil1baa9a92022-06-28 14:47:50 +0100186 }
187
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900188 /// Returns the standard ranges property.
Andrew Walbranb39e6922022-12-05 17:01:20 +0000189 pub fn ranges<A, P, S>(&self) -> Result<Option<RangesIterator<'a, A, P, S>>> {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000190 if let Some(cells) = self.getprop_cells(cstr!("ranges"))? {
Andrew Walbranb39e6922022-12-05 17:01:20 +0000191 let parent = self.parent()?;
192 let addr_cells = self.address_cells()?;
193 let parent_addr_cells = parent.address_cells()?;
194 let size_cells = self.size_cells()?;
195 Ok(Some(RangesIterator::<A, P, S>::new(
196 cells,
197 addr_cells,
198 parent_addr_cells,
199 size_cells,
200 )))
201 } else {
202 Ok(None)
203 }
204 }
205
Jaewan Kimaa638702023-09-19 13:34:01 +0900206 /// Returns the node name.
207 pub fn name(&self) -> Result<&'a CStr> {
Pierre-Clément Tosi60282ae2023-12-21 16:00:02 +0000208 let name = self.fdt.get_name(self.offset)?;
Jaewan Kimaa638702023-09-19 13:34:01 +0900209 CStr::from_bytes_with_nul(name).map_err(|_| FdtError::Internal)
210 }
211
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900212 /// Returns the value of a given <string> property.
Pierre-Clément Tosib244d932022-11-24 16:45:53 +0000213 pub fn getprop_str(&self, name: &CStr) -> Result<Option<&CStr>> {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000214 if let Some(bytes) = self.getprop(name)? {
215 Ok(Some(CStr::from_bytes_with_nul(bytes).map_err(|_| FdtError::BadValue)?))
Pierre-Clément Tosib244d932022-11-24 16:45:53 +0000216 } else {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000217 Ok(None)
218 }
David Brazdil1baa9a92022-06-28 14:47:50 +0100219 }
220
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900221 /// Returns the value of a given property as an array of cells.
Pierre-Clément Tosib244d932022-11-24 16:45:53 +0000222 pub fn getprop_cells(&self, name: &CStr) -> Result<Option<CellIterator<'a>>> {
223 if let Some(cells) = self.getprop(name)? {
224 Ok(Some(CellIterator::new(cells)))
225 } else {
226 Ok(None)
227 }
David Brazdil1baa9a92022-06-28 14:47:50 +0100228 }
229
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900230 /// Returns the value of a given <u32> property.
Pierre-Clément Tosib244d932022-11-24 16:45:53 +0000231 pub fn getprop_u32(&self, name: &CStr) -> Result<Option<u32>> {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000232 if let Some(bytes) = self.getprop(name)? {
233 Ok(Some(u32::from_be_bytes(bytes.try_into().map_err(|_| FdtError::BadValue)?)))
Pierre-Clément Tosib244d932022-11-24 16:45:53 +0000234 } else {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000235 Ok(None)
236 }
David Brazdil1baa9a92022-06-28 14:47:50 +0100237 }
238
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900239 /// Returns the value of a given <u64> property.
Pierre-Clément Tosib244d932022-11-24 16:45:53 +0000240 pub fn getprop_u64(&self, name: &CStr) -> Result<Option<u64>> {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000241 if let Some(bytes) = self.getprop(name)? {
242 Ok(Some(u64::from_be_bytes(bytes.try_into().map_err(|_| FdtError::BadValue)?)))
Pierre-Clément Tosib244d932022-11-24 16:45:53 +0000243 } else {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000244 Ok(None)
245 }
David Brazdil1baa9a92022-06-28 14:47:50 +0100246 }
247
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900248 /// Returns the value of a given property.
Pierre-Clément Tosib244d932022-11-24 16:45:53 +0000249 pub fn getprop(&self, name: &CStr) -> Result<Option<&'a [u8]>> {
Pierre-Clément Tosi410e46a2023-12-24 11:33:11 +0000250 self.fdt.getprop_namelen(self.offset, name.to_bytes())
David Brazdil1baa9a92022-06-28 14:47:50 +0100251 }
252
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900253 /// Returns reference to the containing device tree.
David Brazdil1baa9a92022-06-28 14:47:50 +0100254 pub fn fdt(&self) -> &Fdt {
255 self.fdt
256 }
257
Alice Wang474c0ee2023-09-14 12:52:33 +0000258 /// Returns the compatible node of the given name that is next after this node.
259 pub fn next_compatible(self, compatible: &CStr) -> Result<Option<Self>> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000260 let offset = self.fdt.node_offset_by_compatible(self.offset, compatible)?;
Pierre-Clément Tosi41c158e2022-11-21 19:16:25 +0000261
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000262 Ok(offset.map(|offset| Self { fdt: self.fdt, offset }))
Pierre-Clément Tosi41c158e2022-11-21 19:16:25 +0000263 }
264
Alice Wang474c0ee2023-09-14 12:52:33 +0000265 /// Returns the first range of `reg` in this node.
266 pub fn first_reg(&self) -> Result<Reg<u64>> {
267 self.reg()?.ok_or(FdtError::NotFound)?.next().ok_or(FdtError::NotFound)
268 }
269
David Brazdil1baa9a92022-06-28 14:47:50 +0100270 fn address_cells(&self) -> Result<AddrCells> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000271 self.fdt.address_cells(self.offset)?.try_into()
David Brazdil1baa9a92022-06-28 14:47:50 +0100272 }
273
274 fn size_cells(&self) -> Result<SizeCells> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000275 self.fdt.size_cells(self.offset)?.try_into()
David Brazdil1baa9a92022-06-28 14:47:50 +0100276 }
Jaewan Kimbc828d72023-09-19 15:52:08 +0900277
278 /// Returns an iterator of subnodes
Jaewan Kim4a34b0d2024-01-19 13:17:47 +0900279 pub fn subnodes(&self) -> Result<SubnodeIterator<'a>> {
Jaewan Kimbc828d72023-09-19 15:52:08 +0900280 SubnodeIterator::new(self)
281 }
282
283 fn first_subnode(&self) -> Result<Option<Self>> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000284 let offset = self.fdt.first_subnode(self.offset)?;
Jaewan Kimbc828d72023-09-19 15:52:08 +0900285
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000286 Ok(offset.map(|offset| Self { fdt: self.fdt, offset }))
Jaewan Kimbc828d72023-09-19 15:52:08 +0900287 }
288
289 fn next_subnode(&self) -> Result<Option<Self>> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000290 let offset = self.fdt.next_subnode(self.offset)?;
Jaewan Kimbc828d72023-09-19 15:52:08 +0900291
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000292 Ok(offset.map(|offset| Self { fdt: self.fdt, offset }))
Jaewan Kimbc828d72023-09-19 15:52:08 +0900293 }
Jaewan Kim72d10902023-10-12 21:59:26 +0900294
Jaewan Kimc9e14112023-12-04 17:05:27 +0900295 /// Returns an iterator of descendants
Jaewan Kim1eab7232024-01-04 09:46:16 +0900296 pub fn descendants(&self) -> DescendantsIterator<'a> {
Jaewan Kimc9e14112023-12-04 17:05:27 +0900297 DescendantsIterator::new(self)
298 }
299
300 fn next_node(&self, depth: usize) -> Result<Option<(Self, usize)>> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000301 if let Some((offset, depth)) = self.fdt.next_node(self.offset, depth)? {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000302 Ok(Some((Self { fdt: self.fdt, offset }, depth)))
303 } else {
304 Ok(None)
305 }
Jaewan Kimc9e14112023-12-04 17:05:27 +0900306 }
307
Jaewan Kim72d10902023-10-12 21:59:26 +0900308 /// Returns an iterator of properties
309 pub fn properties(&'a self) -> Result<PropertyIterator<'a>> {
310 PropertyIterator::new(self)
311 }
312
313 fn first_property(&self) -> Result<Option<FdtProperty<'a>>> {
Pierre-Clément Tosi410e46a2023-12-24 11:33:11 +0000314 if let Some(offset) = self.fdt.first_property_offset(self.offset)? {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000315 Ok(Some(FdtProperty::new(self.fdt, offset)?))
316 } else {
317 Ok(None)
318 }
Jaewan Kim72d10902023-10-12 21:59:26 +0900319 }
Jaewan Kimf34f4b82023-11-03 19:38:38 +0900320
321 /// Returns the phandle
322 pub fn get_phandle(&self) -> Result<Option<Phandle>> {
323 // This rewrites the fdt_get_phandle() because it doesn't return error code.
324 if let Some(prop) = self.getprop_u32(cstr!("phandle"))? {
325 Ok(Some(prop.try_into()?))
326 } else if let Some(prop) = self.getprop_u32(cstr!("linux,phandle"))? {
327 Ok(Some(prop.try_into()?))
328 } else {
329 Ok(None)
330 }
331 }
Jaewan Kim52026012023-12-13 13:49:28 +0900332
333 /// Returns the subnode of the given name. The name doesn't need to be nul-terminated.
334 pub fn subnode(&self, name: &CStr) -> Result<Option<Self>> {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000335 let name = name.to_bytes();
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000336 let offset = self.fdt.subnode_offset_namelen(self.offset, name)?;
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000337
Jaewan Kim52026012023-12-13 13:49:28 +0900338 Ok(offset.map(|offset| Self { fdt: self.fdt, offset }))
339 }
340
341 /// Returns the subnode of the given name bytes
342 pub fn subnode_with_name_bytes(&self, name: &[u8]) -> Result<Option<Self>> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000343 let offset = self.fdt.subnode_offset_namelen(self.offset, name)?;
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000344
Jaewan Kim52026012023-12-13 13:49:28 +0900345 Ok(offset.map(|offset| Self { fdt: self.fdt, offset }))
346 }
David Brazdil1baa9a92022-06-28 14:47:50 +0100347}
348
Pierre-Clément Tosi504b4302023-10-30 12:22:50 +0000349impl<'a> PartialEq for FdtNode<'a> {
350 fn eq(&self, other: &Self) -> bool {
351 self.fdt.as_ptr() == other.fdt.as_ptr() && self.offset == other.offset
352 }
353}
354
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000355/// Mutable FDT node.
Pierre-Clément Tosi504b4302023-10-30 12:22:50 +0000356#[derive(Debug)]
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000357pub struct FdtNodeMut<'a> {
358 fdt: &'a mut Fdt,
359 offset: c_int,
360}
361
362impl<'a> FdtNodeMut<'a> {
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900363 /// Appends a property name-value (possibly empty) pair to the given node.
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000364 pub fn appendprop<T: AsRef<[u8]>>(&mut self, name: &CStr, value: &T) -> Result<()> {
Pierre-Clément Tosi410e46a2023-12-24 11:33:11 +0000365 self.fdt.appendprop(self.offset, name, value.as_ref())
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000366 }
367
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900368 /// Appends a (address, size) pair property to the given node.
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000369 pub fn appendprop_addrrange(&mut self, name: &CStr, addr: u64, size: u64) -> Result<()> {
Pierre-Clément Tosi410e46a2023-12-24 11:33:11 +0000370 let parent = self.parent()?.offset;
371 self.fdt.appendprop_addrrange(parent, self.offset, name, addr, size)
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000372 }
373
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900374 /// Sets a property name-value pair to the given node.
375 ///
376 /// This may create a new prop or replace existing value.
Jaewan Kimba8929b2023-01-13 11:13:29 +0900377 pub fn setprop(&mut self, name: &CStr, value: &[u8]) -> Result<()> {
Pierre-Clément Tosi410e46a2023-12-24 11:33:11 +0000378 self.fdt.setprop(self.offset, name, value)
Jaewan Kimba8929b2023-01-13 11:13:29 +0900379 }
380
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900381 /// Sets the value of the given property with the given value, and ensure that the given
382 /// value has the same length as the current value length.
383 ///
384 /// This can only be used to replace existing value.
Jiyong Park9c63cd12023-03-21 17:53:07 +0900385 pub fn setprop_inplace(&mut self, name: &CStr, value: &[u8]) -> Result<()> {
Pierre-Clément Tosi410e46a2023-12-24 11:33:11 +0000386 self.fdt.setprop_inplace(self.offset, name, value)
Jiyong Park9c63cd12023-03-21 17:53:07 +0900387 }
388
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900389 /// Sets the value of the given (address, size) pair property with the given value, and
390 /// ensure that the given value has the same length as the current value length.
391 ///
392 /// This can only be used to replace existing value.
Pierre-Clément Tosic27c4272023-05-19 15:46:26 +0000393 pub fn setprop_addrrange_inplace(&mut self, name: &CStr, addr: u64, size: u64) -> Result<()> {
394 let pair = [addr.to_be(), size.to_be()];
Pierre-Clément Tosi410e46a2023-12-24 11:33:11 +0000395 self.fdt.setprop_inplace(self.offset, name, pair.as_bytes())
Pierre-Clément Tosic27c4272023-05-19 15:46:26 +0000396 }
397
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900398 /// Sets a flag-like empty property.
399 ///
400 /// This may create a new prop or replace existing value.
Pierre-Clément Tosi4ba79662023-02-13 11:22:41 +0000401 pub fn setprop_empty(&mut self, name: &CStr) -> Result<()> {
Pierre-Clément Tosi410e46a2023-12-24 11:33:11 +0000402 self.fdt.setprop(self.offset, name, &[])
Pierre-Clément Tosi4ba79662023-02-13 11:22:41 +0000403 }
404
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900405 /// Deletes the given property.
Pierre-Clément Tosi4ba79662023-02-13 11:22:41 +0000406 pub fn delprop(&mut self, name: &CStr) -> Result<()> {
Pierre-Clément Tosi410e46a2023-12-24 11:33:11 +0000407 self.fdt.delprop(self.offset, name)
Pierre-Clément Tosi4ba79662023-02-13 11:22:41 +0000408 }
409
Jaewan Kim4ae0e712023-10-19 14:16:17 +0900410 /// Deletes the given property effectively from DT, by setting it with FDT_NOP.
Pierre-Clément Tosibe3a97b2023-05-19 14:56:23 +0000411 pub fn nop_property(&mut self, name: &CStr) -> Result<()> {
Pierre-Clément Tosi410e46a2023-12-24 11:33:11 +0000412 self.fdt.nop_property(self.offset, name)
Pierre-Clément Tosibe3a97b2023-05-19 14:56:23 +0000413 }
414
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900415 /// Trims the size of the given property to new_size.
Jiyong Park9c63cd12023-03-21 17:53:07 +0900416 pub fn trimprop(&mut self, name: &CStr, new_size: usize) -> Result<()> {
Pierre-Clément Tosi843845d2024-01-23 23:35:01 +0000417 let prop = self.as_node().getprop(name)?.ok_or(FdtError::NotFound)?;
Jiyong Park9c63cd12023-03-21 17:53:07 +0900418
Pierre-Clément Tosi843845d2024-01-23 23:35:01 +0000419 match prop.len() {
420 x if x == new_size => Ok(()),
421 x if x < new_size => Err(FdtError::NoSpace),
422 _ => self.fdt.setprop_placeholder(self.offset, name, new_size).map(|_| ()),
423 }
Jiyong Park9c63cd12023-03-21 17:53:07 +0900424 }
425
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900426 /// Returns reference to the containing device tree.
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000427 pub fn fdt(&mut self) -> &mut Fdt {
428 self.fdt
429 }
430
Jaewan Kimf72f4f22023-11-03 19:21:34 +0900431 /// Returns immutable FdtNode of this node.
432 pub fn as_node(&self) -> FdtNode {
433 FdtNode { fdt: self.fdt, offset: self.offset }
434 }
435
Jaewan Kime6363422024-01-19 14:00:00 +0900436 /// Adds new subnodes to the given node.
437 pub fn add_subnodes(&mut self, names: &[&CStr]) -> Result<()> {
438 for name in names {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000439 self.fdt.add_subnode_namelen(self.offset, name.to_bytes())?;
Jaewan Kime6363422024-01-19 14:00:00 +0900440 }
441 Ok(())
442 }
443
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900444 /// Adds a new subnode to the given node and return it as a FdtNodeMut on success.
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000445 pub fn add_subnode(&'a mut self, name: &CStr) -> Result<Self> {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000446 let name = name.to_bytes();
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000447 let offset = self.fdt.add_subnode_namelen(self.offset, name)?;
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000448
Jaewan Kim5ab13582023-10-20 20:56:27 +0900449 Ok(Self { fdt: self.fdt, offset })
450 }
451
452 /// Adds a new subnode to the given node with name and namelen, and returns it as a FdtNodeMut
453 /// on success.
454 pub fn add_subnode_with_namelen(&'a mut self, name: &CStr, namelen: usize) -> Result<Self> {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000455 let name = &name.to_bytes()[..namelen];
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000456 let offset = self.fdt.add_subnode_namelen(self.offset, name)?;
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000457
Jaewan Kim5ab13582023-10-20 20:56:27 +0900458 Ok(Self { fdt: self.fdt, offset })
459 }
460
Jaewan Kim5f1a6032023-12-18 15:17:58 +0900461 /// Returns the first subnode of this
Pierre-Clément Tosi973dda32024-02-14 16:08:28 +0000462 pub fn first_subnode(self) -> Result<Option<Self>> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000463 let offset = self.fdt.first_subnode(self.offset)?;
Jaewan Kim5f1a6032023-12-18 15:17:58 +0900464
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000465 Ok(offset.map(|offset| Self { fdt: self.fdt, offset }))
Jaewan Kim5f1a6032023-12-18 15:17:58 +0900466 }
467
468 /// Returns the next subnode that shares the same parent with this
469 pub fn next_subnode(self) -> Result<Option<Self>> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000470 let offset = self.fdt.next_subnode(self.offset)?;
Jaewan Kim5f1a6032023-12-18 15:17:58 +0900471
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000472 Ok(offset.map(|offset| Self { fdt: self.fdt, offset }))
Jaewan Kim5f1a6032023-12-18 15:17:58 +0900473 }
474
475 /// Deletes the current node and returns the next subnode
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000476 pub fn delete_and_next_subnode(self) -> Result<Option<Self>> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000477 let next_offset = self.fdt.next_subnode(self.offset)?;
Jaewan Kim5f1a6032023-12-18 15:17:58 +0900478
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000479 self.delete_and_next(next_offset)
Jaewan Kim5f1a6032023-12-18 15:17:58 +0900480 }
481
Jaewan Kim28a13ea2024-01-04 09:22:40 +0900482 /// Returns the next node
483 pub fn next_node(self, depth: usize) -> Result<Option<(Self, usize)>> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000484 let next = self.fdt.next_node(self.offset, depth)?;
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000485
486 Ok(next.map(|(offset, depth)| (Self { fdt: self.fdt, offset }, depth)))
Jaewan Kim28a13ea2024-01-04 09:22:40 +0900487 }
488
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000489 /// Deletes this and returns the next node
Pierre-Clément Tosi81c5bc72024-01-29 13:39:07 +0000490 pub fn delete_and_next_node(self, depth: usize) -> Result<Option<(Self, usize)>> {
491 let next_node = self.fdt.next_node_skip_subnodes(self.offset, depth)?;
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000492 if let Some((offset, depth)) = next_node {
493 let next_node = self.delete_and_next(Some(offset))?.unwrap();
494 Ok(Some((next_node, depth)))
495 } else {
Jaewan Kima5bc8922024-02-20 01:15:37 +0900496 self.delete_and_next(None)?;
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000497 Ok(None)
498 }
Jaewan Kim28a13ea2024-01-04 09:22:40 +0900499 }
500
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000501 fn parent(&'a self) -> Result<FdtNode<'a>> {
Pierre-Clément Tosidf3037f2024-01-22 15:41:43 +0000502 self.as_node().parent()
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000503 }
Jiyong Park9c63cd12023-03-21 17:53:07 +0900504
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900505 /// Returns the compatible node of the given name that is next after this node.
Jiyong Park9c63cd12023-03-21 17:53:07 +0900506 pub fn next_compatible(self, compatible: &CStr) -> Result<Option<Self>> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000507 let offset = self.fdt.node_offset_by_compatible(self.offset, compatible)?;
Jiyong Park9c63cd12023-03-21 17:53:07 +0900508
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000509 Ok(offset.map(|offset| Self { fdt: self.fdt, offset }))
Jiyong Park9c63cd12023-03-21 17:53:07 +0900510 }
511
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900512 /// Deletes the node effectively by overwriting this node and its subtree with nop tags.
513 /// Returns the next compatible node of the given name.
Jiyong Park9c63cd12023-03-21 17:53:07 +0900514 // Side note: without this, filterint out excessive compatible nodes from the DT is impossible.
515 // The reason is that libfdt ensures that the node from where the search for the next
516 // compatible node is started is always a valid one -- except for the special case of offset =
517 // -1 which is to find the first compatible node. So, we can't delete a node and then find the
518 // next compatible node from it.
519 //
520 // We can't do in the opposite direction either. If we call next_compatible to find the next
521 // node, and delete the current node, the Rust borrow checker kicks in. The next node has a
522 // mutable reference to DT, so we can't use current node (which also has a mutable reference to
523 // DT).
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000524 pub fn delete_and_next_compatible(self, compatible: &CStr) -> Result<Option<Self>> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000525 let next_offset = self.fdt.node_offset_by_compatible(self.offset, compatible)?;
Jiyong Park9c63cd12023-03-21 17:53:07 +0900526
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000527 self.delete_and_next(next_offset)
528 }
529
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000530 fn delete_and_next(self, next_offset: Option<c_int>) -> Result<Option<Self>> {
Jaewan Kim4ae0e712023-10-19 14:16:17 +0900531 if Some(self.offset) == next_offset {
532 return Err(FdtError::Internal);
533 }
534
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000535 self.fdt.nop_node(self.offset)?;
Jiyong Park9c63cd12023-03-21 17:53:07 +0900536
537 Ok(next_offset.map(|offset| Self { fdt: self.fdt, offset }))
538 }
Jaewan Kim4ae0e712023-10-19 14:16:17 +0900539
540 /// Deletes this node effectively from DT, by setting it with FDT_NOP
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000541 pub fn nop(self) -> Result<()> {
542 self.fdt.nop_node(self.offset)
Jaewan Kim4ae0e712023-10-19 14:16:17 +0900543 }
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000544}
545
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000546/// Wrapper around low-level libfdt functions.
Alice Wang9d4df702023-05-25 14:14:12 +0000547#[derive(Debug)]
David Brazdil1baa9a92022-06-28 14:47:50 +0100548#[repr(transparent)]
549pub struct Fdt {
Pierre-Clément Tosief2030e2022-11-28 11:21:20 +0000550 buffer: [u8],
David Brazdil1baa9a92022-06-28 14:47:50 +0100551}
552
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000553// SAFETY: Fdt calls check_full() before safely returning a &Self, making it impossible for trait
554// methods to be called on invalid device trees.
555unsafe impl Libfdt for Fdt {
556 fn as_fdt_slice(&self) -> &[u8] {
557 &self.buffer[..self.totalsize()]
558 }
559}
560
561// SAFETY: Fdt calls check_full() before safely returning a &Self, making it impossible for trait
562// methods to be called on invalid device trees.
563unsafe impl LibfdtMut for Fdt {
564 fn as_fdt_slice_mut(&mut self) -> &mut [u8] {
565 &mut self.buffer
566 }
567}
568
David Brazdil1baa9a92022-06-28 14:47:50 +0100569impl Fdt {
570 /// Wraps a slice containing a Flattened Device Tree.
571 ///
572 /// Fails if the FDT does not pass validation.
573 pub fn from_slice(fdt: &[u8]) -> Result<&Self> {
Pierre-Clément Tosifbb5ee22023-12-21 13:49:59 +0000574 libfdt::check_full(fdt)?;
575 // SAFETY: The FDT was validated.
David Brazdil1baa9a92022-06-28 14:47:50 +0100576 let fdt = unsafe { Self::unchecked_from_slice(fdt) };
Pierre-Clément Tosifbb5ee22023-12-21 13:49:59 +0000577
David Brazdil1baa9a92022-06-28 14:47:50 +0100578 Ok(fdt)
579 }
580
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000581 /// Wraps a mutable slice containing a Flattened Device Tree.
582 ///
583 /// Fails if the FDT does not pass validation.
584 pub fn from_mut_slice(fdt: &mut [u8]) -> Result<&mut Self> {
Pierre-Clément Tosifbb5ee22023-12-21 13:49:59 +0000585 libfdt::check_full(fdt)?;
586 // SAFETY: The FDT was validated.
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000587 let fdt = unsafe { Self::unchecked_from_mut_slice(fdt) };
Pierre-Clément Tosifbb5ee22023-12-21 13:49:59 +0000588
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000589 Ok(fdt)
590 }
591
Jaewan Kim4cf20aa2023-04-03 10:25:38 +0900592 /// Creates an empty Flattened Device Tree with a mutable slice.
593 pub fn create_empty_tree(fdt: &mut [u8]) -> Result<&mut Self> {
Pierre-Clément Tosifbb5ee22023-12-21 13:49:59 +0000594 libfdt::create_empty_tree(fdt)?;
Jaewan Kim4cf20aa2023-04-03 10:25:38 +0900595
Pierre-Clément Tosifbb5ee22023-12-21 13:49:59 +0000596 Self::from_mut_slice(fdt)
Jaewan Kim4cf20aa2023-04-03 10:25:38 +0900597 }
598
David Brazdil1baa9a92022-06-28 14:47:50 +0100599 /// Wraps a slice containing a Flattened Device Tree.
600 ///
601 /// # Safety
602 ///
Pierre-Clément Tosidf3037f2024-01-22 15:41:43 +0000603 /// It is undefined to call this function on a slice that does not contain a valid device tree.
David Brazdil1baa9a92022-06-28 14:47:50 +0100604 pub unsafe fn unchecked_from_slice(fdt: &[u8]) -> &Self {
Pierre-Clément Tosidf3037f2024-01-22 15:41:43 +0000605 let self_ptr = fdt as *const _ as *const _;
606 // SAFETY: The pointer is non-null, dereferenceable, and points to allocated memory.
607 unsafe { &*self_ptr }
David Brazdil1baa9a92022-06-28 14:47:50 +0100608 }
609
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000610 /// Wraps a mutable slice containing a Flattened Device Tree.
611 ///
612 /// # Safety
613 ///
Pierre-Clément Tosidf3037f2024-01-22 15:41:43 +0000614 /// It is undefined to call this function on a slice that does not contain a valid device tree.
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000615 pub unsafe fn unchecked_from_mut_slice(fdt: &mut [u8]) -> &mut Self {
Pierre-Clément Tosidf3037f2024-01-22 15:41:43 +0000616 let self_mut_ptr = fdt as *mut _ as *mut _;
617 // SAFETY: The pointer is non-null, dereferenceable, and points to allocated memory.
618 unsafe { &mut *self_mut_ptr }
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000619 }
620
Pierre-Clément Tosice0b36d2024-01-26 10:50:05 +0000621 /// Updates this FDT from another FDT.
622 pub fn clone_from(&mut self, other: &Self) -> Result<()> {
623 let new_len = other.buffer.len();
624 if self.buffer.len() < new_len {
625 return Err(FdtError::NoSpace);
Jiyong Parke9d87e82023-03-21 19:28:40 +0900626 }
Pierre-Clément Tosice0b36d2024-01-26 10:50:05 +0000627
628 let zeroed_len = self.totalsize().checked_sub(new_len);
629 let (cloned, zeroed) = self.buffer.split_at_mut(new_len);
630
631 cloned.clone_from_slice(&other.buffer);
632 if let Some(len) = zeroed_len {
633 zeroed[..len].fill(0);
634 }
635
636 Ok(())
Jiyong Parke9d87e82023-03-21 19:28:40 +0900637 }
638
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900639 /// Unpacks the DT to cover the whole slice it is contained in.
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000640 pub fn unpack(&mut self) -> Result<()> {
Andrew Walbran84b9a232023-07-05 14:01:40 +0000641 // SAFETY: "Opens" the DT in-place (supported use-case) by updating its header and
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000642 // internal structures to make use of the whole self.fdt slice but performs no accesses
643 // outside of it and leaves the DT in a state that will be detected by other functions.
644 let ret = unsafe {
645 libfdt_bindgen::fdt_open_into(
646 self.as_ptr(),
647 self.as_mut_ptr(),
648 self.capacity().try_into().map_err(|_| FdtError::Internal)?,
649 )
650 };
651 fdt_err_expect_zero(ret)
652 }
653
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900654 /// Packs the DT to take a minimum amount of memory.
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000655 ///
656 /// Doesn't shrink the underlying memory slice.
657 pub fn pack(&mut self) -> Result<()> {
Andrew Walbran84b9a232023-07-05 14:01:40 +0000658 // SAFETY: "Closes" the DT in-place by updating its header and relocating its structs.
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000659 let ret = unsafe { libfdt_bindgen::fdt_pack(self.as_mut_ptr()) };
660 fdt_err_expect_zero(ret)
661 }
662
Pierre-Clément Tosi90e19352022-11-21 17:11:48 +0000663 /// Applies a DT overlay on the base DT.
664 ///
665 /// # Safety
666 ///
667 /// On failure, the library corrupts the DT and overlay so both must be discarded.
668 pub unsafe fn apply_overlay<'a>(&'a mut self, overlay: &'a mut Fdt) -> Result<&'a mut Self> {
Andrew Walbran84b9a232023-07-05 14:01:40 +0000669 let ret =
670 // SAFETY: Both pointers are valid because they come from references, and fdt_overlay_apply
671 // doesn't keep them after it returns. It may corrupt their contents if there is an error,
672 // but that's our caller's responsibility.
673 unsafe { libfdt_bindgen::fdt_overlay_apply(self.as_mut_ptr(), overlay.as_mut_ptr()) };
674 fdt_err_expect_zero(ret)?;
Pierre-Clément Tosi90e19352022-11-21 17:11:48 +0000675 Ok(self)
676 }
677
Alice Wang2422bdc2023-06-12 08:37:55 +0000678 /// Returns an iterator of memory banks specified the "/memory" node.
679 /// Throws an error when the "/memory" is not found in the device tree.
David Brazdil1baa9a92022-06-28 14:47:50 +0100680 ///
681 /// NOTE: This does not support individual "/memory@XXXX" banks.
Alice Wang2422bdc2023-06-12 08:37:55 +0000682 pub fn memory(&self) -> Result<MemRegIterator> {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000683 let node = self.node(cstr!("/memory"))?.ok_or(FdtError::NotFound)?;
684 if node.device_type()? != Some(cstr!("memory")) {
Alice Wang2422bdc2023-06-12 08:37:55 +0000685 return Err(FdtError::BadValue);
Pierre-Clément Tosib244d932022-11-24 16:45:53 +0000686 }
Alice Wang2422bdc2023-06-12 08:37:55 +0000687 node.reg()?.ok_or(FdtError::BadValue).map(MemRegIterator::new)
688 }
689
690 /// Returns the first memory range in the `/memory` node.
691 pub fn first_memory_range(&self) -> Result<Range<usize>> {
692 self.memory()?.next().ok_or(FdtError::NotFound)
David Brazdil1baa9a92022-06-28 14:47:50 +0100693 }
694
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900695 /// Returns the standard /chosen node.
Pierre-Clément Tosib244d932022-11-24 16:45:53 +0000696 pub fn chosen(&self) -> Result<Option<FdtNode>> {
Jaewan Kimb635bb02023-11-01 13:00:34 +0900697 self.node(cstr!("/chosen"))
David Brazdil1baa9a92022-06-28 14:47:50 +0100698 }
699
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900700 /// Returns the standard /chosen node as mutable.
Pierre-Clément Tosi4ba79662023-02-13 11:22:41 +0000701 pub fn chosen_mut(&mut self) -> Result<Option<FdtNodeMut>> {
Jaewan Kimb635bb02023-11-01 13:00:34 +0900702 self.node_mut(cstr!("/chosen"))
Pierre-Clément Tosi4ba79662023-02-13 11:22:41 +0000703 }
704
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900705 /// Returns the root node of the tree.
Pierre-Clément Tosi41c158e2022-11-21 19:16:25 +0000706 pub fn root(&self) -> Result<FdtNode> {
Jaewan Kimb635bb02023-11-01 13:00:34 +0900707 self.node(cstr!("/"))?.ok_or(FdtError::Internal)
Pierre-Clément Tosi41c158e2022-11-21 19:16:25 +0000708 }
709
Jaewan Kimf163d762023-11-01 13:12:50 +0900710 /// Returns the standard /__symbols__ node.
711 pub fn symbols(&self) -> Result<Option<FdtNode>> {
712 self.node(cstr!("/__symbols__"))
713 }
714
715 /// Returns the standard /__symbols__ node as mutable
716 pub fn symbols_mut(&mut self) -> Result<Option<FdtNodeMut>> {
717 self.node_mut(cstr!("/__symbols__"))
718 }
719
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900720 /// Returns a tree node by its full path.
Pierre-Clément Tosib244d932022-11-24 16:45:53 +0000721 pub fn node(&self, path: &CStr) -> Result<Option<FdtNode>> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000722 let offset = self.path_offset_namelen(path.to_bytes())?;
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000723
724 Ok(offset.map(|offset| FdtNode { fdt: self, offset }))
David Brazdil1baa9a92022-06-28 14:47:50 +0100725 }
726
Pierre-Clément Tosi41c158e2022-11-21 19:16:25 +0000727 /// Iterate over nodes with a given compatible string.
728 pub fn compatible_nodes<'a>(&'a self, compatible: &'a CStr) -> Result<CompatibleIterator<'a>> {
729 CompatibleIterator::new(self, compatible)
730 }
731
Jaewan Kim17ba7a32023-10-19 13:25:15 +0900732 /// Returns max phandle in the tree.
733 pub fn max_phandle(&self) -> Result<Phandle> {
Pierre-Clément Tosiecd5bbc2023-12-21 15:12:45 +0000734 self.find_max_phandle()
Jaewan Kim17ba7a32023-10-19 13:25:15 +0900735 }
736
737 /// Returns a node with the phandle
738 pub fn node_with_phandle(&self, phandle: Phandle) -> Result<Option<FdtNode>> {
Pierre-Clément Tosiecd5bbc2023-12-21 15:12:45 +0000739 let offset = self.node_offset_by_phandle(phandle)?;
740
Jaewan Kimc63246d2023-11-09 15:41:01 +0900741 Ok(offset.map(|offset| FdtNode { fdt: self, offset }))
742 }
743
744 /// Returns a mutable node with the phandle
745 pub fn node_mut_with_phandle(&mut self, phandle: Phandle) -> Result<Option<FdtNodeMut>> {
Pierre-Clément Tosiecd5bbc2023-12-21 15:12:45 +0000746 let offset = self.node_offset_by_phandle(phandle)?;
Jaewan Kimc63246d2023-11-09 15:41:01 +0900747
Pierre-Clément Tosiecd5bbc2023-12-21 15:12:45 +0000748 Ok(offset.map(|offset| FdtNodeMut { fdt: self, offset }))
Jaewan Kim17ba7a32023-10-19 13:25:15 +0900749 }
750
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900751 /// Returns the mutable root node of the tree.
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000752 pub fn root_mut(&mut self) -> Result<FdtNodeMut> {
Jaewan Kimb635bb02023-11-01 13:00:34 +0900753 self.node_mut(cstr!("/"))?.ok_or(FdtError::Internal)
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000754 }
755
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900756 /// Returns a mutable tree node by its full path.
Pierre-Clément Tosib244d932022-11-24 16:45:53 +0000757 pub fn node_mut(&mut self, path: &CStr) -> Result<Option<FdtNodeMut>> {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000758 let offset = self.path_offset_namelen(path.to_bytes())?;
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000759
760 Ok(offset.map(|offset| FdtNodeMut { fdt: self, offset }))
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000761 }
762
Pierre-Clément Tosi81c5bc72024-01-29 13:39:07 +0000763 fn next_node_skip_subnodes(&self, node: c_int, depth: usize) -> Result<Option<(c_int, usize)>> {
764 let mut iter = self.next_node(node, depth)?;
765 while let Some((offset, next_depth)) = iter {
766 if next_depth <= depth {
767 return Ok(Some((offset, next_depth)));
768 }
769 iter = self.next_node(offset, next_depth)?;
770 }
771
772 Ok(None)
773 }
774
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900775 /// Returns the device tree as a slice (may be smaller than the containing buffer).
Pierre-Clément Tosidb74cb12022-12-08 13:56:25 +0000776 pub fn as_slice(&self) -> &[u8] {
Pierre-Clément Tosid83741d2024-02-02 10:44:55 +0000777 self.as_fdt_slice()
David Brazdil1baa9a92022-06-28 14:47:50 +0100778 }
779
Jaewan Kimaa638702023-09-19 13:34:01 +0900780 fn get_from_ptr(&self, ptr: *const c_void, len: usize) -> Result<&[u8]> {
Pierre-Clément Tosi60282ae2023-12-21 16:00:02 +0000781 get_slice_at_ptr(self.as_fdt_slice(), ptr.cast(), len).ok_or(FdtError::Internal)
Jaewan Kim72d10902023-10-12 21:59:26 +0900782 }
783
Jaewan Kimb3dcfc22023-09-20 10:20:52 +0900784 /// Returns a shared pointer to the device tree.
Pierre-Clément Tosi8036b4f2023-02-17 10:31:31 +0000785 pub fn as_ptr(&self) -> *const c_void {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000786 self.buffer.as_ptr().cast()
David Brazdil1baa9a92022-06-28 14:47:50 +0100787 }
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000788
789 fn as_mut_ptr(&mut self) -> *mut c_void {
Pierre-Clément Tosi0dcc75e2023-05-02 13:43:55 +0000790 self.buffer.as_mut_ptr().cast::<_>()
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000791 }
792
793 fn capacity(&self) -> usize {
Pierre-Clément Tosief2030e2022-11-28 11:21:20 +0000794 self.buffer.len()
Pierre-Clément Tosi1b0d8902022-11-21 18:16:59 +0000795 }
Pierre-Clément Tosidb74cb12022-12-08 13:56:25 +0000796
797 fn header(&self) -> &libfdt_bindgen::fdt_header {
Pierre-Clément Tosicb92b512024-01-22 15:55:25 +0000798 let p = self.as_ptr().cast();
Andrew Walbran84b9a232023-07-05 14:01:40 +0000799 // SAFETY: A valid FDT (verified by constructor) must contain a valid fdt_header.
Pierre-Clément Tosi0dcc75e2023-05-02 13:43:55 +0000800 unsafe { &*p }
Pierre-Clément Tosidb74cb12022-12-08 13:56:25 +0000801 }
802
803 fn totalsize(&self) -> usize {
804 u32::from_be(self.header().totalsize) as usize
805 }
David Brazdil1baa9a92022-06-28 14:47:50 +0100806}