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

kaurohy

/*
* myfileio.c
* Apps
0 k0 ~! I* g" k' W* J. h */
# ^  }+ S+ C1 T
2 V9 t  a1 n4 h" W- j+ L8 j( J#include "efi.h"
! j$ h2 O+ l  w! E#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;
6 v: t% X4 M9 H    EFI_INPUT_KEY       key;
1 ~! y' c9 m! g) f* d. B    UINTN               index;
   
$ O7 t3 a: d0 I$ W0 F$ ~    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
, |8 {& Y/ R4 j, t& y* K1 H( ]- b( o            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
" `4 |+ ?* g; Z! O' u+ E, p* O        else
            break;
    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
, f% @& w0 S$ M. T# O    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;
) E4 s  {  c. m9 w. b; o    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
4 G7 T8 {# ~( E/ |# l2 Z6 i    EFI_FILE                *FileHandle04;
# m2 l% s) y2 G% E    EFI_FILE                *FileHandle05;
6 \4 s7 q$ z6 I    CHAR16                *FileName;
6 b1 P6 i% Z) }5 h. u% n+ }        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       
) V  T$ N, U8 r
5 C" l1 h; p# W9 {0 P. S2 g# O. B    FileName = L"NewFile.txt";
+ {( g) k" A6 j( s7 r" D        Space_Key = L" ";
       
        BufferA = L"ABCD";
        BufferB = L"EFGH";
        BufferC = L"IJKL";
+ B5 l& e* O2 Y6 U9 D# u; o
; I# L' E) M  g' c( k3 C: K        Bit0 = 0xff;
        Bit1 = 0xfe;
       
6 [. z/ A4 c  D5 i        Key_Space = 0x0020;
       
    InitializeLib (ImageHandle, SystemTable);       

    DestAddr = AllocatePool (BufferSize);       

    Print (L"Value of Bit0 is %x\n", Bit0);
& o7 j+ D! e5 m4 {/ J    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);
. k+ F1 _- e1 m        return EFI_LOAD_ERROR;
    }
' \% I8 L9 X4 Z; Q, ~/ z, E$ v& u) i
    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);
3 g# _0 F- L& H9 \            return EFI_LOAD_ERROR;
        }       
       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
