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

kaurohy

/*
. X8 R: J" q# O$ G! n. U7 G * myfileio.c
8 S7 a: s. C# x: c5 F4 W* `" g8 R * Apps
*/

( |; c3 Q; p5 h* I#include "efi.h"
#include "efilib.h"

; S1 R( h; e6 r1 o! }: }$ p#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

2 u+ I9 C& S$ B( n, ystatic EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
# z& D  B' o, h2 T$ X/ d0 v    EFI_INPUT_KEY       key;
1 V" E& x, N. Y, u    UINTN               index;
# B' T8 a2 }, U; X) J9 v   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
# p# H1 ]: U' F5 T/ r        if (Status == EFI_NOT_READY)
; a" u* A  l' w& g, k* @            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
8 I7 ^. K0 g% [7 W9 g2 r            break;
    }
    if (!EFI_ERROR(Status)) {
8 M, W+ P* E/ U' {/ X) U) j; u1 T        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
6 q8 H  n8 J1 M2 A6 Y+ a   
6 i' N, M1 k  F& }3 l8 e    return Status;
; ~; ?; {4 y) y/ A}

! Y# ~3 i* j/ z! wEFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
+ w! W( _3 v% W2 O7 g: F. V             IN EFI_SYSTEM_TABLE     *SystemTable)
{
% V$ ^/ P9 z0 ^4 _6 @    EFI_STATUS                Status;
" f/ I* v" q$ R( {. K    EFI_HANDLE                *DestAddr;       
! A- g0 @* B. R% M    EFI_LOADED_IMAGE        *FileHandle01;
: @6 l& E$ s! a3 ^3 [5 I    EFI_DEVICE_PATH        *FileHandle02;       
% J7 ~. o; C1 \7 ], L! r    EFI_FILE_IO_INTERFACE        *FileHandle03;
4 Q5 F& I& s. D/ R+ C* H    EFI_FILE                *FileHandle04;
3 i) q3 z% j& B8 H/ e    EFI_FILE                *FileHandle05;
/ T/ ~  ^, h' X$ c    CHAR16                *FileName;
7 j; d% D& Y/ Z. C2 d6 h" U6 I- `        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
4 ]  |' P( W5 C0 }/ C    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
3 _+ v1 L$ F( T: B% V" O. ^# m    int         Bit0, Bit1, Key_Space;       

. }6 z( m# f- l8 {) T    FileName = L"NewFile.txt";
& s7 g) I3 k4 t  _& t        Space_Key = L" ";
       
        BufferA = L"ABCD";
4 V5 p) w( W! R2 N9 W        BufferB = L"EFGH";
) W- W2 ]; i/ h/ X0 C5 p        BufferC = L"IJKL";

