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

kaurohy

/*
* myfileio.c
* Apps
*/
/ j* M* S. r- U- R( j
#include "efi.h"
#include "efilib.h"

#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
+ k; G6 u# d4 B5 j6 e3 i: F
& T+ C2 n* M9 j; k# m/ d5 kstatic EFI_STATUS WaitForKeyOrReset(VOID)
9 z$ x+ A! h7 G3 S{
- J; m8 l4 _1 H1 n    EFI_STATUS          Status;
9 p  w& e; W2 R6 ~! f    EFI_INPUT_KEY       key;
    UINTN               index;
   
  @. p2 L( \4 l% v, p: n& q* Y    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
, C) v1 |. f: k, m        else
. f! I. Z' \7 b4 j, L            break;
) a' b3 j0 F. e- J1 R6 B+ i    }
# t$ W7 p0 V) Z  k  p. Q/ p9 h    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
6 g/ q: K: J  n   
4 [+ `. O9 f6 F1 G7 \1 e    return Status;
}
: j7 B+ R* m) d8 e, t
* o& e# A3 |2 K2 hEFI_STATUS
% O8 B$ [6 {) [: {EFIAPI
9 `" v- j: ?: P6 V7 s- m( B# LMyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
" R% M& C, O( D. S: a- E/ z{
$ V9 G" e" S5 W    EFI_STATUS                Status;
, N4 @" z4 A$ J5 V    EFI_HANDLE                *DestAddr;       
; q* h/ m; u9 v4 J. _- |    EFI_LOADED_IMAGE        *FileHandle01;
) d! H3 }. }$ q1 q    EFI_DEVICE_PATH        *FileHandle02;       
. ?  T0 u; G8 C    EFI_FILE_IO_INTERFACE        *FileHandle03;
- U- w9 T# h3 B/ S$ e    EFI_FILE                *FileHandle04;
" i+ G4 w3 Q5 L- k    EFI_FILE                *FileHandle05;
  L5 R; F6 ]" y8 f. g    CHAR16                *FileName;
! K4 ~0 [2 {) |; U9 \. E        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
+ C! V4 B. _, N6 |    int         Bit0, Bit1, Key_Space;       

  u6 u0 c8 Q; y    FileName = L"NewFile.txt";
( ~' j4 V# G. w6 C( k        Space_Key = L" ";
2 w  `- y) D6 v) y" \4 X  b: c1 ~       
% A( d& T, c8 c4 l9 w3 N/ ]4 d" v5 G        BufferA = L"ABCD";
: r, f7 j, P$ @% R        BufferB = L"EFGH";
. a+ j# w4 X: V2 ]9 H* o        BufferC = L"IJKL";

        Bit0 = 0xff;
        Bit1 = 0xfe;
, j* S6 i6 K! k. b) F       
, e) c- A. S' j, q        Key_Space = 0x0020;
6 M0 {; m6 l( r5 s9 D* i       
    InitializeLib (ImageHandle, SystemTable);       
2 ]+ q7 n. O) g9 J: Y+ d4 P& q7 w
    DestAddr = AllocatePool (BufferSize);       
( t  Q6 k9 ?2 T2 J
    Print (L"Value of Bit0 is %x\n", Bit0);
' b: V- o9 U  z/ w' W    Print (L"Value of Bit1 is %x\n", Bit1);       

       
, t0 r) D4 S/ C+ S) H    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
# [9 I! \( J/ a; D        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
" k0 }, Q" [( P% ^# p9 p        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;
        }       
       
- X. v6 V, \" \6 ]% b0 x    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
8 J. `7 A8 F- u$ O8 _            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
: W: _! D' r/ c% }4 V6 v9 m$ A        }
% |3 x9 J! a% ~. z! h
+ S! F, f) C/ d! m: }    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
+ C8 p8 P: |9 d: b2 k& y0 B) F/ V    if (EFI_ERROR(Status)) {
/ z$ I4 {* R* \/ t: O        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }               
' Y" l/ i4 h5 @               
' C# q: c& l( X. F9 ?7 F    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
    if (EFI_ERROR(Status)) {
" q* q2 f6 \6 t) g; p9 o        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }
       
    Status = FileHandle05->SetPosition(FileHandle05, 0);       
# L9 O+ Q- t" G. s    if (EFI_ERROR(Status)) {
6 ^/ n) l2 j: D0 \        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
% _1 z- E( P+ z) w( Y6 w8 r        return EFI_SUCCESS;
    }
       
/ ?( ]' D! K( {0 _7 ?) ~& f    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
& O% w( n0 `- w4 A; p$ l        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
: s4 l" }' t6 b/ C        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
+ m0 r- ~' a5 a7 e* C- b: L    }       

7 E: F6 U% o5 K2 p) }    Status = FileHandle05->SetPosition(FileHandle05, 1);       
3 n, P( b$ {: ~2 R. }7 T8 d    if (EFI_ERROR(Status)) {
# {+ @8 i- ?0 q' R1 L. D        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
. R2 g6 ?3 {; ^; d9 V7 ?) y       
0 U( x' F0 ~5 s$ _% _% j# T8 e    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
# u& |* a! ~$ C6 {8 L& b        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);
" m. \# V" X& z* F        return EFI_SUCCESS;
7 m2 @0 t/ b& c0 \1 V    }               
* r$ D2 z' ^, V" y       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
- `% W( l5 e- b  W    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
/ y6 _7 X6 t, y" ~        return EFI_SUCCESS;
! v5 T' R8 x2 [) X  V+ T7 E    }
+ `9 }, H/ G* r9 v6 O! `       
$ ~9 V- @1 X: Y1 @    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
        Print (L"File Buffersize is %x\n\n", BufferSize);
0 d* L, R9 v3 U  E( Q: W/ J- X% G    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
5 l' o% i, N2 Z- f# Y, [        return EFI_SUCCESS;
6 _; z. A# `) ?& |* ?8 N    }       
. @+ {5 H) b' B( u
3 _) o) E8 q/ E# g8 F7 w$ k//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
* g4 w  z+ Y3 l& p% q    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }

3 ~3 M( s3 I) u# q& _    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;
5 K; D$ B7 k- S/ A3 E& K1 G    }       
  S# m: B) O/ v& A        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

    Status = FileHandle05->SetPosition(FileHandle05, 10);       
