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

kaurohy

/*
& ]( v7 n2 H/ Q7 ~ * myfileio.c
* Apps
( y, f' {7 t: ?4 q/ w( [9 W */
( y. T7 _+ d6 ]1 y4 k2 E
#include "efi.h"
#include "efilib.h"

#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

static EFI_STATUS WaitForKeyOrReset(VOID)
{
; }9 S6 l# H2 q0 O    EFI_STATUS          Status;
. X2 u3 r7 Y$ ]! w% s) d' B% a    EFI_INPUT_KEY       key;
    UINTN               index;
* M7 E6 D) {$ ^% d1 c   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
" |( o+ f: y- o$ Z8 z            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
, ~# g3 _3 o2 i2 M7 P+ g            break;
    }
, @/ l- [/ t6 v, q' F    if (!EFI_ERROR(Status)) {
' T1 M! F  j0 e5 J        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
( q) n; ~0 f, ?6 V8 _    }
   
+ q% Y$ X+ M3 P6 B    return Status;
}

$ r! ^2 z; T1 p4 E7 W: [# C% [EFI_STATUS
& Y, g' m1 ^3 ~3 P2 S( W" e! zEFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
; _- G' A% w2 I, K. c8 I0 f* J2 P{
    EFI_STATUS                Status;
; i* `7 v7 M, T! K* U5 [, \5 E. f    EFI_HANDLE                *DestAddr;       
" w4 H0 L8 {' K. u    EFI_LOADED_IMAGE        *FileHandle01;
' i5 s9 O$ p( X& K' ^& d& d    EFI_DEVICE_PATH        *FileHandle02;       
9 ]9 n  Z# q9 ?. d( M$ k    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
9 E7 n3 L5 {) s# \    EFI_FILE                *FileHandle05;
) f6 p% a. a# f' A  l    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       

7 b( x3 d% |6 e0 g    FileName = L"NewFile.txt";
$ f  x" F1 H# t, Z9 V( C% R        Space_Key = L" ";
! K$ N# }9 f* `- R$ ?* E, d       
% b/ {+ [* |8 O3 J! |) Y8 w        BufferA = L"ABCD";
        BufferB = L"EFGH";
& p: x  ~9 K& P, @        BufferC = L"IJKL";

        Bit0 = 0xff;
- P' j5 o* h0 ?        Bit1 = 0xfe;
       
        Key_Space = 0x0020;
       
6 u1 g! F, ^; J, D2 P" |) P/ S    InitializeLib (ImageHandle, SystemTable);       

5 N, @6 Y( \& ?. p    DestAddr = AllocatePool (BufferSize);       

    Print (L"Value of Bit0 is %x\n", Bit0);
9 N9 d/ p- F/ H( Z    Print (L"Value of Bit1 is %x\n", Bit1);       

       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
/ a) C) M1 s9 i5 B    if (EFI_ERROR(Status)) {
  g- U# H6 U0 I  k        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }

- O; u6 c2 u# U  U9 _9 @* }    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
    if (EFI_ERROR(Status)) {
' }( d; a0 c: u' }2 x$ e            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
$ ~, @, j/ r) Y6 k. [/ `0 e- x            return EFI_LOAD_ERROR;
  X" d' ~/ c  e* o2 l  o        }       
$ J7 ^7 w8 ?+ p5 }       
: O3 T8 H8 `! `    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
" }+ @- N. `/ v    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
        }
$ X( ~7 W/ f& M( N9 C
. k% n, l. z; O! {, |' L: u% [$ f    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
$ m/ l4 b' o1 }* k: }. o$ @  y( ^        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
# D6 H8 o" w; {        return EFI_LOAD_ERROR;
    }               
+ Q- }  |8 Y  d# R" U; o8 `9 X               
0 @8 @0 d2 G2 }, D1 V1 t7 E  ?- G    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
# w" w" K( e; k& L8 l! R& ]    if (EFI_ERROR(Status)) {
+ F# J0 L: z* @- R3 N. o        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
0 u( p' }: T" x        return EFI_LOAD_ERROR;
/ U" _& r8 m3 E5 o    }
       
    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;
0 J% s2 F0 K& j8 v5 F    }
       
3 O. ^" W& q( @! p- d. ]$ o! K    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
' ]- P) A. c/ Z9 v5 Y        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       
* e; o/ ~3 C6 i/ _% W! l9 t) W
    Status = FileHandle05->SetPosition(FileHandle05, 1);       
" c) @8 J  z9 a3 S    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
, |2 M5 {" n; _0 ]2 i% I0 z$ F        return EFI_SUCCESS;
" c7 P; q# [3 v/ F    }
; ^! H# S8 Q- D4 u, q, k" q$ V       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
# i3 I6 J2 r; u        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
2 V! u3 Q. Y; e7 s) Q        return EFI_SUCCESS;
    }               
