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

kaurohy

/*
9 w+ S: _4 u5 G; v1 i * myfileio.c
* Apps
6 Y" d. A5 ]' t( d6 Q$ p) H8 A; Z */
6 P. I: ^1 ?- R+ b7 x0 T+ x: J
#include "efi.h"
! p( m) F- n5 Z#include "efilib.h"

#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
; M  K! z7 x1 b# f/ G7 ?
static EFI_STATUS WaitForKeyOrReset(VOID)
, J  j1 Z! E; q4 [1 V{
: Y* J, C2 e: A2 o/ f    EFI_STATUS          Status;
6 D: ^1 p3 j. ?$ G" B+ B    EFI_INPUT_KEY       key;
    UINTN               index;
   
4 R% B' @# Y" C& s1 d9 \    for(;;) {
- {. M2 C, S3 G7 G& D        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
! `5 _- I6 C- a& H  x        if (Status == EFI_NOT_READY)
1 H% X" o" R& Y            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
% w* l7 v1 s& Q  p        else
            break;
& i0 E" K+ x* t2 F9 V+ P" X6 N# d    }
. l& h$ G  O& S. R- _    if (!EFI_ERROR(Status)) {
/ K! C% B2 U; J0 L! u2 h. R        if (key.ScanCode == SCAN_ESC)
3 A! w4 G2 Z7 Y/ W9 k/ t8 D            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
/ D* s8 e# H1 k    return Status;
" |/ H! l# H) w( i  a( I  r+ K}

EFI_STATUS
4 Z& q1 f! l4 ^& _* cEFIAPI
) n, b7 L" v* d/ y' R( U4 M5 UMyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
  f4 X, S2 N" \; p{
    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
* w7 K2 `- {0 J    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
) i9 m6 O/ V- n; d0 ^3 O    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
) q' K8 d; ]7 ]2 U- d    int         Bit0, Bit1, Key_Space;       
( s% A0 K& r, H3 l- n5 b0 p* t8 J
7 ]( {( `" @. \    FileName = L"NewFile.txt";
2 ^& M, |5 S. r1 s) }$ p        Space_Key = L" ";
) r' S; z4 ~4 y/ L' E8 t% y9 M       
) t4 T* D( {. H* X8 [% q        BufferA = L"ABCD";
        BufferB = L"EFGH";
        BufferC = L"IJKL";

+ V$ g& J$ `& I/ k2 }9 D' l$ D        Bit0 = 0xff;
4 C; j% T2 N2 `& m        Bit1 = 0xfe;
       
% ~! i, F% Z7 H! D8 q        Key_Space = 0x0020;
$ H; Z% e$ P1 M* F       
    InitializeLib (ImageHandle, SystemTable);       
. j; n3 H2 _0 A3 _0 Y
2 w" q, d; G' n1 @# j    DestAddr = AllocatePool (BufferSize);       

! a6 v2 {5 i( `1 V: `4 l8 v    Print (L"Value of Bit0 is %x\n", Bit0);
8 m- M+ M0 _. V* E' o: e    Print (L"Value of Bit1 is %x\n", Bit1);       
$ A$ v# |& s: k) F# _
       
3 A6 p: @$ g1 R' O3 Z1 ^3 ~    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
: g- V+ Q7 Y. a1 M    if (EFI_ERROR(Status)) {
' I& |) m% U6 a$ n        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
" a7 P2 v5 h% G& t4 H        return EFI_LOAD_ERROR;
    }
' `+ v" P3 D% U' b
    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
    if (EFI_ERROR(Status)) {
2 c3 ^) I3 k$ `) S, H            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
/ e) t* |$ A; g- G3 r; e" j            return EFI_LOAD_ERROR;
        }       
       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
: c8 R  _" G2 }. S3 J8 `: I    if (EFI_ERROR(Status)) {
' [1 Y) `) ]. t2 y5 g            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
        }

6 H9 ?$ G1 s& [. k) r9 r    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
4 ^$ K0 `; t  W/ C    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
+ F2 M  t+ @8 c, V7 o( M    }               
3 M" |6 E. A8 _+ }% ^* j               
/ Q. k, @' B- r    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
, S8 j, L' ?. V2 c* k$ q/ U5 e    if (EFI_ERROR(Status)) {
$ k- D+ y8 m7 j& U        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
6 [' E2 m+ e% Y1 B7 g7 R+ o$ h0 ]        return EFI_LOAD_ERROR;
3 \& O" [- y; v' u0 C$ K    }
7 _* f) {5 i) Q' _; q% _4 W+ V: E       
    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);
        return EFI_SUCCESS;
2 |/ j; i/ M4 p) h0 g) }/ k. x    }
: t: b* Z* L9 j. d       
1 |2 ]" g. C, y/ v2 I) ?, ^    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
. r! W! ^; `% i$ z8 c        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
0 _' R* k, g/ Z1 h! E; p        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
# ?" I+ w: T- o  k/ V$ |7 i3 f    }       
8 u- D' S" P" R7 C7 J
    Status = FileHandle05->SetPosition(FileHandle05, 1);       
$ S% W' z% G$ d0 t6 C( M9 U, p    if (EFI_ERROR(Status)) {
* y  ]; V. e8 z; r* w        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
' _1 t% x' s$ ]6 i- x0 A7 r        return EFI_SUCCESS;
! l$ Y) `+ K6 n; q; t5 Z# v    }
+ c1 \0 }0 I; }6 N8 V9 P+ F) C       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
' C; Q& Q7 c% d3 Z6 }        Print (L"File Buffersize is %x\n\n", BufferSize);
: N5 |2 \+ @' A) }& D  j+ _# S* x    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
& M; k0 j  B6 k5 D( W        return EFI_SUCCESS;
    }               
       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
: J* P, C" y5 m- x: w        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
        Print (L"File Buffersize is %x\n\n", BufferSize);
9 Y, D3 X) L; x. [4 g1 L    if (EFI_ERROR(Status)) {
' M" j7 \) X+ h6 B7 i        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
  X7 w4 }$ m+ A/ M3 `    }       
