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

kaurohy

/*
2 `, I0 X1 u# B1 O  \1 X * myfileio.c
: x1 d  D+ ?7 \5 r6 D6 A * Apps
; ~$ E4 h5 f6 ^& {* l, } */

1 r* R, w+ l  W#include "efi.h"
#include "efilib.h"

+ {$ p/ U7 S2 A/ I1 E#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
+ p0 n9 ?# K. L; y
+ {- J) o3 c( M2 C! [static EFI_STATUS WaitForKeyOrReset(VOID)
; w' ~( i* D% C{
2 ^% O* e/ y; N3 \. [    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
5 o- \! f; W! G' m    UINTN               index;
, f" @% |4 r5 ^0 a   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
# l* P" I" k$ f! K            break;
    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
* N& m# M: _8 `$ x: y  r    return Status;
}

( N1 n( F, ]- s7 AEFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
, i- L$ K3 s2 z- r; O{
    EFI_STATUS                Status;
" S7 W# O6 y; l6 Q    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
7 E0 `7 k, d' i8 |: E1 {: p( Z    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
* U, d6 Z$ m/ w, u0 W3 }    CHAR16                *FileName;
0 z1 k: ^, F; P2 c8 s4 b) f$ F( c! @0 f        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
* n& _0 F  R# {/ l    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
3 H0 \3 |/ ^; O' V7 x* J& e    int         Bit0, Bit1, Key_Space;       

% U0 C6 k% G$ E# z    FileName = L"NewFile.txt";
        Space_Key = L" ";
       
        BufferA = L"ABCD";
        BufferB = L"EFGH";
        BufferC = L"IJKL";
0 l7 S; S0 E, E) Y* c5 d7 }0 l0 W
        Bit0 = 0xff;
        Bit1 = 0xfe;
2 h" l3 r" A6 U       
- `) M. m9 O0 a: a: B5 b        Key_Space = 0x0020;
  c9 n0 ]) u$ Z8 E! N) d  n. ~       
    InitializeLib (ImageHandle, SystemTable);       

; G4 O" c7 ^0 ^9 O, A7 v1 J- ~) b; \    DestAddr = AllocatePool (BufferSize);       
" V+ \) @( g# R4 w
0 f1 A9 g" U& V6 t" a+ ~! l    Print (L"Value of Bit0 is %x\n", Bit0);
, u4 J/ e3 @; m( |5 B* Y# {    Print (L"Value of Bit1 is %x\n", Bit1);       

       
8 @& X* c! e1 r( s3 ?. S$ Q    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
: P; w& J$ I4 B! s' l    if (EFI_ERROR(Status)) {
* a' D9 E, o4 z4 r+ `; x2 ^! r        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
9 `2 l4 u; a" E- W" b) {        return EFI_LOAD_ERROR;
    }

! g$ z: ~/ T. v' n: |- ?6 c0 u    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
+ p( h8 ]: Q0 Z) \4 S+ Y9 B, k    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
1 G  X( `: ~3 A& C            return EFI_LOAD_ERROR;
* x+ b7 Q' p6 Y* U        }       
' r# ~4 d; c6 d/ E, K       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
+ Q& M" Y2 Z! H9 B; v6 L; \. h9 P6 r            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
. A9 ?4 o+ d7 H( |" x        }

' d6 U0 j5 |) M- k1 @7 @    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
: q4 D  Y3 u$ Z6 Q1 H        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
4 E( {; G# q! o+ C! @, A1 \- d    }               
  ]* `0 z9 V) q0 h) c' l               
    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
    if (EFI_ERROR(Status)) {
# L1 ]& w% B* E& Q& p! i        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
  U& W/ u! v. d% l$ j% [    }
       
3 {- O  V& T: l% t3 M  o    Status = FileHandle05->SetPosition(FileHandle05, 0);       
; h/ Z- `& j' Q    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
4 w, q5 V8 E- q8 ^& z    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
5 K# ^' h' r& g4 h        Print (L"File Buffersize is %x\n\n", BufferSize);
6 ?+ O$ S- m: D- a' n    if (EFI_ERROR(Status)) {
* Q/ d+ T) W+ X6 D        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
6 s# m) f! N3 I9 _+ h+ O    }       

$ M2 u3 M! F/ ~* x* F    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);
' i, w/ @0 K2 Y        return EFI_SUCCESS;
    }
       
+ J: }+ X" y/ V7 K& {    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
" b5 Y9 t" j* Z        Print (L"File Buffersize is %x\n\n", BufferSize);
8 K- g" w4 T! R6 I0 @    if (EFI_ERROR(Status)) {
" b& j* H. v2 f4 M  o        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
% \5 s$ n" G! N( [  E% ]        return EFI_SUCCESS;
/ |) k& N  w. F3 l" B6 E    }               
       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