5 Z( C5 k( y+ h' ?4 @  E        Bit0 = 0xff;
        Bit1 = 0xfe;
! Y% q# s( N" Z4 e* t5 p. ]( `% Z; v       
8 R! Q8 B& [9 c+ d        Key_Space = 0x0020;
       
    InitializeLib (ImageHandle, SystemTable);       
3 t2 r) D3 J& p
( e& _* k8 n2 c- }: _    DestAddr = AllocatePool (BufferSize);       

! W$ F0 {+ H) X( i. i# W( q    Print (L"Value of Bit0 is %x\n", Bit0);
    Print (L"Value of Bit1 is %x\n", Bit1);       
- U2 O2 ?: O1 B5 q0 \! ~
       
3 q: l: ?, y2 ?8 r# y    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
- E* _' v( t; W$ z    if (EFI_ERROR(Status)) {
1 i; z9 A$ v3 z6 Q8 P* U/ I7 q* G        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
. Z  e3 h2 O  u* W! O& q        return EFI_LOAD_ERROR;
0 a: d/ \, p3 Q4 H    }

  ~" g6 ?& m% w9 @( H, c( E    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
    if (EFI_ERROR(Status)) {
- O* b9 w' z8 K3 N2 K0 p# B            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
# R, u5 E1 q) a: |            return EFI_LOAD_ERROR;
        }       
" s( L+ h" j( ?7 U0 C% s/ }       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
2 i% L) M5 W+ |  [" B            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
  O0 {* W7 W9 x& L2 Y            return EFI_LOAD_ERROR;
% T+ x0 w' J7 M) r& _        }
; J  u2 t/ U  R% f0 f
0 D, [/ K3 ^! Z0 @' n6 ~    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
2 B7 G: Q" K7 F2 J# ~        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
) M5 R* z, V+ k/ y$ N! W        return EFI_LOAD_ERROR;
3 G5 G" f. o& S% Q# @* w    }               
; M5 h( u( }' K: J; ]! T. S1 {               
    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
' ~# z& C, e- M9 O" W    if (EFI_ERROR(Status)) {
5 ^/ \0 V4 e" H        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }
2 V# n# u8 Q: s; ~4 ?       
8 l1 q( s: P1 l: D0 e6 _! J    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);
        return EFI_SUCCESS;
    }
       
% i0 q1 K7 r+ i6 U: ~    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
! O1 Y3 w0 r- g( Q: L9 O8 H/ u    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
( n/ f/ G! g1 Y    }       
+ U. k: {( B) U: e+ f, w/ _+ Z
    Status = FileHandle05->SetPosition(FileHandle05, 1);       
    if (EFI_ERROR(Status)) {
! M) O9 g# O/ O( n* E% }; m        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
% o  n; L4 c9 R, S9 m; T    }
       
6 l) c8 T7 W& k; l% h1 j, P    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
# y- F- b0 B7 H. Q: D2 R        Print (L"File Buffersize is %x\n\n", BufferSize);
2 c$ l; ]  `+ M  \    if (EFI_ERROR(Status)) {
4 B& N/ [6 R0 |# L. ^2 _        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
& U. n* z3 K/ ]& m/ K6 V' Q5 ^        return EFI_SUCCESS;
2 o7 ^+ S7 y9 X- M8 S    }               
# ?4 ^. N0 x5 t; ^  @! J; V, s       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
4 ]1 ~" g! d: v$ S; L    if (EFI_ERROR(Status)) {
3 e$ c7 P- K4 r# M% V# L9 d        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
, k7 @* t8 E7 W) Q        return EFI_SUCCESS;
    }
; g9 I) i3 D8 h/ Z( Q/ J& J8 ^       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
        Print (L"File Buffersize is %x\n\n", BufferSize);
# Z* j& f, Y' _0 C: K    if (EFI_ERROR(Status)) {
! C( X' h5 e7 _+ d( }/ ?        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
$ j" K& l  j1 v) v% d! g1 k    }       
1 d  E/ B7 ?# i$ b, k
, S0 W7 C; D1 x& x4 n6 s. S- P//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
: ]( E8 A& M0 H9 y: _, x        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);
    if (EFI_ERROR(Status)) {
  e0 E9 h1 [, Q+ S8 j        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
! J% Y' g3 _4 `& a  x0 Z3 X/ y3 K    }       
) ~6 E2 k3 S; H5 n        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
5 O) S4 W) i* \5 @//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
# G: ~, M! a3 T
    Status = FileHandle05->SetPosition(FileHandle05, 10);       
    if (EFI_ERROR(Status)) {
8 E% s5 u. H' f% `5 F        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
. w" h- e* G6 Q' Y& _/ E; d        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
; v4 Z& ~  |  a& L% _8 p    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
/ M: Z: F6 a( N2 \3 C/ u    }

7 q3 Z8 H& V9 Q7 c7 G    Status = FileHandle05->SetPosition(FileHandle05, 12);       
- }0 x3 L  q. w& i! Z  S    if (EFI_ERROR(Status)) {
, @% P+ b( T* H5 e        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
0 O" E! `! _. h3 X9 C2 a- [# p- K    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
( ~- M) x8 _2 ~) V& R        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
  h) [( e% J) F! y3 J& L/ M  m        return EFI_SUCCESS;
8 B+ l/ J+ \" c) G/ A& g    }
* W% _* f& W/ h, x* g% w8 U$ G
    Status = FileHandle05->SetPosition(FileHandle05, 14);       
% X% B, H1 ?9 N3 C; C* `- F    if (EFI_ERROR(Status)) {
; m$ J7 {" ~- ]" n) d+ c* X$ }# e: @        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
7 ~& K0 V& h$ g       
7 P) A" l+ C. x5 ~3 i" ^; u! G    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
1 Z1 G$ k+ Z5 A' I; r    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
( O; r. ^* R! c& g+ j3 {        return EFI_SUCCESS;
    }

    Status = FileHandle05->SetPosition(FileHandle05, 16);       
. l6 q9 }9 c9 }. @" W- v* I) z    if (EFI_ERROR(Status)) {
$ B; K8 n  z$ c3 l: B1 n6 R        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
9 q% ?2 v3 y& q- w+ w    }
2 w% z; T! _5 }; L       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
2 B+ x& q) W; U" E        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
. I$ g. I/ \2 G! R3 ^0 }/ w  f    }       
. l% G3 v! ^" j
//---------------------------------------------------------------------------------------------------------------------------------------------------------------

    Status = FileHandle05->SetPosition(FileHandle05, 18);       
3 b$ r8 J5 P/ k' B    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
  q) s. V5 O6 ~        return EFI_SUCCESS;
    }
       
0 B# E7 e& ?6 {/ X- t    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
4 g4 p: h9 F5 l3 y  [" s+ e3 s        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
% N; V, I( Z* y, b- @  P! i+ H' Q5 |       
    FreePool(DestAddr);       

" \9 {9 y, @/ C# K    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
1 A5 H) e& }/ {& Z    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");
    WaitForKeyOrReset();

7 O* |2 X( }4 T- c/ J; ?3 C    return Status;
: G9 u; b1 H% J1 T5 i3 |}

chenhy

//**********************************************************************
9 @2 I5 P# J; h- Z* n  _//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
8 p* L0 ^6 s; r* D3 n! V7 y: C$ o//**********************************************************************
! j5 r( b7 Y4 r$ o6 D, Q- h//**********************************************************************
' M2 D- R( P- D/ l/ M4 E
#include "efi.h"
! P( e6 }" @5 O, z# u/ t3 O#include "efilib.h"
. G& A- |$ i: y
#define        Not_Found_Me        0x00000000
1 z$ V* m  N6 ^6 F4 r( r#define        Found_Me        0x00000001

* ~6 M% i4 z$ i4 s. C4 kstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
6 ~; c2 p- ]! T* b    EFI_STATUS          Status;
* f) {+ f8 X" ]% w    EFI_INPUT_KEY       key;
    UINTN               index;
   
    for(;;) {
2 d9 f7 R( V# J! z* |6 }0 ~8 H        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
7 o, o. ?1 L7 a% h" q3 U            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
% V/ |6 v0 i6 {8 W1 Z0 L8 a        else
/ P+ a# k) e( z6 p            break;
    }
9 d7 _! t  K2 }% D# ]/ g    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
0 C/ S" T$ B3 @. D            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
+ P* T! u. P6 a2 t    }
   
    return Status;
}

UINTN
) J0 H2 I6 b& u0 P4 G8 ]AccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
( r: h& o3 _! l) P4 K) e    )
{
        UINT8  i;
        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;
' M, ^) k9 P! a! f0 Y: F1 D
        //get device attr
) A, ^) m) _5 R: z        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
& z5 A+ k- y6 R$ j: t0 b% T$ Y    if ((PCI_ATTR[1] == 0x00) && (PCI_ATTR[2] == 0x1e) && (PCI_ATTR[3] == 0x00)){
) i" j( ?2 E5 M$ y* _7 b' c/ z          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));       
" R8 u+ R0 Q" p2 p% Z- {( G7 O6 U          //print register value
8 H, x6 l% k4 \8 X2 B          for (i = 0;i < 4;i++)
8 Z7 Q4 L& @$ T) m& D+ [* P          {
0 @- i7 m# N$ b7 m2 e9 H, W5 @        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                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));                  
          return Found_Me;
- Q0 r, G, j+ J$ n! M4 i; C9 J    }

* ~$ G: P) K( c' _" T        return Not_Found_Me;
6 z* y6 x6 c6 W$ O, q8 C}

/ t. l" [  s9 D( XEFI_STATUS
8 O3 f0 a8 W" A6 `InitializePciApplication (
! ?4 j3 o- G  }2 q2 b: J1 y    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
    )
{
2 G# T/ T5 \  b& H  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
, q: M" x) i1 t! o5 _  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
7 Z5 G3 V+ z$ y/ ~  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
+ P: L' F" ^5 v# ~8 {! C6 Z
    //
, I* Z5 {+ J" l$ W    // Initialize the Library.
    //
* R8 `( k" X6 A; W8 \/ s( d    InitializeLib (ImageHandle, SystemTable);
    //
0 J' \+ x, J  X( E2 Z5 ^: U    // Find the PCI driver
1 m- j$ [2 Z! Q- E- B! W" e    //
    Handle = NULL;
    BufferSize = 1;
( E; Z% L& k0 N% J2 W# _5 E2 h    do
" t/ J, Y6 h' l/ ]; ^9 M    {
3 l  r$ v% g  a5 v      if (Handle)
      {
  n+ W$ o  ~; I0 H  L- M  `) g) }        FreePool (Handle);
