init. 1AQQW051HEADmaster

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                      **
+//**                                                                  **
+//**********************************************************************
+//**********************************************************************