[原创]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);
# y- v' h, X2 |% x    }6 [/ B  J) G, W/ b
   Print(L"\n");
& s4 c4 x( a  s4 g3 h: b- a- R% J% m    ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));    0 J% X7 s$ ~$ L* _
   return Found_Me;, q  i( @$ k) `% ?: W. ^7 A
}
3 ~) \( {7 v) n7 ?
) }. t5 G) H# \ return Not_Found_Me;
+ k& F! n3 X. W7 `7 u* R}
+ M/ C; J) ?0 s! p+ X
6 L4 M2 c1 X' rEFI_STATUS+ \- w5 h: n9 n7 {
InitializePciApplication (
) [! e5 |" I6 \2 y& k: B+ ` IN EFI_HANDLE          ImageHandle,
4 U- H8 F* U; L IN EFI_SYSTEM_TABLE    *SystemTable
: L- L5 E* n. H- M6 ~ )" f4 r1 s7 b3 a
{
/ T# G2 ~$ E+ o1 X, i, ^$ g/ M' t  EFI_STATUS             Status;9 E/ @- s  P( E) y! Q
  EFI_HANDLE             *Handle;
  @/ q. |$ W/ h+ S# H  UINTN                BufferSize, Result;
5 K* f8 L+ n+ T7 ?) q- |) x* u  UINTN                NoHandles, Index;9 D; A/ s2 r5 X0 ^$ o1 F. p" Y
  EFI_PCI_IO_PROTOCOL    *PCI_DEVICE_TARGET;3 l: [) ~* m; v$ q. \- \0 b: l

7 |7 F+ @+ W2 I5 [$ B$ F4 m" G! G //
# l" _* @+ y6 q9 A0 s3 T3 O, R3 K+ j // Initialize the Library.
+ r, K; l; D' c6 r //
( ^" ^5 [/ u2 r: u# M: d/ P9 N2 g InitializeLib (ImageHandle, SystemTable);8 I  B* h8 z. G- W: J  ?
//+ A$ r9 h) d0 P
// Find the PCI driver
9 J7 q" Y' G2 _" _! b0 x4 E) R, i //* H* E! p* _4 V- k, ]" j% w
Handle = NULL;
/ B* ?* m% d9 q2 [1 u. [% T BufferSize = 1;7 N- q2 D1 e% Z- g: P$ K
do
+ O5 p3 g3 |7 T, {3 x% l {
, @3 Y6 Y1 g% H. X- U    if (Handle)) j: V2 G/ p1 {
   {+ w3 b5 Z) P% ?) F! ^. f2 q
      FreePool (Handle);+ ]# e4 G7 z: @& @3 y! \  @
   }6 _) w2 v7 _& y- h! b  f; x
& J; H) B4 x  W6 h6 a
   Handle = AllocatePool (BufferSize);! x2 q& g& n" G6 x, X4 G
   Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
5 C( v& R8 I; C. J% t# K$ p3 d# g' _, `( C9 |3 {- A4 e0 a' J
} while (Status == EFI_BUFFER_TOO_SMALL);
0 P: ^$ H; ~+ P# h8 q ASSERT (!EFI_ERROR(Status));
) Q$ Y  c. N, J0 q! s7 D, L6 H
% W- E# U& X2 E0 l6 G9 H NoHandles  = BufferSize / sizeof(EFI_HANDLE);) l( a4 I! C) ]% D- N1 }
Index = 0;* k4 X0 A0 k! D3 c9 L% P
ST->ConOut->ClearScreen (ST->ConOut);/ V, W( A. H& l; T6 O2 ]2 w
Result = Not_Found_Me; * c8 p6 [) K+ ~3 c4 [/ b
do$ U6 @8 |0 Q2 x! p  P5 M
{ 4 n4 o8 |/ ~) E3 l& o
if(Index == NoHandles - 1)' w, q. K5 c( Z4 Z+ P
      Index=0;( [- r) A  B6 F% ?  l
      else
, R* z3 _/ M$ u+ E2 h( Y7 O2 E4 O3 ] Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);. k2 \+ w- F& m7 u( F
Print(L"THIS IS DEVICE %d \n",Index);$ u. `8 X( N8 @: `5 ?4 C7 ?
if (!EFI_ERROR(Status))- I6 F$ w- t3 K1 o9 p! h
{) X, ?4 Z2 f- B% Q: q* W& F
   Result = AccessPciConfig(PCI_DEVICE_TARGET);# W2 I$ v( s' c% d% c+ |
//-    Print(L"Again now Result is %x \n",Result);    " N- [1 n( K+ L
}
$ Q8 w. H; T2 b7 ~' [& E0 b/ P& D Index++; 2 b& ^( p# H4 {# ~2 R0 ^5 C/ i1 w
} while(Result == Not_Found_Me);
+ j) i3 {5 f; a7 z4 D6 C / r2 H/ m1 g9 q' A
Print(L"Please hit any key to exit this image.");
5 D; X$ J: r' ~2 J5 s, ] WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
4 n. r' t. m% ~
  K8 a' q, v; P return EFI_SUCCESS;
4 \' V4 f: L9 O/ Y! r) m3 y}

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

感激不尽