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

kaurohy

/*
* myfileio.c
1 ?3 T7 l9 e( R4 y# j. j$ i& F * Apps
*/

" R' a# M/ \' k#include "efi.h"
: v! J7 m2 V% ]#include "efilib.h"
* k3 ?* q( K6 V; [" @3 H' p! P
#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
1 |+ T) A- S1 [7 E  h. S$ y7 t2 m
: R* P* [  o1 X8 b* H. `- Qstatic EFI_STATUS WaitForKeyOrReset(VOID)
' ~. T$ H% T* \: b0 p{
" f5 ~% z* Z. I1 a4 R    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
: w( H& i* U: ^" }% [$ e   
    for(;;) {
+ h, m2 P2 ^# p2 e; [8 c        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
' A+ w0 Y. z7 @) U! O        if (Status == EFI_NOT_READY)
0 s( c2 K6 @9 w6 l            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
    }
% g# O- J' R9 z- B) ^    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
' s. o% Y6 H4 c- l! B            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
% P; W7 e/ ^* k# I; w0 y7 J6 ^5 L   
    return Status;
}

# E1 d* |' w9 ^6 g# v2 u, j/ d6 GEFI_STATUS
0 `" c1 d  s8 H4 AEFIAPI
1 d+ _; A* {+ w- zMyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
{
    EFI_STATUS                Status;
+ t; v, Z/ T$ N6 O    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
3 Z/ N  Z# c3 k    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
2 B7 k& b3 l0 q" E9 O        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
- K, @4 T$ Z. j9 X9 D: W    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       

" X' U( H/ Q! L, Q) H    FileName = L"NewFile.txt";
+ m! U! o" f1 K( l9 V' V5 |        Space_Key = L" ";
, ?5 \$ S# B" g6 J2 j+ t0 Z# X% D       
        BufferA = L"ABCD";
% d7 b- y% a$ J4 A        BufferB = L"EFGH";
        BufferC = L"IJKL";
/ p# u: C' j* n8 F. {7 Q; W
# \1 j8 O' @8 c        Bit0 = 0xff;
        Bit1 = 0xfe;
- E) T$ s' T: D8 W. G1 P. A       
        Key_Space = 0x0020;
/ W, C, b7 Y! V       
/ F1 n) k; z# Z1 S7 L" ?    InitializeLib (ImageHandle, SystemTable);       
( Q! }/ k' D- W5 ^* J. r  f
    DestAddr = AllocatePool (BufferSize);       

" [; ~; Y% c; c/ _! Q    Print (L"Value of Bit0 is %x\n", Bit0);
' Z; D* p6 Q7 W0 r1 R* ]5 r    Print (L"Value of Bit1 is %x\n", Bit1);       
# Q2 b0 `6 I1 k! z5 h' A
& N+ d* _. Y) a& o( u5 I+ o       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
- v' ?- X& U" X! D/ J! }1 _& P        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
; ], W2 j2 p3 A) r% H0 r2 }        return EFI_LOAD_ERROR;
) K, k( J1 P7 Y    }

7 t' z6 _# D' O; R    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
! A# l( t9 o. [6 K: z. ?4 k    if (EFI_ERROR(Status)) {
" w5 \3 ?1 S, Q9 P4 o9 [3 d            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
1 h! V% ]4 }6 x; [& ~            return EFI_LOAD_ERROR;
9 @3 e& M: d: K( h) @        }       
       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
