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

kaurohy

/*
3 q% V: ]7 x1 ~* B& N. S% a * myfileio.c
  Z6 K2 D" u2 x& R * Apps
+ X' F, {$ k7 }8 o0 q$ L */
  n4 [  Z. d+ P
#include "efi.h"
* N# S' u9 p- N#include "efilib.h"

#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
2 X# G0 P! m- F
static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
* j- x  o1 f* o3 R+ s$ h    EFI_INPUT_KEY       key;
    UINTN               index;
/ ~0 H9 s3 Z. I! |   
    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)) {
* `8 p" v) Y0 R1 u8 Q+ f4 D9 R1 j        if (key.ScanCode == SCAN_ESC)
  }* R6 n0 {7 w% c* R. Z1 x            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
# c. C' z+ c8 j4 G0 P. @# Y   
( v* f' ^1 c9 u% X, a- `9 c, r    return Status;
}

, A1 J: i& b, e! \0 JEFI_STATUS
% }+ Z" J% l) A. |EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
$ A' j1 n. l7 r{
  [" w. }6 c  h! W    EFI_STATUS                Status;
& J7 B  U4 W* \3 `    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
' y5 Q% x" L4 B0 O    EFI_FILE                *FileHandle05;
% V$ A2 r. Q: E& A9 a' S    CHAR16                *FileName;
7 z0 X. Q5 T% ]( h( a        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
3 T/ T* j8 L! Z: m/ t    int         Bit0, Bit1, Key_Space;       

    FileName = L"NewFile.txt";
        Space_Key = L" ";
       
9 f0 J3 p0 U0 t9 n        BufferA = L"ABCD";
        BufferB = L"EFGH";
: [& ]2 j" |2 }: ~$ B/ g; G2 C1 V        BufferC = L"IJKL";
' v+ c* S6 i) C  ~7 G
2 R8 m2 J4 L) ?" `        Bit0 = 0xff;
! g: ~- W- ?+ E  s4 D% u2 r! a* v        Bit1 = 0xfe;
$ |9 n* ?0 a" I/ W0 m       
        Key_Space = 0x0020;
, N8 }- S2 ^4 u9 ]       
* B5 d& Y. d9 h5 L) {    InitializeLib (ImageHandle, SystemTable);       

/ g8 y; i2 L5 |( Y1 h5 o% E    DestAddr = AllocatePool (BufferSize);       

    Print (L"Value of Bit0 is %x\n", Bit0);
2 Q. j, p/ N/ }  [/ r    Print (L"Value of Bit1 is %x\n", Bit1);       
1 b; C! r. }8 v  v
3 O7 D8 s$ d1 G       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
, S+ a2 @1 e* v" c  q; u7 Y" ^    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
7 {) O2 g( D" u6 S  u& Y* s7 ?        return EFI_LOAD_ERROR;
3 p% Q, n( p# G4 Q8 R    }
( l( Q$ D7 _5 r; _8 F5 L# K) c$ h
9 J+ Z$ d% q  P( h8 a, v    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
) w* t4 D8 P) m* X" c    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
. L/ V4 f1 R* T  V3 p            return EFI_LOAD_ERROR;
" r8 y9 H% H; z" m        }       
       
    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;
) @, j( j' m. v3 Q& y  ^        }
2 K' m+ s+ }8 s! M& e- N) K' S4 y6 C
/ r+ A. }& B4 q* X* z8 v    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
2 }8 D# r( y5 Y/ T0 \/ u; z! V) c$ m    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
1 {7 ]6 O) s8 B5 D. U/ c        return EFI_LOAD_ERROR;
9 O: D% ]2 J/ R# j0 a1 q    }               
' e/ b7 o! {) g' i3 u               
* b, C* ]8 E6 \! Z    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
    if (EFI_ERROR(Status)) {
" L* N! ~% b8 C# j3 Z( Y# t7 R        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
1 n5 u% `3 i+ \4 V4 }. D/ D        return EFI_LOAD_ERROR;
3 c6 r8 `- ~% a* u0 M  q( V    }
" c+ f: z; V! h1 W! {, t8 p; f% O       
    Status = FileHandle05->SetPosition(FileHandle05, 0);       
    if (EFI_ERROR(Status)) {
8 J1 M$ J8 |. l        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
- K) ]# T9 B  X% u1 C    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
6 b$ e0 y$ z. c    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
6 \8 h1 }6 @# Y* o' h2 p# g        return EFI_SUCCESS;
7 F9 E+ q1 p) t    }       

    Status = FileHandle05->SetPosition(FileHandle05, 1);       
  g, \- T& n0 x    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
