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

kaurohy

/*
* myfileio.c
* Apps
*/
5 R" ~! m* X5 z$ I' o# @7 A
. j9 f! o, }1 o: p# W# L#include "efi.h"
0 Q9 C' G9 v: w+ ]9 O#include "efilib.h"

8 e8 R5 v% P$ ]9 p" K0 i#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

8 @, i& _' b: @8 p6 d$ P% \+ Wstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
   
; n# Y) W4 F' y0 I    for(;;) {
8 F. t: Z* c2 Y! @        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
5 q1 [+ H' j0 m+ a$ Q# E        if (Status == EFI_NOT_READY)
2 b2 v  R; x% q% c, ~            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
: e% C% u, [: [. `% r' F        else
6 ^& \' E# X5 o; y" L. r3 j3 ?% p            break;
    }
    if (!EFI_ERROR(Status)) {
: s: a! X7 T6 l1 I$ L0 ^2 S2 [# l        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
9 r. h6 ~# E9 P0 j# V: \& o   
- p! K: G4 d& q: I3 v' b    return Status;
* I2 X$ j: B1 d) n5 i# [- x}
$ N( T/ A: G" L5 \7 Q
EFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
' N( i! Q5 o8 j* M; c+ J& z0 Z             IN EFI_SYSTEM_TABLE     *SystemTable)
: O* R$ L) ?  x5 I{
( K( \7 l: I/ @" j    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
" _# g" ~$ K+ g9 O9 X# o7 B2 u    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
9 S% [: ?" L4 q4 \5 b        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       

    FileName = L"NewFile.txt";
        Space_Key = L" ";
       
9 [( p) \8 a; \. }0 o. a) T( G        BufferA = L"ABCD";
/ X! ?# K- g# @& ]( z- ]1 i        BufferB = L"EFGH";
8 J& x/ |# L- h% J' C7 h        BufferC = L"IJKL";

        Bit0 = 0xff;
2 J* u5 c& Y# |  y8 J$ N        Bit1 = 0xfe;
& B& z/ A/ k4 w7 L" _       
        Key_Space = 0x0020;
       
8 A; N: d: G  t# l    InitializeLib (ImageHandle, SystemTable);       

' W! G% G# C9 I. J! D! H. `' L7 M    DestAddr = AllocatePool (BufferSize);       
: C* T: B  I5 p3 j, o3 k; K6 n
3 S% A5 T; b7 f; _: h    Print (L"Value of Bit0 is %x\n", Bit0);
    Print (L"Value of Bit1 is %x\n", Bit1);       

2 V, i$ a* O- C: c7 Y       
( l( q" `8 R; @/ f( P    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
- N% \! D2 t, B    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }

4 m2 U; j- b  Q; {/ z. j# j6 ]& |    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
' P# L$ O2 Z  h) |- O$ q  e: u    if (EFI_ERROR(Status)) {
- m: g# K. F" V. T$ V7 b            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
! [0 K& _2 y4 A! X$ \+ G) V% m/ f            return EFI_LOAD_ERROR;
$ K3 V5 f! d% I, E, G        }       
       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
2 i. C# f. F- c: P9 A. a            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
* C: w% b& Y! V. H3 g7 S            return EFI_LOAD_ERROR;
& V7 N0 _& t; T, {        }

    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
) F$ n/ x$ m1 e+ R        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }               
               
% T  u) W/ o$ D0 q) U9 T, U    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);
0 E! ?; V5 A' [. p5 g1 T/ C        return EFI_LOAD_ERROR;
    }
       
    Status = FileHandle05->SetPosition(FileHandle05, 0);       
    if (EFI_ERROR(Status)) {
. u7 h! w4 ?" c: e        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
, N! Q5 _- Z% b        return EFI_SUCCESS;
    }
