SecurityPkg OpalPasswordSmm: Fix S3 resume failure.

Changes includes:
1.Check SMM device list before update it to avoid duplicate creation.
2.Clean up the configuration buffer before use it in S3 resume phase.

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Eric Dong <eric.dong@intel.com>
Reviewed-by: Feng Tian <feng.tian@intel.com>
This commit is contained in:
Eric Dong 2016-10-11 16:00:21 +08:00 committed by Star Zeng
parent 19e3aa7a8a
commit 0acd8df45f
2 changed files with 76 additions and 21 deletions

View File

@ -1691,6 +1691,14 @@ NvmeControllerInit (
Nvme->Cid[0] = 0; Nvme->Cid[0] = 0;
Nvme->Cid[1] = 0; Nvme->Cid[1] = 0;
Nvme->Pt[0] = 0;
Nvme->Pt[1] = 0;
ZeroMem ((VOID *)(UINTN)(&(Nvme->SqTdbl[0])), sizeof (NVME_SQTDBL) * NVME_MAX_IO_QUEUES);
ZeroMem ((VOID *)(UINTN)(&(Nvme->CqHdbl[0])), sizeof (NVME_CQHDBL) * NVME_MAX_IO_QUEUES);
ZeroMem ((VOID *)(UINTN)Nvme->BaseMem, NVME_MEM_MAX_SIZE);
Status = NvmeDisableController (Nvme); Status = NvmeDisableController (Nvme);
if (EFI_ERROR(Status)) { if (EFI_ERROR(Status)) {
DEBUG ((DEBUG_ERROR, "NvmeDisableController fail, Status: %r\n", Status)); DEBUG ((DEBUG_ERROR, "NvmeDisableController fail, Status: %r\n", Status));

View File

@ -483,6 +483,53 @@ done:
return Status; return Status;
} }
/**
The function extracts device information from OpalDeviceList and creat SmmDeviceList used for S3.
@param[in] OpalDeviceList Opal device list created at POST which contains the information of OPAL_DISK_AND_PASSWORD_INFO
@param[in,out] SmmDeviceList Opal Smm device list to be created and used for unlocking devices at S3 resume.
@retval EFI_SUCCESS Create SmmDeviceList successfully.
@retval Others Other execution results.
**/
EFI_STATUS
CreateSmmDeviceList (
IN LIST_ENTRY *OpalDeviceList,
IN OUT LIST_ENTRY *SmmDeviceList
)
{
LIST_ENTRY *Entry;
OPAL_DISK_AND_PASSWORD_INFO *PciDev;
OPAL_SMM_DEVICE *SmmDev;
for (Entry = OpalDeviceList->ForwardLink; Entry != OpalDeviceList; Entry = Entry->ForwardLink) {
PciDev = BASE_CR (Entry, OPAL_DISK_AND_PASSWORD_INFO, Link);
SmmDev = AllocateZeroPool (sizeof (OPAL_SMM_DEVICE));
if (SmmDev == NULL) {
return EFI_OUT_OF_RESOURCES;
}
SmmDev->Signature = OPAL_SMM_DEVICE_SIGNATURE;
ExtractDeviceInfoFromDevicePath(&PciDev->OpalDevicePath, SmmDev);
SmmDev->PasswordLength = PciDev->PasswordLength;
CopyMem(&(SmmDev->Password), PciDev->Password, OPAL_PASSWORD_MAX_LENGTH);
SmmDev->Sscp.ReceiveData = SecurityReceiveData;
SmmDev->Sscp.SendData = SecuritySendData;
DEBUG ((DEBUG_INFO, "Opal SMM: Insert device node to SmmDeviceList:\n"));
DEBUG ((DEBUG_INFO, "DeviceType:%x, Bus:%d, Dev:%d, Fun:%d\n", \
SmmDev->DeviceType, SmmDev->BusNum, SmmDev->DevNum, SmmDev->FuncNum));
DEBUG ((DEBUG_INFO, "SataPort:%x, MultiplierPort:%x, NvmeNamespaceId:%x\n", \
SmmDev->SataPort, SmmDev->SataPortMultiplierPort, SmmDev->NvmeNamespaceId));
InsertHeadList (SmmDeviceList, &SmmDev->Link);
}
return EFI_SUCCESS;
}
/** /**
Main entry point for an SMM handler dispatch or communicate-based callback. Main entry point for an SMM handler dispatch or communicate-based callback.
@ -521,7 +568,6 @@ S3SleepEntryCallBack (
UINT64 Address; UINT64 Address;
S3_BOOT_SCRIPT_LIB_WIDTH Width; S3_BOOT_SCRIPT_LIB_WIDTH Width;
UINT32 Data; UINT32 Data;
OPAL_DISK_AND_PASSWORD_INFO *PciDev;
OPAL_HC_PCI_REGISTER_SAVE *HcRegisterSaveListPtr; OPAL_HC_PCI_REGISTER_SAVE *HcRegisterSaveListPtr;
UINTN Count; UINTN Count;
OPAL_SMM_DEVICE *SmmDev; OPAL_SMM_DEVICE *SmmDev;
@ -530,25 +576,28 @@ S3SleepEntryCallBack (
Status = EFI_SUCCESS; Status = EFI_SUCCESS;
mOpalDeviceList = OpalSupportGetOpalDeviceList(); mOpalDeviceList = OpalSupportGetOpalDeviceList();
if (IsListEmpty (mOpalDeviceList)) {
//
// No Opal enabled device. Do nothing.
//
return EFI_SUCCESS;
}
for (Entry = mOpalDeviceList->ForwardLink; Entry != mOpalDeviceList; Entry = Entry->ForwardLink) { if (IsListEmpty (&mSmmDeviceList)) {
PciDev = BASE_CR (Entry, OPAL_DISK_AND_PASSWORD_INFO, Link); //
// mSmmDeviceList for S3 is empty, creat it by mOpalDeviceList.
SmmDev = AllocateZeroPool (sizeof (OPAL_SMM_DEVICE)); //
if (SmmDev == NULL) { Status = CreateSmmDeviceList (mOpalDeviceList, &mSmmDeviceList);
return EFI_OUT_OF_RESOURCES; if (EFI_ERROR (Status)) {
return Status;
} }
SmmDev->Signature = OPAL_SMM_DEVICE_SIGNATURE; }
ExtractDeviceInfoFromDevicePath(&PciDev->OpalDevicePath, SmmDev); //
// Go through SmmDeviceList to save register data for S3
SmmDev->PasswordLength = PciDev->PasswordLength; //
CopyMem(&(SmmDev->Password), PciDev->Password, OPAL_PASSWORD_MAX_LENGTH); for (Entry = mSmmDeviceList.ForwardLink; Entry != &mSmmDeviceList; Entry = Entry->ForwardLink) {
SmmDev = BASE_CR (Entry, OPAL_SMM_DEVICE, Link);
SmmDev->Sscp.ReceiveData = SecurityReceiveData;
SmmDev->Sscp.SendData = SecuritySendData;
InsertHeadList (&mSmmDeviceList, &SmmDev->Link);
if (SmmDev->DeviceType == OPAL_DEVICE_TYPE_NVME) { if (SmmDev->DeviceType == OPAL_DEVICE_TYPE_NVME) {
continue; continue;
@ -592,10 +641,8 @@ S3SleepEntryCallBack (
} }
} }
if (!IsListEmpty (mOpalDeviceList)) { Status = S3BootScriptSaveIoWrite (S3BootScriptWidthUint8, 0xB2, 1, &mSwSmiValue);
Status = S3BootScriptSaveIoWrite (S3BootScriptWidthUint8, 0xB2, 1, &mSwSmiValue); ASSERT_EFI_ERROR (Status);
ASSERT_EFI_ERROR (Status);
}
return Status; return Status;
} }