Initial import.
git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@3 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
@@ -0,0 +1,266 @@
|
||||
/*++
|
||||
|
||||
Copyright (c) 2006, Intel Corporation
|
||||
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.
|
||||
|
||||
Module Name:
|
||||
|
||||
MonotonicCounter.c
|
||||
|
||||
Abstract:
|
||||
|
||||
Produced the Monotonic Counter Services as defined in the DXE CIS
|
||||
|
||||
Revision History:
|
||||
|
||||
--*/
|
||||
|
||||
#include "MonotonicCounter.h"
|
||||
|
||||
//
|
||||
// The Monotonic Counter Handle
|
||||
//
|
||||
EFI_HANDLE mMonotonicCounterHandle = NULL;
|
||||
|
||||
//
|
||||
// The current Monotonic count value
|
||||
//
|
||||
UINT64 mEfiMtc;
|
||||
|
||||
|
||||
//
|
||||
// Event to use to update the Mtc's high part when wrapping
|
||||
//
|
||||
EFI_EVENT mEfiMtcEvent;
|
||||
|
||||
//
|
||||
// EfiMtcName - Variable name of the MTC value
|
||||
//
|
||||
CHAR16 *mEfiMtcName = (CHAR16 *) L"MTC";
|
||||
|
||||
//
|
||||
// EfiMtcGuid - Guid of the MTC value
|
||||
//
|
||||
EFI_GUID mEfiMtcGuid = { 0xeb704011, 0x1402, 0x11d3, { 0x8e, 0x77, 0x0, 0xa0, 0xc9, 0x69, 0x72, 0x3b } };
|
||||
|
||||
//
|
||||
// Worker functions
|
||||
//
|
||||
EFI_STATUS
|
||||
EFIAPI
|
||||
MonotonicCounterDriverGetNextMonotonicCount (
|
||||
OUT UINT64 *Count
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Arguments:
|
||||
|
||||
Returns:
|
||||
|
||||
--*/
|
||||
{
|
||||
EFI_TPL OldTpl;
|
||||
|
||||
//
|
||||
// Can not be called after ExitBootServices()
|
||||
//
|
||||
if (EfiAtRuntime ()) {
|
||||
return EFI_UNSUPPORTED;
|
||||
}
|
||||
//
|
||||
// Check input parameters
|
||||
//
|
||||
if (Count == NULL) {
|
||||
return EFI_INVALID_PARAMETER;
|
||||
}
|
||||
//
|
||||
// Update the monotonic counter with a lock
|
||||
//
|
||||
OldTpl = gBS->RaiseTPL (EFI_TPL_HIGH_LEVEL);
|
||||
*Count = mEfiMtc;
|
||||
mEfiMtc++;
|
||||
gBS->RestoreTPL (OldTpl);
|
||||
|
||||
//
|
||||
// If the MSB bit of the low part toggled, then signal that the high
|
||||
// part needs updated now
|
||||
//
|
||||
if ((((UINT32) mEfiMtc) ^ ((UINT32) *Count)) & 0x80000000) {
|
||||
gBS->SignalEvent (mEfiMtcEvent);
|
||||
}
|
||||
|
||||
return EFI_SUCCESS;
|
||||
}
|
||||
|
||||
EFI_STATUS
|
||||
EFIAPI
|
||||
MonotonicCounterDriverGetNextHighMonotonicCount (
|
||||
OUT UINT32 *HighCount
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Arguments:
|
||||
|
||||
Returns:
|
||||
|
||||
--*/
|
||||
{
|
||||
EFI_STATUS Status;
|
||||
EFI_TPL OldTpl;
|
||||
|
||||
//
|
||||
// Check input parameters
|
||||
//
|
||||
if (HighCount == NULL) {
|
||||
return EFI_INVALID_PARAMETER;
|
||||
}
|
||||
|
||||
if (!EfiAtRuntime ()) {
|
||||
//
|
||||
// Use a lock if called before ExitBootServices()
|
||||
//
|
||||
OldTpl = gBS->RaiseTPL (EFI_TPL_HIGH_LEVEL);
|
||||
*HighCount = (UINT32) RShiftU64 (mEfiMtc, 32) + 1;
|
||||
mEfiMtc = LShiftU64 (*HighCount, 32);
|
||||
gBS->RestoreTPL (OldTpl);
|
||||
} else {
|
||||
*HighCount = (UINT32) RShiftU64 (mEfiMtc, 32) + 1;
|
||||
mEfiMtc = LShiftU64 (*HighCount, 32);
|
||||
}
|
||||
//
|
||||
// Update the NvRam store to match the new high part
|
||||
//
|
||||
Status = gRT->SetVariable (
|
||||
mEfiMtcName,
|
||||
&mEfiMtcGuid,
|
||||
EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_RUNTIME_ACCESS | EFI_VARIABLE_BOOTSERVICE_ACCESS,
|
||||
sizeof (UINT32),
|
||||
HighCount
|
||||
);
|
||||
|
||||
return Status;
|
||||
}
|
||||
|
||||
VOID
|
||||
EFIAPI
|
||||
EfiMtcEventHandler (
|
||||
IN EFI_EVENT Event,
|
||||
IN VOID *Context
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Monotonic count event handler. This handler updates the high monotonic count.
|
||||
|
||||
Arguments:
|
||||
|
||||
Event The event to handle
|
||||
Context The event context
|
||||
|
||||
Returns:
|
||||
|
||||
EFI_SUCCESS The event has been handled properly
|
||||
EFI_NOT_FOUND An error occurred updating the variable.
|
||||
|
||||
--*/
|
||||
{
|
||||
UINT32 HighCount;
|
||||
|
||||
MonotonicCounterDriverGetNextHighMonotonicCount (&HighCount);
|
||||
}
|
||||
|
||||
EFI_STATUS
|
||||
EFIAPI
|
||||
MonotonicCounterDriverInitialize (
|
||||
IN EFI_HANDLE ImageHandle,
|
||||
IN EFI_SYSTEM_TABLE *SystemTable
|
||||
)
|
||||
/*++
|
||||
|
||||
Routine Description:
|
||||
|
||||
Arguments:
|
||||
(Standard EFI Image entry - EFI_IMAGE_ENTRY_POINT)
|
||||
|
||||
Returns:
|
||||
|
||||
--*/
|
||||
{
|
||||
EFI_STATUS Status;
|
||||
UINT32 HighCount;
|
||||
UINTN BufferSize;
|
||||
|
||||
//
|
||||
// Make sure the Monotonic Counter Architectural Protocol is not already installed in the system
|
||||
//
|
||||
ASSERT_PROTOCOL_ALREADY_INSTALLED (NULL, &gEfiMonotonicCounterArchProtocolGuid);
|
||||
|
||||
//
|
||||
// Initialize event to handle overflows
|
||||
//
|
||||
Status = gBS->CreateEvent (
|
||||
EFI_EVENT_NOTIFY_SIGNAL,
|
||||
EFI_TPL_CALLBACK,
|
||||
EfiMtcEventHandler,
|
||||
NULL,
|
||||
&mEfiMtcEvent
|
||||
);
|
||||
|
||||
ASSERT_EFI_ERROR (Status);
|
||||
|
||||
//
|
||||
// Read the last high part
|
||||
//
|
||||
BufferSize = sizeof (UINT32);
|
||||
Status = gRT->GetVariable (
|
||||
mEfiMtcName,
|
||||
&mEfiMtcGuid,
|
||||
NULL,
|
||||
&BufferSize,
|
||||
&HighCount
|
||||
);
|
||||
if (EFI_ERROR (Status)) {
|
||||
HighCount = 0;
|
||||
}
|
||||
//
|
||||
// Set the current value
|
||||
//
|
||||
mEfiMtc = LShiftU64 (HighCount, 32);
|
||||
|
||||
//
|
||||
// Increment the upper 32 bits for this boot
|
||||
// Continue even if it fails. It will only fail if the variable services are
|
||||
// not functional.
|
||||
//
|
||||
Status = MonotonicCounterDriverGetNextHighMonotonicCount (&HighCount);
|
||||
|
||||
//
|
||||
// Fill in the EFI Boot Services and EFI Runtime Services Monotonic Counter Fields
|
||||
//
|
||||
gBS->GetNextMonotonicCount = MonotonicCounterDriverGetNextMonotonicCount;
|
||||
gRT->GetNextHighMonotonicCount = MonotonicCounterDriverGetNextHighMonotonicCount;
|
||||
|
||||
//
|
||||
// Install the Monotonic Counter Architctural Protocol onto a new handle
|
||||
//
|
||||
Status = gBS->InstallMultipleProtocolInterfaces (
|
||||
&mMonotonicCounterHandle,
|
||||
&gEfiMonotonicCounterArchProtocolGuid,
|
||||
NULL,
|
||||
NULL
|
||||
);
|
||||
ASSERT_EFI_ERROR (Status);
|
||||
|
||||
return EFI_SUCCESS;
|
||||
}
|
Reference in New Issue
Block a user