0 f" E0 w/ h# U2 `    }
/ D: g- J6 F, Y& u; E       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
; B5 v" L3 A. ~- X    if (EFI_ERROR(Status)) {
0 d4 a/ `* v6 Y/ j' B        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
% l5 L4 r4 o: i' ]# }4 S  i. c    }               
) b+ n& G8 ?( d, P) \9 E4 g       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
8 k7 X& g$ C5 E6 {2 K! H: F+ v8 D, x    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
/ l  U  Z- U1 ^2 {; w5 s$ l1 ~    }
# _/ D9 {; C/ P7 h; e: w       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
1 F$ ~( r6 t2 Z* F9 l        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;
    }       

& e7 N3 d! }: Q" Q' W9 c8 t1 `//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
4 w/ t7 f, p" |: w( h5 p    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
+ _4 |+ {  U7 s0 z2 s        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }

5 K- J+ O' }( m6 G; A    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
# C5 I% S' i2 l: @, a  f* s        return EFI_SUCCESS;
    }       
0 ]3 E+ H' `9 w" @1 }/ q. o        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
# z. m9 P. _. E//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

4 v$ ^; @( @- ?# J    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);
( S$ I% r) ^' b, a0 D  A        return EFI_SUCCESS;
    }
. {) |# e( U& \# A2 m7 P- I8 R* b: @       
% ?! p' A! ?) S+ l% {2 T( y9 `    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
5 Q" E: J; y$ q5 \/ C5 b        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
* N1 Y+ k% `& C    }
7 T7 r- n. a. p  b5 M
    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
1 v$ f1 j8 g) @) u9 f& c5 \        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
/ M2 V1 w: T1 }+ H$ W        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
* E- v% m+ _  A' [    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
/ A  D$ h7 s$ n$ z        return EFI_SUCCESS;
. r+ q( t/ L$ |) c% ~+ M3 @    }

, P6 O) q2 M. |( F: J  d    Status = FileHandle05->SetPosition(FileHandle05, 14);       
+ s! k  l& z/ i' k" o* K    if (EFI_ERROR(Status)) {
0 |# S# A( c( `0 D+ |8 y( Z        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
6 Z7 ~( j! G; h6 J/ }) N        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
! q& h8 E/ K% ^7 }    if (EFI_ERROR(Status)) {
* ?* X: o& ]1 _" ^) s1 ]        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
4 _$ R2 @/ c" T" i! k1 E2 f    }
# z) V" k! n8 a
    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;
    }
: ~( J3 O5 C& L0 x) H! \3 z       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
5 w$ `( H6 z* ^5 j: C        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
: }6 p. U4 _" x7 R1 g! m5 c        return EFI_SUCCESS;
    }       
2 @9 z1 d1 ~* p& R0 g0 q* ^
//---------------------------------------------------------------------------------------------------------------------------------------------------------------
. U8 y0 L: H+ L5 q1 j  F1 i1 p
    Status = FileHandle05->SetPosition(FileHandle05, 18);       
0 P  L9 R0 x4 _* \3 b    if (EFI_ERROR(Status)) {
2 k! Y; k* _2 y6 O  L# |        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
: P0 T# \* }9 W$ {- \; ?& Z* [    }
       
+ c) a: A4 w' R. a& W    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
* d9 E; v% A  h8 z* B& W+ S) F    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
  P& J' D0 o1 c) Y        return EFI_SUCCESS;
+ n. Q  S8 B/ I+ E" ^    }       
0 P5 L# Y$ i; v! Q( U       
7 ^) V. {( `4 k    FreePool(DestAddr);       

- d) B6 F  Y$ l3 H    Status = FileHandle05->Close(FileHandle05);
' {& x, x7 V1 H: X3 \( ^    if (EFI_ERROR(Status)) {
5 H; p# a: l1 E        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
; H. i7 m( B& I( J    }
       
% t  @6 M: d# {. E0 P" }* I    Print(L"File Name = %s has already been created.\n", FileName);
% z, P$ ?* o2 \. i
) t4 {0 b, J7 L0 D6 H    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
0 ~3 i# b) ~7 j5 P7 G6 |& f" }- S    WaitForKeyOrReset();

    return Status;
}

chenhy

