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

kaurohy

/*
* myfileio.c
* Apps
- ]. c$ u/ `6 x4 r */

#include "efi.h"
6 L( i( Y7 |+ R+ _0 _8 o: y0 j#include "efilib.h"
/ [8 ]  u/ H3 O, z
#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
: X( y! @0 I5 u1 h
static EFI_STATUS WaitForKeyOrReset(VOID)
$ Q4 x4 t% Q& L# b1 y{
: D2 m  f+ E8 e- A! [    EFI_STATUS          Status;
- Z; l5 [) {" @    EFI_INPUT_KEY       key;
0 Z& c: G; N# f! _' k4 ]* A# ^    UINTN               index;
   
) [8 k" ~. J. P! z  |    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
0 o% N+ V1 H0 s; p8 \9 i, f            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
& T% b1 @: K+ W; D6 r9 I+ S, c        else
            break;
    }
2 M+ r# ?2 F, n; }7 W( C    if (!EFI_ERROR(Status)) {
3 Q& @) o. }& V% w6 i1 ?        if (key.ScanCode == SCAN_ESC)
- {. Y* O+ E' z            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
+ T6 l% I1 Q6 S0 }9 y   
9 t# D, g7 S- B' W    return Status;
}
; I4 f( V6 A6 R
EFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
% `9 K" F1 H0 a" l3 L& |: z{
" N) D1 K5 d. c& W! a    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
( l4 Q/ H+ B( a: I" @    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
/ b8 ?. k* N' f* d8 A* X; m5 L    EFI_FILE_IO_INTERFACE        *FileHandle03;
  |: Z5 r. @9 l. d6 X    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
" V0 H' [' |) A. p3 t9 F# n- e8 J    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       
4 E* U8 ?+ x9 Q1 C; O' \7 b
    FileName = L"NewFile.txt";
        Space_Key = L" ";
       
        BufferA = L"ABCD";
        BufferB = L"EFGH";
        BufferC = L"IJKL";
+ u" `; C" V" N0 N, ^# d8 T, H% _
2 r0 o, c+ S* E8 j; w        Bit0 = 0xff;
        Bit1 = 0xfe;
       
% d# j! `6 P) u6 l4 Z7 d; g/ y        Key_Space = 0x0020;
$ U) l/ j$ a6 h$ S- K# r# `' }       
" Z* U9 }" w" ?! k  U8 ^6 \    InitializeLib (ImageHandle, SystemTable);       

    DestAddr = AllocatePool (BufferSize);       
: v. J$ d" D$ ]5 R+ g" e+ a
    Print (L"Value of Bit0 is %x\n", Bit0);
5 K" ]4 N1 A& \) `, M6 n    Print (L"Value of Bit1 is %x\n", Bit1);       
/ v3 `- n/ X7 p1 X* G! U$ y
2 z& ], J8 x: k- r  ?+ C/ A       
# `% J7 v1 U/ ^. `    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
3 i! [1 I& Z3 a; Y/ [    if (EFI_ERROR(Status)) {
4 p1 W8 I: ^+ P  s1 \        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
1 |9 J6 F- H% C8 V; G# x6 P! y        return EFI_LOAD_ERROR;
    }
- l' d* k  k6 B* B1 p, e
    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
; y: Z9 x& a$ e6 O$ @. V    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
        }       
1 Y& D9 n5 i+ H6 t8 d2 w& J* Z       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
5 T: I! z$ x6 E% e% v' Z5 S            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
' h5 D9 |8 R- W            return EFI_LOAD_ERROR;
        }
  W: Y4 V- ^. k) W' X
9 r$ J6 @) H5 P3 a6 a7 H  j    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
2 B) F; z3 J# @! o( Z    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
; O- D& s0 V! B+ J- t1 y0 [9 ]8 m        return EFI_LOAD_ERROR;
8 b, p, e0 R: P( B; `    }               
               