: G9 r9 [( J7 w* M  T$ c% F6 y    if (EFI_ERROR(Status)) {
$ X. Y  I" I7 p& n            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
  ~3 N' Z  i2 |/ p8 x! }        }
% R# H+ O9 j# W9 P) u* x+ y
    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
* Y+ |$ h0 @0 z$ L6 d        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
, f# D8 q# q$ i& F/ f        return EFI_LOAD_ERROR;
    }               
+ N+ _9 H5 V6 G               
8 y3 z$ d" J5 Y    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
# @7 M9 i  B' \, v: {7 l    if (EFI_ERROR(Status)) {
        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
5 E% X) g- I# p; k; P    }
6 p5 H) c7 p) ?/ o5 t       
: ]* B8 s6 ^, z. }% ?! T    Status = FileHandle05->SetPosition(FileHandle05, 0);       
1 [# h# _3 h. }( {; m4 v) A0 F0 U    if (EFI_ERROR(Status)) {
, [5 |- ~1 x5 O. q; ?. U7 h* \3 [( p        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
& M" ?3 c( u. G; U- ^4 S3 ?7 o    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
) R) H  |' [3 b1 |        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
% F# r8 j) s: B5 Q) D* v" m        return EFI_SUCCESS;
    }       

: Q/ M! O& t8 t0 V; Y5 d    Status = FileHandle05->SetPosition(FileHandle05, 1);       
    if (EFI_ERROR(Status)) {
7 a! z4 e. G1 y- Q        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
$ [5 a/ q/ H  B2 q, ?+ Z    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
) G3 l9 O+ r- N* y! T        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);       
1 K6 N7 r3 h+ b5 M    if (EFI_ERROR(Status)) {
. J# M2 u4 F8 |8 r        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
; c+ a4 w: G+ e0 ~8 P, s        return EFI_SUCCESS;
- B- Y7 X8 K9 w6 a, v- W4 d* n  K6 E    }
) E* ~9 h+ s2 E+ \. b8 R8 b$ @4 r9 Z       
0 n% ^+ D0 t4 f  U' S3 a    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
, N3 b* Z4 W$ ^' H* h- i        Print (L"File Buffersize is %x\n\n", BufferSize);
3 Y8 W2 o: v/ [3 u& t* o    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
* u7 A* H9 Y* G- G0 E) |    }       
4 q8 C6 w; O# W% E8 P% j1 j% J3 k
1 j# h' y* B/ v+ M! m//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
0 d, j1 t! s. v1 W. N* p    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
) T7 z; ~4 J' i% U    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;
/ p+ {$ k  {" k- p( C% \    }
% L& f9 B& `4 u& R
6 e, v- d' ^3 l3 V    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
; I# A1 Y8 c$ a( e: X    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
& B$ ^6 ^' F. D! N( u. O  F    }       
. y+ o. g% W2 {6 Q( R        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
" f* ?" C: u1 i$ M/ ~//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

    Status = FileHandle05->SetPosition(FileHandle05, 10);       
    if (EFI_ERROR(Status)) {
, S! M  p9 w1 t        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
( }% c2 I5 |9 e" u4 Q- C       
* w1 E* o1 T3 j& D9 b: B+ Q    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
8 L1 B# |# W8 w: k, @( b+ A        return EFI_SUCCESS;
7 H8 j1 d" K% ]" `( M4 y    }
5 m* ~" i. r+ L& Y# B% N5 Q$ k6 K
4 Z# @8 t8 U1 [& D' k7 S    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
2 Y. _* B4 z9 [) V        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
6 f" j4 i. m2 u& ^- M        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
4 h( X- b. |6 z; s    if (EFI_ERROR(Status)) {
3 H" C% w* M6 z8 O8 L3 q        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
% {* s& F2 o: S        return EFI_SUCCESS;
. B6 Q/ w( R5 b' j    }
4 J2 Y( K$ c/ [
7 k; ?6 n+ t5 v3 J; H! {( c2 }1 _    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);
; l7 }) S" `; i$ o        return EFI_SUCCESS;
    }
       
8 J0 D; s( h3 K& J' k; L. l1 o    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
' P  |& B* r5 T% [  R) t/ G( s    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
/ b7 Z& j! ^$ M2 R$ S0 J        return EFI_SUCCESS;
    }

    Status = FileHandle05->SetPosition(FileHandle05, 16);       
5 s8 Y( O7 d5 o4 V& d    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
( r4 L, B7 x' P' O! S; X: f' I    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
0 ?3 |) ~" r7 O0 Z3 `. Y$ e    if (EFI_ERROR(Status)) {
2 ~9 O7 E. i$ A$ E( g" G        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
( L) }3 g+ e6 P' T2 W% n; F$ T        return EFI_SUCCESS;
6 s2 y0 I& {1 K$ t+ f* ?. U    }       
" u7 D0 P0 t. E- E, @7 d0 |
//---------------------------------------------------------------------------------------------------------------------------------------------------------------

    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
/ F+ W- p% L$ k6 M; V; G+ a$ r        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);
5 H  b* X) @, P  C0 X    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
  S2 I( w; E1 p' h  D* h/ {        return EFI_SUCCESS;
    }       
1 [5 p( U2 G5 x5 ]- \0 [+ L( `       
    FreePool(DestAddr);       

; z; A, H/ m5 g1 X5 K    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
6 m0 K( @7 S, U0 P( Z3 r% u& ?! z8 I    }
' q5 B& s  ^2 n/ ~2 f! s' r6 O       
# U3 T" h: m1 f8 Y    Print(L"File Name = %s has already been created.\n", FileName);

1 {9 W3 [- A2 x. L8 q1 Y% A+ x) S    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
' ~" T  W* ^; V/ a9 [    WaitForKeyOrReset();

$ m* M( k) I$ [& ]    return Status;
  J; m+ |0 L, Y}

chenhy

//**********************************************************************
& r3 u0 @# I! e0 J8 p//**********************************************************************
1 I3 B* u' }$ z/ ]* U//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
, f  p0 I8 M4 C+ y//**********************************************************************

#include "efi.h"
8 u6 U) z- |  o4 y#include "efilib.h"
) x$ _" K9 ~* D5 t+ I4 E7 p
! u* r, L8 Z4 y# d, U# w' O#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001

static EFI_STATUS WaitForKeyOrReset(VOID)
/ V. @- T% I; g& _' l8 [{
    EFI_STATUS          Status;
1 s- o+ @6 Y2 _: }/ }" J    EFI_INPUT_KEY       key;
2 `4 C5 a* o! J* {    UINTN               index;
% F, b5 t3 I9 |  ^! R   
  g" c- [) e' {# P, D  c% {    for(;;) {
$ ]( q4 K0 z& ?/ M! q9 `        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
- j: ]/ I* o9 R+ Q) O8 Z' k* r* a        else
$ Q8 B$ m; Z/ C: z$ |            break;
0 s0 u  P) ]: v! k9 x    }
. V7 e. Q2 M9 O: k! ^    if (!EFI_ERROR(Status)) {
8 o3 h5 u! R1 m        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
3 g* g; @, T8 w! N    }
" U4 O8 f$ J8 h+ l   
    return Status;
}

UINTN
' ?. e( H7 {- d) _% F9 Z+ vAccessPciConfig (
* M. G" L* P2 y1 u5 E. M    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
/ F8 S# N7 O8 n" ]9 A" \5 m. y; _8 ?0 G6 s    )
{
5 {  t( f" L, T        UINT8  i;
        UINTN  PCI_ATTR[4];
- o( b) S. i( |$ d* b, o        UINT8  PCI_REG[4], pci_offset=0;
+ B0 b8 \# Q+ e" _# t& g
+ o) ^' Q# O6 d& Z, G. ?        //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)){
. I$ O1 P" X/ n0 I4 R, \8 _2 W          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));       
  `% e5 O" V+ l          //print register value
) A7 V: |; z- V          for (i = 0;i < 4;i++)
          {
/ I  A7 @- Z1 i  w- k4 C        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
8 C1 K+ d. _+ b. K          }
          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
! t& ^4 k& C' r) O" g% L    }

        return Not_Found_Me;
5 H) t0 I: ?7 s# |; s( M6 R}
$ S$ l& R  W; J. e$ O
- j. T2 n, c2 N0 KEFI_STATUS
2 }8 Y3 k& S+ k) d- ^- eInitializePciApplication (
5 F" U* v# h/ u, d) X7 Q3 t( M    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
    )
) E+ n1 p1 N6 K7 s2 ^7 v) h{
  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
/ |$ |; C2 c- u; O2 c  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

    //
    // Initialize the Library.
    //
    InitializeLib (ImageHandle, SystemTable);
' ~# @$ {+ a1 f1 L+ G1 w7 |    //
. v! ~( N7 c2 ~    // Find the PCI driver
3 O0 [1 R1 v% L' o: T# c    //
( y& v2 r$ T; y, F    Handle = NULL;
    BufferSize = 1;
    do
+ A  J6 |  O0 [; b9 z7 w    {
      if (Handle)
      {
$ ]* L( q' a# I. E        FreePool (Handle);
( p. {5 N1 L$ I0 k      }
               
! q3 C- n2 [' j5 k& P; R      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
" c/ G' K& U( R* q3 ?
6 S, L5 j7 M; c9 w5 j! Z4 W    } while (Status == EFI_BUFFER_TOO_SMALL);
7 K% J! s' W$ O0 F' x4 @    ASSERT (!EFI_ERROR(Status));
: t" y8 F4 P! _
+ H6 t: x- ?+ z/ t* j    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
5 M9 {! O8 y2 [; g2 [* j% O    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
0 C$ n2 D( J* Z% a    Result = Not_Found_Me;       
& B3 ?4 c- F0 q- z0 j# h2 j$ J    do
4 V+ Q) y* a+ v' R+ a% x5 p4 ~1 C    {       
        if(Index == NoHandles - 1)
2 I& A' `1 ~6 Z         Index=0;
; Q  f# e) O1 `) |) l% V6 ]& _          else
6 v; P, u* S& P* ?; H    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
- _8 C6 v+ {% n% J, z        Print(L"THIS IS DEVICE %d \n",Index);
6 L# R' H: r9 I& Q( H) f    if (!EFI_ERROR(Status))
    {
( ], y" W) t! _! n# ?          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
6 j0 s, v3 A* c, A; n    }
    Index++;               
    } while(Result == Not_Found_Me);
       
% L& R" `$ ^5 |. p8 k5 ]* j        Print(L"Please hit any key to exit this image.");       
: M9 `( s! B9 b' a; f    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
; j! ?: J8 \( a: y# G& A7 S               
7 i2 n. r" Q, |% }    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习