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

kaurohy

/*
* myfileio.c
9 i! S: D0 V& Y, \* v( h * Apps
*/

- Q( w8 }9 Y5 Z% Z4 C#include "efi.h"
4 t" H% K: J/ w8 F#include "efilib.h"

  I7 u" n: q1 g2 X( ~* t#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
) [9 q+ [" p! q0 n8 J5 {
static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
; j. K0 x! }7 Q7 H    EFI_INPUT_KEY       key;
    UINTN               index;
   
* h  i3 ]/ a8 P" U) o( M4 [    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
, Y9 ~# i$ ^& \  H        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
, z, h, }, w( e6 \            break;
    }
3 r4 U! V3 W& H0 n: O- b# \    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
2 l' K0 |6 Z, z; T6 t            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
: w# U+ R+ X3 [4 H    return Status;
}
& h/ }0 e0 b1 ]3 _6 J( p! a  \
- I9 G" u$ ]1 Z5 `EFI_STATUS
EFIAPI
; j" S# Q* S, O& L# O6 m5 fMyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
{
% ^- O3 J* i* B, i9 V, i    EFI_STATUS                Status;
3 H# l# v9 B, X( N5 O2 `1 p3 S    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
3 E; c* v8 q6 P$ d* m5 O+ y  I    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
; g% {* B5 m1 X! v    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
2 n0 \+ z" ~" r/ w7 V* f    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       

) f( e+ ?$ X% c% C    FileName = L"NewFile.txt";
        Space_Key = L" ";
       
        BufferA = L"ABCD";
        BufferB = L"EFGH";
        BufferC = L"IJKL";
% d; h$ F+ `) |& y( \4 w/ y
9 M1 \) b8 D9 g- U0 L9 S% z9 g& @        Bit0 = 0xff;
* P2 v4 V% h+ X/ s        Bit1 = 0xfe;
0 [# L8 n3 w- v2 k       
        Key_Space = 0x0020;
* h& d2 b5 X& |, d0 o       
    InitializeLib (ImageHandle, SystemTable);       
, P( g6 {5 @. i3 C" R
3 C; t! ^0 i8 ^$ U, ^4 D8 T    DestAddr = AllocatePool (BufferSize);       
2 n$ D9 Z: U3 X/ l) T; ]
    Print (L"Value of Bit0 is %x\n", Bit0);
5 [/ C2 \* Q  z) u$ ]    Print (L"Value of Bit1 is %x\n", Bit1);       
% Z8 O- n3 Z0 i  F" z: q
       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
% S# m* @9 @9 j) |$ ^9 @5 b6 t    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }
: i' \- B" `& K1 n0 H+ w7 M
    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
    if (EFI_ERROR(Status)) {
6 y  s" W- q5 a            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
        }       
( @7 s. f. A. N       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
) e5 J% i: u! x- A& E( a    if (EFI_ERROR(Status)) {
' y" ^! i$ ^/ k/ i) }" {8 H            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
0 e- f6 P, q% ]/ ?( b- h, U        }

! r9 I+ V/ u4 d6 }' k/ w    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
4 M" V- _) D6 x# A+ s6 v        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }               
1 Z  x5 q% v! T8 v  _               
$ s, }0 w  Y& [- `    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
2 w, `' |: `3 l! n8 G    if (EFI_ERROR(Status)) {
        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
6 r' Z* i' R" X; I' o8 T        return EFI_LOAD_ERROR;
& B& V% e6 V" g! M) ~0 O$ a    }
       
    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);
+ j, N0 R; V% U8 a% _  w        return EFI_SUCCESS;
    }
       
" L1 P4 c3 i4 x/ l    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
) e/ D$ G( p4 S5 L$ h! }% f    if (EFI_ERROR(Status)) {
' q4 d" ~$ q/ c        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       

; `  A  I# N* Y- }) m- q    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);
        return EFI_SUCCESS;
8 O5 |/ }, J* ^" ^5 {1 f: y    }
1 ^# d! q& j3 ^' b       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
/ w/ w% o8 P0 G9 [    if (EFI_ERROR(Status)) {
& U- F- q% p! M        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
, ~3 Q4 Y- q4 n% f9 j) H        return EFI_SUCCESS;
6 x5 b0 i2 [, m7 R6 s    }               
3 Z5 l- T$ e% j       
# t7 r6 O. U& k3 ~6 G+ X    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);
" w) A) d, J9 V+ x0 C8 l1 Y        return EFI_SUCCESS;
    }
       
, p" d0 s1 f3 y; ~    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
        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;
# k0 k7 S( D! M( |    }       
5 G$ R9 h; W" `
- w% K# n8 w+ i8 }//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
0 M% b0 |. S! C0 v        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }

