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

kaurohy

/*
* myfileio.c
2 R- ^* _5 V7 E * Apps
' M7 r2 p# F8 u" q: V/ F+ m */
* m2 w# R/ T; ~2 W
#include "efi.h"
) {& L. F1 {1 V' r, T& q#include "efilib.h"
- s# k3 ]+ J7 a  _
' W. `  t7 e- F+ c! K#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

2 l7 h. V( ~% ]; z1 Z# {static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
2 [' q; Q& T! b$ K& n2 ~' ~) V: u' a9 f; w   
5 N( l: J# Q) }$ l) f# |    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
3 O, g/ v4 O1 Q# B. z        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
" _3 I1 T  }  N+ k6 x; @            break;
    }
    if (!EFI_ERROR(Status)) {
2 w0 u2 t7 l6 N: h! ]! D        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
# V( x& o* h0 H- D1 N   
    return Status;
2 n( }/ e1 C  u0 w) A}
; N3 I5 ]8 i4 c' \% y8 @1 f  T& D
EFI_STATUS
+ O/ [% t* P% i0 E; F/ ?; f' SEFIAPI
- H2 [0 M0 D0 L: @MyfileioMain (IN EFI_HANDLE           ImageHandle,
  D, e# [& S: d# p. A             IN EFI_SYSTEM_TABLE     *SystemTable)
{
+ K$ n$ ^7 ^( c. t# N4 F8 [    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
. x* W, X1 M8 G* ^! A6 U& d    EFI_FILE_IO_INTERFACE        *FileHandle03;
. x7 v6 H8 o6 }7 o- ^    EFI_FILE                *FileHandle04;
& B& H2 o4 P( f    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
$ f9 f& z8 W1 `  A  S        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       
. O5 s& i- X3 U) E
    FileName = L"NewFile.txt";
! r+ m0 z9 R$ @' Z        Space_Key = L" ";
       
- E; `9 G' e  J4 [        BufferA = L"ABCD";
        BufferB = L"EFGH";
        BufferC = L"IJKL";

        Bit0 = 0xff;
% p) d" g8 w* ~' ~0 a        Bit1 = 0xfe;
. n! m, w4 L1 l8 {; @       
        Key_Space = 0x0020;
       
    InitializeLib (ImageHandle, SystemTable);       

    DestAddr = AllocatePool (BufferSize);       
4 d; ?, Q2 c' ^2 r
    Print (L"Value of Bit0 is %x\n", Bit0);
  ~( w' q, z9 U/ j. O    Print (L"Value of Bit1 is %x\n", Bit1);       
/ L9 b! z8 }6 z" [. z; y  O
1 N. a" Q3 y  S: C       
" ~, G% y' Y2 C    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
+ E* `5 d# n: z& [% p0 Z' w    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
2 p0 _# ^+ i# C/ O) n7 V        return EFI_LOAD_ERROR;
    }
# a6 H9 Q; V' M: ^/ O4 ^: c
3 b7 j' @0 q) {    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
# w% @2 E7 j1 f% g    if (EFI_ERROR(Status)) {
, F, ^4 {6 \7 n' z( A            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
# f: \- E) p9 d' @0 }1 G* ^            return EFI_LOAD_ERROR;
        }       
5 J1 K" P* r* o8 F" z       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
7 [% E: W/ c2 I9 I2 o* [    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
% [% E" l9 m" r) ^+ h7 e0 o            return EFI_LOAD_ERROR;
7 X% x# G; {& T! w0 c2 N6 }% m1 w2 J  `        }

1 _: n* o4 C9 c- m! V& n% x    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
3 y' Q4 c+ v! Z- ], p, ^+ o    if (EFI_ERROR(Status)) {
) _4 W: B+ F- o0 l% p) d        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
" t& ^& r8 Z* y9 g5 L        return EFI_LOAD_ERROR;
    }               
$ I! q" |, t/ a# K  ~2 z0 O- b( D               
! @+ a. j" d. q, p& y    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
* I8 x( R; G5 S0 U' N' E7 G# u0 V    if (EFI_ERROR(Status)) {
        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }
+ H. U+ c2 E( |4 S% J       
    Status = FileHandle05->SetPosition(FileHandle05, 0);       
% ~3 X" [  b9 |: I0 N    if (EFI_ERROR(Status)) {
: y- N9 U* d7 ^4 n# d        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
/ l6 T( \. _, Z' h$ M1 S        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
% O$ P: q3 |- E        Print (L"File Buffersize is %x\n\n", BufferSize);
( ~8 l, M; W# K! K  ~! Q    if (EFI_ERROR(Status)) {
$ v" e9 S3 m( E- `  U5 t8 x        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
3 Z: P) Y: |- ~        return EFI_SUCCESS;
    }       
1 C3 q; o* q/ C" H  j& N
$ [  K6 u5 |- P9 R* Q( Y7 T    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);
# T$ R; p8 n7 Y5 j1 P2 I        return EFI_SUCCESS;
& c, V  Y2 m, ~3 ?/ f$ ^$ q5 c6 ^    }
( G6 E# w" f* D5 a: ^& V2 ?       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
' I+ e# k$ e- a        Print (L"File Buffersize is %x\n\n", BufferSize);
! r/ A, M9 d% ^' L, R  H  O    if (EFI_ERROR(Status)) {
6 `& r; `$ e+ I6 e5 \        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }               
/ U  M% G: i4 b       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
2 T& m5 f% r8 ]        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
( d0 d* @6 Q, q# F8 b        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
, N3 j% G: U8 E( r7 u# |+ b$ D        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 P7 N0 Q  D2 m4 ^4 p        return EFI_SUCCESS;
    }       

+ x$ U* ]9 k' i" b9 B& d- i' B) l//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
( i9 z9 s1 ]+ d+ V. y- x    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);
2 Y5 U2 n" P% e! I! D$ W        return EFI_SUCCESS;
. a& O6 U; `$ T" i    }
0 }, R( s' m+ ?, d2 R
    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
' w; c+ s! G3 }        return EFI_SUCCESS;
; ~+ c7 A" V4 h" T    }       
        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
( x+ O+ k( o! l/ v' N//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
6 U$ v! A, I4 s) P) H
/ N, u% s3 m" y, e, c6 D3 [, q' v    Status = FileHandle05->SetPosition(FileHandle05, 10);       
1 V& {7 K) Z- t% j/ |9 E    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
1 f- M7 F. p4 P1 H        return EFI_SUCCESS;
    }
! T  N1 ?; f  K4 F) ?+ V$ v       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
& S* k# W9 f) @* I; V    if (EFI_ERROR(Status)) {
& X' ]9 \4 Y: P' d- X5 W) t        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
2 l9 f( @% s) m& C        return EFI_SUCCESS;
+ r# M0 y. y; g! t4 q1 C    }

. x$ G+ H! D/ W* a    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
/ {, a/ s) n% |3 D4 x        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
$ K7 k2 t5 C/ `4 b4 o       
' y8 g0 |( R3 G# ?6 `0 j    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
8 H# _7 Q( D. U        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
( _9 ?5 Z' t3 {7 d8 K: J- t( o        return EFI_SUCCESS;
1 _. A; O* d, K# B5 p0 O    }

' r+ E6 `) U  E7 }/ c    Status = FileHandle05->SetPosition(FileHandle05, 14);       
    if (EFI_ERROR(Status)) {
  k1 x$ ]# ~+ F        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);
4 }9 W- |9 H! j7 t# I. L* p    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }

