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

kaurohy

/*
* myfileio.c
* Apps
*/

#include "efi.h"
#include "efilib.h"
# n! ^$ }; x8 m! u7 ?
3 g& y% j1 y- X& ?! M# W( {#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
- u; }4 a$ p8 S( |0 s4 Z' \
) @5 n2 y# m9 g! |  Sstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
1 [9 E/ E3 y. I8 u( C    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
) K6 D4 T, V# Z4 l: Z0 [    UINTN               index;
   
    for(;;) {
% o( a- ]" X; d( v9 x* u2 i        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
( n! g) F" h" `4 n        if (Status == EFI_NOT_READY)
6 f, `, Q* ]' Y/ K; m% M            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
    }
" ?3 _3 [! r' [- ^; I" s5 m$ h# O    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
7 N- J- J2 V! i1 r* F6 j+ h% w; f    }
   
& T' e3 [& K! z0 b    return Status;
}

; ?, u9 t4 g# Q& z. mEFI_STATUS
/ B* X8 Y, h$ _0 k+ \* ?: aEFIAPI
$ U* s! m. j: \& N% b& ?" `  g+ hMyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
) W( N+ G) E; @1 r- f0 Y* |{
    EFI_STATUS                Status;
, y+ t% X! L6 ], X1 w7 a7 f    EFI_HANDLE                *DestAddr;       
) s# P! w4 z) }2 ~0 p+ l    EFI_LOADED_IMAGE        *FileHandle01;
$ G, H+ P% F' l    EFI_DEVICE_PATH        *FileHandle02;       
8 I! ~& j# Z* Q( E2 [% R/ M0 X" I' i" ^    EFI_FILE_IO_INTERFACE        *FileHandle03;
& m! W& W; [8 c0 _& b) B% P    EFI_FILE                *FileHandle04;
2 U9 Y9 s) J- Z; |1 ?! v/ P  H7 P    EFI_FILE                *FileHandle05;
: v7 c/ H! }, B7 i" v    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
. }% g/ i$ B9 M! Y* l  p- e    int         Bit0, Bit1, Key_Space;       

! ?- W  d0 f: C' G! \    FileName = L"NewFile.txt";
0 S! t! w- O: h3 C        Space_Key = L" ";
; {3 y8 g5 V1 V* Z, N/ q       
        BufferA = L"ABCD";
+ |" _, Z, c: `; @; b) {, ~9 r        BufferB = L"EFGH";
+ G8 v+ N1 ]9 K  s7 m* r7 V        BufferC = L"IJKL";

4 t. {$ P8 _1 f% ~  L! ]" r! `        Bit0 = 0xff;
/ I3 w! D% K5 _# i$ \3 _- `' E2 E' v        Bit1 = 0xfe;
       
        Key_Space = 0x0020;
: |# k+ K2 f% }       
    InitializeLib (ImageHandle, SystemTable);       
3 e2 U3 W" I$ @2 L) F
    DestAddr = AllocatePool (BufferSize);       
0 L) T3 \/ q' e# y6 c8 \
. P% G) q0 ~8 a; _6 W) O, A    Print (L"Value of Bit0 is %x\n", Bit0);
" ?! b: i8 B# T. T! ?8 M* h' [( ^    Print (L"Value of Bit1 is %x\n", Bit1);       

7 @8 Z$ n1 z1 B: w0 u       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
3 V4 D: x1 M' Z1 [  y    if (EFI_ERROR(Status)) {
" }0 t/ ~) L8 e" p/ R4 g        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
0 v7 ?7 Y- n  n# `        return EFI_LOAD_ERROR;
, F9 o7 F: ?1 a6 Q* |! k9 `9 a* U6 J    }
6 q9 E5 w" l5 W1 ]0 O
0 A) ]. w1 C2 y2 }. x$ ~8 b    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
6 E, B0 D" X$ K    if (EFI_ERROR(Status)) {
3 q, `/ ~. @1 h( _( I, p            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
9 q, B/ m3 C+ V            return EFI_LOAD_ERROR;
        }       
       
. D; F9 Q2 U  t# {8 g- R6 ]+ q    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
+ C: Z! S, Y; L' E9 Z: X. |    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
7 Y  c- ]0 |' j8 V! J        }
% t. p* h) T, u% U
    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
