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

kaurohy

/*
* myfileio.c
* Apps
: i) \1 ^; X# K. l */

#include "efi.h"
#include "efilib.h"
, k' S, h! g8 Q1 U. f! ^
#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

/ N$ A+ H8 i% _) p0 L- Mstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
3 ?6 q2 x9 y; o* K9 e; s/ P, a    UINTN               index;
   
- H# B! Q% J2 y' [/ [/ h    for(;;) {
0 Y5 d* t4 j5 \: X! K+ s. M        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
: J- ^- b+ ^2 x            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
5 _1 z% v# x4 c3 C- m4 P0 k) a    }
7 \3 f7 }1 L3 [  b5 _* i) W. y    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
' u4 }8 z% G) V, n8 z6 G    }
   
    return Status;
: {/ N+ V. z# h/ G6 h. n% q}

$ r; g! |& e+ A; P9 S: w7 `EFI_STATUS
EFIAPI
$ Y) m8 \: Y& q; N/ \* R& JMyfileioMain (IN EFI_HANDLE           ImageHandle,
" ?5 ^$ N" m" x  d% d9 i& q  u. X- ?1 o             IN EFI_SYSTEM_TABLE     *SystemTable)
6 J* d6 B5 q, u1 x{
    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
/ ?* y+ }% x' Y! ?6 v( w( Q    EFI_LOADED_IMAGE        *FileHandle01;
. C9 r9 v5 h$ o, G2 s" o8 j& }% `/ y0 Q5 n/ X    EFI_DEVICE_PATH        *FileHandle02;       
9 @' C" T5 x( a6 G7 i- T6 u    EFI_FILE_IO_INTERFACE        *FileHandle03;
6 _/ r5 W0 K% s. @+ X, y* F    EFI_FILE                *FileHandle04;
% f, {6 S, m3 u/ P" _% C- N: m    EFI_FILE                *FileHandle05;
% I# N4 Z! f* ^- I1 t! Z6 x  g    CHAR16                *FileName;
3 D9 F5 [" J( c% ~- q3 e        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       
7 a- |) x9 h' ]
    FileName = L"NewFile.txt";
        Space_Key = L" ";
! S% K# M: ^$ Q9 U5 l: v       
        BufferA = L"ABCD";
4 i/ O7 ?% F7 _1 @- J8 S        BufferB = L"EFGH";
        BufferC = L"IJKL";
% o5 G7 ?+ }: c! A" U3 l% f# X
3 A* q+ p5 M. n" ^( M* p0 H: I        Bit0 = 0xff;
        Bit1 = 0xfe;
# O" O$ b$ N6 ?% P$ \" x       
        Key_Space = 0x0020;
4 a3 Q9 L6 V, n4 K1 {' o7 ], l/ `5 N       
/ w2 ~; a5 w- x" Q+ l    InitializeLib (ImageHandle, SystemTable);       

* g# ?. z. P' V: d3 ?+ y6 k    DestAddr = AllocatePool (BufferSize);       
2 _4 X* g! r. \) V9 P6 u
; X4 R0 B4 l' f! p7 l0 L3 y7 h% b4 x    Print (L"Value of Bit0 is %x\n", Bit0);
* L4 E) P0 q' @' q    Print (L"Value of Bit1 is %x\n", Bit1);       

       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
4 Y! ?0 ~& {4 Z' b    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
) w8 \9 @: V( [. L5 ^/ ^( ]* v        return EFI_LOAD_ERROR;
    }

/ d+ A0 w. N, ]% O- e/ r( a5 n5 H) M    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
    if (EFI_ERROR(Status)) {
6 P* Q' F$ l3 e3 O5 C            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
+ L" U8 g( C1 Z3 X+ a            return EFI_LOAD_ERROR;
        }       
6 V9 q% |' G% N       
1 I) r& W* X' ?0 ?" a0 Z    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
4 F/ q! J! D- j! a            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
9 P/ Q+ F6 c8 X! M+ X7 W        }

    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
* Y6 h# A& ^$ {. H4 p    }               
- e  A" f4 y3 }- k4 z1 C               
0 v1 e4 b$ f, ]2 I    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
    if (EFI_ERROR(Status)) {
- k; J5 }4 T/ d! s- `; a        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }
( k( b. F2 J5 v' I6 m3 Y       
( P( ?; a: Z% m    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;
. [6 Z7 T( ^& {" C6 f' f7 ^    }
9 \  R- d+ V! ?; p6 T6 d. p       
0 F8 `" Q1 ], _/ q$ J; @    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        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;
    }       
