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

kaurohy

/*
* myfileio.c
* Apps
8 N; m0 M4 B7 w; H: W */

9 @$ {7 ^2 \( I. V: s% \5 ]: H#include "efi.h"
) X% b: v& \0 m#include "efilib.h"

#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

+ e: y) A% J* j2 h' t, K+ K. Mstatic EFI_STATUS WaitForKeyOrReset(VOID)
8 F8 X! O( q1 A- c; c7 N{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
1 K% Y. e6 V, g7 X% Q6 T    UINTN               index;
7 ], A! ?6 p# e3 [   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
/ H0 @. l' o! M' H; ^; L        if (Status == EFI_NOT_READY)
0 r/ U. q; ^! x. K3 j            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
    }
' U# y8 d7 ^4 D+ r. o; Z  s3 _    if (!EFI_ERROR(Status)) {
$ h5 Z" k: f$ X3 Y7 ^/ Z( v! _        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
    return Status;
# n7 m! [/ p* L& r, Y/ a+ M}
$ C: s1 Z4 a! U* p& G# M0 B
  N; e4 ]8 d8 l9 ]6 @0 v/ F3 eEFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
, s/ p$ N% n4 y             IN EFI_SYSTEM_TABLE     *SystemTable)
. R1 G7 q" V' {2 U' Y- H5 L{
( q- N: D# l+ b" m3 u3 i; n    EFI_STATUS                Status;
6 p) x/ ^, k3 o# @8 n- ~    EFI_HANDLE                *DestAddr;       
" I: `5 X) ^7 R9 x! t" C    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
* R- ]. ?) `1 x5 Q! T    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       
* b$ }" ]/ {' r
: |1 N! x5 z# `    FileName = L"NewFile.txt";
        Space_Key = L" ";
       
        BufferA = L"ABCD";
6 K' J: h+ A$ C* b8 w        BufferB = L"EFGH";
5 C* ?  Q6 Y5 c7 a  @3 T2 b) Y        BufferC = L"IJKL";
: P$ ?& r+ P# H( D! S
8 _/ ~, P* Y' i        Bit0 = 0xff;
/ A, d3 \* _! r        Bit1 = 0xfe;
1 w) \" D0 k& m8 [, p       
        Key_Space = 0x0020;
. b7 Y$ U1 W; w( \* h) v- B$ n       
    InitializeLib (ImageHandle, SystemTable);       

    DestAddr = AllocatePool (BufferSize);       

    Print (L"Value of Bit0 is %x\n", Bit0);
) z4 A  C/ P2 n( I8 t    Print (L"Value of Bit1 is %x\n", Bit1);       

       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
) \0 e& X1 ?5 R: \& w  H    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
4 z1 u  r9 @" d$ U$ M        return EFI_LOAD_ERROR;
    }
- ?/ ^  u. C/ O$ o6 Z
    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
    if (EFI_ERROR(Status)) {
( C, H' q6 F! W1 L" S' T            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
7 E" k* W/ h, b, z3 k" Y, D- `            return EFI_LOAD_ERROR;
" q# U, \! F8 N4 q        }       
0 I# G# ]7 H. l, R9 E       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
# p# \" S8 z4 M% J% h& B            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
7 C$ d- ?! w: U) r, e3 J! [" _  d2 e            return EFI_LOAD_ERROR;
6 w2 i0 l  B2 G( ], K        }

    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
) b; L8 k  l% y/ y    if (EFI_ERROR(Status)) {
$ H5 [8 N" F+ i% _( f1 A5 D( U        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
3 K. u. w! u$ L, M% j        return EFI_LOAD_ERROR;
    }               
               
    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
8 b" E8 @7 W+ S8 N    if (EFI_ERROR(Status)) {
        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
" M- k# X7 o1 p1 ]9 e+ E2 V) b        return EFI_LOAD_ERROR;
    }
       
/ s* z4 f% J& d6 V5 ^! `! Q    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);
1 \- J, U1 H" @        return EFI_SUCCESS;
    }
       
0 S+ j4 u# {8 s- q  H) s8 K    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);
1 Q2 D2 n/ K! O/ f$ t4 x        return EFI_SUCCESS;
    }       

    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;
. i) p. y& M8 T3 ~* K/ a) P    }
       
1 g$ P* g9 D6 p, D) t/ h8 y' g    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;
    }               
       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
