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

kaurohy

/*
* myfileio.c
* Apps
*/

, R6 @- D# Q9 J#include "efi.h"
) ~% q4 q9 J$ i1 w#include "efilib.h"
$ _: n9 C; h2 |0 H: W
#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

# l- Q7 H, b0 R1 [+ u* d8 z! Tstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
) A5 H- J- r$ R. m: Y; |) l& H    EFI_INPUT_KEY       key;
- d' i) d; |% T) e$ B" |8 {    UINTN               index;
, P  w: N9 ~( z+ n- x   
  }8 q, {6 h- m+ W* y7 O    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
/ e$ |5 ~% s1 D6 g# Q        else
            break;
    }
    if (!EFI_ERROR(Status)) {
/ [3 Q: A5 }$ A7 V$ D! W        if (key.ScanCode == SCAN_ESC)
' r6 X) Q0 n# s2 D4 l) q/ {) {            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
9 Q) g, J3 H' y2 J. e    }
   
    return Status;
* }) w4 e. [7 Y( F* g}

8 {5 g8 @. A+ VEFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
" s1 j" @2 l  v9 v* N( U* D{
' p7 P5 U/ l4 L: ~    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
* b; h' c, @! ~, Z# G# N! o; h# ?    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
; N5 v. Z# l- H" B  S: J    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       

    FileName = L"NewFile.txt";
        Space_Key = L" ";
$ X1 x9 L) O/ \1 Z       
        BufferA = L"ABCD";
        BufferB = L"EFGH";
        BufferC = L"IJKL";

        Bit0 = 0xff;
9 s, Y) v: b7 c3 j% W0 X# u/ b) x: j        Bit1 = 0xfe;
1 z. ?0 V$ I+ z  f       
, p; |) H6 o8 D+ v6 x        Key_Space = 0x0020;
       
    InitializeLib (ImageHandle, SystemTable);       
5 t& c* m$ S: |% H
4 X" ]4 ?* @7 a$ `    DestAddr = AllocatePool (BufferSize);       
" @  {1 Q% t4 T" c7 b' H
$ `( @( H( S) H% t, s# ~    Print (L"Value of Bit0 is %x\n", Bit0);
' @4 C* i' S/ }- ?  c    Print (L"Value of Bit1 is %x\n", Bit1);       
- y; J, K! X* w+ w3 v. I) |
- A+ i8 r6 S9 w! f7 _9 L: U: i7 i       
" F, _5 g6 c; N    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
& [* X; V- S$ F+ V* v0 k0 G        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
) c- J4 Z4 i4 [) O        return EFI_LOAD_ERROR;
8 b4 r3 K' ~. L& H: U9 h    }

    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;
        }       
       
- K9 Z; c; r7 R8 c    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
* B+ J7 M2 l) a+ Q+ l            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
3 \6 B8 b8 R# F9 F: q2 W/ V9 |* o        }

' \1 i4 Y" ^% v- K4 s% q    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
6 A+ t3 o$ [, Y+ Y: W    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
* n) h) s. r$ i4 G    }               
               
- e! w4 v, D6 C: K9 l    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
    if (EFI_ERROR(Status)) {
! X* Z1 [# V- ?& Q/ c        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }
       
    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;
! m0 T6 }* T8 j( g( k# t    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
/ U5 x5 a, B2 t( h' H- O! y% [        Print (L"File Buffersize is %x\n\n", BufferSize);
. }, q1 b: ]" {( ]6 \    if (EFI_ERROR(Status)) {
" f8 Q: b6 @; c) Y' C        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       

    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 }. G) A. u3 P. q       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
" ^; l  C3 J+ r        Print (L"File Buffersize is %x\n\n", BufferSize);
4 H. ~$ A* x7 j9 I" Y* Y' b# n    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
9 p* p% P0 T7 Q7 h8 ]    }               
" ?$ ]3 e3 V4 v# H1 O       
& d1 w# C4 w/ Y% R" i    Status = FileHandle05->SetPosition(FileHandle05, 2);       
/ V/ P6 P3 t* q9 ^5 g$ b9 k# A    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
6 v" y3 W( G' _' T$ G9 D+ E        return EFI_SUCCESS;
4 j3 _2 Q3 l7 y# k$ v    }
- K' |3 i2 A+ X! {. h% f" m; F       
, {$ t8 _4 g( ~% `+ w# F    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
! G, u7 r% ]: W4 S) E1 T        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);
( ~$ @5 e, p# P        return EFI_SUCCESS;
0 u: r8 Z) o1 }3 ]    }       
( m, S  P  Q! m, r. i+ A; r7 E* E
: i5 W  f9 B; ?% C: J  F5 p//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
" G* b& m. k) V8 o6 J    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
, Z0 W+ x% O" A+ t) k  X" j
# b" Z+ @  Y9 @# F1 L    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
$ t/ x8 U+ W# ^    if (EFI_ERROR(Status)) {
$ t- c) z. l6 e" h5 ?        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
+ A- z* ]9 N1 M' g    }       
/ v3 S7 n4 U1 l8 h" j        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
2 e5 Y4 }0 g8 n4 A1 E0 Y% s8 C
    Status = FileHandle05->SetPosition(FileHandle05, 10);       