. V( e. N3 q# {2 T" z( h
: k6 A, _. m& C    Status = FileHandle05->SetPosition(FileHandle05, 1);       
; L& y* M8 E+ Y9 P5 `    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
2 m$ @- J2 r: f- P6 e4 _: a3 z7 r        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
; g6 L- E3 x* N, }; W2 x! b" L        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
+ b1 |- U6 }" R6 F3 u: T        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
) \3 E! {/ J3 T: V- ?) m        return EFI_SUCCESS;
    }               
, |" _5 t& Z- X$ t+ Z) l0 {       
% a* U* `1 a& S3 Y! T    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
" b, z* }# z+ S' C  r# O3 V        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
& x# r: y8 i3 O9 A9 ~        return EFI_SUCCESS;
$ S3 j! x! ?" c" \& S3 c' t- Y    }
       
    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);
: Y- B& j5 A7 P% H        return EFI_SUCCESS;
    }       

//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
9 ?- q) L( ]: ?# E) f" K& l    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
( n9 a- d% b$ n3 b, X! O9 l5 `* J; ^    }
7 ~1 d1 P  L5 P: \4 x( V0 Q
    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;
# l$ q$ S8 y% |2 y- d    }       
        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
, Q6 {% a' g, i. u4 z1 o
    Status = FileHandle05->SetPosition(FileHandle05, 10);       
    if (EFI_ERROR(Status)) {
5 x  A; \' o& @: u  s        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
* p. Z6 D; X; @/ \       
( c! L' _* w( y, j- ?+ Y: H    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
+ j% t4 C1 k2 T5 l# }4 ~    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
( b& G8 E: i, U2 {4 Q! k' `  y        return EFI_SUCCESS;
% v' p1 C1 y  U  ^: J4 U% `# l    }
6 b, j5 z7 E# G# @
1 J, ~7 P2 H4 l& S  d    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
( J, ^% i" r3 h! R+ O: k& [        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
& ?6 V2 g) R9 v/ n% b7 Z2 A( n        return EFI_SUCCESS;
    }
       
5 P2 i8 x" w) W7 Y# D    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
( B7 e  ?0 r6 g# q" X$ v# t4 x5 k        return EFI_SUCCESS;
    }

    Status = FileHandle05->SetPosition(FileHandle05, 14);       
    if (EFI_ERROR(Status)) {
& a$ n. H- y1 Z+ k! X+ b7 R- C        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
  ^3 B' ]9 T% X  I: V, R        return EFI_SUCCESS;
# C. J) Y& ^( Q# m  i. L6 d    }
       
: u+ f9 l# B& [- Q& E9 S    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
4 T. R- Q; A  f        return EFI_SUCCESS;
# K, Z& V, R$ \( A    }

    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);
; b3 k5 O* y# w/ j8 N        return EFI_SUCCESS;
    }
5 D7 m5 Y  x2 f( i       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
  d# I# ^6 r7 m. B. |% B    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
( i0 p1 e8 Z: P$ ~1 V( t        return EFI_SUCCESS;
    }       

//---------------------------------------------------------------------------------------------------------------------------------------------------------------

    Status = FileHandle05->SetPosition(FileHandle05, 18);       
6 L% B# z! ^$ a6 S    if (EFI_ERROR(Status)) {
3 i' U5 S3 W0 n0 p& \        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
! W; P4 |) C/ G- w) {        return EFI_SUCCESS;
    }