; q& o- L& c) g9 n; v* k1 J2 S    Status = FileHandle05->SetPosition(FileHandle05, 16);       
7 x/ a( E  V8 \, h    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
" ], o6 ^+ j% ^+ J5 N        return EFI_SUCCESS;
    }
: _/ k' n' c0 _; m. \       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
3 K; ^( x+ p, p  C, g2 \  T    if (EFI_ERROR(Status)) {
& z- o- S; T# f! t) R1 E        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
/ y. d6 ~1 b" Z1 T9 Y" o, b        return EFI_SUCCESS;
    }       

2 y" j9 |3 [/ P- e/ @//---------------------------------------------------------------------------------------------------------------------------------------------------------------
5 B3 A5 h# C+ {. s" ^/ X
+ E7 C5 Q, ]3 H% I    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
& H5 i! @* A, N0 ?" r9 g4 b! C1 X        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
: i- n7 w( d  V/ O& x9 f        return EFI_SUCCESS;
0 l" Y5 f2 [8 [/ h& Q- W; [0 \) l    }
       
2 }2 U' V( Q7 @! v; f6 E# \    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;
: c$ H! w% C! Z+ q/ A    }       
5 x, N2 o) A- G) [- V" b" G       
    FreePool(DestAddr);       

2 G% O  A- a' L4 Z2 q4 |    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
4 C% Z8 a2 w3 S    }
' k% s& [' D+ @# l- x  _) m       
" c/ M; P3 `; {6 q0 f! r    Print(L"File Name = %s has already been created.\n", FileName);

    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();
5 a- g/ K8 [  y/ ?; c5 _
* _9 o" h* P+ J/ P; A    return Status;
}

chenhy

