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

kaurohy

/*
( k: Y5 }6 h$ t  _( N) J * myfileio.c
* Apps
" V) ?# a* B6 U" e1 B( ~7 S8 Y */

#include "efi.h"
#include "efilib.h"
- g! ]% l. S; G% f! x
#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

static EFI_STATUS WaitForKeyOrReset(VOID)
4 a1 z  m. P' o8 r1 _) O  E) d) g{
* m7 m2 \$ r' a2 l    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
& e; h0 o" T! ~: g+ x- f$ A# s, w% @        if (Status == EFI_NOT_READY)
5 n' p0 @% F1 Q            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
. [: h5 m  n2 q& Z3 y7 ]        else
9 h2 Y) d- d" [# R* \            break;
- C+ {% l$ I4 J3 A& c6 w1 e    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
' B+ F( v' D2 k: {* j            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
( R* h3 a# q1 m/ c    return Status;
}

+ G, q. Q4 ^0 Y) C# [: O, K' f- QEFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
( o: u  o/ K- _' Z0 `{
+ k: g# ~: M; u# Y  O3 L    EFI_STATUS                Status;
& o+ B3 j7 X/ r: H  C/ \+ L    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
1 o; x2 P% U' H    EFI_FILE                *FileHandle04;
3 ~# h- u* ?. P1 k    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
# v$ ~- F! ~/ x    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       
5 d: j$ p$ N5 P8 {$ O1 _) F4 j
    FileName = L"NewFile.txt";
5 e0 V( s+ e# s; z- R* G# q3 z4 ]        Space_Key = L" ";
. L# u* S% r: s5 A+ Y       
6 p' b4 Q1 U# V+ |/ R3 {        BufferA = L"ABCD";
        BufferB = L"EFGH";
: X0 k* o! b+ }, E        BufferC = L"IJKL";
/ a; Z+ i* `: D7 F8 e7 y: X
        Bit0 = 0xff;
        Bit1 = 0xfe;
& C- J0 _- ]8 ~) X2 U       
        Key_Space = 0x0020;
       
    InitializeLib (ImageHandle, SystemTable);       
  F9 ^2 Y0 s. a7 n& g5 {
, N5 r' F: {% u1 V0 o    DestAddr = AllocatePool (BufferSize);       
( T+ U: F3 j7 ?6 j+ \0 j# {7 w" u
2 c5 Y, A: [9 e( w: N    Print (L"Value of Bit0 is %x\n", Bit0);
    Print (L"Value of Bit1 is %x\n", Bit1);       

       
2 }/ X, K+ ]& R# x9 K: g7 R! L    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;
    }

5 \/ \- R9 U, x, E' X- l; C- `    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
. W4 _, @3 @* e2 X7 @2 ]( `: ]    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
9 O; W2 u# Q3 k$ A- ~% B7 G            return EFI_LOAD_ERROR;
        }       
       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
# X3 q+ `6 [$ x1 f    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
        }

    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
3 I7 f& _- }: g) D0 r        return EFI_LOAD_ERROR;
    }               
0 a% V  ^) E7 d2 y% s& r( n               
    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
    if (EFI_ERROR(Status)) {
        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
% h" f' X5 f- z    }
3 b: @" Q* B( t  q       
    Status = FileHandle05->SetPosition(FileHandle05, 0);       
/ ?# C  v; x( d, B+ n) \5 p    if (EFI_ERROR(Status)) {
* y; K$ c1 A) P) V# d4 @        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
5 ?* y/ o6 _) V( u6 E3 I; o       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
! o! |. K  r! J9 f6 a7 x        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: ]% `) N8 F9 @# N0 C        return EFI_SUCCESS;
' @: ~1 `! m  I  k    }       
+ q# M8 A1 i9 X* r( c5 ^
2 _% `+ V  `7 S    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);
        return EFI_SUCCESS;
    }
3 b5 }# a3 p1 H  |       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
- \4 I( z% ?; w3 e( M& a: ^        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;
3 W5 I% {3 R; K5 X) N  g    }               
       
# u/ a: A- P  k    Status = FileHandle05->SetPosition(FileHandle05, 2);       
# [) w5 Q7 g- o5 a5 \2 p    if (EFI_ERROR(Status)) {
* H$ X' ?% Z, |! Q        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
1 e! Y2 w8 ?2 u: ?& n# Y        return EFI_SUCCESS;
    }