* {: a0 S  _8 s    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
; g2 K$ Z, @5 F( B# u+ g# R9 D            return EFI_LOAD_ERROR;
        }
' V8 x7 e4 @) W  ^: o- }/ B0 n
    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
& p- R! _. Y: w    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
  u- A4 m6 x( \# F0 R' }        return EFI_LOAD_ERROR;
    }               
0 W8 w9 d" I7 m5 _: a; Q               
    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
3 |/ C# g7 e% n) m. E  G( R% o8 V    if (EFI_ERROR(Status)) {
' X- U5 S4 ]1 N0 ^1 i0 ?" t        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
0 R& p: [9 f  t5 p3 c        return EFI_LOAD_ERROR;
    }
       
! l4 v0 q) m  N3 k% }) n6 z    Status = FileHandle05->SetPosition(FileHandle05, 0);       
4 c/ ]0 H9 \" c+ Q7 P7 M* T# P  x    if (EFI_ERROR(Status)) {
) D8 @: Q1 y/ {" X- L0 r        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
) v, _( t1 g0 @) s8 A9 t        return EFI_SUCCESS;
( j7 u) [2 ]+ f4 N! b    }
9 P+ f/ I) ?" w, \       
+ d/ J* `2 o- D    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
! S. e% |) \" `( `        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
) f3 Q+ D2 o9 D( r. @# d+ U) C        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
$ V7 e. y) d4 x! C* i8 F& w5 D        return EFI_SUCCESS;
    }       

) G5 Z( D: ]& O  l& n4 A8 H    Status = FileHandle05->SetPosition(FileHandle05, 1);       
4 ~% ~, k6 }. r- M  }8 W2 W    if (EFI_ERROR(Status)) {
" |0 l9 X5 u6 W        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
+ {9 Z* K& ^( q2 H" K7 K, S" z        return EFI_SUCCESS;
    }
) e0 r" e& ~3 F, q* b       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
! p0 N! ?' }' K& V2 G        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
  A5 \' j, m7 b: D1 V9 M        return EFI_SUCCESS;
    }               
6 y" b- _7 w. J       
. k; [$ \: K" V# c$ |% S5 b, D: q    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;
    }
3 [$ o, `; N+ f* i       
. M# }5 J" ~* K1 [    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
8 T1 {+ E8 ?6 v; W0 I' @  U$ R2 L        Print (L"File Buffersize is %x\n\n", BufferSize);
2 L- a! ~' \1 @    if (EFI_ERROR(Status)) {
( [. W6 F1 `& @  |        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
8 X  R% C2 {+ `0 R6 j' ~- p        return EFI_SUCCESS;
    }       

//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
$ G2 v* _! n3 C9 C: u    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
  x( @6 ?* y6 ?+ o% `7 |    if (EFI_ERROR(Status)) {
4 |$ u; Q. z+ m$ T        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
% `7 C  v  Y* O6 ]: E
    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
. t1 ^* P& F* r( C0 G2 ~        return EFI_SUCCESS;
    }       
        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
7 b0 Q/ S. Q6 y( U. f/ N% e
    Status = FileHandle05->SetPosition(FileHandle05, 10);       
/ |) a. W  z# q- w- f6 b. e    if (EFI_ERROR(Status)) {
. n6 Y) u8 e" t) }3 m! W  n        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
; q7 F& Z* Z1 c; U8 r        return EFI_SUCCESS;
- |  l5 {7 Q3 x& l    }
# X( O( {3 O; ~) Y- H       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
4 ]9 z( H. E9 B; o/ E. H# e    if (EFI_ERROR(Status)) {
& o/ D( X% I0 [: K; @        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
* I2 X7 |% r7 L* G+ d' }* t! o7 q
0 F/ R3 j6 i2 {& k. }1 C    Status = FileHandle05->SetPosition(FileHandle05, 12);       
6 ?9 t9 Z# ^' `$ F, e; h    if (EFI_ERROR(Status)) {
0 u2 N% N* k; T        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
7 N+ \& e3 b% e) D* x" k1 i" `        return EFI_SUCCESS;
    }
' x, p' R; h% K* T) \       
6 m1 I, R9 j' I& A, q    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
$ z7 \" C; L+ j8 O. F8 K# V0 h    if (EFI_ERROR(Status)) {
# ?) c9 [0 X) A6 g1 B$ [2 A( H- c4 t4 s        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
6 d* C- J1 f$ \/ V        return EFI_SUCCESS;
4 n" H$ R- y2 @1 _    }
/ P, J1 t5 l' K1 G+ e7 V9 @
7 E' u% N& ]- I/ ?& |( S& D" A% g3 ^    Status = FileHandle05->SetPosition(FileHandle05, 14);       
    if (EFI_ERROR(Status)) {
7 [$ B" b& r; X1 P. L( |        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
& k# Z% P# e) L: A7 E        return EFI_SUCCESS;
    }
2 h4 W2 F# ^: a9 p       
1 O: O% q! U# R8 Q! H% k: l6 z- H' u% J    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
1 V' l) N7 ~  V2 p9 G- V" x! W3 T    if (EFI_ERROR(Status)) {
        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)) {
- W) J$ @2 T2 B+ u3 M1 F1 v        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
. E4 ]5 t" ]- `- w       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
% b, X9 x1 }) n' q    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
7 S1 W" `  R( I- u9 n: K5 }        return EFI_SUCCESS;
  F1 @2 V. Q8 _/ {1 ^9 `    }       
$ M# e1 m% A" A# n5 T& g& n
* [# ~: r& b, t& [" D//---------------------------------------------------------------------------------------------------------------------------------------------------------------

    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
! C9 ^& B, G3 X/ O) P        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
# f+ z. D. v& H) Y2 E    }
& |# {/ _8 k1 F       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
) R; a4 b9 A, {8 C# N    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
+ J0 u; d5 A: _# N8 g% G9 P2 D, T    }       
       
9 g; w' e$ O! x+ ?, f  H    FreePool(DestAddr);       
) H5 D1 {- L* T6 I
5 H8 R) p! B& J( o& G$ q    Status = FileHandle05->Close(FileHandle05);
" ]) C- P" h' f- a3 D$ D; v6 @    if (EFI_ERROR(Status)) {
8 k' R0 q4 s, Y5 D2 h        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
1 p2 ]0 F* O" ~& i8 G       
    Print(L"File Name = %s has already been created.\n", FileName);
; ~8 w9 h$ f# u# s& [
    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
% e. Q+ p( `; K    WaitForKeyOrReset();

) f  Y  h' W: D- l% O( T" {    return Status;
7 i5 X+ J0 @$ f1 \+ d}