8 v9 p" d% p" K0 g$ W2 \       
    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);
        return EFI_SUCCESS;
4 `" u2 Y! Z- W0 i. X    }
  q# ?5 U! k& j. U4 p       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
3 p* v9 S$ q6 [! T% W. S9 [. \8 o        Print (L"File Buffersize is %x\n\n", BufferSize);
1 M$ J3 g9 k$ d0 R    if (EFI_ERROR(Status)) {
6 }& O: k# K8 j0 G/ N  K' H        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       

//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
+ G# _1 U3 B9 ~9 Q- i    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);
; u$ p6 c9 e% S* q        return EFI_SUCCESS;
% _7 r2 V3 K$ P6 P9 }    }

    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
1 t5 J6 Y+ N9 Q; w2 s        return EFI_SUCCESS;
0 M$ f9 F" j& q, b) G    }       
        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);       
7 q  d) f3 D  u; @' P    if (EFI_ERROR(Status)) {
+ g5 Q2 g2 D# X( g6 y% {        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
6 n+ N& a8 u; t! x, ]& k2 ^0 Z        return EFI_SUCCESS;
    }
0 i( @. X0 T5 i: Q, 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);
/ s" b2 \3 J, E$ g% C/ A& Q* N" T3 t        return EFI_SUCCESS;
. V9 H! G& O5 z& G  c    }
. P, {. P7 y0 z/ n/ F
    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;
; l7 w  O" c# B; A  T    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
8 R/ M4 K6 L( x1 f$ O    if (EFI_ERROR(Status)) {
1 N* m4 \' m8 G& ~        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
4 d' C3 \5 W3 F; y        return EFI_SUCCESS;
    }
3 V7 A0 t5 n( y9 m- S& F! X
    Status = FileHandle05->SetPosition(FileHandle05, 14);       
    if (EFI_ERROR(Status)) {
! O: W* X+ G- c  U# a8 Y: i) y        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
4 b1 c7 c$ h; R" D    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
9 I8 @* I, n$ _    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
% |& a1 t2 P6 M4 L  _# K! w    }

1 O( e2 Y# i- {8 {: ?$ D4 G    Status = FileHandle05->SetPosition(FileHandle05, 16);       
    if (EFI_ERROR(Status)) {
0 W% o. s( t, a8 }        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
( F. I. w+ [' L8 R        return EFI_SUCCESS;
7 a" ^. L1 W+ @2 H2 Y6 O    }
1 b. q  \. Y, f2 v       
; C( h; ?7 ^& j4 ^: W/ t5 U: ~; O8 {    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
# y8 M1 m9 ~; z5 s/ {        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
  K3 y7 m9 x% g8 u& R0 A' c$ [        return EFI_SUCCESS;
$ M# H) _* x; j) ]/ x1 i( i- v- q    }       

, b7 ]2 p7 Y' |+ D, q//---------------------------------------------------------------------------------------------------------------------------------------------------------------
2 q- z$ s1 i. l
) e0 t1 s  V0 c9 X/ Q0 p7 E    Status = FileHandle05->SetPosition(FileHandle05, 18);       
+ a/ N" Z$ E9 I    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
1 Y0 F. t' D( a- E+ _3 a    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
5 o2 j3 r& a6 ]2 s( v7 b4 w# I    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
       
    FreePool(DestAddr);       

2 P* p2 G6 `- ], e+ N    Status = FileHandle05->Close(FileHandle05);
9 k+ Q6 q: v: Y% O( }2 f+ \    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
# W, t. E" B9 j$ k- c! w( U        return EFI_SUCCESS;
    }
       
& u2 o4 ?" g4 ]1 ]' T    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();

