Reviewed-by: Eric Dong <eric.d...@intel.com>
> -----Original Message----- > From: edk2-devel [mailto:edk2-devel-boun...@lists.01.org] On Behalf Of > Dandan Bi > Sent: Monday, October 16, 2017 11:37 AM > To: edk2-devel@lists.01.org > Cc: Dong, Eric <eric.d...@intel.com>; Zhang, Chao B > <chao.b.zh...@intel.com> > Subject: [edk2] [patch] Security/OpalHii.c: Handle NULL Request or Request > with no elements > > According to UEFI spec, for the ExtractConfig function in > EFI_HII_CONFIG_ACCESS_PROTOCOL,If a NULL is passed in for the Request > field or if a ConfigHdr is passed in with no request elements, all of the > settings being abstracted by this function will be returned in the Results > field. > > The implementation of ExtractConfig function in OpalHii.c misses to handle > above cases.This patch is to do the enhancements. > > Cc: Chao Zhang <chao.b.zh...@intel.com> > Cc: Eric Dong <eric.d...@intel.com> > Contributed-under: TianoCore Contribution Agreement 1.1 > Signed-off-by: Dandan Bi <dandan...@intel.com> > --- > SecurityPkg/Tcg/Opal/OpalPasswordDxe/OpalHii.c | 46 > +++++++++++++++++++++++++- > 1 file changed, 45 insertions(+), 1 deletion(-) > > diff --git a/SecurityPkg/Tcg/Opal/OpalPasswordDxe/OpalHii.c > b/SecurityPkg/Tcg/Opal/OpalPasswordDxe/OpalHii.c > index 4881e72..e3bde42 100644 > --- a/SecurityPkg/Tcg/Opal/OpalPasswordDxe/OpalHii.c > +++ b/SecurityPkg/Tcg/Opal/OpalPasswordDxe/OpalHii.c > @@ -1278,10 +1278,16 @@ ExtractConfig( > EFI_STRING *Progress, > EFI_STRING *Results > ) > { > EFI_STATUS Status; > + EFI_STRING ConfigRequest; > + EFI_STRING ConfigRequestHdr; > + UINTN BufferSize; > + UINTN Size; > + BOOLEAN AllocatedRequest; > + EFI_HANDLE DriverHandle; > > // > // Check for valid parameters > // > if (Progress == NULL || Results == NULL) { @@ -1292,22 +1298,60 @@ > ExtractConfig( > if ((Request != NULL) && > !HiiIsConfigHdrMatch (Request, &gHiiSetupVariableGuid, > OpalPasswordStorageName)) { > return EFI_NOT_FOUND; > } > > + AllocatedRequest = FALSE; > + BufferSize = sizeof (OPAL_HII_CONFIGURATION); ConfigRequest = > + Request; if ((Request == NULL) || (StrStr (Request, L"OFFSET") == > + NULL)) { > + // > + // Request has no request element, construct full request string. > + // Allocate and fill a buffer large enough to hold the <ConfigHdr> > template > + // followed by "&OFFSET=0&WIDTH=WWWWWWWWWWWWWWWW" > followed by a Null-terminator > + // > + DriverHandle = HiiGetDriverImageHandleCB(); > + ConfigRequestHdr = HiiConstructConfigHdr (&gHiiSetupVariableGuid, > OpalPasswordStorageName, DriverHandle); > + Size = (StrLen (ConfigRequestHdr) + 32 + 1) * sizeof (CHAR16); > + ConfigRequest = AllocateZeroPool (Size); > + if (ConfigRequest == NULL) { > + return EFI_OUT_OF_RESOURCES; > + } > + AllocatedRequest = TRUE; > + UnicodeSPrint (ConfigRequest, Size, L"%s&OFFSET=0&WIDTH=%016LX", > ConfigRequestHdr, (UINT64)BufferSize); > + FreePool (ConfigRequestHdr); > + } > + > // > // Convert Buffer Data to <ConfigResp> by helper function BlockToConfig( ) > // > Status = gHiiConfigRouting->BlockToConfig( > gHiiConfigRouting, > - Request, > + ConfigRequest, > (UINT8*)&gHiiConfiguration, > sizeof(OPAL_HII_CONFIGURATION), > Results, > Progress > ); > > + // > + // Free the allocated config request string. > + // > + if (AllocatedRequest) { > + FreePool (ConfigRequest); > + ConfigRequest = NULL; > + } > + > + // > + // Set Progress string to the original request string. > + // > + if (Request == NULL) { > + *Progress = NULL; > + } else if (StrStr (Request, L"OFFSET") == NULL) { > + *Progress = Request + StrLen (Request); } > + > return (Status); > } > > > /** > -- > 1.9.5.msysgit.1 > > _______________________________________________ > edk2-devel mailing list > edk2-devel@lists.01.org > https://lists.01.org/mailman/listinfo/edk2-devel _______________________________________________ edk2-devel mailing list edk2-devel@lists.01.org https://lists.01.org/mailman/listinfo/edk2-devel