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

kaurohy

/*
* myfileio.c
* Apps
/ F& l3 Y" R* V7 f+ ]. o */
4 r: h2 R( I$ m# s
/ N& x- o* Q- m; U8 `; r; b6 S% Q#include "efi.h"
: Q) W" f8 C$ c6 a#include "efilib.h"

, n2 |- I& a& |8 a" s  e, E7 f#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
- S& f2 J+ t$ q) |# _9 P
6 l5 ~: {$ |! v' Z, x6 Vstatic EFI_STATUS WaitForKeyOrReset(VOID)
  C8 L3 ~* a8 ?$ J4 l# q. T+ o; _* F{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
) G2 }* _5 Q8 _  H9 y; [    UINTN               index;
% D4 ^& B: ]- M7 F, Q   
    for(;;) {
& F' M9 z, b* @6 w        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
8 V% q) A7 \5 {* ^. g% k$ z        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
- X. Z3 _8 r6 [8 o& x3 Q            break;
    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
5 P* b6 \& G! d9 t7 H    return Status;
}

EFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
{
    EFI_STATUS                Status;
0 J0 \% {$ ?3 ~1 x7 C    EFI_HANDLE                *DestAddr;       
1 X4 x5 h& y. ]9 c    EFI_LOADED_IMAGE        *FileHandle01;
+ T+ {* W9 V: s" P    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
7 U7 D6 D- K& z: c9 [    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
4 ]6 W4 q4 m# \( u8 x    int         Bit0, Bit1, Key_Space;       
# Y8 h% x7 p6 {
- U- x- V' n7 Z! t7 k    FileName = L"NewFile.txt";
, u4 }. i5 x* R- [        Space_Key = L" ";
       
; R" L8 u5 n  @/ `9 q( X- p! M        BufferA = L"ABCD";
        BufferB = L"EFGH";
' }* i  J# g% m. W2 y* U        BufferC = L"IJKL";

        Bit0 = 0xff;
6 u+ y* h% A6 F- q        Bit1 = 0xfe;
: j6 M9 d/ H3 H1 w& y( @6 E       
# L: g. Y/ @' w        Key_Space = 0x0020;
       
    InitializeLib (ImageHandle, SystemTable);       

5 D. _+ V' d2 P* }* _$ Z8 [9 N    DestAddr = AllocatePool (BufferSize);       

0 T) b7 J5 V* ]" B4 g" o- f    Print (L"Value of Bit0 is %x\n", Bit0);
    Print (L"Value of Bit1 is %x\n", Bit1);       

' h* i7 I' ]/ \; q: ]       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
3 o$ w& R" n5 K1 m        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }

    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
    if (EFI_ERROR(Status)) {
$ s% p" v% G/ J4 w            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
        }       
       
' e8 w) E& T# e2 {    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
* G% S" g" t, n; Z# J            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
* d8 f, \% H# A" D+ P            return EFI_LOAD_ERROR;
0 y( ?* Z7 B" [4 ~/ C        }

6 i4 Z" e2 D6 V2 @* u2 q    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
# d; r2 K; M( d. P. h0 S    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
2 R/ ~4 ~9 K5 A, |* }% u* p    }               
; ]! t8 w$ J) u' u$ P7 z/ f' }% U+ `               
    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
; j/ i5 i3 p: g0 c# M    if (EFI_ERROR(Status)) {
# b- w6 \* \- C1 j" K. U7 H        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
9 l+ \3 K0 P0 C4 r  u1 E+ e        return EFI_LOAD_ERROR;
; {& _4 h1 g) t* o* h( B! X    }
       
5 `; i3 J9 _* N    Status = FileHandle05->SetPosition(FileHandle05, 0);       
    if (EFI_ERROR(Status)) {
6 C  D1 O# o, \% w, _. \        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
  i, W/ @6 f& q" s+ k5 ~! q        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
# T7 C! L' j" D* {4 i        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
& q' T% b3 K9 s( n7 c    }       

    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);
0 m  x4 B$ X) s        return EFI_SUCCESS;
5 ?$ r) l2 ?) T/ O# j* }/ z    }
& J/ y+ ]3 c3 ~7 g       
' s3 R9 q+ O9 C$ L    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
& y! b; Q6 \3 x+ L        Print (L"File Buffersize is %x\n\n", BufferSize);
: P. p1 q  ~# j: Z* Y! [    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
$ P4 y8 D0 @' T% c5 d. G$ H5 k        return EFI_SUCCESS;
    }               