5 t( R$ R4 r6 R       
9 e4 f. o3 s! O0 {' I3 S* l6 j0 [    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
" f" {* i1 F6 X0 y    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
/ u4 E2 A* d+ \/ d. E$ E% T, S        return EFI_SUCCESS;
8 B# L" F& d% @6 N    }       

    Status = FileHandle05->SetPosition(FileHandle05, 1);       
' d0 Y. ]4 m) b$ |" e( [6 b    if (EFI_ERROR(Status)) {
4 m: q$ w/ c7 |! k( D. ~5 m6 b# h        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
7 E- W. @) z. S5 U- u' p, H0 b    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);
( Z3 m; w4 }9 b, U7 [* F        return EFI_SUCCESS;
6 X, n  a: ~% [  G1 z- t) A' E  I    }               
7 o" B* W  }4 h       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
0 p: L7 r3 s/ Q% m9 e    if (EFI_ERROR(Status)) {
3 _7 N9 I. Y$ G6 ]1 r        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
4 o& Z; C3 f1 s6 }        return EFI_SUCCESS;
    }
8 c% m7 N) ]& _0 `       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
4 e# h; Y: A7 H8 y        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
' K7 D& z% e" u4 G1 F& o        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       

# J" s2 f/ _8 n) b7 b( |8 O//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
: s, b- T8 e* S0 \    Status = FileHandle05->SetPosition(FileHandle05, 2);       
* H/ c5 I7 L; N( t% t    if (EFI_ERROR(Status)) {
/ L5 C1 }' l$ S5 O# w# N9 n        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
5 O* d4 I+ y# i/ r; b5 |: N. y        return EFI_SUCCESS;
    }
! T$ a4 E4 N) @: f0 Z/ P" x1 K
8 Y1 R) B, t* u1 u$ Y* E    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
' T# H8 Z; e0 H- Y- j! f, w    if (EFI_ERROR(Status)) {
3 d0 i. I9 e. }$ @; b1 O- f- x# S8 N( i        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
  y" X" ]6 Y' X& G0 X4 L& Z    }       
1 V0 c6 H* r2 G3 I; s        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
0 b, }* Y  Q) i# ^' o  ]2 o//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

' g: Z9 H5 ?  q  N0 [0 B4 W. L/ b    Status = FileHandle05->SetPosition(FileHandle05, 10);       
4 j' L$ O. @$ B9 z/ k, Q    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
6 {% m' Y8 n1 f" M* G- A! G       
# W! C* b" W  V' Z( G    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
" f8 h, j+ X% i5 f3 O9 d: P        return EFI_SUCCESS;
    }

    Status = FileHandle05->SetPosition(FileHandle05, 12);       
4 a: y0 O  ~: Q0 k    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
+ G# [1 {( t* F0 u        return EFI_SUCCESS;
5 B6 P5 H" f( e! s9 F' F9 K    }
       
& L/ E  ^" P8 C! j  o    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
5 i- Y7 B0 `( N6 Y    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
. s. s2 A3 w) i' F$ {2 P    }
4 }2 o0 A7 R, f* P5 [1 Z
    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;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
7 s0 N8 D& B  W4 c2 j/ F. J& Q4 P+ \    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
& Y5 |3 ~: L8 J: F    }
3 K" u$ W: T9 I2 f6 ~+ q) z/ Z
    Status = FileHandle05->SetPosition(FileHandle05, 16);       
    if (EFI_ERROR(Status)) {
7 b% f9 k, |. m& l0 I        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
$ o  X1 i4 s6 U* q) D        return EFI_SUCCESS;
0 @7 L2 u  C. d# u3 {    }
' R2 [4 I0 @# o5 Q. J       
- z; @( [( J( G. {) W7 z# S1 C    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
6 v3 {- ?0 l* C& m    if (EFI_ERROR(Status)) {
+ V" d" a, B4 B- |) J        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
1 Q7 e3 N) m  |5 G2 w# G1 j$ W    }       

: }/ X  Z) T  r$ W//---------------------------------------------------------------------------------------------------------------------------------------------------------------
. }8 j& J. `5 _2 _' t5 H2 T
    Status = FileHandle05->SetPosition(FileHandle05, 18);       
' z  C, a1 q$ G! `2 X8 l    if (EFI_ERROR(Status)) {
0 N) L6 Y7 e% \  E        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
/ x! Y6 b1 @6 g  z    }
       
0 f! I6 k* t" O    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
       