8 q6 `* r+ Q3 |    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
" L8 V! }+ e' S    }       
        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
2 c5 g. C* m9 [( U) I
. Y1 H) l( P$ G/ T% z. U/ z    Status = FileHandle05->SetPosition(FileHandle05, 10);       
    if (EFI_ERROR(Status)) {
4 E- H3 _2 V( ~2 c. n        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
# z0 M$ _5 C# d( f8 i: R       
9 U6 S# u# N/ j: 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;
    }

' W3 ~! @8 A; s1 }( Q    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
* s- G  Z  y/ S1 W5 c        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
  \& \3 x3 B; h. Z" I: L7 h( k        return EFI_SUCCESS;
    }
2 W. V" t4 o* ?- S* ~: ?6 q* O       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
8 E8 q6 _) R2 k7 y) M8 r    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
: @8 u" F, ?2 y4 x5 z        return EFI_SUCCESS;
    }

    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;
1 I3 W$ r3 N  q) Q$ g& Y    }
8 ^9 [  f& o9 h: J* R       
, }3 r5 ^/ M+ T& Y8 J8 O8 q    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
+ @8 I2 ^+ Q$ O7 E, Q    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
. x( Q; P$ ?% _& [3 h" [/ X" X( [    }

$ w& L; H' Z' J9 c9 \0 k! V1 i    Status = FileHandle05->SetPosition(FileHandle05, 16);       
    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
" N5 O% R% m% b( i  U' B        return EFI_SUCCESS;
+ j, k6 R# o& J" K! {    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
4 u2 ?- \" Q3 Q8 K. P6 }    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
9 O- f% Y' A* y+ }# {        return EFI_SUCCESS;
    }       

7 N# l  I, S  L/ [4 \+ G' f7 I//---------------------------------------------------------------------------------------------------------------------------------------------------------------

  B* M2 \9 `) S7 O+ I) v/ }    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    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, &BufferSize, BufferC);
