//**********************************************************************
//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 **
//**********************************************************************
//**********************************************************************
#include "efi.h"
#include "efilib.h"
#define Not_Found_Me 0x00000000
#define Found_Me 0x00000001
static EFI_STATUS WaitForKeyOrReset(VOID)
{
EFI_STATUS Status;
EFI_INPUT_KEY key;
UINTN index;
for(;;) {
Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
if (Status == EFI_NOT_READY)
BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
else
break;
}
if (!EFI_ERROR(Status)) {
if (key.ScanCode == SCAN_ESC)
RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
}
return Status;
}
UINTN
AccessPciConfig (
IN EFI_PCI_IO_PROTOCOL *PCI_DEVICE_TARGET
)
{
UINT8 i;
UINTN PCI_ATTR[4];
UINT8 PCI_REG[4], pci_offset=0;
//get device attr
PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);
if ((PCI_ATTR[1] == 0x00) && (PCI_ATTR[2] == 0x1e) && (PCI_ATTR[3] == 0x00)){
Print (L"Device found at BUS = %02x DEV= %02x FUN= %02x \n",PCI_ATTR[1],PCI_ATTR[2],PCI_ATTR[3]);
ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));
//print register value
for (i = 0;i < 4;i++)
{
PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
Print(L"Register0x%d value = %02x \n",i, PCI_REG[i]);
}
Print(L"\n");
ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));
return Found_Me;
}
return Not_Found_Me;
}
EFI_STATUS
InitializePciApplication (
IN EFI_HANDLE ImageHandle,
IN EFI_SYSTEM_TABLE *SystemTable
)
{
EFI_STATUS Status;
EFI_HANDLE *Handle;
UINTN BufferSize, Result;
UINTN NoHandles, Index;
EFI_PCI_IO_PROTOCOL *PCI_DEVICE_TARGET;
//
// Initialize the Library.
//
InitializeLib (ImageHandle, SystemTable);
//
// Find the PCI driver
//
Handle = NULL;
BufferSize = 1;
do
{
if (Handle)
{
FreePool (Handle);
}
Handle = AllocatePool (BufferSize);
Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
} while (Status == EFI_BUFFER_TOO_SMALL);
ASSERT (!EFI_ERROR(Status));
NoHandles = BufferSize / sizeof(EFI_HANDLE);
Index = 0;
ST->ConOut->ClearScreen (ST->ConOut);
Result = Not_Found_Me;
do
{
if(Index == NoHandles - 1)
Index=0;
else
Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
Print(L"THIS IS DEVICE %d \n",Index);
if (!EFI_ERROR(Status))
{
Result = AccessPciConfig(PCI_DEVICE_TARGET);
//- Print(L"Again now Result is %x \n",Result);
}
Index++;
} while(Result == Not_Found_Me);
Print(L"Please hit any key to exit this image.");
WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
return EFI_SUCCESS;
}
回复 更多评论