//**********************************************************************
//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
. q  H2 j$ g! S7 T1 T//** By Lawrence 2009/11/20 @Taiwan**                          
/ `8 k! ?. b* T4 E//**********************************************************************
//**********************************************************************

; w5 V8 T9 K: T: A" R0 a; G4 n' g" e#include "efi.h"
#include "efilib.h"

2 N4 w0 U! R  f- B. }#define        Not_Found_Me        0x00000000
6 q5 M& a" d" L! E. C#define        Found_Me        0x00000001
% g& E. n5 n; `1 B, B3 \9 b" N
* h! x0 m$ K& N% _6 l$ Sstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
# t8 o* z* B% T! W% X. f% ?    EFI_INPUT_KEY       key;
6 S$ ^' v% Q! f$ O- O    UINTN               index;
   
8 W5 ~. Q  B: t9 l- [5 s3 M    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
3 h& Q. y/ l6 b            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
    }
: ~+ `) K, W- z- N    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
) T, r7 l. C4 j3 o( Y+ F            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
/ ]* U3 [) p" a% K! S7 l    }
   
    return Status;
}

UINTN
- h+ p' @4 {% P& y6 GAccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
6 t; _8 o- j5 D# |+ m0 C1 t' ]4 G{
        UINT8  i;
- |! N% v( x# L- l3 ^% \        UINTN  PCI_ATTR[4];
# q5 ?; l+ E" x9 f, m" P& M) M* I1 h        UINT8  PCI_REG[4], pci_offset=0;
9 x* b: N7 Y: i& N! i
2 B7 T. D2 _! @6 {* ~: u/ E        //get device attr
8 Q, ~2 }. H# v5 y        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
2 `2 Y: \/ Z$ v( S    if ((PCI_ATTR[1] == 0x00) && (PCI_ATTR[2] == 0x1e) && (PCI_ATTR[3] == 0x00)){
  E4 {7 b( J% n4 o4 J, {2 q- H3 {: M          Print (L"Device found at BUS = %02x         DEV= %02x        FUN= %02x  \n",PCI_ATTR[1],PCI_ATTR[2],PCI_ATTR[3]);
* U: x3 k4 A: q# T5 ?# C  A          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
          //print register value
          for (i = 0;i < 4;i++)
          {
, F# x+ t0 u1 b, I4 R  O5 U        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
1 l& V9 Q% D$ l( r& ?& L                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
          }
          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
    }

        return Not_Found_Me;
}

EFI_STATUS
InitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
! m8 p; s4 M8 [% k2 L. P& F$ o    IN EFI_SYSTEM_TABLE     *SystemTable
    )
{
  EFI_STATUS              Status;
4 |0 W7 \6 n, j. s+ j% `1 `  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
1 \& \" t# f; f  N  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
) s0 q, }( N# M
# [4 t  ?3 n- Q$ D9 }    //
5 u* J$ j3 a/ R4 f/ l) G    // Initialize the Library.
* d9 d. S, b( ]6 Q" V: Q5 p    //
) c0 |3 f5 n5 z. K' [1 T    InitializeLib (ImageHandle, SystemTable);
  C) F, z4 n4 j0 i, Z    //
    // Find the PCI driver
    //
    Handle = NULL;
5 p" ~4 q  j0 B9 D9 R8 p' x    BufferSize = 1;
+ N7 B) n3 o4 X    do
/ A9 s+ r* ^" q2 u* M- ]" |    {
) W6 G2 W! o3 c9 I$ t! ?      if (Handle)
      {
( j$ `9 k# @: q        FreePool (Handle);
- J5 y% E) o6 D1 U# U      }
: a( Z' y" \6 f1 [* g5 P! j! v) R               
3 ?) t9 p  S) Y$ t; K( G      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
- y6 u$ a; B( G% `* t' L
    } while (Status == EFI_BUFFER_TOO_SMALL);
  o( ^4 z9 j1 Z: W$ Y5 v* T    ASSERT (!EFI_ERROR(Status));

    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
9 \- {- f+ Y6 ~+ M, O    ST->ConOut->ClearScreen (ST->ConOut);
: z0 e3 |% C: [( D/ l0 k4 I    Result = Not_Found_Me;       
    do
    {       
4 [* I5 o- {( f$ ]( c3 q# k        if(Index == NoHandles - 1)
         Index=0;
! \/ R* Y  D/ {$ ^& s- c1 Q          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
! C0 L! b0 d: [5 S" ~" f        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
8 p/ C+ h3 Z' B* U9 R  n8 P: q//-          Print(L"Again now Result is %x \n",Result);          
$ ^; l) V$ U% _    }
5 m" G! l  h; h, G" _9 y; Q) Z    Index++;               
& u) S" k- j! r9 w4 X( X: O7 i    } while(Result == Not_Found_Me);
       
' ]  `, W( g0 P9 N        Print(L"Please hit any key to exit this image.");       
* Q# N* S; d6 q    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
+ @: L2 m3 V! [( @; J$ [# }               
& d5 U& l& ^4 _% Q, m    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习