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

kaurohy

/*
* myfileio.c
& U; S) y# w) P& P. d * Apps
*/
0 @3 b+ r1 i+ q3 m% N7 ]
#include "efi.h"
( g1 }. ^5 a1 Z5 C5 ~#include "efilib.h"
$ r. C7 s- F+ {3 r4 j/ q( i
+ i, y; P8 G6 W+ q+ k#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
4 X- O4 L, t* Z- G' ~" K
static EFI_STATUS WaitForKeyOrReset(VOID)
{
/ x$ J0 U) q3 E    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
4 h' L# L( K4 `, \  j" E& d( P    UINTN               index;
   
    for(;;) {
" J# Q8 [- F  Y, b- M        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
6 C: R2 h/ I. m2 G+ h, l! E        if (Status == EFI_NOT_READY)
9 s( V; [- k- p: L6 Z0 H            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
4 a, D- p! S! t. C            break;
    }
& `5 L1 T8 t% S0 d; l    if (!EFI_ERROR(Status)) {
$ y  _- W: Z% `" Z% |) w        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
" A  V% Z3 N/ I) D! {    }
% i( F% t0 x6 D) l/ N- r0 T! q   
    return Status;
}

EFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
4 D% W4 i5 i. p: ^% t. z5 d{
, i! g6 f# A. S3 y: E. Z    EFI_STATUS                Status;
3 i: N. s. H- L    EFI_HANDLE                *DestAddr;       
$ ?  `/ D( t) C" X    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
7 D- E+ C9 M+ l" w    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
3 I, U+ a5 T5 t4 |: S- o7 W    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       

1 a# ~# y+ w  u/ l    FileName = L"NewFile.txt";
        Space_Key = L" ";
% _0 q: M7 c& V       
& Y, g1 b5 ?9 C        BufferA = L"ABCD";
        BufferB = L"EFGH";
        BufferC = L"IJKL";

        Bit0 = 0xff;
- i  J9 D6 A. _+ C) ]        Bit1 = 0xfe;
       
        Key_Space = 0x0020;
8 |1 u; j: M7 C+ M8 ]2 ]       
- J! K5 ~* l5 ]# b0 S% S9 t) |+ _% T    InitializeLib (ImageHandle, SystemTable);       

    DestAddr = AllocatePool (BufferSize);       

    Print (L"Value of Bit0 is %x\n", Bit0);
    Print (L"Value of Bit1 is %x\n", Bit1);       
& P: W1 ^; @+ P2 y8 D
       
) n' U9 ~5 N5 c: f    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
$ I7 v' n8 {6 [# F, @    if (EFI_ERROR(Status)) {
& z; |4 g$ y3 @. p        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);
% p2 d/ }: N. m/ _    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
! M# I/ q1 P7 A1 S        }       
7 r3 M6 b. E; ^1 M3 G       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
& {  ]6 u) E  N$ T3 F0 c            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
. ~. |& D  A  P  ^( u8 w! _6 \            return EFI_LOAD_ERROR;
        }

7 X5 R2 S. P: d  r    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
% h) n) y$ o# e0 S0 X. I        return EFI_LOAD_ERROR;
    }               
' w8 b- S$ ^6 s- D. r4 @' Q               
    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
- D0 J5 S5 c( T! e" a# ?    if (EFI_ERROR(Status)) {
        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
  u* w3 J! d6 K, Y6 l  v. u) G        return EFI_LOAD_ERROR;
    }
       
9 k( _& p0 W" w    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;
    }
$ v5 {: z; M( `, @9 r1 L, B1 z       
, n6 u5 s* y, r2 y    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
7 @& F) X# c1 U2 m# |0 {9 ~        Print (L"File Buffersize is %x\n\n", BufferSize);
' m9 B- g' g% s) |& b, |+ n    if (EFI_ERROR(Status)) {
2 I" |5 J) T$ Q; w, Q        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       

    Status = FileHandle05->SetPosition(FileHandle05, 1);       
) z! e# ^# n2 d+ l& A; L    if (EFI_ERROR(Status)) {
, Q: d/ `  B( @3 ^( |, X        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
1 p8 _1 n4 j) `' `  {        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
/ g1 R0 |$ l) y/ C4 }        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
% E: K+ B! j9 ?; |$ C        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }               
       
