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

kaurohy

/*
* myfileio.c
; l) l$ r; q. x% ? * Apps
*/

#include "efi.h"
1 ~! G$ i/ ]4 ^* o; q3 J3 B% Y#include "efilib.h"
6 ^& r; ]5 M; ^8 R2 @
#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

6 X7 O8 t4 T# d. Zstatic EFI_STATUS WaitForKeyOrReset(VOID)
/ f  l/ \; y) r, Z: y' r0 b2 a{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
   
( C- i& N7 T- e2 p% H6 \+ _- X: L    for(;;) {
! c, ]! `& n* ~# W, c9 v8 t        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
1 w. Q  `' W; z& h  r            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
3 i, N/ F6 u& N2 q7 @$ k        else
4 p1 W3 B  ~; X, f9 h5 d3 V& c            break;
+ j* u4 O1 O/ u& E/ N" w* {    }
. w/ r! ~1 M& k  {* H# c    if (!EFI_ERROR(Status)) {
6 ?) n/ w* [5 ?; _% F, P0 ]        if (key.ScanCode == SCAN_ESC)
+ l/ o/ t2 p9 G! ^            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
) P0 L6 t0 q$ {( r" \! ?2 j    return Status;
  K* [- k! ]. H& j" v3 E% J$ P}

EFI_STATUS
0 C2 P6 a4 q8 a2 [, k; FEFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
$ U# `" ?+ Q3 T- }/ Q6 j             IN EFI_SYSTEM_TABLE     *SystemTable)
6 ?. |) c) V! X6 K5 n% k{
' [- G, \6 X6 }7 w    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
; N- e2 N& l) ~& M" O" z% Z    EFI_DEVICE_PATH        *FileHandle02;       
: @* V3 d$ q8 ]! N  f; P# u/ ~    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
4 ^! d& b& @' z    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
5 f3 N" K5 P+ `. w4 R  [    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       
7 t- N9 f3 i( {- U3 Z
    FileName = L"NewFile.txt";
7 K* C' ], f, W: o  ~! q+ B        Space_Key = L" ";
( ]/ m7 v+ S2 c* y  M, }/ A       
: V" @0 v5 e0 b" s        BufferA = L"ABCD";
6 w. w' L! B- y* O        BufferB = L"EFGH";
        BufferC = L"IJKL";
" B1 k- L- x) G) U$ O
        Bit0 = 0xff;
! d0 `: I& K* L1 U        Bit1 = 0xfe;
       
3 I( I, K4 a4 K5 l7 n6 O$ o% ^        Key_Space = 0x0020;
# r$ p8 \0 x0 T  A( Y       
6 L$ B! x7 p  Y% R+ {- f' x9 m! a    InitializeLib (ImageHandle, SystemTable);       
9 n& E8 F( ?& _! O
( F1 n( D7 x3 ~* }    DestAddr = AllocatePool (BufferSize);       

    Print (L"Value of Bit0 is %x\n", Bit0);
* _: L, ]4 y( a4 G# B    Print (L"Value of Bit1 is %x\n", Bit1);       

, `0 w+ H: B8 j5 [+ i; x       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
) l  k8 t5 |! ?    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }

