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

kaurohy

/*
' y3 M7 d/ i# B; G% D * myfileio.c
* r" A* e" f0 |5 K * Apps
) R4 k' |+ U) L& F. Y: H- {/ N% w */

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

0 \7 L/ t, c# H+ C# U#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
2 e+ l9 G8 F: u6 T$ e, m* _
static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
0 R$ U. Q6 t- x! e" }4 J# Q            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
7 U& b: O, }0 q( q7 H- I. g4 p        else
6 [- a1 X" c0 `, N            break;
    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
3 l- j$ ^' d1 a5 r8 v' V! [4 D            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
9 y0 I, e" @- A2 V5 X    }
   
    return Status;
  {* y! ^& G. ]2 h# X9 Y& E& }+ R}

EFI_STATUS
EFIAPI
! }, Z1 m0 T$ H8 T8 @7 P5 I% S9 rMyfileioMain (IN EFI_HANDLE           ImageHandle,
6 g4 L; J. t5 g; B8 w: `             IN EFI_SYSTEM_TABLE     *SystemTable)
{
    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
- E- a4 H5 P. ?! r    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
) W& a; X  M& u% z    EFI_FILE                *FileHandle05;
% ~  W* U( U! N9 \1 ?$ G6 j    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
8 {) H) X& ?6 \( x- d; r    int         Bit0, Bit1, Key_Space;       
% L$ A! X0 A. m* @/ ?+ u, ?+ T
: ~5 T6 N9 A/ }/ P$ A' {! D: d( A    FileName = L"NewFile.txt";
* w6 H5 w8 J$ X* t7 e4 k) @        Space_Key = L" ";
       
        BufferA = L"ABCD";
        BufferB = L"EFGH";
  E+ B: G8 Y3 q1 W        BufferC = L"IJKL";

        Bit0 = 0xff;
# {/ x, t4 a) f' _% B        Bit1 = 0xfe;
- S8 B, M2 P* M. ^; D       
        Key_Space = 0x0020;
6 {0 J; n4 v, A+ m, l1 b5 ^. f       
7 W7 }2 v+ @3 t4 O" ]    InitializeLib (ImageHandle, SystemTable);       
5 s+ G! i/ s" B! a, B6 z3 z5 L- Q6 j
$ f0 Z2 F8 g8 e! j    DestAddr = AllocatePool (BufferSize);       

    Print (L"Value of Bit0 is %x\n", Bit0);
- p' y3 G8 ~' T, Z    Print (L"Value of Bit1 is %x\n", Bit1);       
' U  ?6 S! \" B
% _! }. X; L$ w; d0 K2 ^       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
7 i% J8 S- V0 F5 G2 l5 }) T        return EFI_LOAD_ERROR;
    }
' b* [7 v8 J2 M& U' H
    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);
            return EFI_LOAD_ERROR;
        }       
       
- O  D3 [; t) Y- G8 D3 ^" w    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
" `! _, e9 I& t5 N, s8 y: c' q6 g            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
0 \1 E+ P# n2 c8 W        }
% g7 l1 J  @( j. W: p8 g; Z
    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
; F; g$ X( r& O" C    if (EFI_ERROR(Status)) {
4 M+ S4 J3 A; l. f) n8 c( ]" R        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }               
7 |+ A  E( s. x, m  v               
    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);
$ F3 X7 z) x$ n, K; S        return EFI_LOAD_ERROR;
    }
( Y# R% H5 l1 J5 M  |7 Z. I: R       
    Status = FileHandle05->SetPosition(FileHandle05, 0);       
    if (EFI_ERROR(Status)) {
4 G- ~6 ^# N( C7 N' `        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
3 b2 o1 X' T( y7 t6 L* k        return EFI_SUCCESS;
; e- n, [' L# H. T1 B    }
       
7 [, I' M% Y6 X& P% K# d% P    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
5 u* n: h5 ~0 p2 _( Q        Print (L"File Buffersize is %x\n\n", BufferSize);
# O' O; }! f- r: B    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       
+ g4 E  D; O8 L0 r: O9 }
    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);
1 ~; i/ h, H) e) y        return EFI_SUCCESS;
    }
- T) E9 y2 r8 o( b" G) g       
) w1 F; S# Q  x$ \/ k9 P1 j    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
, ]$ d* k& n8 ?        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;
+ T6 O3 U4 ?0 G; |  P, Z3 s    }               
- Y* o1 t7 o( j/ s; N0 e       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
: G! ]$ y" N2 O2 P; F    if (EFI_ERROR(Status)) {
$ T3 \& k9 t" u% j2 ]* ?, j        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
$ j& ^% d, K) C% J' ?$ e7 q       
6 O* Y1 U" [' p6 l# r' e; w+ G5 W    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
4 v6 Q6 p  r: y( ]5 j        Print (L"File Buffersize is %x\n\n", BufferSize);
6 K# D3 G) }6 u% ?5 \    if (EFI_ERROR(Status)) {
  \' }! ~  [$ @$ ^        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