& r- x8 y; }" Q! q# w6 N* p    FreePool(DestAddr);       
" x6 P- k. N( b
    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
) `' b/ u0 k) u/ |        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
: h% N) ]& n% k& N    Print(L"File Name = %s has already been created.\n", FileName);

: g- t& m$ p& }: O1 ~    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();

    return Status;
& w1 m. K5 H( v1 f2 ^) U% e}

chenhy

//**********************************************************************
//**********************************************************************
( W5 \+ L% t8 F* v$ }3 T5 ]) i! P//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
//**********************************************************************
7 {  Y- w8 \; X. q# Y# m
#include "efi.h"
& \) I1 S! {' S#include "efilib.h"

" E2 A/ X$ p' Z- s( n#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001

static EFI_STATUS WaitForKeyOrReset(VOID)
; L8 i6 Q4 q* S' n+ Q{
1 J$ x& x, B6 V8 M# f9 o0 o    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
* ?7 m( `( P8 q, D7 M        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
- ?- o2 N# @8 \8 R4 W# i  w1 j            break;
' h# \$ `" b# v    }
9 w- b. Z' o7 N1 W) T1 s4 Y    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
  u7 u- ?. J& r' [            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
4 g/ R! [/ b* D: p   
    return Status;
}

! t  `) @: r. E1 h0 yUINTN
AccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
3 i% o7 O. M1 k* G) P7 J( E! h. R    )
{
        UINT8  i;
        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;
/ x- C. L( w% M5 F( j4 K, B
% I) |: [" b5 G2 M7 A' 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)){
          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));       
( x5 [  I. k7 e' \' T) `" p& B- j# r          //print register value
) B/ `0 i$ o- N; M# |! v5 ?! O          for (i = 0;i < 4;i++)
          {
        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
' i! R/ l% g8 f! v, d, k                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
  u3 j  W* Y, j          }
          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
    }

$ K( {8 Y3 Z2 y. j        return Not_Found_Me;
/ ?4 [. w  Q+ [  h# u}

EFI_STATUS
InitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
    )
) |$ y  H7 P- G/ k{
+ [( e* @( j3 q0 T  EFI_STATUS              Status;
0 }! E2 M- W& _: P- Z' s  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
# e4 w/ X) w$ G, E3 |  UINTN                   NoHandles, Index;
: ?% x# z% Y9 T6 l# D4 E  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
* @- b+ m% L; O0 _# V
    //
    // Initialize the Library.
* e" M+ F- v2 V/ {9 _# J    //
: l0 Q6 B. F  ?    InitializeLib (ImageHandle, SystemTable);
    //
    // Find the PCI driver
    //
  B6 H6 s9 ~5 h& b0 l    Handle = NULL;
( A- b5 q3 `# B    BufferSize = 1;
- J% B2 P  T1 @4 c$ J$ ]3 k    do
, |: v& r7 O) W9 c5 u    {
: A0 L) P9 O8 W2 x0 {' d      if (Handle)
      {
        FreePool (Handle);
6 q4 P% f3 E( d  y& M      }
+ y  u( ^# N7 s) {/ w8 q               
$ t: ~+ T# Q; p: w$ T* K      Handle = AllocatePool (BufferSize);
) F; q3 u. Z2 f% i6 \, }7 T7 i0 W7 ]      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
8 p2 ~/ d4 y  c+ h, u6 h4 R
# P+ \1 S5 M& a. m0 c, b' O    } while (Status == EFI_BUFFER_TOO_SMALL);
% r, k- _) T' j+ C) ^/ l    ASSERT (!EFI_ERROR(Status));

: c2 s# n$ e, O# l8 M( z    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
" ~0 ~7 s" t* @8 N$ |    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
" n$ H, T2 n0 Z! A/ V. j* T( s    do
4 {8 i) H$ r# N- r% @    {       
        if(Index == NoHandles - 1)
$ \* l$ l' x7 E; Q. @" R' L         Index=0;
9 }  _2 P0 ]5 X7 w( X. _' e          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
5 d# l: z6 S5 R/ W$ N/ Z( B% K7 ~    if (!EFI_ERROR(Status))
) k. `/ k( V, ^  @0 ?    {
7 f- d/ R3 M, X( n% B5 U& a          Result = AccessPciConfig(PCI_DEVICE_TARGET);
' U0 \0 n4 p9 S, I: \1 h$ N) ]//-          Print(L"Again now Result is %x \n",Result);          
( [; M0 e1 D6 Q& d: s    }
; j9 j3 D: K7 e  r7 q    Index++;               
" h, A) @, t0 h, i3 l) F    } while(Result == Not_Found_Me);
       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
& ^* Q0 x5 E6 c% U& Z    return EFI_SUCCESS;
6 Q# Q( a0 F/ j& W; X) Y- `}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习