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

kaurohy

/*
* myfileio.c
4 u; y! q* k4 w, a. ^( C * Apps
*/
, t; W; v3 Z- x& I7 b. @  V( d
) C  M* b: ~! G9 Q3 D#include "efi.h"
0 C. ?/ g7 ?0 G#include "efilib.h"
, V7 g' A- X+ n: K% J- n6 _; n4 P+ C
# T& g# b8 c% |8 \% E( D; y; i: V9 ]#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
6 G( W# ^5 L. B9 \/ S! B8 v- n
8 h0 {  B0 P( pstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
/ a# S+ ]' G1 O+ n+ g( z+ k    EFI_STATUS          Status;
7 y8 i/ @( U% R& @    EFI_INPUT_KEY       key;
    UINTN               index;
   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
2 ~! ~# R2 h& O4 I  r& G        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
; W2 y; P6 l: S6 E8 r" K8 z        else
            break;
    }
    if (!EFI_ERROR(Status)) {
7 A5 ^+ V2 l. w/ F6 S/ W        if (key.ScanCode == SCAN_ESC)
1 C4 X# E# Z+ b) p: C# H; |) n! k) H) L0 S            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
( F" y+ R7 {3 y  u* z* b8 [   
    return Status;
  B$ J7 A; \: J( W8 ]( S2 v2 u}
3 h  e& q0 |4 @" w+ U
% G* l! C' ?  @8 N; CEFI_STATUS
3 ~! e% k+ H, s: q1 n% SEFIAPI
  {) v) S3 h. c( p# S. k) PMyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
( q' Z# F4 T5 g$ C% D* [# P{
    EFI_STATUS                Status;
% F, F/ X( m/ T, {0 R, I! c! k' B    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
# A; F. A8 @8 f7 I& O    EFI_DEVICE_PATH        *FileHandle02;       
- {( {8 H% f. \' u1 ^6 ?1 z! x% {    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
" ?* H  ?- U$ `! S3 h+ j    EFI_FILE                *FileHandle05;
  F/ }- Q+ V  I. B) L4 a1 A    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
5 c- q) ^) V$ I$ L" w, F    int         Bit0, Bit1, Key_Space;       

    FileName = L"NewFile.txt";
        Space_Key = L" ";
* L1 R- Y4 e- F       
        BufferA = L"ABCD";
% ~; Y( Z/ D" Y        BufferB = L"EFGH";
- E- p/ e* m9 L- O5 r$ X        BufferC = L"IJKL";
5 |) t( n0 I* |
        Bit0 = 0xff;
        Bit1 = 0xfe;
: e3 O  u$ R1 H2 X' ]       
! Q  M1 ~) L3 X, _        Key_Space = 0x0020;
- [; S  o' S, ^$ ^1 ^       
    InitializeLib (ImageHandle, SystemTable);       

; |* Z" ?, k  B' D    DestAddr = AllocatePool (BufferSize);       
7 G: @) F- A3 D. V3 c
    Print (L"Value of Bit0 is %x\n", Bit0);
    Print (L"Value of Bit1 is %x\n", Bit1);       

, y3 D/ j5 i. s' D: N) D+ {5 C       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
4 i9 r  G% u' h5 W1 i: R1 ^    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
# ~" i8 @9 \/ a        return EFI_LOAD_ERROR;
/ U/ }! @" e3 @" l2 q5 z! P    }
9 L1 @* G8 h8 X9 N2 W+ q; {
3 `$ h! N6 K% F" P    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);
. s" s; y# r2 E            return EFI_LOAD_ERROR;
        }       
       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
% r1 S! f) K. v! c: L& Z            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
        }

# Z  W! ?4 |2 ]9 K) q' G7 T    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
; b' {! q, K+ @8 b    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }               
               
. I& ~4 J" @( {6 t/ I1 m: [4 @0 n    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
& _0 d# [* C0 |2 z# z    if (EFI_ERROR(Status)) {
1 W, o# @, a) [" @) h        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
: e& r' a0 w. j& t        return EFI_LOAD_ERROR;
  _- ^3 k8 s1 i% T7 k& ~    }
8 [$ r! j, B0 {3 a5 I' }; O       
2 Z+ U9 y7 X  @: f; a3 L; q# F    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;
    }
       
0 Y2 ~8 k  u. n+ E5 c    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
( Q" D! _- J" j  f( R    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
7 A5 ~$ N# o' p* [$ V4 T- A    }       

  _. w0 z# r) u0 N    Status = FileHandle05->SetPosition(FileHandle05, 1);       
% b7 h- Z4 y) g: A    if (EFI_ERROR(Status)) {
& v/ z$ K" t7 S# f& w        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
# Z  G* C. g; i" [- c( L5 l1 [8 M        return EFI_SUCCESS;
    }
       
" e3 ?1 [& t! ~$ S9 o, D, G    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
# d3 a6 x  H$ H" L        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
% N9 p2 ?. i0 j6 J5 `        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
7 d# r5 T! X$ S$ X$ G        return EFI_SUCCESS;
    }               
& B* b/ Y: ?$ b# U       
7 [: G6 h9 z+ i% C4 F! A    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
  Y! M% B4 \  v/ d* V* a        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
/ |* H- H) y0 |9 @       
1 ^5 c& Q; h+ Z6 G0 b) P) F    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
6 E) `5 o% {; Q7 T0 \: n, z        Print (L"File Buffersize is %x\n\n", BufferSize);
3 c3 t9 V$ }/ d5 [    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
8 K4 C. [: [! k        return EFI_SUCCESS;
    }       

//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
3 m- ]8 H+ V( ?* k# `$ p! R! o    Status = FileHandle05->SetPosition(FileHandle05, 2);       
2 ]/ E- i4 d# l: B" r) I( }    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
' |7 Y% @4 f4 V" O1 U+ C" T        return EFI_SUCCESS;
    }
