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

kaurohy

/*
* myfileio.c
' P/ V- m8 U, u$ b$ E * Apps
*/
& S9 z& b( U* ?# w( C, t1 @, u! A
  f+ [/ B) d8 y; o: ~#include "efi.h"
, {3 @' O- a- R" J4 Q( e#include "efilib.h"

#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
% r3 P) I) q3 J* M/ \
static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
& c4 `. t* l  r, r9 Q/ A9 W/ J8 d8 t    EFI_INPUT_KEY       key;
    UINTN               index;
   
5 c# C; B1 s4 O    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
' N& e: a. h0 a        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
2 B& t, D- F) I) k( T            break;
    }
' W* ?6 ^% @7 `9 c$ V4 h    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
    return Status;
}
5 y. D% ?7 l2 T; r2 K
) K" t- b; y! E9 e9 T' b, xEFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
& A* E" q  Q& D- g8 y% [1 \% E{
  Z* ?1 T/ k: M    EFI_STATUS                Status;
& E2 |" J% R* G4 J    EFI_HANDLE                *DestAddr;       
, ]9 ?0 P6 Q5 z. O+ p, R    EFI_LOADED_IMAGE        *FileHandle01;
% f' q: V8 q) r3 T" v    EFI_DEVICE_PATH        *FileHandle02;       
9 c; {, J" s1 \  Z5 t% [/ Q    EFI_FILE_IO_INTERFACE        *FileHandle03;
1 d+ Q- \/ X9 o& ^! b    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
7 w8 |6 E, b6 Q' h. a6 ^0 z4 `        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
! {9 A9 g# N, b9 [6 U# u    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       

( G3 C( O( H0 ?' ^+ O9 [* O    FileName = L"NewFile.txt";
        Space_Key = L" ";
5 e" |" h6 e/ r2 _6 w, f. Z       
        BufferA = L"ABCD";
        BufferB = L"EFGH";
& T+ Y. x- W& ^" F! t5 j8 ~        BufferC = L"IJKL";

        Bit0 = 0xff;
        Bit1 = 0xfe;
       
        Key_Space = 0x0020;
- Z" Z7 O) ]2 m9 Z  X       
    InitializeLib (ImageHandle, SystemTable);       
# g' ^: u. I# Z. F
; A7 q1 M" }0 r$ o$ V8 q( r, K    DestAddr = AllocatePool (BufferSize);       
5 n3 k5 p6 g# i6 i9 C
    Print (L"Value of Bit0 is %x\n", Bit0);
5 P" G& u6 i& Z1 X    Print (L"Value of Bit1 is %x\n", Bit1);       

8 a/ V3 L5 U/ D/ B, g7 J4 }       
; w% B9 b) H; }+ @5 R1 a! [    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
0 T. @' Z0 _- m. |+ Y/ _% L    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
/ o) f  O, r$ J2 l& ~# P        return EFI_LOAD_ERROR;
    }
' ^) @( ~0 P# E, I; G6 D
    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
, A. J3 H. y5 ?7 U6 t- B" @5 @    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
+ f$ z7 `7 {1 N            return EFI_LOAD_ERROR;
        }       
- Z$ X# p* _$ `" A7 I( F       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
# K6 K- Y; Z: T1 o: E            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
' S! `* B3 C4 C# H  q            return EFI_LOAD_ERROR;
        }
) H. \! e% \3 @5 c# @6 J9 f6 x( o% z
' S- l; Q0 m# }0 W1 X6 A! @    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
6 N7 ?' J; [+ M  E        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
2 A5 U9 Z$ s7 V! O1 T& {        return EFI_LOAD_ERROR;
- R1 i. H5 L1 Q3 ?5 V    }               
               
    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);
+ M( f8 A; {9 V( D; F* |7 E! J! A5 M        return EFI_LOAD_ERROR;
    }
9 h& S$ I' Z: Z& r       
    Status = FileHandle05->SetPosition(FileHandle05, 0);       
    if (EFI_ERROR(Status)) {
7 T, O, _- w; t! P( ?8 ^        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
% Z# A. C$ C+ q* _# y( j$ I        return EFI_SUCCESS;
    }
: P  S7 c3 ^1 U& G1 n/ U       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
& M- U% L% T' `. i- v( R+ A        Print (L"File Buffersize is %x\n\n", BufferSize);
; q7 ~$ j9 U. ~- G9 a/ ?! s+ H1 ?    if (EFI_ERROR(Status)) {
  c4 u' \) K+ R8 \/ Z        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
5 d; p6 f1 O# j    }       
% f; x0 Q  M& q
+ _& D. C/ @+ y, \1 Z4 d% j. C    Status = FileHandle05->SetPosition(FileHandle05, 1);       
    if (EFI_ERROR(Status)) {
1 ~2 L0 K/ Q6 K" y) e( S+ C4 o        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
) n/ t& Q0 F: ]- ?# F! j6 e, g    }
! b) j* l7 \" v' ~       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
, @" B- N, Y8 z5 l* A7 s) G    if (EFI_ERROR(Status)) {
1 o9 V! ?' a8 d! W2 _5 o9 T        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
, O! o2 X0 ]/ b/ P6 y/ s  B1 A% q        return EFI_SUCCESS;
$ t* d: B3 G+ w9 B$ ^    }               
       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