9 ]/ ~, M8 u4 Z    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
2 Q" I0 A0 H9 u! }, L* B& 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);
        return EFI_SUCCESS;
$ \, d3 {5 T* f: Z+ j    }

    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
! f, U) o$ L% X) P) D; K        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
: U2 `, N1 W4 O+ ?        return EFI_SUCCESS;
( ^$ p! q3 D& y( g) g# [    }
6 e8 n+ h; Z& g7 G8 q       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
* k2 Q- o4 B- u7 p1 h9 b        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
* z4 W/ E  p" |8 p! k3 q& \3 E
    Status = FileHandle05->SetPosition(FileHandle05, 14);       
    if (EFI_ERROR(Status)) {
) X5 r1 c8 u, T9 T9 x        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
! C9 \& f4 V, [/ i9 _        return EFI_SUCCESS;
9 [+ Y! z; Z7 T4 a    }
       
3 s9 v& I0 [+ _# o/ h    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
% l8 t; \) k0 S. V" |    }

    Status = FileHandle05->SetPosition(FileHandle05, 16);       
    if (EFI_ERROR(Status)) {
3 M1 }/ d, o. e7 H        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
+ H" ~  _+ I5 s9 h4 Y; Z" Z        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
- o. A1 ^6 b9 r1 G9 _9 |: U3 K3 b    if (EFI_ERROR(Status)) {
2 e9 k' V, d9 K0 x/ F        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
2 c% v2 P& L* v% J/ L/ l$ U! i    }       
) J4 U: @$ R6 m( b
//---------------------------------------------------------------------------------------------------------------------------------------------------------------

3 h: d- {0 V( Y& g    Status = FileHandle05->SetPosition(FileHandle05, 18);       
8 ?* x" q8 h8 g& c* d/ @    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, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
       
! L5 i' W5 B1 Z* j0 S' L    FreePool(DestAddr);       

    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
: _( N3 I) {1 E  _% u, L5 [- N        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
    Print(L"File Name = %s has already been created.\n", FileName);
  ~3 t( `7 Y$ W, X: H
    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
! [; _, c( U6 a$ X    WaitForKeyOrReset();

    return Status;
}

chenhy

//**********************************************************************
//**********************************************************************
8 t0 _* f$ ]" I//** Demo code for PCI configuration space I/O access program **
" q0 W6 V7 |5 v9 o//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
//**********************************************************************

( g5 l& Z. A  A+ w' K#include "efi.h"
#include "efilib.h"
5 \9 [+ D! p$ z9 `
#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001
, [6 H2 S$ C$ p6 d6 ~( U/ P8 x5 H
0 n5 ~" f* d  h: Z6 hstatic EFI_STATUS WaitForKeyOrReset(VOID)
: o1 o% Z3 T6 R{
: E1 Y7 m0 D/ O: Z3 ?    EFI_STATUS          Status;
+ L. U3 T( w' u( m" R2 e# X: L" Z    EFI_INPUT_KEY       key;
    UINTN               index;
   
  P: n" V; [7 b% [( e; Z    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
! \+ I& @1 g. t6 Q' ]            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
1 `6 a, k3 x% \6 \% z5 x    return Status;
}

# }0 Y, D; G* o, r" M) j  i  ^UINTN
AccessPciConfig (
. S5 i: g' h5 m$ f    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
. X  ?+ G4 c5 `9 f    )
{
7 g. t7 l' E! N% y- H6 `9 q        UINT8  i;
! d) Q" T, p- o0 }( C        UINTN  PCI_ATTR[4];
  F! O8 K  B6 W) N1 D# L        UINT8  PCI_REG[4], pci_offset=0;