5 k! w$ M! `/ c7 q6 w3 w, a    Status = FileHandle05->SetPosition(FileHandle05, 2);       
, z& m' C/ _, a" p2 J) @" Z+ c2 ^    if (EFI_ERROR(Status)) {
' f$ C& \1 `$ ]% j( @        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
0 e# ?& a1 Q% v, Y4 c; Z% v* x/ ^        return EFI_SUCCESS;
' _' E6 X* j% K: j# A) [    }
2 V' |' H# ^" r% b0 X! O5 u  I; b       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
) g3 k4 f/ Q5 n9 G2 l        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
0 l0 E7 q. f: Z, \1 ^& D        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
8 w8 t, ]3 R6 x+ L' P, @# B    }       
9 v# L. ^- u/ ^& B: B2 ~
//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
0 T/ u' e2 {( |1 l! c' Q1 m; p    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
. D0 ^9 ~( |1 c- `: Z9 d) p% K    Status = FileHandle05->SetPosition(FileHandle05, 2);       
3 K0 u9 d0 g& o5 K# N    if (EFI_ERROR(Status)) {
& Z7 ]% v; t, {: ~  K        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
) Z! {( c+ t; q+ L. K$ O        return EFI_SUCCESS;
    }
- ^: p0 a4 i, Q  `$ J
    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
3 N" {+ k7 ?2 O$ |6 P( _! _8 ?    if (EFI_ERROR(Status)) {
$ p- ~& [6 Z% Z1 X' @" ]# v        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
4 \! i9 J- S0 m        return EFI_SUCCESS;
' t, X6 K8 C# n' P: R    }       
8 t) V# O0 y' h; u& B0 x: ?/ R! j        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
% m# U) ^: m% F4 x: W% w) o
2 a$ H5 ]8 V7 [' q    Status = FileHandle05->SetPosition(FileHandle05, 10);       
. M/ q' V! B" Z, z  ~' ]" I    if (EFI_ERROR(Status)) {
! X. r. b& w# t' y. C        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
* i8 c) L" B+ Q    }
5 ^4 [7 |5 X$ O( _9 a, o       
/ `+ j$ c! _% m! q- a3 a; V    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
" ]* j. _/ F$ a  i1 h    if (EFI_ERROR(Status)) {
7 u# b! k; l6 h  e6 n! Z        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)) {
5 ^& c; M1 d: ?5 X6 B& C7 z        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
3 C2 u* |7 Z0 m5 W, A" I2 I8 m9 S    }
# j4 m" b6 t2 L; c       
) ?- m- A0 T, b/ f  N; b! F) d  y    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
- B8 ]# a3 @2 {' j- ]        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
- U4 U4 D! @/ e3 U  P: ]+ `        return EFI_SUCCESS;
! w; I: F$ t3 ?7 @- B    }
$ W4 X. T) {% K8 I3 }* o! ^
    Status = FileHandle05->SetPosition(FileHandle05, 14);       
* l: b& H6 X4 \+ {; a( O    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
1 O+ o& u) ^" K: _* k        return EFI_SUCCESS;
. H5 Q: W4 E8 J+ I/ m    }
( V. x  l: p7 a$ A* D       
/ a8 o3 }! w" Y* X! E; N/ A    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
$ X2 ?) ^* L1 e" @5 T6 f* D* O    if (EFI_ERROR(Status)) {
, c/ Q3 i* F1 ]: e# e3 P* `, I        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);
& I! b. u! W0 `) j    if (EFI_ERROR(Status)) {
! D) c% O9 l8 L" d( S3 |: V        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
4 U5 p( Z% M0 t/ ~    }       

! w. b% ]: D8 v* j1 w//---------------------------------------------------------------------------------------------------------------------------------------------------------------
& d1 Z2 _; T, N0 [* R; m+ c
5 D* S% l) L0 q4 c    Status = FileHandle05->SetPosition(FileHandle05, 18);       
0 m0 V/ q) a. `% `- @6 W) y9 @    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
# W/ f/ J( E1 a% B& D        return EFI_SUCCESS;
+ V2 ^( M9 Z; ]3 G) D3 I) Y; ~    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
0 F/ ], }& q! ?3 H! ~7 x        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
6 i# B3 L7 r% ?  m* F; A( {        return EFI_SUCCESS;
    }       
: g6 j' ?2 Q) a       
    FreePool(DestAddr);       

2 B6 ~& @; t+ M5 x6 F8 E    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
" D: |( h4 k; q        return EFI_SUCCESS;
    }
# D! y6 h8 U6 i' w$ `& a       
    Print(L"File Name = %s has already been created.\n", FileName);

