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

kaurohy

/*
4 M+ `8 P5 J  x" P- K8 W * myfileio.c
* Apps
$ z4 L/ b% l- O. U */

& T, j6 ]' e/ e# f  T#include "efi.h"
#include "efilib.h"
: o. J, s8 O/ Q+ c# e
#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
8 ~* L& K! m1 Y+ h9 {6 G
static EFI_STATUS WaitForKeyOrReset(VOID)
{
7 Y- Z% W& L& a    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
  y+ q2 g* @' ]/ \    UINTN               index;
  k! m4 ~! R3 b% i   
0 r9 @; e+ |) k( ^    for(;;) {
! S! e! O/ |4 Q; Z* O1 C! K        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
4 K/ P2 ^8 Z% X' n$ X% v% K( @; H+ F" r        if (Status == EFI_NOT_READY)
7 ~' ?% v7 [% Y4 S            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
2 H4 C5 D: q7 M6 @% M# {        else
            break;
    }
/ q% }' V3 z% i    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
8 V' j/ o6 @! e: y            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
& W8 _8 c0 @3 Z) L    }
   
4 }* n3 C8 z% S8 G$ N: R    return Status;
0 o) F8 e/ J1 F3 a8 w4 ?* l$ t! i}

& e9 C; K& j2 j5 kEFI_STATUS
% ?# D& ]6 |( W$ OEFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
5 M: K9 ?) C1 n/ ]! n5 I4 _+ M8 I{
2 V7 i! A0 E% x* r6 P! S. l    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
. [5 C! e' o7 s% S- ~  P    EFI_LOADED_IMAGE        *FileHandle01;
+ |- u3 F$ v$ z0 f3 U. G    EFI_DEVICE_PATH        *FileHandle02;       
: W  y! p9 {# f% m$ w    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
& U: E9 o" V  c3 J# e( v' p    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
5 ^  e( n. z$ c5 {        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
+ B. ?& K" }% z: ~4 n3 x    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
% f! g/ l' w  B& z, J    int         Bit0, Bit1, Key_Space;       

. e/ k- M$ Q6 P: \+ S    FileName = L"NewFile.txt";
1 {2 w8 h9 A4 n* y' a9 T+ i3 J        Space_Key = L" ";
       
8 a7 K  \7 s6 `  K# {        BufferA = L"ABCD";
* j) P: b1 C. b, x        BufferB = L"EFGH";
% ^) Z; y) S9 I6 T        BufferC = L"IJKL";

0 K# S; n: [7 w        Bit0 = 0xff;
        Bit1 = 0xfe;
1 s6 a# D, g4 e" B& s       
/ t/ n0 R+ [4 k$ w: `* s/ y* R        Key_Space = 0x0020;
. m4 L/ v4 u  A$ v       
    InitializeLib (ImageHandle, SystemTable);       

$ Y* C3 q1 ]  `) k: l+ u4 q7 f, n- L    DestAddr = AllocatePool (BufferSize);       
+ S* q4 z% Q9 {, @6 q2 Z" c
9 W3 F! }( O; l3 x) Z    Print (L"Value of Bit0 is %x\n", Bit0);
$ _0 S: \$ H. ]! K    Print (L"Value of Bit1 is %x\n", Bit1);       

       
2 ^# ^, \; N* t4 x    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }

  ^6 s+ z* [6 }" S) ~5 e/ }: _& m    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
8 P+ t( Y' G0 l4 K# q$ W    if (EFI_ERROR(Status)) {
- E" y, M2 Z5 t- d/ G; L- M- W7 V            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
& `$ H* U- R  g) M- ?  o0 X            return EFI_LOAD_ERROR;
- V8 z% q5 p. I( M( ]8 T        }       
3 Y, c) ]; Y9 U       
! x  S. A/ ^1 n6 J    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
* H. [( z, B: ~3 a* Y9 A! Q5 @3 z            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
) N8 Y; H. g0 d9 k' S" t            return EFI_LOAD_ERROR;
        }

    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
6 w' [6 }6 z! l1 m    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
! b# y3 t( X! p' r7 h        return EFI_LOAD_ERROR;
    }               
8 q. \+ j- O. j" K6 X; Z, \8 W               
    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
* R7 _& i# q. x: h0 E7 e- F    if (EFI_ERROR(Status)) {
6 K) H6 Z7 N: m  w! r/ {3 \0 F        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
- o4 s& Z% l6 w$ g, \' {$ b3 C        return EFI_LOAD_ERROR;
    }
       
    Status = FileHandle05->SetPosition(FileHandle05, 0);       
    if (EFI_ERROR(Status)) {
: x: M5 k, X4 s' N9 J        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
! @- O# A9 Z- G; |7 ^        return EFI_SUCCESS;
    }
       
