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

kaurohy

/*
- r! G3 v$ a& u( ] * myfileio.c
* Apps
*/
; ]6 M1 Z6 t5 J+ q' B/ j( K
+ V: ~1 ^6 u" P/ ]% J+ N1 h#include "efi.h"
#include "efilib.h"
  j# D5 |& i4 J$ t
#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

; d& |0 Q- {2 @$ J/ C( kstatic EFI_STATUS WaitForKeyOrReset(VOID)
" t( i, v7 d3 b& t2 j9 p{
; B# k  W3 N6 S$ P& O    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
5 d5 a% }$ `3 z1 w3 _, @    UINTN               index;
   
% D6 e% K0 p' z' o; \    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
+ K5 e# d" p0 |1 e        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
6 N. C  {5 `8 n3 g& d! i        else
4 P3 Z. U6 Q" C$ ^+ Y4 x            break;
( q( T/ k% W5 o% N) t    }
    if (!EFI_ERROR(Status)) {
) C% @# a3 j' C4 o- X7 p' i* @# J        if (key.ScanCode == SCAN_ESC)
( }: m8 @3 J4 Q$ G            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
    return Status;
: d+ s+ k5 k9 Q6 t- x. p}
* v! r( m4 a3 C3 r- q
. u! c! \7 ^/ N+ n( \EFI_STATUS
5 P( w. W; }& G% P. f/ xEFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
, q% ]. S3 I9 i$ I% F) e& F{
' f. z: D+ K4 ~6 T# n6 s8 }    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
' O" \9 j/ T+ e+ m; F' h" y    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
& k  M2 y3 e$ b1 P    EFI_FILE_IO_INTERFACE        *FileHandle03;
3 Z) y. u6 D5 o    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
5 f& P! H% ]8 H; F        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
1 _+ W5 Y7 f9 @# }" R/ K; {& v    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
5 [! g) e& l% H. r. [/ n    int         Bit0, Bit1, Key_Space;       

    FileName = L"NewFile.txt";
        Space_Key = L" ";
) ?% r/ f9 |7 ]% b$ k( K0 I' w8 ~       
0 Z8 P# n; M  S  s        BufferA = L"ABCD";
        BufferB = L"EFGH";
  z0 E% J; a1 s' m5 N# d* F8 `        BufferC = L"IJKL";

        Bit0 = 0xff;
        Bit1 = 0xfe;
+ B/ g- D  F# C& v: Q       
7 Z4 b6 q0 j1 N( Q- S        Key_Space = 0x0020;
# `2 c2 K) O8 D8 n       
1 j6 F* b( i1 t; p1 N    InitializeLib (ImageHandle, SystemTable);       
2 @+ ?0 b2 N6 o, b: C' O9 j8 ]
    DestAddr = AllocatePool (BufferSize);       
0 T1 N' w/ a% `* B9 t( g; d. \
    Print (L"Value of Bit0 is %x\n", Bit0);
    Print (L"Value of Bit1 is %x\n", Bit1);       

       
) [& o7 `& @5 Y2 t    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
" M) r' Y# n) k        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
$ `0 B  K) N% D, j    }

    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
1 O9 Q+ p. |9 F' C    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
6 _  R. e- T7 Z            return EFI_LOAD_ERROR;
        }       
