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

kaurohy

/*
* myfileio.c
* Apps
*/

5 O! `8 D9 _5 Z#include "efi.h"
) q& b- }3 O' Q+ c- c* ?#include "efilib.h"
" V# T4 e& D2 z+ b
- }# N( n/ \! V& J: f" X4 c#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

9 S' H( N+ S  l# _) Bstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
3 }( G; s4 c& c& ^- I    EFI_STATUS          Status;
" E0 h9 P+ C- k! f) o    EFI_INPUT_KEY       key;
6 a  t' }8 w" @) X! {# A/ b    UINTN               index;
   
: W* u& L5 A6 p. f1 Z6 C& A5 v/ V3 E8 {    for(;;) {
1 \& q; O3 L+ x8 Y8 p, v0 Q        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
- O  G6 c6 c3 Q9 U" `" m# K        if (Status == EFI_NOT_READY)
- V/ t+ ^# v2 o0 z  p            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
) o7 s' r7 J! ^! G) Q; n/ e% m        else
            break;
2 R5 o1 ?% t& a+ b; f    }
9 _* V  w8 |" D' h1 S    if (!EFI_ERROR(Status)) {
% {+ M7 f( Y7 H/ y7 c        if (key.ScanCode == SCAN_ESC)
0 f! U. l5 _. Q# E            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
# V, v0 {3 b3 Q* H' ?    }
   
    return Status;
& z* ?7 z+ g! r  I$ |4 s}
8 h$ m& f1 h& p
EFI_STATUS
EFIAPI
. _: o% w& K) ?  D+ N$ W3 {MyfileioMain (IN EFI_HANDLE           ImageHandle,
. G6 K( X7 j+ f6 P% g7 {             IN EFI_SYSTEM_TABLE     *SystemTable)
3 F! M" U. e8 k5 ^{
8 `/ @0 ]0 F8 o: N6 X    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
3 b+ Y6 z. g8 }9 `" O; m  ?( x. x    EFI_LOADED_IMAGE        *FileHandle01;
0 s: c- M* R, z" d    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
7 H" ]; W, Q1 K+ U! x    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       
7 v4 ?0 I% i7 X& _% Q
    FileName = L"NewFile.txt";
* b! ^5 D: w( m+ o. W5 d6 Y. u        Space_Key = L" ";
       
- [7 U. O6 A8 P1 _! V* U        BufferA = L"ABCD";
        BufferB = L"EFGH";
        BufferC = L"IJKL";
* F, M6 ?' V: Q9 M$ h: l
        Bit0 = 0xff;
4 v" U4 @. k9 t4 \        Bit1 = 0xfe;
       
        Key_Space = 0x0020;
       
; K1 ?* e% s+ T( S( V6 N! x    InitializeLib (ImageHandle, SystemTable);       
: D5 h0 Q" g0 V
* s+ X6 S6 R7 E: `( y    DestAddr = AllocatePool (BufferSize);       
$ E# p6 P* V/ j) d& ]' B
" O# a: h% l" E1 B& I5 E0 o    Print (L"Value of Bit0 is %x\n", Bit0);
! [" l3 x' a/ [' o! m    Print (L"Value of Bit1 is %x\n", Bit1);       

       
7 h9 s1 e, E$ f    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
, e" l* ^# K8 H: J1 L% x( D4 t6 m        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }

9 a4 C( G# {) _; v, m    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
! ]2 ~' z: z* \/ [1 g6 N    if (EFI_ERROR(Status)) {
9 A& a* e6 Z# ]) ^' @4 ?            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
        }       
       
3 b8 I+ J* f; O: v2 x, z    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
$ A& P0 }: w, N8 J/ M& k            return EFI_LOAD_ERROR;
+ u' |3 S+ ?' o, H& b8 R        }

    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
2 m( d8 Q) g. ?8 g        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
$ i, m* t" B" D- D: N    }               
) |* x, X- o. W& K- G! b, r& D               
    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;
5 Z. o( Z# H7 n; o/ t    }
       
$ z  P7 F* u* a9 ]$ t6 [    Status = FileHandle05->SetPosition(FileHandle05, 0);       
% s/ ~% P* V" b! A    if (EFI_ERROR(Status)) {
" _1 x& l, E; Q3 O( [        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
. x" C, D4 M) f9 {  M9 Y, z4 D    }
! {8 T+ F( A2 j7 V7 g! t) i/ [$ a       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
1 ~. z" P) m9 G- w' o& x* i    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
6 h& V" _1 \( }# R1 B2 @        return EFI_SUCCESS;
1 A! q/ m! B9 S: }% \4 t8 Y    }       
& c. J; O8 V4 O: U
    Status = FileHandle05->SetPosition(FileHandle05, 1);       
* v6 o* X* C0 R' m1 k* w    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, &Bit1);
        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;
    }               
% `8 y3 W: a8 I- W! K0 h       
' t  K$ e' v- w9 E* p    Status = FileHandle05->SetPosition(FileHandle05, 2);       
  |. O$ c, f6 h# g. y    if (EFI_ERROR(Status)) {
; ^  n2 ]) _" F        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
# o$ b. N5 e& o; e1 G$ N# m        return EFI_SUCCESS;
    }
. f- U( K/ J- g) X9 N2 o2 P. o/ u       
/ X3 K" V0 I5 @4 q/ G- z" D# ?4 L    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);
) v% z: Q9 ?5 C        return EFI_SUCCESS;
    }       
* e3 {% ]& |- K8 S: O, P0 f
9 R* P; q8 p* U  \, E$ X  y' J//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
9 X3 b5 y  s( C    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
+ b/ a, }; E, j4 k2 x    Status = FileHandle05->SetPosition(FileHandle05, 2);       
8 u% v: [0 Y: W( I    if (EFI_ERROR(Status)) {
0 h. J& ?) W6 r2 ~        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
; B/ f* }6 y; u7 Q        return EFI_SUCCESS;
    }
7 B  J: R9 h2 T# c4 a4 v; T
+ Q# P4 B( _# l/ a) U: C5 l& g    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
# `/ N8 E: u7 p+ @    if (EFI_ERROR(Status)) {
" ^/ u' D, ^; K0 L        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
$ r, O% k* Q1 C8 i0 K    }       
+ A! v  O" n: f3 k- W        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);       
    if (EFI_ERROR(Status)) {
4 ^% |* u. V2 E) I5 U& y8 w        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
  W# w1 e/ N. n5 }! V; m( a    }
       
9 ~$ g& s! B& V- y8 F    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
9 P3 f0 X# P9 _; R    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
/ w( ~) X) T* a$ U% j
    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