; `9 L8 y6 e4 ?' l9 D5 c3 h6 V    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);
  G! \" }, V$ T. s4 B, o        return EFI_SUCCESS;
" K0 X4 K1 `( S5 b- x5 ]    }       
$ k4 b% K6 e0 f! w0 k! Q) h9 `$ M
- v  q7 Y+ b2 q. G    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);
5 l, v- }0 Q* j        return EFI_SUCCESS;
    }
2 J$ h5 N) S3 Q       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
% I/ ~% X& H( r/ f        Print (L"File Buffersize is %x\n\n", BufferSize);
) }8 U/ ^% e# H, s$ j2 e) @    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
2 _* a: N6 A/ t/ X    }               
4 S" f5 P1 s1 D' ?       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
+ y- w. g$ F. C# A0 s8 R        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
( V5 |! U0 J( S  r6 I7 U$ V        return EFI_SUCCESS;
( y. `& L5 d2 J; D$ y- ~4 R    }
1 T+ `# U2 A2 k% _4 q4 K! g       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
        Print (L"File Buffersize is %x\n\n", BufferSize);
4 s( w0 p! Q7 `" r; T    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
2 T3 T+ N. h. ]' }2 T, U        return EFI_SUCCESS;
: T9 k0 q' k3 s0 L0 K    }       

//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
! R5 O2 D. M: m$ X$ a7 h    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
/ r* |* }( c; J- G7 Q$ o3 L* s    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);
" f3 X; [+ k9 ~$ _# H        return EFI_SUCCESS;
0 e0 |  p0 [* D  X/ G' R1 j    }

6 P* M- c4 z8 p/ [; p4 R! b; s    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
1 f2 T. E& j! ~    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
/ {7 o( u) B: Y# J6 J: R    }       
6 ^* s' B. U' V* }3 a. {        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
- y( a- D4 W+ x! Z. N7 v
) {* a7 O2 S% s, W    Status = FileHandle05->SetPosition(FileHandle05, 10);       
    if (EFI_ERROR(Status)) {
* E( q0 E0 A: K& S) }. d0 {        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
$ Q! _2 `  q& {8 X6 Z3 U$ ^        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
7 U" o6 w! X) y* k    if (EFI_ERROR(Status)) {
! I, p8 N; R, D; }        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
! \+ T9 l- [0 s) s        return EFI_SUCCESS;
    }
7 s: ]) Z( T. e+ J9 _1 S! }
    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
0 b9 v. F1 F6 a  X. Y) u: H$ T/ A    }
       
# ^! }" _, Z2 Q* h2 P- v' r    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
1 u( U. P" X/ {    if (EFI_ERROR(Status)) {
7 E3 ~) P" u8 n" t        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
. F; h0 g7 ]6 H        return EFI_SUCCESS;
: a4 u; W8 q4 j    }
4 ?0 \+ _* f& \1 s8 |) E( U9 o
: o5 L6 u: a8 T' t    Status = FileHandle05->SetPosition(FileHandle05, 14);       
  _1 N+ z) _0 x0 ?8 x3 a/ }" O    if (EFI_ERROR(Status)) {
9 k1 X3 a" Q: {8 O        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
/ F, r+ S, m$ {" h    }
# B% R7 l( h0 g7 p( ], I7 i8 |       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
% |7 y) }0 @! w8 w  ]$ ?    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
. J: t* t, K1 m
    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);
4 f1 _0 w3 I3 @, Z/ t2 W        return EFI_SUCCESS;
6 l. R1 ~7 A# c8 j    }
       
8 n" X8 j5 b" j& {    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
) K( U/ x9 r" {$ r# ]    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
2 R# b% X) M' R1 M, X        return EFI_SUCCESS;
+ b4 k+ x, S6 Z0 a/ R; I  S    }       
7 a0 E# |2 L( B3 ~4 o5 a. A" [
0 N% P3 m( I2 u1 [) g% M//---------------------------------------------------------------------------------------------------------------------------------------------------------------
5 n' w  P7 b  y* Q1 I" B
    Status = FileHandle05->SetPosition(FileHandle05, 18);       
) r" r% ]9 k) ?. |6 }    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, BufferC);
4 x$ U1 O8 A3 C    if (EFI_ERROR(Status)) {
( L' S1 H  T9 X5 v( M) b        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
# v( ?1 g5 Z+ a/ z: |        return EFI_SUCCESS;
  ]. V) c9 P4 |: {" G    }       
5 ~% B- p, o2 B4 K  S% B8 @/ i       
7 w/ U+ X8 b% o# ]9 z: C    FreePool(DestAddr);       

    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
8 X/ N( }2 l) X$ q$ y    Print(L"File Name = %s has already been created.\n", FileName);
4 D7 W( Y0 b# o/ J; v
0 g' J8 @/ L7 n    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
6 T7 W2 `% q- }# X) T+ o    WaitForKeyOrReset();
+ w3 \' x% k% s! F( l  U
9 h1 T( V$ `8 S% G& \    return Status;
}

chenhy

//**********************************************************************
; n/ G2 C' {5 J2 b4 P6 R//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
- C! ?) y2 M3 _% V! d4 D//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
' `! T- i. z. G6 q0 J! I  l& ~//**********************************************************************

# Y% ?* N/ H" C#include "efi.h"
: c1 f' n" F: o* M+ Y+ d. }#include "efilib.h"

5 i% f% R, ^  w+ ~& H#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001

static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
. X8 D7 }9 d' [2 }+ A! Y   
1 n$ b9 N1 E0 ~" `    for(;;) {
6 u! q# q' V5 y* Q% X' S" S6 A        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
; g& v! h7 A2 o5 c/ K3 @8 b        if (Status == EFI_NOT_READY)
8 N8 F- I/ L. S/ \* `( c! Y1 Q0 O. u# b            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
    }
    if (!EFI_ERROR(Status)) {
$ `" A6 {  v% a2 _# b        if (key.ScanCode == SCAN_ESC)
9 q/ W- u! I* q) E& i3 P; @            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
% b  v0 }: _* o   
9 V1 `6 H: C' A1 u  a0 J    return Status;
0 _1 ~  n5 C  D}
) X  m  C! R6 H* K7 S# v
UINTN
2 g& x7 }2 l; D  S) y* UAccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
{
+ M9 p& P4 c  H, V& Y: o$ B        UINT8  i;
        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;
' t8 _- V# C1 o* E) T, z& r
7 j; ]7 w7 |. C" R* S/ o        //get device attr
" X0 ~3 P& a9 {) Y        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));       
3 o, w8 o  U% q* [( ]& b          //print register value
4 ^; w* Q+ i+ O2 a          for (i = 0;i < 4;i++)
8 J6 w/ `, ~4 D/ x* m5 q+ m  w- J          {
3 r" M' t! A! _7 I        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
+ d% v; v9 a0 [; G/ c                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
9 z+ j# f- V, o9 s          }
          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
    }
& K- I. ^; {+ k7 E
' a  N' I5 o3 f6 o5 G; H        return Not_Found_Me;
}

" J& v% ~- q& o$ \EFI_STATUS
8 C' `( ^* \: J. ]InitializePciApplication (
$ z% |' \! k0 Q0 d/ \    IN EFI_HANDLE           ImageHandle,
  ]# t! r" q2 C4 n3 o$ P, S4 E    IN EFI_SYSTEM_TABLE     *SystemTable
    )
! G9 t- R" a- {# H  R{
/ n; G9 f( }5 T0 _  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
0 H5 e6 e: `1 m7 O% r9 b% s  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
/ I: ]3 a" o" n  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
* e  c+ H: s; ^+ x) `" s
. a% i8 d& r7 C, Z    //
    // Initialize the Library.
    //
" X# T& l; r! a6 J/ q( G3 |. ^    InitializeLib (ImageHandle, SystemTable);
    //
* V3 y7 }6 Y1 b0 V. W/ u    // Find the PCI driver
    //
    Handle = NULL;