3 E6 I) \) G+ _0 t5 ^       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
& z, ~/ b* f- G5 L    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
- U. M* z8 s' P1 M' m       
    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;
    }       

# v4 r" V. l/ n( t% y4 }//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
% l( P. D# z+ V! n    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
' `# J4 {+ d9 A4 B2 f8 M/ a. ^        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
5 W2 \! a6 v: J6 v! r* p% K    }

8 n2 ~5 Y/ m/ u; g% D8 n+ [    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
, D, `$ N' b- `. {0 A! c& b% B    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
* H$ f7 Z8 W1 G% Y# m2 E7 ^        return EFI_SUCCESS;
4 y) O) A% g. y+ [' P# ~) y, V    }       
        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
5 Q2 r5 ^$ s$ z& Z//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

    Status = FileHandle05->SetPosition(FileHandle05, 10);       
    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);
2 u$ D4 K$ w# }/ R: ~, ^* z( m- }    if (EFI_ERROR(Status)) {
3 x1 p3 A' |- ~" h( A, Z& x  l, U        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
) y/ H9 g/ k) @: t4 Y; ^2 \: e
4 y4 p& ]8 Q7 ~6 w# n  p    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);
) o& o; a: L$ x+ K" S        return EFI_SUCCESS;
    }
       
+ i* `* a) O2 a2 H1 s: e" Z    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
  e! J( e9 X. h# `9 X3 l: m        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
0 z& t- ~- \8 L( |8 i7 o5 Z! H        return EFI_SUCCESS;
+ b, C6 D/ e" P- Y) r6 a2 ^    }

6 j' f% D, X' ]+ d" _  N$ X    Status = FileHandle05->SetPosition(FileHandle05, 14);       
    if (EFI_ERROR(Status)) {
3 u1 y/ B( h) q9 K+ t& F        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
: _3 o: Q3 ]/ [# ~' z        return EFI_SUCCESS;
    }
( ^% \" _" `6 F% ]4 x       
' k  S8 _# Y! J& G  _5 N    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
# A7 y  Z3 j# J& X# X3 }* v    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
' L5 g8 H7 b: X4 D
: O4 g& R7 U3 H8 ^7 M' K! _    Status = FileHandle05->SetPosition(FileHandle05, 16);       
    if (EFI_ERROR(Status)) {
9 P, X- }  v2 {5 l- X. N        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
7 E& Y. U: {2 k% h+ ^7 s+ U       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
" E) E" i  d8 I3 B8 d    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
. ^! w4 G+ j1 k- k3 K# I        return EFI_SUCCESS;
    }       
: L! {! M4 o5 s) z
( w( z0 U+ }- g3 T//---------------------------------------------------------------------------------------------------------------------------------------------------------------
4 w. d) |; c$ |' e
% J! T2 I1 {8 o5 m$ F    Status = FileHandle05->SetPosition(FileHandle05, 18);       
( B( X2 S( L8 i    if (EFI_ERROR(Status)) {
: I; H& B) `- O9 R        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
" p8 E, s0 [& s; q4 h1 x        return EFI_SUCCESS;
    }
5 t( N9 B/ r7 m  W  ]$ }       
" j) n% ]7 ?6 m& s$ v4 l3 K+ N    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
9 {1 {0 P1 s1 x5 X1 u( w    if (EFI_ERROR(Status)) {
  R8 R  i# \; ~3 s7 ?        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
  ^' @& J; J' j/ e" Y        return EFI_SUCCESS;
, _' k$ y4 ^4 L5 ?( D' {    }       
       
! o: k3 J) x8 a0 A0 n    FreePool(DestAddr);       
1 Z' S5 v: d0 L) D0 Y4 x$ E
. i1 W- X0 }/ b' d% `% g    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
) y# }- J/ N. s: V% s7 R- h$ |        return EFI_SUCCESS;
4 |, Z5 y* U2 @+ ^4 U    }
       
