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

kaurohy

/*
7 C' C' W6 \  V/ Q4 x * myfileio.c
$ F# \2 V: @) A$ C9 e  o3 G$ R * Apps
3 p8 g& H. O' ~0 j- Q+ \% w */
  Z: @( `. \! y1 F8 P- ]
9 ~4 `0 ~: l  q  H# Y#include "efi.h"
#include "efilib.h"
& z( c* _7 K2 F! Z
#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

0 |6 c% U4 B- v6 G5 E5 Dstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
( e( t! L+ s4 D& ]    EFI_STATUS          Status;
/ A1 h( b& a1 Y    EFI_INPUT_KEY       key;
    UINTN               index;
   
    for(;;) {
3 y' b/ m# d; i5 ~        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
) G: B, `' ]0 s% U) _        else
            break;
    }
    if (!EFI_ERROR(Status)) {
9 F9 c7 n# I, z$ A/ `( A        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
# q. \8 w2 S4 ?* s9 ?% d7 t6 g    }
* x5 P. M% f% F, A3 J- a   
3 s* x" V4 S0 P$ b0 x4 u    return Status;
5 G9 s- @) ]3 X  Q}
- _6 B* x9 M" o
) i; ?/ O! [$ V# W0 v' @! mEFI_STATUS
* ?7 w; Q: @% c4 U+ B: @# }EFIAPI
0 l# i; H4 c: p$ V1 Q2 rMyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
, R# F6 @7 A: d: j4 u) F, Z6 @{
    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
0 G6 u4 ^' v" h# S& A    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
. j0 q3 D7 }, N% l- X" _5 Y    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
) q7 i4 l9 Z: s6 v4 A6 L    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
% ~5 L" l. d  u7 v/ _    int         Bit0, Bit1, Key_Space;       

% [. M( M. l4 ^) x" T: D    FileName = L"NewFile.txt";
        Space_Key = L" ";
% r0 j( O: V2 Z9 `) @3 F       
, d5 U! E% {1 W" U$ I        BufferA = L"ABCD";
" T6 X% w& Q2 z6 |9 F        BufferB = L"EFGH";
' E1 M" c1 W$ W# p* H2 e! @4 W        BufferC = L"IJKL";
7 T* F* O; k% J) Y- P+ I
        Bit0 = 0xff;
        Bit1 = 0xfe;
9 l& A' ~  W) n4 U0 v0 r4 }       
        Key_Space = 0x0020;
       
    InitializeLib (ImageHandle, SystemTable);       

    DestAddr = AllocatePool (BufferSize);       

* `' n' A( Q5 C; f. S; K% |    Print (L"Value of Bit0 is %x\n", Bit0);
: ~( ~( N2 h0 e  P& {, {3 M  v    Print (L"Value of Bit1 is %x\n", Bit1);       
* m$ N8 N+ u+ V2 _
7 j8 I" m. \5 {       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
& l, B  L4 D+ D7 g) J" b( l' S' ~    }

    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
3 C3 f7 [# R2 Y8 U3 l( a- F- N. J/ ~    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
  L' p4 F6 |3 O6 C        }       
       
  U+ g# e# g) O+ f4 e2 N+ |    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
) t- X! \) W' ]6 P7 E6 X$ I7 h            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
, |8 x  y5 g" L* u$ P            return EFI_LOAD_ERROR;
1 Q% N8 E& H  R2 v        }
8 o( p, Z6 ?2 H0 b
( w) r! w1 o0 a  q% L    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
, C( n6 a" r& i% j( _    if (EFI_ERROR(Status)) {
% Q, L5 i/ O3 d2 i2 t  ^        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
0 @( D& R0 r  m( h, Z0 h8 k        return EFI_LOAD_ERROR;
    }               
, R4 w# s/ u- X% Y0 ~& y               
: E5 U( x2 c. [0 n6 P3 G& m$ i    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);
        return EFI_LOAD_ERROR;
  w$ e6 j2 W8 O5 \  v, c    }
       
    Status = FileHandle05->SetPosition(FileHandle05, 0);       
5 J" [& s( |, e; X    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
/ y7 i! i+ p0 I, T6 L' z    }
       
# @$ R& P, V2 n% x4 M& f2 l( ]    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        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);
        return EFI_SUCCESS;
8 v# e. B" c" q! w    }       

5 V& ^* g9 _( ^! ~    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);
$ x, B# z/ F8 j  k- w        return EFI_SUCCESS;
/ n8 Q" a5 o: K3 d: E    }
! t) Q6 u) ^0 W! H- ]# [6 H0 I/ P       
; E' n& V0 f# K    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);
        return EFI_SUCCESS;
; A" l1 b; @; i: b7 c" `$ ~    }               
% y! e/ a1 |4 Z  S5 [       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
5 X1 [! L, e: ^& k    if (EFI_ERROR(Status)) {
' \- `+ K8 s/ a8 d, q( o/ |5 H        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
+ y! Y. Q+ n2 H       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
        Print (L"File Buffersize is %x\n\n", BufferSize);
* e$ c6 D; r, Y3 W' |/ R    if (EFI_ERROR(Status)) {
, A0 ~) Q7 Y% E  _, G7 l# C3 x        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
1 T) p% W( p+ b5 q9 P' v        return EFI_SUCCESS;
    }       
# @3 z: m. h, I$ U8 ~
//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
  P9 l$ F' v% u2 ]9 S# D    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);
+ f0 T6 o( L  r! F' \/ f3 N: e        return EFI_SUCCESS;
    }
7 j7 d/ X; k+ x7 t7 L
    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
/ O4 O! k) A; w2 w        return EFI_SUCCESS;
2 S9 C, W/ }) D* K1 l    }       
. Q2 x9 F0 q1 t) G" V- v        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);       
$ Y- B: ?8 e7 j0 Q: x# u    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
# d; O2 t( [+ U1 d7 e! M! _        return EFI_SUCCESS;
% D' [$ j) u# E9 v+ @: d7 ^    }
0 ~" h% U$ M# a4 v' N7 o2 @       
. p1 {' V( I. D3 O' b    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;
    }

