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

kaurohy

/*
* myfileio.c
( A4 d. ~0 M5 @( `4 x * Apps
*/

#include "efi.h"
#include "efilib.h"
* m9 O- h, i' T- O% o' _
#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
" K5 F. I/ e2 B8 C- u
static EFI_STATUS WaitForKeyOrReset(VOID)
! m" h! i. u$ ~' {. |2 a{
    EFI_STATUS          Status;
/ Q% C! o7 {5 [    EFI_INPUT_KEY       key;
    UINTN               index;
- {3 D5 L0 h2 x% {- Y   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
: S  t5 e' V) U7 K9 J3 v& a* S# d            break;
    }
' i$ J/ b$ l6 k    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
7 ~! Z( O: U; b' M            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
! [1 [) u$ Y4 G% D4 i    return Status;
7 t+ ]+ V8 C! ]3 Y+ ~  z4 R6 B}

EFI_STATUS
( S& K/ D% H( M* c; ?: @$ IEFIAPI
: O( \9 U9 n9 l9 oMyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
{
    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
6 k6 z* o1 k3 S4 i; l# l, Y    EFI_DEVICE_PATH        *FileHandle02;       
9 _7 G9 F- ?4 J7 Z( y; C! H% o1 z    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
/ m4 ]" S! a  |* v0 B        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
) L, K5 B6 X6 `5 V9 i7 F    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
: `. a6 K" t$ i4 v! i    int         Bit0, Bit1, Key_Space;       

    FileName = L"NewFile.txt";
        Space_Key = L" ";
       
        BufferA = L"ABCD";
        BufferB = L"EFGH";
        BufferC = L"IJKL";

        Bit0 = 0xff;
        Bit1 = 0xfe;
# ?, z3 E6 \4 m/ e5 l6 C       
' F- a! ~/ c9 l        Key_Space = 0x0020;
       
    InitializeLib (ImageHandle, SystemTable);       

    DestAddr = AllocatePool (BufferSize);       
* k' z/ h, f) O( ?+ J
. L. s& W' |  w( n, h8 i1 |    Print (L"Value of Bit0 is %x\n", Bit0);
    Print (L"Value of Bit1 is %x\n", Bit1);       

       
# v5 v$ V; m! m- t) m: t    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
1 v" i9 h+ R* H6 t    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
  i+ i% ^- q* W" N        return EFI_LOAD_ERROR;
. b/ ^# y# ^- H! v6 n  d    }

, r! B* k$ Y  Y9 @    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
  K. k& ^9 `) ~# M% ~! m" @+ k* M# p3 R    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
+ C& A- E  i1 y        }       
' Z: T5 _& F1 Q' A       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
6 _7 l+ T- x$ a            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
2 ?7 Z: M  f5 W; }2 z+ R! }            return EFI_LOAD_ERROR;
3 ^& L" Q% J6 |$ v; m: a$ E* ^; D9 i        }

    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
' x: [& l: `3 C# Z+ U. J    if (EFI_ERROR(Status)) {
" v. S9 I. o" Z6 Z/ L# c/ O' d8 S8 ]        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
9 y' T# b, p5 j  M5 a; u+ \" p    }               
! \1 b! h: H3 k8 k9 V               
% Q1 I% B  g/ ?* j    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
    if (EFI_ERROR(Status)) {
1 T( Y# [" Y/ {: t1 Q* \0 M        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
4 R5 [1 D0 N7 V$ q        return EFI_LOAD_ERROR;
- ^* \. ], I1 @0 M4 _! g    }
; T' |8 [/ t, A  R: X* A8 D       
    Status = FileHandle05->SetPosition(FileHandle05, 0);       
/ f$ Z% \# u& `7 \    if (EFI_ERROR(Status)) {
1 j, ?% P7 ~" E3 e9 h! n! m        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
) O0 ?5 a; ]2 Y4 O        return EFI_SUCCESS;
    }
# n* T/ H$ ^  A' b# m       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
& u3 @+ I& l9 |/ l        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
! @' d! l' w$ U0 j. Y1 x        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
0 W+ X6 P& T) |3 L* z/ G' R, y    }       
  ^8 y" R( S$ j
    Status = FileHandle05->SetPosition(FileHandle05, 1);       
    if (EFI_ERROR(Status)) {
- B. _' B% w5 J% Z9 ^        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
) m  N! J# v7 B  C       
0 v) B% [& b) o$ j, I    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
2 j4 ?6 `8 F% T  u% f) Q# ?        Print (L"File Buffersize is %x\n\n", BufferSize);
4 [) K1 r& h4 f) ]5 B* H0 F    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
3 g# Z; ~" c+ O1 q        return EFI_SUCCESS;
    }               
       
0 c3 u9 G  Y7 i' E6 X6 N/ j+ s8 F    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
% H$ G& R( P. ?* g( C) A        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
/ l7 W: H( n+ ~" a" J; ?& y        Print (L"File Buffersize is %x\n\n", BufferSize);
' U* \6 U( A/ K6 I    if (EFI_ERROR(Status)) {
: k  c! q/ z& h7 V; e: ~        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       

; L# E4 r4 J+ B//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
2 O/ h. x* X/ R    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
* K! [! y/ a5 J    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
" c8 g: ^1 [8 U        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
. b' l9 ?. h# S$ o& t        return EFI_SUCCESS;
    }
/ D( m3 S1 S9 g8 }3 K6 V* y+ ?' ?  d
    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
4 Y! ]( ]6 u2 }/ \4 [    if (EFI_ERROR(Status)) {
9 q# n* q6 ]; n: a* G2 O8 ~$ X4 m        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        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 U8 j" |1 |+ q% s3 x% Y//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
! Z: a3 Y$ |4 r) k$ h
; C1 u2 l0 Y% e" u' F" d: w    Status = FileHandle05->SetPosition(FileHandle05, 10);       
, g% ]9 U+ A5 u6 f+ x    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
8 a0 r. f) Y( j! ?8 B) O9 Z    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
! U9 l. O9 h4 N' A$ Q; P        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
6 W. K' J1 |( \3 \  h    }

9 x2 a1 r; q  V2 T& p    Status = FileHandle05->SetPosition(FileHandle05, 12);       
: D' T1 n8 X: @$ N5 t2 S    if (EFI_ERROR(Status)) {
/ }! c" l2 F$ C3 ~1 ?8 u        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
8 ]5 i& l1 o5 \4 L: a$ w5 N        return EFI_SUCCESS;
    }
       
1 k! m/ r, f9 ^& G8 c5 }# ?! R$ H    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
/ n) Y9 U7 O$ W5 {( e        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
. Y! A& {6 n  K2 w2 j4 N        return EFI_SUCCESS;
6 L  @: z& Y5 ?& u    }

- d% |% b/ h  F$ A" Y& d$ [' F/ ?: R    Status = FileHandle05->SetPosition(FileHandle05, 14);       
- o/ n% x1 i9 ~9 t& ?/ g: a    if (EFI_ERROR(Status)) {
8 R# K1 a! i2 T1 O2 i4 q# s        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
+ B/ q! T0 w- I$ {    }
( q9 \) I  G  q" s% S0 |& E       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
, r. p! N  Q6 k2 [  }7 D2 d2 q    if (EFI_ERROR(Status)) {
+ ]- x2 o) C2 n        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
9 H# j/ c; ~" @& c+ s- n        return EFI_SUCCESS;
    }
% k. s! i* H; ]+ ~% u
6 }6 d: O" ?* o( n    Status = FileHandle05->SetPosition(FileHandle05, 16);       
) X/ A* O5 p6 Y8 {+ b    if (EFI_ERROR(Status)) {
$ k! |/ D7 f5 M/ t7 v) {' g* m        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
9 q7 l& l  O" B4 R       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
5 c0 N$ r9 T; E6 b9 ~+ ]    if (EFI_ERROR(Status)) {
, f1 J2 g2 y8 j. x: H8 ]0 c        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
: p. H# ~( I1 j( i/ a9 ?        return EFI_SUCCESS;
3 n2 C. z5 d& {$ z1 t8 i    }       

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

  z- d3 [# L3 ^  D' Z, Y4 ?    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
% N7 }  q% Z6 h) s3 E( `# ]# }        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
& b- w' }9 f& y; b. p        return EFI_SUCCESS;
    }
       
