hc
2024-03-25 edb30157bad0c0001c32b854271ace01d3b9a16a
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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
/** @file
*
*  Copyright (c) 2017, Linaro, Ltd. All rights reserved.
*
*  SPDX-License-Identifier: BSD-2-Clause-Patent
*
**/
 
#include <PiDxe.h>
 
#include <libfdt.h>
#include <Library/BaseLib.h>
#include <Library/DebugLib.h>
#include <Library/DxeServicesLib.h>
#include <Library/MemoryAllocationLib.h>
#include <Library/OpteeLib.h>
#include <Platform/VarStore.h>
 
STATIC
VOID
EnableDtNode (
  IN  VOID                        *Dtb,
  IN  CONST CHAR8                 *NodePath
  )
{
  INT32                           Node;
  INT32                           Rc;
 
  Node = fdt_path_offset (Dtb, NodePath);
  if (Node < 0) {
    DEBUG ((DEBUG_ERROR, "%a: failed to locate DT path '%a': %a\n",
      __FUNCTION__, NodePath, fdt_strerror (Node)));
    return;
  }
  Rc = fdt_setprop_string (Dtb, Node, "status", "okay");
  if (Rc < 0) {
    DEBUG ((DEBUG_ERROR, "%a: failed to set status to 'disabled' on '%a': %a\n",
      __FUNCTION__, NodePath, fdt_strerror (Rc)));
  }
}
 
/**
  Return a pool allocated copy of the DTB image that is appropriate for
  booting the current platform via DT.
 
  @param[out]   Dtb                   Pointer to the DTB copy
  @param[out]   DtbSize               Size of the DTB copy
 
  @retval       EFI_SUCCESS           Operation completed successfully
  @retval       EFI_NOT_FOUND         No suitable DTB image could be located
  @retval       EFI_OUT_OF_RESOURCES  No pool memory available
 
**/
EFI_STATUS
EFIAPI
DtPlatformLoadDtb (
  OUT   VOID        **Dtb,
  OUT   UINTN       *DtbSize
  )
{
  EFI_STATUS                        Status;
  VOID                              *OrigDtb;
  VOID                              *CopyDtb;
  UINTN                             OrigDtbSize;
  UINTN                             CopyDtbSize;
  INT32                             Rc;
  UINT64                            SettingsVal;
  SYNQUACER_PLATFORM_VARSTORE_DATA  *Settings;
 
  Status = GetSectionFromAnyFv (&gDtPlatformDefaultDtbFileGuid,
             EFI_SECTION_RAW, 0, &OrigDtb, &OrigDtbSize);
  if (EFI_ERROR (Status)) {
    return EFI_NOT_FOUND;
  }
 
  CopyDtbSize = OrigDtbSize;
  CopyDtb = AllocatePool (CopyDtbSize);
  if (CopyDtb == NULL) {
    return EFI_OUT_OF_RESOURCES;
  }
 
  Rc = fdt_open_into (OrigDtb, CopyDtb, CopyDtbSize);
  if (Rc < 0) {
    DEBUG ((DEBUG_ERROR, "%a: fdt_open_into () failed: %a\n", __FUNCTION__,
      fdt_strerror (Rc)));
    return EFI_NOT_FOUND;
  }
 
  if (PcdGet8 (PcdPcieEnableMask) & BIT0) {
    EnableDtNode (CopyDtb, "/pcie@60000000");
  }
  if (PcdGet8 (PcdPcieEnableMask) & BIT1) {
    EnableDtNode (CopyDtb, "/pcie@70000000");
  }
 
  SettingsVal = PcdGet64 (PcdPlatformSettings);
  Settings = (SYNQUACER_PLATFORM_VARSTORE_DATA *)&SettingsVal;
  if (Settings->EnableEmmc == EMMC_ENABLED) {
    EnableDtNode (CopyDtb, "/sdhci@52300000");
  }
 
  if (IsOpteePresent()) {
    EnableDtNode (CopyDtb, "/firmware/optee");
  }
 
  *Dtb = CopyDtb;
  *DtbSize = CopyDtbSize;
 
  return EFI_SUCCESS;
}