#include "Fvb.h"\r
\r
//\r
-// Event for Exit Boot Services Callback\r
+// Event for Set Virtual Map Changed Event\r
//\r
-STATIC EFI_EVENT mExitBootServicesEvent = NULL;\r
+STATIC EFI_EVENT mSetVirtualMapChangedEvent = NULL;\r
\r
//\r
// Lib will ASSERT if more FVB devices than this are added to the system.\r
//\r
STATIC FVB_ENTRY *mFvbEntry;\r
STATIC EFI_EVENT mFvbRegistration;\r
-STATIC BOOLEAN mEfiFvbInitialized = FALSE;\r
STATIC UINTN mFvbCount;\r
\r
/**\r
if (NULL != mFvbEntry[Index].Fvb) {\r
EfiConvertPointer (0x0, (VOID **) &mFvbEntry[Index].Fvb->GetBlockSize);\r
EfiConvertPointer (0x0, (VOID **) &mFvbEntry[Index].Fvb->GetPhysicalAddress);\r
- EfiConvertPointer (0x0, (VOID **) &mFvbEntry[Index].Fvb->GetVolumeAttributes);\r
- EfiConvertPointer (0x0, (VOID **) &mFvbEntry[Index].Fvb->SetVolumeAttributes);\r
+ EfiConvertPointer (0x0, (VOID **) &mFvbEntry[Index].Fvb->GetAttributes);\r
+ EfiConvertPointer (0x0, (VOID **) &mFvbEntry[Index].Fvb->SetAttributes);\r
EfiConvertPointer (0x0, (VOID **) &mFvbEntry[Index].Fvb->Read);\r
EfiConvertPointer (0x0, (VOID **) &mFvbEntry[Index].Fvb->Write);\r
EfiConvertPointer (0x0, (VOID **) &mFvbEntry[Index].Fvb->EraseBlocks);\r
// Register SetVirtualAddressMap () notify function\r
//\r
Status = gBS->CreateEvent (\r
- EVT_SIGNAL_EXIT_BOOT_SERVICES,\r
+ EVT_SIGNAL_VIRTUAL_ADDRESS_CHANGE,\r
TPL_NOTIFY,\r
FvbVirtualAddressChangeNotifyEvent,\r
NULL,\r
- &mExitBootServicesEvent\r
+ &mSetVirtualMapChangedEvent\r
);\r
ASSERT_EFI_ERROR (Status);\r
\r
- mEfiFvbInitialized = TRUE;\r
-\r
return EFI_SUCCESS;\r
}\r
\r
return EFI_INVALID_PARAMETER;\r
}\r
\r
- return mFvbEntry[Instance].Fvb->GetVolumeAttributes (mFvbEntry[Instance].Fvb, Attributes);\r
+ return mFvbEntry[Instance].Fvb->GetAttributes (mFvbEntry[Instance].Fvb, Attributes);\r
}\r
\r
/**\r
**/\r
EFI_STATUS\r
EfiFvbSetVolumeAttributes (\r
- IN UINTN Instance,\r
- IN EFI_FVB_ATTRIBUTES Attributes\r
+ IN UINTN Instance,\r
+ IN OUT EFI_FVB_ATTRIBUTES *Attributes\r
)\r
{\r
if (Instance >= mFvbCount) {\r
return EFI_INVALID_PARAMETER;\r
}\r
\r
- return mFvbEntry[Instance].Fvb->SetVolumeAttributes (mFvbEntry[Instance].Fvb, &Attributes);\r
+ return mFvbEntry[Instance].Fvb->SetAttributes (mFvbEntry[Instance].Fvb, Attributes);\r
}\r
\r
/**\r