+ y" i7 o' k' {0 S
    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
+ ~9 k& T: `. M    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
! l0 h$ Y- R+ E9 E        return EFI_SUCCESS;
    }       
' M/ ~8 h7 j3 T. x. E0 l* y" r        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
4 c0 B- b: j* I( g; W
    Status = FileHandle05->SetPosition(FileHandle05, 10);       
3 r. m( ]( j5 {9 K    if (EFI_ERROR(Status)) {
5 ^6 g; w" K; k+ X2 L        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
% }; H' c: U0 L  o& ]        return EFI_SUCCESS;
' U1 T& j5 J* c2 B    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
! }* d# E. Z/ j" v- q# A    if (EFI_ERROR(Status)) {
6 c4 A0 Q) Q% b5 d* \6 d        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
! s' O; W7 u! h! t  w" G        return EFI_SUCCESS;
" z( D3 `6 q4 j& C  `    }
) \9 d4 A* C" M& [
& m, i" _  `( X$ G7 t0 K! H. h    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
4 `, M6 `' H4 C" a7 J        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
. q. U& K$ U3 g0 m. k+ e  F        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);
/ K1 P. M/ P4 M* d0 s0 J        return EFI_SUCCESS;
    }

9 H3 `1 I, b7 ?0 ]' x    Status = FileHandle05->SetPosition(FileHandle05, 14);       
    if (EFI_ERROR(Status)) {
# D7 a  e+ I8 {        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
6 l- L* Z0 s3 e        return EFI_SUCCESS;
    }
7 k7 p8 n8 N% H! q- U       
) z' h8 H6 E5 R9 R    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
+ _! i+ ^' i7 {+ V* R        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
0 J. ^* E$ C2 f# E& N        return EFI_SUCCESS;
    }

    Status = FileHandle05->SetPosition(FileHandle05, 16);       
    if (EFI_ERROR(Status)) {
3 P. T; |6 A& }/ [! e! ^" q        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
6 |) N( U$ ~/ ?8 F  ?* I( J1 `    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
9 }+ p' ~" g2 H5 ~% e# Y5 q    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       