/ m) `- G' p4 a! w' C3 A    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
" z0 T3 N5 Z. w8 P0 P        return EFI_SUCCESS;
    }
       
) Q- W! h5 I7 K7 y/ {8 ]. D    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
+ l0 N0 h# Q& ?9 p  E1 `" O4 t        Print (L"File Buffersize is %x\n\n", BufferSize);
- O) k+ g8 \+ G+ Q2 c+ U. k    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
. S9 I6 F6 D' S: Z3 C        return EFI_SUCCESS;
    }       
/ c" K+ o' l% c- P6 Y6 `6 a) }3 H
//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
+ g3 z9 v, C- c( V/ t' `    Status = FileHandle05->SetPosition(FileHandle05, 2);       
/ C# a+ Z0 J% A$ Q7 x& Y( H* o1 Y    if (EFI_ERROR(Status)) {
. d0 H4 U4 M$ q: i+ b/ u- p        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
7 A1 K) }& P! n, u) R0 r        return EFI_SUCCESS;
    }
2 @, h" w: _/ i5 H8 M' c
% B- s% [% x. F0 ?; \6 B/ N    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
& d" V0 D: V5 o        return EFI_SUCCESS;
    }       
% `+ W0 g- ~: a  s( F; b8 K        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
  U2 V( R6 J* v! r, O//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
" ]& S6 b; L9 ~  T; ?6 a
    Status = FileHandle05->SetPosition(FileHandle05, 10);       
( F" |" A6 z6 K    if (EFI_ERROR(Status)) {
; G. Y0 a% }7 Z; O* Q7 f        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
% N3 x: }* x+ n        return EFI_SUCCESS;
: J8 R9 s( q0 b& ]3 m8 E5 {0 _6 p    }
       
& \% M* ~$ ^  b& b2 Z+ X4 n% M    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
) I6 j2 U# |' \1 Q9 d8 p        return EFI_SUCCESS;
    }
1 A5 Y5 S, ^' J: y+ E
4 l: e5 J: g8 ?; S( U. T6 o    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
- k8 Z/ ?1 H# ^" Y) {        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
& k# b5 m% p; s  u0 ]- P4 _       
' t/ w- m9 @; H# _    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
8 Y% p+ s6 Q( C5 n6 u    if (EFI_ERROR(Status)) {
& d8 f, E! o7 G: Y        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
" m) }# L. j% x- z" B    }

! c; ~6 }9 L0 p2 ~& ^4 l4 i    Status = FileHandle05->SetPosition(FileHandle05, 14);       
    if (EFI_ERROR(Status)) {
# O, f6 H' M' A5 f+ G7 E1 b7 R        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
( q! [% F' @2 L        return EFI_SUCCESS;
  T# }5 Q) z: @  }; u2 p    }
, C4 j: Q" C; `9 Z% ?, x9 s/ c       
  O- J% l' x- Q1 G& s4 P7 b    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
* I- r& z9 D6 E; Q1 l6 L. V5 i        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
1 Z' B% c3 }% `3 k" V        return EFI_SUCCESS;
    }

    Status = FileHandle05->SetPosition(FileHandle05, 16);       
) ?9 _7 D# q( g    if (EFI_ERROR(Status)) {
9 O( f/ I6 @' D' ^4 P        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
$ v4 [+ N, N  @8 b( ]! o! p- B        return EFI_SUCCESS;
    }
1 ^: m3 ?7 K! i' I  L8 O" ?% m7 I       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
. o" `5 c! ?  p# s) Q6 h' \1 @        return EFI_SUCCESS;
! N$ U+ d7 [9 S* W4 e    }       
) a, q9 N4 q' K# J! n
//---------------------------------------------------------------------------------------------------------------------------------------------------------------
! p6 P/ i$ D; {& x  n7 T
$ G3 `2 F) f1 B; c2 S9 C    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
) S2 Z, H) g, c! n        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
  G. o# {% e) E6 ~+ N3 U- Z        return EFI_SUCCESS;
$ z- v  \# N$ g    }
       
: p6 x0 H4 a) S4 ^) }' |9 J    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
1 ^' w* {  n1 E& A' h        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
" W$ R8 E7 M/ B- c    }       
0 A( P7 r) M. Z$ P' ?       
    FreePool(DestAddr);       
: U: p- w& x9 `& Q
% ?8 E+ y5 m+ T% J    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
/ s  S7 b- i0 o% Q: a. E- a        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
  p  H" a3 w1 F. L% o# a        return EFI_SUCCESS;
    }
