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

kaurohy

/*
* myfileio.c
9 i9 c: t. S1 p- U * Apps
& s0 B2 Y- V' z8 _% M */
9 `- P5 _' r7 a
1 H- P) E0 H4 }* }" |) Y6 L#include "efi.h"
#include "efilib.h"

#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

static EFI_STATUS WaitForKeyOrReset(VOID)
. ~8 N, i& h" o{
- K# c  X9 f7 D2 }& {$ o5 L" ^    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
5 N% g; ?$ ~" h( _5 l0 \# Y. _   
    for(;;) {
6 |; e" g+ m8 H% m( y- C        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
    }
& {0 O0 r6 D6 I3 p3 F! |+ D( t3 i. L/ V    if (!EFI_ERROR(Status)) {
4 L2 `/ \( j* E, D8 r3 S        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
9 x. B& E4 Y1 b' A4 W    }
6 E1 n8 K, ?# \$ O   
    return Status;
}
7 ~5 {% @2 A1 o" S
6 |* K6 h3 j" r6 x) I- n1 |EFI_STATUS
3 u" a# ?2 V( L# w% J6 P+ HEFIAPI
* O9 E% |' l: `, R' E3 j5 LMyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
{
7 Y) P1 d2 d  S: z7 p9 H4 D- i) g: H    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
3 B4 X# x) ~! S) h    EFI_LOADED_IMAGE        *FileHandle01;
: {* o, _$ s9 h, o% A5 m" V    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
3 ^3 v' _/ p- f* y# N: ^    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       
0 q; f8 o, x  g) v. ~
    FileName = L"NewFile.txt";
/ j; o" u1 `+ b% [2 |        Space_Key = L" ";
, \5 k2 P  }- ]# k) Z+ u       
7 e; r3 D; N5 `; f& b        BufferA = L"ABCD";
5 k( U  `. i( m        BufferB = L"EFGH";
        BufferC = L"IJKL";
# v9 t4 A8 b; F" q
        Bit0 = 0xff;
        Bit1 = 0xfe;
       
+ ~- y- x. B/ X9 J        Key_Space = 0x0020;
; z3 [) b5 g7 p. _5 U       
- ~, F5 H) t, @7 ]$ }    InitializeLib (ImageHandle, SystemTable);       
9 w; R( o6 `0 @6 S
% x) M) n* [; ]& H# f- j    DestAddr = AllocatePool (BufferSize);       
1 q! w7 F8 y1 [6 b; c+ G$ V
5 I% g* E- t) p; T: H2 r    Print (L"Value of Bit0 is %x\n", Bit0);
    Print (L"Value of Bit1 is %x\n", Bit1);       

) R1 n3 ]" c6 l8 H1 l' F       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
. I& p+ E1 L: S( S" D        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
: }3 h+ \9 n* M        return EFI_LOAD_ERROR;
    }

    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;
5 ]7 j7 Z# V% r) w* T        }       
       
4 R7 q: o" o5 F2 V9 F    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
; B' P6 M% T/ ^! A9 ?: }) I- x- [            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
0 f% b( [% ~3 J$ X2 o        }

/ |1 T: B& d# {    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
/ t5 G: u  s' h! H- h        return EFI_LOAD_ERROR;
    }               
: P0 A: [& e. v4 q" `+ @% P5 _               
    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);
8 Y) E1 _% u' A. g: D, X        return EFI_LOAD_ERROR;
$ z' e$ x% Y5 q  S) h6 m' G    }
       
3 J( F% R1 ^) ~" a; z5 ?    Status = FileHandle05->SetPosition(FileHandle05, 0);       
. M- N! R  n' ]9 J. {1 n2 q& W& D    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
- }4 {& N6 X2 D# Z        return EFI_SUCCESS;
2 |  V1 v; P# L: B    }
9 P* l6 D; M& y8 Q       
/ p) ^: f: l6 S5 `% k    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);
& F. i8 T/ f3 _$ l, z        return EFI_SUCCESS;
    }       

    Status = FileHandle05->SetPosition(FileHandle05, 1);       
/ V2 t' B' j8 H1 q; D" ?$ T; w0 q    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
# g" f! x* n% A, E6 i        return EFI_SUCCESS;
1 F$ J4 p" q, j) c* b+ L& s    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
' z4 J' ~) m3 y' [, `, H    if (EFI_ERROR(Status)) {
3 @1 }6 Z+ }/ o        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
- b$ X0 S( m% p% o$ F& N+ i        return EFI_SUCCESS;
    }               
