ArmPkg/ArmScmiDxe: Fix ASSERT error in SCMI DXE

[mirror_edk2.git] / ArmPkg / Drivers / ArmScmiDxe / ScmiDxe.h

diff --git a/ArmPkg/Drivers/ArmScmiDxe/ScmiDxe.h b/ArmPkg/Drivers/ArmScmiDxe/ScmiDxe.h
index 29cdde173659c701116b021a3c437a92b473e4e5..222e54f4dca558d9b1fedddf3f96fd977898c7b2 100644 (file)
--- a/ArmPkg/Drivers/ArmScmiDxe/ScmiDxe.h
+++ b/ArmPkg/Drivers/ArmScmiDxe/ScmiDxe.h
@@ -17,8 +17,9 @@
 #ifndef SCMI_DXE_H_\r
 #define SCMI_DXE_H_\r
 \r
+#include "ScmiPrivate.h"\r
+\r
 #define MAX_PROTOCOLS        6\r
-#define PROTOCOL_ID_MASK     0xF\r
 #define MAX_VENDOR_LEN       SCMI_MAX_STR_LEN\r
 \r
 /** Pointer to protocol initialization function.\r
@@ -35,7 +36,8 @@ EFI_STATUS
   );\r
 \r
 typedef struct {\r
-  SCMI_PROTOCOL_INIT_FXN Init;\r
-} SCMI_PROTOCOL_INIT_TABLE;\r
+  SCMI_PROTOCOL_ID Id;            // Protocol Id.\r
+  SCMI_PROTOCOL_INIT_FXN InitFn;  // Protocol init function.\r
+} SCMI_PROTOCOL_ENTRY;\r
 \r
 #endif /* SCMI_DXE_H_ */\r