% c- K9 ~5 n/ Q" D; U    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
- E+ j/ X7 f/ p0 C    if (EFI_ERROR(Status)) {
( N8 g) i. V, B$ f% ^5 n) h        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
) d- D$ d5 i$ U1 s7 v6 `    }
" m' d1 B' A& `+ n' ~9 n6 o& b" 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;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
0 U4 c" J9 Y2 Z9 b8 V    if (EFI_ERROR(Status)) {
+ n* J5 ?5 c. Y  D        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
: k3 \( A% Z$ ^: m    }

    Status = FileHandle05->SetPosition(FileHandle05, 16);       
4 B- v; R* G$ e' u/ A6 M( v    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
  G3 _. k% X& F4 S    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
; R1 @. H- Y# s    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
5 W# _# q# |$ Q- u8 t        return EFI_SUCCESS;
    }       
8 f* r; J+ d$ l1 ?
* p: O$ a4 X, s& }: i2 R//---------------------------------------------------------------------------------------------------------------------------------------------------------------
  Y2 N* j& K; o
8 E' T2 R2 P- x' n4 N    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
6 z" F3 V( \* y* w5 N) g8 `        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
+ E- v# _$ ]3 \1 Q$ a, A* X    }
       
  n& K7 x- l1 h8 z3 u  {! [8 |    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
9 e. T* |8 h5 H! g8 v. h* f6 @! M        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
) N* a% K% i) g# x! O/ k        return EFI_SUCCESS;
    }       
       
% u0 M( j# g' n) ]& u2 D    FreePool(DestAddr);       

    Status = FileHandle05->Close(FileHandle05);
6 p. w" _/ T  E3 ^- T% X7 ~    if (EFI_ERROR(Status)) {
5 {4 |% }: U) F/ _; K        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
$ o- q7 y1 K$ O+ |) ?# p# u  ?/ Q        return EFI_SUCCESS;
) L, |5 q: ]5 {9 W* p2 l    }
       
