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

kaurohy

/*
* myfileio.c
3 G: d( Z% n  Q3 `! b# E4 G% n * Apps
8 Z. `+ ?  E6 ?3 ]8 M2 v */

) H! \+ ^2 z9 a9 [#include "efi.h"
#include "efilib.h"
5 f2 R9 E5 u. w
#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

' \( i9 ]: b' K5 Y# q. r1 @static EFI_STATUS WaitForKeyOrReset(VOID)
  g+ w+ f! j, g/ o  f, t{
3 B. q# U' U' D; X* ~    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
8 S9 S# B: v( f! c   
- L: N& @! z) }; B. m1 o    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
0 e6 w2 u. k2 W+ s' G9 P            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
" j: V; e2 l% y) O) ~4 A    }
    if (!EFI_ERROR(Status)) {
5 a/ c3 h- g; F# F6 l        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
7 D2 B1 _9 _6 m1 \6 p7 B2 A    }
   
    return Status;
, E( e: X' \# r# S. t7 W" {! y}

EFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
{
- a4 s3 G) l% j' h- i. }: ]( X    EFI_STATUS                Status;
. Z/ q9 a- x# h3 T! ^    EFI_HANDLE                *DestAddr;       
) P1 V5 W, {# [* |) |8 Q    EFI_LOADED_IMAGE        *FileHandle01;
# O3 ?- s2 b: i1 ]8 q; G3 ~    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
8 }/ Y1 Y- \  Q5 p% x; Q) N1 t7 M: F    EFI_FILE                *FileHandle05;
0 V- w" n0 K* k    CHAR16                *FileName;
, t& k1 g/ W* a3 I, m+ e        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
2 \9 h, J, b+ u, d4 O# O1 S* M    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
0 M( ~5 c( E8 H, u. F, _1 ^; y    int         Bit0, Bit1, Key_Space;       

& @2 _9 V( s. B8 i    FileName = L"NewFile.txt";
% h0 ~- s: e' s1 w& r        Space_Key = L" ";
: W' u1 e' o/ c: p3 L$ ]* v! P# s       
        BufferA = L"ABCD";
        BufferB = L"EFGH";
6 ^) A. Q' Y' p6 L6 M0 S2 C& q: q        BufferC = L"IJKL";
  w2 [8 A' v& u
        Bit0 = 0xff;
        Bit1 = 0xfe;
       
0 f, W( @5 a( w: l/ m" T& Y        Key_Space = 0x0020;
" c( B. a3 y4 }       
    InitializeLib (ImageHandle, SystemTable);       
. y1 G) J3 M3 o. |0 Q) W* Y! Y
    DestAddr = AllocatePool (BufferSize);       

( e: {2 `) l* T4 W& ^7 M    Print (L"Value of Bit0 is %x\n", Bit0);
    Print (L"Value of Bit1 is %x\n", Bit1);       

       
2 e3 e1 D5 J* y" \1 S+ s2 W- b# X- Z    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
+ ^3 s4 g5 y6 N; u" }7 ~    }
+ s& a4 H0 t5 Y8 y
! F* k2 W8 F( U- o! ~+ c    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
9 N1 P8 R( P; y: ^: R8 c    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
' B7 D; c9 r( P% A( B5 Y        }       
       
9 s8 U6 \5 ^3 P: ~( r    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
* q2 k; H: e+ X. Y' e3 N            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
. @' E9 Q5 S/ w' O+ r; `" ^            return EFI_LOAD_ERROR;
        }
( i4 t  B% H0 }7 g" H
    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
+ ]& X+ @+ k( X8 C4 B7 N    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
3 R, |3 H, G# A9 m( Z0 X, q+ c) R" J    }               
3 t5 P; w% }5 u( l" O. P               
    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
    if (EFI_ERROR(Status)) {
        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
" O$ B/ D# [1 k" E2 r" e    }
5 Z2 w) r* G; c- b8 K3 `2 j0 |* ?! V       
    Status = FileHandle05->SetPosition(FileHandle05, 0);       
    if (EFI_ERROR(Status)) {
  X8 }9 l+ ~) L" J) _; N        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
  ?. c8 T" x# b4 o6 x5 N    }
       
    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;
( I* J7 B9 b4 I    }       

. Q5 K. R) I- g    Status = FileHandle05->SetPosition(FileHandle05, 1);       
    if (EFI_ERROR(Status)) {
8 q8 E5 p2 k! a/ F. e0 E0 H5 s+ k        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
0 l/ b* f, ^: [5 O6 m        return EFI_SUCCESS;
/ m3 o6 r) E9 L* V% f$ `+ ^! P6 ?* l    }
: A. Y4 d1 o* S/ H3 N& {       
' n9 C5 V3 u0 c, h5 ~+ e    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
% V( N; _5 F; Y3 U2 f/ J        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);
+ o; {6 z  [) V5 @; V2 t# {5 u        return EFI_SUCCESS;
    }               
+ P3 i! m9 q: T  I- @3 |* B$ L       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
) S( P* e& A* L6 r# ]  Y        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
& ]) J: j4 k* G, ^  x        return EFI_SUCCESS;
    }
