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

kaurohy

/*
4 \# {1 c" l* ?& {2 O2 U% O. t * myfileio.c
5 ^& t+ W& m" H6 E6 F * Apps
*/

#include "efi.h"
! X- }$ O( S' d& B4 b) p. _#include "efilib.h"
! R( h( \) C1 G
! |% x4 j5 G# }* a1 ~#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
% `4 R2 N# v6 w6 }) _7 p
+ U9 s4 `# D6 v9 k$ b9 B0 m# Mstatic EFI_STATUS WaitForKeyOrReset(VOID)
: {6 R; K. q+ }/ t{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
0 k( h4 S7 \" I; c8 x) m   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
* Y8 A5 @3 S1 ]        if (Status == EFI_NOT_READY)
+ |2 {, g6 X! L3 U, n5 L5 o2 ^            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
+ W0 E) J& N7 `4 h* O# l! _        else
- x5 f2 e* i1 \+ c) p            break;
    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
5 M' d; }* G6 ~- E# @4 u    }
5 s! B; |% c: R. P8 M   
    return Status;
. d* _& e/ E+ ~; D+ P2 [" a8 g9 _}

/ y! U# t( J* o  ^% Q3 w& Q# LEFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
5 f+ ~; F# `! Z$ [9 c! _             IN EFI_SYSTEM_TABLE     *SystemTable)
{
    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
! Y; R; j5 _& G) z    EFI_LOADED_IMAGE        *FileHandle01;
8 p4 |) F0 E# V  ]$ P    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
; f6 v4 U, i& [( R7 p5 @5 {0 r# {( ^4 I    EFI_FILE                *FileHandle05;
; b9 X  B' s& Q    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
8 O( b; e- X+ g# c$ h    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       

& Q, A! K0 h: t% S6 ]- P" A4 p, v% f4 `    FileName = L"NewFile.txt";
        Space_Key = L" ";
3 K% r# l' @( F/ W) b% D: \2 |1 E       
# }" a* w- w; [$ J9 t        BufferA = L"ABCD";
        BufferB = L"EFGH";
        BufferC = L"IJKL";

        Bit0 = 0xff;
% U; w+ ]/ h! u# u0 Q% M3 A6 j        Bit1 = 0xfe;
       
2 J& v( W1 V, n3 U        Key_Space = 0x0020;
       
    InitializeLib (ImageHandle, SystemTable);       

    DestAddr = AllocatePool (BufferSize);       
9 @3 k/ `6 _! @
    Print (L"Value of Bit0 is %x\n", Bit0);
    Print (L"Value of Bit1 is %x\n", Bit1);       
8 g0 i9 o; P# F: y
7 Q' o* T, U& z       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
& h+ N9 e4 M. M% J        return EFI_LOAD_ERROR;
+ p2 r& }; I1 u9 }- k* |' Q, g: ]9 y# x    }

3 ^2 _9 s# P& s! C; H! ?! p. R    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
; H: K4 U1 R# @: V  R: F    if (EFI_ERROR(Status)) {
! @" E2 p. [1 F2 l! f; V            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
" @8 L- C0 o% q7 ?/ O        }       
       
6 h, x* w# j1 V0 ?( k    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
6 ?* r; t9 B5 W            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
+ q: f, V, V7 F- H        }

* u# ~# z  o) E5 _4 Y    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
' {9 v6 k7 {3 i; v        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
% V1 G6 q* |+ {# r2 s# j    }               
               
    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
, t7 v" w- v3 @# |, \  G0 z# n    if (EFI_ERROR(Status)) {
        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
/ \$ n# m& x8 i. J* e9 L0 \    }
       
7 {+ C- A2 c: _4 f$ C9 p: ^, @. q    Status = FileHandle05->SetPosition(FileHandle05, 0);       
    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
& O* u7 U9 X' l        return EFI_SUCCESS;
* b: \9 s& V9 U: W: L    }
4 Q' [% x5 [5 a- q: a# V' C       
: ]4 H2 P/ w& r( o4 x- A6 W6 N    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
. g* S% B- V8 k. Z; ^1 e        Print (L"File Buffersize is %x\n\n", BufferSize);
) l& x+ d% m$ g) N$ l    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
# R, Q6 Y8 Z" @5 V, _- L9 E+ }        return EFI_SUCCESS;
0 f1 y! W, F# _    }       

1 z3 n! V7 o! W6 y/ Z1 a3 U* @: V    Status = FileHandle05->SetPosition(FileHandle05, 1);       
    if (EFI_ERROR(Status)) {
2 w. x7 k1 `3 ^* @        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
1 ?+ N" @/ }" s  k' i    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        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);
2 D0 A4 ]9 a: }9 D9 c$ W: g        return EFI_SUCCESS;
    }               
       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
/ I2 g2 `) w" u" J- r( z% N        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
; C, r; I8 c. P    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);
        return EFI_SUCCESS;
    }       

//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
& M+ k7 i! D1 s9 u    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    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);
        return EFI_SUCCESS;
    }
! _8 v+ `* R0 Q
: a+ o% c2 i$ g% u* X9 o6 t+ ^4 m    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
( h6 [0 x' _! k        return EFI_SUCCESS;
  Z: r# _+ `9 w% U    }       
        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
( }, z8 e- z( r, J) o$ L' A! S//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

+ J3 J" V+ \% ?$ q; M    Status = FileHandle05->SetPosition(FileHandle05, 10);       
$ T4 Q3 G& H" o8 z    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);
1 D- l. a% b* R" C& F' f    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
" c/ O+ `$ F8 [9 u9 U        return EFI_SUCCESS;
    }

    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);
' S  w+ F4 \. z, h2 u+ ?        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
, c. H" r6 I" j. `" x    if (EFI_ERROR(Status)) {
# \* D# f6 Y3 U) j" k; U        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
8 L: l- g4 G  _4 L- @( e& o: N( M& Q    }
) t" X( L/ X& [6 R$ l* _
2 ]% m( }7 O+ r* z/ k$ {1 ?- l/ ?$ W    Status = FileHandle05->SetPosition(FileHandle05, 14);       
7 r3 x* y4 w4 u& @9 z( h    if (EFI_ERROR(Status)) {
( J5 V  @) n! T) I7 {% Z( M        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
* H$ F* Q/ N* z! ^. l8 G9 _2 T    }
       
& i; [- `4 O) i7 G8 L    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
$ q; z" R5 j) L0 F+ n        return EFI_SUCCESS;
$ X9 n; K9 M9 F4 u6 E6 f    }
1 g5 _' V. h- N# w
    Status = FileHandle05->SetPosition(FileHandle05, 16);       