chenhy

//**********************************************************************
//**********************************************************************
* D9 |% k, p* v! p8 d//** Demo code for PCI configuration space I/O access program **
/ f/ a* s' i' t9 b8 ?//** By Lawrence 2009/11/20 @Taiwan**                          
' v' g( X% @. A- i0 C//**********************************************************************
8 _0 q! a( F; O; L7 Y" Y/ e% k3 B//**********************************************************************

#include "efi.h"
4 H# _  n3 z- w+ b#include "efilib.h"

7 I' L) A5 B/ d" d( G2 f#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001
4 e$ K1 }4 y6 s7 {3 D
8 P; E5 j2 X; r- T$ h8 \static EFI_STATUS WaitForKeyOrReset(VOID)
# h+ p% o* y8 }; @{
: N& |2 e! k8 @& E5 U8 A6 V( ~    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
   
    for(;;) {
1 `5 a9 x# C* I5 X/ @        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)) {
        if (key.ScanCode == SCAN_ESC)
$ n, J+ B9 u) Q. i6 t/ m            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
+ K1 P/ T5 m" \8 H    }
( V) G( c9 u# F  u- R, @2 ^: y   
9 T1 i) c! [; a1 [2 ?- G! s    return Status;
}

UINTN
AccessPciConfig (
3 S: x. a# P; Q1 Y6 }; a- |    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
' r* _1 f' [3 D2 d7 I- Z1 n% ^) }    )
{
: p( j3 J; n, t, ?) {) I        UINT8  i;
        UINTN  PCI_ATTR[4];
# P, z9 R: o' x3 c! }        UINT8  PCI_REG[4], pci_offset=0;

        //get device attr
* l8 }) p& Z+ O' T( a: x1 b        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));       
- J2 y. a, n- f1 U- G; {          //print register value
7 e$ U2 j6 x/ w1 R          for (i = 0;i < 4;i++)
          {
, [% D* ?4 M/ ^; D        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");          
3 F3 m* }8 K  f& L  F7 A          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
: N3 L! s& U4 V7 j2 U          return Found_Me;
9 a1 S" I" _% C4 B% a0 `/ z    }
* ^, m' Q# o0 O/ y+ {( F, @
: q8 a0 d; U& @# W! F  I        return Not_Found_Me;
}

EFI_STATUS
InitializePciApplication (
# w: k3 j( m+ f: G0 j; s+ Z    IN EFI_HANDLE           ImageHandle,
; \* M; @, A+ h5 s; ?- J% ]1 U    IN EFI_SYSTEM_TABLE     *SystemTable
    )
- o5 S3 O7 Y# r8 G; B{
  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
8 ]( M# f, D( U4 W/ ~- [  UINTN                   BufferSize, Result;
2 j8 k1 k0 |7 T1 A' J/ ?/ u; h  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

. N0 r, @. `" y% e! V# T7 P0 O4 e3 h, {    //
    // Initialize the Library.
' R3 X8 _' L- d4 e9 n* S. d    //
- X2 g+ h) V9 R: O& L+ s  C    InitializeLib (ImageHandle, SystemTable);
* X0 f' ~; \" c  b  N1 w    //
    // Find the PCI driver
$ d& T3 t4 S5 M7 D    //
    Handle = NULL;
+ x- c$ X/ l# y    BufferSize = 1;
8 {+ H8 d" A6 T4 H, U    do
9 R( O# w! F2 D% p( B  C- J    {
      if (Handle)
# P( Z& V& d* d' V& f) S0 ^      {
        FreePool (Handle);
5 {/ M- L  R, g7 _; s3 L- R5 K" B      }
0 h' s' [% x1 `% X5 v: D, r               
      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
+ Z2 C0 {) L5 m: K. I
    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));
3 V7 K  i  e/ O/ R/ ^
2 |# ?" ?% \/ d$ P* [/ N    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
. N# r0 Z7 Y$ R" ]    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
    do
    {       
        if(Index == NoHandles - 1)
         Index=0;
: R" u9 c! ~: C          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
7 O/ G) d5 r2 b# e! }1 Y$ f        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
' U% e# Q& X5 i: s9 X+ G4 G* b* X          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
    }
    Index++;               
2 X! Q6 u- j8 O9 b3 H7 Z7 O8 B    } while(Result == Not_Found_Me);
: V5 j$ c: _& d9 |       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
    return EFI_SUCCESS;
3 I. P6 c& W8 b! i! K# e" J* @" F}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习