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

kaurohy

/*
7 |! K  \- l& Q  p/ d; \* z2 D3 a$ Z * myfileio.c
4 n7 L  Z1 S4 \7 ^# {4 R * Apps
*/
( h* I9 [6 r+ i0 A6 W0 b
#include "efi.h"
3 X, C/ D! L' A5 x#include "efilib.h"
1 {+ d0 \" x) v& P# n( `9 R
- s* Y3 H: Z  l1 b#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

. @8 p$ }; J# U: v, nstatic EFI_STATUS WaitForKeyOrReset(VOID)
% m( r+ b+ [* Y7 A: T9 U{
, Z; Q; k6 C7 p    EFI_STATUS          Status;
( K5 o0 H: p0 e    EFI_INPUT_KEY       key;
    UINTN               index;
& f( J5 i! G' v' D3 p' ~6 f0 W   
    for(;;) {
- d  y5 z6 j! g5 O        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
. ^4 {- G; c& k        if (Status == EFI_NOT_READY)
7 m* _+ c- G3 S& ~0 D$ k            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
0 w6 J  z; W, [9 E    }
9 c% D8 e$ L2 Q; r/ Q$ H8 [    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
( U+ C! r1 O" d* H8 b    }
/ ~/ y$ u5 K2 m/ D: G# s5 R   
- ?4 K. d, W* X& f; V    return Status;
. ^+ S, I( A7 y/ E# a8 v; M& V/ E}
7 E! V5 g9 C$ g8 V# b
EFI_STATUS
EFIAPI
( X% p  J% D/ _) E1 V5 T5 z- BMyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
{
    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
2 F3 d6 i2 N4 F    EFI_FILE_IO_INTERFACE        *FileHandle03;
2 ]/ C# f0 n( B, g, u; R# [+ ?  \# j    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
: c) R) i% g: H    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
5 d1 d! X% K. o) ]+ }    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       
: r2 `5 L$ a, f
+ Y  N5 o7 c+ a    FileName = L"NewFile.txt";
0 \0 T& q( A# }9 e2 T        Space_Key = L" ";
       
0 G' _% R2 N* [0 u! y% F) r1 N        BufferA = L"ABCD";
        BufferB = L"EFGH";
        BufferC = L"IJKL";
" B9 E& |) @, v
        Bit0 = 0xff;
        Bit1 = 0xfe;
       
, \$ X9 K* A  |" {        Key_Space = 0x0020;
       
    InitializeLib (ImageHandle, SystemTable);       

) z* d! f) M1 H2 |# m' v- z    DestAddr = AllocatePool (BufferSize);       

  O" C8 T/ `9 S" v4 k% V8 {$ n    Print (L"Value of Bit0 is %x\n", Bit0);
. U1 P7 z, K1 k8 K% e! x    Print (L"Value of Bit1 is %x\n", Bit1);       
1 W; @! T7 c& r+ ]  k' y
3 ^+ L" ^( t" ?$ B, B6 P$ f. V       
" `' }' [5 Y0 L7 H% m1 n    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
6 E, `' ], X% _: T( a        return EFI_LOAD_ERROR;
; k* B/ J8 R9 w    }
7 F/ L0 s. F, o- |+ d
    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);
% v8 v: u. w4 y" D4 _8 p            return EFI_LOAD_ERROR;
        }       
       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
- x' }- O6 `/ ?    if (EFI_ERROR(Status)) {
8 C! v1 R; ]- T            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
" i, l9 \9 ^/ k/ E, y            return EFI_LOAD_ERROR;
        }

: j* C- _  h& B    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
6 ^- G7 W% s3 ?" J    if (EFI_ERROR(Status)) {
6 C. B+ s4 n; v$ Q9 z0 {        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
" I# h& [6 Z4 k7 q        return EFI_LOAD_ERROR;
) b" `+ X, W* ]    }               
               
% j  j1 M# K3 s/ m    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
+ g0 b" E" Z+ q) V+ x  l0 G    if (EFI_ERROR(Status)) {
        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
/ U) Y# D/ N+ f' n    }
       
, @; y+ e/ t+ a% L" `, e1 L' t    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;
2 l0 ]8 g& u/ A% t$ L    }
+ T' q  j" c& u8 O$ n" }" |/ s       
) P( d2 X/ v6 Q% t$ i# a+ p    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
$ ^+ a6 n. d; i$ C7 h$ I' ^    if (EFI_ERROR(Status)) {
# ]- S* h1 `7 B: a        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
: I) q, z2 W2 p& X3 u& g9 _5 }9 E        return EFI_SUCCESS;
* r. i3 W7 d$ `+ d+ u2 O    }       
+ F: A% H. k0 [5 V( ^
2 K2 P1 ^4 {5 \# e    Status = FileHandle05->SetPosition(FileHandle05, 1);       
: ^$ b. J6 o5 }: e) \3 C    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);
2 P5 s4 D* @$ k" `' ]4 h        Print (L"File Buffersize is %x\n\n", BufferSize);
# p) V# u9 U/ k1 {( V) j    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
7 G4 |* y9 L6 D$ u+ t        return EFI_SUCCESS;
3 ]6 u% _" A7 T+ t" G1 o- b    }               
       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