6 u) O1 G# f7 ?9 a+ _6 M- t  Z       
    Print(L"File Name = %s has already been created.\n", FileName);

! ~* ?# L. ]+ U6 |5 }8 C  J! u    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
) S4 r! _6 j. D3 O3 F) G    WaitForKeyOrReset();
0 y& n( q# `5 W" V$ [
    return Status;
7 b, X: M. X: }}

chenhy

//**********************************************************************
//**********************************************************************
$ M9 y! ]. u8 n0 {% U" e1 E- @//** Demo code for PCI configuration space I/O access program **
  \1 V' r' `6 u//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
//**********************************************************************
7 ]5 q% C& h8 W) b" m
7 N! A; x5 M# y2 _5 g7 L#include "efi.h"
& `4 X. M$ }3 z, d7 V* i* w: V% t#include "efilib.h"
5 ?& ^1 R2 u& `7 d% o
# g* ]. b. y8 x' V# P#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001
7 \) C2 \! O  q+ X  k/ q) p5 _
static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
! U7 ^' j" Q7 Y! g; n; H7 u    EFI_INPUT_KEY       key;
5 |. o6 M" G, K/ N( i3 ?  c# W7 T; [    UINTN               index;
   
    for(;;) {
: a+ R) Q+ j  B" x6 E# L        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
* Q9 j3 z; g% R5 ^) N- w3 L; P+ l        if (Status == EFI_NOT_READY)
+ q" J) Q3 f% m7 H; L            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
' W- ]- K5 P& {4 y3 h* e2 n    }
    if (!EFI_ERROR(Status)) {
3 Z$ p/ x5 |2 y, b        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
    return Status;
! L$ n/ ], U: t2 i) a! b}

UINTN
AccessPciConfig (
( _: {; o: e5 A& I# q    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
2 T* ^( H& g) T# z# X, R! _1 J    )
{
4 G7 J" ?, q) l, ?2 }9 z        UINT8  i;
        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;
5 k9 W. O4 L8 j" t  B
" e# J' T% v( `2 y6 v0 b4 b4 z        //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)){
% F/ t% g6 h, j! h9 f# `2 T* p/ C          Print (L"Device found at BUS = %02x         DEV= %02x        FUN= %02x  \n",PCI_ATTR[1],PCI_ATTR[2],PCI_ATTR[3]);
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
          //print register value
          for (i = 0;i < 4;i++)
          {
0 n% `- U) v6 ?+ h        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
          }
: z; ?, i: e. h3 [$ Z2 M          Print(L"\n");          
; Y) [! ?  A: s0 ]- k          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
. l! p! K2 c7 n. y7 {# y) G- E          return Found_Me;
    }
% C6 D' A% ~  u) H% O
        return Not_Found_Me;
}

6 V7 o) b: z/ R2 L  j$ _( CEFI_STATUS
InitializePciApplication (
2 m, k8 [' K7 G4 y' k+ y) D2 j    IN EFI_HANDLE           ImageHandle,
& ~4 C% N7 {2 ]0 _& o  k    IN EFI_SYSTEM_TABLE     *SystemTable
    )
$ Y1 \! |& r) X5 }) G4 G& `{
  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

    //
    // Initialize the Library.
    //
- z# k( @9 g# U8 W) X0 W    InitializeLib (ImageHandle, SystemTable);
    //
* t  \5 w3 a% ]' A4 j2 @& S    // Find the PCI driver
6 O' a7 y2 X7 m7 E6 E. E& ]) O; |    //
  C, S% S+ K& A% f/ `    Handle = NULL;
6 L6 S8 i0 F8 V    BufferSize = 1;
    do
    {
      if (Handle)
+ ~) q8 y6 g- r. L) Z6 t      {
        FreePool (Handle);
      }
               
      Handle = AllocatePool (BufferSize);
5 b, \# p" ]8 {" t      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

# k" q, f  z3 t    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));
* t; q# A5 U1 G8 ]* v" q' X1 X7 _
    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
3 @6 K0 k/ t( ~- P+ _7 @    Index = 0;
) E8 t- P/ ^2 u+ w+ q- }6 S    ST->ConOut->ClearScreen (ST->ConOut);
4 ^, d+ m% b0 c$ Q5 B    Result = Not_Found_Me;       
    do
    {       
6 x+ X' q# R* G* {7 f        if(Index == NoHandles - 1)
         Index=0;
          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
5 R  M$ p- g- W0 j7 V! B        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
/ V' \9 ?: R: l          Result = AccessPciConfig(PCI_DEVICE_TARGET);
3 W+ i. e* A% N0 c( Q3 y0 E8 U" k//-          Print(L"Again now Result is %x \n",Result);          
    }
    Index++;               
* y' @4 K3 `" C    } while(Result == Not_Found_Me);
2 v' l3 M' |( @1 P" B; V9 l       
1 K5 @( @' p( [1 V$ Y3 x        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
4 d0 ^/ A" m" D4 U/ N               
' |4 e% U; q3 ]    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习