2 x7 c! A+ P" d7 G7 r    return Status;
}

chenhy

//**********************************************************************
! y! r0 D1 w  D. v//**********************************************************************
1 |% l0 m: E4 x! R5 P/ K//** Demo code for PCI configuration space I/O access program **
- U: A$ e* B' I1 c( d//** By Lawrence 2009/11/20 @Taiwan**                          
( n1 Z- z! p1 q/ |; P+ _//**********************************************************************
* t+ x) F" z) ]2 ^6 N3 @$ V* J//**********************************************************************

#include "efi.h"
# _2 m# C6 ?& g% a6 H#include "efilib.h"
% n; |2 s9 t2 P7 P9 q
#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001
' }6 K! }6 L% V( P: q+ k
static EFI_STATUS WaitForKeyOrReset(VOID)
{
( D6 ]8 p, \- W+ E* ]* x% Q; S& L    EFI_STATUS          Status;
9 @8 T2 @' w% _$ e* U- q! V1 U    EFI_INPUT_KEY       key;
    UINTN               index;
8 L3 g( _# g! Q; `9 S; ]   
    for(;;) {
' @1 m  }2 \4 I5 n: y        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
' X# V( O1 z0 I( J4 m& o2 I            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
5 F: M. [  Z1 N8 A: c    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
9 R1 t' n" z8 A8 L% n- A' I* e    }
   
5 m0 H4 y, u! v2 x8 D    return Status;
# Z$ f% R0 j- t}

UINTN
+ ]8 s. J* h6 p4 j4 f3 Y4 jAccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
7 ]! B; W+ T8 Z$ r1 W) l+ ?    )
, l4 c% w, {+ s4 i{
+ u( A! Z' L  b! O4 m' m        UINT8  i;
        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;
$ v) `* k9 L+ ^8 y* v
        //get device attr
+ ~; A, Y: W" H  \- A$ V/ ]        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]);
( \& m- {$ N+ G: x0 o          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
5 s  U/ W$ C# I          //print register value
          for (i = 0;i < 4;i++)
          {
( X/ j# h7 F& I        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                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));                  
: j( B  a( y- A( {! s, \          return Found_Me;
    }

' }0 L7 F" Y3 z        return Not_Found_Me;
! |" y8 i/ h1 m1 u}

EFI_STATUS
InitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
; e/ |( w* c4 I3 H0 `3 ~    IN EFI_SYSTEM_TABLE     *SystemTable
    )
, R5 j: X: n& P, F% H+ [/ V{
: Z" R9 }1 _$ ?/ m1 f; L  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
, ^# V( i8 ~# `, `6 ^  u- Z& y0 Y0 R  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

    //
    // Initialize the Library.
2 M6 O, T* t# T% K2 p    //
2 u7 B' A  s; E, E3 G: \    InitializeLib (ImageHandle, SystemTable);
    //
7 A. Z5 Y: Q) t+ M1 K. O# T    // Find the PCI driver
    //
    Handle = NULL;
! Y) e4 `" g! N7 V' K8 k$ C  \    BufferSize = 1;
- N$ `! p, u9 X, x9 w! a5 I    do
    {
" f, ~* {; D! E7 N      if (Handle)
      {
        FreePool (Handle);
      }
               
      Handle = AllocatePool (BufferSize);
4 }3 f6 m/ q( I      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
$ u+ s4 o) c# z$ k
    } while (Status == EFI_BUFFER_TOO_SMALL);
; w, ]4 j& D( b( T$ I) z, M# B0 O    ASSERT (!EFI_ERROR(Status));

    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
# C( e" g! n* o- J    Index = 0;
! H9 Y0 B7 a, V& ?7 _% D    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
    do
    {       
' P' X( q! ~" w8 t( h1 L        if(Index == NoHandles - 1)
3 T: S7 ~! t( _) J; J7 I3 a9 C! v6 X         Index=0;
          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
* m$ U; e& e( l+ Y        Print(L"THIS IS DEVICE %d \n",Index);
; G( C: p7 G0 f( A7 z    if (!EFI_ERROR(Status))
    {
  U- \5 c6 ^0 c( @          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
    }
    Index++;               
    } while(Result == Not_Found_Me);
       
( {9 E* Y/ p( `$ X' C        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习