/ Q- |6 e& d6 D+ a4 h3 q8 h
" o' Y0 k; }* u$ K+ d  O//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
* q. g% x" d: l7 |7 K    Status = FileHandle05->SetPosition(FileHandle05, 2);       
+ l% p( U' ~. Y    if (EFI_ERROR(Status)) {
4 |% H* n* |) l5 Z% P        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
; B4 C/ `3 `. F5 Y2 i& e4 @        return EFI_SUCCESS;
3 Y; {1 H" w, G1 O    }

# E( D0 x: Q# K$ F  |    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;
    }       
        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

! I5 e. k5 K- p/ T    Status = FileHandle05->SetPosition(FileHandle05, 10);       
7 j, N) D- t, H) f% f8 N2 {5 {    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
9 ]5 \( w1 Q1 e/ N9 _) q        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
2 W, H  f+ {; J6 ?7 F! K        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }

    Status = FileHandle05->SetPosition(FileHandle05, 12);       
- [, m3 d* h: {8 Q9 R% ], r) \    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
8 t0 h$ C" Y4 N        return EFI_SUCCESS;
    }
' L- A: ^5 n& c1 p( S- r* v       
! |2 e8 a" ^. d1 I% l    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
2 F, U6 d: a) k' K6 b" X    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
$ n1 L' }; Q; D, i5 a        return EFI_SUCCESS;
3 @; s7 G: c: ^/ I. S* [8 m0 u% ^  {" l8 z    }
$ [% r# }; [# }. k
    Status = FileHandle05->SetPosition(FileHandle05, 14);       
( T$ Q0 _8 q, a  u    if (EFI_ERROR(Status)) {
. K2 F0 i- n  J6 V4 ^9 [        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
: _. x( y: j4 \, w, Q% U        return EFI_SUCCESS;
    }
2 F$ m& w3 B# r       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
) n7 p: _! @0 n2 ]+ R) y2 k2 f/ h    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
! ^( r/ l" y6 D+ Y' C        return EFI_SUCCESS;
    }

    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);
        return EFI_SUCCESS;
6 a" }8 x" p4 ~, \8 t" D, _& e    }
: c2 p  O6 ^; p8 {       
; r% ~4 M! S- V9 N# a. z+ F    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
. ~$ T' w. e& H5 v  l    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
" Y8 v, Y: C1 X/ X        return EFI_SUCCESS;
, p9 ~0 P8 S( _4 y0 ~# B& B    }       

//---------------------------------------------------------------------------------------------------------------------------------------------------------------
& X2 u2 h- h" U- f( w1 i' k2 O
    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
2 H, b% s# j" q( o( }        return EFI_SUCCESS;
    }
9 c4 P1 y- _7 ]' D! e: H6 u7 n! J* \       
    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;
/ L: [' X& {* K3 X- X7 n    }       
       
5 @8 B$ F/ L$ F( P- A4 `$ Y    FreePool(DestAddr);       

    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
! ~' _* d' H9 v7 q: U        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
1 E/ Y& _, D/ A& ~    }
$ m, k. U3 a$ K       
    Print(L"File Name = %s has already been created.\n", FileName);
* |. f/ w7 X  ^: e' k- g/ l8 t- p# N3 m
    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();
4 R# y- f$ F) [0 K3 D
3 m8 C! s* v: K/ r& G    return Status;
, x8 b# D" S6 M2 H8 D1 V}

chenhy

//**********************************************************************
  ^+ W+ F1 E2 g! g9 m# Z//**********************************************************************
/ G% B! s8 n9 C1 M5 I) ?- p//** Demo code for PCI configuration space I/O access program **
. R) B6 m& ~2 D! D2 C; m//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
. w8 `" p! A; o: T//**********************************************************************
7 V, h- B1 H" q+ I
, J3 m( o  ]% y- I+ _#include "efi.h"
#include "efilib.h"

#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001

; i* \7 O+ l$ h+ E/ p' B6 ?static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
   
) `. q7 u- E4 i5 y6 _& C# h    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
: Q0 M, J' \5 P9 L    }
    if (!EFI_ERROR(Status)) {
, }; @/ h) f2 J- ~        if (key.ScanCode == SCAN_ESC)
, X  T5 V& z6 |7 U            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
4 T! x* a: Y0 n  e/ g    }
# I! ?: V" b9 P1 c6 r- O0 e" a! _   
    return Status;
}
! K1 X, E+ \) z
# A  `6 ]0 O0 g. ~* DUINTN
+ N4 N0 M) x1 x; kAccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
7 B4 M, i- J8 t! x- V    )
{
# ?5 g7 Y  ^0 u9 q$ g        UINT8  i;
/ H; I; Q- W0 O6 g7 P( e/ A        UINTN  PCI_ATTR[4];
0 O5 q5 n: y/ m( R        UINT8  PCI_REG[4], pci_offset=0;
5 t' O* x8 ]- x+ y# v
        //get device attr
" |+ ?7 E& a* t" W- A! v* X        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
7 [% w, |1 E+ A    if ((PCI_ATTR[1] == 0x00) && (PCI_ATTR[2] == 0x1e) && (PCI_ATTR[3] == 0x00)){
) |/ ~& B; x7 g2 @4 ~7 l          Print (L"Device found at BUS = %02x         DEV= %02x        FUN= %02x  \n",PCI_ATTR[1],PCI_ATTR[2],PCI_ATTR[3]);
6 c( q0 b6 }1 \          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
% r+ G  C+ i% O/ o! ]          //print register value
          for (i = 0;i < 4;i++)
          {
; {0 P0 p% [4 P7 w! p        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
0 o6 c# R# d- {: m. B          }
( `. w* X3 `/ [          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
6 i3 \0 P) Q+ V  A          return Found_Me;
* {  U; n  ~9 k$ N- E) J3 Z# Q$ ]* s    }
; b  d9 v5 K/ }- Q
5 n3 `$ [8 [# w, v# G. h, T        return Not_Found_Me;
}
( ]$ g1 K" X8 ~! m' C- g
EFI_STATUS
7 Y/ r' s  y' n' e1 G4 T( ~4 CInitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
    )
( B; F$ l( G! Z, ?{
& \3 k  Y1 a% P% O! j0 D+ Y  Q( y  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
4 u; M5 t, Z7 x/ v: u
- _' i9 T* }  g( \    //
& ^! \# P" c) v' {3 K2 R: I  u    // Initialize the Library.
# X2 e* k; o6 g% T1 F1 h    //
    InitializeLib (ImageHandle, SystemTable);
/ p1 k0 T& c* o0 e3 F( S, @    //
    // Find the PCI driver
) Z3 l; q, w! z3 A5 ^; k( V    //
' ^3 M+ X3 {. y  Q2 R0 i2 T0 F0 m    Handle = NULL;
0 [" h3 e/ l3 g6 P3 C2 B& \9 G$ P    BufferSize = 1;
, V) ~# C8 @, \  S    do
    {
      if (Handle)
      {
& U9 Y. ]/ v9 _/ {" t        FreePool (Handle);
      }
" m" v1 G" i) h: b" S               
: A0 ?' s/ Q% K1 M' q; t( x- m      Handle = AllocatePool (BufferSize);
7 S( f# I; |* q& O      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
* d; h+ q. [. ^
7 K. X& V  O: D% m0 J% T/ A7 d& X6 |    } while (Status == EFI_BUFFER_TOO_SMALL);
! E9 D  a( F8 g; ^: C    ASSERT (!EFI_ERROR(Status));
7 C- @! N3 v0 C0 Z) o7 L0 B' {
    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
& ]( k/ y* m+ @3 j    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
    do
7 `0 G& M& A8 F8 A4 U# f$ q    {       
9 Y2 p, l( G; n7 m7 f9 M4 E  o        if(Index == NoHandles - 1)
         Index=0;
; F% T, E2 _. S1 m" I. W          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
7 p, g* s( M$ ?. e        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
8 ?8 ^* ?$ x2 Q8 `2 f//-          Print(L"Again now Result is %x \n",Result);          
    }
    Index++;               
: X" q% a8 {1 B, L    } while(Result == Not_Found_Me);
       
8 y9 T) s  ^& \( c2 [+ v# J% Y$ A        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
0 ~$ f4 y* T5 G) \8 J               
& \3 `6 u3 }7 F1 {8 O: u    return EFI_SUCCESS;
& w% h% f; }0 r2 v+ b4 \}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习