! }0 F9 o9 [. `# L8 _2 K4 {    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
: `5 E' i7 e5 f. V7 K        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
       
' a# j# m6 N9 f; l    FreePool(DestAddr);       

5 ?, |8 n  W9 y% [, h    Status = FileHandle05->Close(FileHandle05);
, u: m, u7 Y  K( l( f  f) T1 n( r! o, F    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
7 o- ^- Z# V9 ?/ Z' C       
    Print(L"File Name = %s has already been created.\n", FileName);

    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();
6 m. g7 f. q, m1 {0 o
  k% x* g5 ]0 m/ [4 V    return Status;
3 g& p! O- @$ y6 u, f# ]( G& c}

chenhy

//**********************************************************************
//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
  }3 R* C, h1 {5 w//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
+ ^5 E1 A: p- J9 P! A//**********************************************************************
1 F0 O/ f! w+ U9 P5 A
& `* Y8 ?# S, P5 M& ^, k#include "efi.h"
#include "efilib.h"
0 d, l8 w5 u# d5 ?/ x& h2 F
#define        Not_Found_Me        0x00000000
  x' U7 |6 z% J0 |, Q- H#define        Found_Me        0x00000001

static EFI_STATUS WaitForKeyOrReset(VOID)
{
4 p, ]3 u& T8 _% ~, o: v6 F    EFI_STATUS          Status;
5 A% W- z; h" Y+ U    EFI_INPUT_KEY       key;
    UINTN               index;
) i: ]* X$ t7 A0 p6 Y& R   
& F. T' ]' {. B3 ]* B- R6 r$ ^+ o$ m    for(;;) {
8 j; ~% ^5 z, w! L+ q9 O        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
' q) f0 G# V- r8 b  Z! o$ ]% J; h, c5 @        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
6 S7 V6 t( ~( A9 ~! z5 u            break;
    }
0 Q, v3 w" L- }7 F    if (!EFI_ERROR(Status)) {
  s5 C9 ]% m  k4 y1 h        if (key.ScanCode == SCAN_ESC)
3 ^* I' \& U" l# H; V3 c  u) p            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
; A0 Y6 s  H5 f" ?' d' d4 ~    return Status;
. t! a. e  h; u, a. l$ m+ m" C/ `2 i}
6 K/ {0 T1 C+ W/ U' u1 k
UINTN
% q1 d4 x' R# I7 d: }* t8 b2 oAccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
{
        UINT8  i;
        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;

0 _/ c- W7 [2 t0 P! @        //get device attr
8 R9 [7 U/ ^' F$ l4 t) C        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]);
6 k/ [/ V6 `; B          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
0 [: W) M7 Q* B          //print register value
* G& J# ]5 ^5 a3 \- V          for (i = 0;i < 4;i++)
/ o% H0 ^" W4 Z          {
        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
9 |9 d! C6 n: T3 g                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
- }& W* T" ?5 H* R# U* j0 }2 C          }
          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
    }

        return Not_Found_Me;
7 Y" Q0 ^% l9 \; E% j}

) u# O3 O/ z( l7 `& D+ oEFI_STATUS
8 B5 H* h) J) W# F4 d6 U- @2 N& CInitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
4 z0 U8 z3 W8 v$ e6 W; {- y    )
8 \/ `. u7 ~0 _  I2 f* d# I( c/ N" I{
% R3 ]) b. K; b  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
* S4 g  e/ A4 t: m9 F5 d& y  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
$ E0 |6 g7 J$ J$ G# D, P
, E8 t' _9 X; w& E, I    //
    // Initialize the Library.
: }  j2 z$ \5 g3 Z5 |    //
; H. ^# Q" f% x: `. O% @5 D7 M$ d    InitializeLib (ImageHandle, SystemTable);
    //
: X5 r# U; k, M) c6 H$ k8 R    // Find the PCI driver
2 y- s" m: E+ r7 L( I, y    //
  p8 m8 R& j8 _  W/ w# O  }    Handle = NULL;
    BufferSize = 1;
    do
    {
* J, L% H3 D9 a5 k) ?1 a      if (Handle)
      {
        FreePool (Handle);
      }
               
( \- O( ~, m! q      Handle = AllocatePool (BufferSize);
6 i; \; x( D. S" K5 B      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));
& T8 D% N- D3 B, M8 P
    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
    do
    {       
        if(Index == NoHandles - 1)
4 ?# m0 }. R6 w+ ~- ?         Index=0;
: s  \$ I; n$ f. X+ `* O          else
; Q6 ^) d7 C' J5 V7 D. ~$ d    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
$ U* [1 y+ Z' e' K) n, _) R* X        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
& v2 _1 G1 b# g9 E          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
    }
    Index++;               
    } while(Result == Not_Found_Me);
       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
0 |, r% D# ~  [               
    return EFI_SUCCESS;
6 n" W# ~8 {& `$ \}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习