1 v$ S7 Y2 j5 Y/ ^/ R    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
7 A$ T" d! S  y4 t% C& |) s        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
$ o) E5 H. {3 N+ \! r        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
2 e* w* z( p3 t  G! k  h3 E    if (EFI_ERROR(Status)) {
1 _4 t& i7 W* C" D- m+ w8 M% Q        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
! |# _* I# {) z7 |* U1 S        return EFI_SUCCESS;
3 l) M/ o" o1 D& M- Z9 P: w    }
9 a& \! ]  b7 A7 G4 `
    Status = FileHandle05->SetPosition(FileHandle05, 14);       
& i1 |" s' x( x, m    if (EFI_ERROR(Status)) {
  x5 ~, q+ F; _% A& _        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
4 p7 n0 F/ q6 x8 R; @: \! N        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
$ s8 T$ S5 d8 J! g    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
  n5 f: ?/ Z: A3 P& ~% M        return EFI_SUCCESS;
9 A/ Z" W; ^4 r3 v1 [5 w5 s    }
6 H- R8 W$ e' V
8 E5 o7 |* j2 U/ h' T6 E    Status = FileHandle05->SetPosition(FileHandle05, 16);       
) K* ~9 H' [2 W6 O$ x" [    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
8 q8 b- v! y. ?+ q8 n& h6 R' }       
( D# ~- E- G" M0 E; Z8 \6 W0 [    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
0 P& T9 l( d- l$ d+ Y' x' n    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
" T% q, s- W' |0 [+ u& @    }       
0 z) L) G+ o4 E" v$ t) ~; L
//---------------------------------------------------------------------------------------------------------------------------------------------------------------

    Status = FileHandle05->SetPosition(FileHandle05, 18);       
4 W: x$ V1 s2 z    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
+ z9 G# X4 Y$ A  b6 g( F- w% O5 d# g0 B    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
5 I% \& l8 y! y( b, O* Z    if (EFI_ERROR(Status)) {
0 P5 K5 o+ C7 ^' C        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
* ~* p- \" e- g3 @        return EFI_SUCCESS;
    }       
: \% ^2 B8 A" B; n' r4 u       
    FreePool(DestAddr);       

    Status = FileHandle05->Close(FileHandle05);
: U% v+ K* F8 Z; I6 `+ ?    if (EFI_ERROR(Status)) {
$ R8 j1 _5 b4 T/ H) j        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
0 K% @; n) U6 T0 W4 [        return EFI_SUCCESS;
. w6 r; J3 t# l$ `+ P/ f; N    }
: i! O3 T9 J2 @  p       
4 o/ |7 ~" ~' X4 G# `: O' Q6 F    Print(L"File Name = %s has already been created.\n", FileName);
" V7 b5 ^' M  \# J
: h; H0 \9 {1 I. ~( W: ^& i    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();

    return Status;
}

chenhy

//**********************************************************************
4 r" o: k3 I1 L- ^7 c1 |2 |+ o//**********************************************************************
3 c, F) r4 q# x4 |9 _//** Demo code for PCI configuration space I/O access program **
2 \% V# D  Q8 }- l//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
//**********************************************************************

4 L! [* m% |8 b# d, N7 T% P% J  L8 s#include "efi.h"
  n) [3 ?( t1 q% B; [9 D8 t#include "efilib.h"
( L: e$ l9 b8 i7 o* C* [: H
+ Y: V2 j( Q& L- r; \& G- C#define        Not_Found_Me        0x00000000
& J; f+ b, p8 S7 J* h3 Z$ I) o#define        Found_Me        0x00000001

static EFI_STATUS WaitForKeyOrReset(VOID)
% k+ p0 w/ T% C2 c( H% `- G{
$ k7 C/ x" q. i# u% h+ B1 U    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
1 z$ ~! T" f( U/ v0 a    UINTN               index;
2 R; R5 P. a; _4 L- Z& U   
    for(;;) {
% F% h6 }) p7 Y3 z        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
8 y' j% u# N& `        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
+ X( T2 o2 Y0 x7 g* `6 r( f        else
            break;
    }
; \# N( k- T& z$ w    if (!EFI_ERROR(Status)) {
) s6 @4 U) p* l9 b% V$ t        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
$ r# s: S% w, F  y    return Status;
- J" c, [: F  D) }! g# f}
* f1 K- E2 F$ r9 D) z
UINTN
# u8 n1 z7 C) o$ VAccessPciConfig (
, T+ {2 U3 F! v/ C, p    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
( d- g" Z: ~7 _: _- J    )
8 A! i. l3 n+ r! E/ W{
% R1 O& ^  W; Y0 d        UINT8  i;
; K$ O% o1 u' r2 m) N' l) m) @2 J        UINTN  PCI_ATTR[4];
9 h& X; G# [2 ~. _        UINT8  PCI_REG[4], pci_offset=0;
+ i" `2 h7 l9 e
4 x* I3 s* f: o        //get device attr
        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
, H& B7 ]5 T0 z3 q    if ((PCI_ATTR[1] == 0x00) && (PCI_ATTR[2] == 0x1e) && (PCI_ATTR[3] == 0x00)){
3 x; @! b" N* M          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++)
5 g! ?/ g' |* Q: m; v          {
        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
# [1 V- i& [; k0 e) N          }
& k0 z% a' e+ Y. H3 r          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
6 p- z6 f! g) }8 ?+ q          return Found_Me;
, j' K2 T* m* j* w    }
4 V, @6 M0 z  T3 `2 g
; B/ K3 f. O* M6 h8 y        return Not_Found_Me;
9 P1 M6 ]% H& ?" Y! t4 Y}

EFI_STATUS
! T& Q0 P, c3 Y5 fInitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
& z# X5 U& l( t2 s: @5 ?    )
8 q3 Y" r$ I* u' S* E# j{
3 X$ L& L# e1 W" `5 q8 @5 H  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

* A( z- m/ w: E% f  o2 r4 g2 s    //
    // Initialize the Library.
    //
' l. E& a8 ^" }& {' T9 a! ]6 R    InitializeLib (ImageHandle, SystemTable);
    //
    // Find the PCI driver
    //
    Handle = NULL;
    BufferSize = 1;
    do
    {
      if (Handle)
      {
        FreePool (Handle);
, Y& H+ d1 j4 m/ J      }
               
+ P9 Z6 M1 F. }+ v5 `8 T, Y      Handle = AllocatePool (BufferSize);
# ]! B8 U6 Z) B4 \      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

    } while (Status == EFI_BUFFER_TOO_SMALL);
3 i( U) O! n* W* F# h% T) y    ASSERT (!EFI_ERROR(Status));
" n* ?; _& H1 G0 A, i  J
    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
1 m: J7 A; s% ^    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
' A6 F# a1 d& a    Result = Not_Found_Me;       
% U& J( D$ ]7 o) |( N    do
4 B5 Q7 K8 _- Z# F5 h# s    {       
8 R+ a7 T- U: B2 \* ^        if(Index == NoHandles - 1)
         Index=0;
4 S1 ~& I! Q- A0 b( u( B, U          else
  y' i, Q" y: U+ {0 ~. y2 t    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
: m0 w8 c- T8 ]' E" X        Print(L"THIS IS DEVICE %d \n",Index);
7 Z' n0 N7 U# ~    if (!EFI_ERROR(Status))
* A; d; `6 T: c: o5 q3 W2 X# `2 d    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
. Q' q2 U9 f, p7 _. x9 \7 H3 P//-          Print(L"Again now Result is %x \n",Result);          
" M/ T% ?% l* q' ^* ^    }
& P/ Y  V9 V" q    Index++;               
& W" s* j$ G+ ?6 B6 l    } while(Result == Not_Found_Me);
       
6 Q. S4 {% i, m% l& t' }        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
    return EFI_SUCCESS;
0 k7 U& e$ B2 U1 U; O2 E4 E9 }}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习