9 f" g- e+ l3 O3 I      }
* j, \  I6 l- P- `3 N6 h# {7 X: G4 w               
5 L' F# B% [. k- P1 E7 b      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
3 p& s& H7 \8 g" i5 J) S% P- T
    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));
, c/ {" s7 S3 ]  j5 @9 a+ c; l
0 P  V. i2 ]) b( t$ A9 Q& ?) Q    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
+ X5 s6 x0 N# K0 G4 j# R9 Y$ o: F    Index = 0;
/ k" D* C! t5 y! [7 n  m    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
' e/ V* U8 W# m    do
9 a" X0 f9 ^0 H    {       
. J& b2 _& m, o. o* j8 ]9 i& J        if(Index == NoHandles - 1)
         Index=0;
5 q) w% z* W4 C9 ~, u          else
. l  F( q( W1 C! e; W1 s    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
- J6 C9 Q# S( c  A        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
2 V2 E0 t4 Q& e1 e          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
    }
    Index++;               
    } while(Result == Not_Found_Me);
" Z" e6 f; o4 x2 Y% l( G7 F0 m       
' Q: i+ }' [& Y8 Q9 o9 d6 |        Print(L"Please hit any key to exit this image.");       
  g1 h  |% Y5 @3 t0 j4 R    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
3 N4 S; [/ k/ E, `  i, X: k               
    return EFI_SUCCESS;
* `7 i$ \. O* q4 Y( s}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习