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

kaurohy

/*
9 @4 i# v5 }  X! s' V * myfileio.c
& H1 `6 D# |! y5 G0 { * Apps
6 k  s8 v  k) T# V! ? */

- B- F! R5 F9 ]6 h7 e#include "efi.h"
* c) M+ N5 M* L9 I# }, J3 u2 r3 T#include "efilib.h"

#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

* v( w( K  c( R7 Kstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
; C- }! T1 y$ c4 z% m& f    EFI_INPUT_KEY       key;
, C6 b# u4 }- j    UINTN               index;
  [, {& N! z7 d7 o  M2 A   
/ m( l7 s# \8 Q8 p+ i; \- o    for(;;) {
( s. t- o; x% B% {        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
8 S, S) v' q9 w7 Z% N3 f+ ^        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
' \  J  z6 G# E& L        else
            break;
7 z$ _% o' M# m7 P1 E    }
! t7 b5 k5 e% m; O& R5 ^7 T- m    if (!EFI_ERROR(Status)) {
. m* d( x% A; q2 j, B        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
    return Status;
}
- }( }! U8 j* U$ ]( A6 T7 f+ v& q
, N7 P1 c8 B: A; SEFI_STATUS
8 i- N) x8 R- L' N' ~EFIAPI
3 h, I% H. z! h% t& AMyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
{
    EFI_STATUS                Status;
3 o) A& p. ~+ v3 [5 q* h" A1 v    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
+ k6 S! X9 V1 ~) f4 _5 C    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
5 b* o. v" v2 y    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
1 S/ E! m) K: C9 {. B9 Q1 W! V9 ]' ]    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
8 e, \0 w/ O1 N9 ^6 v; {" N, |5 \/ y. {    int         Bit0, Bit1, Key_Space;       

; _2 |4 U: i2 W  U; x* l    FileName = L"NewFile.txt";
, h) r; L+ f2 b! j+ U/ S* ]        Space_Key = L" ";
$ H. s. S5 i- `1 g# F       
        BufferA = L"ABCD";
        BufferB = L"EFGH";
4 L+ L; X" }7 j1 M' J        BufferC = L"IJKL";

        Bit0 = 0xff;
        Bit1 = 0xfe;
       
2 a% s, e! _- p' m        Key_Space = 0x0020;
& `4 O2 k4 R% T" `, Q0 \       
4 |& h! W* z# }5 B; R    InitializeLib (ImageHandle, SystemTable);       
- e3 f! D: s9 s; c9 [5 ~
    DestAddr = AllocatePool (BufferSize);       
  [" E1 l+ P) I: a  z4 z% V" w6 b
    Print (L"Value of Bit0 is %x\n", Bit0);
% F+ C% n( w. J: N' B5 j; n+ |    Print (L"Value of Bit1 is %x\n", Bit1);       
3 z- v7 W! n0 p* w  g. W% j0 i. a
. z. P; M' J1 E+ M5 [& I       
& M/ e) n8 ?: c$ |' q! r1 c7 X5 M    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
9 F) h* |- x( }/ r        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
- L, u5 ~' ~- T) t: i4 _* {    }
3 ]/ W% \; G3 t4 R
* J$ t5 ^9 t* ]; \    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
8 B9 }* r; G# H7 c5 z    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
        }       
9 W1 c4 h& }) J' {4 w/ S, k0 j       
    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);
            return EFI_LOAD_ERROR;
        }
! d) S; X3 H; Y5 Z0 {
) [* v( t; F; I  z, G1 ]    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
# {1 f! F! u) {0 P        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }               
               
    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
9 g' u- Z8 G2 f    if (EFI_ERROR(Status)) {
2 r* i5 J3 w8 e- K& U        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
& l5 b8 i/ j! H) l        return EFI_LOAD_ERROR;
    }
