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

kaurohy

/*
% ~9 z7 W7 a7 O7 T9 y0 T5 Y' W4 U/ x * myfileio.c
8 v/ b. d, D5 f% H  V9 p9 b0 _ * Apps
- m8 x( z1 u& `, s1 N+ F% E3 m */

#include "efi.h"
  `1 V, e; U- C' ]$ m1 i#include "efilib.h"
1 c! Z- K7 S% {
- j. f: i/ x% ?/ ~3 c#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

& j& ~. S- a+ K" [2 j2 y+ [: gstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
   
! e. _2 a7 S) x    for(;;) {
7 `0 t: C; }: a0 s        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
7 }$ s' X4 J) p- Y$ y4 l4 V' Z            break;
    }
8 @* ]5 J# Y1 i% ?9 f2 K+ T& Y    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
9 R. j7 C; \1 W7 K    }
   
' \0 }) ^2 O' [/ X, ]+ K6 h: N* i    return Status;
( R4 I+ e/ G& S5 t' H+ {  H}

; ~, i; `6 T% G% HEFI_STATUS
& y! N1 R* O& PEFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
6 x6 i' Q* N0 c) r; u& k( K{
    EFI_STATUS                Status;
1 B3 I- \* b& }) v! U6 Q* C    EFI_HANDLE                *DestAddr;       
- q# F% F% q- H5 \0 V    EFI_LOADED_IMAGE        *FileHandle01;
' A. j# h& j' F+ r    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
  ]( W- }+ X* q( @/ A    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
* B% [6 A. b, l* p" r. a    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
7 Z7 e( }4 x+ K  Z" x    int         Bit0, Bit1, Key_Space;       
4 M- x* C4 I- N
; c' R0 N. r' f    FileName = L"NewFile.txt";
9 N/ J) _0 X  W" H( ~        Space_Key = L" ";
       
- F* y- O- t, y, e! h        BufferA = L"ABCD";
0 Q+ b; i% r* ?+ M9 o) r3 J7 j        BufferB = L"EFGH";
        BufferC = L"IJKL";
" g8 c" e; f, U2 n. G' }, u7 p
        Bit0 = 0xff;
        Bit1 = 0xfe;
7 z$ ~/ N% c8 f( Y8 x& Y       
        Key_Space = 0x0020;
6 @* M( _2 @, ~       
    InitializeLib (ImageHandle, SystemTable);       
) d8 v. ~, }" U1 T& y
    DestAddr = AllocatePool (BufferSize);       
/ n9 N  v, Z+ I+ R0 \) z# G
% `/ G/ k& I8 g4 N+ g+ b: i    Print (L"Value of Bit0 is %x\n", Bit0);
$ [- g# e1 G2 u; e& g    Print (L"Value of Bit1 is %x\n", Bit1);       

7 d3 }9 m2 X7 n7 |+ ]       
5 O" [" C8 x0 v2 c    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
" m' t# t9 k* A( P! ^% H2 x    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }

5 Y5 O# N1 l- `( |7 H% n% @, U, f    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;
        }       
       
7 |6 {4 s* t8 c! z" h    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;
        }

