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

kaurohy

/*
* myfileio.c
- l$ R: M& h3 W' Z * Apps
*/

; i% ^1 |  v4 ?0 C7 b#include "efi.h"
#include "efilib.h"

#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
% ^- d8 V6 w" A$ Q6 @
6 M" f( P" P* d$ O2 I: b% u# T0 N* ~static EFI_STATUS WaitForKeyOrReset(VOID)
{
. `4 P' Y, N5 r. L5 t% Q/ E. _! ?    EFI_STATUS          Status;
$ s' ]9 ?( m' P" @; w9 c% Y    EFI_INPUT_KEY       key;
    UINTN               index;
( l: x# E7 W  F$ X% l' c   
    for(;;) {
9 P$ o0 n/ z! r4 V' S# B8 r        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
( ?) O$ J) t: H' }# @9 g        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
/ `7 `0 u. x4 D; q0 n3 i4 V        else
. r- V9 D7 Y2 z3 x- X& w            break;
    }
    if (!EFI_ERROR(Status)) {
$ b6 V4 A, @; Y! J+ @; w. o( s        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
5 d1 p" ~! G/ u4 T9 ]- Z3 H) i8 O    }
   
! s+ {% v& j: }- }1 m    return Status;
. K/ C( x! O8 t$ ?3 y4 v) q}
6 J1 o& H" A0 X3 b
EFI_STATUS
4 b6 t5 i' x8 ^- h8 e: a/ i# \EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
: F4 ^' r6 s4 U6 l0 R" f- E% G             IN EFI_SYSTEM_TABLE     *SystemTable)
{
6 m4 c( h) `# k  F    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
% D" ~$ {: F: x* v    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
, ^! ]. j6 K0 `    EFI_FILE                *FileHandle05;
  U( O/ I# O) Z    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
6 g( T+ {2 y3 D& s, _. u    int         Bit0, Bit1, Key_Space;       
7 Z3 y2 i6 l# l2 ?# X- s3 J
    FileName = L"NewFile.txt";
5 `4 a' G4 }8 D- t- @  {& g  {        Space_Key = L" ";
1 _9 R& T0 G6 A( |0 K; k       
$ f: d7 P6 b3 G        BufferA = L"ABCD";
8 H" Z& d+ z. t3 U; Q% ^3 g" x- e        BufferB = L"EFGH";
        BufferC = L"IJKL";
" ^" u/ l0 X5 \* b" |
' y+ Z9 \3 b+ r% K* q6 o* v        Bit0 = 0xff;
. n* b' `+ M8 ^# O        Bit1 = 0xfe;
       
        Key_Space = 0x0020;
       
! T" ~, x" E! _6 p7 X7 k    InitializeLib (ImageHandle, SystemTable);       

; k0 h: y& |& ?    DestAddr = AllocatePool (BufferSize);       
0 ~0 s4 Z, L! b' C
    Print (L"Value of Bit0 is %x\n", Bit0);
  C$ d' o' Z- t$ m    Print (L"Value of Bit1 is %x\n", Bit1);       

/ r+ ^' m: \. j: m! K, W       
  k5 W1 W$ O- C. Y7 [) `/ M/ A    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
6 E+ }4 F/ M- x( K( O    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
* ~; D0 w! I- d  ^        return EFI_LOAD_ERROR;
    }

    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
8 R* G% ?( y: P# @( h' |+ R- f1 c) `    if (EFI_ERROR(Status)) {
5 v1 b7 F, T3 l3 ~0 U            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
7 `5 I& k& G% W" z6 d- {/ p            return EFI_LOAD_ERROR;
        }       
       
1 y! q/ t: U; a0 \2 {: _) |7 c; E9 S8 @    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
/ Z. ]& m1 |) n/ [* |# W    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
, v1 l; O. @! ]$ ?$ E& ?        }
* J4 |" m8 ]2 B
: T$ E3 j. G8 K    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
6 u% z& {, S+ J3 B8 D$ S        return EFI_LOAD_ERROR;
6 {% A: k8 J) v1 [$ x    }               
               
7 R) I; N- }% N9 A% ^2 A+ W8 `* @    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
    if (EFI_ERROR(Status)) {
% l2 W) D) a* g  v        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
7 l% K; P8 Z+ G) P$ u( j1 n        return EFI_LOAD_ERROR;
& N. @5 c3 `3 M, m; u    }
       
7 M$ F0 |+ l" w/ ^6 E/ W& Q    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);
. _* q2 T5 O, o        return EFI_SUCCESS;
1 |% M# x. Q5 F4 g7 P+ l9 W$ y    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
/ L: w- B0 U, ]- i- e        Print (L"File Buffersize is %x\n\n", BufferSize);
' q/ L9 L/ i% R% j) `6 w- u7 F$ y    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
- r: c0 m2 N& V" K5 g" V& e% C        return EFI_SUCCESS;
    }       

