/** @file\r
Report Status Code Router PEIM which produces Report Stataus Code Handler PPI and Status Code PPI.\r
\r
- Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
+ Copyright (c) 2009 - 2011, 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_PEI_RSC_HANDLER_CALLBACK *CallbackEntry;\r
UINTN *NumberOfEntries;\r
UINTN Index;\r
+ UINTN FreeEntryIndex;\r
UINTN *FreePacket;\r
\r
if (Callback == NULL) {\r
return EFI_INVALID_PARAMETER;\r
}\r
\r
- Hob.Raw = GetFirstGuidHob (&gStatusCodeCallbackGuid);\r
- FreePacket = NULL;\r
+ Hob.Raw = GetFirstGuidHob (&gStatusCodeCallbackGuid);\r
+ FreePacket = NULL;\r
+ FreeEntryIndex = 0;\r
while (Hob.Raw != NULL) {\r
NumberOfEntries = GET_GUID_HOB_DATA (Hob);\r
CallbackEntry = (EFI_PEI_RSC_HANDLER_CALLBACK *) (NumberOfEntries + 1);\r
- if (*NumberOfEntries < 64) {\r
+ if (FreePacket == NULL && *NumberOfEntries < 64) {\r
+ //\r
+ // If current total number of handlers does not exceed 64, put new handler\r
+ // at the last of packet\r
+ //\r
FreePacket = NumberOfEntries;\r
+ FreeEntryIndex = *NumberOfEntries;\r
}\r
for (Index = 0; Index < *NumberOfEntries; Index++) {\r
if (CallbackEntry[Index] == Callback) {\r
//\r
return EFI_ALREADY_STARTED;\r
}\r
+ if (FreePacket == NULL && CallbackEntry[Index] == NULL) {\r
+ //\r
+ // If the total number of handlers in current packet is max value 64,\r
+ // search an entry with NULL pointer and fill new handler into this entry.\r
+ // \r
+ FreePacket = NumberOfEntries;\r
+ FreeEntryIndex = Index;\r
+ }\r
}\r
Hob.Raw = GET_NEXT_HOB (Hob);\r
Hob.Raw = GetNextGuidHob (&gStatusCodeCallbackGuid, Hob.Raw);\r
}\r
\r
CallbackEntry = (EFI_PEI_RSC_HANDLER_CALLBACK *) (FreePacket + 1);\r
- CallbackEntry[*FreePacket] = Callback;\r
- *FreePacket += 1;\r
+ CallbackEntry[FreeEntryIndex] = Callback;\r
+ \r
+ if (*FreePacket == FreeEntryIndex) {\r
+ //\r
+ // If new registered callback is added as a new entry in the packet,\r
+ // increase the total number of handlers in the packet.\r
+ //\r
+ *FreePacket += 1;\r
+ }\r
\r
return EFI_SUCCESS;\r
}\r
CallbackEntry = (EFI_PEI_RSC_HANDLER_CALLBACK *) (NumberOfEntries + 1);\r
for (Index = 0; Index < *NumberOfEntries; Index++) {\r
if (CallbackEntry[Index] == Callback) {\r
- CallbackEntry[Index] = CallbackEntry[*NumberOfEntries - 1];\r
- *NumberOfEntries -= 1;\r
+ //\r
+ // Set removed entry as NULL.\r
+ //\r
+ CallbackEntry[Index] = NULL;\r
return EFI_SUCCESS;\r
}\r
}\r
return EFI_DEVICE_ERROR;\r
}\r
\r
- for (Link = GetFirstNode (&mCallbackListHead); !IsNull (&mCallbackListHead, Link); Link = GetNextNode (&mCallbackListHead, Link)) {\r
+ for (Link = GetFirstNode (&mCallbackListHead); !IsNull (&mCallbackListHead, Link);) {\r
CallbackEntry = CR (Link, RSC_HANDLER_CALLBACK_ENTRY, Node, RSC_HANDLER_CALLBACK_ENTRY_SIGNATURE);\r
-\r
+ //\r
+ // The handler may remove itself, so get the next handler in advance.\r
+ //\r
+ Link = GetNextNode (&mCallbackListHead, Link);\r
if ((CallbackEntry->Tpl == TPL_HIGH_LEVEL) || EfiAtRuntime ()) {\r
CallbackEntry->RscHandlerCallback (\r
Type,\r
Report Status Code Router Driver which produces SMM Report Stataus Code Handler Protocol\r
and SMM Status Code Protocol.\r
\r
- Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
+ Copyright (c) 2009 - 2011, 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
return EFI_DEVICE_ERROR;\r
}\r
\r
- for (Link = GetFirstNode (&mCallbackListHead); !IsNull (&mCallbackListHead, Link); Link = GetNextNode (&mCallbackListHead, Link)) {\r
+ for (Link = GetFirstNode (&mCallbackListHead); !IsNull (&mCallbackListHead, Link);) {\r
CallbackEntry = CR (Link, SMM_RSC_HANDLER_CALLBACK_ENTRY, Node, SMM_RSC_HANDLER_CALLBACK_ENTRY_SIGNATURE);\r
-\r
+ //\r
+ // The handler may remove itself, so get the next handler in advance.\r
+ //\r
+ Link = GetNextNode (&mCallbackListHead, Link);\r
CallbackEntry->RscHandlerCallback (\r
Type,\r
Value,\r