* \& W7 z! N" W. E       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
( m; b& Q( F$ \8 b  y8 r            return EFI_LOAD_ERROR;
0 g* y7 ~/ q! S5 I) |" n        }

* {- V! e# [: Y0 {+ f2 g    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
4 p" w9 L& p  P) z    if (EFI_ERROR(Status)) {
9 U" a3 N5 ]( ^( p        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
; T% H8 v7 e1 l6 U' u( s, n        return EFI_LOAD_ERROR;
    }               
  O, L& H$ G* {6 |$ \% ~: v               
    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
3 {; A7 D& |/ U% |    if (EFI_ERROR(Status)) {
) E% D5 Q6 g& F( p, Y  x6 C! b4 L        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
+ l- `5 t8 x# B' L5 h/ a    }
       
; X7 h# s  t: n$ v4 e    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);
        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
$ q# X& u+ q* ]) U    if (EFI_ERROR(Status)) {
! {& U0 \. M1 j, a. ^- n        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       
% p- |) H& y  e  s
    Status = FileHandle05->SetPosition(FileHandle05, 1);       
    if (EFI_ERROR(Status)) {
+ U% \$ ^( X  a/ m- h* \( t7 `* }        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
5 v# V6 m' I' S8 m% g) ]    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
7 o! Q9 l1 c/ \& R6 O    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
6 Y. ~- t/ x' k) M" q, q" t        return EFI_SUCCESS;
    }               
       
    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);
" n- y& x) O2 u  M3 V4 x        return EFI_SUCCESS;
% n7 n8 o) O2 w4 W4 W6 m    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
# A9 u5 H, @/ }5 |/ Z' D        Print (L"File Buffersize is %x\n\n", BufferSize);
7 w8 P+ |: ^7 D8 I7 |; }" Z4 f    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
1 ~  ~+ ~, A) X& i& M        return EFI_SUCCESS;
    }       

, h. k& S; S4 N& [8 i//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
! h6 }% F7 L9 s" A$ E& o; \    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
" I6 |( V* x/ A4 [$ u5 C    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->Read(FileHandle05, &BufferSize, BufferB);
    if (EFI_ERROR(Status)) {
( G% S. m) F! w1 r        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
9 f9 o4 ^4 R7 e  w        return EFI_SUCCESS;
    }       
! U& j: j( g! W8 a! D4 s9 Z# s        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

5 H  a# i4 @* o0 K, K    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);
5 C# r* X( W/ X' Q6 F        return EFI_SUCCESS;
    }
9 ?2 d9 n2 ?0 E2 K# \  |       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
# g! \8 `+ q6 w( h& M- p( G" }* u  g        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
. ]! E( @1 z; e- T7 \
& h& @1 d* ~! P9 W7 y/ X    Status = FileHandle05->SetPosition(FileHandle05, 12);       
4 m- R3 b" }. v. e% x: C    if (EFI_ERROR(Status)) {
5 t% F- G- ~! H: X: y2 i+ q        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
" G1 L1 R% W) Y9 o5 D" Y        return EFI_SUCCESS;
    }
       