7 r! U: `6 o% i, ?+ J- v% m        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
; O) y7 }% ?) |' @% [6 {    }
: f) {, P8 i, K" c) k1 r       
& W3 h+ i$ A; M' v    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
5 l* K0 J! i6 y  a        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;
, o- K  S3 k, ?9 O1 {    }       
0 k$ s( f# k3 r0 ?
//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
0 y! [2 w! \9 h: ?    if (EFI_ERROR(Status)) {
4 X% k4 V5 R7 J, w6 o$ d        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
: U+ v( V/ U0 u6 O! O        return EFI_SUCCESS;
# D2 j. P( Y* b; i2 h) v    }
% F; G/ L3 B& E+ Y6 T1 a
: U! K/ K2 a1 T( c$ V$ a    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
$ F) K9 e6 b+ j3 U    if (EFI_ERROR(Status)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
2 V8 ~) p& J; K/ @% a5 i" b% X        return EFI_SUCCESS;
; y! M3 p" a) L/ v    }       
; Y, Q) u6 i! s) N        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
& c) c9 E0 v7 O+ ^//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

    Status = FileHandle05->SetPosition(FileHandle05, 10);       
5 t+ B4 @& Q4 z. B* [8 T    if (EFI_ERROR(Status)) {
7 k( ^( l. w0 w: Z% @        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
* Y8 F: s6 K/ n8 |7 s( q7 N* {    }
% f" K& M5 {. o2 E       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
0 c! c8 C# e0 Y2 w        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
( @. }# b4 j3 U* A  q9 M" K! `' y    }
  Z: K& |% V$ e- t) o+ }
    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
0 P6 v& {8 W9 O" u3 t7 J        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
1 Q6 u6 p2 H* U, H5 F5 ^# ^2 y" n2 K! T        return EFI_SUCCESS;
  ^/ `4 S- b3 `: e    }
       
! _/ ]" f& y$ g% j, s1 h    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
5 Y8 S) h' A. j0 S" q    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
% [. w3 a' P/ O3 p! A! a    }

4 w, |2 {, e, Q" ]    Status = FileHandle05->SetPosition(FileHandle05, 14);       
- ?$ x# D! c( F' E    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
6 z4 Z" `7 S& v% b" y( o) G       
% G/ P2 Q% @! G3 F0 ^8 J    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
4 h; g4 g. B, @+ h2 E+ l        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
) ~/ q' c8 Z6 {9 f0 u% e        return EFI_SUCCESS;
    }