& i$ l0 o+ j8 |: `    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
( W; r! |$ Q( M- X7 F        return EFI_SUCCESS;
7 l8 F4 ]1 `: o& q( E0 o    }
8 a6 K6 u+ ]5 |: V6 ], `0 \1 a       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
# }- ~. v( P7 n3 l    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
( F0 T5 Q2 B1 m/ t! O    }       

//---------------------------------------------------------------------------------------------------------------------------------------------------------------
6 h* g+ R3 \$ P0 k
) h; z; q/ S. _+ q5 W, l    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
! a0 _- [) _/ L$ i: W' M        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
4 s/ v% Z- h0 M& @% U, {        return EFI_SUCCESS;
0 p* k8 g" ]8 m( t& m1 y    }
1 h, a8 n# S( X; G; w9 v       
& L+ f# D8 L5 v7 P; U* o7 Z    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
7 w9 c) ?4 v0 {' ]# I    if (EFI_ERROR(Status)) {
! V% A) g% w7 |( i# D) R        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
/ Q! g- K3 K, ]6 h; n        return EFI_SUCCESS;
    }       
- s. s0 j! G& i5 {$ N0 ?+ x       
) G: z; M" v/ Z2 \$ _    FreePool(DestAddr);       
4 _* B* V0 E# X( h7 |$ Q
& x) b9 V7 J0 }& g    Status = FileHandle05->Close(FileHandle05);
4 h) C) l1 `3 N8 ~4 ?    if (EFI_ERROR(Status)) {
- W  L  G1 G! C( w. R  F! M% k3 S4 o        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
! ?/ h1 s4 ?8 f% _        return EFI_SUCCESS;
' Q, e" l; }0 g1 D8 r3 V$ ]4 J, g9 j    }
9 D) s9 j( {5 q3 b+ c) D, ^       
    Print(L"File Name = %s has already been created.\n", FileName);
" ~5 w- c4 G9 ^* u; a
    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
( w1 e( V' _" X8 y% e    WaitForKeyOrReset();

    return Status;
% `! }4 \; h+ a2 O7 d}

chenhy

//**********************************************************************
4 [( b2 A! p% D& C//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
/ C7 z5 q. ~, C/ ?. u( U8 u8 e' N//** By Lawrence 2009/11/20 @Taiwan**                          
( x4 ^' g4 h  h3 U0 _: o//**********************************************************************
5 v# r, A4 c- R0 e& {//**********************************************************************

#include "efi.h"
+ ~$ e7 \1 i& y0 Y+ n' p#include "efilib.h"

8 h! c. f* w% s1 d- |% r#define        Not_Found_Me        0x00000000
6 n3 Y8 E; B# D) m#define        Found_Me        0x00000001
2 z: W: Q; D0 @3 Y2 p9 H
static EFI_STATUS WaitForKeyOrReset(VOID)
{
6 f4 }  ]2 H/ K8 D    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
   
    for(;;) {
+ N2 ]# X) c$ O        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
" ]) s# s  H$ P2 A/ K7 M        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
$ o( O  w( G8 R( D0 p            break;
- {, ?+ J5 ]( i3 S    }
0 _7 X# K' s$ f    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
% s, D/ y4 L9 T( z% d+ L/ E2 D7 A- }            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
& g' i: R" |9 }* h    }
   
    return Status;
) M& ?; V2 n- w+ d}

4 O, j: N$ e3 \UINTN
4 |* P* s9 n- i) u2 P' w$ AAccessPciConfig (
7 a: _7 [6 D2 W0 f1 k    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
3 b% L4 _- l/ q% w{
: w; R5 ~+ E( t2 T2 H- v$ B        UINT8  i;
        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;

, S* N# r" b3 k- n' D6 ?        //get device attr
        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
7 l# m/ y9 R( Y( p    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]);
* ~# k  \4 @& }: O+ n8 E          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
. a5 R# i& _4 o" A2 R# j8 z1 T          //print register value
0 T/ Q( P5 w0 s2 D- g. `          for (i = 0;i < 4;i++)
          {
0 F/ b# K: j- z3 w        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# U0 G  |6 w( c          }
          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
    }

        return Not_Found_Me;
}

) @1 d8 B$ X1 E$ G; E9 bEFI_STATUS
InitializePciApplication (
( ?7 `4 W1 O* G" h. i, t  t/ d7 y    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
9 @  n8 M$ n$ M8 d    )
) [. o% D- @) Y4 L7 R! X  q{
  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
) G) ]: l6 F2 \& h4 m+ y$ g  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
: R) [" k( c& h0 Q& d7 T7 W; x6 |) e
    //
$ B1 `# {- E$ y% Q, R* p    // Initialize the Library.
0 u( T5 m- F% \, Z" ~* y9 _$ e7 L* e' t    //
    InitializeLib (ImageHandle, SystemTable);
    //
    // Find the PCI driver
, X0 N: J8 [0 l( d    //
    Handle = NULL;
    BufferSize = 1;
+ r+ }. w& b: K! J. P8 {8 f. d; {7 l    do
    {
      if (Handle)
      {
; a6 X3 o8 Q. F' k  R6 F        FreePool (Handle);
- i" _) I6 u. X% {3 O      }
               
. q, ]. ?  i5 v# h* c) O      Handle = AllocatePool (BufferSize);
2 n8 W; p  p3 p      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
  @1 j: c7 C) F0 k9 C5 N2 {
    } while (Status == EFI_BUFFER_TOO_SMALL);
2 [, q( `1 ?' v- Y) Y  j    ASSERT (!EFI_ERROR(Status));

1 f( J/ L% ^5 s6 g    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
% ~) L6 d1 Y* H% `8 O, ~9 r    Index = 0;
/ }" @2 p) K1 u( i  w! }    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
  u' @+ l" R* Y. x) Q    do
( t! d: c0 Z/ L    {       
        if(Index == NoHandles - 1)
         Index=0;
  q+ Y- N0 N7 `5 t7 C: _0 ?          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
! _; Y' N. Z' C& H) {0 p$ p        Print(L"THIS IS DEVICE %d \n",Index);
$ n/ w3 H* P9 ]; ~3 s' V    if (!EFI_ERROR(Status))
    {
( K' t& n) |4 T' u# m          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
    }
- n. M; n! ^; s# d1 u    Index++;               
    } while(Result == Not_Found_Me);
8 P# o$ o% ?9 v/ `# U. v5 Q0 Y- {1 x       
& U2 p, \. V6 S( ^0 n2 J        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习