" g' K6 v& g4 U; N7 F  b# N: h" J        return EFI_LOAD_ERROR;
/ \% v7 n6 \) L    }               
7 x3 G9 r) V% n  D" D% r               
    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
    if (EFI_ERROR(Status)) {
4 Z5 x% o: O8 `. [( |9 x+ A" ^        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
3 ^7 L7 K3 E! M  ^  W        return EFI_LOAD_ERROR;
    }
6 N" N! H, k1 w1 [       
& C3 K) k5 \: L7 @( p5 t    Status = FileHandle05->SetPosition(FileHandle05, 0);       
2 w9 l# G& @: a2 ^0 W    if (EFI_ERROR(Status)) {
0 X: {' o; ~, G7 f0 H9 a7 c0 i& h        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
- v2 A; A  {3 @    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
+ ?2 ?% y% k/ U# x    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, 1);       
    if (EFI_ERROR(Status)) {
  O& D4 f& ]5 `5 x( f* m8 W. j( V        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
0 l. E4 B1 x8 s( J        return EFI_SUCCESS;
# I. S1 [& {; b" l    }
; h# [+ y! w3 B5 x" c( E( B       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
8 b. Y4 ?, b3 ?8 a    if (EFI_ERROR(Status)) {
2 s1 y, T" {; z. `' x/ ?& R7 p5 D! M        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
" t* ?1 v2 }  i, i3 q1 M    }               
       
8 j6 G5 C& m% X    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
) d3 |9 l6 _  Z0 F! A8 r0 J        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
$ h5 e$ N) X' D       
# A' s, u0 j0 ^" S! P9 D    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
* _5 O7 C( ^0 R        Print (L"File Buffersize is %x\n\n", BufferSize);
" G  H; b" `" m1 U    if (EFI_ERROR(Status)) {
* T: c' p, x8 I( d: B        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
# n; V' S+ v! G. S- J! e$ t- l    }       
1 Y% k% V& N' H/ S: ^! `' L6 M
//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
9 c4 c' x. ?& N3 u    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
! k) F* q5 _0 \) X( w9 ?- o    if (EFI_ERROR(Status)) {
$ A; T! p+ f! L( |+ F# `        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
- ^* d2 W8 ~+ {7 ~        return EFI_SUCCESS;
. k% L8 [0 p" V! a: Q    }

1 X! r. i$ t% a2 h3 B7 w9 V- w    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
5 b1 F1 G/ e+ E$ M    if (EFI_ERROR(Status)) {
0 ?/ S; b. A. d- V1 @$ ^2 B  t        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
( @) s0 b* z. Z5 x        return EFI_SUCCESS;
' w, d& a& t& l; O9 c    }       
        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

+ Z2 Y5 ~* e. C' q& \    Status = FileHandle05->SetPosition(FileHandle05, 10);       
    if (EFI_ERROR(Status)) {
" A' g$ ^  d1 [& A        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
; W7 N, R& d  `. `7 O+ J  E        return EFI_SUCCESS;
    }
& x3 i" N; z& Q/ s5 w  o       
' z% A: h. N' v; F4 z% X# j5 {    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
9 n$ y# c5 n: r# T+ A! O; `! w    if (EFI_ERROR(Status)) {
$ p! z/ g( L& s; N9 i0 p+ {        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
; Y, E! Z9 N4 H        return EFI_SUCCESS;
" o  V. h$ U, u    }

, q. Q6 [) P& ]. |) k% M( B$ E    Status = FileHandle05->SetPosition(FileHandle05, 12);       
& H9 L7 m. s" T& W- C9 a    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
) ]% G3 W- ^/ n! ~        return EFI_SUCCESS;
* x% N. y) p. O1 |$ g    }
) P1 `) d9 _8 Y! w' ~8 x       
, l/ Z: m' a5 z4 [. s    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
$ G% [9 K5 t/ M! ^2 a8 @9 O" N        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
" o' D) v( p& e% Y; |% s    }

6 T0 s" ]' I/ `$ T; F% n    Status = FileHandle05->SetPosition(FileHandle05, 14);       
- W% E2 O. A7 B    if (EFI_ERROR(Status)) {
' b& O5 I- W% o9 T        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
) H% b7 g  @2 A  N9 b: L        return EFI_SUCCESS;
    }
& k% t6 ?% o- y       
! ~! S* y* T% }+ L0 I3 C    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
/ p4 Y4 \' w  g5 v, u" `        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
* Q2 N1 }; y* Y  c* p0 ^    }
" [  Z9 t$ _2 P7 f' v; L" f
    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);
" Q. z; D: m7 }2 ]$ ^4 {        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
4 ]% F( S! S- }6 N) q8 z1 q+ v- E0 _    if (EFI_ERROR(Status)) {
: S. L) S% m" J7 h/ T: P# y        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
; O5 R9 V2 M. I- m) D    }       