4 a0 o: a: m! U% F- f+ F    if (EFI_ERROR(Status)) {
) t$ I2 T/ I6 d, x7 }        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
5 N! \" D% t; r/ X       
" n, ~+ M* l2 R, N1 `: l. }8 @    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
9 K- D3 U2 `- z& `2 A        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
- h5 @+ s- Y6 N2 i, w$ h( [    }

2 q" q+ V* U' z4 U    Status = FileHandle05->SetPosition(FileHandle05, 12);       
& t$ n) l) K: d; p; h, g    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
" o& k7 x! x9 y/ @0 v        return EFI_SUCCESS;
    }
' O% [# Z! J; W' R+ m' R( D       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
0 d# v! l' t3 E    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
) O" g  Y  p  ]) z5 R/ r: M        return EFI_SUCCESS;
0 P7 V  x2 l/ N! }) K- I8 N    }
# m$ G2 \4 b# b. h  j! M' I
    Status = FileHandle05->SetPosition(FileHandle05, 14);       
* H$ c7 }! b: B  \! u6 H  }    if (EFI_ERROR(Status)) {
% p* L' B  ?' ]: o1 L; ?1 ?        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
( v0 _$ C) F4 F4 G( t8 M3 G       
) F# F+ y' n4 E! ~6 @5 @    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
; I/ E. Z: m  p% k        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }

