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

kaurohy

/*
* myfileio.c
* Apps
- U4 e/ Y5 N, h1 w% k) | */
) c% A; G& W) b$ `; m) I0 p2 T
#include "efi.h"
  v' {& W' l! |2 Z1 m1 \+ A; ^1 J#include "efilib.h"

#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

static EFI_STATUS WaitForKeyOrReset(VOID)
{
5 c) ?/ H- F2 Q6 l    EFI_STATUS          Status;
" V- }3 ~4 d4 n    EFI_INPUT_KEY       key;
& c; D: u; w% c# ~3 A    UINTN               index;
0 D! R* u! Z. M- R. [   
) K- l# _2 I1 h( `% k+ {    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
. F, j. v9 y- }        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
    }
4 x/ e! M" R$ P+ ?    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
! G7 X9 S# @7 l& O4 M   
    return Status;
: b" P& {1 V" h3 w( I! T* f3 _}

EFI_STATUS
" d" o( H6 ]; u# j9 V* bEFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
2 B1 c, y& s5 R! W5 q; n5 U  z$ M             IN EFI_SYSTEM_TABLE     *SystemTable)
: [2 u* L* V* p  m4 g% k* w6 c{
    EFI_STATUS                Status;
6 c  [5 Y, n1 v0 t. H    EFI_HANDLE                *DestAddr;       
. a8 P4 \  Y/ o7 w    EFI_LOADED_IMAGE        *FileHandle01;
% _* u1 G0 o  C" y5 I' R) [) X& Q    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
2 p. r7 s7 U6 ?    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
0 j( K/ [/ A- P+ e    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       

    FileName = L"NewFile.txt";
% ]) B, l# J' D: O6 ]+ j        Space_Key = L" ";
       
        BufferA = L"ABCD";
" _  ~# `. B3 e' j* ~        BufferB = L"EFGH";
8 A; K0 H2 x, z+ o; i: ]5 `        BufferC = L"IJKL";

        Bit0 = 0xff;
. d+ u& |5 Q; f8 t  ^9 r& C3 I2 r        Bit1 = 0xfe;
       
; O& c0 }* z! S) c+ q        Key_Space = 0x0020;
8 O0 o' s2 Q* V  G  j3 S6 u       
! _- x. L+ w2 e) j    InitializeLib (ImageHandle, SystemTable);       

4 o8 [2 z. u  j* Y' _* J    DestAddr = AllocatePool (BufferSize);       

    Print (L"Value of Bit0 is %x\n", Bit0);
    Print (L"Value of Bit1 is %x\n", Bit1);       
3 b! l6 }  E$ H1 z# _
       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
) Z  d- `4 }% _, M/ X5 l9 i2 p$ u    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
' d' p, }  ~% ~* r( ~; F! j    }

; a' H8 Y4 u; I$ I  ]0 n! T    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);
            return EFI_LOAD_ERROR;
        }       
       
8 e+ D0 d6 v/ ^; q, O' c1 c    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);
) ]% ]9 l9 |0 q8 ~  j            return EFI_LOAD_ERROR;
5 l1 @) d) G1 x; j2 F        }

7 x. c, _+ Y) Y5 E+ q, a8 r, i    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
* T. n/ z) Z. t2 I8 w    }               
               
6 H0 ]1 z% ~- E5 J1 w    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
- P5 b: m0 R: `8 e) l! q    if (EFI_ERROR(Status)) {
- ]% S+ O+ E1 o9 Q+ Y% C        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }
       
6 e7 t# u& I6 q# t% W6 O1 B0 w' I7 a    Status = FileHandle05->SetPosition(FileHandle05, 0);       
6 u) U& g  P* b5 I. M    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
6 n% d/ m$ ?; v% Q        return EFI_SUCCESS;
" @- B' @. \# i( c    }
# S. y. l* h' Y: _9 l       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        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;
! D7 J, Q5 w& n# R. ~: D& b+ N    }       

    Status = FileHandle05->SetPosition(FileHandle05, 1);       
    if (EFI_ERROR(Status)) {
4 }* W9 g1 ?  @        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
$ i+ r! ~9 P- ?0 p* m    }
1 z. g; p  z& z. A       
! e# V/ ]0 v% d    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
  j: A0 B: [! Z# B        Print (L"File Buffersize is %x\n\n", BufferSize);
8 N3 ~! I/ [; a    if (EFI_ERROR(Status)) {
2 a9 Z' u% r: Z) u        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }               
2 e7 H0 n* X0 U- [2 ]7 w' H       
1 v8 ]# R% ~+ r4 ]  L6 K# C) N    Status = FileHandle05->SetPosition(FileHandle05, 2);       
0 Y5 k; {3 c! g0 L# r    if (EFI_ERROR(Status)) {
; n" [9 q9 v+ {) P+ `        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
: t3 f  O* ]8 r) x4 u" ], N       
# ^$ n% N/ |9 r( N) O- e6 U6 ^1 {    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
        Print (L"File Buffersize is %x\n\n", BufferSize);
