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

kaurohy

/*
8 h: f1 j" j! S* d  s- d6 J0 s * myfileio.c
* Apps
*/
4 h5 L- V; m7 q; P& v' L
#include "efi.h"
2 N& F7 a) E. H# [+ @& c3 o#include "efilib.h"
2 `1 }/ d$ R8 A
; D, ?/ @- _+ Q( G% m#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

static EFI_STATUS WaitForKeyOrReset(VOID)
{
3 k" J0 n' D* Z3 X! R7 @' c4 _    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
   
    for(;;) {
( d1 Y; z( X3 I% g4 x/ s        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
, E$ E& q, G; {        if (Status == EFI_NOT_READY)
5 p1 f! ], }! `# m$ Y            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
    }
0 ^& \) \8 N3 o$ F" L: X! w6 ?    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
+ z" _* R+ U8 N! Q5 Y6 G5 z/ {" _   
    return Status;
}

1 O2 N6 o( [$ j! J% l; BEFI_STATUS
0 d6 V5 q* r; ^) G# q+ d" m2 W+ EEFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
{
2 e, [: _. j* }! f: o* D    EFI_STATUS                Status;
: R( r9 y; x$ v2 u) P* b    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
& A0 I: c, c. \% s5 n    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
0 }1 n7 `) s( D9 ?& `7 L5 B    EFI_FILE                *FileHandle04;
+ z4 `2 y# L3 P3 x# y4 b2 s# _    EFI_FILE                *FileHandle05;
+ x1 e2 I! ?3 K    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
+ H1 H7 l5 A* y0 g0 ^- p8 q    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       
& v8 M+ P& b% q: t, S$ K: i
    FileName = L"NewFile.txt";
        Space_Key = L" ";
       
        BufferA = L"ABCD";
4 W2 a0 v) @& i+ m* i7 s        BufferB = L"EFGH";
        BufferC = L"IJKL";

        Bit0 = 0xff;
        Bit1 = 0xfe;
0 `  l, Y2 g9 I$ g! _5 \" G       
        Key_Space = 0x0020;
6 T$ y# i3 c' K/ d& A       
    InitializeLib (ImageHandle, SystemTable);       

4 J3 W; z1 L) S/ t( t) s    DestAddr = AllocatePool (BufferSize);       

    Print (L"Value of Bit0 is %x\n", Bit0);
9 u9 e4 N0 y8 ?    Print (L"Value of Bit1 is %x\n", Bit1);       

       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
9 h2 P: V1 `% Z' ^- _5 L% _( Y+ t    if (EFI_ERROR(Status)) {
% M# A. C( s8 v( U        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }
: u8 O% P  g7 Q2 A9 ]9 g0 l7 @& h
0 k4 O1 E, x/ v. F5 h% `5 J, f; H0 D* J    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
, f+ Q$ [4 d8 A+ |            return EFI_LOAD_ERROR;
        }       
       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