; y: p" b8 v4 ~' a! l) r$ y    Status = FileHandle05->SetPosition(FileHandle05, 1);       
    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
8 z1 k* I# L8 e! _% c0 B# x        return EFI_SUCCESS;
' V5 S- b* a6 ~# _8 \- J    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
9 ~# J, p% N7 q  D9 w    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
, s" Z+ d! a) k  M8 C2 C; e        return EFI_SUCCESS;
6 g* r) V  A4 K! ^    }               
1 V: n9 s+ c& B7 \; t8 e5 b- W       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
6 z) G( E: n2 _/ x# d/ S3 t/ N    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, BufferA);
% H5 v6 |+ Z$ _4 f& N; D" V3 v        Print (L"File Buffersize is %x\n\n", BufferSize);
  I: I! E4 n; j- Y; ?    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
" g8 F( Y/ z5 U1 M    }       

+ T" g  I+ i" u, o4 m4 l//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
% ~* `1 F+ ~, @4 \  W3 X) {) r    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    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;
  K; w3 q  h3 P    }
5 h/ k9 z. ~$ t! s# F, x2 N- k, p
4 N6 v. y* s* I) H    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
) ]4 u8 D% f9 R9 b  B# J3 `    if (EFI_ERROR(Status)) {
( `7 m, q" B+ M+ G        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
+ i" c2 A% `& R9 Z! v7 D        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
8 V4 o% T# a! _: F
$ a6 W9 M" e9 M- Q    Status = FileHandle05->SetPosition(FileHandle05, 10);       
+ M4 T  x/ n5 i/ T% E# m, L    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
8 K% Z" c% J3 t- F       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
& h5 s! h- }$ @% q4 v* s    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
! P. T2 |% _2 I) O7 b/ s* w        return EFI_SUCCESS;
    }
$ a* Z$ o6 D/ M9 g! V
    Status = FileHandle05->SetPosition(FileHandle05, 12);       
9 T' }. n2 Z- ]; d; [6 D    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
( d! ]8 A6 z! }) y! V        return EFI_SUCCESS;
    }
+ b0 m# @! r/ Q3 _8 s' x       
* f9 Y7 F- e& d0 W7 Q& c/ K    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
( m% t- H4 s0 ^& H* _' _
    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;
    }
       
$ ~- y6 M8 O! b" U' ]5 `$ }0 c/ c    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
' \- D) F0 l) u    if (EFI_ERROR(Status)) {
7 a% T- m5 W- Z* Y        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
! {# ^. b; n7 G9 \    }
1 I! n% J$ h6 y+ \! S% `
    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;
    }
       
! j$ J2 A( \. ~9 B1 v# T# m    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
& C. h: `4 l8 v/ u    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
9 s" j, y% A  U6 C" |9 R        return EFI_SUCCESS;
: N* J/ M  r' R# z& F; `. E    }       
' Q/ l: U; i1 r- V& ]) j9 T
% S: Z: ?. g6 K$ K) Q# s//---------------------------------------------------------------------------------------------------------------------------------------------------------------
0 g9 G- D0 }) `4 L2 h, H  q
; {, _( ]" U& V% M- h    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
6 W; d1 m3 j$ o8 v$ f        return EFI_SUCCESS;
0 Z5 b; M5 i7 ?7 s; I8 }' \* Z    }
6 }* A% g5 P$ I. V3 y0 A       
1 K# B9 m5 S# Z    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
1 k% Y1 F7 C1 g, l& f, f        return EFI_SUCCESS;
4 d( |% r! {  k3 u$ u    }       
: f  R6 ~# e2 m       
    FreePool(DestAddr);       
; v( R6 d6 v  Q3 D/ H' c
$ u8 F8 p8 S' l' D7 Y1 @; ?* i    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
) `/ ~  e( a7 p/ f) p, r2 v5 y        return EFI_SUCCESS;
    }
       
  y1 ^$ x/ E5 z1 t2 ~    Print(L"File Name = %s has already been created.\n", FileName);