+ a4 W/ l4 h0 J% o# {3 h/ y0 z    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
+ S( U( B2 ^( N) R        return EFI_SUCCESS;
0 W; F1 D. Q2 G( j4 G& y    }       
       
    FreePool(DestAddr);       

" H+ q% r- f/ q, d3 s, j  }4 n* F; T    Status = FileHandle05->Close(FileHandle05);
9 t/ X; M4 V: ^    if (EFI_ERROR(Status)) {
, E+ i( W. ?# ^* o& h# i        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
0 L' C# c/ o# _) W        return EFI_SUCCESS;
    }
       
0 w8 _& X5 I; F, ]% D0 f  M    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");
0 H: ~4 V8 M5 n$ i$ q    WaitForKeyOrReset();
! r/ L! K& k0 q9 F* [
    return Status;
}

chenhy

//**********************************************************************
7 u& s( K8 u: D% w* s& t( N: e//**********************************************************************
# B( ]6 H# g" ?! d6 o$ x  `3 p//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
2 x6 u- |6 k3 w) S" r//**********************************************************************
: f+ Y3 v( i" U5 M  {  q4 b//**********************************************************************
% `9 ?7 h* v: H7 `! U# x
#include "efi.h"
5 S9 b7 A. C% Y9 k- c- ^3 x4 }#include "efilib.h"

#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001
7 K; d2 W/ E3 E( ?8 N( w+ C& _
static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
8 {2 a+ Y0 A$ ^% h. k    EFI_INPUT_KEY       key;
    UINTN               index;
   
    for(;;) {
' n) R( {0 m* e- ~' l" R8 T2 O# c        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
# W% T5 S& Q4 I; {8 t) o" f! V. |            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
2 ^& ^0 ]- }7 o  t8 l+ G        else
2 M9 X" ]  t2 t6 p4 |            break;
4 x( Q$ \; {% t+ Q/ c+ o' e    }
- s' q$ b+ f+ C, D    if (!EFI_ERROR(Status)) {
+ d# m7 M6 e; N, j& x3 z# z        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
    return Status;
4 O- `2 s) f; w6 {# k}
' o5 f# g8 F& W9 T
UINTN
* i3 {, [* w; qAccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
+ t% L( C4 n. a8 `* X! y& }{
        UINT8  i;
        UINTN  PCI_ATTR[4];
0 s' b$ `+ Z. b  l" {) p( w/ {3 j6 [2 r        UINT8  PCI_REG[4], pci_offset=0;
2 Y  N) p7 w2 T; e( z
        //get device attr
8 u/ E, @* Q+ z: t9 `& H1 h        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]);
+ N" _: C  [, K' {          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
          //print register value
          for (i = 0;i < 4;i++)
8 u8 S" ?1 g1 m7 v% w          {
        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
) |  F& r; U* t          }
4 B' T0 W# p: D          Print(L"\n");          
3 T7 Q% m; d/ _- P% R) E          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
: |; K/ @, u; o! G; m          return Found_Me;
    }
% Q" ^: s* [5 x5 w; f/ w: b
) w/ `; ^7 E% ^. U/ F2 b6 r        return Not_Found_Me;
- E  y2 l+ u6 H' v8 }+ `4 Y, ?}

' e' r" i/ J# u, y) x& D$ }EFI_STATUS
8 a! Z6 E% q2 I* m) W0 [2 Y/ B4 UInitializePciApplication (
' X* Q/ ~8 B3 _3 [    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
    )
6 ~2 o1 ~+ r+ B( P9 b8 X{
6 h* L# w5 i7 O6 O* r' a  EFI_STATUS              Status;
4 z6 F- y& @3 G8 R3 `4 W2 J. @  EFI_HANDLE              *Handle;
" C/ l9 O$ ?& t- v& t  UINTN                   BufferSize, Result;
2 g; I# y6 V" b" C- b  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

    //
    // Initialize the Library.
, e5 @. i! k% v1 H    //
  P9 v, v* W* O+ D* W& P    InitializeLib (ImageHandle, SystemTable);
    //
. C3 \' J: T, ?$ z8 V    // Find the PCI driver
    //
    Handle = NULL;
) T+ p  a+ y$ H4 X4 D9 w    BufferSize = 1;
    do
    {
      if (Handle)
      {
& _* Y* L/ B2 v; t. P; p  F( a        FreePool (Handle);
+ V" E4 {, x" E: O; U8 a; i      }
               
* y8 c: m9 P: I( `7 E. L/ B      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

    } while (Status == EFI_BUFFER_TOO_SMALL);
. D4 L- Y# x4 O( k8 ~) Z    ASSERT (!EFI_ERROR(Status));
# N# F" h. j) x0 [. c0 c
    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
- j& Y+ }: K4 f, D  y+ [4 T    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
5 C3 U+ K8 ], o* o" K2 _! `    do
    {       
* N* d1 ~8 A7 ^3 B, |8 q3 W        if(Index == NoHandles - 1)
         Index=0;
          else
( K6 C/ {$ u5 U2 m- W    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
. t) i9 i' N$ f        Print(L"THIS IS DEVICE %d \n",Index);
3 n# A1 @2 W& s. i    if (!EFI_ERROR(Status))
    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
$ Q: T& `+ }7 e. N) d! |    }
8 [  ]+ C0 [! h7 |& C    Index++;               
2 V+ ?8 s  D$ ~) p* x    } while(Result == Not_Found_Me);
       
9 X# Y4 e( ?* H2 U7 n, n8 @        Print(L"Please hit any key to exit this image.");       
! m- q, U8 C( C+ r, O+ ?- v# w    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
7 G2 ~" @5 k+ R               
! ]4 H/ f- b) f$ L    return EFI_SUCCESS;
: ]. W/ l" w' n6 f+ J4 S& X& a( M}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习