8 f& u' w/ X# Q       
  \4 q0 N" }& I8 J3 }7 H) {    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);
0 h1 y; \4 z' M' t* d2 i: _        return EFI_SUCCESS;
    }       

//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
' ?6 t2 r1 }, n! d+ e' E1 l0 X    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
8 n6 R  }$ F& F. {4 ^        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
. h# N% X) c, ^& b0 g- y        return EFI_SUCCESS;
    }

2 s- V+ d# P/ j4 t4 n    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
! f! t- k% \$ k. Z+ s    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
2 A0 `, c( N) z& X: S) r        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);
3 q( G$ _5 k9 l( R//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
1 C* d, K( E% h3 o
- ?% o: C3 o- g$ E    Status = FileHandle05->SetPosition(FileHandle05, 10);       
    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
( _% `4 o3 @- M$ r        return EFI_SUCCESS;
; Q+ I5 i& H! G5 O! E    }
: _" Q6 o9 K; ?$ |+ o9 k" T       
& I% U8 }5 b  C$ e. g- N    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
. f1 {2 ~% q7 _9 V    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
6 W% p" f9 C' k& b3 j( x3 }8 _
- t% T4 Z" b, h: p$ i5 @0 {    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
+ O* Y- }) q" e4 P3 v; I, Y! ?        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);
/ }* a) p% G) d! G" M    if (EFI_ERROR(Status)) {
' l7 l0 N2 g' O7 r( ?  @        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
1 T* _8 k+ G6 N        return EFI_SUCCESS;
    }
3 r) C' y0 H/ H* y
    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);
        return EFI_SUCCESS;
  y6 Y! @1 G' A$ ^( [, S* D6 p; r    }
       
0 ^9 ?4 @- T: _5 W/ _) s6 E; x- Y$ W    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
; R- P  B2 d" r8 j3 J& @0 {* H$ P; d        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }

    Status = FileHandle05->SetPosition(FileHandle05, 16);       
    if (EFI_ERROR(Status)) {
% b  C& G. h/ P& z        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
! s& }) W2 }7 U; R- S( O9 O        return EFI_SUCCESS;
    }