9 [  a8 l* B2 K0 v& {. M    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
( Q( a3 K. L9 O, ?: T& r7 C    WaitForKeyOrReset();
! `, V7 ^% q8 j6 h0 L
8 b- G/ B& x/ ]7 E' m1 f    return Status;
}

chenhy

//**********************************************************************
+ G  n/ S2 I" f2 p0 a% d0 I//**********************************************************************
2 |. g; z8 s5 Y+ P//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
8 M) q( T$ d9 t! c2 }) k//**********************************************************************
//**********************************************************************
7 L2 p6 y# O8 ]+ u4 K4 X" A; N
9 w) f5 ?6 c) i5 o8 r#include "efi.h"
#include "efilib.h"
7 p) S& X/ u/ q  f
) z- K8 y! ]( T#define        Not_Found_Me        0x00000000
$ V" c! b. l7 k1 L4 A; k#define        Found_Me        0x00000001

static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
* J/ M- Z7 m7 _2 K! q2 C/ G    UINTN               index;
   
/ h  @: {" z& `% U3 _3 U4 d( F    for(;;) {
) B+ ?- l' `2 o, I        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
9 ]; G0 u& l* C! |  y            break;
    }
# J0 D/ f* ~$ v' X) u$ L    if (!EFI_ERROR(Status)) {
3 b1 w# s& {+ j/ p        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
) g7 I( O, G$ [0 t: Y. u0 d  D    return Status;
4 R: Z: W4 Y; k, c8 J$ X}

9 ]5 ~' z* z; T1 j# w# eUINTN
2 m0 x9 f/ v9 ]) VAccessPciConfig (
- K/ i% }. e; q+ x/ `/ X1 {4 b" q    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
- |: C4 y0 i$ m    )
{
% ?/ W1 m' ^0 g2 l) a' e/ N9 l/ G5 g        UINT8  i;
3 K* V7 L5 n: U: r, L$ J7 T4 ^        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;
. }3 K' s: y+ e' A
        //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]);
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
          //print register value
          for (i = 0;i < 4;i++)
  b7 o+ W  z2 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);
* u4 @  G2 _3 ~8 r3 m$ }          }
          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
    }
* f* o2 F7 l, s. N9 e7 b
        return Not_Found_Me;
; q+ A! u) Q) V- R8 m2 R( x}
, t0 B# I3 m; M* a3 H' |$ e, Z9 i- P
EFI_STATUS
InitializePciApplication (
* R7 y, B# l& O3 ?# h2 t# k2 H( k- F$ F    IN EFI_HANDLE           ImageHandle,
! Q/ J: A8 m3 H8 ^+ g- ~    IN EFI_SYSTEM_TABLE     *SystemTable
( ^7 n9 ]- F+ g) Y2 V5 F- }, \0 {    )
' T. F5 I$ g( ?0 E* Q* z/ V{
  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
# O; s0 S3 B; }; ~9 l- h  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
  f4 j2 l! @, b* s
    //
% O" h+ s" |9 i7 G    // Initialize the Library.
5 C/ V" N3 R) q" Y$ l* E1 q    //
. e/ \: M8 F+ q! H1 h8 m+ q    InitializeLib (ImageHandle, SystemTable);
    //
0 H4 B/ O3 ~9 J8 d) ^  o    // Find the PCI driver
* A' Y6 b0 e. i    //
6 R: G6 H( R2 l: z    Handle = NULL;
    BufferSize = 1;
- V' C6 w% X5 d    do
    {
4 M+ v" a  \! K      if (Handle)
3 _. D+ R! u# ^6 z+ [      {
        FreePool (Handle);
, h& ^8 X) Q& F7 f      }
8 U/ o" V  l4 e6 ]2 [) t) @, O- }               
      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

; l( v0 `9 H2 o' P& |    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));
; Y+ H& J- Z; `
: u  f) @% Q4 R2 K! d, ~    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
+ D5 v6 V$ I+ Z: E/ \3 r    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
9 L: T4 m  N5 i9 G    Result = Not_Found_Me;       
& g* {2 G' @9 T# }" }$ y6 ]3 r    do
    {       
( t& w: R4 R8 G: c8 n        if(Index == NoHandles - 1)
         Index=0;
          else
3 A, a4 X% s- q    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
6 t8 ^+ q$ u4 D% d          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
7 P. m: L. v" x4 f3 h+ V$ T    }
    Index++;               
    } while(Result == Not_Found_Me);
       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
% q5 l0 m& V) i3 S  T' t2 e# D  U" @               
    return EFI_SUCCESS;
0 |4 s3 e2 u; c; M- X2 w}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习