' n* \9 l' `' x& p3 j2 t       
    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);
6 R- o$ q! V& H7 N9 _2 ^1 Q        return EFI_SUCCESS;
    }
" }4 Q7 o$ k, f* g+ [       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
$ Z8 N3 Y4 R* O" a) b( j/ J5 H        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 z" N; O- Q5 I  v: d        return EFI_SUCCESS;
9 N4 `7 o3 e) F/ U, @2 r: x    }       
1 u/ D; l! f* C5 l  Q$ V/ E/ N
. d; I$ E! b  z4 F( ^% b    Status = FileHandle05->SetPosition(FileHandle05, 1);       
    if (EFI_ERROR(Status)) {
- e4 A. F0 X6 P- g9 m( U+ _        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
0 \, t) j* w! E+ W4 ^    }
9 o3 x$ C0 z% Y8 o: C  b1 o       
% g8 `9 X8 ^0 Z    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
$ I+ x" }$ S: \! [9 p( f        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
7 Y9 B3 p  ]9 x        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
5 X- B  _  g! j* o2 O% E0 J        return EFI_SUCCESS;
% z5 w( G$ X" s# u0 v5 Q9 ]) d3 m6 m4 L    }               
* o& z8 X- V. m* x       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
& L+ n/ `7 L6 b7 F8 O    if (EFI_ERROR(Status)) {
5 p" I- i) T' H2 L6 a/ ?        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
  I2 g+ ]8 S$ h       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
3 c0 j+ o4 n/ \        Print (L"File Buffersize is %x\n\n", BufferSize);
6 N4 ^1 r/ f' P  q    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
, U. _" W5 e4 D9 ?# e; `    }       

//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
& X# p, y  j: E; `  j5 G    Status = FileHandle05->SetPosition(FileHandle05, 2);       
. t; \. g* x* _8 p    if (EFI_ERROR(Status)) {
( B; @  s! S6 t        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)) {
        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
: e* y( R+ h& _: k5 }        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
% Z/ X8 q* ^: W  {. n" e//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
$ `6 g! u. {; u
8 R) n+ l" y4 Z" ]+ n6 ?2 g2 `+ W    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);
1 r/ O( ]* b: E! Z+ w/ F        return EFI_SUCCESS;
) a6 a/ _4 ~1 |3 f6 `: f# F    }
       
$ N# [5 U/ ^" z7 Y2 @( E    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
& _5 K$ R" l9 L, K    if (EFI_ERROR(Status)) {
( [7 u5 {8 i+ J6 q- s        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }

1 y: Z3 w. H# C3 J# w+ l    Status = FileHandle05->SetPosition(FileHandle05, 12);       
" e* V  T7 B6 H5 f: l    if (EFI_ERROR(Status)) {
+ t' u" C) p+ @2 `+ ^        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
9 [3 v$ ?& v7 \4 A# L    }
5 I. r. H$ \& T. c$ M6 i' }       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
, h* z9 X, w2 m* J5 ^    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
$ _# X* K) U$ }" p. K, M+ `        return EFI_SUCCESS;
    }
' X' z7 J. h$ f/ v$ i5 V  G
    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;
    }
, F5 P$ Y! c1 Y       
) j- Y9 s0 A+ Y* J/ y8 L0 t    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 y# A% G- T" g* Y9 U" [    }

    Status = FileHandle05->SetPosition(FileHandle05, 16);       
2 X4 S. L/ l  a3 Q) n    if (EFI_ERROR(Status)) {
' n/ E, |' D6 }, G" |: R' ^        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
* o: |1 f+ |' F        return EFI_SUCCESS;
    }
