MockMpService.h

Go to the documentation of this file.

 
#ifndef MOCK_MP_SERVICE_H_
#define MOCK_MP_SERVICE_H_
 
#include <Library/GoogleTestLib.h>
#include <Library/FunctionMockLib.h>
 
extern "C" {
  #include <Uefi.h>
  #include <Protocol/MpService.h>
}
 
struct MockMpService {
  MOCK_INTERFACE_DECLARATION (MockMpService);
 
  MOCK_FUNCTION_DECLARATION (
    EFI_STATUS,
    GetNumberOfProcessors,
    (
     IN  EFI_MP_SERVICES_PROTOCOL  *This,
     OUT UINTN                     *NumberOfProcessors,
     OUT UINTN                     *NumberOfEnabledProcessors
    )
    );
 
  MOCK_FUNCTION_DECLARATION (
    EFI_STATUS,
    GetProcessorInfo,
    (
     IN  EFI_MP_SERVICES_PROTOCOL   *This,
     IN  UINTN                      ProcessorNumber,
     OUT EFI_PROCESSOR_INFORMATION  *ProcessorInfoBuffer
    )
    );
 
  MOCK_FUNCTION_DECLARATION (
    EFI_STATUS,
    StartupAllAPs,
    (
     IN  EFI_MP_SERVICES_PROTOCOL  *This,
     IN  EFI_AP_PROCEDURE          Procedure,
     IN  BOOLEAN                   SingleThread,
     IN  EFI_EVENT                 WaitEvent               OPTIONAL,
     IN  UINTN                     TimeoutInMicroSeconds,
     IN  VOID                      *ProcedureArgument      OPTIONAL,
     OUT UINTN                     **FailedCpuList         OPTIONAL
    )
    );
 
  MOCK_FUNCTION_DECLARATION (
    EFI_STATUS,
    StartupThisAP,
    (
     IN  EFI_MP_SERVICES_PROTOCOL  *This,
     IN  EFI_AP_PROCEDURE          Procedure,
     IN  UINTN                     ProcessorNumber,
     IN  EFI_EVENT                 WaitEvent               OPTIONAL,
     IN  UINTN                     TimeoutInMicroseconds,
     IN  VOID                      *ProcedureArgument      OPTIONAL,
     OUT BOOLEAN                   *Finished               OPTIONAL
    )
    );
 
  MOCK_FUNCTION_DECLARATION (
    EFI_STATUS,
    SwitchBSP,
    (
     IN EFI_MP_SERVICES_PROTOCOL  *This,
     IN  UINTN                    ProcessorNumber,
     IN  BOOLEAN                  EnableOldBSP
    )
    );
 
  MOCK_FUNCTION_DECLARATION (
    EFI_STATUS,
    EnableDisableAP,
    (
     IN  EFI_MP_SERVICES_PROTOCOL  *This,
     IN  UINTN                     ProcessorNumber,
     IN  BOOLEAN                   EnableAP,
     IN  UINT32                    *HealthFlag OPTIONAL
    )
    );
 
  MOCK_FUNCTION_DECLARATION (
    EFI_STATUS,
    WhoAmI,
    (
     IN EFI_MP_SERVICES_PROTOCOL  *This,
     OUT UINTN                    *ProcessorNumber
    )
    );
};
 
MOCK_INTERFACE_DEFINITION (MockMpService);
MOCK_FUNCTION_DEFINITION (MockMpService, GetNumberOfProcessors, 3, EFIAPI);
MOCK_FUNCTION_DEFINITION (MockMpService, GetProcessorInfo, 3, EFIAPI);
MOCK_FUNCTION_DEFINITION (MockMpService, StartupAllAPs, 7, EFIAPI);
MOCK_FUNCTION_DEFINITION (MockMpService, StartupThisAP, 7, EFIAPI);
MOCK_FUNCTION_DEFINITION (MockMpService, SwitchBSP, 3, EFIAPI);
MOCK_FUNCTION_DEFINITION (MockMpService, EnableDisableAP, 4, EFIAPI);
MOCK_FUNCTION_DEFINITION (MockMpService, WhoAmI, 2, EFIAPI);
 
static EFI_MP_SERVICES_PROTOCOL  MP_SERVICE_PROTOCOL_INSTANCE = {
  GetNumberOfProcessors,
  GetProcessorInfo,
  StartupAllAPs,
  StartupThisAP,
  SwitchBSP,
  EnableDisableAP,
  WhoAmI
};
 
extern "C" {
  EFI_MP_SERVICES_PROTOCOL  *gMpServiceProtocol = &MP_SERVICE_PROTOCOL_INSTANCE;
}
 
#endif // MOCK_MP_SERVICE_H_