, Y7 M& ?- R4 C* Y/ d6 c9 J    if (EFI_ERROR(Status)) {
) G' x+ S, a$ D/ h& D  v, \        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
# B& T$ T  k: e+ t1 q        return EFI_SUCCESS;
' |: \0 s! J& k8 P" u# X, l    }
  C' F) a; h; {, X3 I       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
/ A8 {9 R& P8 i- }        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       
# L9 u1 z! K. m( l5 ?
//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
, @: F! t  T* g/ \7 f1 B7 f    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
0 c. ?0 v( J, \7 x+ W5 _2 ]6 Q    Status = FileHandle05->SetPosition(FileHandle05, 2);       
. e' y% G# H1 |- n: [    if (EFI_ERROR(Status)) {
4 o5 o% Q9 q0 R) P4 F        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
; ]" Z* |8 \: I8 \/ ^    }
6 W, k. D7 z+ A) g! A
    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
; c0 D( J2 C6 r+ c    if (EFI_ERROR(Status)) {
& u' K+ u* w6 N1 A( @: g        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
( P8 A" Y  O  Z# c- F0 E  h* Y        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
' d) D$ j9 X  o# j: H% F* W- F//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

; o7 C0 }( r! a0 G% |    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);
! ^9 k% w6 d/ U5 `' n* F& f3 }( B; Y0 R        return EFI_SUCCESS;
9 I; m. \9 K! b" k) n' h! B    }
       
8 H9 g: F8 r& _& n    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
% @0 M$ D/ }! O        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
) C* |4 R7 r; v( X" e$ t        return EFI_SUCCESS;
    }

    Status = FileHandle05->SetPosition(FileHandle05, 12);       
7 p+ D4 d0 t) K$ C8 \: g1 Q    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
5 _/ j* |0 ]0 W3 I! g        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
# c+ _4 |3 A8 u+ q        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
) ~! N" n$ d% z" i        return EFI_SUCCESS;
! U  }' S2 o% {7 f. ]    }
0 |' m- q' K  z& D9 g7 X/ Q
    Status = FileHandle05->SetPosition(FileHandle05, 14);       
    if (EFI_ERROR(Status)) {
4 E2 O4 _. |  t- K, p. d# O1 J        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
" Q6 p  G; ], h# B    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
& b% j- ~' d# z% \    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
+ @- _% a4 [5 ~        return EFI_SUCCESS;
    }

    Status = FileHandle05->SetPosition(FileHandle05, 16);       
* Q4 f# ^" c( p# k    if (EFI_ERROR(Status)) {
, G; ~: Z/ o; T: a        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
* @0 B0 ?. `2 @+ N2 L        return EFI_SUCCESS;
    }
       
! e" t1 \) K+ _  U( f- N1 l. P( z    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
6 Y8 @2 t: z* m$ y8 E, u    if (EFI_ERROR(Status)) {
, j$ `$ q" s! ]- J! A0 `        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
. }; _& ^3 f8 I- _# |8 [3 P3 U        return EFI_SUCCESS;
    }       
8 ~  Q7 k2 O# v6 \( @% C* ~
//---------------------------------------------------------------------------------------------------------------------------------------------------------------

    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);
5 I4 ]# @* Q7 T' o        return EFI_SUCCESS;
    }