9 o  R4 m, p. k" E7 x       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
/ b5 H. _/ B# I7 n( }+ r" g  x" R; O        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
  A& g0 f3 d  b$ w# I3 A        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       
* J- T. o& H3 e$ B1 n- V; `: s. ~: O
7 q! ?: x, r" s9 W4 d- }/ f7 X//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
. V- X; ?( ~/ O  n' f    Status = FileHandle05->SetPosition(FileHandle05, 2);       
: q* o( M- l0 |4 D; j3 M6 P    if (EFI_ERROR(Status)) {
/ `/ n; W( _! Z        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
  g8 d7 L6 }9 _$ B( g8 A& F; o        return EFI_SUCCESS;
' H) ?4 B" O1 r5 _# {0 a+ d4 |    }

    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
2 S" H" g" t7 ~1 [, {3 M" K: M        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

    Status = FileHandle05->SetPosition(FileHandle05, 10);       
) Z( A- @( m7 R+ t. f  E    if (EFI_ERROR(Status)) {
" M6 N2 z: q8 z* ]! S+ |+ ?        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
  K" {+ e/ V4 _" A- m- ]7 ~    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
$ I# X6 O1 `) N9 {/ m  e    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
& [2 N( C+ w  x7 |" T9 o$ Y9 M    }

    Status = FileHandle05->SetPosition(FileHandle05, 12);       
5 u8 P' R4 y: X, H- U    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
( i  G& w2 w; I- h9 ?) L        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
4 z4 R/ m  O5 ?1 B$ F* t        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
! T: R( P: E4 e. \) }    }
0 z4 l3 k) ^8 G
    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);
        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
9 j/ X5 a) C# w6 y6 g: w    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
" E: K) g+ X: |5 n8 `2 T8 l
1 Q+ d% B% [, u4 f8 Q. E    Status = FileHandle05->SetPosition(FileHandle05, 16);       
" g  h+ g' |# d8 z0 o% k    if (EFI_ERROR(Status)) {
$ M& U1 B. O+ v4 o7 m        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
5 p- \8 P: m, N- r- Z2 s, C    }
6 H; _0 l9 B( h, I4 t9 T& C       
  L( U* W& k; |    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
2 D5 T0 {# a) X        return EFI_SUCCESS;
    }       

1 a# \  D" p3 Q) c3 ]0 g- G//---------------------------------------------------------------------------------------------------------------------------------------------------------------
# E( W+ ^% ]0 ]; |4 C: d: d
6 C) U; U1 x+ X- Z5 I: P2 l    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
8 }# v- ~8 `+ q% \. V- M1 ]        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
' G. _2 K( w: H& W% e  A        return EFI_SUCCESS;
    }