$ C& u+ O6 x8 C! N" M    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
4 d+ }& [9 j" ~5 {; x    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
! j, x) b' H5 a! E' {( d            return EFI_LOAD_ERROR;
        }       
7 m' C1 o" J2 @. K( 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;
5 u3 b2 f4 V" c9 G2 G% E        }

% @8 V- [7 K% b" f4 i+ r3 F    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
3 t0 H7 R& L1 D; B6 A2 k  v5 x    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
. ~3 r; i) i! i( J& R, M' E4 y    }               
  o% B6 h' v8 M/ ?% a, J               
* {6 Z. y& d- V+ m1 h6 ^9 r- A    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;
9 X8 a( |! O1 B) B# ]1 k    }
       
    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;
    }
       
% r# C! {2 K' U9 U# A1 h    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
- X9 p6 R, ~5 \    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
! G1 f( j- v# x        return EFI_SUCCESS;
    }       
2 k7 J# ]( u' o. W7 I4 l% I
6 e( }) M' [; A* j- V4 q    Status = FileHandle05->SetPosition(FileHandle05, 1);       
    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
+ B1 \( W5 `3 A0 k( [        return EFI_SUCCESS;
5 `, a9 s! @" ?8 ^    }
& f, E2 W" S$ n% A& h3 M2 u       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
5 A" W' H; `5 G+ y' a: ?        Print (L"File Buffersize is %x\n\n", BufferSize);
2 D+ k% ^8 z/ Z2 [5 `/ w    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
1 |9 ]+ Q6 n" B3 L) Y    }               
0 G# J! a, P% B, `0 z: J6 ?       
, l" a+ x( |% H9 s5 r; A    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
" u/ U, A2 r+ g& ?* k1 \        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
, V3 [$ S. ?1 g- @& l    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
: m$ X8 ^0 J8 @        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
- P- j3 q0 g6 K. r        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
; m+ e/ N+ c4 o' c  s7 p$ Q$ I& |, C2 i        return EFI_SUCCESS;
    }       
: m9 t  H: J, Y; Y$ e; N& H$ J" @
/ v" v  o+ R9 R! @1 g2 }/ c//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
. U$ s! m, K6 h; f& y    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
0 S8 o; d) y" H# Q# c        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
; ^! H8 O' @9 C" Q: X/ z/ o    }

, v! N, X+ B' S" I    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
7 V1 v- p, V1 R6 H    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
2 l( Q: o4 M2 M1 o  c+ K' a' m7 l5 a        return EFI_SUCCESS;
; U' J' g3 }; r( _' k' i    }       
; c5 B. i4 N4 h) t2 u1 {        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
  t' @5 u, ~. P: {! i" F//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

    Status = FileHandle05->SetPosition(FileHandle05, 10);       
# W7 ^4 c* A. _9 {( L    if (EFI_ERROR(Status)) {
6 M) o; s" Q4 o1 v0 [8 }# l" h1 @        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
7 X/ G. c" O. \+ l        return EFI_SUCCESS;
" }6 o/ M# @7 s; V    }
- _. a# \4 u4 k, N, `- q% h       
- K5 i! E* C: r2 n    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
; [5 a4 }( C& q" p2 G% T$ Y' O9 c    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
: N! p! Y# B' X    }
1 l4 l9 \& D' t6 l% t6 K( Z
    Status = FileHandle05->SetPosition(FileHandle05, 12);       
- D) o$ X6 ?3 ?' R" J    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);
    if (EFI_ERROR(Status)) {
+ Z6 H! z  V( Y; d  M( y+ p        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
' K7 B# n/ ]2 g# O0 d: c% W$ a" [) X: @    }
3 N$ J0 h. b& z1 L; w: [
6 e* S( h& `" g    Status = FileHandle05->SetPosition(FileHandle05, 14);       
- Q. Y5 R2 _, R    if (EFI_ERROR(Status)) {
. q$ r2 t+ Z: Z3 I        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
/ }; ~# B. o# x. f7 i8 [! O" t0 c4 T9 L    }
. m8 u# o8 J8 V/ M4 C       
- I' k$ O* h0 O8 O3 F5 J- W    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
; E0 ?7 B: I9 b1 C        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
  \* i0 Y( p6 ^! K1 a) J5 F        return EFI_SUCCESS;
    }

4 W4 V; B* o3 D* Z5 w5 l8 |    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;
    }
# z9 p1 N5 L/ |; Q- O/ f       
0 v# t$ j: t% F: \; _9 s) c! J9 _    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
- D9 v; H4 t; x0 g* @. R        return EFI_SUCCESS;
    }       

//---------------------------------------------------------------------------------------------------------------------------------------------------------------

    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
7 P- X0 U( l) P* ]        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
9 S6 g% ?4 k' I& c. k2 t        return EFI_SUCCESS;
; k0 v4 `6 V# |3 a, {    }
" M( p" `* r0 E* C; w6 c       
5 g- I$ t# w3 m' w' ~$ u8 E    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;
    }       
4 O3 n/ x) m/ j7 i       
. Q+ z: E) X9 ^" k    FreePool(DestAddr);       
3 w3 g$ e$ L2 o% P% b
7 Q) k* T# T2 u3 h4 S    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
$ Q! w, c$ @* J2 U2 ^  Y    }
9 A- e& k# }3 G; a/ {       
    Print(L"File Name = %s has already been created.\n", FileName);
# g" D9 p  A  x) R5 P% ?2 [# F7 K4 H( U
    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
) R) |5 \2 d  F. h& ]) r4 d* _    WaitForKeyOrReset();

' s; C/ p& x' X    return Status;
}

chenhy

//**********************************************************************
//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
6 t% t1 M/ q1 m; T. x//**********************************************************************
, A: g+ J  M: z' N3 \; t$ K8 j//**********************************************************************

#include "efi.h"
#include "efilib.h"
' Y, V/ J; Q# n4 I0 c  S
" r0 W! h, }6 I#define        Not_Found_Me        0x00000000
; L" h5 d, F0 F* u; }" @0 }9 D#define        Found_Me        0x00000001

static EFI_STATUS WaitForKeyOrReset(VOID)
6 T7 k- P$ c  D: P& g{
! r. u' L% p3 K+ s- j8 h; v    EFI_STATUS          Status;
9 S  o3 F! S, D) H& z2 K    EFI_INPUT_KEY       key;
    UINTN               index;
   
    for(;;) {
2 G# ?' A( o6 O* w0 {        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
6 c+ {/ Z& w) o/ l2 S. a5 `  Y+ ~        if (Status == EFI_NOT_READY)
1 U+ `0 T& J7 G4 _* u4 r            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
) ?& Y, U" m3 g1 f        else
( l! |0 F, h/ F. Y. N, b            break;
    }
' \8 f% ?, `; Q, \! \# w    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
2 R3 z: P: o+ ?" Z6 Q4 z* h            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
3 o7 V; Y/ m% Z  y   
/ g' t8 z+ B0 |. g" j    return Status;
}

UINTN
AccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
: N" J5 \: n6 M! |3 Y{
# Y) f" b$ T  }9 U. s5 _        UINT8  i;
$ u, `5 k! z' ~        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;
8 A" k2 P+ H' {
  }) z: r3 z3 d+ H        //get device attr
) A2 m& i& }- x& ~& u/ D        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]);
) f+ l' D9 Y. e  f0 [! X6 a          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
5 V2 k! r1 v  g. X0 y          //print register value
          for (i = 0;i < 4;i++)
          {
. N1 \5 W, l' p        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
          }
* S& _% q0 `0 E+ x# Z7 @! D  s          Print(L"\n");          
( t/ l, M4 K9 O; J4 y5 b          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
$ m/ j- s& J) o; V- Y- F          return Found_Me;
3 m1 o* \" g  a5 Q; _  g    }

& R- ^+ l8 u6 E* E        return Not_Found_Me;
) o# a3 Y- h3 S+ H* |& u}
' Y) Q6 @% }' s/ c" |. |
4 i# `+ p8 F- WEFI_STATUS
InitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
    )
{
  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

    //
$ Y5 r) e! s( G$ o; k" q    // Initialize the Library.
    //
    InitializeLib (ImageHandle, SystemTable);
    //
6 c( a  x% F% V3 J$ w    // Find the PCI driver
- A0 I% t4 N+ [- O4 m7 y/ a) o    //
; `* `, W& t' u" ^    Handle = NULL;
# g5 K. P* X3 Y3 z  J    BufferSize = 1;
' o# [4 u0 Z$ }5 T7 Z! j    do
    {
      if (Handle)
& W& }; F8 E. Q' A' w0 [      {
: r2 }, ?, a; s' E/ H        FreePool (Handle);
* }+ M+ X: L/ s. e0 u      }
5 s0 C) R8 i9 ^4 x( U               
      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
% {+ _, B. A+ `. q! f" L
    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));
; y& N1 p" Y, S$ J' N- z
    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
' {9 v/ `6 K% e/ o4 T7 _% |: F    ST->ConOut->ClearScreen (ST->ConOut);
3 b* h) S+ R* a8 Q    Result = Not_Found_Me;       
    do
, Y/ f: w0 g2 h) {" E1 Q( n    {       
        if(Index == NoHandles - 1)
         Index=0;
          else
# }% b! v4 V9 Q" z  S+ i# T    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
% {! }9 p. x' }8 V, b% x+ m        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
; ^+ g4 P) I7 |( j# @# \% B          Result = AccessPciConfig(PCI_DEVICE_TARGET);
2 Z4 w6 c) Z9 V$ G//-          Print(L"Again now Result is %x \n",Result);          
1 N3 M, W) t# v  ?2 w    }
0 }4 D0 Y: Y3 k# ?( G% F2 h3 R    Index++;               
    } while(Result == Not_Found_Me);
$ @1 ~- [1 t, W       
        Print(L"Please hit any key to exit this image.");       
; R0 D3 R: c) t+ s" K  l    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
) Z, _+ _0 o& q' n- d    return EFI_SUCCESS;
9 e# c7 A, l! D3 X}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习