2 ]/ h1 l+ B) Y# y' j    BufferSize = 1;
2 l# a; B( T5 ?' m    do
0 m) V6 C: c& v0 q4 Y9 C    {
      if (Handle)
      {
        FreePool (Handle);
" z+ t: R0 \4 j4 h8 h      }
               
5 L( O: U# K% R4 S3 T  G" |$ n      Handle = AllocatePool (BufferSize);
* ~1 f( o8 v% [$ G* c8 T% U      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

    } while (Status == EFI_BUFFER_TOO_SMALL);
/ F* r, m6 X! h/ ?4 f    ASSERT (!EFI_ERROR(Status));
  S0 |3 t  G, }9 k
- H- L1 l* Y  A. c; A. J; k2 A    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
/ m4 A& F  D' u    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
- _9 Z% Z: c5 b& f/ D5 ~: E    Result = Not_Found_Me;       
1 E3 E( I5 ^$ _& ]4 p    do
    {       
        if(Index == NoHandles - 1)
8 \8 ^" b3 I; X4 U, M         Index=0;
          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
' o$ y, K) E; e3 b' x    if (!EFI_ERROR(Status))
& n8 V1 \# N! Y0 z7 u* D+ E0 i3 D    {
! h& b2 j) t7 u4 G1 \( T, v          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
+ a$ \% I, x' O0 }& m0 \    }
    Index++;               
6 R, c6 N$ _6 |1 _1 K0 O/ O    } while(Result == Not_Found_Me);
       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
' Y- v5 R- ?5 w. i9 K0 p    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习