//---------------------------------------------------------------------------------------------------------------------------------------------------------------
8 [2 W+ H8 }1 [- p- J* c: d' ]
    Status = FileHandle05->SetPosition(FileHandle05, 18);       
4 [# @% d# U3 U6 m0 a2 F. }    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
" P& K+ N& K' N3 A  H    }
6 a8 Z# a" m. Y# ?+ @& }: `9 `       
! q+ z  h' _; p# [$ L" d    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
6 {9 q7 G  u: j9 j9 c    if (EFI_ERROR(Status)) {
% M0 V0 j/ p- T        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
       
5 b) i: {6 j* P- @9 [. D2 _8 T    FreePool(DestAddr);       
  g& b: B: U" L1 M2 V* D
    Status = FileHandle05->Close(FileHandle05);
7 z. J6 X4 r4 G8 M) B+ M    if (EFI_ERROR(Status)) {
3 a9 v" S3 m/ t4 p8 ?% H: k& I" ~        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
9 }6 T$ W+ M& H- E9 o        return EFI_SUCCESS;
) ^' j8 b  z0 k* i& t    }
       
7 Z% z: |  R7 n5 T8 e& h    Print(L"File Name = %s has already been created.\n", FileName);

9 J( }4 E* h- Q* A6 Q* W, E- |    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
2 G6 _7 u; }2 _) q* M. w    WaitForKeyOrReset();

    return Status;
  N2 S8 `# R. w# j7 }3 X- B( C}

chenhy

//**********************************************************************
//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
* S2 O# [$ `- s4 q7 @% r4 l" P//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
; U4 u  S3 n- ^3 s4 {//**********************************************************************

#include "efi.h"
- [. ~" c6 e. s2 i/ w) a; G$ B#include "efilib.h"
6 \, D0 W( c0 m; S: Z2 G3 q, c3 g. s
#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001

static EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
) Y0 s5 Y2 x# i    EFI_INPUT_KEY       key;
    UINTN               index;
! C. p1 V3 [* C% k# A+ K   
0 ]$ C' j2 G$ ?  ]0 S1 \/ F    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
3 [" M& {8 ?; R4 l- k* W8 J0 z            break;
    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
    return Status;
+ X/ i* L9 h9 |4 [  m+ j2 K$ h- S! r9 x}
0 n! T2 H6 }% l# g
5 `! S1 {" m' \8 |. ]6 p; U$ A& _UINTN
AccessPciConfig (
9 e; u9 s2 h0 H. ~1 z7 i    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
/ G0 n. I* n  Z2 R4 U9 t: I+ r- x    )
{
        UINT8  i;
7 d: ?  {5 l/ n0 v. ]        UINTN  PCI_ATTR[4];
& H7 U# }) }8 c' S: V        UINT8  PCI_REG[4], pci_offset=0;
9 z6 H# M* z' S; [8 e1 L  r# G, g, P1 ~/ z
3 j+ C& t" H6 N) P        //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]);
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
1 ?" h' L0 N* o5 m" K! z5 i/ r          //print register value
          for (i = 0;i < 4;i++)
( Z. Q+ T4 u; W1 k3 H2 L- X          {
        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");          
* ^7 S9 g; E8 X9 z4 ~          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
: t3 j, k6 Y% y9 X          return Found_Me;
# q0 T6 Y8 f' L/ ^3 q+ h    }

        return Not_Found_Me;
}

  t' I4 H: ?+ IEFI_STATUS
; K8 q& z9 g7 Y6 D9 }" p1 F$ zInitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
    )
{
0 `: @8 `/ t- i) D$ M  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  o, ^6 n$ _0 @) N9 \  UINTN                   BufferSize, Result;
2 X. d% {7 U5 P# w  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
7 w( I8 M: Z5 ]% D( U4 T3 [" L
& l8 |- S3 \; Q" H! V- K* l    //
0 \- u" y6 Z5 z: p& T; C- Y  m    // Initialize the Library.
# U( o& s; r% {" k% i    //
    InitializeLib (ImageHandle, SystemTable);
    //
0 m/ l& ]# X: d# G' }7 m3 H    // Find the PCI driver
    //
    Handle = NULL;
; k/ g6 s5 O* N" h    BufferSize = 1;
    do
* o; N. A: ^! P+ x* Q" a( I    {
      if (Handle)
      {
        FreePool (Handle);
      }
               
! R2 u1 y2 Y$ Y4 p. I      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
1 _$ T9 t. D. ~! X; M8 ?
    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));

# \, |7 p. {5 G0 X: P9 W, T    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
! [1 S9 Q" A+ z6 i) o3 s    Index = 0;
% S. J1 U' f2 l* l+ W    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
2 O/ _, E( `% j; o/ p+ i    do
1 h& v- {: c8 p- t$ ^$ b* a3 Z' I    {       
' d6 B/ d* N+ b+ M7 y        if(Index == NoHandles - 1)
3 F5 R3 ~0 E+ p7 V/ o9 |9 B: D         Index=0;
          else
4 h3 I/ t' z1 q) T    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
    }
    Index++;               
, V# A' i0 T$ o3 _* @6 }6 ]    } while(Result == Not_Found_Me);
2 {) e  q  ?# _% D       
9 L6 U2 C- C8 C9 W) `        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
    return EFI_SUCCESS;
: [" w% Z7 y) s1 g7 v}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习