* U% _5 \- B: h3 G8 w# K& {0 w" m    Print(L"File Name = %s has already been created.\n", FileName);
5 Z1 h3 }6 {0 h. h1 Y5 d3 W
    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();
; o, y8 `0 e8 y& l' K
    return Status;
}

chenhy

//**********************************************************************
0 ]- O1 S9 J7 `5 K8 f( O$ A% o  W3 s: E//**********************************************************************
& x; }( Q6 I) E' J# ^//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
//**********************************************************************

#include "efi.h"
#include "efilib.h"
% j- H- [* i0 c
#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001

# e4 `' ]+ e$ n9 Y  n8 T$ Sstatic EFI_STATUS WaitForKeyOrReset(VOID)
5 i& {5 r( |; c1 n4 q8 I# m) z{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
5 q7 Y$ L2 x1 l3 W    UINTN               index;
   
    for(;;) {
6 y2 d. C$ n' A+ \4 M        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
7 W) ?6 }4 R# v! A% e            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
# K1 Q7 T& n/ h; Q' N/ b* F- u            break;
    }
    if (!EFI_ERROR(Status)) {
3 M: X3 ?6 v; A8 o' R        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
2 w3 Y! |; h5 G4 R0 q$ _   
    return Status;
}
( v8 g7 S5 b, w1 B" R
/ Q4 F: c" X/ m1 f4 i/ F/ rUINTN
& E" g- `* ~: r% hAccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
1 V0 J9 y% E( z6 x7 ?7 t    )
{
        UINT8  i;
        UINTN  PCI_ATTR[4];
2 j2 u; W' M: z2 o: L5 |        UINT8  PCI_REG[4], pci_offset=0;

: y3 O" j, T; t: `  K        //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)){
          Print (L"Device found at BUS = %02x         DEV= %02x        FUN= %02x  \n",PCI_ATTR[1],PCI_ATTR[2],PCI_ATTR[3]);
/ i5 y6 C( c( d% {3 t% a1 C          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
8 G. {( T) V( I7 C" W/ p          //print register value
5 x$ Z6 G9 F7 M1 A+ F% y3 C          for (i = 0;i < 4;i++)
2 R; W3 x  b/ t. t# v! V          {
        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
          }
& c/ ?5 F  a3 b: Z- z          Print(L"\n");          
; Q0 z* a* ^$ y6 ~3 a4 n          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
) I# o# w) K/ \$ R" K6 @    }

& b' W+ S. I& o/ V9 P* _$ p, B        return Not_Found_Me;
}
0 u4 M$ j1 m" Z" z
2 D( E0 V* ~) L  F" f1 FEFI_STATUS
InitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
! \5 Q3 ~) V' m2 |4 ?; n6 E    )
{
  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
! ?/ P# R* `( f8 W  UINTN                   BufferSize, Result;
3 y  n: r* h# d1 ?0 b9 Z. Y  UINTN                   NoHandles, Index;
% g' J) Y6 _' l8 v% ^4 }  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
$ t, q. k5 @9 E1 S. @" {1 b; X
    //
    // Initialize the Library.
# g6 ~4 O1 U' P8 J$ T2 u( Q( _    //
2 X1 z: l7 P# E1 I    InitializeLib (ImageHandle, SystemTable);
+ i* I: Q7 B, S, X( h    //
    // Find the PCI driver
    //
" P* Q. q" m2 k    Handle = NULL;
    BufferSize = 1;
    do
    {
% U5 L( c7 Y/ L# m0 C& J      if (Handle)
      {
        FreePool (Handle);
      }
               
$ Y4 g; [. ]9 V" |      Handle = AllocatePool (BufferSize);
: U; F; W( m6 u/ q! B6 p- z% S      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

4 u8 j8 Q4 d. E$ L    } while (Status == EFI_BUFFER_TOO_SMALL);
7 B6 X0 p0 a5 u+ g! }1 G" E+ m* [    ASSERT (!EFI_ERROR(Status));
9 i% z+ j' v+ E0 r8 h
- A% o- |8 F" A' t    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
# `+ t" a" ~3 e; q/ |    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
& Q" ?. L2 c4 ]# n    do
" q# F. z0 z+ N    {       
        if(Index == NoHandles - 1)
         Index=0;
          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
8 i: P$ V$ e; v# ^5 f        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
- a! t3 N$ ]5 k: s* K, t//-          Print(L"Again now Result is %x \n",Result);          
7 I/ a+ e8 i0 d. ?5 U" ^9 y    }
    Index++;               
    } while(Result == Not_Found_Me);
       
/ i2 f$ O, u: y0 Y2 L8 Z, p        Print(L"Please hit any key to exit this image.");       
6 d  \4 h' f0 X1 b& X9 q7 [7 [" K    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
! b5 C/ k% E5 U: [. b9 @               
    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习