7 z3 T$ \( J/ _, M//---------------------------------------------------------------------------------------------------------------------------------------------------------------

+ A$ D! b% r( s& K# F& E$ e3 @" f    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;
* i. [( Q( L& B    }
* B! b1 e0 w% m       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
+ N' s- X  q+ a" }9 {    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
5 [" n7 S/ y+ }2 v4 I# z& V        return EFI_SUCCESS;
6 P) s" g& a. C2 ~    }       
       
/ ?* m: C* f& x' @8 V' c    FreePool(DestAddr);       
( w- q9 S9 U2 m2 y6 s; Y
) P- D" b5 M" o    Status = FileHandle05->Close(FileHandle05);
6 _9 {$ F. N( C& E    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
' b) l: \6 P% K, G, u        return EFI_SUCCESS;
3 b+ q( Y. O' g  b0 Z0 p    }
       
    Print(L"File Name = %s has already been created.\n", FileName);
/ K, b3 D! F5 R7 h
3 w9 [; v$ Y. d    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();
5 Q0 _  A) E! j
    return Status;
. q: t. N& L  ~+ {, T) \: V# c}

chenhy

//**********************************************************************
7 q1 J  E0 Z  X//**********************************************************************
- X$ D5 E9 m4 _/ H9 A; @//** Demo code for PCI configuration space I/O access program **
0 N# J( B. N# @; A8 g6 p1 ^//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
//**********************************************************************

! C" J; w. n/ l9 f2 u  @; C+ i#include "efi.h"
#include "efilib.h"
% l% u* X, x4 {2 L! I5 O/ V
#define        Not_Found_Me        0x00000000
" N! d* S! d$ s, t. J0 \. x#define        Found_Me        0x00000001
+ o( K6 v) X! O2 x3 M( F4 ?
static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
; ]; T# [% r( j0 G3 B, p2 T    UINTN               index;
8 `' ^$ ?  }# F# P! Y( b6 j+ E   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
9 t* V8 w- l5 L* x        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
  Y1 Y* h2 Z) Z* Q/ Q1 C3 D8 z; X    )
{
) f; b# G7 b: w: o% x        UINT8  i;
. i+ x( A# G" ?        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;

- l2 T( s1 V- M' S; i        //get device attr
        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
, ?' h* @5 Q+ J* S0 _; ?& k    if ((PCI_ATTR[1] == 0x00) && (PCI_ATTR[2] == 0x1e) && (PCI_ATTR[3] == 0x00)){
8 w: E5 e; h/ @+ X8 B          Print (L"Device found at BUS = %02x         DEV= %02x        FUN= %02x  \n",PCI_ATTR[1],PCI_ATTR[2],PCI_ATTR[3]);
4 i6 o- Z) O  J" q: R: }          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
          //print register value
4 f# f. K% b* ]9 h4 W& z! q" r: Y          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);
          }
          Print(L"\n");          
# c9 c& h. _* z5 Y0 t( B4 w3 Z6 l          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
* |2 m. q/ O" [' [7 E          return Found_Me;
    }
- F0 \. f# X% P% k
; X1 k0 c/ q) t- _        return Not_Found_Me;
}

EFI_STATUS
9 B; l! _# M4 d/ M3 ]; S; LInitializePciApplication (
8 m: ^0 v. q( k1 x    IN EFI_HANDLE           ImageHandle,
" m8 [; x( x5 V4 n    IN EFI_SYSTEM_TABLE     *SystemTable
7 ?- }2 P; y' z5 N9 A, g/ |$ f    )
{
9 S2 i6 h- r! R$ a  EFI_STATUS              Status;
$ l  y' R0 |6 o$ f6 Y: t% N8 M" E# M  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
( s+ {8 X8 w$ }" \7 ~! r
0 h1 M" {& t) \( q: _5 Y    //
/ d. l1 Q. e" h3 H6 t    // Initialize the Library.
    //
    InitializeLib (ImageHandle, SystemTable);
, a6 ]; j& L* ]' ~, a( g    //
; U9 p' w! R$ A- q0 \0 P. z6 ]    // Find the PCI driver
    //
' D& p3 \$ S( s  E& E, K    Handle = NULL;
    BufferSize = 1;
3 t; w$ L. v6 _    do
0 ?/ X/ }$ R0 [" b2 N% i7 X) J    {
      if (Handle)
      {
& O! `. f4 U' W/ z& c        FreePool (Handle);
8 y5 d4 J) c) l  T: A; u* _      }
               
      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

1 N# Q0 O0 B/ f5 ^    } while (Status == EFI_BUFFER_TOO_SMALL);
8 h: i( q& _0 F# X4 |    ASSERT (!EFI_ERROR(Status));
8 ?1 l: g" ~8 p
2 Q6 _$ Q% P% {. @8 T    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
) ^& b  t9 Q- ~; g6 N% [    Index = 0;
& K! a0 s- k1 {# P" g3 r2 [* X    ST->ConOut->ClearScreen (ST->ConOut);
2 R) @6 D' u& A/ A. N/ a, K3 d' H( f    Result = Not_Found_Me;       
    do
    {       
        if(Index == NoHandles - 1)
         Index=0;
          else
" f6 `% Y* G% P' M* z; V; u    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
: C( k( |- B. {4 ^& x$ p; a        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
& d. F! B0 ~* n. W5 m    {
+ j# Y6 Z- W/ B- [! V, ?          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
    }
& u3 P% Y& P5 x  F    Index++;               
    } while(Result == Not_Found_Me);
' w7 w! V2 S: I8 _# K- H3 l       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习