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

kaurohy

/*
* myfileio.c
* Apps
. y1 L" }# h6 g: N9 h; \; e3 a  H */
! [$ O1 W$ E/ T4 D3 Z
. g+ T5 [- r" R% P2 r" Z4 V#include "efi.h"
4 B5 I  {9 H; e* D$ K# u#include "efilib.h"

9 c$ y0 @* @# l! m#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

static EFI_STATUS WaitForKeyOrReset(VOID)
{
0 G/ w  K1 Y/ N8 d* l    EFI_STATUS          Status;
( l% D0 X6 _  I) E" D" h$ P9 J    EFI_INPUT_KEY       key;
% N  [. d4 ~, p. g    UINTN               index;
: b% ^3 G* g1 \$ b   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
' R8 k6 L$ s! N9 t        else
! n2 f$ L5 I4 A            break;
' N- G) V" P; A8 m3 h2 p  b5 X9 ]    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
" Q+ e. [9 I: Q4 }' j3 I            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
! |0 P! Z) F+ k. s6 \" s* i% [   
+ }! V9 ?% x. p2 p- Q0 ~. |    return Status;
}
) C7 n; G" j9 f. k' l
EFI_STATUS
EFIAPI
) X9 I) C" T" Y1 O. AMyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
{
    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
3 `( m7 ]1 B% ~/ R; L! C    EFI_LOADED_IMAGE        *FileHandle01;
" i. f5 m4 Y" f8 `" R+ Y7 ?    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
6 H. h! A" t7 G. ^" V2 Z8 T( Y4 e    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
/ `7 Z: q' o$ D: f+ i4 r. f% }        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       

# f3 N2 b% i! c$ B( F    FileName = L"NewFile.txt";
# p: M7 ^) ?8 T% X        Space_Key = L" ";
       
        BufferA = L"ABCD";
        BufferB = L"EFGH";
) V! k% U! P  L" u! m; ^        BufferC = L"IJKL";

        Bit0 = 0xff;
/ x$ y% p2 {' o( A- k' S        Bit1 = 0xfe;
3 J4 m7 ]0 a9 l, @9 _: t! \       
' N: p. n" T$ t2 J( q        Key_Space = 0x0020;
       
    InitializeLib (ImageHandle, SystemTable);       

, Q$ }! z/ r# C! U% z* w4 g/ l    DestAddr = AllocatePool (BufferSize);       

    Print (L"Value of Bit0 is %x\n", Bit0);
4 g8 b; g& a. b# J2 T$ G4 v) z6 n    Print (L"Value of Bit1 is %x\n", Bit1);       
# v  X' {( I# T9 M6 H* `) d5 O
       
1 l) {2 ]# g! E7 e, s    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
( y/ y# |6 f# G3 I    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
# w5 k1 M  _  w    }
4 {- g, v6 I3 C5 \5 {/ B
6 l8 ]/ K8 N% @5 c( [1 y    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
, e* ]( Z8 J, Q    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
+ Y* m! t) i1 x$ |  i" x        }       
       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
) |( V7 f$ w$ V    if (EFI_ERROR(Status)) {
8 o! {1 K* m3 A/ b) y7 ]* _            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
1 F5 W' K9 ]5 z5 g* D! b        }
+ ~, j5 K( z! X7 \6 W. J5 {
    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
: c! \  m6 ?7 T; Y1 H% C4 [% e    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
9 b- b) g- O  W        return EFI_LOAD_ERROR;
: P% ?5 h5 c7 a7 g3 }2 i: p    }               
               
; [9 r- [" j6 O. j: P7 t    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
3 o& g" ]' |& M3 d* c2 S; P    if (EFI_ERROR(Status)) {
5 |; K1 Q; g( \% Z        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
# x; f) Z! z+ X6 g        return EFI_LOAD_ERROR;
    }
0 }% ^' @- s5 m4 W       
5 j( R! K) K0 Y    Status = FileHandle05->SetPosition(FileHandle05, 0);       
1 r0 P- e  @3 ^/ Q7 g    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
1 U8 _' y  }$ ^% z* Z& _        return EFI_SUCCESS;
$ g5 V( H: E& m0 i# L- Z    }
       
; M) Y' y% H. h- o    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
8 Z- q7 S; B/ R7 |1 e9 O" n& F        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;
, ^7 S* Y0 q6 B- O6 t6 d3 o( T    }       

+ s1 S, F5 @( ]9 V, q8 ^; _8 e9 @    Status = FileHandle05->SetPosition(FileHandle05, 1);       
    if (EFI_ERROR(Status)) {
8 f: S" q7 G: @& O& O        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
) d+ Q( W0 ]! Y, V6 {. O: Q2 r0 u        return EFI_SUCCESS;
# W) F, c3 U3 Y    }
       
, k: j# a1 v) @% b. M/ C# ?    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
3 Y, {0 R- }1 i; i/ C    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
0 O1 Y; {# Y4 e0 S. T        return EFI_SUCCESS;
    }               
6 c8 \. ^+ K: }+ Z8 e2 |       
9 w' t- q8 g# V" A# W( X3 M  v1 t    Status = FileHandle05->SetPosition(FileHandle05, 2);       
0 [" O; x- K/ D  A: v6 y    if (EFI_ERROR(Status)) {
3 j6 w5 m* a% Z9 E$ q: B3 Z        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
7 i6 T' u, k8 `& N* P. V& l        return EFI_SUCCESS;
7 H3 u" B( A" ~+ i1 ?    }
       
% G( u5 f$ g+ l) o9 g: X    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
3 Z  \' C2 g8 D! k. ?! A        Print (L"File Buffersize is %x\n\n", BufferSize);
- A: x, q% I' x% n* G    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
7 l- _- B  ]. P2 L$ q6 N        return EFI_SUCCESS;
' g) X1 r) g6 _! H% T% {$ T    }       

* K5 n9 C" C3 F//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
5 R8 _% p& Z, ?% p( A$ l  s3 N    if (EFI_ERROR(Status)) {
6 G1 y% @! c$ f& t        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
1 B3 e) p& M8 {& s( `& j        return EFI_SUCCESS;
, ~# D6 n' Q  N& u) M) s    }

    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
    if (EFI_ERROR(Status)) {
1 d- I5 k2 G: g( t        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
) m# {- l% z; h8 R. E3 W0 {" @        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);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

    Status = FileHandle05->SetPosition(FileHandle05, 10);       
9 q+ Y8 n! s* f/ j    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
6 c' C* a% z4 E/ j% ^4 m    }
; j, R8 i4 d# ~1 W+ `       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
: V/ G8 h% B3 R' r- I4 x        return EFI_SUCCESS;
. ~8 M- G2 n& m4 q( I    }
4 X2 l8 G$ L# t
    Status = FileHandle05->SetPosition(FileHandle05, 12);       
9 n% D% W! \, \2 {    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
9 F  t! P/ R6 o* `+ P2 a. k    }
* N7 Y9 g; N6 f9 _! p       
; ]% D7 c! x' [    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
& y  x4 \, B% v' u    if (EFI_ERROR(Status)) {
/ F0 n' u% p1 C7 `7 H# ^        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
+ i5 w2 V4 z* U) H        return EFI_SUCCESS;
1 [' w$ l0 {0 i( N/ L    }

8 t1 E& {* Q. z8 X# j) U    Status = FileHandle05->SetPosition(FileHandle05, 14);       
6 \1 T: z# x# F4 F& _& d    if (EFI_ERROR(Status)) {
9 S; c1 A& n' q6 ?, m8 E8 ^        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
3 v% T; l; Z8 w6 j/ L    }
       
; r# R+ H! }0 P4 t+ S: v; E8 {" s    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
8 v) X8 n4 H9 D4 c        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
' R9 j  U: t% I1 |9 ]
4 u3 \; }: n' Q2 G7 ?- ^    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;
    }
$ l' @+ y' V. s; a$ k* ?! }3 W0 q* M2 m       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
$ W2 t+ H( o- a        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
2 T: N& g2 [7 Z3 L  {; c  L3 I        return EFI_SUCCESS;
    }       

+ W& e* [5 Z& T- W//---------------------------------------------------------------------------------------------------------------------------------------------------------------
5 J" L# k& V" ~2 a9 o2 z. g5 O
    Status = FileHandle05->SetPosition(FileHandle05, 18);       
( p  }& }! K  e' c; r' o0 T$ p    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
- O- y) j* n0 F: m" q9 f( d       
6 S9 P- Y4 F; @. O! v. x+ Y) D    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
! \# _% T- g* W/ i' N    if (EFI_ERROR(Status)) {
  f6 t, I2 T- S* b        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
" _' r8 O7 j) ^/ @5 z       
. D+ h& h$ k' b! s* k* H1 s    FreePool(DestAddr);       

    Status = FileHandle05->Close(FileHandle05);
% O! H9 {2 z* I) H6 G    if (EFI_ERROR(Status)) {
# m3 ~5 C1 C$ V0 g4 \+ z        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
2 b8 a$ U8 T5 M% T( h        return EFI_SUCCESS;
( Y* b! h/ Y* ^. t) J% @. E5 u    }
' G& r# Y& Z# z) X* @; w8 |7 m1 ~       
    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");
% W% ~5 U9 N9 S3 c" d) k: O    WaitForKeyOrReset();

& B: `- k  R# H7 L! P    return Status;
}

chenhy

//**********************************************************************
* [- |$ m! ]& p' m1 p/ j4 A) W% `//**********************************************************************
2 d( v2 t$ J; W5 K/ O1 i( N//** Demo code for PCI configuration space I/O access program **
, h+ W+ |( z  I: [6 [. h% \$ C//** By Lawrence 2009/11/20 @Taiwan**                          
2 x' V7 e+ ~# c//**********************************************************************
# A7 N6 C  @! i: t//**********************************************************************
# l) j# c9 d/ S, F, ^( C* }
#include "efi.h"
#include "efilib.h"
1 Z$ v1 p8 M7 k& n8 _
' S7 k% s5 P6 r: M' n2 k- E5 D4 ^#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001

8 w4 m2 s  g  b( G8 Mstatic EFI_STATUS WaitForKeyOrReset(VOID)
: Y. p; t0 Z6 @0 W2 f{
; P: |3 Z  ~* F5 H    EFI_STATUS          Status;
0 L; r& r# }0 }4 B5 }1 c3 _    EFI_INPUT_KEY       key;
! T) s  i/ b- r& D9 ?* O, d+ q    UINTN               index;
2 ]7 @: n6 t: J. i6 T" X- _   
" i9 }+ `: h  d8 g8 r    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
+ d& T; {- o; @; c        if (Status == EFI_NOT_READY)
, S, A7 F% @9 y4 a/ \            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
! d3 s$ v) \/ M    }
    if (!EFI_ERROR(Status)) {
( t, A3 t' D. `: N' H' \        if (key.ScanCode == SCAN_ESC)
& I; v7 ?) Z5 N" n, a            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
1 B9 C2 r9 B: l) m4 s5 I3 A    return Status;
}
  }, G% B% y5 J2 `" x  Y/ y
, c7 f; g9 a3 a& T; m! FUINTN
: G% \6 P- x2 IAccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
{
        UINT8  i;
        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;
! m2 F4 J  `. v& j7 F' N
2 V- H; \4 F2 S* V        //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)){
: |9 R2 R" }' x8 G5 d0 |          Print (L"Device found at BUS = %02x         DEV= %02x        FUN= %02x  \n",PCI_ATTR[1],PCI_ATTR[2],PCI_ATTR[3]);
3 o3 V9 g" g/ U5 `7 I          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
' E/ ]% y) H9 w! M! m6 s          //print register value
          for (i = 0;i < 4;i++)
          {
0 V6 r/ k" @1 _: _* A        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; }# ]5 P7 v: u" G! t          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
    }

        return Not_Found_Me;
5 b3 v4 w5 W) k& O}

EFI_STATUS
InitializePciApplication (
. l) U9 ]) J0 ^$ j3 G2 a    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
8 e; r1 `0 c0 ~: Z& S5 U    )
8 ?4 Z: w7 q4 S{
! L+ M1 b9 L6 Q/ F6 b  _) y+ `5 K  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
5 A. c( M+ ~( h7 U* D2 z2 y+ w  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

, b; J' j) I. A. Y/ o$ [0 n% S    //
# h) U: g1 H, A; v! n    // Initialize the Library.
" A; l( ?2 z% M( V, u" X2 F' W5 F    //
" f, {( x' t; l& O' r2 ?: y+ Y# g    InitializeLib (ImageHandle, SystemTable);
" C  V8 V0 z" f: t+ n    //
# g9 k1 ?: m; K: R, J( s1 G    // Find the PCI driver
2 B. Z9 B" y  n# I8 \# q* k2 a    //
% b9 {4 W) ^, v. ]0 t, c    Handle = NULL;
6 _+ d9 |% P0 E    BufferSize = 1;
    do
    {
5 V- _/ e, ~6 {$ j      if (Handle)
! q1 j6 l& a+ M+ O: @1 p: w, ^3 a      {
% ]9 i4 G8 v& x0 z+ p7 l2 c0 V        FreePool (Handle);
& m/ x3 x) h5 `, i5 k$ W1 t      }
               
      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
, v0 R  R1 [: h% h& r% }
    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));
! _9 c: \% m" ^* S5 K% F
" b" N; y3 d! f; V: W9 ]) D; {    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
& h2 ~4 \' ]) z, z' X    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
    do
    {       
; g1 P! v7 D! p3 V        if(Index == NoHandles - 1)
+ C- [! O) n$ p. M         Index=0;
) C3 a1 f# P" ]          else
: f8 e* a. _% ^1 }  j  M6 B1 L' L    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
7 O) d5 y) \+ m! n# A$ k6 D7 j        Print(L"THIS IS DEVICE %d \n",Index);
+ w! N1 f( _' J. O) ]    if (!EFI_ERROR(Status))
    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
+ b! H4 f2 s6 ~: L//-          Print(L"Again now Result is %x \n",Result);          
9 u( w' |+ w$ X0 W% ]% j) s2 s    }
# k1 X% [0 h1 ~# R    Index++;               
0 P& g: }8 S; a& ^" }% s, g    } while(Result == Not_Found_Me);
       
; V* m9 y6 J, d( e- Y        Print(L"Please hit any key to exit this image.");       
/ P  D- r# n+ ]$ x! [0 n    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
5 O9 h5 `" b8 N1 c3 o, @               
    return EFI_SUCCESS;
& X& s9 v( l, ^7 C2 X" d. Q}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习