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

kaurohy

/*
* C$ K: z. H! ?! O * myfileio.c
* Apps
3 P" ?. S( h! p# n */

#include "efi.h"
% V% F1 i( x. d2 g5 R7 X#include "efilib.h"
9 R' R, b8 b1 J# G7 {6 m
! ^4 T9 I. f( H. w3 j5 m( ^#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
$ Y8 @: a2 F9 U1 l7 M) ?6 u    EFI_INPUT_KEY       key;
    UINTN               index;
* V; k. k& E5 z3 `9 b   
    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
+ [& Z- w; P0 g9 D! a        else
7 C: V  ~- U1 }: j( e0 U$ C% u- }            break;
    }
    if (!EFI_ERROR(Status)) {
9 Q7 o5 o  W) z! K* R! ~/ t2 y        if (key.ScanCode == SCAN_ESC)
5 m/ z* g9 Z# u! c. x/ B- I            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
0 W- X( T/ x2 z$ X( {   
3 X. M' g+ R, }    return Status;
}

EFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
{
3 `1 `, R. h: m4 ]  ~. N: q    EFI_STATUS                Status;
4 L) O0 }3 `5 C& a" ]4 x    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
# _0 {% ?4 w( i0 W2 a% E/ f    EFI_FILE                *FileHandle04;
. C7 C, g5 v, r  `7 k    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       

    FileName = L"NewFile.txt";
8 ~; M3 ^& \( @, h7 S* e        Space_Key = L" ";
       
" O. f$ C1 k" _$ [        BufferA = L"ABCD";
- n  Q/ F! `. C( D1 S2 T# w# l" R        BufferB = L"EFGH";
        BufferC = L"IJKL";
4 G8 m$ a+ _) b: C5 W' P' C( B
/ x: L, s/ V$ h# [3 ]7 l) x' I        Bit0 = 0xff;
7 I5 z# u3 B3 b" L! k( g        Bit1 = 0xfe;
       
8 J0 b) {2 d9 ^3 d* y) v! x        Key_Space = 0x0020;
       
4 M7 l1 X- ^9 W/ n0 G, F! t% I- D3 {    InitializeLib (ImageHandle, SystemTable);       
0 q( g  J8 \4 e8 V' |: ?  z
    DestAddr = AllocatePool (BufferSize);       
( W% l; v9 D5 H8 ]0 j! d+ q
) ^( n/ `5 z/ P    Print (L"Value of Bit0 is %x\n", Bit0);
0 O* j% |4 u, F' W    Print (L"Value of Bit1 is %x\n", Bit1);       

       
$ ], @& K% ]- j( A    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
0 ~: I& q$ P* c% f( v        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
9 h$ U  |, g" q! _3 v        return EFI_LOAD_ERROR;
    }

- q+ R$ f0 z: t) u    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;
6 I& ^! G* Q, W/ E/ T8 l8 |$ F3 v        }       
       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
0 I4 D" o2 L/ E' q3 A; J  ~7 u    if (EFI_ERROR(Status)) {
8 ~" k5 x& X7 L; v* i' m( a            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
2 @$ O& e% _% n/ N+ I" V            return EFI_LOAD_ERROR;
6 N7 [& P9 B( r) T. x+ l        }
8 q8 U# q: x' {4 e% N; |
    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
+ i7 n3 g, r$ C" r* n2 o# Z        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
8 P7 ^$ V# r; m0 ^; r' u+ S0 f: ?        return EFI_LOAD_ERROR;
    }               
3 N' H4 n% Y0 m+ b               
    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
    if (EFI_ERROR(Status)) {
# B0 }" k' z% }, V        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
6 m# V; ?) V9 \8 u    }
       
% _3 n! K( h$ C    Status = FileHandle05->SetPosition(FileHandle05, 0);       
    if (EFI_ERROR(Status)) {
# {" N; [  `9 K6 O# N8 M/ X# P8 x: ]        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
! D' A5 v( V2 W* h8 Z4 ?        return EFI_SUCCESS;
    }
       
2 s+ l! f( m; k$ q. b3 j    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
- B1 x7 G9 E  @        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
3 E# }* Z- m+ m6 L/ q8 Z4 B& ^6 ?% E        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       

# k) E) m8 H; @' O8 O! |" N$ t/ C2 g    Status = FileHandle05->SetPosition(FileHandle05, 1);       
    if (EFI_ERROR(Status)) {
! w& @/ u. p4 _# O' V7 c        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
  b8 v# l) o4 Z3 o  h, s6 r    }
: ^  n0 Q7 b4 N0 S. N' Y       
" C) F0 ^& R% J* P! a" s, d    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
' A9 m0 d4 {4 d) ?        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;
    }               
       
4 o2 q. S  J  N+ ]# X) u    Status = FileHandle05->SetPosition(FileHandle05, 2);       
! X; p( z3 |, W! g6 e  [" q5 J, e    if (EFI_ERROR(Status)) {
4 ~2 ]2 ?* g7 j- Z6 P( G; Z        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
" r; y, n8 s# f    }
! u  Y0 w. J$ E1 d, m       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
! u' r- f$ a5 z, l- O        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
7 `. O# R8 X1 f2 d7 f) g5 c        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
/ h) p5 s3 z* _$ u; N: ^3 S        return EFI_SUCCESS;
# k8 G  A4 X* d3 K/ L0 m9 C    }       
7 p, M( s* V3 Z8 z4 x1 |. E
$ [3 F9 b: a: @. u5 I8 s6 t//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    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);
9 w5 D' o! I8 f        return EFI_SUCCESS;
+ E/ x. m( t9 Q3 W) v! n    }

    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
    if (EFI_ERROR(Status)) {
$ e. d$ t; z: P3 D# ]& d        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
3 e, X% p3 U) V' }! v  M    }       
( c) k9 d7 N% M  w  g0 @2 q& {        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
. v" Y% C7 N- M; J' ?% I5 d& z
    Status = FileHandle05->SetPosition(FileHandle05, 10);       
0 T( ]) I. L1 ]- p    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
: G  p' ~. ]/ S' {2 b    }
# n# [. S# a% {6 j: M. K' J       
* r0 X- {* T- z# A2 \" U) S, {: v    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;
1 O  S( p3 S0 m6 C/ k* g    }

- m3 P  t: M  {: R, ?# o    Status = FileHandle05->SetPosition(FileHandle05, 12);       
6 d# W5 X; R% L& V2 y* @    if (EFI_ERROR(Status)) {
8 e; N# s/ R2 M# _        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
+ Z3 n( y: ^" A0 O/ G! ~    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
4 b& E: Z* c; _" @# j! G0 Y+ c    if (EFI_ERROR(Status)) {
- n, X, y5 O1 ?9 z7 n" ?        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
% \' X. M  l. A" d# Z        return EFI_SUCCESS;
    }

8 ~& `& E* Z# g/ @- ?6 I# b+ G+ J    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;
8 s& C, \. y) g7 ~; J% L    }
       
% Y/ u0 X# ~6 k  \8 e/ X5 e    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
( t, [$ d. I* C! m3 O6 \% W. ?2 C    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
, y; I5 @+ f# T' ~0 x        return EFI_SUCCESS;
6 I' h; T# N+ q2 l! S$ {  z1 q    }
; A7 k( k/ P: ?; |8 b2 c9 E; D$ O
    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);
3 O8 a  F+ t+ b* X% S# w$ p7 P- n, V        return EFI_SUCCESS;
6 X) r' s7 M; L  X    }
$ W/ v6 i6 T  \. m1 r- G       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
) x; z8 d* I% o& _. C    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
: C' y0 x, C' m6 u4 k    }       

; Q% d. ?2 F  A5 P. P- h//---------------------------------------------------------------------------------------------------------------------------------------------------------------

0 W% K, s2 v! ~    Status = FileHandle05->SetPosition(FileHandle05, 18);       
6 y4 \. S+ w+ U; c    if (EFI_ERROR(Status)) {
' D" M7 S) r; ~  |        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
1 E" ^" g) _- h  |    }
       
. T  F' G; {2 l    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
6 w8 _3 }7 J: `. K4 g    if (EFI_ERROR(Status)) {
- l9 d8 h5 a$ j. _        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
% P) [# Q' w' O0 q. u        return EFI_SUCCESS;
    }       
1 a0 I' P; l) s4 c6 G; `       
    FreePool(DestAddr);       

    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