: V& R, h2 k9 D5 i       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
1 k2 f, @( A0 w4 t& ^0 _1 N! U3 @        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
2 f, r8 b. J& D- b8 M3 m        return EFI_SUCCESS;
* w6 S4 K  {; d" @' s; T( {    }
       
9 Z# t6 T/ a: \0 L    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
        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);
1 [3 b" w. f9 {. s* _* l4 u        return EFI_SUCCESS;
    }       
3 R8 p  {1 k# U8 O, @
4 c9 |' _/ N* M) p//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
7 \! j5 ]" F& J/ W9 p) P! p! X+ i3 S    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
( s# Y, d" F+ c( }% a  l6 Q/ p( b    Status = FileHandle05->SetPosition(FileHandle05, 2);       
: G) n% x9 J: W  q7 N, V2 O    if (EFI_ERROR(Status)) {
+ ^, A8 f1 h4 P6 D        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
: Y5 ]1 d9 `( j: {. l        return EFI_SUCCESS;
  B) O' X: L: C' S    }
: [8 l8 |: F5 g$ u- ]2 M
0 [; v% A9 e/ S3 w    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
( s. O9 C) z/ n        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);
& h6 N) d1 i8 f/ R" @) T; B( T& W//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

" a. ~  C6 @$ U" O7 f0 ]    Status = FileHandle05->SetPosition(FileHandle05, 10);       
    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)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
+ o$ z. f) ~7 [, j2 E    }

    Status = FileHandle05->SetPosition(FileHandle05, 12);       
7 I$ o. I$ H, Z1 X& B4 g/ e    if (EFI_ERROR(Status)) {
8 f+ C) p8 U; ~4 s. s% [/ |        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
! W" H# |: V+ z1 K; D1 I8 Q        return EFI_SUCCESS;
# A4 ^/ C( X+ H* T+ o    }
) a) B1 H7 o% g       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
& g; c( e8 i% y7 y8 U1 E    }
0 F! Y% l; e; P4 g  O' g) R
! J+ J* R4 v6 `; |0 s, x7 W$ e    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;
, `8 i. T/ j2 q/ A" R  |    }
: P) l' f' `. u6 Z9 F       
- G0 t# ]: \4 l    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
( R" e/ J7 d! k    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
6 k& `7 q2 S& ^! O4 b
! P; T9 P6 ~6 D7 `, p  \6 P    Status = FileHandle05->SetPosition(FileHandle05, 16);       
) P% |3 f3 r( s8 h4 L    if (EFI_ERROR(Status)) {
4 E6 `7 |: t# X2 l; C        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
; q/ B* T2 Q5 H/ h. o       
9 d' ~" `* o% s* r$ P6 m    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       

//---------------------------------------------------------------------------------------------------------------------------------------------------------------
6 T; g/ f" w) G: v" ?; s; z, ?
    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
/ I  |  `3 `( R- {* [        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
  A! j9 T- V; c. w5 R$ z1 e5 Q    }
6 T% s. w5 O) H5 \7 _       
    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;
    }       
       
& f- d4 W3 L/ q    FreePool(DestAddr);       

& ^% T, J; d. v7 x    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
1 w* F# A2 b% h6 A! h. C        return EFI_SUCCESS;
/ f/ F- O/ E7 T; g  ?4 ]1 j    }
: f# p: x6 E, D& @8 X; G! r# P       
2 C# X/ b+ N% Y3 G1 Y5 }3 ^    Print(L"File Name = %s has already been created.\n", FileName);
* x/ z- O, Y! O- }% m
    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
