4 * Copyright (c) 2016, Linaro Ltd. All rights reserved.<BR>
6 * This program and the accompanying materials are
7 * licensed and made available under the terms and conditions of the BSD License
8 * which accompanies this distribution. The full text of the license may be found at
9 * http://opensource.org/licenses/bsd-license.php
11 * THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
16 #include <Library/BaseLib.h>
17 #include <Library/DebugLib.h>
18 #include <Library/UefiDriverEntryPoint.h>
19 #include <Library/UefiBootServicesTableLib.h>
20 #include <Library/HobLib.h>
24 #include <Guid/FdtHob.h>
25 #include <Guid/PlatformHasDeviceTree.h>
27 #include <Protocol/FdtClient.h>
29 STATIC VOID
*mDeviceTreeBase
;
35 IN FDT_CLIENT_PROTOCOL
*This
,
37 IN CONST CHAR8
*PropertyName
,
38 OUT CONST VOID
**Prop
,
39 OUT UINT32
*PropSize OPTIONAL
44 ASSERT (mDeviceTreeBase
!= NULL
);
45 ASSERT (Prop
!= NULL
);
47 *Prop
= fdt_getprop (mDeviceTreeBase
, Node
, PropertyName
, &Len
);
52 if (PropSize
!= NULL
) {
62 IN FDT_CLIENT_PROTOCOL
*This
,
64 IN CONST CHAR8
*PropertyName
,
71 ASSERT (mDeviceTreeBase
!= NULL
);
73 Ret
= fdt_setprop (mDeviceTreeBase
, Node
, PropertyName
, Prop
, PropSize
);
75 return EFI_DEVICE_ERROR
;
84 FindNextCompatibleNode (
85 IN FDT_CLIENT_PROTOCOL
*This
,
86 IN CONST CHAR8
*CompatibleString
,
92 CONST CHAR8
*Type
, *Compatible
;
95 ASSERT (mDeviceTreeBase
!= NULL
);
96 ASSERT (Node
!= NULL
);
98 for (Prev
= PrevNode
;; Prev
= Next
) {
99 Next
= fdt_next_node (mDeviceTreeBase
, Prev
, NULL
);
104 Type
= fdt_getprop (mDeviceTreeBase
, Next
, "compatible", &Len
);
110 // A 'compatible' node may contain a sequence of NUL terminated
111 // compatible strings so check each one
113 for (Compatible
= Type
; Compatible
< Type
+ Len
&& *Compatible
;
114 Compatible
+= 1 + AsciiStrLen (Compatible
)) {
115 if (AsciiStrCmp (CompatibleString
, Compatible
) == 0) {
121 return EFI_NOT_FOUND
;
128 IN FDT_CLIENT_PROTOCOL
*This
,
129 IN CONST CHAR8
*CompatibleString
,
133 return FindNextCompatibleNode (This
, CompatibleString
, 0, Node
);
139 FindCompatibleNodeProperty (
140 IN FDT_CLIENT_PROTOCOL
*This
,
141 IN CONST CHAR8
*CompatibleString
,
142 IN CONST CHAR8
*PropertyName
,
143 OUT CONST VOID
**Prop
,
144 OUT UINT32
*PropSize OPTIONAL
150 Status
= FindCompatibleNode (This
, CompatibleString
, &Node
);
151 if (EFI_ERROR (Status
)) {
155 return GetNodeProperty (This
, Node
, PropertyName
, Prop
, PropSize
);
161 FindCompatibleNodeReg (
162 IN FDT_CLIENT_PROTOCOL
*This
,
163 IN CONST CHAR8
*CompatibleString
,
164 OUT CONST VOID
**Reg
,
165 OUT UINTN
*AddressCells
,
166 OUT UINTN
*SizeCells
,
172 ASSERT (RegSize
!= NULL
);
175 // Get the 'reg' property of this node. For now, we will assume
176 // 8 byte quantities for base and size, respectively.
177 // TODO use #cells root properties instead
179 Status
= FindCompatibleNodeProperty (This
, CompatibleString
, "reg", Reg
,
181 if (EFI_ERROR (Status
)) {
185 if ((*RegSize
% 16) != 0) {
187 "%a: '%a' compatible node has invalid 'reg' property (size == 0x%x)\n",
188 __FUNCTION__
, CompatibleString
, *RegSize
));
189 return EFI_NOT_FOUND
;
201 FindNextMemoryNodeReg (
202 IN FDT_CLIENT_PROTOCOL
*This
,
205 OUT CONST VOID
**Reg
,
206 OUT UINTN
*AddressCells
,
207 OUT UINTN
*SizeCells
,
212 CONST CHAR8
*DeviceType
;
216 ASSERT (mDeviceTreeBase
!= NULL
);
217 ASSERT (Node
!= NULL
);
219 for (Prev
= PrevNode
;; Prev
= Next
) {
220 Next
= fdt_next_node (mDeviceTreeBase
, Prev
, NULL
);
225 DeviceType
= fdt_getprop (mDeviceTreeBase
, Next
, "device_type", &Len
);
226 if (DeviceType
!= NULL
&& AsciiStrCmp (DeviceType
, "memory") == 0) {
228 // Get the 'reg' property of this memory node. For now, we will assume
229 // 8 byte quantities for base and size, respectively.
230 // TODO use #cells root properties instead
232 Status
= GetNodeProperty (This
, Next
, "reg", Reg
, RegSize
);
233 if (EFI_ERROR (Status
)) {
235 "%a: ignoring memory node with no 'reg' property\n",
239 if ((*RegSize
% 16) != 0) {
241 "%a: ignoring memory node with invalid 'reg' property (size == 0x%x)\n",
242 __FUNCTION__
, *RegSize
));
252 return EFI_NOT_FOUND
;
259 IN FDT_CLIENT_PROTOCOL
*This
,
261 OUT CONST VOID
**Reg
,
262 OUT UINTN
*AddressCells
,
263 OUT UINTN
*SizeCells
,
267 return FindNextMemoryNodeReg (This
, 0, Node
, Reg
, AddressCells
, SizeCells
,
274 GetOrInsertChosenNode (
275 IN FDT_CLIENT_PROTOCOL
*This
,
281 ASSERT (mDeviceTreeBase
!= NULL
);
282 ASSERT (Node
!= NULL
);
284 NewNode
= fdt_path_offset (mDeviceTreeBase
, "/chosen");
286 NewNode
= fdt_add_subnode (mDeviceTreeBase
, 0, "/chosen");
290 return EFI_OUT_OF_RESOURCES
;
298 STATIC FDT_CLIENT_PROTOCOL mFdtClientProtocol
= {
302 FindNextCompatibleNode
,
303 FindCompatibleNodeProperty
,
304 FindCompatibleNodeReg
,
306 FindNextMemoryNodeReg
,
307 GetOrInsertChosenNode
,
313 OnPlatformHasDeviceTree (
320 VOID
*DeviceTreeBase
;
322 Status
= gBS
->LocateProtocol (
323 &gEdkiiPlatformHasDeviceTreeGuid
,
324 NULL
, // Registration
327 if (EFI_ERROR (Status
)) {
331 DeviceTreeBase
= Context
;
334 "%a: exposing DTB @ 0x%p to OS\n",
338 Status
= gBS
->InstallConfigurationTable (&gFdtTableGuid
, DeviceTreeBase
);
339 ASSERT_EFI_ERROR (Status
);
341 gBS
->CloseEvent (Event
);
346 InitializeFdtClientDxe (
347 IN EFI_HANDLE ImageHandle
,
348 IN EFI_SYSTEM_TABLE
*SystemTable
352 VOID
*DeviceTreeBase
;
354 EFI_EVENT PlatformHasDeviceTreeEvent
;
357 Hob
= GetFirstGuidHob (&gFdtHobGuid
);
358 if (Hob
== NULL
|| GET_GUID_HOB_DATA_SIZE (Hob
) != sizeof (UINT64
)) {
359 return EFI_NOT_FOUND
;
361 DeviceTreeBase
= (VOID
*)(UINTN
)*(UINT64
*)GET_GUID_HOB_DATA (Hob
);
363 if (fdt_check_header (DeviceTreeBase
) != 0) {
364 DEBUG ((EFI_D_ERROR
, "%a: No DTB found @ 0x%p\n", __FUNCTION__
,
366 return EFI_NOT_FOUND
;
369 mDeviceTreeBase
= DeviceTreeBase
;
371 DEBUG ((EFI_D_INFO
, "%a: DTB @ 0x%p\n", __FUNCTION__
, mDeviceTreeBase
));
374 // Register a protocol notify for the EDKII Platform Has Device Tree
377 Status
= gBS
->CreateEvent (
380 OnPlatformHasDeviceTree
,
381 DeviceTreeBase
, // Context
382 &PlatformHasDeviceTreeEvent
384 if (EFI_ERROR (Status
)) {
385 DEBUG ((DEBUG_ERROR
, "%a: CreateEvent(): %r\n", __FUNCTION__
, Status
));
389 Status
= gBS
->RegisterProtocolNotify (
390 &gEdkiiPlatformHasDeviceTreeGuid
,
391 PlatformHasDeviceTreeEvent
,
394 if (EFI_ERROR (Status
)) {
397 "%a: RegisterProtocolNotify(): %r\n",
405 // Kick the event; the protocol could be available already.
407 Status
= gBS
->SignalEvent (PlatformHasDeviceTreeEvent
);
408 if (EFI_ERROR (Status
)) {
409 DEBUG ((DEBUG_ERROR
, "%a: SignalEvent(): %r\n", __FUNCTION__
, Status
));
413 Status
= gBS
->InstallProtocolInterface (
415 &gFdtClientProtocolGuid
,
416 EFI_NATIVE_INTERFACE
,
419 if (EFI_ERROR (Status
)) {
422 "%a: InstallProtocolInterface(): %r\n",
432 gBS
->CloseEvent (PlatformHasDeviceTreeEvent
);