summaryrefslogtreecommitdiff
path: root/Core/EM/Ahci/AhciInt13Dxe.c
diff options
context:
space:
mode:
authorraywu <raywu0301@gmail.com>2018-06-15 00:00:50 +0800
committerraywu <raywu0301@gmail.com>2018-06-15 00:00:50 +0800
commitb7c51c9cf4864df6aabb99a1ae843becd577237c (patch)
treeeebe9b0d0ca03062955223097e57da84dd618b9a /Core/EM/Ahci/AhciInt13Dxe.c
downloadzprj-b7c51c9cf4864df6aabb99a1ae843becd577237c.tar.xz
init. 1AQQW051HEADmaster
Diffstat (limited to 'Core/EM/Ahci/AhciInt13Dxe.c')
-rw-r--r--Core/EM/Ahci/AhciInt13Dxe.c412
1 files changed, 412 insertions, 0 deletions
diff --git a/Core/EM/Ahci/AhciInt13Dxe.c b/Core/EM/Ahci/AhciInt13Dxe.c
new file mode 100644
index 0000000..20724ad
--- /dev/null
+++ b/Core/EM/Ahci/AhciInt13Dxe.c
@@ -0,0 +1,412 @@
+//**********************************************************************
+//**********************************************************************
+//** **
+//** (C)Copyright 1985-2014, American Megatrends, Inc. **
+//** **
+//** All Rights Reserved. **
+//** **
+//** 5555 Oakbrook Pkwy, Suite 200, Norcross, GA 30093 **
+//** **
+//** Phone: (770)-246-8600 **
+//** **
+//**********************************************************************
+//**********************************************************************
+//**********************************************************************
+// $Header: /Alaska/SOURCE/Modules/AHCI/INT13/AhciInt13Dxe.c 1 11/25/14 12:09a Kapilporwal $Revision:
+//
+// $Date: 11/25/14 12:09a $Log: /Alaska/SOURCE/Modules/AHCI/INT13/AhciInt13Dxe.c $
+//
+//
+//
+//****************************************************************************
+
+//<AMI_FHDR_START>
+//****************************************************************************
+//
+// Name: AhciInt13Dxe.c
+//
+// Description:
+// This file will register one CALLBACK function, AmiLegacyBootNotify(),
+// it will collect AHCI Int13 runtime data from gAhciI13Data and
+// send it to SMM for AHCI INT13 SMI handler.
+//****************************************************************************
+//<AMI_FHDR_END>
+
+//---------------------------------------------------------------------------
+#include <Token.h>
+#include <AmiDxeLib.h>
+
+#if defined(PI_SPECIFICATION_VERSION)&&(PI_SPECIFICATION_VERSION>=0x0001000A)&&(CORE_COMBINED_VERSION>=0x4028B)
+#include <Protocol/SmmCommunication.h>
+#include <Protocol/LegacyBios.h>
+#include <Protocol/LegacyBiosExt.h>
+#include <Protocol/PciIo.h>
+#include <Protocol/BlockIo.h>
+#include <Protocol/PDiskInfo.h>
+#include <Protocol/PIDEController.h>
+#include <Protocol/PIDEBus.h>
+#include <Protocol/PAhciBus.h>
+#include "AInt13.h"
+#include "AhciInt13Smm.h"
+
+AHCI_I13_RTDATA *gAhciI13Data = NULL;
+extern EFI_GUID gAint13SmmDataGuid;
+EFI_GUID gEfiLegacyBiosProtocolGuid = EFI_LEGACY_BIOS_PROTOCOL_GUID;
+EFI_GUID gEfiDiskInfoProtocolGuid = EFI_DISK_INFO_PROTOCOL_GUID;
+EFI_GUID gEfiEventLegacyBootGuid = EFI_EVENT_LEGACY_BOOT_GUID;
+
+//---------------------------------------------------------------------------
+
+//<AMI_PHDR_START>
+//---------------------------------------------------------------------------
+//
+// Name: IsSataDeviceInAhciMode
+//
+// Description: Check input BBS device is a SATA DEVICE in AHCI mode.
+//
+// Input: BBS_TABLE* - Pointer to an entry in BBS table
+//
+// Output: UINT8 TRUE - It is a SATA device and in AHCI mode
+// FALSE - It is not a SATA device or not in AHCI mode
+//
+//---------------------------------------------------------------------------
+//<AMI_PHDR_END>
+
+UINT8
+IsSataDeviceInAhciMode (
+ IN BBS_TABLE *BbsEntry
+)
+{
+
+ if((BbsEntry->DeviceType == BBS_HARDDISK || BbsEntry->DeviceType == BBS_CDROM)) {
+ if(BbsEntry->Class == MASS_STORAGE || BbsEntry->SubClass == AHCI_CONTROLLER ){
+ return TRUE;
+ }
+ }
+
+ return FALSE;
+}
+
+//<AMI_PHDR_START>
+//---------------------------------------------------------------------------
+//
+// Name: InitializeAhciI13Data
+//
+// Description: It initializes AHCI INT13 runtime data pointer (gAhciI13Data).
+//
+// Input: BBS_TABLE* - Pointer to an entry in BBS table
+//
+// Output: VOID
+//
+//---------------------------------------------------------------------------
+//<AMI_PHDR_END>
+
+VOID
+InitializeAhciI13Data (
+ IN BBS_TABLE *BbsEntry
+)
+{
+ UINT16 StrSeg = 0, StrOff = 0;
+ UINT32 DevOutfitAddr = 0, DevOutfitStartAddr = 0;
+ UINT8 AInt13No = 0, DevNo = 0;
+
+ if(!gAhciI13Data) {
+ // INT13 Drive number (0x80~0x8F)
+ AInt13No = (UINT8)(BbsEntry->InitPerReserved >> 8);
+ DevNo = AInt13No & 0x7F; // Drive index
+
+ // Get address of drive description string i.e. AHCI_I13_RTDATA.DevOutfit[DevNo]
+ StrSeg = BbsEntry->DescStringSegment;
+ StrOff = BbsEntry->DescStringOffset;
+ DevOutfitAddr = (UINT32)((UINTN)StrSeg << 4) + (UINT32)StrOff;
+
+ // Now get address of description string of 1st drive i.e. AHCI_I13_RTDATA.DevOutfit[0]
+ DevOutfitStartAddr = DevOutfitAddr - (sizeof(DEV_BBS_OUTFIT)*DevNo);
+ // Now get base address of AHCI_I13_RTDATA structure
+ gAhciI13Data = (AHCI_I13_RTDATA*)(DevOutfitStartAddr - EFI_FIELD_OFFSET(AHCI_I13_RTDATA, DevOutfit));
+ }
+
+ return;
+}
+
+//<AMI_PHDR_START>
+//---------------------------------------------------------------------------
+//
+// Name: TransferAhciInt13SmmDataToSmm
+//
+// Description: It uses EFI_SMM_COMMUNICATION_PROTOCOL API to transfer data
+// from Non-SMM mode to SMM mode.
+//
+// Input: VOID* - Pointer to data to be transfered
+// UINTN - size of the data
+// EFI_GUID* - Pointer to GUID identifier for the data
+//
+// Output: EFI_STATUS
+//
+//---------------------------------------------------------------------------
+//<AMI_PHDR_END>
+
+EFI_STATUS
+TransferAhciInt13SmmDataToSmm (
+ IN VOID *pData,
+ IN UINTN DataSize,
+ IN EFI_GUID *pGuid
+)
+{
+ EFI_SMM_COMMUNICATION_PROTOCOL *gSmmCommunication = NULL;
+ EFI_SMM_COMMUNICATE_HEADER *SmmCommunicateHeader;
+ UINT8 *CommunicateBuffer = NULL;
+ UINTN CommunicateBufferSize;
+ EFI_STATUS Status;
+
+ if ( pData == NULL || DataSize == 0 || pGuid == NULL ) {
+ return EFI_INVALID_PARAMETER;
+ }
+
+ // Calculate Size of Communication buffer
+ CommunicateBufferSize = (OFFSET_OF (EFI_SMM_COMMUNICATE_HEADER, Data)) + DataSize; // Header size (without data) + data size
+
+ // Allocate memory for Communication Buffer.
+ Status = pBS->AllocatePool( EfiBootServicesData,
+ CommunicateBufferSize,
+ (VOID**)&CommunicateBuffer );
+ if ( EFI_ERROR( Status )) {
+ ASSERT_EFI_ERROR(Status);
+ return Status;
+ }
+
+ // Copy SMM Communicate Header Here
+ SmmCommunicateHeader = (EFI_SMM_COMMUNICATE_HEADER *)CommunicateBuffer;
+
+ // Copy data GUID
+ pBS->CopyMem( &SmmCommunicateHeader->HeaderGuid, pGuid, sizeof( EFI_GUID ) );
+
+ // Updated data length
+ SmmCommunicateHeader->MessageLength = DataSize;
+
+ // Copy Data Here
+ pBS->CopyMem( &SmmCommunicateHeader->Data, pData, DataSize );
+
+ // Locate EFI_SMM_COMMUNICATION_PROTOCOL
+ Status = pBS->LocateProtocol (&gEfiSmmCommunicationProtocolGuid, NULL, (VOID **) &gSmmCommunication);
+ if (EFI_ERROR(Status)) {
+ ASSERT_EFI_ERROR(Status);
+ return Status;
+ }
+
+ // Send data to SMM using protocol API
+ Status = gSmmCommunication->Communicate (gSmmCommunication, CommunicateBuffer, &CommunicateBufferSize);
+ if (EFI_ERROR(Status)) {
+ ASSERT_EFI_ERROR(Status);
+ return Status;
+ }
+
+ // Free memory allocated for Communication Buffer.
+ Status = pBS->FreePool(CommunicateBuffer);
+ if (EFI_ERROR(Status)) {
+ ASSERT_EFI_ERROR(Status);
+ return Status;
+ }
+
+ return EFI_SUCCESS;
+}
+
+//<AMI_PHDR_START>
+//---------------------------------------------------------------------------
+//
+// Name: AmiLegacyBootNotify
+//
+// Description: This function will be called upon legacy boot event. It
+// collects information about all AHCI devices present in the system and
+// send it to SMM so that AHCI INT13 SMI handler can use it.
+// Operation:
+// 1. Locate EFI_LEGACY_BIOS_PROTOCOL and get All BBS entries and look
+// for entry corresponding to a drive in AHCI mode
+// 2. If a valid entry is found then it will use AHCI INT13 runtime
+// data and AHCI bus interface data to fill data required by AHCI
+// INT13 SMI handler.
+// 3. This newly created data structure will be transfered to SMM using
+// SmmCommunicationProtocol API.
+//
+// Input: EFI_EVENT - Event
+// VOID* - Pointer to Context
+//
+// Output: VOID
+//
+//---------------------------------------------------------------------------
+//<AMI_PHDR_END>
+
+VOID
+AmiLegacyBootNotify(
+ IN EFI_EVENT Event,
+ IN VOID *Context
+)
+{
+ EFI_STATUS Status;
+ UINT16 HddCount = 0;
+ HDD_INFO *HddInfo = NULL;
+ BBS_TABLE *BbsTable = NULL;
+ UINT16 BbsCount = 0;
+ UINT8 DevNo = 0;
+ UINTN SegNum;
+ UINTN BusNum;
+ UINTN DevNum;
+ UINTN FuncNum;
+ UINT16 i = 0;
+ EFI_LEGACY_BIOS_PROTOCOL *LegacyBios;
+ EFI_DISK_INFO_PROTOCOL *DiskInfo = NULL;
+ SATA_DEVICE_INTERFACE *SataDevInterface = NULL;
+ SMM_AINT13_DRIVE_INFO *pDriveInfo = NULL;
+ AHCI_INT13_SMM_DATA *AhciInt13SmmData = NULL;
+
+ // Run this function only once
+ pBS->CloseEvent(Event);
+
+ // Locate EFI_LEGACY_BIOS_PROTOCOL
+ Status = pBS->LocateProtocol( &gEfiLegacyBiosProtocolGuid,
+ NULL,
+ &LegacyBios);
+ if (EFI_ERROR(Status)) {
+ return;
+ }
+
+ // Get BBS_TABLE
+ Status = LegacyBios->GetBbsInfo( LegacyBios,
+ &HddCount,
+ &HddInfo,
+ &BbsCount,
+ &BbsTable);
+ if (EFI_ERROR(Status)) {
+ return;
+ }
+
+ // Allocate Memory for AhciInt13SmmData buffer.
+ Status = pBS->AllocatePool( EfiBootServicesData,
+ sizeof(AHCI_INT13_SMM_DATA),
+ (VOID**)&AhciInt13SmmData );
+ if ( EFI_ERROR( Status )) {
+ ASSERT_EFI_ERROR(Status);
+ return;
+ }
+
+ // Initialize drive count to 0
+ AhciInt13SmmData->DriveCount = 0;
+
+ // Loop through all BBS entries
+ for (i = 0; i < BbsCount; i++) {
+
+ // Process if this BDS entry is corresponding to a HDD in AHCI mode
+ if (IsSataDeviceInAhciMode(&BbsTable[i])) {
+
+ // NOTE: We don't need EFI_DISK_INFO_PROTOCOL this is just to get SATA_DEVICE_INTERFACE structure.
+ Status = pBS->HandleProtocol ( (EFI_HANDLE)BbsTable[i].IBV1, &gEfiDiskInfoProtocolGuid, &DiskInfo );
+ if(EFI_ERROR(Status)) {
+ continue; // Goto next BBS entry
+ }
+ SataDevInterface = ((SATA_DISK_INFO *)DiskInfo)->SataDevInterface;
+
+ // Fill data in AhciInt13SmmData
+ pDriveInfo = &(AhciInt13SmmData->DriveInfo[AhciInt13SmmData->DriveCount]);
+ pDriveInfo->DriveNum = (UINT8)(BbsTable[i].InitPerReserved >> 8);
+ pDriveInfo->PMPortNum = SataDevInterface->PMPortNumber;
+ pDriveInfo->PortNum = SataDevInterface->PortNumber;
+ SataDevInterface->AhciBusInterface->PciIO->GetLocation(
+ SataDevInterface->AhciBusInterface->PciIO,
+ &SegNum, &BusNum, &DevNum, &FuncNum
+ );
+ pDriveInfo->BusNo = (UINT8)BusNum;
+ pDriveInfo->DevNo = (UINT8)DevNum;
+ pDriveInfo->FuncNo = (UINT8)FuncNum;
+ pDriveInfo->DeviceType = SataDevInterface->DeviceType;
+ if(pDriveInfo->DeviceType == ATAPI) {
+ pDriveInfo->Lun = SataDevInterface->AtapiDevice->Lun;
+ pDriveInfo->BlockSize = SataDevInterface->AtapiDevice->BlockSize;
+ } else {
+ // Initialize AHCI INT13 runtime data
+ if(!gAhciI13Data) {
+ InitializeAhciI13Data(&BbsTable[i]);
+ }
+ // Get drive index in device parameter array of gAhciI13Data
+ DevNo = pDriveInfo->DriveNum & 0x7F;
+
+ // Fill the information using gAhciI13Data
+ pDriveInfo->wMAXCYL = gAhciI13Data->DevParam[DevNo].wMAXCYL;
+ pDriveInfo->bMAXHN = gAhciI13Data->DevParam[DevNo].bMAXHN;
+ pDriveInfo->bMAXSN = gAhciI13Data->DevParam[DevNo].bMAXSN;
+ pDriveInfo->wLBACYL = gAhciI13Data->DevParam[DevNo].wLBACYL;
+ pDriveInfo->bLBAHD = gAhciI13Data->DevParam[DevNo].bLBAHD;
+ pDriveInfo->bLBASPT = gAhciI13Data->DevParam[DevNo].bLBASPT;
+ }
+ pDriveInfo->RCommand = SataDevInterface->ReadCommand;
+ pDriveInfo->WCommand = SataDevInterface->WriteCommand;
+
+ AhciInt13SmmData->DriveCount += 1; // Update drive count
+ }
+ }
+
+ // Save AhciInt13SmmData in SMM
+ if(AhciInt13SmmData->DriveCount != 0) {
+ Status = TransferAhciInt13SmmDataToSmm ( AhciInt13SmmData, sizeof(AHCI_INT13_SMM_DATA), &gAint13SmmDataGuid );
+ }
+
+ // Free the Memory Allocated for AhciInt13SmmData Buffer.
+ Status = pBS->FreePool(AhciInt13SmmData);
+ if (EFI_ERROR(Status)) {
+ ASSERT_EFI_ERROR(Status);
+ return;
+ }
+}
+
+//<AMI_PHDR_START>
+//---------------------------------------------------------------------------
+//
+// Name: AhciInt13DxeEntry
+//
+// Description: This is entry point function. It registers a call back function
+// for Legacy boot event.
+//
+// Input: EFI_HANDLE - Standard EFI Image handle
+// EFI_SYSTEM_TABLE* - Pointer to System Table
+//
+// Output: EFI_STATUS
+//
+//---------------------------------------------------------------------------
+//<AMI_PHDR_END>
+
+EFI_STATUS
+AhciInt13DxeEntry(
+ IN EFI_HANDLE ImageHandle,
+ IN EFI_SYSTEM_TABLE *SystemTable
+)
+{
+ EFI_STATUS Status;
+ EFI_EVENT Event;
+
+ InitAmiLib(ImageHandle, SystemTable);
+
+ Status = pBS->CreateEventEx (
+ EFI_EVENT_NOTIFY_SIGNAL,
+ TPL_CALLBACK,
+ AmiLegacyBootNotify,
+ NULL,
+ &gEfiEventLegacyBootGuid,
+ &Event
+ );
+
+ return Status;
+}
+#endif
+
+//**********************************************************************
+//**********************************************************************
+//** **
+//** (C)Copyright 1985-2014, American Megatrends, Inc. **
+//** **
+//** All Rights Reserved. **
+//** **
+//** 5555 Oakbrook Pkwy, Suite 200, Norcross, GA 30093 **
+//** **
+//** Phone: (770)-246-8600 **
+//** **
+//**********************************************************************
+//**********************************************************************