summaryrefslogtreecommitdiffstats
path: root/ArmVirtPkg/HighMemDxe/HighMemDxe.c
blob: 4d56e6236b54eeac81e80550c924cd66347d57f9 (plain)
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
/** @file
*  High memory node enumeration DXE driver for ARM Virtual Machines
*
*  Copyright (c) 2015, Linaro Ltd. All rights reserved.
*
*  This program and the accompanying materials are licensed and made available
*  under the terms and conditions of the BSD License which accompanies this
*  distribution.  The full text of the license may be found at
*  http://opensource.org/licenses/bsd-license.php
*
*  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
*  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR
*  IMPLIED.
*
**/

#include <Library/BaseLib.h>
#include <Library/UefiDriverEntryPoint.h>
#include <Library/DebugLib.h>
#include <Library/PcdLib.h>
#include <Library/HobLib.h>
#include <libfdt.h>
#include <Library/DxeServicesTableLib.h>

EFI_STATUS
EFIAPI
InitializeHighMemDxe (
  IN EFI_HANDLE           ImageHandle,
  IN EFI_SYSTEM_TABLE     *SystemTable
  )
{
  VOID             *Hob;
  VOID             *DeviceTreeBase;
  INT32            Node, Prev;
  EFI_STATUS       Status;
  CONST CHAR8      *Type;
  INT32            Len;
  CONST VOID       *RegProp;
  UINT64           CurBase;
  UINT64           CurSize;

  Hob = GetFirstGuidHob(&gFdtHobGuid);
  if (Hob == NULL || GET_GUID_HOB_DATA_SIZE (Hob) != sizeof (UINT64)) {
    return EFI_NOT_FOUND;
  }
  DeviceTreeBase = (VOID *)(UINTN)*(UINT64 *)GET_GUID_HOB_DATA (Hob);

  if (fdt_check_header (DeviceTreeBase) != 0) {
    DEBUG ((EFI_D_ERROR, "%a: No DTB found @ 0x%p\n", __FUNCTION__,
      DeviceTreeBase));
    return EFI_NOT_FOUND;
  }

  DEBUG ((EFI_D_INFO, "%a: DTB @ 0x%p\n", __FUNCTION__, DeviceTreeBase));

  //
  // Check for memory node and add the memory spaces expect the lowest one
  //
  for (Prev = 0;; Prev = Node) {
    Node = fdt_next_node (DeviceTreeBase, Prev, NULL);
    if (Node < 0) {
      break;
    }

    Type = fdt_getprop (DeviceTreeBase, Node, "device_type", &Len);
    if (Type && AsciiStrnCmp (Type, "memory", Len) == 0) {
      //
      // Get the 'reg' property of this node. For now, we will assume
      // two 8 byte quantities for base and size, respectively.
      //
      RegProp = fdt_getprop (DeviceTreeBase, Node, "reg", &Len);
      if (RegProp != NULL && Len == (2 * sizeof (UINT64))) {

        CurBase = fdt64_to_cpu (((UINT64 *)RegProp)[0]);
        CurSize = fdt64_to_cpu (((UINT64 *)RegProp)[1]);

        if (PcdGet64 (PcdSystemMemoryBase) != CurBase) {
          Status = gDS->AddMemorySpace (
                          EfiGcdMemoryTypeSystemMemory,
                          CurBase, CurSize,
                          EFI_MEMORY_WB);

          if (EFI_ERROR (Status)) {
            DEBUG ((EFI_D_ERROR,
              "%a: Failed to add System RAM @ 0x%lx - 0x%lx (%r)\n",
              __FUNCTION__, CurBase, CurBase + CurSize - 1, Status));
            continue;
          }

          Status = gDS->SetMemorySpaceAttributes (
                          CurBase, CurSize,
                          EFI_MEMORY_WB);

          if (EFI_ERROR (Status)) {
            DEBUG ((EFI_D_ERROR,
              "%a: Failed to set System RAM @ 0x%lx - 0x%lx attribute (%r)\n",
              __FUNCTION__, CurBase, CurBase + CurSize - 1, Status));
          } else {
            DEBUG ((EFI_D_INFO, "%a: Add System RAM @ 0x%lx - 0x%lx\n",
              __FUNCTION__, CurBase, CurBase + CurSize - 1));
          }
        }
      }
    }
  }

  return EFI_SUCCESS;
}