; u/ g; W0 g5 ~0 C& g) c9 a) K       
" s* \5 W2 V9 R* o& [8 L  l    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
3 H* ?1 R! B% i" V' U! n        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
, t2 U( C- S( P2 l- {' o/ R1 B    }       
       
    FreePool(DestAddr);       

8 _# ~0 `! I, }5 q- a/ b% @    Status = FileHandle05->Close(FileHandle05);
% @! c% m4 M+ N    if (EFI_ERROR(Status)) {
6 V1 i7 I$ A% _. F4 Q$ t, g; D        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
; o  j+ h# d& X6 Q        return EFI_SUCCESS;
- ~: ?: L8 F, t( Q    }
0 Z$ {% Q  p. t6 H# h       
; f! F) E1 b0 F3 i    Print(L"File Name = %s has already been created.\n", FileName);

" W" [2 p9 A7 R" x3 {    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();

    return Status;
) h, y8 N; T  \$ j- \6 r}

chenhy

//**********************************************************************
//**********************************************************************
6 C6 f$ a( J" ?+ x4 p6 `8 i" Z//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
* P$ w2 X1 X! j//**********************************************************************
6 R# X0 m/ G0 c1 k. Y. h% o# V//**********************************************************************

; V$ l5 B: w5 X# ~' L) }#include "efi.h"
#include "efilib.h"

) F; {' Y3 x1 G! U' ^" l" t#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001

' @4 Q' q4 Z3 C( M6 Q6 t2 k' Gstatic EFI_STATUS WaitForKeyOrReset(VOID)
" f( T. O& D, t" e% l* g( Y{
1 y3 R! k- T/ v' V    EFI_STATUS          Status;
/ G+ `) W8 \+ q    EFI_INPUT_KEY       key;
5 t3 U  M& \9 Q    UINTN               index;
8 H1 _1 T& T) ^" N2 S   
    for(;;) {
5 h4 G4 z7 N' m$ n        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
7 b: P: j8 C7 v- k7 `' E% t        else
            break;
0 S7 s2 W6 L5 R" v1 M    }
' W- R9 G) b% H8 m# s: @! d    if (!EFI_ERROR(Status)) {
9 J* _6 U0 B3 J- o+ B        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
9 P( g1 C: O7 D7 B/ A8 e   
    return Status;
, \# H& w: G" ^6 z9 ?, i}

UINTN
AccessPciConfig (
3 D4 z% `" j: k9 u! k    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
6 a( X- Z: ~/ G; K    )
{
        UINT8  i;
! Z8 @0 v4 L; j+ @        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;

' I# ]  R2 x5 `0 d2 V( ]0 u4 ?& U        //get device attr
        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
2 e0 y8 N: a/ x! D0 T    if ((PCI_ATTR[1] == 0x00) && (PCI_ATTR[2] == 0x1e) && (PCI_ATTR[3] == 0x00)){
8 B$ k6 H  y! y/ A          Print (L"Device found at BUS = %02x         DEV= %02x        FUN= %02x  \n",PCI_ATTR[1],PCI_ATTR[2],PCI_ATTR[3]);
5 K% M( P0 K3 E/ c- ?8 ?: L8 D          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
! z' O: F6 }0 B4 N& n          //print register value
          for (i = 0;i < 4;i++)
) ]9 n  r! ]! V7 z% |* S          {
        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
/ l8 f4 ~/ B" C7 F                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
3 F" W& e2 p; i9 P: |& K+ h/ K; o          }
          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
. @  K* |2 o, ~' w, \          return Found_Me;
    }

        return Not_Found_Me;
% f8 b9 E+ X% S; f" c}
, G: ^1 Z0 Y1 w+ O, O' o; O
, V, a& r  M/ v+ OEFI_STATUS
$ J: ?! |3 d3 w0 R0 v+ D; rInitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
    )
, M! @1 B" c# g( Y{
# u; y. A9 l& j4 p/ C/ R/ b; l3 W  EFI_STATUS              Status;
3 `8 D3 E  S& {5 r9 A  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
9 ~. A  \* X5 a  v2 m9 P  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

8 q' S3 i% F% i. Z    //
    // Initialize the Library.
5 l" S4 P7 b/ A! L& o' B5 ~- o: S    //
    InitializeLib (ImageHandle, SystemTable);
0 N7 W+ I/ [- X" L% T9 e    //
    // Find the PCI driver
    //
. Z! m9 @/ Z! F5 I+ j% P* @    Handle = NULL;
    BufferSize = 1;
' h& |' R" T# l* v; r    do
; C$ ~$ _8 x& J, t5 b5 x, A0 `! J    {
      if (Handle)
      {
        FreePool (Handle);
      }
+ }5 Y' G, O9 u               
      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
! w/ F. }- Q% q; e4 Z- O) B
: C* {" i1 r8 t. o+ B    } while (Status == EFI_BUFFER_TOO_SMALL);
5 F3 z' _- G( e9 r0 H    ASSERT (!EFI_ERROR(Status));

) H, V* ^6 s% |    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
% Y* V, |2 C' r" \$ z    Index = 0;
6 Q3 d9 B  X' q1 U    ST->ConOut->ClearScreen (ST->ConOut);
1 }) l! o; ?; J' T  N    Result = Not_Found_Me;       
    do
) Q% ~* l! @  q1 A    {       
        if(Index == NoHandles - 1)
8 O# G" s" o6 R( B) ~" m7 U         Index=0;
6 S$ j3 Q) o7 Y. D% _' w: O3 V8 |& `          else
9 ~1 S& I  O( e2 u, i7 G    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
2 k% j& g8 \4 p( O    if (!EFI_ERROR(Status))
    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
    }
    Index++;               
0 v% C, s$ m1 N% }9 q    } while(Result == Not_Found_Me);
       
        Print(L"Please hit any key to exit this image.");       
& N7 u$ p4 ~  m    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
! f5 l( ?) n- |5 @               
/ u' K6 _: @- M. s& m    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习