3 Z% ?. ^  Z7 j       
7 v! u& K% V$ l$ `    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
/ r1 F& |1 O& J3 L' T% [    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
& @  X) v  w4 Y: F0 E! P" s1 m/ S        return EFI_SUCCESS;
3 I+ B4 J6 N7 J4 w/ }* I# e    }       
. Y( H, _& V5 k) m1 k$ e       
* r, Z: W) o3 }( b2 C    FreePool(DestAddr);       
. K4 j4 I2 P& \) u1 h" m! p! Y
    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
# q2 |: I, E7 w- }2 c0 s; B* D        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
3 a; Y  t, |9 M1 n3 ^1 q2 i) P, J    Print(L"File Name = %s has already been created.\n", FileName);
) k! |4 S$ B) C' f
    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
) _3 R4 W) V) h' m9 y    WaitForKeyOrReset();

) f1 C5 W5 B0 O: a7 E3 \    return Status;
- c9 C' T4 S$ i1 i* r  i/ R& @}

chenhy

//**********************************************************************
7 y% |8 Q( J2 a//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
* Z( [  c, g5 ?1 f# h) a( S//** By Lawrence 2009/11/20 @Taiwan**                          
1 L% y( @5 a8 e- G" T/ Q//**********************************************************************
8 o* p  ?+ n8 s! [7 p//**********************************************************************

#include "efi.h"
9 d: Z! L  f5 j2 A; s% }. b5 W$ ]#include "efilib.h"
) ]0 U" }  r+ F9 z6 B: l3 d" d6 l
4 V; K1 }2 K- h, W, B9 I#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001
- v# k2 p# F# v, a$ W/ U
static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
, m& D  a3 S+ m" p    EFI_INPUT_KEY       key;
    UINTN               index;
* z) C; e8 i- G  \; }/ r9 V1 m   
" D# {6 z8 E2 w    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
) r+ ~, @: f' J2 x/ t. S, U            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
5 r4 }" w" l1 |% {, Q8 H1 P        else
            break;
* N' ]* `, [3 w4 t2 A    }
    if (!EFI_ERROR(Status)) {
* U% J2 {: l2 v% ~2 a        if (key.ScanCode == SCAN_ESC)
4 [* @5 S- U2 K. g( y; L6 [            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
    return Status;
}

UINTN
  [, D. ~7 [' u2 n( H7 K( @AccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
{
) f1 i) F( g" D$ ]8 Y2 {5 C9 Q        UINT8  i;
        UINTN  PCI_ATTR[4];
+ _# ]6 h6 G9 g4 }3 D        UINT8  PCI_REG[4], pci_offset=0;
$ t) u, R7 }; Q9 R* J5 ^
8 m- f( m7 ]) {) k9 P4 @& J        //get device attr
  s* K9 P$ b* Q* Y" a0 @        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++)
          {
* Y2 d+ A* V" @( D+ t: k        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
$ _' o' e1 T. }/ R; \; @7 L                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
6 S! G9 r, s; \" q          }
) m; m; m6 \$ f8 {6 r          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
    }

        return Not_Found_Me;
}

% B+ V) c% e/ _, u* c5 H& eEFI_STATUS
" P8 K: q7 }6 |) {InitializePciApplication (
4 Z0 e( O! O) `    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
$ y$ Q0 o% k2 p6 }* N$ Q; l% K4 Y    )
{
  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
0 `4 J5 d7 x+ s: r$ e$ T  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

' D  l; M4 l5 ^# x* B    //
    // Initialize the Library.
1 j' P- u; I7 L6 ?3 B    //
) j1 G1 q* h1 U1 E8 C$ o2 j    InitializeLib (ImageHandle, SystemTable);
  Z  y, ]; b: |) U' z3 o5 U$ j) q5 y    //
  h- n6 ?' ], X/ l2 x+ ~! g    // Find the PCI driver
; N& ~$ H- ]* T9 ~9 v    //
# l1 b# w) m( ]9 D    Handle = NULL;
4 ]1 o6 L% O+ h9 B3 [/ J    BufferSize = 1;
    do
  d8 t, M, V/ v    {
; u0 l$ T- Z# f6 e; v0 i8 @+ L      if (Handle)
& N* i( ]+ b& X) o( n+ l      {
        FreePool (Handle);
" L4 i6 Y0 ~  p7 i      }
               
+ s! E/ ~  k, z) k6 E5 W3 O3 k! X6 O      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));
) M2 z# y# o1 g6 R/ V* h: H
/ t) o& d6 T* i# T  S    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
4 E3 Y8 G# b/ F& a8 ]/ \    Result = Not_Found_Me;       
    do
2 O: d9 ^" D! Y: H: K7 w, }    {       
        if(Index == NoHandles - 1)
  w/ b( T6 N+ n$ U( Y5 b         Index=0;
, Q. t7 J- k/ ], J# ^' d  l          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
) f+ l+ A' j. e) Z        Print(L"THIS IS DEVICE %d \n",Index);
) @! @- V$ b( r. ]& G; g3 Y    if (!EFI_ERROR(Status))
    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
- n+ i( Q, z7 c% N//-          Print(L"Again now Result is %x \n",Result);          
    }
    Index++;               
0 N* Y) W7 T7 c8 I+ L3 A" d    } while(Result == Not_Found_Me);
       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
5 b1 |7 y- ?  N- q3 s5 O/ A               
( }9 o7 [4 l1 s) v6 R) L% S1 J    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习