; j. |6 J* i2 f# o        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);
        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
. E9 m1 r0 ?5 F7 [        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
, @) s- z) L) A; }/ F% ^    }       
+ Y" O  Z+ L$ Y3 F1 y2 \7 c8 ~
4 P( [1 A7 ~& D+ r& F1 Y- t//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
9 u; `1 g/ W8 u3 ?* T) @( I" m    if (EFI_ERROR(Status)) {
# X! R5 z# P8 f6 D( F        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }

    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
( Y. ^. C) o! x% K/ {$ N    if (EFI_ERROR(Status)) {
- B2 P* r- ]) F. F& A6 K/ a' P; ?        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
1 w- H/ P- s3 d        return EFI_SUCCESS;
    }       
* o) a) ^) r4 H/ u$ T* D6 j        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
7 V% X1 k; W6 p; J; t# q
    Status = FileHandle05->SetPosition(FileHandle05, 10);       
    if (EFI_ERROR(Status)) {
2 v7 o/ _# P: v+ B. i" t6 P        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
3 Y7 ~4 _' o* r& ]6 h- h3 x       
; a; k* ^0 T8 {! ~0 x: H    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
& I4 F  w6 y9 `* d+ H( L        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
2 {$ C9 Z' w5 \        return EFI_SUCCESS;
    }

    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;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
5 C2 |  a4 L) {7 K. R/ f    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
! Z$ w9 \# V, @1 x- v& L        return EFI_SUCCESS;
    }
5 N1 j( K% B  D" h% l
4 Z) ^: h! w/ {2 `    Status = FileHandle05->SetPosition(FileHandle05, 14);       
8 V* A: O7 E8 r$ ]# x* D9 C, f    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
0 e6 m! F( R, T* L. o  _* R4 G        return EFI_SUCCESS;
    }
       
1 |5 n4 `4 \" e6 N    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
! c2 j  m( z# Q! v) e' l. L        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }

    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;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
1 {" X) F! R, a% f% `) {        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
6 L7 Q4 B! S6 C. k4 b& `        return EFI_SUCCESS;
    }       
* i$ [6 {* K; z5 Z2 ]; _
//---------------------------------------------------------------------------------------------------------------------------------------------------------------

    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);
        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
0 ^5 x0 j0 Y/ O3 v    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
6 V( v# v3 w8 e, t4 S        return EFI_SUCCESS;
% d7 z7 Y- u! U% o% S; v  c    }       
       
! N9 j$ }0 q& ?& r. `6 Y6 O    FreePool(DestAddr);       
$ g; ]1 ]5 K" \$ c
    Status = FileHandle05->Close(FileHandle05);
5 H0 N/ m# G& Z7 r" I  `    if (EFI_ERROR(Status)) {
2 ?' V$ x1 Q1 u+ d( L8 I9 x3 M, b        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
7 ^2 s6 Q) s) A, V/ r) E        return EFI_SUCCESS;
    }
$ H1 o3 I" W( i4 I, E" Q       
    Print(L"File Name = %s has already been created.\n", FileName);
! Z3 k8 ^- \5 c2 T4 P  j, e* s
+ i- ~1 d. V) d! N% I% F2 f* D    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
, V! X* W, S, k3 w  ^) v    WaitForKeyOrReset();

3 e0 q9 \8 g" C0 ]0 z  g! p    return Status;
6 j, n# S; V1 X0 r1 h0 Q7 y}

chenhy

//**********************************************************************
//**********************************************************************
0 ?. a% `0 z( `, u7 K0 m//** Demo code for PCI configuration space I/O access program **
1 f9 J0 d: V6 E& |3 W* }//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
//**********************************************************************

#include "efi.h"
#include "efilib.h"