+ [" u. _$ W  {       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
2 Q9 g5 |/ x. h" Y5 O% _        return EFI_SUCCESS;
+ R* n+ V! t0 g' f  b! Y+ P6 q    }       
       
    FreePool(DestAddr);       
2 a# H: Q! q4 G3 Y: Q
9 k1 R, Z/ \8 Y& h1 W3 n    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
, L5 X: A- I( y- U) z8 ^    }
4 R9 J' \9 B* P1 ]       
6 u: A$ X; E+ P1 @# H' a5 `    Print(L"File Name = %s has already been created.\n", FileName);
' [; y2 i' [9 m2 {& ~4 }% u
* g5 n, e6 l6 [8 v# i. ~, B( x0 h    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();

    return Status;
}

chenhy

//**********************************************************************
//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
6 e8 l9 y5 e; [9 Y. k//** By Lawrence 2009/11/20 @Taiwan**                          
; W% W8 \  q# h; F9 h//**********************************************************************
//**********************************************************************
' v4 H+ f1 n2 E  n/ t: X
#include "efi.h"
1 `& p- k. f- R#include "efilib.h"
8 i8 I: c$ J/ F* t6 c
#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001
0 ~9 G7 Q' c) u. M8 \
static EFI_STATUS WaitForKeyOrReset(VOID)
* Z0 i8 e% B/ K& p{
. s5 j1 H7 X  D: x' |4 p    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
* n, d3 @% ~! v$ v5 C9 H; _        if (Status == EFI_NOT_READY)
( I# K( B8 s/ @7 d/ C; m: }; t            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
# U' M0 H! v: l& U$ O" D            break;
    }
    if (!EFI_ERROR(Status)) {
! p2 C1 G7 w$ b3 f6 t, [        if (key.ScanCode == SCAN_ESC)
; z4 |4 O3 Q2 k; x" |/ ?8 ~. |            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
9 ^0 `! R1 t# X5 p4 p7 N# n   
) W0 ?% o3 G% g1 K. R    return Status;
% P4 Y( M3 n8 H" x! r5 @! B9 D}
# _  f( M" L' @. q, ~6 k1 L* ]; _
UINTN
/ W  R8 b- z! WAccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
$ n+ ~" l2 p6 a2 Y: x0 x# Z" f+ Z{
        UINT8  i;
        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;
, V5 ^# Z! l  b; f- K
        //get device attr
4 i/ Q; E/ w9 }) b1 y" F8 O        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));       
* \1 ?4 @) D/ F2 T* s5 ]          //print register value
          for (i = 0;i < 4;i++)
: }! k8 X, x* y0 j( P. o7 K          {
0 j% e3 b+ r7 N: }# R        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");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
    }
6 S' y  ?4 w/ Q( l: L* o
& b$ S4 T5 v! @) b5 D        return Not_Found_Me;
- d+ k9 ?4 C) j; p% o1 r}
* G7 Y! ?1 u, u* }( B
EFI_STATUS
+ J; C- U2 v, w; `9 T  b) ?3 `InitializePciApplication (
7 D& y2 O4 E  B  q    IN EFI_HANDLE           ImageHandle,
5 f' F  e2 Q. H. ^( U1 x    IN EFI_SYSTEM_TABLE     *SystemTable
    )
{
  EFI_STATUS              Status;
) i6 v, O) d5 ^- m/ K4 ^. Q  EFI_HANDLE              *Handle;
6 R0 M' {& t9 m9 w2 R) M* q  UINTN                   BufferSize, Result;
9 M! Z5 e% R) E7 J  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
/ X( ^1 V0 s. f8 K
$ k3 H3 j* w: O. q5 Z    //
    // Initialize the Library.
& N; w7 E2 e. O3 y    //
7 G# `7 F( n5 z& M+ i; w    InitializeLib (ImageHandle, SystemTable);
9 U+ u% [9 [/ [5 f; G    //
    // Find the PCI driver
7 G7 l9 A+ W* |9 }0 ^2 y1 q/ t    //
' G3 l. N: g* j9 K( V4 x    Handle = NULL;
$ m& u: b0 e) L3 H6 ?% V) [/ _    BufferSize = 1;
    do
    {
$ `. a- y0 o8 o: Y      if (Handle)
      {
        FreePool (Handle);
      }
# R5 ?2 r  R% n/ j               
+ X; i- L7 E# C4 J1 h0 n      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
' R2 V, D% D4 z: G6 l  y8 g
; c( W; ?' [9 a: s    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));
+ ?$ H; _1 n" g4 L
2 `$ ?# \4 h" v; W3 O# D( T    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
9 @0 ~5 \5 x8 Z3 H    Index = 0;
# o, F+ s1 y5 A    ST->ConOut->ClearScreen (ST->ConOut);
. v4 G! V! M. m! y    Result = Not_Found_Me;       
    do
    {       
        if(Index == NoHandles - 1)
         Index=0;
2 d3 m0 R: ~5 y5 o. @0 i          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
: l' ]# X3 [- y" t5 C    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
* c) v: g  I/ ]    }
! J% P3 z6 b$ G* `' i    Index++;               
5 y. K9 u; g5 R9 `( v    } while(Result == Not_Found_Me);
) B- y  |0 i& X  z6 p1 d! Q: U       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
9 G1 Z! V7 ?- a( }6 d               
; `+ }2 p+ g9 }6 O% ^( A    return EFI_SUCCESS;
8 ~9 y% H; J' n4 x$ v7 P. u2 R}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习