EFI_STATUS Status;\r
FDT_CLIENT_PROTOCOL *FdtClient;\r
CONST UINT64 *Reg;\r
- UINT32 RegElemSize, RegSize;\r
+ UINT32 RegSize;\r
+ UINTN AddressCells, SizeCells;\r
EFI_HANDLE Handle;\r
UINT64 RegBase;\r
\r
ASSERT_EFI_ERROR (Status);\r
\r
Status = FdtClient->FindCompatibleNodeReg (FdtClient, "xen,xen",\r
- (CONST VOID **)&Reg, &RegElemSize, &RegSize);\r
+ (CONST VOID **)&Reg, &AddressCells, &SizeCells,\r
+ &RegSize);\r
if (EFI_ERROR (Status)) {\r
DEBUG ((EFI_D_WARN, "%a: No 'xen,xen' compatible DT node found\n",\r
__FUNCTION__));\r
return EFI_UNSUPPORTED;\r
}\r
\r
- ASSERT (RegSize == 16);\r
+ ASSERT (AddressCells == 2);\r
+ ASSERT (SizeCells == 2);\r
+ ASSERT (RegSize == 2 * sizeof (UINT64));\r
\r
//\r
// Retrieve the reg base from this node and wire it up to the\r