* ~6 ]  L4 i$ g. r. B
! v; {3 w, B4 I% p( U/ r        //get device attr
        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
    if ((PCI_ATTR[1] == 0x00) && (PCI_ATTR[2] == 0x1e) && (PCI_ATTR[3] == 0x00)){
. w- M: i( s3 ~  G4 t& M9 m          Print (L"Device found at BUS = %02x         DEV= %02x        FUN= %02x  \n",PCI_ATTR[1],PCI_ATTR[2],PCI_ATTR[3]);
7 b( h  m! A. g' W' r          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
5 }' i& \2 E+ N: k. H0 e          //print register value
, V3 X3 ]! g2 o1 H7 a          for (i = 0;i < 4;i++)
: ]! ^  t# {$ \$ T. f          {
        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
) E7 z. d% z/ I5 D# b( R          }
          Print(L"\n");          
- U/ o+ ]' s5 `# e# I) \' y          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
    }

% |$ g- Z9 Y) J  g& l1 Y  p8 |        return Not_Found_Me;
}
8 F6 {* W* M' Q. n$ Y$ [7 G9 L& R' O
* t  x7 @9 c6 |7 o* ~: [9 F  CEFI_STATUS
1 u; ?8 A) P9 @% h. T; YInitializePciApplication (
7 x' R) Z4 o$ p    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
1 Y& [" i4 h: M3 P. s    )
8 k6 T+ [& q- J6 P; a9 N, D{
. }( I6 @/ B( `, _( f6 b9 |  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
& u" ^2 x- ?1 @- R  B  UINTN                   BufferSize, Result;
4 h% z2 J. Z  ~2 r  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

* I% Y" l$ t& X1 ]" W4 _    //
" D0 V4 l, J) e    // Initialize the Library.
+ M1 d4 A7 d! d$ k6 L& |    //
9 p! P7 w& C* t" j    InitializeLib (ImageHandle, SystemTable);
& a* y, P, D; m# z    //
0 @7 Q$ k9 G) X1 }$ O% y* d% p    // Find the PCI driver
, o0 v, g0 K* G: p" Q    //
8 w0 m3 u* `, T/ w5 g$ u    Handle = NULL;
$ L0 T0 E8 Q$ i4 L    BufferSize = 1;
    do
    {
      if (Handle)
      {
; q. A4 e2 T7 E# o) R% C8 I        FreePool (Handle);
: \( G& W. j  Z$ H. W+ W% ?      }
( d4 w. q* d8 B- {, L, k               
      Handle = AllocatePool (BufferSize);
# a9 y8 M6 T+ N! M      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

# \) S, }; q) G, D    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));
& O" x$ @& B5 u. R
5 U/ S9 X) Z" ^    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
; `" |$ c; E% d6 J1 g. [" r( I0 w6 h    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
5 l8 G; s6 B7 a$ i8 r    do
    {       
( @4 B/ q! q4 S8 C. c        if(Index == NoHandles - 1)
) |0 z' y; t- g5 g" u         Index=0;
0 ]# i7 E; l! t; f/ K- _: ]) e7 n: Y          else
+ p- q. D$ A1 |( q    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
( v5 F7 ]$ x7 ?6 g+ R        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
, \; V. p+ K9 w- H9 T: q8 f9 M//-          Print(L"Again now Result is %x \n",Result);          
" V& k5 e; @) v    }
    Index++;               
    } while(Result == Not_Found_Me);
6 L+ ?2 G& x* q       
; {9 E  k2 J5 H/ i        Print(L"Please hit any key to exit this image.");       
' N% E5 e  X+ B2 @/ v  f    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
0 g6 I5 x3 n& r4 r               
, g6 D! [* v0 }& C7 g    return EFI_SUCCESS;
& D0 ]" y: v# c' _4 `4 b3 {  e}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习