: V  r. ^+ |4 }+ L+ y    Print(L"File Name = %s has already been created.\n", FileName);
. P0 @3 C, L2 E
    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();
  I, V/ ^; ^6 Z
) s, F% G) R4 M1 [/ h8 Y+ V    return Status;
: X7 i1 z( J, g6 `1 L}

chenhy

//**********************************************************************
9 @2 f  k0 G7 F- Q: {) @' p8 T$ |//**********************************************************************
" O6 P/ V; J- X//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
//**********************************************************************
. w& ]- Y' B/ b" h
#include "efi.h"
#include "efilib.h"

#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001

static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
   
    for(;;) {
* [0 ]7 G( b3 o6 z$ ]. H6 Y- \: o        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
* M) e" s) u: }. q2 n9 D            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
    }
: b* d) g- h' ^( M! e2 c    if (!EFI_ERROR(Status)) {
2 D! G; k$ b/ S5 K: W        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
# o1 U) K' Y- y4 X2 \+ {" ^    }
; V  g% O* p5 z8 n   
    return Status;
}
0 n- E+ s' n/ \0 X
9 @. M% u/ |3 m7 HUINTN
* E, R6 {, |) y- c# [AccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
{
        UINT8  i;
& f$ R9 a2 w8 V/ a5 r; g5 t        UINTN  PCI_ATTR[4];
8 \* k+ N3 F, E8 @+ }' W5 L        UINT8  PCI_REG[4], pci_offset=0;
# N: S: ?9 [1 o0 ?) G
* v2 T5 p% s- h) e) B$ p3 C        //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]);
9 @  H% }& S" l$ Q, `          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
, r/ ?5 k4 o. Y6 y7 o& t# x          //print register value
& t" Z- k- e+ w          for (i = 0;i < 4;i++)
+ s8 ^. ]( s0 @) E          {
        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");          
1 `4 @1 ?: y6 L$ U& O          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
* J, r+ R3 t& z* t          return Found_Me;
+ h" t& L$ x# V, ]) o1 L    }

        return Not_Found_Me;
}
# P: r/ m2 a8 B. Q( k
EFI_STATUS
InitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
    )
{
  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  ^: T/ ]5 C# J. X, y' G$ f  UINTN                   BufferSize, Result;
, p% n0 P! x! |% S% ^  UINTN                   NoHandles, Index;
* x  \$ W" l/ ^. \. K$ V7 T  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
4 }8 f6 X: y$ b& w
    //
- u" k+ M' ?2 V; G$ f* _& k    // Initialize the Library.
$ q, Z1 O2 z% R2 @    //
    InitializeLib (ImageHandle, SystemTable);
    //
    // Find the PCI driver
. `/ y" T% h6 V% \# ^    //
    Handle = NULL;
3 M) a& e: ?6 [* a8 r3 F& x    BufferSize = 1;
    do
. T( i% U$ }  D    {
      if (Handle)
      {
        FreePool (Handle);
& Q& ~# w% k, O, ^2 @8 W- M      }
               
. u; R) k0 C1 s" Q, D, z' p  V) {      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
. Y1 S3 L+ a+ f, {  r( c
% W5 f7 K& Z2 }3 b# ~9 G9 K" ]% {+ x    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));

" `  l8 A( M  t. ]; r    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
, K" p6 z3 U7 ]: [    ST->ConOut->ClearScreen (ST->ConOut);
) i5 R* p6 k' R' x- e    Result = Not_Found_Me;       
4 w; N) O" T3 n7 t! b7 M2 \# Q) E    do
    {       
3 w& A$ f) e2 {6 e( y7 w2 O        if(Index == NoHandles - 1)
" @) Q# f3 [  z         Index=0;
8 u6 l- Q1 z8 c, y' _( e  B/ E$ Z          else
* m" W, j8 q* H2 S! R$ \, h    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
0 j3 `) K$ ~# b' G          Result = AccessPciConfig(PCI_DEVICE_TARGET);
' T' F, H9 H7 t//-          Print(L"Again now Result is %x \n",Result);          
: A$ x$ M8 n% W7 p1 l7 N/ E    }
    Index++;               
0 _3 G  f0 c7 ?8 N2 y1 e$ J    } while(Result == Not_Found_Me);
- |5 A# W+ ^1 L) s: `6 M       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习