# q: [. u/ }' @% w" C: w    }       

//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
+ m" O; B& b+ B1 Y    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
: \" w% q0 f5 D; ^1 i) e" }$ `    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);
* g9 D5 c+ x; |+ s9 C$ @6 T# ~        return EFI_SUCCESS;
    }

    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
    if (EFI_ERROR(Status)) {
% Q* K$ r) q& b: y  P        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
: Q9 s0 L& e# }+ t8 m: m        return EFI_SUCCESS;
    }       
        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
+ _0 _/ l4 d$ {+ m; D8 R2 |
    Status = FileHandle05->SetPosition(FileHandle05, 10);       
0 H. E+ L5 c. I; l) J    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);
; I" L, L1 {) r( b( J    if (EFI_ERROR(Status)) {
, V1 z$ s( b& ^( e: q; T0 v        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
# v6 |5 u( \# p    }

+ |2 G/ {7 @  A  N5 U    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);
. C# w( d6 A7 b. k# a+ m  A" F' U        return EFI_SUCCESS;
    }
       
8 `" v4 U: n- |6 D    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
$ _( K0 j( b2 }5 Y7 Y5 n  m! N        return EFI_SUCCESS;
' A$ w  D% @2 }/ B* k5 C    }

    Status = FileHandle05->SetPosition(FileHandle05, 14);       
7 G! n& e4 T5 B1 p& u  r    if (EFI_ERROR(Status)) {
0 {! w) T2 p4 E4 w  Z( j. h3 q        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
+ M# w) ^5 w$ d- T' f       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
3 o! V! e* g& c0 b; i        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
( T9 J, ~. S1 y0 M    }
% t8 P0 d8 K- w, k/ q
    Status = FileHandle05->SetPosition(FileHandle05, 16);       
    if (EFI_ERROR(Status)) {
5 g( K" z7 V! `, N8 G8 X        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
. k/ @  N" g4 g' F       
0 I* F, l# X2 S    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
6 {/ z5 E8 O$ W- R0 o        return EFI_SUCCESS;
    }       

//---------------------------------------------------------------------------------------------------------------------------------------------------------------

    Status = FileHandle05->SetPosition(FileHandle05, 18);       
/ u% j: _& G- F3 ?' q$ J0 r+ V, \    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
' O& O. {+ ]" O( v& P6 ^7 v% p        return EFI_SUCCESS;
8 y  u6 a* e$ }- f; F& y5 t1 n8 G6 f    }
8 g( s( X* X' T7 Q       
5 d2 B) u4 m0 a1 J" g# V2 @    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
+ V: r% A; P. F8 ^# q        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
! W6 K  H) O- u; A, a* M        return EFI_SUCCESS;
    }       
! {# b; N1 H# a) s2 ]' e0 \       
    FreePool(DestAddr);       

; f" I6 y3 y2 X( {. X) _    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
1 g+ G) |8 z; P2 K% r: b0 D        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
8 a. U, I* F$ B/ ]: w        return EFI_SUCCESS;
    }
