[原创]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);
' K2 ^* J, k' C8 [: H# D' N    }
! C1 q" u: g% T% Y* d    Print(L"\n");    ' F6 J4 s& x6 ]
   ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));    : t: T' d- p# ~' H: J
   return Found_Me;# q; a5 ~* a8 a
}& K  F( U7 }+ L6 N

1 `- U$ c: f1 t+ T5 m return Not_Found_Me;
3 T1 |2 u; [$ ^: w; u7 A5 D* G}
0 l0 Q/ O8 D5 o+ I( x( G$ P8 H
7 V( Z& `2 I5 oEFI_STATUS
5 D$ L" ~: V" T% I1 bInitializePciApplication (; L  I& ?7 a9 k) j  u
IN EFI_HANDLE          ImageHandle,/ x) l( t: [  n
IN EFI_SYSTEM_TABLE    *SystemTable2 n* x/ S6 e* @. K8 w* R! E- A
)2 h, L2 Q% U3 ^& c; \' E1 p
{# t4 R% F3 _6 {/ Y. P& P
  EFI_STATUS             Status;
$ k2 J+ \; H5 R0 q+ q# ?, ]4 `" d6 y  EFI_HANDLE             *Handle;! U- x8 I* j0 v
  UINTN                BufferSize, Result;2 q+ V3 h6 k) Z( A4 u
  UINTN                NoHandles, Index;9 O/ A. S- N4 K3 t" I
  EFI_PCI_IO_PROTOCOL    *PCI_DEVICE_TARGET;
/ k! d! d! ?3 t  R+ \- ~  V- F# i9 ], c. R9 B: k- x+ A8 H
//; R! T  Z& n( F
// Initialize the Library." @3 A7 G. a) P
//, F+ u, `; s: f% W7 B
InitializeLib (ImageHandle, SystemTable);* G* C9 k, O# D1 s! f! o4 U+ `
//, V- q, |1 q: p4 M. N0 R) {1 u
// Find the PCI driver( q" I6 J( N5 o6 a8 Y
//2 ?& Q9 C6 L: d" C7 O$ |
Handle = NULL;' f: S& w) x7 J# B
BufferSize = 1;# ~$ g0 Z1 `" m+ t6 i
do
0 C# `* D. a3 a/ |/ D& y {
: c0 `" F) Z! B    if (Handle)0 E, Y/ R. ]; k2 m# w% |1 ?
   {
+ p1 ?2 ?( l* n$ U$ z0 X1 l* h; N       FreePool (Handle);" S. N" J5 @$ ?, t9 ^
   }
3 E+ ^. Z# `" p* a, W8 h3 \/ [
8 m% ^5 e( S! D9 m6 o5 O    Handle = AllocatePool (BufferSize);
# @2 x& o5 [4 R( }: k    Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
8 ]% o- t6 Q/ Z6 j+ z3 l; d( M' A- L5 y
} while (Status == EFI_BUFFER_TOO_SMALL);) B7 H; l5 y' ?# w6 k
ASSERT (!EFI_ERROR(Status));
: p- t; c, b% E$ a& J* v7 q' V; [8 t3 f0 e
NoHandles  = BufferSize / sizeof(EFI_HANDLE);8 [, s( A) x; P: E, ?
Index = 0;, i1 J3 j$ S  {; z/ a; {
ST->ConOut->ClearScreen (ST->ConOut);
- E8 a- \9 y' l2 a3 P& t, J# Q Result = Not_Found_Me;
/ t. N/ w4 t. H3 _ do; l- ?/ `1 ^$ h( n
{ 0 B' J7 V  W5 n5 z( w. t
if(Index == NoHandles - 1)7 e3 n7 i& }4 D
      Index=0;
% t. ^4 ?+ [* B       else2 `" |3 ^0 C( P: K
Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);1 x, h/ [/ ^" s0 s3 s
Print(L"THIS IS DEVICE %d \n",Index);
9 K. q# D, W! x, j9 N if (!EFI_ERROR(Status))+ C( i5 M0 P) i; G  f
{
7 q+ J9 R: L" s. S    Result = AccessPciConfig(PCI_DEVICE_TARGET);  X1 m5 ^: k/ [9 S9 I3 ^
//-    Print(L"Again now Result is %x \n",Result);
/ d" r" G8 y! P }3 ]- o, j) l/ n# d! k
Index++; 1 c/ o( \6 B! q, S
} while(Result == Not_Found_Me);) y0 d! ^. m) h+ _3 \
  Q  U4 e  z. ]8 i8 ^" h
Print(L"Please hit any key to exit this image."); * T  Y. _' ?: c9 M- ~1 C" M
WaitForSingleEvent (ST->ConIn->WaitForKey, 0);7 I. T& `& ?' M+ _

9 \9 p  W: n4 T' z8 s2 a return EFI_SUCCESS;. ~$ |  _. w) J
}

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

感激不尽

浏览过的版块