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

kaurohy

/*
7 q& Q7 i1 e, x6 V7 ^1 O9 A * myfileio.c
' g8 l, e- I" X. }  q# @# o * Apps
*/
# j; o' K0 N4 g# ~
( ]! J4 @; g) ]#include "efi.h"
) f+ o- X& T' _5 Y; b#include "efilib.h"
5 j9 r3 M2 K9 H% l/ T
#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
, x, T0 y7 d) j+ `+ `" R  Q
static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
3 M9 A: b# T9 S6 p- D' y6 h( `( q    EFI_INPUT_KEY       key;
    UINTN               index;
   
, J$ f# ?. H" _( L8 b    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
: l2 q4 p/ b- v        if (Status == EFI_NOT_READY)
' V8 o: t9 `) s8 X            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
% d- W* p. L- k) H7 t! E        else
' \6 q4 V# n" A: E. i  S8 W            break;
    }
    if (!EFI_ERROR(Status)) {
$ O& x6 M9 ~$ R3 o! Y1 }( h+ q        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
    return Status;
3 P4 x5 r5 t: j}

* P; A8 B) ^- ^; rEFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
{
5 y1 r% q' }9 O' S# O4 f" e    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
6 b6 x- W, Y6 ]2 T7 ^    EFI_LOADED_IMAGE        *FileHandle01;
/ ?$ C: U" E- U5 k3 ]& a4 l    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
" z1 Z* ?3 I( {9 ~/ {  Z5 t4 C! V    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
" Q# v% v/ U6 U: r+ n  J1 Y, b        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
9 I# E' ]% H) a- n6 ^9 d    int         Bit0, Bit1, Key_Space;       

    FileName = L"NewFile.txt";
& P5 d6 S' V* K- w. T. [  ?        Space_Key = L" ";
       
        BufferA = L"ABCD";
        BufferB = L"EFGH";
& s: y$ s3 W0 H2 @/ A' L4 _4 G5 W        BufferC = L"IJKL";
  C7 b  h+ _- X" [7 U
; `# f+ ]! {' k* ~4 O) z        Bit0 = 0xff;
        Bit1 = 0xfe;
* R! p3 M0 _' O; Z       
        Key_Space = 0x0020;
9 n" [6 l# P( ?; N4 [; U6 |; K9 L       
    InitializeLib (ImageHandle, SystemTable);       
, Z1 P* I/ c9 p5 R
    DestAddr = AllocatePool (BufferSize);       
; w( g" M& v; M8 }/ y% \8 R0 J& g& i
    Print (L"Value of Bit0 is %x\n", Bit0);
* g5 R  k# s. Q& T    Print (L"Value of Bit1 is %x\n", Bit1);       
1 B8 @( h7 t. U! H6 D; C
       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
; l% b& v2 X( g1 i7 i    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
1 E' [" u9 B- Z        return EFI_LOAD_ERROR;
    }

* l4 p# J& {, f- ^" z    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);
0 g: o2 \5 |! D5 t/ q7 d1 Q' X) r            return EFI_LOAD_ERROR;
, M; q7 {; V( H& A: p: B        }       
       
9 f# d  H; J0 h    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
4 w+ m. [3 L, K! h4 `$ t- Y6 Y            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
. Z: j5 X3 }7 T& B6 S" Q0 w        }
: [" J, `7 X4 R- o2 E8 o# E
, Q. Q9 {  `/ b+ c    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
* m& M5 a9 n0 J7 z    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
2 C+ A- l# _% W/ Z        return EFI_LOAD_ERROR;
    }               
               
/ I4 ]/ i- c# b5 q    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
    if (EFI_ERROR(Status)) {
7 M2 j7 {( Q4 U- I5 B% i        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
) c: V) d* Z, P3 a        return EFI_LOAD_ERROR;
# n9 @! i, p1 B& g& M/ v    }
       
3 L: j" B9 d2 M    Status = FileHandle05->SetPosition(FileHandle05, 0);       
    if (EFI_ERROR(Status)) {
" U: J/ U0 u. x! K' @9 p; n        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
0 ]3 e: R8 c- h9 T. P       
$ J( T* \7 a  H$ y2 v8 T& g" |    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
: g2 u+ ]- Z9 |        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
1 ~6 q3 ?: ~, n- }        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
+ N9 z# {& b4 d8 I+ q5 r, G: G2 S* @        return EFI_SUCCESS;
    }       

    Status = FileHandle05->SetPosition(FileHandle05, 1);       
% |( U2 D  m; j    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
: |7 s: q. a% q. o3 W        return EFI_SUCCESS;
; ?) F: u) a. U% G) x    }
       
, `- ^# W; u9 T2 r! L    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
: Y+ }5 p$ l. L. T2 E, s        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }               
* x6 z4 l4 L" M1 _6 }! [; }       
* K9 I9 Z* |( D  v: `    Status = FileHandle05->SetPosition(FileHandle05, 2);       
2 G0 d1 [; g, [7 f4 O4 _    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
( y0 s/ r8 B' ]9 S' J9 x        return EFI_SUCCESS;
' h; M- l0 V: q  b( v    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
        Print (L"File Buffersize is %x\n\n", BufferSize);
6 ]( o9 D  M" s; w    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
2 B0 u4 V! X% M        return EFI_SUCCESS;
    }       

4 E5 Y( N. U% {/ G//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
! j$ P: C) V$ ^7 |2 Q6 s# Z    Status = FileHandle05->SetPosition(FileHandle05, 2);       
* b, s: b6 T) j' \( w    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
9 ~9 N0 Z+ |4 s6 j! A        return EFI_SUCCESS;
    }

8 w# m' G% ~' [6 ]. ?5 w' ~$ P( G    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
$ m4 _$ n6 V* B8 _- N    if (EFI_ERROR(Status)) {
) o% f% p; o9 {; ~        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
9 E4 p6 p6 J# `' ]* k8 w5 m  y        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);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

, N7 I2 r, d2 r. x* L" T' m    Status = FileHandle05->SetPosition(FileHandle05, 10);       
! {7 G1 ]3 T3 [, X4 q0 G% v    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
2 s# l+ v2 F- ?( b. m* C       
4 S4 k& c3 I" S' `, e# c2 U    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
) e. [7 f4 }3 O9 d* v  B' e5 C        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
: o# `: ?7 h7 u3 C" A6 R3 Z9 M        return EFI_SUCCESS;
    }
  T7 E) d7 S0 B# _& I& f