' ?5 P) ]# w6 {( k5 k#define        Not_Found_Me        0x00000000
; F9 N7 [+ b4 G; P5 U  h#define        Found_Me        0x00000001
$ x8 q% m8 F+ w
static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
5 [+ S2 l# R! f  j9 e0 X) q4 @0 {    EFI_INPUT_KEY       key;
    UINTN               index;
6 {4 K0 K4 @& ?' ~   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
8 T0 v8 _8 J2 @$ A+ u  N+ U        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
* @: ~  ~6 }5 S7 F        else
+ u3 Z0 z9 T2 v6 i- {/ _            break;
, Y! ]9 x* f2 H4 m* G8 F: A" ?    }
# ^  A! a$ L: z) _    if (!EFI_ERROR(Status)) {
7 D0 Q9 p- l+ {7 p1 X+ e- `, L        if (key.ScanCode == SCAN_ESC)
- y# m5 x9 w4 G% G2 M2 C2 A            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
$ g7 U% ]0 F4 d, }3 Q; x! X; T    }
5 j3 d8 m+ I. X+ b1 R* I1 [, E   
    return Status;
}

UINTN
AccessPciConfig (
/ S  e# E. l. [% ^    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
$ G+ ~, i0 S# ~1 i" c. ?    )
3 ~3 G' i# @* m" x/ a, a  I{
$ o/ u% B7 ]0 T& e' f+ {' U& C, U4 k        UINT8  i;
7 V  ^1 l# I. o% G2 J2 x        UINTN  PCI_ATTR[4];
$ V( V* I1 F& n- s3 F        UINT8  PCI_REG[4], pci_offset=0;
: x$ l7 v( ~( [' c5 X8 N7 G. P- ?
        //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)){
  R6 q. A3 p2 O! X. M          Print (L"Device found at BUS = %02x         DEV= %02x        FUN= %02x  \n",PCI_ATTR[1],PCI_ATTR[2],PCI_ATTR[3]);
. x# r1 [) u& U, V( z          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
          //print register value
          for (i = 0;i < 4;i++)
          {
        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
, c( J, L% b0 h- d* }1 j                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
0 t2 l- M& w0 ]: a          }
          Print(L"\n");          
% l$ O  X3 M* Q1 {- q) ^1 g6 ?          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
* O2 ^+ h9 c4 H+ n! @7 }) |          return Found_Me;
    }
  B* v3 R/ }. [2 k, ?: h+ Z
$ C5 q4 q' l" W  Q2 S3 k3 Z0 ]        return Not_Found_Me;
}

( W- H4 \( o! s# T: h. e2 o0 lEFI_STATUS
/ _- ~6 q+ t; f! |- wInitializePciApplication (
# s5 q+ v7 v8 d6 G: \    IN EFI_HANDLE           ImageHandle,
& ~% _3 i# n3 i* s/ E$ Y    IN EFI_SYSTEM_TABLE     *SystemTable
    )
. i& u, ]  A: Y9 ]! U1 A{
  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
1 m( K0 ?* {( d4 R4 `5 \6 e  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
3 _9 j1 R7 _$ g$ W  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
/ B9 d+ [# e7 `+ z$ [
  i6 U9 w) N- w" F( V    //
1 O4 R% [; o6 ^, K. j6 w    // Initialize the Library.
    //
    InitializeLib (ImageHandle, SystemTable);
$ y% o) B. X# A; p9 c( K2 r    //
  }3 y. M9 V( H, m; _2 H    // Find the PCI driver
    //
    Handle = NULL;
$ O$ H+ S8 o% `3 N) V6 y  Y: {. l    BufferSize = 1;
    do
    {
  G8 g; e( ^( Z! A- J      if (Handle)
6 D0 U' A! ^  {! ?2 P' x+ R      {
9 M  D7 ~& r0 E& T' [) L        FreePool (Handle);
/ U$ n' g8 g; U5 k6 ~; N5 Q      }
               
      Handle = AllocatePool (BufferSize);
  `1 e$ R" |* M  n6 @      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));
  I' x# y/ Q( D% e4 {; X
    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
. ^, ?  @6 y( G% p; u) j    Index = 0;
" x: ]+ ?. T4 {; D4 k9 w    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
3 h6 H% R# z; F. J' k3 L# e: q9 u    do
    {       
8 k" j  i$ S3 Z        if(Index == NoHandles - 1)
  |; |: Q6 Z0 I* }' H8 w! W1 s         Index=0;
          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
$ ?* m/ D& a: f        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
. K$ b; |6 t0 |! I    }
8 E9 j% \- T2 S    Index++;               
    } while(Result == Not_Found_Me);
       
. m0 [- C& K+ G7 D( `# l+ A  S        Print(L"Please hit any key to exit this image.");       
$ S/ c6 G7 ?2 S# J    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
0 X0 h2 y, h& o$ F    return EFI_SUCCESS;
# d9 N' a/ z2 l}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习