4. It save all the mapping info in NV variables which will be consumed\r
by platform override protocol driver to publish the platform override protocol.\r
\r
-Copyright (c) 2007 - 2014, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2007 - 2015, Intel Corporation. All rights reserved.<BR>\r
This program and the accompanying materials\r
are licensed and made available under the terms and conditions of the BSD License\r
which accompanies this distribution. The full text of the license may be found at\r
EFI_HANDLE *mDriverImageHandleBuffer;\r
\r
INTN mSelectedCtrIndex;\r
-EFI_STRING_ID mControllerToken[MAX_CHOICE_NUM];\r
+EFI_STRING_ID *mControllerToken;\r
UINTN mDriverImageHandleCount;\r
-EFI_STRING_ID mDriverImageToken[MAX_CHOICE_NUM];\r
-EFI_STRING_ID mDriverImageFilePathToken[MAX_CHOICE_NUM];\r
-EFI_LOADED_IMAGE_PROTOCOL *mDriverImageProtocol[MAX_CHOICE_NUM];\r
-EFI_DEVICE_PATH_PROTOCOL *mControllerDevicePathProtocol[MAX_CHOICE_NUM];\r
+EFI_STRING_ID *mDriverImageToken;\r
+EFI_DEVICE_PATH_PROTOCOL **mControllerDevicePathProtocol;\r
UINTN mSelectedDriverImageNum;\r
UINTN mLastSavedDriverImageNum;\r
UINT16 mCurrentPage;\r
EFI_CALLBACK_INFO *mCallbackInfo;\r
+BOOLEAN *mDriSelection;\r
+UINTN mMaxDeviceCount;\r
\r
HII_VENDOR_DEVICE_PATH mHiiVendorDevicePath = {\r
{\r
EFI_STATUS Status;\r
UINTN Index;\r
UINTN DevicePathHandleCount;\r
+ UINTN NewStrSize;\r
CHAR16 *NewString;\r
EFI_STRING_ID NewStringToken;\r
CHAR16 *ControllerName;\r
return EFI_SUCCESS;\r
}\r
\r
+ mMaxDeviceCount = DevicePathHandleCount;\r
+ mControllerDevicePathProtocol = AllocateZeroPool (DevicePathHandleCount * sizeof (EFI_DEVICE_PATH_PROTOCOL *));\r
+ ASSERT (mControllerDevicePathProtocol != NULL);\r
+ mControllerToken = AllocateZeroPool (DevicePathHandleCount * sizeof (EFI_STRING_ID));\r
+ ASSERT (mControllerToken != NULL);\r
+\r
for (Index = 0; Index < DevicePathHandleCount; Index++) {\r
if (FakeNvData->PciDeviceFilter == 0x01) {\r
//\r
// Export the driver name string and create item in set options page\r
//\r
Len = StrSize (ControllerName);\r
- NewString = AllocateZeroPool (Len + StrSize (L"--"));\r
+ NewStrSize = Len + StrSize (L"--");\r
+ NewString = AllocateZeroPool (NewStrSize);\r
ASSERT (NewString != NULL);\r
if (EFI_ERROR (CheckMapping (ControllerDevicePath,NULL, &mMappingDataBase, NULL, NULL))) {\r
- StrCat (NewString, L"--");\r
+ StrCatS (NewString, NewStrSize/sizeof(CHAR16), L"--");\r
} else {\r
- StrCat (NewString, L"**");\r
+ StrCatS (NewString, NewStrSize/sizeof(CHAR16), L"**");\r
}\r
- StrCat (NewString, ControllerName);\r
+ StrCatS (NewString, NewStrSize/sizeof(CHAR16), ControllerName);\r
\r
NewStringToken = HiiSetString (Private->RegisteredHandle, mControllerToken[Index], NewString, NULL);\r
ASSERT (NewStringToken != 0);\r
@param ImageHandle The Image handle\r
\r
@return Handle to Driver binding\r
- @retval NULL The paramter is not valid or the driver binding handle is not found.\r
+ @retval NULL The parameter is not valid or the driver binding handle is not found.\r
\r
**/\r
EFI_HANDLE\r
{\r
EFI_STATUS Status;\r
UINTN Index;\r
+ UINTN NewStrSize;\r
CHAR16 *NewString;\r
EFI_STRING_ID NewStringToken;\r
EFI_STRING_ID NewStringHelpToken;\r
VOID *EndOpCodeHandle;\r
EFI_IFR_GUID_LABEL *StartLabel;\r
EFI_IFR_GUID_LABEL *EndLabel;\r
+ EFI_LOADED_IMAGE_PROTOCOL **DriverImageProtocol;\r
+ EFI_STRING_ID *DriverImageFilePathToken;\r
+ UINT8 CheckFlags;\r
\r
//\r
// If user select a controller item in the first page the following code will be run.\r
return EFI_NOT_FOUND;\r
}\r
\r
+ mDriverImageToken = AllocateZeroPool (DriverImageHandleCount * sizeof (EFI_STRING_ID));\r
+ ASSERT (mDriverImageToken != NULL);\r
+ mDriSelection = AllocateZeroPool (DriverImageHandleCount * sizeof (BOOLEAN));\r
+ ASSERT (mDriSelection != NULL);\r
+\r
+ DriverImageProtocol = AllocateZeroPool (DriverImageHandleCount * sizeof (EFI_LOADED_IMAGE_PROTOCOL *));\r
+ ASSERT (DriverImageProtocol != NULL);\r
+ DriverImageFilePathToken = AllocateZeroPool (DriverImageHandleCount * sizeof (EFI_STRING_ID));\r
+ ASSERT (DriverImageFilePathToken != NULL);\r
+\r
mDriverImageHandleCount = DriverImageHandleCount;\r
for (Index = 0; Index < DriverImageHandleCount; Index++) {\r
//\r
EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
);\r
if (EFI_ERROR (Status)) {\r
- FakeNvData->DriSelection[Index] = 0x00;\r
+ mDriSelection[Index] = FALSE;\r
continue;\r
}\r
- mDriverImageProtocol[Index] = LoadedImage;\r
+ DriverImageProtocol[Index] = LoadedImage;\r
//\r
// Find its related driver binding protocol\r
//\r
DriverBindingHandle = GetDriverBindingHandleFromImageHandle (mDriverImageHandleBuffer[Index]);\r
if (DriverBindingHandle == NULL) {\r
- FakeNvData->DriSelection[Index] = 0x00;\r
+ mDriSelection[Index] = FALSE;\r
continue;\r
}\r
\r
(VOID **) &LoadedImageDevicePath\r
);\r
if (LoadedImageDevicePath == NULL) {\r
- FakeNvData->DriSelection[Index] = 0x00;\r
+ mDriSelection[Index] = FALSE;\r
continue;\r
}\r
\r
(VOID **) &BusSpecificDriverOverride\r
);\r
if (EFI_ERROR (Status) || BusSpecificDriverOverride == NULL) {\r
- FakeNvData->DriSelection[Index] = 0x00;\r
+ mDriSelection[Index] = FALSE;\r
continue;\r
}\r
} else {\r
- FakeNvData->DriSelection[Index] = 0x00;\r
+ mDriSelection[Index] = FALSE;\r
continue;\r
}\r
}\r
//\r
// First create the driver image name\r
//\r
- NewString = AllocateZeroPool (StrSize (DriverName));\r
+ NewStrSize = StrSize (DriverName);\r
+ NewString = AllocateZeroPool (NewStrSize);\r
ASSERT (NewString != NULL); \r
if (EFI_ERROR (CheckMapping (mControllerDevicePathProtocol[mSelectedCtrIndex], LoadedImageDevicePath, &mMappingDataBase, NULL, NULL))) {\r
- FakeNvData->DriSelection[Index] = 0x00;\r
+ mDriSelection[Index] = FALSE;\r
} else {\r
- FakeNvData->DriSelection[Index] = 0x01;\r
+ mDriSelection[Index] = TRUE;\r
mLastSavedDriverImageNum++;\r
}\r
- StrCat (NewString, DriverName);\r
+ StrCatS (NewString, NewStrSize/sizeof(CHAR16), DriverName);\r
NewStringToken = HiiSetString (Private->RegisteredHandle, mDriverImageToken[Index], NewString, NULL);\r
ASSERT (NewStringToken != 0);\r
mDriverImageToken[Index] = NewStringToken;\r
//\r
DriverName = DevicePathToStr (LoadedImageDevicePath);\r
\r
- NewString = AllocateZeroPool (StrSize (DriverName));\r
+ NewStrSize = StrSize (DriverName);\r
+ NewString = AllocateZeroPool (NewStrSize);\r
ASSERT (NewString != NULL); \r
- StrCat (NewString, DriverName);\r
- NewStringHelpToken = HiiSetString (Private->RegisteredHandle, mDriverImageFilePathToken[Index], NewString, NULL);\r
+ StrCatS (NewString, NewStrSize/sizeof(CHAR16), DriverName);\r
+ NewStringHelpToken = HiiSetString (Private->RegisteredHandle, DriverImageFilePathToken[Index], NewString, NULL);\r
ASSERT (NewStringHelpToken != 0);\r
- mDriverImageFilePathToken[Index] = NewStringHelpToken;\r
+ DriverImageFilePathToken[Index] = NewStringHelpToken;\r
FreePool (NewString);\r
FreePool (DriverName);\r
\r
+ CheckFlags = 0;\r
+ if (mDriSelection[Index]) {\r
+ CheckFlags |= EFI_IFR_CHECKBOX_DEFAULT;\r
+ }\r
+\r
HiiCreateCheckBoxOpCode (\r
StartOpCodeHandle,\r
- (UINT16) (DRIVER_SELECTION_QUESTION_ID + Index),\r
- VARSTORE_ID_PLAT_OVER_MNGR,\r
- (UINT16) (DRIVER_SELECTION_VAR_OFFSET + Index),\r
- NewStringToken,\r
- NewStringHelpToken,\r
+ (UINT16) (KEY_VALUE_DRIVER_OFFSET + Index),\r
0,\r
0,\r
+ NewStringToken,\r
+ NewStringHelpToken,\r
+ EFI_IFR_FLAG_CALLBACK,\r
+ CheckFlags,\r
NULL\r
);\r
}\r
\r
HiiFreeOpCodeHandle (StartOpCodeHandle);\r
HiiFreeOpCodeHandle (EndOpCodeHandle);\r
+\r
+ if (DriverImageProtocol != NULL) {\r
+ FreePool (DriverImageProtocol);\r
+ }\r
+\r
+ if (DriverImageFilePathToken != NULL) {\r
+ FreePool (DriverImageFilePathToken);\r
+ }\r
+\r
return EFI_SUCCESS;\r
}\r
\r
//\r
SelectedDriverImageNum = 0;\r
for (Index = 0; Index < mDriverImageHandleCount; Index++) {\r
- if (FakeNvData->DriSelection[Index] != 0) {\r
+ if (mDriSelection[Index]) {\r
SelectedDriverImageNum ++;\r
}\r
}\r
//\r
SelectedDriverImageNum = 0;\r
for (Index = 0; Index < mDriverImageHandleCount; Index++) {\r
- if (FakeNvData->DriSelection[Index] != 0) {\r
+ if (mDriSelection[Index]) {\r
//\r
// Use the NO. in driver binding buffer as value, will use it later\r
//\r
\r
**/\r
EFI_STATUS\r
-CommintChanges (\r
+CommitChanges (\r
IN EFI_CALLBACK_INFO *Private,\r
IN UINT16 KeyValue,\r
IN PLAT_OVER_MNGR_DATA *FakeNvData\r
}\r
\r
Status = EFI_SUCCESS;\r
- if (mCurrentPage == FORM_ID_DRIVER) {\r
- KeyValue = KEY_VALUE_DRIVER_GOTO_ORDER;\r
- UpdatePrioritySelectPage (Private, KeyValue, FakeNvData);\r
- KeyValue = KEY_VALUE_ORDER_SAVE_AND_EXIT;\r
- Status = CommintChanges (Private, KeyValue, FakeNvData);\r
- //\r
- // Since UpdatePrioritySelectPage will change mCurrentPage,\r
- // should ensure the mCurrentPage still indicate the second page here\r
- //\r
- mCurrentPage = FORM_ID_DRIVER;\r
- }\r
\r
if (mCurrentPage == FORM_ID_ORDER) {\r
KeyValue = KEY_VALUE_ORDER_SAVE_AND_EXIT;\r
- Status = CommintChanges (Private, KeyValue, FakeNvData);\r
+ Status = CommitChanges (Private, KeyValue, FakeNvData);\r
}\r
\r
return Status;\r
}\r
}\r
\r
- if (((KeyValue >= KEY_VALUE_DEVICE_OFFSET) && (KeyValue < KEY_VALUE_DEVICE_MAX)) || (KeyValue == KEY_VALUE_ORDER_GOTO_PREVIOUS)) {\r
+ if (((KeyValue >= KEY_VALUE_DEVICE_OFFSET) && (KeyValue < KEY_VALUE_DEVICE_OFFSET + mMaxDeviceCount)) || (KeyValue == KEY_VALUE_ORDER_GOTO_PREVIOUS)) {\r
if (KeyValue == KEY_VALUE_ORDER_GOTO_PREVIOUS) {\r
KeyValue = (EFI_QUESTION_ID) (mSelectedCtrIndex + KEY_VALUE_DEVICE_OFFSET);\r
}\r
UpdateDeviceSelectPage (Private, KeyValue, FakeNvData);\r
}\r
} else if (Action == EFI_BROWSER_ACTION_CHANGED) {\r
- switch (KeyValue) {\r
- case KEY_VALUE_DEVICE_REFRESH:\r
- case KEY_VALUE_DEVICE_FILTER:\r
- UpdateDeviceSelectPage (Private, KeyValue, FakeNvData);\r
- //\r
- // Update page title string\r
- //\r
- NewStringToken = STRING_TOKEN (STR_TITLE);\r
- if (HiiSetString (Private->RegisteredHandle, NewStringToken, L"First, Select the controller by device path", NULL) == 0) {\r
- ASSERT (FALSE);\r
- }\r
- break;\r
- \r
- case KEY_VALUE_ORDER_SAVE_AND_EXIT:\r
- Status = CommintChanges (Private, KeyValue, FakeNvData);\r
- *ActionRequest = EFI_BROWSER_ACTION_REQUEST_SUBMIT;\r
- if (EFI_ERROR (Status)) {\r
- CreatePopUp (EFI_LIGHTGRAY | EFI_BACKGROUND_BLUE, &Key, L"Single Override Info too large, Saving Error!", NULL);\r
- return EFI_DEVICE_ERROR;\r
- }\r
- break;\r
+ if ((KeyValue >= KEY_VALUE_DRIVER_OFFSET) && (KeyValue < KEY_VALUE_DRIVER_OFFSET + mDriverImageHandleCount)) {\r
+ mDriSelection[KeyValue - KEY_VALUE_DRIVER_OFFSET] = Value->b;\r
+ } else {\r
+ switch (KeyValue) {\r
+ case KEY_VALUE_DEVICE_REFRESH:\r
+ case KEY_VALUE_DEVICE_FILTER:\r
+ UpdateDeviceSelectPage (Private, KeyValue, FakeNvData);\r
+ //\r
+ // Update page title string\r
+ //\r
+ NewStringToken = STRING_TOKEN (STR_TITLE);\r
+ if (HiiSetString (Private->RegisteredHandle, NewStringToken, L"First, Select the controller by device path", NULL) == 0) {\r
+ ASSERT (FALSE);\r
+ }\r
+ break;\r
+ \r
+ case KEY_VALUE_ORDER_SAVE_AND_EXIT:\r
+ Status = CommitChanges (Private, KeyValue, FakeNvData);\r
+ *ActionRequest = EFI_BROWSER_ACTION_REQUEST_SUBMIT;\r
+ if (EFI_ERROR (Status)) {\r
+ CreatePopUp (EFI_LIGHTGRAY | EFI_BACKGROUND_BLUE, &Key, L"Single Override Info too large, Saving Error!", NULL);\r
+ return EFI_DEVICE_ERROR;\r
+ }\r
+ break;\r
\r
- default:\r
- break;\r
+ default:\r
+ break;\r
+ }\r
}\r
}\r
\r
}\r
\r
/**\r
- The driver Entry Point. The funciton will export a disk device class formset and\r
+ The driver Entry Point. The function will export a disk device class formset and\r
its callback function to hii database.\r
\r
@param ImageHandle The firmware allocated handle for the EFI image.\r
mCallbackInfo->PlatformDriverOverride.GetDriver = GetDriver;\r
mCallbackInfo->PlatformDriverOverride.GetDriverPath = GetDriverPath;\r
mCallbackInfo->PlatformDriverOverride.DriverLoaded = DriverLoaded;\r
+\r
+ //\r
+ // Locate ConfigRouting protocol\r
+ //\r
+ Status = gBS->LocateProtocol (\r
+ &gEfiHiiConfigRoutingProtocolGuid,\r
+ NULL,\r
+ (VOID **) &mCallbackInfo->HiiConfigRouting\r
+ );\r
+ if (EFI_ERROR (Status)) {\r
+ goto Finish;\r
+ }\r
+\r
//\r
// Install Device Path Protocol and Config Access protocol to driver handle\r
// Install Platform Driver Override Protocol to driver handle\r
goto Finish;\r
}\r
\r
- //\r
- // Locate ConfigRouting protocol\r
- //\r
- Status = gBS->LocateProtocol (\r
- &gEfiHiiConfigRoutingProtocolGuid,\r
- NULL,\r
- (VOID **) &mCallbackInfo->HiiConfigRouting\r
- );\r
- if (EFI_ERROR (Status)) {\r
- goto Finish;\r
- }\r
-\r
//\r
// Clear all the globle variable\r
//\r
mDriverImageHandleCount = 0;\r
mCurrentPage = 0;\r
- ZeroMem (mDriverImageToken, MAX_CHOICE_NUM * sizeof (EFI_STRING_ID));\r
- ZeroMem (mDriverImageFilePathToken, MAX_CHOICE_NUM * sizeof (EFI_STRING_ID));\r
- ZeroMem (mControllerToken, MAX_CHOICE_NUM * sizeof (EFI_STRING_ID));\r
- ZeroMem (mDriverImageProtocol, MAX_CHOICE_NUM * sizeof (EFI_LOADED_IMAGE_PROTOCOL *));\r
\r
return EFI_SUCCESS;\r
\r
\r
FreePool (mCallbackInfo);\r
\r
+ if (mControllerToken != NULL) {\r
+ FreePool (mControllerToken);\r
+ }\r
+\r
+ if (mControllerDevicePathProtocol != NULL) {\r
+ FreePool (mControllerDevicePathProtocol);\r
+ }\r
+\r
+ if (mDriverImageToken != NULL) {\r
+ FreePool (mDriverImageToken);\r
+ }\r
+\r
return EFI_SUCCESS;\r
}\r