/ c9 r( S# z2 t- }% D. e    WaitForKeyOrReset();
: X+ a' Q& D  E  L
, Y' y( t7 Q* c7 ~# }0 {: l) t3 \    return Status;
; _# }& v; k) S  J% n' Q}

chenhy

//**********************************************************************
//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
( w: e. `8 v) h//**********************************************************************
0 i1 [! r' j8 N4 k! p//**********************************************************************

0 b! [9 V; G) Z, h7 \#include "efi.h"
#include "efilib.h"
3 t  R9 s# C* S0 L- V# C1 I
/ x% Q7 H2 I# O, b/ P7 i0 m$ W#define        Not_Found_Me        0x00000000
- S/ W: \1 s  D8 f( p& y#define        Found_Me        0x00000001

$ U1 j& f  A( I' |2 Z! Sstatic EFI_STATUS WaitForKeyOrReset(VOID)
4 d1 I4 y& C4 W9 i0 L' f) k7 L{
  {7 F% w$ X, Y6 ~4 w/ J% @    EFI_STATUS          Status;
; J0 n( I, a# s    EFI_INPUT_KEY       key;
    UINTN               index;
4 h  E1 k+ ^. ^/ [   
" V8 n9 }  c% s8 l    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
- H# r3 Q% X" a  g8 `, v' I8 M: y        if (Status == EFI_NOT_READY)
0 |% _- {* z, s. T# x            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
8 K9 D# \; m) _) ~! P; l% L: ~6 S            break;
    }
    if (!EFI_ERROR(Status)) {
$ s. v8 @0 d1 P( E! n4 d        if (key.ScanCode == SCAN_ESC)
$ |  |9 f# S- F            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
    return Status;
/ x& O8 i. |/ f/ \8 u/ f}
/ x) G7 Q; E4 D
UINTN
AccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
1 ]! g/ J. b) ~$ i5 f2 T# E- u    )
: o, G, q% @7 \9 X* z{
        UINT8  i;
3 N2 n2 `7 I1 F0 p( g        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;
$ E' Y5 o, ]0 H' a  E1 t+ d/ |  x
  r, k. ^( \4 m% O1 C1 o/ {, l        //get device attr
        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
9 q" f5 p- c3 {8 T& f; D6 o    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));       
          //print register value
1 y% B) }3 v: S1 e7 `" K          for (i = 0;i < 4;i++)
; k% n1 ?( o5 }$ y          {
" i* C$ ~# ~$ K: Y  B- i) z) g) h" Q        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
; h& p) W# w$ U1 i          }
4 H! x# N% T7 v          Print(L"\n");          
$ B) _! @6 @' [& `+ Z2 H3 J" X          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
9 c. |5 M! q; E8 H    }

1 o) q1 q9 \* _, ^) T# w        return Not_Found_Me;
8 i; x) x+ s7 O" [+ k6 E# d4 C}

& G2 ?. Y' A* B" a/ ]) O1 G8 GEFI_STATUS
0 H  m$ x: S: r+ T3 qInitializePciApplication (
7 B4 G1 ^  k9 _    IN EFI_HANDLE           ImageHandle,
; w8 C+ y" m* v    IN EFI_SYSTEM_TABLE     *SystemTable
0 p2 R6 t3 j8 \; K* K" W    )
5 i9 i- p6 Q" b8 N1 n  z' m2 v{
  EFI_STATUS              Status;
) N3 R2 J8 t) q  EFI_HANDLE              *Handle;
- n/ A) m# L# R, [8 m" @  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
& }: o5 [1 ?! C  a, `  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

    //
% n3 D! z) J6 p3 S! \    // Initialize the Library.
& Z3 K$ S9 c/ p" S0 m% {    //
    InitializeLib (ImageHandle, SystemTable);
* C0 S! J' X. q    //
    // Find the PCI driver
& Y+ J: B' e5 r, L0 P3 d6 \    //
    Handle = NULL;
2 j2 m  g) h8 X+ b& {    BufferSize = 1;
6 E5 T. p+ I( Q4 G/ o    do
    {
      if (Handle)
      {
' c1 q. Q9 Y+ [3 m2 R$ l/ S" R  f        FreePool (Handle);
      }
               
      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

( `3 s: t, \8 o- Y3 l( Q. a    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));
: g  W; K: ~' v7 w
/ W; T8 k, v3 [; S    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
2 z: _/ v1 k: y    Index = 0;
4 _5 I9 R$ \0 S+ T) Z& q    ST->ConOut->ClearScreen (ST->ConOut);
0 e* n3 F& s3 A! `8 l7 s! t    Result = Not_Found_Me;       
    do
! ^; p, o' O% D    {       
. q6 I  `. u% |1 V8 h' b. \0 A        if(Index == NoHandles - 1)
" {  O) U$ P* S2 G; h  ~3 v         Index=0;
$ z* h1 O9 d. N. d) ~( ?* K1 Y          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
* ]" u1 g9 j2 u2 n; k; w        Print(L"THIS IS DEVICE %d \n",Index);
; I/ ?% A( a& X    if (!EFI_ERROR(Status))
    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
    }
! D3 K& _' R% _4 e8 H/ ]    Index++;               
    } while(Result == Not_Found_Me);
       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
$ X) e* p) ^! P) N: Z) _9 S* U: F    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习