/ [) i( S" T; |9 [, |# I( {- r       
2 A/ r4 R+ t, {) O    Print(L"File Name = %s has already been created.\n", FileName);

. T* i0 p  T1 V( j1 N5 {: k4 v    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
- c: T( I/ m" [    WaitForKeyOrReset();

8 h: m+ ?. U) _% y  X* \    return Status;
+ H: A  C+ R4 W( l4 ?}

chenhy

//**********************************************************************
//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
6 S4 `) I+ d0 F4 B* S//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
//**********************************************************************

6 k! ~) U: W9 h: h#include "efi.h"
+ Q4 d* W9 R' c  q#include "efilib.h"

+ z- a/ m. l4 K#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001

/ j5 Z0 ~, J/ d. @# a; D) n0 Rstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
: L$ ]  Y; O6 o  u5 Q: ?% D% j  `        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
6 ]1 m, p3 U( t$ ?        else
            break;
    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
* J4 K; y  z8 \1 q. k+ {! ~! `    }
$ I! q/ S- ~7 p0 x; y' y8 b. s" z2 q   
    return Status;
! `, \& r2 R0 I# b" c- V) f2 M' s}
8 |" S, ^& a. {/ o5 K
UINTN
) Y3 r" f0 o4 T, n4 T- z: j3 F" PAccessPciConfig (
, N5 v6 _2 _* Z0 h, ^    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
{
        UINT8  i;
        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;
0 H; Q9 W" R0 t" S
        //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)){
% s& y: a9 v& ]% |* z1 V" Z4 B          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
: _: `' L) m6 y( C9 W# f5 \% _          for (i = 0;i < 4;i++)
5 k$ z/ e7 @' V          {
$ S/ C" j. [+ e# n& P* m        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");          
0 N, f, D0 l2 F. ~          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
    }

7 X5 A1 I* z, R1 b        return Not_Found_Me;
}

EFI_STATUS
7 F. W2 P% c, C# h" t9 iInitializePciApplication (
1 s  K9 x5 W3 B    IN EFI_HANDLE           ImageHandle,
. q% e, F2 K( h& m9 @    IN EFI_SYSTEM_TABLE     *SystemTable
" t! Y( r7 b7 R    )
, Y7 L9 V5 x/ D. l7 P; C' E{
  EFI_STATUS              Status;
. f- J2 E% I$ O" g/ i  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
6 x1 w- M$ G7 H. |5 D( o  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
6 D" H2 }% t5 [
    //
, k7 K/ f. C0 V0 W; G. P6 F# N. z    // Initialize the Library.
    //
    InitializeLib (ImageHandle, SystemTable);
    //
# i( ]8 t1 [' S+ G( k) b& K    // Find the PCI driver
    //
    Handle = NULL;
- h3 N0 L3 i& i" e  P    BufferSize = 1;
    do
( c2 {$ E# t  W2 F4 J, E' n    {
      if (Handle)
5 j$ g" H( q0 O! y      {
        FreePool (Handle);
      }
% L7 q+ V2 X7 L, n: Y! f: ]2 @# u               
      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
3 ?1 `; U% b" _4 M( C. X  n4 `
    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));

9 X% E2 x5 @+ L$ q) v# R. X    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
* Y% X, R4 ]7 Z) O6 Z% e& Y2 d; d    Index = 0;
2 P; Y7 d9 ^; s% |* a    ST->ConOut->ClearScreen (ST->ConOut);
5 b( Q' t* y/ J4 e    Result = Not_Found_Me;       
    do
    {       
        if(Index == NoHandles - 1)
3 H/ ?, x" R3 `7 m: Q0 O         Index=0;
0 U! R6 j: H8 G2 a5 O, T6 \, a          else
' p- Z* Q- x- T% T  Q    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
. \" [& }( v" }$ P* r1 u    if (!EFI_ERROR(Status))
    {
% S! {. e0 h. ~* C5 W6 x          Result = AccessPciConfig(PCI_DEVICE_TARGET);
  Y+ J3 e4 a0 ^3 J8 d//-          Print(L"Again now Result is %x \n",Result);          
! |) |  @; I# P1 s    }
    Index++;               
    } while(Result == Not_Found_Me);
7 @7 n- W2 i6 |' i       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习