; ^! C, ?9 T( P% }( g* v5 R8 P    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
0 n4 G! ~, H$ v5 S( u5 g: e% u( g    if (EFI_ERROR(Status)) {
3 L% |4 S$ P3 x7 i# ]        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
2 S* `" f$ c& Q. P    }

    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);
. U$ t( V6 S( A. S+ q& u. }  u        return EFI_SUCCESS;
    }
# `5 P& P) }" B! Q) n4 v' i6 ?$ b3 |       
, I5 l" I! r. u, I6 q/ k: T8 b    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
9 V( Z& P1 v* j( m    if (EFI_ERROR(Status)) {
6 l, g; ?+ [) o( r        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }

    Status = FileHandle05->SetPosition(FileHandle05, 16);       
7 }$ f4 U( q+ T! n* m' r: i4 u" S    if (EFI_ERROR(Status)) {
  R+ I* k2 u  F. ^        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
8 ]% ?* G3 |# f# b$ C+ w        return EFI_SUCCESS;
    }
- ?3 h0 Y/ f: G0 x       
    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;
5 P7 C9 N( X4 s8 b. l3 D    }       

) N$ L+ Z% I2 C  X//---------------------------------------------------------------------------------------------------------------------------------------------------------------
- ?; U$ c( K+ q$ ?$ K
    Status = FileHandle05->SetPosition(FileHandle05, 18);       
2 [8 A* Y/ O/ y2 a+ z( `* w$ K    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);
    if (EFI_ERROR(Status)) {
/ t  @) ~( L* z/ ~$ D        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
- k& s# s: u1 `! ]/ q+ Q. u" B3 h    }       
       
- O' m# e0 F. z0 x! T2 Y    FreePool(DestAddr);       
0 C# T3 {7 }  A
    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
0 F4 f: r8 V; \' r* R, D4 y) {, f    }
       
( |( {) i6 R& L) t( {* P    Print(L"File Name = %s has already been created.\n", FileName);
; i. j9 f$ T- v' d9 I# U
; \- L! w2 V. D: e2 s% F* m1 i    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
+ v5 t' n1 i5 m2 E! s    WaitForKeyOrReset();

    return Status;
" Z  l" J; \1 d' G1 Q}

chenhy

//**********************************************************************
//**********************************************************************
: c9 [+ N) z7 h) {//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
+ S# p+ A; h8 [+ J- k+ X- j, B9 i- I//**********************************************************************
//**********************************************************************
6 q# i, n2 `  ]2 V1 E5 E( b5 Y
( ~5 f+ ^7 K7 |0 T$ [" c8 c#include "efi.h"
#include "efilib.h"

) \1 N/ \: H7 n8 Y. j) c#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001

static EFI_STATUS WaitForKeyOrReset(VOID)
# T5 R9 q$ p8 j0 B; }{
1 l: a* r6 I; n$ b6 n* e: `    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
9 L% o" G8 p2 s2 |, W8 N    UINTN               index;
   
! u# |- c2 W2 z" n) O    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
. u4 z! k) l  x- r8 N        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
6 u+ H+ ?+ R: j! |( P* k5 P    }
   
/ s  J  E8 Q& x    return Status;
2 D8 |5 ^' K5 ]0 z% M}

; G5 K" d* n. \5 c# |* W! TUINTN
AccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
. ^' M6 q; w8 q; P. w8 p/ _    )
1 R0 y# W% N/ g( J, Q{
# u1 e. y$ N4 ^* G        UINT8  i;
2 `, U2 z. \! ?" F) j) f        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;

        //get device attr
4 @6 k+ o  j4 _# w- l2 [. e% F" B        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)){
; R5 N. F% m) ?2 }          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));       
. A2 B3 A& i) H4 i, O          //print register value
          for (i = 0;i < 4;i++)
          {
) E1 d) d3 `# b) f        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
: u  F" c4 d" L6 ?          }
          Print(L"\n");          
5 Q) L; X( T& g* W          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
    }

        return Not_Found_Me;
}

EFI_STATUS
InitializePciApplication (
' }. W5 O/ G$ G, A& X) o    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
: T1 _( s2 N7 \" Z0 U  ?    )
7 X: v( k# j+ L  A$ o{
  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
* ~' q0 x% y" `1 A0 |  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

    //
/ S) c  ~6 l: U% h/ G/ B4 W    // Initialize the Library.
    //
; }$ }6 M5 j) f. C+ m    InitializeLib (ImageHandle, SystemTable);
" |8 p% g; H1 T: X    //
    // Find the PCI driver
0 \% \  j0 E" r- A9 A' ~    //
8 j5 O1 Z' j2 ?$ d" N8 B    Handle = NULL;
7 b6 p* F3 S% u' @/ {8 ~" e; v    BufferSize = 1;
    do
' ^- ]# q2 o8 u2 |, b& x    {
. ^; K, Y% O3 W9 _4 m" X! c0 ~; w      if (Handle)
      {
8 [  ]  x$ \& g        FreePool (Handle);
      }
* g. B8 o4 N: s8 o7 k& |5 w1 K3 Z/ ?               
      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
; ^5 s% m+ K: d3 M! H# c9 J
' p$ a1 b4 {  t( j3 L, b: f    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));
! r' e* E$ l. s' B+ U  g+ [5 D
    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
5 w5 Y9 R; E- T* C    Index = 0;
- ^- U. P. l0 [    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
8 j; b6 e5 A4 b; w    do
    {       
        if(Index == NoHandles - 1)
9 m8 B3 j& P! w5 O0 B$ J8 Z         Index=0;
  J) R& T8 {. f$ o' u' w  [0 R; k          else
1 F% f4 d8 Y0 |$ [- j, g3 [    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
8 p9 U7 o$ _& i  F    if (!EFI_ERROR(Status))
    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
. z' F( L0 F1 j! `$ h//-          Print(L"Again now Result is %x \n",Result);          
9 v2 b( A! |: Q" ?& V    }
    Index++;               
    } while(Result == Not_Found_Me);
% f* O0 P4 y8 I( i       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
( L6 H4 ?1 o/ L. @7 n               
    return EFI_SUCCESS;
! B+ }4 P2 b* s' R9 R}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习