( a) u+ w7 |( P! b* q+ c    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
, ^: E0 K! x; c$ I; L. g" z    if (EFI_ERROR(Status)) {
% {; C8 d9 q# L. Z: Q7 J        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }               
               
    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
3 ?+ f: h! e% b    if (EFI_ERROR(Status)) {
        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
+ K" @" J7 X% s0 t" B        return EFI_LOAD_ERROR;
    }
       
    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;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
  c1 n; `( f0 O* @    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       
! R- p" ?: C/ ?8 m! A( p
    Status = FileHandle05->SetPosition(FileHandle05, 1);       
5 P: o# w+ z5 B' c5 k    if (EFI_ERROR(Status)) {
$ Z7 @/ J# v# i, N2 ~        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
6 C) I2 J! P' N) u$ Y# A    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
  J1 E5 _$ P  q% c7 D        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
5 m2 Z' w. ~; l/ s# m7 f        return EFI_SUCCESS;
    }               
       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
! ]$ r" ^: g9 s' ^, N0 F: {    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
4 @0 n9 c5 \, t$ u5 ~        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
" D9 p' Q! r$ ^; A/ f        Print (L"File Buffersize is %x\n\n", BufferSize);
1 p8 D  c$ h7 Y+ S0 M' {    if (EFI_ERROR(Status)) {
' c; `: t4 F1 x# B5 u. A( \        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
7 L& J4 z, ~: O3 t8 f        return EFI_SUCCESS;
( A  r7 u  Z( ?" W  x+ P) A. l& T' ~    }       

- N( o& \  P0 z* ?3 ?# K( K. Y//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
  D& N! J/ K* \9 i    if (EFI_ERROR(Status)) {
5 E8 K1 c3 M2 o0 X8 O        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
0 ~, a7 H, P% i& |0 H' ^' W% E        return EFI_SUCCESS;
- c9 D! ~# t9 K- v5 s( t    }

; w2 j1 v# @6 N' O    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;
6 l" ~7 }9 Q, K) g    }       
* @: H# e1 |8 J' B% x# D+ p# p        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
& {* A% B1 @7 D2 {$ t( E9 t, Y//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

    Status = FileHandle05->SetPosition(FileHandle05, 10);       
4 O0 j0 Q  ^% T9 B    if (EFI_ERROR(Status)) {
/ M4 G. m" X  y        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
6 P/ L+ L+ f% K( V) [; r8 `        return EFI_SUCCESS;
    }
9 q5 |1 O& C6 H  E" o, ~       
% i% W+ o) v5 Z3 m& L  t: X( V    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
1 v" C% F, L; B7 w        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
  i, S$ j+ u! l+ R5 ^8 X
    Status = FileHandle05->SetPosition(FileHandle05, 12);       
: q" S2 `* a: U8 E9 F6 u1 g- v1 |    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
/ M7 u9 M, B4 t* K6 R        return EFI_SUCCESS;
    }
  e# A' z- f4 S       
/ @, Y. ]' u/ I' H3 S' @8 D    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
, V2 \; U5 m2 ~7 G: g0 K7 ^) D    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
' x/ [1 F8 R: e! T6 Q        return EFI_SUCCESS;
+ a8 T+ H" m1 V& u! x    }
* \+ O' U. P0 z+ H' l
, t8 N* b2 T% z6 s0 w    Status = FileHandle05->SetPosition(FileHandle05, 14);       
+ [( Q1 I) u8 J4 f2 M    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
2 ?' p( X3 k0 X0 @/ i    }
+ t) T* X# D, A; A8 u       
' u* Y8 Y$ \2 K& u7 A% O2 }" a    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
5 H$ i& b7 r( i6 a, y' @& D+ s  a    if (EFI_ERROR(Status)) {
0 p$ a+ C$ v- A4 b! H( Q' g: o        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
) I2 H; Q! V, v5 ]( y6 f
# u2 d7 N% b; p) M7 E    Status = FileHandle05->SetPosition(FileHandle05, 16);       
    if (EFI_ERROR(Status)) {
2 q. S" F+ U# E        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
3 v! e/ Y# v+ I; B4 a2 ~) |    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
, s- y8 y+ m$ T2 H3 \7 T/ g    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
' w2 Y2 ~; P  G    }       

//---------------------------------------------------------------------------------------------------------------------------------------------------------------
5 g/ w: \$ ]5 [) w6 c' D8 ]4 s
% D3 B" ~5 t0 B" B6 Q1 x' x. B, g    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
* `7 h  Y" e' }        return EFI_SUCCESS;
    }
       
$ E7 _* |# @9 @* z1 V' a: P1 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;
    }       
+ C+ y$ a9 a# W       
    FreePool(DestAddr);       

: r9 f7 F* ]" L9 w2 e    Status = FileHandle05->Close(FileHandle05);
4 ~# E9 N. E- w  T" Y( I    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
2 g/ ~0 c/ L7 b$ Q, [" k) t        return EFI_SUCCESS;
5 h' G1 X) `  S; ~6 }. z    }
8 l, {- n* R6 l       
    Print(L"File Name = %s has already been created.\n", FileName);

. o  b9 H, X* ~8 G  e/ z% V    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();

    return Status;
5 B! @3 A# m: F1 c$ l1 K}

chenhy

//**********************************************************************
. I0 k" _; f& c% n5 F- l! V//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
//**********************************************************************
; ]# s2 l+ |. w( b1 ~" A
#include "efi.h"
#include "efilib.h"

6 E' t% p" E7 Q) o+ F4 O#define        Not_Found_Me        0x00000000
* p4 }0 g3 ~  H3 X' Q#define        Found_Me        0x00000001

static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
" s* _; I9 g4 T! ^7 T% L: `+ d   
) Q  S- }. R+ ?' w3 t    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
$ ~! Q/ B  j* v0 r- g9 o  \+ Q2 G        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
! k3 I1 q  I4 W& M, C) {/ [        else
            break;
    }
3 \6 ^6 i' P7 h6 {    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
5 c: I& g) m# F& F! v   
, x; W# {' e2 u9 j" x    return Status;
}
9 L* W6 w  _( _7 d& c1 V
UINTN
8 y6 u4 Z8 N' p. V& }& }AccessPciConfig (
* E- Y& j! G: g2 A    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
# [9 z. B: R3 a$ v2 N! U0 r    )
{
* p- @% s; w- T$ f& K$ B( P% R        UINT8  i;
0 r9 W! |  [6 L5 ?        UINTN  PCI_ATTR[4];
7 o( }" [  ]: m/ C& B9 D) K7 c' I        UINT8  PCI_REG[4], pci_offset=0;
  m6 B# C2 t. ]1 {) J6 Z" S
; S  e5 @$ `  T, N        //get device attr
        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
: n" Q. f% F9 h+ Z* k" L    if ((PCI_ATTR[1] == 0x00) && (PCI_ATTR[2] == 0x1e) && (PCI_ATTR[3] == 0x00)){
% d- X- D. \$ }4 `  |: l          Print (L"Device found at BUS = %02x         DEV= %02x        FUN= %02x  \n",PCI_ATTR[1],PCI_ATTR[2],PCI_ATTR[3]);
' B) S) q% ^: ^' L2 u          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
          //print register value
          for (i = 0;i < 4;i++)
          {
+ i3 U- `4 Z' u; L' J8 C  d        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
+ i- D* @% T6 J5 v                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
          }
% b+ f( t3 e, \2 d& P4 m) W          Print(L"\n");          
7 Z: R( s+ `/ \# d3 D+ N1 y& A          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
9 V' q  O5 s' ^% O6 V    }

! M8 J* [, @5 I        return Not_Found_Me;
2 X  Y- k5 A0 c  ]5 @, T}
3 p( e# b- P& C
; F" M6 {2 Y4 {5 QEFI_STATUS
InitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
    )
, r3 T9 ~; d7 q) Q9 M{
; L& O# W, B, X4 R6 t6 J, @# d  EFI_STATUS              Status;
- N: G+ |8 P6 k5 Y4 U! J  EFI_HANDLE              *Handle;
+ p6 w4 F  J0 `' v* v  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
# p( p: T- Z! H% t  |. ]/ _# h8 O  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

    //
    // Initialize the Library.
    //
    InitializeLib (ImageHandle, SystemTable);
    //
    // Find the PCI driver
1 a, n% w. k1 f- n5 E8 b3 C    //
    Handle = NULL;
    BufferSize = 1;
3 r# g# p9 _& Q$ `$ }! N( a    do
' J8 @. p  m9 W7 y; {8 Q% C! u$ m; a    {
      if (Handle)
      {
' n  p( ?6 J7 j. a& L+ L        FreePool (Handle);
& J4 v) P' C% \$ u) ^5 ?) Y      }
               
6 t* n) s( W. K9 H$ H      Handle = AllocatePool (BufferSize);
& {, P: i" G/ S/ I% T      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
2 R/ m9 u3 S7 h
    } while (Status == EFI_BUFFER_TOO_SMALL);
) P. Z' n" ~' W& o    ASSERT (!EFI_ERROR(Status));
) T* ]) I) ?  P
6 y- _3 ~9 p/ i' ^$ g3 g. h    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
7 O; x# R5 A6 z% V    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
0 e" R- a/ P. }) z# V1 S7 m    Result = Not_Found_Me;       
    do
    {       
, u, E6 b, E7 Q8 r" h# B. O        if(Index == NoHandles - 1)
8 o8 m3 t4 P% e# {& v7 B" Y9 Q9 h         Index=0;
          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
! O& Y; L* m) z5 k2 c# g    if (!EFI_ERROR(Status))
( n# N8 H0 ?: G6 k! T! G    {
5 h) y& m7 o1 }          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
    }
- {4 Y  I7 }$ L( h. b9 e. H    Index++;               
    } while(Result == Not_Found_Me);
. [3 [5 j' N- c/ H0 R       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
4 I& {3 Q- u+ T* x& y$ `( C    return EFI_SUCCESS;
5 q( r& w* w( b/ N}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习