: j8 ^: q  V" H- V. v  E1 g
, A0 J1 i4 d6 D3 R+ y% N& h: f/ p    Status = FileHandle05->SetPosition(FileHandle05, 16);       
2 P" E8 _1 h, F. `' @& @8 Y    if (EFI_ERROR(Status)) {
2 t( w6 e- G, Y        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
( R6 y" _( m8 ~# ^' C! ?6 e       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
* k4 ?% m5 z6 D+ W8 K* _( A    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
- W/ n( Z3 M' i% i. G        return EFI_SUCCESS;
    }       

//---------------------------------------------------------------------------------------------------------------------------------------------------------------
* I6 k% ~$ Z  K0 W6 V9 a* R
    Status = FileHandle05->SetPosition(FileHandle05, 18);       
; W! n9 x, k# q8 O    if (EFI_ERROR(Status)) {
/ f3 y" z- R! r; x; j3 N! t        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
# v; d" T0 A! E; d! {' p    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
1 o: Q1 H' e" G) Y% |& v  J        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
+ ?6 e0 Q$ `; |8 Y# O( [) D: f% j    }       
       
4 @6 U) Z9 F' x/ I' x! \$ Z  `: r1 ~    FreePool(DestAddr);       
  N$ n' o; Q$ ?% Z8 o
    Status = FileHandle05->Close(FileHandle05);
9 ~9 z' I0 J4 H: V0 f3 `    if (EFI_ERROR(Status)) {
* Q' X& S7 D1 z: b- S, b. v        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
4 _4 k' x# o0 w: H+ P' c) Y       
( R7 V* m* X2 t* J1 [    Print(L"File Name = %s has already been created.\n", FileName);

  B( K% R: Q2 ~& N- K    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();

    return Status;
" n" Q( n+ d/ |; P8 x}

chenhy

//**********************************************************************
/ h# v, k: g- E( H//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
/ M% ~% `& m7 {5 l) a//**********************************************************************

#include "efi.h"
#include "efilib.h"
+ e( D; M! C( H3 h, b; ?
* d  p4 X" R5 B+ d& C- `#define        Not_Found_Me        0x00000000
3 n( J6 F( {1 ?$ W2 l- F#define        Found_Me        0x00000001
2 Y# I# d  |, v9 M
static EFI_STATUS WaitForKeyOrReset(VOID)
7 b1 E/ Z" E! s5 q{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
- V) x+ {! @# b7 ], b    UINTN               index;
- c5 i, H  F5 I   
    for(;;) {
0 x& O- T4 F; c        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
7 @4 `9 T$ K" ]8 S; a2 z        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
1 K, X- Z, P1 z* \        else
            break;
" `8 u. y, k: {( K+ ?    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
1 N/ U% d$ E; z            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
1 ~% \) B$ i0 B6 d    }
   
    return Status;
}
9 ^: T- ~6 p' k  O
/ \+ Q9 u5 g/ iUINTN
" ^$ w5 N6 F3 _, w8 j# W$ K) MAccessPciConfig (
" _; s* l3 }' c8 W$ L    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
. w9 t- G2 b) U- t- Q    )
{
        UINT8  i;
, A( x$ g; }1 b! l5 m( N6 j) ]% R( F        UINTN  PCI_ATTR[4];
$ d/ H6 ?+ ?& E) X! _+ J2 Y: y" Z. R        UINT8  PCI_REG[4], pci_offset=0;
/ b% q, Z: L7 ]# L
5 s$ M4 \5 ^# c$ l, t" O6 }        //get device attr
8 o! S/ g2 R: a  `. `7 P8 T        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)){
& d" I% _4 ~3 O" Y6 N2 c          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));       
9 f4 Z: s( ^( E          //print register value
* x1 P) N, n/ J* G          for (i = 0;i < 4;i++)
          {
        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
7 E9 W8 C  ^. Y, ^; ]6 ]                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
+ P/ s2 G& |# j4 X5 U) ?% p. Q$ Q          }
          Print(L"\n");          
1 d2 K4 r' ~; N, F! I4 c( _          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
; ~1 {8 D8 h* S' }6 d* Y0 {4 h          return Found_Me;
    }
$ n% j0 n- |( D5 r
        return Not_Found_Me;
}

3 J. \# M) E6 D. X* W& O$ tEFI_STATUS
InitializePciApplication (
- d$ l$ D+ l  j5 r    IN EFI_HANDLE           ImageHandle,
5 E) o( f5 ]- m6 R1 r. y, Y    IN EFI_SYSTEM_TABLE     *SystemTable
; N" z, y: N' K: l* }/ b$ u' F5 u    )
{
8 g# C+ k; i9 a9 M# F1 u  EFI_STATUS              Status;
: G! d% T; n; J- e- ~6 V  EFI_HANDLE              *Handle;
/ S$ b7 X5 j& O* g: O2 N5 J6 M3 `. f" @3 v  UINTN                   BufferSize, Result;
. o% E+ _3 }' e  w8 p: N9 I% s  UINTN                   NoHandles, Index;
6 ^* j, d& i% h7 W  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
) p" C( I4 c* u- f
    //
    // Initialize the Library.
5 L" o5 W8 x# ^    //
4 j5 v$ u7 H. N    InitializeLib (ImageHandle, SystemTable);
7 X/ j5 F, E0 I  N: I) c$ e2 C    //
    // Find the PCI driver
4 H4 ^' h# _& J+ P    //
% `$ J) B' |6 e$ y    Handle = NULL;
    BufferSize = 1;
, `( o% T& g# J( |" c    do
    {
, A2 u0 o" {1 E1 J9 M) P. w6 x      if (Handle)
$ a& h5 o6 S; i2 ^+ q+ k      {
        FreePool (Handle);
5 q/ L* l+ p" ~0 g      }
               
. T$ U7 W6 h5 n8 Z; u1 [* s      Handle = AllocatePool (BufferSize);
6 {2 ^7 G; p; Z$ ?5 `      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));

    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
: G0 h' W# A6 a( e$ T    Index = 0;
' e4 G* |4 w5 u8 G. S/ S    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
1 Z+ Y2 j# z5 q" e! ~    do
    {       
, y; R7 o# r- M        if(Index == NoHandles - 1)
         Index=0;
          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
( l  g) h. C. E! I$ x! V/ [    if (!EFI_ERROR(Status))
: [# U4 c5 {1 c( p    {
0 D; r( Z6 W% w3 b! G; t          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
    }
    Index++;               
    } while(Result == Not_Found_Me);
       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习