* R8 \3 w+ \6 p, {    Status = FileHandle05->SetPosition(FileHandle05, 16);       
3 {5 w' S5 m4 H: _    if (EFI_ERROR(Status)) {
& r% {! a5 P& f* t4 U- g        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
  z3 i7 ^" m3 I! d" a. M) U       
2 Q" v0 J9 W/ x2 @) p3 o! N/ g    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
/ X9 y: U' g- F6 K) _: U4 ^" Z    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
- A1 Z4 C7 ~( @" i3 J# E        return EFI_SUCCESS;
    }       

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

    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
6 T0 ]; ]# g8 @; l4 X3 ]) a        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
( {+ J+ {4 r7 v1 k5 d. B' r8 |        return EFI_SUCCESS;
: Q, I: S& z1 ^( m) r' J: ~    }
# Q2 W2 _, N8 k' [- F! m* t. I       
1 A" S; m' N0 L- ]. Z$ ^    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
+ Y1 Z3 Y3 O3 x& ]5 T    if (EFI_ERROR(Status)) {
* O! A6 Z! A' X5 r        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
( k/ u1 G. F+ g) V. k% M. T' t1 W    }       
- w3 ~# [3 Z2 l, z       
( \, D6 Q& ?" h+ M    FreePool(DestAddr);       
$ |' e2 l2 j" T7 `
    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
( M  b- W9 _. y& r2 d9 H# F% p        return EFI_SUCCESS;
, A% X, @: l3 T! t8 n( B; Q    }
       
    Print(L"File Name = %s has already been created.\n", FileName);
: ?4 Q5 U  r9 w4 o) r8 B: Z
    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();

# r2 u. z- h6 j    return Status;
2 C& ]! t& _" G' t1 A1 Q}

chenhy

//**********************************************************************
# w% _3 _8 T) W6 o2 f2 U. P! y//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
6 O! y' o% ~' q  @//**********************************************************************
% T. K' @: A* J) ]  K/ O  D//**********************************************************************
8 I& z/ u2 {# w' M& e' s
3 S( N  K. h. `) Q' T$ E$ U#include "efi.h"
$ W# X# `# c4 [6 b- ~3 \#include "efilib.h"

#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001
- l& M- r& R7 m4 e/ v
static EFI_STATUS WaitForKeyOrReset(VOID)
- Z% e$ T; E. W& v6 d0 U! {3 Y. k{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
+ Z6 g/ W) _% `* K. Q            break;
    }
1 {2 F+ f- g6 m4 E3 S- i! X" l; ^/ P    if (!EFI_ERROR(Status)) {
& Y" U" A- s9 y- |2 w        if (key.ScanCode == SCAN_ESC)
0 ?3 F' C' n! L4 z            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
1 l9 I9 P& j' \7 y8 z# I) v    }
& g8 U/ ^+ @, y/ y+ R4 s   
    return Status;
}

- O! l% {5 h3 W: ?  `UINTN
5 I" r" W( A( q  TAccessPciConfig (
" {  H7 K: s; a7 C    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
( a# a  P$ [  J, p* D6 h    )
{
        UINT8  i;
        UINTN  PCI_ATTR[4];
' j. W# p7 g. \. j2 h  \0 @+ A        UINT8  PCI_REG[4], pci_offset=0;

8 [, r1 h; z/ b        //get device attr
: S9 {& k3 e% [: C1 Y" w/ v7 p/ O% ^        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
2 ]2 ?2 b: M& j* 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));       
6 i- `7 [5 o( a& ?/ l          //print register value
          for (i = 0;i < 4;i++)
          {
0 C, f9 J" q# S. B% S( R        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
9 F, z! E- l) \: T                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
. z3 |" V& q; y$ t4 l# J; ~          }
, I, }0 y1 b4 {4 Q; J          Print(L"\n");          
* N& j# S0 u5 a1 h3 k+ Y2 B          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
% ^+ V) D) Y& i: x0 H          return Found_Me;
  w1 T) |; K' V" y* N0 e    }

        return Not_Found_Me;
% k, k% ~" S3 @( x2 A}
. z& ]. F; R6 a  \- L/ {
EFI_STATUS
9 L5 h4 O( Z! O5 u9 X4 |: sInitializePciApplication (
6 b$ n% V5 S: Q: @7 x6 H: x    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
    )
{
  EFI_STATUS              Status;
( d* s$ t$ s$ j) u* P  EFI_HANDLE              *Handle;
" o8 q6 E8 Z, n  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
4 S( {' {4 p0 q1 T  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
6 d* w: y+ i$ z* E
    //
* L2 k% O. h/ W0 O6 k    // Initialize the Library.
    //
( z1 L. K% n1 @* x" B    InitializeLib (ImageHandle, SystemTable);
5 t! N: A/ Q8 s7 N, p    //
+ n+ H% R: W" E; o7 e5 l, z" X    // Find the PCI driver
    //
    Handle = NULL;
    BufferSize = 1;
0 w' R8 h6 r. J* e. \    do
    {
      if (Handle)
2 c. z# J0 J- f% }" a: Z% w% e      {
        FreePool (Handle);
      }
               
      Handle = AllocatePool (BufferSize);
* b& T: ]/ B# D5 O5 P      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

0 `. L/ Z1 t* p8 Y6 D9 n3 U* [/ l2 ?. {( c    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));

    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
! M! n0 |+ y0 p9 m/ t+ e    Index = 0;
0 R+ d7 |, b, ^. p: v4 Z# S5 k    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
    do
# z. ]/ ]$ a# m: L    {       
        if(Index == NoHandles - 1)
         Index=0;
          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
2 A* P  S6 Q& E( I, ]" ~: o    if (!EFI_ERROR(Status))
    {
6 C2 E6 t  W( j# D6 F; X          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
- {! t9 E6 J, d* B! y$ `, l    }
9 ?1 }) B' D: i6 {" i; v4 c    Index++;               
    } while(Result == Not_Found_Me);
       
% y+ o! B1 J; L$ t* m7 |        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
) G* m0 S' Z- z/ P$ m% W% J0 S8 D               
4 {) ^' D+ ^& \    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习