[原创]EFI application：PCI read & write 程序源码

kaurohy · 发表于 2009-11-11 16:21:18

/*
* myfileio.c
* Apps
*/

#include "efi.h"
#include "efilib.h"

#define FILE_ATTRIB_CREATENEW       EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

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;
}

EFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE          ImageHandle,
         IN EFI_SYSTEM_TABLE    *SystemTable)
{
EFI_STATUS Status;
EFI_HANDLE *DestAddr;
EFI_LOADED_IMAGE *FileHandle01;
EFI_DEVICE_PATH *FileHandle02;
EFI_FILE_IO_INTERFACE *FileHandle03;
EFI_FILE *FileHandle04;
EFI_FILE *FileHandle05;
CHAR16 *FileName;
CHAR16    *BufferA, *BufferB, *BufferC, *Space_Key;
UINTN BufferSize = 8, BufferSizeKeySpace = 2;
int       Bit0, Bit1, Key_Space;

FileName = L"NewFile.txt";
Space_Key = L" ";

BufferA = L"ABCD";
BufferB = L"EFGH";
BufferC = L"IJKL";

Bit0 = 0xff;
Bit1 = 0xfe;

Key_Space = 0x0020;

InitializeLib (ImageHandle, SystemTable);

DestAddr = AllocatePool (BufferSize);

Print (L"Value of Bit0 is %x\n", Bit0);
Print (L"Value of Bit1 is %x\n", Bit1);

Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
if (EFI_ERROR(Status)) {
      Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
      return EFI_LOAD_ERROR;
}

Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
if (EFI_ERROR(Status)) {
         Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
         return EFI_LOAD_ERROR;
      }

Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
if (EFI_ERROR(Status)) {
         Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
         return EFI_LOAD_ERROR;
      }

Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
if (EFI_ERROR(Status)) {
      Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
      return EFI_LOAD_ERROR;
}

Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
if (EFI_ERROR(Status)) {
      Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
      return EFI_LOAD_ERROR;
}

Status = FileHandle05->SetPosition(FileHandle05, 0);
if (EFI_ERROR(Status)) {
      Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
Print (L"File Buffersize is %x\n\n", BufferSize);
if (EFI_ERROR(Status)) {
      Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
      return EFI_SUCCESS;
}

Status = FileHandle05->SetPosition(FileHandle05, 1);
if (EFI_ERROR(Status)) {
      Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
Print (L"File Buffersize is %x\n\n", BufferSize);
if (EFI_ERROR(Status)) {
      Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
      return EFI_SUCCESS;
}

Status = FileHandle05->SetPosition(FileHandle05, 2);
if (EFI_ERROR(Status)) {
      Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
Print (L"File Buffersize is %x\n\n", BufferSize);
if (EFI_ERROR(Status)) {
      Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
      return EFI_SUCCESS;
}

//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
Status = FileHandle05->SetPosition(FileHandle05, 2);
if (EFI_ERROR(Status)) {
      Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
if (EFI_ERROR(Status)) {
      Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}
Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

Status = FileHandle05->SetPosition(FileHandle05, 10);
if (EFI_ERROR(Status)) {
      Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
if (EFI_ERROR(Status)) {
      Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

Status = FileHandle05->SetPosition(FileHandle05, 12);
if (EFI_ERROR(Status)) {
      Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
if (EFI_ERROR(Status)) {
      Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

Status = FileHandle05->SetPosition(FileHandle05, 14);
if (EFI_ERROR(Status)) {
      Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
if (EFI_ERROR(Status)) {
      Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

Status = FileHandle05->SetPosition(FileHandle05, 16);
if (EFI_ERROR(Status)) {
      Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
if (EFI_ERROR(Status)) {
      Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

//---------------------------------------------------------------------------------------------------------------------------------------------------------------

Status = FileHandle05->SetPosition(FileHandle05, 18);
if (EFI_ERROR(Status)) {
      Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
if (EFI_ERROR(Status)) {
      Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

FreePool(DestAddr);

Status = FileHandle05->Close(FileHandle05);
if (EFI_ERROR(Status)) {
      Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

Print(L"File Name = %s has already been created.\n", FileName);

Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
WaitForKeyOrReset();

return Status;
}

chenhy · 发表于 2009-11-30 09:50:02

//**********************************************************************
//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**
//**********************************************************************
//**********************************************************************

#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);
0 _2 l8 R1 x& J8 h  R    }
7 r# ~$ }- d5 N" T8 b! s% X    Print(L"\n");
; v) _0 i9 q5 U+ x+ t# b    ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));
8 |7 X& D* Z4 _# W" T2 M8 W% g% H    return Found_Me;
+ G" ?9 o- ]" V }9 C  R# D$ z7 T& f+ r4 A7 L- w
4 a9 K9 V1 L: `  z" o# \
return Not_Found_Me;1 P5 z1 t3 ?- P$ R
}9 C; L. [% U: X/ Z4 u) b3 n

+ I/ I* M0 g5 T% CEFI_STATUS/ P, g# T: l" o
InitializePciApplication (
2 P6 i) e$ A5 ]# ]! \3 E6 J IN EFI_HANDLE          ImageHandle,' w) j/ U' H) s* F1 G0 r4 D
IN EFI_SYSTEM_TABLE    *SystemTable  g" h" i! x& {7 w& O, r% @
)5 V, K! G# d% C
{" p& v$ b3 G% i$ A; d5 b: b) d8 r3 Y7 p
  EFI_STATUS             Status;
% d% [2 \2 V: m! F1 U; \1 v( x  EFI_HANDLE             *Handle;
6 ?& ~/ G# [2 `& t  UINTN                BufferSize, Result;" H# |) v- l. t1 Z/ @$ u2 d
  UINTN                NoHandles, Index;* U7 [" H2 i( C' R
  EFI_PCI_IO_PROTOCOL    *PCI_DEVICE_TARGET;
2 C: A' E. K! o4 B( y/ B
, A6 z4 g$ L7 ]# K, P) u //: G9 `2 t4 X' n6 D
// Initialize the Library.9 g4 t' @! c. F
//
; X/ T" D9 ~1 l* r7 ~: y/ H InitializeLib (ImageHandle, SystemTable);9 B) K% H$ x% Z, N  C8 ?3 A0 k
//
/ E! L1 m$ u/ Q% ^0 ~) i3 ?, b( B( `. _ // Find the PCI driver- V6 k3 R6 C8 i) W4 g3 H
//
& k( X% P$ d1 l7 e Handle = NULL;
/ I: E3 s' h2 k3 c BufferSize = 1;. j! D1 g9 m/ \6 j, U  V
do
6 L5 Z& w$ r1 l4 r& h+ @ {
# [; P: w1 i, V2 V    if (Handle)3 n8 Q4 \1 v* k
   {5 K) m+ P& h* K1 a5 X
      FreePool (Handle);) |5 c/ v4 c! K* K
   }
/ N9 I! W) O( v
5 r) u; W( I- P$ F& x* z    Handle = AllocatePool (BufferSize);3 z6 S) H. `% g" x
   Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
0 P$ g! U7 ~0 w1 _" ]  y
1 $ w/ k& s/ r  |0 }' e$ ?- z6 O5 g } while (Status == EFI_BUFFER_TOO_SMALL);
5 D8 m; u3 S8 i% O- K ASSERT (!EFI_ERROR(Status));3 D: f2 p+ P' A

! t, R* a, t' M- { NoHandles  = BufferSize / sizeof(EFI_HANDLE);  K8 M' x5 ^" k9 A4 Y7 x7 `2 ?
Index = 0;3 X* x; j6 j3 _! J) $ i% ?) w( M
ST->ConOut->ClearScreen (ST->ConOut);: b" r# }" s; p7 h$ U
Result = Not_Found_Me;
) [8 C% z+ x- d4 a4 ~  E* U do
, W5 y3 y+ l8 l+ S { % J  c; `, g' ]" x
if(Index == NoHandles - 1)1 K# K+ |+ [- j9 ?" c4 A
      Index=0;7 u0 i. t# r# L( U( j
      else
/ y' c- U1 a: X4 k+ O Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
! i" X/ Z: I( j/ _ Print(L"THIS IS DEVICE %d \n",Index);! H+ U" r" R& N1 X& [2 N$ S
if (!EFI_ERROR(Status))8 T( j- u3 |  U& M9 N! v- k
{
% n5 f% Q. `' ?6 I+ O    Result = AccessPciConfig(PCI_DEVICE_TARGET);- E: m8 h  o7 g1 ~) {
//-    Print(L"Again now Result is %x \n",Result);
" n8 F% I' d0 h5 S }
* s8 e5 t! K$ ? Index++;
6 w1 o- `' `7 ` } while(Result == Not_Found_Me);
; K! Q: s9 L4 T  \8 i" t 0 S+ ^& j) G  {
Print(L"Please hit any key to exit this image."); + x2 M0 j( q& o4 T6 F' c
WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
7 F, v, V; v9 z6 J
9 r4 `9 }5 \/ J2 I/ S return EFI_SUCCESS;4 c( w$ b& y; T" M% V
}

xjf2088 · 发表于 2010-4-16 14:47:39

小弟对你有如滔滔江水。。。

dhua · 发表于 2010-4-30 14:51:38

都是强人。向你们学习！

shaopy · 发表于 2010-8-23 12:52:26

楼主好厉害我接受bios tranning 两周也没有写出这样的的底层程式！楼主强人哈！学习了！

蓝色永恒 · 发表于 2010-8-23 14:07:13

景仰一下，哈哈

zywwin · 发表于 2012-5-21 08:57:06

新手，学习

		自动登录	找回密码
密码			加入计匠网

[原创]EFI application：PCI read & write 程序源码

我寫了一個 File io 的 EFI shell app

pci register access sample

感激不尽

浏览过的版块