$ H. ?5 }1 Q: d  |6 l% q3 @( s+ ^7 w3 q        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
; ]/ R6 u" C9 @; N- z: X        return EFI_SUCCESS;
    }
9 L- C7 @' x5 `4 H- d* R: y0 J       
    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");
( [* q8 x! ^1 [3 }: w    WaitForKeyOrReset();

/ ~( s3 W; y' l5 i    return Status;
}

chenhy

//**********************************************************************
! j3 k* R' o1 i7 ~* v- t) c8 A# Q//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
//**********************************************************************
) o- ]7 n) j3 V$ Q& Z( s+ c  T
#include "efi.h"
" S8 R( S: O5 K+ |. t#include "efilib.h"
+ K4 z' n9 ~- J/ g/ G
2 n$ l9 D2 x+ b; X& @- ^#define        Not_Found_Me        0x00000000
" ]6 T  C+ K6 O5 A- N5 O#define        Found_Me        0x00000001
1 J  d: F6 q2 Z% C) Q1 X
# w* I: h: u1 b+ H1 I& Hstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
: W$ g4 f  e' {9 W' V7 }, {1 e    EFI_INPUT_KEY       key;
    UINTN               index;
' A! F8 R2 A9 w3 i   
! ^! T; M" E( P% ?" c    for(;;) {
6 E2 g- n% T" p7 G% w        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
# K* c# ?" j4 G! q2 Z3 b& B        if (Status == EFI_NOT_READY)
/ W/ Y, O' v5 B1 K  R            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
/ B% N+ V7 ?3 z; ?6 m0 [0 B: Q- m/ b        else
1 S, B0 g8 T- k& ?3 @1 T" L            break;
+ b. a0 ~" `- v# _    }
    if (!EFI_ERROR(Status)) {
' _% S' z+ J# [2 i' `& Z        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
# B6 o, d3 {6 n3 k* R# I    }
   
    return Status;
5 p7 X! c" W, |2 V9 I' k. J}

2 E* N3 e/ q- [. C+ a; Z$ yUINTN
AccessPciConfig (
2 k6 u& ~+ Y7 ]- L    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
0 {9 n- H9 E- B$ Q2 j3 p6 g{
: l0 D+ [' X9 u+ h        UINT8  i;
; `* ]" k! O9 P% R1 Z        UINTN  PCI_ATTR[4];
0 u3 r, b7 p( B; A        UINT8  PCI_REG[4], pci_offset=0;

        //get device attr
        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
: W" p# }5 i$ N+ W! ]+ @: b    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]);
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
7 s& t$ i: y7 k* b! B7 Y9 P. @          //print register value
$ W# [& C$ D" b- w" ~4 i  s          for (i = 0;i < 4;i++)
          {
) ]* I6 q$ J% _8 i2 s        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
( g1 w" ]# [( G% \. @; J) R                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
          }
          Print(L"\n");          
$ ^- z) j% S# V. O* @          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
    }

4 b; K. Z  g" Z6 p7 N        return Not_Found_Me;
}
& ~9 P5 }  l3 c$ A% C5 s
7 g- e6 v3 M9 d# x' R3 w/ gEFI_STATUS
: B& E- f& [* G+ `  OInitializePciApplication (
6 `- f) E! Q) `9 h& M: k% G" ^    IN EFI_HANDLE           ImageHandle,
* j- e) O+ ?1 V2 _& H! W    IN EFI_SYSTEM_TABLE     *SystemTable
    )
{
; T8 V4 C- {: V  EFI_STATUS              Status;
& Y( ^5 W( C$ R  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
& `7 s7 p( {, n) Z) v& S  UINTN                   NoHandles, Index;
! n' m, B) t" G. j5 S5 R9 J  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

    //
    // Initialize the Library.
    //
+ \6 f1 ^- z+ f! D7 c    InitializeLib (ImageHandle, SystemTable);
0 P9 [0 m3 o3 w6 ~/ O( g6 f    //
" p7 g* G$ t4 H1 @* B1 _    // Find the PCI driver
: M6 {% l3 B: }8 K7 N    //
9 E7 u* f3 E( h& M$ i    Handle = NULL;
    BufferSize = 1;
# i. {5 h  d% R' \    do
    {
' \0 F5 R4 H2 w" V8 q- ^. j; |      if (Handle)
      {
        FreePool (Handle);
/ S/ G& ~6 b7 H- ]/ q! o" Q! q      }
& u8 k6 z& w7 T' q  U0 O$ Y               
      Handle = AllocatePool (BufferSize);
, J, O  H. d! \4 f      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

2 v$ \" V3 U) ?, E; k/ A$ r* z& c    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));
' l; r8 r, [) o0 K5 P+ I5 I7 n
- N7 ^, }- j6 o- K0 c    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
- j1 m. r4 A- O# s4 n& Z# k    Result = Not_Found_Me;       
    do
    {       
        if(Index == NoHandles - 1)
         Index=0;
          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
" ~. u9 R5 }5 E# \: [+ t+ J        Print(L"THIS IS DEVICE %d \n",Index);
3 S) Z) j' ~9 \    if (!EFI_ERROR(Status))
    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
5 I# @$ o- p0 I% }//-          Print(L"Again now Result is %x \n",Result);          
    }
    Index++;               
& t* J9 W2 z7 N( l8 G$ |9 @, Q    } while(Result == Not_Found_Me);
       
( F, K$ P5 N7 C+ s+ n' Y( F        Print(L"Please hit any key to exit this image.");       
0 V! Q% m; ]( ?: v( q    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习