//**********************************************************************
: O! e: d' i' m  k8 Z//**********************************************************************
4 W( v! U& E( f$ p" F* Q//** Demo code for PCI configuration space I/O access program **
5 z/ L  N5 {( b& K' s" d0 _3 x//** By Lawrence 2009/11/20 @Taiwan**                          
% g% ^' z  w. w  J//**********************************************************************
//**********************************************************************
* E! M( ^9 }5 x, y
# Q& H1 p) `4 E. M1 `6 w. X#include "efi.h"
#include "efilib.h"
/ W! \- H8 V( e6 s
( r7 k) m  l6 ?3 w/ H4 J5 ?0 @#define        Not_Found_Me        0x00000000
) @+ |2 T9 Z. r3 j/ M3 R  A#define        Found_Me        0x00000001
. `2 A9 `, o: O; J8 m
& g) t/ s9 [0 _0 S# R+ H! \static EFI_STATUS WaitForKeyOrReset(VOID)
{
* n) W6 o0 j$ A6 [- @$ N! l- I4 v    EFI_STATUS          Status;
- G; v( @1 b# U; ^; h& f    EFI_INPUT_KEY       key;
. ^" R$ p$ h. l2 `; H0 a( k    UINTN               index;
: H+ I+ J. I# [) F   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
# G% t6 W( r" S2 N) Q% `: x% U% m            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
3 Q9 a: E/ n. d/ Y        else
) g9 Z9 H0 M& R            break;
    }
    if (!EFI_ERROR(Status)) {
( l- P6 t1 Q  d8 M  Z' P" o# t0 C7 d        if (key.ScanCode == SCAN_ESC)
% ]+ p6 V9 ]1 D            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
0 J/ D+ F% ~( P" \, ?) B! g+ V% ~    return Status;
2 i# P+ r+ O: L! y}
3 j& J$ Z  J3 o4 X+ t  Z3 f
+ A* L9 l! H0 G8 U$ m" h( UUINTN
AccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
{
        UINT8  i;
        UINTN  PCI_ATTR[4];
( @9 ?1 ?* R& j+ G        UINT8  PCI_REG[4], pci_offset=0;
/ M8 p" w6 B2 M& g
        //get device attr
3 Q1 s# r2 q# F1 h$ g. ]& ^        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]);
9 {2 r7 d$ B" r, o          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);
          }
          Print(L"\n");          
- P  F) `, O( ], l- G, F9 Z          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
' t8 d& u) O$ V# i8 t( T. b          return Found_Me;
8 s& [7 Y( y8 l& t    }

8 R: u) A: V8 D& k* _        return Not_Found_Me;
; @% g5 ?3 e5 w}

EFI_STATUS
InitializePciApplication (
5 v( i: Q2 m& C3 E5 p    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
' [2 ]. Q8 t$ x    )
/ _: ^- ]( H* F. m9 B7 I{
  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
, Y- c' [7 i" w& V4 _. P& b  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

    //
$ x" a4 f- I& Y! u5 L    // Initialize the Library.
    //
    InitializeLib (ImageHandle, SystemTable);
  C" w' b, {/ J, ~3 K" L5 ^    //
    // Find the PCI driver
! V( g" o3 i$ k- @    //
; M) B/ Z- U0 `: O- d9 N) n    Handle = NULL;
    BufferSize = 1;
0 V% }, q. i6 f- T. F: x. b    do
    {
      if (Handle)
      {
        FreePool (Handle);
* m7 }' f/ o4 `/ Q4 ]* B2 }      }
+ F% s+ E# Z% ^/ m               
      Handle = AllocatePool (BufferSize);
3 Y% z0 Q+ s. l$ U      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

: }( G: z7 z3 K, N4 V! V- z    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));

    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
4 N8 {! F+ Q+ `    ST->ConOut->ClearScreen (ST->ConOut);
  C1 H; `( f5 {# D: G# X' O3 ~' h    Result = Not_Found_Me;       
0 W; F% E) @* \( J% y" D    do
    {       
        if(Index == NoHandles - 1)
         Index=0;
          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
% J- I! V" ~1 X3 O5 ^          Result = AccessPciConfig(PCI_DEVICE_TARGET);
8 F: X' x" s, l) I3 H//-          Print(L"Again now Result is %x \n",Result);          
    }
2 g: ^1 a* Y8 \5 J9 w+ D6 Y5 v! K    Index++;               
( b4 k& F# q' K( I    } while(Result == Not_Found_Me);
/ B% K+ a7 K$ t0 x* t       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
! j/ g" j0 J8 ?* T. K$ s) g& Z$ b               
    return EFI_SUCCESS;
* \7 y# L/ K- P* \- S}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习