: F& O9 r) J2 `       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
- C) T# [3 O6 ?$ K/ Q3 {/ R& f    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
1 Z1 c6 s1 s1 R. `# d/ k! G) {" @. e        return EFI_SUCCESS;
1 I6 U  m4 y( Y$ j% U+ c    }       
2 n# c* A; n/ {& Z8 w# l
//---------------------------------------------------------------------------------------------------------------------------------------------------------------
# r0 R0 q1 ^5 {4 B
$ @8 x- [: B, R$ o1 V2 v    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
. |/ g& U( o# |# n9 V1 l, V        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
/ y( K( z- b  [: q9 B        return EFI_SUCCESS;
5 d% ~' Q( `5 K4 Q; a& w' l    }
8 D' ?' |" v8 ^       
5 _, M" }2 \- K    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
, p4 N3 h, V8 k- {, a* o! e        return EFI_SUCCESS;
" t. e5 l# T0 `. e$ s    }       
0 h$ p. t. @! k+ m5 u- G0 s       
$ |" \( X3 y8 y5 E6 T0 c    FreePool(DestAddr);       
& @; O8 u3 w- {, S
& l9 T7 \# X  s0 C0 o% u2 m4 @    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 @6 s9 u! w, P% R    }
       
    Print(L"File Name = %s has already been created.\n", FileName);
; G* |; Y: E, B; Q
    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
& l" [5 t0 Q4 A    WaitForKeyOrReset();
5 r" R/ W  `; n' w. a
    return Status;
}

chenhy

//**********************************************************************
//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
4 d( O& U0 S# k/ C9 z//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
//**********************************************************************
: Z1 S9 Z' R. v  u! ~8 ]9 F
" U3 \5 o7 T8 }5 B- a; O/ d#include "efi.h"
#include "efilib.h"
  d: B6 H7 Q! [8 |! X# ]6 l8 b
#define        Not_Found_Me        0x00000000
9 n$ n* h7 F. n# U9 k& N. b#define        Found_Me        0x00000001
& q( i  L' a  ]7 M6 l" ~
1 E5 G4 K1 c' H# `5 Q) q* zstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
8 E3 ^9 v* c0 N9 \, q6 z/ W    EFI_INPUT_KEY       key;
& b! T" H2 c, g8 O    UINTN               index;
6 q+ `8 _3 i2 b# F   
    for(;;) {
, ^# H0 l" a+ q        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
5 t* t, H9 s, b        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
7 j2 s7 I8 I/ m6 }( P, V            break;
' M; f+ i9 t6 G, y    }
. x" G- d- X$ s    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
: P/ e( B  ^" W+ F) X1 [            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
+ k- `. E) p4 |* `* w) D    }
0 {6 N" o+ M1 e( p$ C   
    return Status;
}

) ^6 O' w! c5 S6 n( ]- U$ yUINTN
# R/ f, }" ~, B6 M8 E) lAccessPciConfig (
6 ?4 y2 H3 _' ^$ c    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
; h; E2 {/ Z4 m8 u5 K; i    )
; c7 X4 `& u2 `$ ]{
        UINT8  i;
        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;
  |# u  A0 d! e% k: o( z& E" i
3 l9 `* L$ ?! M        //get device attr
5 v. M/ m; e! n        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
' {1 q7 R6 c) N    if ((PCI_ATTR[1] == 0x00) && (PCI_ATTR[2] == 0x1e) && (PCI_ATTR[3] == 0x00)){
0 v  _) l8 P2 U2 j8 t5 j' M- T          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));       
          //print register value
2 b# a1 ^+ s' O6 `, x* U, n) j          for (i = 0;i < 4;i++)
' [  f: E% N6 ^% Y4 _          {
6 P' Y7 P* M4 A; Z5 O3 d% r        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
  Q  t6 {) p& _- ^, _2 Y0 b                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
% G, Z2 W6 b, f6 Y8 `8 K/ U' w: M7 T1 ?          }
          Print(L"\n");          
  J6 @0 E' l" V. O( ]1 T& r# K# q  |          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
  b/ ^7 N5 ^* a" |1 @: Z$ Z          return Found_Me;
    }
. H. \4 V- i- `6 b* |$ C% S( |
        return Not_Found_Me;
) ]0 k% `4 C/ s# t}
: e0 p! O' I0 Z! M
; H% ]+ n: R: n( ^) {  KEFI_STATUS
InitializePciApplication (
: N0 P9 x& c& W6 ]+ H' B/ }    IN EFI_HANDLE           ImageHandle,
2 P' `# {* R# v9 L. r, [    IN EFI_SYSTEM_TABLE     *SystemTable
    )
; D) E  Z1 }- h{
+ c  y" h* ^- x( S9 F, r' F  EFI_STATUS              Status;
$ u) _2 r& ^- S  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
5 |1 o6 p' }# h# A. q8 K0 C
    //
    // Initialize the Library.
+ d& F  x6 w" U; d9 ~: f    //
8 i' s' M! `. W5 ]% J9 ^    InitializeLib (ImageHandle, SystemTable);
$ f" E/ D+ M$ [6 i4 E5 s    //
" C" P$ q8 h0 w    // Find the PCI driver
    //
    Handle = NULL;
- H- d( F# v% x; {; f0 {' [4 n    BufferSize = 1;
    do
    {
      if (Handle)
+ u5 ?2 [. l; p      {
        FreePool (Handle);
      }
$ T! n! c* y; A6 _               
0 T/ S# m2 s3 L9 f* Y0 G, i      Handle = AllocatePool (BufferSize);
3 g3 Q( s2 Z& B      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
5 Y& p1 I5 V3 P% g8 h( [1 g
    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));

2 ]' T  w; |& w$ O( ]    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
  f0 c  x+ ~& ^( W* T% r    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
: y1 w$ s" K5 w6 ^$ e    Result = Not_Found_Me;       
( l" }4 R) m6 c* B! m: `+ S    do
    {       
        if(Index == NoHandles - 1)
         Index=0;
          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
' R3 B' Q7 _4 a" p% {        Print(L"THIS IS DEVICE %d \n",Index);
3 n; G7 F1 b$ V6 R1 k) J    if (!EFI_ERROR(Status))
( F0 O7 r2 E9 |+ I    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
. X6 f" G3 A1 E//-          Print(L"Again now Result is %x \n",Result);          
6 r8 a" w% }, s. X    }
    Index++;               
    } while(Result == Not_Found_Me);
% u- _. t0 ~5 P7 m) g- v9 D7 s; c       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
7 A2 E; n2 w- ?6 l9 j$ Z5 B: M4 S( g% @               
    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习