@@ -17,8 +17,8 @@ extern "C" {
1717 #include <Protocol/MmReportStatusCodeHandler.h>
1818}
1919
20- struct MockReportStatusCodeHandler {
21- MOCK_INTERFACE_DECLARATION (MockReportStatusCodeHandler );
20+ struct MockEfiMmRscHandlerProtocol {
21+ MOCK_INTERFACE_DECLARATION (MockEfiMmRscHandlerProtocol );
2222
2323 MOCK_FUNCTION_DECLARATION (
2424 EFI_STATUS ,
@@ -34,17 +34,14 @@ struct MockReportStatusCodeHandler {
3434 );
3535};
3636
37- MOCK_INTERFACE_DEFINITION (MockReportStatusCodeHandler );
38- MOCK_FUNCTION_DEFINITION (MockReportStatusCodeHandler , Register , 1 , EFIAPI );
39- MOCK_FUNCTION_DEFINITION (MockReportStatusCodeHandler , Unregister , 1 , EFIAPI );
37+ MOCK_INTERFACE_DEFINITION (MockEfiMmRscHandlerProtocol );
38+ MOCK_FUNCTION_DEFINITION (MockEfiMmRscHandlerProtocol , Register , 1 , EFIAPI );
39+ MOCK_FUNCTION_DEFINITION (MockEfiMmRscHandlerProtocol , Unregister , 1 , EFIAPI );
4040
41- EFI_MM_RSC_HANDLER_PROTOCOL MmRscHandlerProtocol = {
42- Register ,
43- Unregister
44- };
45-
46- extern "C" {
47- EFI_MM_RSC_HANDLER_PROTOCOL * MmRscHandlerProtocolServices = & MmRscHandlerProtocol ;
48- }
41+ #define MOCK_EFI_MM_RSC_HANDLER_PROTOCOL_INSTANCE (NAME ) \
42+ EFI_MM_RSC_HANDLER_PROTOCOL NAME##_INSTANCE = { \
43+ Register, \
44+ Unregister }; \
45+ EFI_MM_RSC_HANDLER_PROTOCOL *NAME = &NAME##_INSTANCE;
4946
5047#endif
0 commit comments