% S( F0 [  Q6 d: N* p    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
  H  U! [$ m( P$ n3 t, ^% d% j            return EFI_LOAD_ERROR;
4 \% S1 |; [5 ~* U4 C' d0 D9 n        }

    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
8 q! G0 k. i- c  }$ V( q    if (EFI_ERROR(Status)) {
# e7 w, R# z2 N5 k8 `4 d        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }               
               
8 a3 C" l6 `6 F) Z, X+ g    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);
& h, L; F8 x/ B6 J9 h: E        return EFI_LOAD_ERROR;
    }
       
9 F. a6 G8 A6 H+ L" n$ Z    Status = FileHandle05->SetPosition(FileHandle05, 0);       
" l- N/ E5 f1 n/ I/ [  y1 k, l    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
/ T. f/ `: v9 ~( u        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
- x" j& X+ ^. V9 Q* \7 @2 F( R    if (EFI_ERROR(Status)) {
, f0 i  k! q# h0 A$ d! N        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
  g1 N' z& E3 D; m        return EFI_SUCCESS;
    }       
( c% [' p$ O! V
    Status = FileHandle05->SetPosition(FileHandle05, 1);       
0 ?6 G. e5 Y& ~+ Z3 W9 a" J    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
" n( A7 E9 F( [! a    }
  f, c; @  R, f( Q% Y       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
( P5 O# H1 }; g6 ~    if (EFI_ERROR(Status)) {
! h5 n: J% N& K  T: x3 Q  K        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
# {! B- s# M% v( @) G' |        return EFI_SUCCESS;
    }               
       
9 {0 t$ T' A: D8 _2 }1 B5 i( ^    Status = FileHandle05->SetPosition(FileHandle05, 2);       
% r4 W% i' B9 o* s% X) {    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
6 \$ _9 M- A+ g/ l        return EFI_SUCCESS;
    }
7 D2 p/ E2 q5 H5 A( H( u1 m; m. J       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
        Print (L"File Buffersize is %x\n\n", BufferSize);
6 R1 g  a1 d/ H4 Y6 r    if (EFI_ERROR(Status)) {
- V8 L! G- p! j% M        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
- `& e6 w/ J, ~1 m6 G; }/ }. U0 A        return EFI_SUCCESS;
    }       
6 R2 q$ ^. W5 e7 [! p+ ^
1 C' o$ W: I. r//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
  g( f  H0 @3 ?. e. Y4 t+ W2 h7 c    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
' |( `2 B+ g, B6 {6 i* H) g    if (EFI_ERROR(Status)) {
5 x) I1 r$ \. v" M, ?! c0 z        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
! b$ ~6 [9 P* G6 `) E  O, N    }
7 ^5 a+ w" ^2 I$ `( O
    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
2 q# [! }- ]# h5 V3 f  I, i    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
1 p* `5 E3 g% g0 ~# D        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);       
    if (EFI_ERROR(Status)) {
5 v1 p! W% Q4 D9 a& {        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
" ]. G0 f4 e% D$ M0 X- x; R- [/ A9 U        return EFI_SUCCESS;
7 A! T" V) [+ Z8 Z; b/ K2 Q. M    }
       
# X+ r8 s  ]) j& V( E' h( S; `: K5 U    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
  ]; L1 ]5 J1 p( W0 D) E; l* J        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
6 A& {6 G9 p! }2 ^% G5 N  D
    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;
    }
3 X. h! f7 _2 I6 N; A$ u       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
! r% i% v  }' |7 x+ F        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
1 I" m" N' G" s/ m5 d        return EFI_SUCCESS;
# y) U8 P" C) Y) V5 N4 y; b    }
5 g4 U* P: G0 ^
    Status = FileHandle05->SetPosition(FileHandle05, 14);       
    if (EFI_ERROR(Status)) {
( I0 m% Y4 X2 V, K/ T" s. T/ h% l        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
. Q; ]* K$ y3 P/ _! U* {    }
1 y7 E' E+ Q# J       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
  D7 r8 h/ k/ r  J+ ~4 S    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
' D0 L& V% Q2 R. g1 Z1 C6 y8 g3 H- Z        return EFI_SUCCESS;
    }
1 G# U5 X1 Y% U7 @8 r2 v
) H2 [  A( K% d% p7 w( a    Status = FileHandle05->SetPosition(FileHandle05, 16);       
    if (EFI_ERROR(Status)) {
" b+ U* q/ l% z  g4 J' K: h        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
0 z1 ~  p6 B" p" M# k8 Q        return EFI_SUCCESS;
    }
# v& a" Y# @3 F  b' W; W, \) _       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
" ~5 t5 y, F# H8 y! `: m    if (EFI_ERROR(Status)) {
  [$ _0 k+ W0 f  I% Q" b& N: a        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
% e0 Y( T( d- O0 g% ~! y        return EFI_SUCCESS;
$ }4 @0 Q) a6 D6 h) {1 r    }       
+ t/ {  O( h% S8 [) ^
//---------------------------------------------------------------------------------------------------------------------------------------------------------------

    Status = FileHandle05->SetPosition(FileHandle05, 18);       
0 }- P, N# {, E2 J    if (EFI_ERROR(Status)) {
) [- C3 R( {0 T+ Y+ w7 i4 N0 z7 ]        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
4 A3 U) x; `% n( l9 I  R; u        return EFI_SUCCESS;
  D& c  ~5 g6 z" m- e6 B    }
       
. \1 u: l' Z6 j1 p3 Z- J5 F    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
1 Q  O1 P. D- y9 X3 Z    if (EFI_ERROR(Status)) {
* `1 @; \+ a( R$ A5 r        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
- x" R# \6 I' c$ a0 b        return EFI_SUCCESS;
5 j( q7 A- j% D; \3 C# D( r    }       
       
/ M6 J* d5 y2 T9 b6 {    FreePool(DestAddr);       
* z! q8 T( d& Y' D8 t$ y0 i3 P
    Status = FileHandle05->Close(FileHandle05);
: ^5 ]: o8 m; Y/ ~+ `    if (EFI_ERROR(Status)) {
% R0 p5 k. c4 b        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
+ Z' u1 I! h. C% k1 r        return EFI_SUCCESS;
/ j5 ]+ F5 M5 |& r! c    }
' Y8 W+ \' B# Y       
. i7 Q* j# D* Q( `) B1 C    Print(L"File Name = %s has already been created.\n", FileName);
9 G3 [* h/ t- U
    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();
- V* B( v% T, `# g
+ f) z7 L& A9 U, j& B$ V0 C    return Status;
}

chenhy

//**********************************************************************
! E% s) a* C& C4 H//**********************************************************************
% V+ N) p0 q. D7 T$ F. ~) o//** Demo code for PCI configuration space I/O access program **
/ R! W% S2 m. J//** By Lawrence 2009/11/20 @Taiwan**                          
7 _6 g0 l+ `) Z; d//**********************************************************************
' ~, C, u& T+ x/ j; N0 @2 a//**********************************************************************

, V/ p. p5 h! t$ Y6 B#include "efi.h"
#include "efilib.h"
! i# U7 K1 ]/ ?7 S# c: R
#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001
$ {" S3 z& H2 T6 q5 w
static EFI_STATUS WaitForKeyOrReset(VOID)
' w& T) P/ K% O) R{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
  G( X! @  Q: j0 }6 [   
, O  x" `$ x( m7 G1 v/ w    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
- A. M2 b( w& ^        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
  V& K* M+ o  p5 k! P3 G2 s        else
            break;
    }
! B# f# o: a6 L5 q  b    if (!EFI_ERROR(Status)) {
# C0 k4 T$ c7 S2 q. X  _- x7 C        if (key.ScanCode == SCAN_ESC)
4 e0 g: J( a0 R9 M            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
! E% i' r& H' I   
    return Status;
}

UINTN
! [! j. U4 f; U. w$ |1 ZAccessPciConfig (
6 M3 y  L* b! |7 {9 J0 I: w    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
' G+ ]: d( O! U" n; H4 G& \    )
9 L8 e$ v7 J" A{
- D8 ^+ f5 {0 J! w( A7 X        UINT8  i;
5 m5 u% Z3 `: r0 B* v        UINTN  PCI_ATTR[4];
# P4 o6 Z! B5 T! \# i: c/ {- T        UINT8  PCI_REG[4], pci_offset=0;
2 O& k7 ^* c" Y; E( L5 Q
3 L0 Q3 ]" H7 T. N- C; e+ }0 j- [        //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)){
" O7 n0 B3 z6 C2 k# S0 a          Print (L"Device found at BUS = %02x         DEV= %02x        FUN= %02x  \n",PCI_ATTR[1],PCI_ATTR[2],PCI_ATTR[3]);
# ]% J- p& O1 x+ O, m9 w          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
          //print register value
  g9 e5 Q* S" ~* [3 k  P$ a% ^          for (i = 0;i < 4;i++)
          {
* v; `5 g6 S. f. B' o8 u5 e) n        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
: Q4 T' N1 f+ \$ y4 [                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
          }
0 S  O! X) F' }, ?. U          Print(L"\n");          
( _: ?8 P1 ]# E6 _, j/ Z) v          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
7 N+ D' Z- K% ^5 s3 |          return Found_Me;
0 [+ @+ R; z# J: f3 D. ~$ n7 }: X. }    }
/ `! }; f$ t! C$ B. W
# n2 c8 d* y$ h8 ?- Z1 R; Y        return Not_Found_Me;
+ c/ M" {3 n8 a6 y' `( E}
5 D( V$ O3 Y3 P) ~* Y
" c, k0 D9 I. U( X4 I2 ^3 cEFI_STATUS
* }& t+ T1 y. W5 n* ]3 [InitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
8 m3 k( K' J* r    )
{
  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
9 A+ X: Q* ^- h4 q  UINTN                   BufferSize, Result;
$ m% _" M: e/ S  UINTN                   NoHandles, Index;
9 ]2 v! w2 C8 U9 p3 K4 |) `  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
7 g( N% b* g% n8 Z. p
4 g9 x2 n: W. V) @    //
7 z& A0 m0 H: Y+ X    // Initialize the Library.
* j2 `; c+ `$ F" \' z3 e$ u    //
    InitializeLib (ImageHandle, SystemTable);
/ L  J% Q  a8 O    //
' q4 i2 m% k4 X) @1 z! J    // Find the PCI driver
' K$ ^3 O+ K: i* C; r" L8 y0 C    //
+ o' I5 T! a, q) ^7 h    Handle = NULL;
, p# p  L6 G' z, @6 H% H$ \    BufferSize = 1;
    do
7 g9 C+ Y3 \0 o) B    {
4 W3 Y4 [/ \$ `% o+ k* s$ v+ F      if (Handle)
      {
        FreePool (Handle);
: W6 ~$ w( c& ~* m' w& W/ t      }
               
- m! V0 @9 Z  Z3 k/ J( Y      Handle = AllocatePool (BufferSize);
9 r, a# G4 h5 G' L      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
- d5 K: U" D% i. _
    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));

* A- E6 ?' b' u0 Q; M    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
, T; h0 k: R  w0 d0 T' M$ F    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
; p* P2 O, P/ J4 t. [" C/ M* \    do
    {       
        if(Index == NoHandles - 1)
6 V6 B3 n' D* {$ D+ w! x         Index=0;
& }% n# K* |5 E3 B* j          else
, p3 ?( D1 m  i9 K* d9 i) B    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        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);          
/ b3 J2 n5 C/ v; ]5 r5 C( Q    }
7 g+ R/ L- s7 t8 D9 I/ @: F    Index++;               
    } while(Result == Not_Found_Me);
       
        Print(L"Please hit any key to exit this image.");       
4 j' G0 z  N% N) Y; |0 [' Z3 B    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
    return EFI_SUCCESS;
- P6 @& u1 ]1 C}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习