5 Y' L! n: d& B" A4 ^9 d/ p  G6 ^    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
+ p" S2 d/ r5 p! j# T    WaitForKeyOrReset();

$ B5 J% f+ B: C    return Status;
, r' H1 T; L4 q. k: M}

chenhy

//**********************************************************************
& J- ^) H9 _; K! Z//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
. P6 z3 a( e2 E  P% d' _//**********************************************************************
/ a7 @" Z( k4 W, v  _3 A: W% A# s$ A//**********************************************************************

#include "efi.h"
#include "efilib.h"
, [8 o# |6 J2 _/ i
#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001

* q, ]- d( K$ `( e; r7 \static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
8 U. W) p  v$ t, N- m    EFI_INPUT_KEY       key;
+ h6 I, T! J: |% C% W& m" V    UINTN               index;
   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
/ U% o2 A% i5 ~0 n4 \        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
0 M7 ?5 U& o6 o) Q% Z! w* s        else
            break;
" b! O( h% a7 M5 Y0 R    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
3 h; F' D/ J4 B    }
, m' A7 Q" J: q7 J   
    return Status;
# y3 U- ~) _1 j4 |$ X6 t1 a/ ^}

UINTN
+ U$ C7 h& l) u* [2 [- s( R& d- u. `$ xAccessPciConfig (
9 @4 W  n* D! Y, h/ l1 w4 T* i    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
8 F* A, p# P9 x3 k, H' j8 H) L{
        UINT8  i;
. e7 r1 E0 q8 z! k' {) u9 l        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);       
& {0 }( R# t5 s* c/ T6 U0 n    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
$ K$ h6 \4 X: y          for (i = 0;i < 4;i++)
! o9 p7 d7 [: N: i% W, u          {
        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
" C& M# u' k. [4 i. |, |# ]                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
! h! G- ]' z' C# Q% O3 e7 I* {          }
% {( c. Z' O- L/ b+ X          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
    }

        return Not_Found_Me;
! s9 ~' W6 ?. M4 A: z' u$ q# o4 p}

) s0 T1 Q5 Z0 V+ E) _7 @+ qEFI_STATUS
InitializePciApplication (
( j6 K* t' Q1 G/ a! C    IN EFI_HANDLE           ImageHandle,
8 U1 W; Y3 s8 s4 c( K6 o2 b) W    IN EFI_SYSTEM_TABLE     *SystemTable
; z9 s9 m! b& ]- D7 o" F% G0 g, K2 P    )
" z1 q7 A% h* E2 T{
( U! E9 r) ?( l. V, K" o; j  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

    //
    // Initialize the Library.
/ q; ?9 f5 U0 }- a6 g    //
    InitializeLib (ImageHandle, SystemTable);
2 V: N" g5 m7 r# g8 p    //
    // Find the PCI driver
, ~, S" i% o3 h4 Y+ y( \    //
% L' H2 ~( o1 x8 K& j1 [    Handle = NULL;
, t5 w* X7 _/ O& b- [3 p: h7 p    BufferSize = 1;
    do
    {
      if (Handle)
      {
        FreePool (Handle);
% }$ ^8 r2 K3 @$ P& ~) b      }
! K  R/ F' \* ], {9 y4 h               
      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
" j  T5 l2 `2 |- B  d
- t: @/ e7 [( [4 [# L9 m3 H    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));

    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
- M' P% E  s3 D    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
+ H' ?6 ]; R  `; W; s' Q/ F    Result = Not_Found_Me;       
* q, i7 f0 F5 R" m- d0 X    do
    {       
        if(Index == NoHandles - 1)
         Index=0;
) U: ^+ R$ z( O0 l1 n1 v4 T& x' y) e          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
% q6 D& ^3 |# U3 S7 N# e        Print(L"THIS IS DEVICE %d \n",Index);
1 p, ^1 E6 f7 O( Q; R    if (!EFI_ERROR(Status))
9 G' b+ o" ]; [+ P( T3 ~    {
9 R  @- T. F& c1 [          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
$ A  H- e8 ~% P! c1 g& t1 G- {0 l/ v    }
    Index++;               
9 \0 h+ L. \( K' |1 b8 z    } while(Result == Not_Found_Me);
. M4 U/ u8 j" y3 Q2 L       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
( @, w6 Q2 Y, \: l$ s    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习