. J: g9 r  P/ }+ x! A5 h- h- j  V       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
# C2 g; ~# Z& w. K% G    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
& s: u9 c! w/ ~/ \1 s4 B& K/ }
9 N% ~% f1 a+ t/ U1 `( h//---------------------------------------------------------------------------------------------------------------------------------------------------------------
3 ~7 H; F' d' E1 f5 D8 x( f
/ d% X$ d6 n' c! m4 c9 h    Status = FileHandle05->SetPosition(FileHandle05, 18);       
- r* q5 F: L8 S  ?& [% J    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
# k9 B1 A& [2 U8 R    }
8 C7 g. l6 n5 \5 }       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
- e& t, x) x6 U    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
! p& g3 e, k& l: n       
    FreePool(DestAddr);       
3 c  m% [; R9 L+ C2 z3 i) t% L
' i; W8 _, p/ x* |$ A8 p    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
! T- v" ]  \0 c, r        return EFI_SUCCESS;
    }
1 n- t% y) e; h  G% S5 Q+ @& [. O+ F! o       
    Print(L"File Name = %s has already been created.\n", FileName);

( c! k1 Q& x5 P    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
2 q& k7 g5 _+ r" f    WaitForKeyOrReset();

2 h) z7 ?2 e( M7 @" S5 q    return Status;
}

chenhy

//**********************************************************************
1 d+ y5 N( p3 b% ^//**********************************************************************
: Q1 B; N7 d6 r  |# t$ t" Y//** Demo code for PCI configuration space I/O access program **
- v4 r) i! [& @7 m//** By Lawrence 2009/11/20 @Taiwan**                          
( l6 M1 U4 A, y% O* x//**********************************************************************
! C8 D* n* w/ P& S* I+ E//**********************************************************************
) [! u1 x" C; {7 }4 a' \. t
; x3 u- t" h) g% h- F# P#include "efi.h"
- Y& v4 M: }6 m4 p0 b/ p6 ?#include "efilib.h"

#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001
8 @$ p; A' _$ R: M
8 y* M3 V, X# k" H& J) f6 Lstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
2 @" U  {; f4 [    EFI_INPUT_KEY       key;
    UINTN               index;
: n0 k- m, H3 v/ y& H   
; h+ V! b3 \! u    for(;;) {
) [7 U) p8 |3 u+ M3 E        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
# b& H! h. }9 I7 A& P6 y0 h# q        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
    }
5 Z" P. e4 O( g4 Y    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
6 h. W6 t  h  N9 D4 x    }
   
    return Status;
; C2 P4 x! ^& }7 |) I}
# `6 g0 f5 r8 V/ h
UINTN
. ~4 U" K& L; B! P+ V( A( [AccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
( u4 d& B# P) O( o4 y7 q0 Q$ Z{
        UINT8  i;
* E' V  I, h0 W( l9 \6 \6 F$ J2 ~        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;

        //get device attr
        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
* u6 J& I' \8 H9 v3 B* m' Q- _3 f    if ((PCI_ATTR[1] == 0x00) && (PCI_ATTR[2] == 0x1e) && (PCI_ATTR[3] == 0x00)){
( U; N* N9 p" V          Print (L"Device found at BUS = %02x         DEV= %02x        FUN= %02x  \n",PCI_ATTR[1],PCI_ATTR[2],PCI_ATTR[3]);
/ a  a  m. Y" D; N3 ]5 {          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
          //print register value
) g+ P4 x& i! x6 U$ _          for (i = 0;i < 4;i++)
          {
: y: t* O8 e  M* t8 l+ y2 O        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
          }
          Print(L"\n");          
! S2 g: f0 y3 K+ s; z6 Q$ ^          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
4 I+ c+ N4 l* t9 y4 U    }
& v) S$ n0 \3 R4 {
# o+ x; m: P5 ~" C: F: Q2 `        return Not_Found_Me;
. y  k. R* C: f* e) h}
7 L; Z8 v9 t4 U! D
EFI_STATUS
InitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
! _* g7 K% H# B% h    IN EFI_SYSTEM_TABLE     *SystemTable
1 w! |5 w7 {3 g8 A, |    )
* s4 M; `+ e7 M- F( d9 Q{
  EFI_STATUS              Status;
: |0 _* S7 n& p' f0 Y6 @$ O0 [$ ?2 h  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
- n# ^5 S2 l! q/ J& t
    //
" J) g4 ~& V: C    // Initialize the Library.
/ E3 Q: c' k4 M3 R% U    //
    InitializeLib (ImageHandle, SystemTable);
( g- [1 i& h: k. z' X5 g: \9 M    //
) a0 o0 o' D" C# M7 h    // Find the PCI driver
    //
5 m9 p$ r% d* M+ q" @$ H  E    Handle = NULL;
8 M: o# Y1 ]. w2 v    BufferSize = 1;
2 m7 B5 T" b- x  k- G) R; \    do
9 O9 s+ p$ d4 ~3 }5 g    {
      if (Handle)
      {
        FreePool (Handle);
% g' t! U$ Z+ `, ~8 z9 f; g      }
! O: H4 y' n1 n               
* F+ a; {  K  j8 W/ ?' d5 l      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

1 {7 f% x; n7 _    } while (Status == EFI_BUFFER_TOO_SMALL);
7 U4 n5 N& i( _- j. U4 a    ASSERT (!EFI_ERROR(Status));

  Y2 z6 c& ^' g, J/ c    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
: @, u/ @1 `% U4 [    ST->ConOut->ClearScreen (ST->ConOut);
. C, r, M0 }: }% \" F* p' H4 b. s) }    Result = Not_Found_Me;       
* c3 O. t6 \! N( O3 P. o0 i) }3 X    do
    {       
& ]7 L6 j4 D/ J8 q1 ^% R        if(Index == NoHandles - 1)
- K. w' g* v1 o2 j3 U         Index=0;
+ ^7 r/ J$ r0 T: o: s" w& c          else
0 n. F" U/ ?  h. |' n3 Y9 H( p5 v. W    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
, q% v' m- A) g2 x  d% U2 A/ p0 H    if (!EFI_ERROR(Status))
+ m) \- ^) ^, u( \/ A; W6 P    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
) h, e  a- E; J0 [    }
    Index++;               
( I" A4 O7 J  J7 c& B9 V    } while(Result == Not_Found_Me);
! h( p) ~4 v9 Z) Z! u( k       
& X! \) R3 Z' G/ [) \        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
. O" l, M" ]4 a" w/ Y* _' d               
0 _; Y% B/ n$ z! J" c2 \3 v, R    return EFI_SUCCESS;
4 O5 |9 e7 o. Z}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习