, _$ E" ?, |0 e# V    if (EFI_ERROR(Status)) {
. |2 @2 t- m6 @: B$ J; h        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
6 G2 v' i0 l9 K5 ~% v5 U/ C    }       
! r- |- X( B9 {
6 G3 D; ~1 y8 W& [: f$ ]2 c* H//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
7 I- J3 D1 ?. n+ Z7 S    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
: M8 x% v4 S' l  I& U% j    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
& g4 l7 R/ P$ j9 x! `  X9 D1 B        return EFI_SUCCESS;
    }

    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
# b$ ?% ^( S/ o( A% X2 u    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
. |! O) h" F1 d6 b: w        return EFI_SUCCESS;
; Y' |+ c0 H* L& w* D7 A( h4 `    }       
0 ~9 r3 O+ ^# n9 ]) v) c8 `4 U        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
5 s: r' i0 u3 A4 L( g6 ~# S
8 A( X" o& i5 Y0 Z& H    Status = FileHandle05->SetPosition(FileHandle05, 10);       
    if (EFI_ERROR(Status)) {
) C# `2 x* ]7 x9 u( I/ h. l2 q        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
& _# D$ H" N1 w  A$ ]    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
; \7 a7 `1 ?' f8 g6 L    if (EFI_ERROR(Status)) {
$ N3 W8 u: ^" |/ k0 I  Y3 i        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
9 p9 [" G9 S! U9 H    }
0 i7 a9 o% [% U/ a7 _3 M  a* S
+ {5 ]: Q4 J) ^8 i7 S& M, M    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
3 ]0 g5 k$ U' I, c% _, y        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
) s, ^' V3 X: Y% W, D" T        return EFI_SUCCESS;
& r4 m. y7 W5 {" V3 |  S1 q    }
       
3 B7 D, @( v3 \; n0 Z    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
. x" t. U" w& R- f: F/ b    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
( ?- v9 I+ \! C7 L) }$ [1 t0 [' s2 d    }
/ `+ V/ q6 n. Z! @
    Status = FileHandle05->SetPosition(FileHandle05, 14);       
! R* o2 s$ J+ _4 X  K    if (EFI_ERROR(Status)) {
7 o5 L/ m5 T1 R! q; ~8 z        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
. x2 Q, R. t. Y$ o! I2 V- m9 n        return EFI_SUCCESS;
( K( P4 h9 G- Y+ r  d" y9 F8 G    }
       
7 A6 n6 T3 U: a' a! v    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
" z8 @) s: d$ B, t. R    if (EFI_ERROR(Status)) {
! O! h7 P1 p/ q9 J( G. ~        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
& o! |. l# S( S# z6 l
% f5 ]. _% t8 t5 S    Status = FileHandle05->SetPosition(FileHandle05, 16);       
    if (EFI_ERROR(Status)) {
* K( X4 D7 P, \+ O$ Y) G        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
& p  r% M( o1 s( q3 [       
: u9 O2 [( m" A( h4 }" l, |! ?    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
% x4 A9 d7 I% j" U, I        return EFI_SUCCESS;
    }       
" E( {7 o2 O  y1 \/ R( @9 F& x
3 Y; x, F- x. ]//---------------------------------------------------------------------------------------------------------------------------------------------------------------
  b8 L! j# @/ s4 t# x9 M
7 t. b' E9 W+ G' F0 x* O' K# U9 y; p    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
9 h# N  z6 d$ O1 ?        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
) u. L/ d/ f) Q    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
       
    FreePool(DestAddr);       

    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
: v: h, Q! t! B$ g. s        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
# G! J9 k9 I9 A$ V0 @# u/ Y- n    }
       
    Print(L"File Name = %s has already been created.\n", FileName);

  V! n& T" p! o) S# c$ m& z3 p- y+ ^    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
" U) g! P, ~7 v$ x    WaitForKeyOrReset();

    return Status;
}

chenhy

//**********************************************************************
//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
8 y6 P5 e9 u# d- E7 Z3 ^* {  i; Y/ X//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
4 B! \: d# }, w' m7 M0 R" m6 o//**********************************************************************

7 Y; b# D- o% A, E#include "efi.h"
3 ]& R2 u1 X- C% ?#include "efilib.h"
5 a& G- }' ~& u0 m% Q  X8 r2 R
: x6 ]+ s9 _2 K- p6 f8 H- w#define        Not_Found_Me        0x00000000
' ]0 C0 Z7 F  k" ^: C#define        Found_Me        0x00000001
: _) k! |1 v5 }0 D7 t6 r7 e
static EFI_STATUS WaitForKeyOrReset(VOID)
{
, {: `) n, }8 X& R    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
2 O0 y0 T6 _' v    UINTN               index;
2 ^4 L0 b$ u$ }) u0 c7 ^: V1 Y   
    for(;;) {
' c5 I- N1 B' t" ]( K) O        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
, ?: W' E+ c3 k; X8 N        if (Status == EFI_NOT_READY)
; W5 W6 V0 t4 t8 F, k            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
    }
+ b% j" _( u( z$ R) |% ^    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
0 o: t" f4 M/ y4 [) l  {( U! Z   
7 c* k( ?6 o* u& j/ l; Q% q7 y    return Status;
; w$ n7 q' E% q7 v' T}

UINTN
% O+ t3 b) V! B7 [6 t7 b! FAccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
. Q$ q+ C. |5 m( M{
; E7 c0 i; t- ~) d! X% T3 X. U$ ~( T        UINT8  i;
# n$ @" u4 G: T, }1 R6 q        UINTN  PCI_ATTR[4];
* n# t* b4 K6 p; O        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);       
    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
  d5 V9 O( c( c; Z, j2 ]; t          for (i = 0;i < 4;i++)
          {
7 Q- C9 j; I8 A9 ^; C: B        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
& w: U2 n7 T* j9 k* W  i0 d8 h: o! Z                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
          }
  Q% H  S% h3 E: N0 x* D          Print(L"\n");          
3 x8 Y+ k7 B! P3 i          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
, Y5 {. P0 b/ c7 b& ]          return Found_Me;
- C1 |8 w7 k( S+ Q* g% v$ F2 W/ b    }
" C* O; a7 y- p
& u0 \0 T; U5 X8 Z1 @) R0 u, F$ ^        return Not_Found_Me;
}
; Z6 N5 E7 H8 \
EFI_STATUS
7 z% L) f! m2 z) T0 H/ sInitializePciApplication (
; [; A2 v" [" x1 c/ v8 e    IN EFI_HANDLE           ImageHandle,
. Z5 k! Y4 y( z# H) m0 Z0 p    IN EFI_SYSTEM_TABLE     *SystemTable
    )
( X0 s! q* d6 _5 @6 Z; ^% H* q{
  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
0 m+ G2 K: `* g9 }4 }  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
% Y, |1 m6 w; n& o- r' i: k
    //
+ h  A7 N! B: G' F# b    // Initialize the Library.
    //
. \* p1 l* ~* X; [. F$ T    InitializeLib (ImageHandle, SystemTable);
    //
    // Find the PCI driver
0 b* _  P6 c: S+ o5 E    //
    Handle = NULL;
$ M5 r. v( z  W% p, B& W( _    BufferSize = 1;
% H( t, d( d; C4 i  G    do
5 A+ b7 E) y# P% [. H    {
5 }+ J/ A+ I! u      if (Handle)
' \( x! @: ^- d# d      {
; C( _8 g: n# u4 s' V6 B. |5 r6 b        FreePool (Handle);
      }
               
      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

    } while (Status == EFI_BUFFER_TOO_SMALL);
+ l$ s% Q6 ?. i$ M3 O    ASSERT (!EFI_ERROR(Status));

7 k$ H  N9 q7 s6 _7 o, ^8 s    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
& H# u6 K. ]) g$ [    Index = 0;
/ Y" S( m# B+ u    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
    do
    {       
        if(Index == NoHandles - 1)
         Index=0;
+ A/ L; l9 v- j0 |2 W          else
; x0 m2 S! r8 ^1 ]) H    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
2 M4 d2 ?: S' }: @    if (!EFI_ERROR(Status))
1 K( C) k( H8 X: W* B# j4 E4 B    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
    }
3 R  o0 ~& g4 A2 [    Index++;               
    } while(Result == Not_Found_Me);
       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
% ?2 G: s4 d" s7 h' n               
    return EFI_SUCCESS;
/ k) a0 f$ b6 p! A" ~' [}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习