4 s; R9 k; b4 i- |/ w8 Q! d    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
    if (EFI_ERROR(Status)) {
: _, [+ z7 j# e; d8 O, d        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
& B6 V! u' u" r7 ~        return EFI_LOAD_ERROR;
" ?' F3 L& ?; w# |- y3 X1 M; w# c    }
' _7 c1 [; v) M! U( 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);
& E+ t3 E( u) g- j: j- ~; }        return EFI_SUCCESS;
+ ~/ j5 F- g) }    }
8 [. I( n$ q* U9 w' M! r8 u       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
. J% c! e0 V* K$ n' _9 g    if (EFI_ERROR(Status)) {
1 _+ h' D$ H2 f5 m2 w3 [7 Z        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
" \- t7 }) P" r/ I/ b$ C        return EFI_SUCCESS;
    }       

    Status = FileHandle05->SetPosition(FileHandle05, 1);       
* c  ~  ?+ @$ J6 p# S4 R& n    if (EFI_ERROR(Status)) {
5 i3 L4 ?1 z4 M: H        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);
) U& M( i* X# b# M        return EFI_SUCCESS;
$ C) Q0 u3 a5 C: P2 k    }               
       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
$ |2 ^! P/ r, }/ S    if (EFI_ERROR(Status)) {
1 K9 n* O) [; e# ?7 T& X- S/ [        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
7 [/ \# q+ b# V        return EFI_SUCCESS;
; h; `; F: Y8 x4 n    }
       
4 W1 _0 s3 T5 A" i% y    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
5 J$ ]% E  U9 @: e        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
8 a& n: e# c  a) k; W( k& J- f        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
' B% g+ h& B6 Z/ c# Q+ d    }       

//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
4 q+ _% w6 R, s0 G% t- u0 ^        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
! P) E# s9 \9 C) X
: J5 R+ R+ Z* w    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
  Z4 S! q, K: X    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
' g' _( z6 X1 [( _7 p, b9 V    }       
        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
6 P7 E: ?$ M! L. F, _
7 }8 W* q& h; P+ ~6 c, j    Status = FileHandle05->SetPosition(FileHandle05, 10);       
    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);
, s3 B1 }$ Y$ ?' I    if (EFI_ERROR(Status)) {
4 ~, q% A/ s  E& [9 J9 f5 g        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }

    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
* s' |  B5 C4 e+ V, x3 E1 g        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
9 s9 R& ^2 }, [4 e% f        return EFI_SUCCESS;
# Y$ ~$ D6 @3 O  x    }
0 q3 i( Y6 o) m6 p       
5 S) g) J6 }3 q. K4 p0 m    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
  \* B0 |6 p8 U$ }! P- I+ k    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
: u; w8 |# c! ^+ r/ K        return EFI_SUCCESS;
    }

8 |3 n9 V4 B6 h6 [& f    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;
7 G6 Z0 S" o! y" `6 }* i( r    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
$ [9 U0 g: P3 F2 y# L6 C7 V4 U    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
) i2 {" r/ z  F- x9 u5 B1 A    }

    Status = FileHandle05->SetPosition(FileHandle05, 16);       
    if (EFI_ERROR(Status)) {
- F8 s8 Z/ l  o/ P        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
, _$ E2 A9 v  h        return EFI_SUCCESS;
    }
) \, W: \0 J9 [- M4 e0 J       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
2 d+ P% j4 M) r+ A# s2 X5 g        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       

+ p0 d4 ?; Z' y! U' Z//---------------------------------------------------------------------------------------------------------------------------------------------------------------

- ^. g" j0 P" n5 T/ y: _    Status = FileHandle05->SetPosition(FileHandle05, 18);       
4 V5 A2 K5 `9 l9 A    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
0 V. F& k5 M5 m; @7 Z. t        return EFI_SUCCESS;
1 L: F$ j1 \7 P9 W9 n6 b* n    }
       
( u/ b" Y: Q$ K5 j    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
7 {6 B( Z. j2 n9 a        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
3 Y  J& Z! Y6 J4 j$ ~/ i        return EFI_SUCCESS;
0 O: G/ ^5 f9 I. [' x    }       
       
" y% r/ Z( S5 [    FreePool(DestAddr);       

    Status = FileHandle05->Close(FileHandle05);
/ w5 {/ X7 p4 F4 u: ?5 ?3 `+ J    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
/ z  {. ?- @; C( w" N. P        return EFI_SUCCESS;
    }
       
" `: E0 b6 d9 V" M, R& h    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");
& ^4 G. k, |9 O    WaitForKeyOrReset();

    return Status;
}

chenhy

//**********************************************************************
//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
+ P/ h9 [) z  U//**********************************************************************
//**********************************************************************
8 N8 L3 m4 s' C0 m; \4 L4 }* F2 X
4 [1 x: T  Q4 I# J4 K" y#include "efi.h"
#include "efilib.h"

#define        Not_Found_Me        0x00000000
' _8 D( I( ?+ U* l) H2 S6 y! B#define        Found_Me        0x00000001
! R5 Z9 h( X) D; z% k! X3 _  _
static EFI_STATUS WaitForKeyOrReset(VOID)
{
/ i  |' [! N! N& q* x; k    EFI_STATUS          Status;
- U' C7 v  j4 I    EFI_INPUT_KEY       key;
- [& L: K, E1 z1 t# p/ |    UINTN               index;
   
    for(;;) {
2 p0 L, B% v3 w: w* y$ Y! c- m: Y        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
3 ]4 H$ T' T8 V4 a, _            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
& [3 Q8 X1 b5 \. X/ p$ y        else
            break;
    }
8 {4 Z& V( ]+ H( o: x* g0 P( O0 I( j    if (!EFI_ERROR(Status)) {
# Z7 S1 I5 T3 E% _0 b5 c        if (key.ScanCode == SCAN_ESC)
: C+ p; _. @% Y) A( L2 n4 ~            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
6 Q6 p7 E5 T$ q+ @    }
   
4 |% `0 X! m5 x0 Q" U5 p    return Status;
8 @, m' `( f0 U6 p% E9 W/ C, _}

4 {% p( _8 |5 J& ]5 bUINTN
AccessPciConfig (
) V* E3 s2 \$ S# m8 E% x) V    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
, K8 N6 u# `0 o7 R! S2 S$ D{
        UINT8  i;
        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;
0 F7 o' z( n; E6 o9 k& u
        //get device attr
6 g2 ]9 d! u$ i; Y/ I        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++)
          {
# S3 P& u, D( M' y' g        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
0 f7 X5 L$ c/ \" U                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));                  
8 ]1 o# T) E1 O% {. e          return Found_Me;
/ R" l) W/ V2 L" r. B    }
+ v! G' t8 h( b4 w5 E
- r5 }; Y$ i. f6 N3 }/ e, [$ z8 _        return Not_Found_Me;
}

/ p3 e5 s+ J9 _& o7 {0 ~EFI_STATUS
  R4 \5 L$ ^3 s9 Y% G  OInitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
7 K6 M, b6 E% o/ ?! x1 s    IN EFI_SYSTEM_TABLE     *SystemTable
    )
{
. Z0 p6 P& {5 }" {8 Z- ]5 N! {( Q) k, I  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
& X, m/ _( s: Q! j$ {  UINTN                   BufferSize, Result;
# L5 E4 ?* s" G0 z  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
  C. e/ J  F% r- G- E
    //
    // Initialize the Library.
    //
    InitializeLib (ImageHandle, SystemTable);
' @% s2 }6 q/ t9 x# }  |! C    //
9 F9 B+ D9 |, H, I6 W$ [3 A    // Find the PCI driver
    //
! j3 F- U4 ]1 n; Z    Handle = NULL;
    BufferSize = 1;
6 @1 X- u4 \; H' F9 b) K    do
    {
      if (Handle)
      {
' N. X. T2 W: B4 |! n  k! [3 h        FreePool (Handle);
      }
               
. o* u* v0 p, d" g7 s      Handle = AllocatePool (BufferSize);
2 \: c: G& z5 g4 d/ A2 O; {      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
+ }7 z1 c3 [$ V% G3 Q- v  }. D1 H
1 f$ T% v. A7 U4 H- t* b    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));
; |3 q; Z0 L- r1 P- I0 Z
    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
5 ~7 m' r/ C8 ]) a9 ?    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
) M( }" i, c6 \' Q    Result = Not_Found_Me;       
1 H! N$ q& j. d( L    do
    {       
9 S# B& Q4 H: C- t0 k" M# g        if(Index == NoHandles - 1)
         Index=0;
$ s- a. x. M! X+ H          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
9 B' Z5 f* Q. ?) G. \- o    if (!EFI_ERROR(Status))
$ M7 {+ Z/ J  y$ T    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
    }
    Index++;               
    } while(Result == Not_Found_Me);
. |7 q1 `+ v  T$ `8 D8 [       
        Print(L"Please hit any key to exit this image.");       
3 a5 D/ ~8 s7 p- W+ y) N    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
# p! l8 a! ^* A/ O               
    return EFI_SUCCESS;
4 f; r2 u) F% H! `7 N' {7 _& F}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习