2 l) k, t3 A; ~    Status = FileHandle05->SetPosition(FileHandle05, 12);       
- l8 O% d/ g% A& h+ R    if (EFI_ERROR(Status)) {
4 H: r( o8 f% v  e4 ~        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
$ G5 D/ z2 e0 A& T        return EFI_SUCCESS;
+ U) D* J; d4 f: y7 G/ v4 Z    }
) ^- n" F& Z6 v; A$ A/ M) q, K9 g       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
1 v! u) w: T9 M; T) ?& k        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
0 S' f5 j( M$ `. {( W: Z4 ^$ |" n    }
- @1 L- o& T, d# q3 n5 C
    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);
, l" l+ q9 u& C) Z) ?        return EFI_SUCCESS;
    }
       
3 R1 w1 s. ]$ \% Z/ y" ^9 E    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
9 S) ~. q' k. H, Q2 e        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
2 V. g! h( F1 K+ v9 {        return EFI_SUCCESS;
% t: k, \& P! E8 |    }
+ b9 O4 y. i5 C/ Y4 B
& o. T$ D8 E: C3 m$ S3 c! y    Status = FileHandle05->SetPosition(FileHandle05, 16);       
    if (EFI_ERROR(Status)) {
* Z+ |) v; H* N( @, O$ u$ \9 R        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
2 I# k9 \$ Y2 o! g9 F+ [        return EFI_SUCCESS;
    }
, C! L5 n4 p3 {  v' c) e       
/ D4 ^8 d( Q- }" z    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
3 }2 C7 f6 a* D' {    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
: S/ m$ E) r, V0 [/ e        return EFI_SUCCESS;
, O8 c% x" ?4 B9 E8 h3 u    }       

. p# h4 B9 ?) M. Y6 V0 F  I8 P//---------------------------------------------------------------------------------------------------------------------------------------------------------------

  O- E. V+ U: L* |. |/ y. h4 g    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
- ~9 Q6 m% q; ?( ^! O& i/ M        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
9 B5 K& t$ A4 T+ p$ b: m0 Q    }
: X3 l: E, Q1 l! x       
+ d. n/ a5 P, ~# W" Z; U# e    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
: o% ^7 y$ \' E$ ?7 b+ s9 J. ^        return EFI_SUCCESS;
2 Q9 A- s- l# L" ^4 \0 G2 z    }       
. k* g/ P" @0 e( T% c0 Y- \4 v       
+ g- q. A  ~- V3 T6 K5 v7 f    FreePool(DestAddr);       

    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
% `8 z: J1 G# n        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
: ?, v. |: s- a7 P: x' v1 }1 h    }
       
0 O* E: j3 G0 l2 f: P# T    Print(L"File Name = %s has already been created.\n", FileName);
3 X* T' j( r2 s
    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();
5 c9 g6 X  I3 x, x" W7 J) Q7 f9 @. J7 V
2 i0 v, Y3 ]: D    return Status;
}

chenhy

//**********************************************************************
7 P0 v: q% n/ f8 A. J//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
# t1 l0 |5 H2 {- S2 X6 x( S' `//**********************************************************************
; U7 w/ Z' F" {) \+ v; k//**********************************************************************
2 @; y( c6 R3 N1 F
#include "efi.h"
& o0 y4 k; J4 N8 ?; n- [; O#include "efilib.h"

8 V. l+ o* J2 U* w9 v#define        Not_Found_Me        0x00000000
4 Q( J2 `# p% U/ [. S#define        Found_Me        0x00000001

# h/ P) l; F0 |0 E: `! ]) _static EFI_STATUS WaitForKeyOrReset(VOID)
{
0 {2 w: n  w2 \+ ?    EFI_STATUS          Status;
$ J+ @8 E% B4 m1 J    EFI_INPUT_KEY       key;
; R# V  A; H( o# L    UINTN               index;
   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
$ t5 D* r+ h3 Y7 o( p7 z3 C        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
- R1 _' W1 w' Z& K        else
            break;
    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
8 B* I) m+ a7 R2 _    }
+ `- z. _0 y& q1 M# M& P- x   
: u6 s0 J- `; Q. X! G  Q/ Z    return Status;
/ E, k, `* D: [) i}

3 Q, p: O" n  S/ e+ O9 |" G! YUINTN
AccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
{
        UINT8  i;
        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;

) u& X+ S9 e" v        //get device attr
        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
) @. F1 A( ^. |$ W    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]);
4 n: [: |) r, _  a9 r7 n8 k          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
1 `) [8 ^0 L" {( t! d# ^( C$ D( D" e          //print register value
          for (i = 0;i < 4;i++)
' L. c( p! |6 P- ]          {
, \) J% Z! S' k% S9 B        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
( z0 I5 n; h: j, n1 r! ?$ Y                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
, K8 X! P, t" e; T$ L          }
          Print(L"\n");          
  g: P5 o& E3 m% {: x7 j          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
+ P* R* O/ Y/ U1 F; W+ ?          return Found_Me;
+ f- D4 ~# @% G    }
6 k& L8 {& F9 g$ S3 ^; |$ {* [% M
$ H1 U  S% |9 `' |5 T        return Not_Found_Me;
" y8 D& D. ?5 [9 D" U7 q1 w}
: N7 \# `! U8 Z
/ P+ ]  {; X& [4 K# \6 G0 R8 LEFI_STATUS
+ q9 V& n9 H5 X/ k; lInitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
4 s1 u  u9 F6 W. _+ d    )
{
6 e: f/ |" K+ J8 N  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
, ?6 ^) z* t& {/ i9 J4 a  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
" \4 r# [! U5 J( ^* ?4 \
  r+ d4 |: i- b9 s    //
    // Initialize the Library.
+ G, w! I. b* T# z- g    //
    InitializeLib (ImageHandle, SystemTable);
    //
    // Find the PCI driver
3 B7 y8 I0 e) G' i* Y1 w    //
, U% a  q! j' O4 m% _8 j    Handle = NULL;
    BufferSize = 1;
" l' O/ @6 p: b* w5 {- B& Z    do
    {
      if (Handle)
/ R6 x8 M! q" T! |& _- O      {
; X7 P2 G9 I; I, [        FreePool (Handle);
# @$ Y# Y! Y/ ^( p7 h. R5 d/ Y      }
5 z! o' P8 M% A% \( x0 L7 @               
- Z  v" q/ P  k& x      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
& c1 t8 E% k/ F6 d+ F
    } while (Status == EFI_BUFFER_TOO_SMALL);
" D. P, M+ ]7 g8 c0 Z2 {# R; g- S6 F    ASSERT (!EFI_ERROR(Status));

    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
- K/ p3 T' z- q4 J. [) f& \! l. _    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
    do
    {       
        if(Index == NoHandles - 1)
$ }  T: \+ I* \6 B         Index=0;
          else
9 _+ E# Z. H7 \1 o    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
$ z" O/ e1 N9 k" Y/ ?        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
# T; i3 b$ Y8 z0 |          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
    }
    Index++;               
1 b4 S3 O4 _; k- T" p* Z* T- `    } while(Result == Not_Found_Me);
! F, B  I2 g2 y       
: V3 S2 i- e& V7 N* K; l        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
- W4 Y' v: c2 R" P               
! I+ |' _" \" p9 O6 t    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习