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

kaurohy

/*
, e( t, u9 D. O9 v7 T/ j1 Z' h * myfileio.c
* Apps
) g. q' X$ R* i$ ?- s) h, l */

#include "efi.h"
#include "efilib.h"
$ k) q& J6 t$ _5 @. r7 [, t7 h+ [' C+ @
2 F8 [! l% w) A#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
' v& i) e( }4 [5 R' C8 O5 H
1 L, M8 Y: |( F' xstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
6 ]% |4 a- R) k- M2 w   
    for(;;) {
' F( ^8 ]4 ~: x  C* m        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
0 X: w7 G" C9 f: ]8 v            break;
    }
    if (!EFI_ERROR(Status)) {
+ l/ _; c# n. U( v        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
    return Status;
}

EFI_STATUS
EFIAPI
; p! }* F& O! c+ e9 {8 w, c0 ?3 eMyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
* H' d) q( E" q( }) A2 ]/ G0 x' t{
    EFI_STATUS                Status;
% E& ]5 y4 w: c. N. K. x    EFI_HANDLE                *DestAddr;       
+ P5 J5 ~. D; P+ S; G& ]    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
$ h, s$ }7 T* u9 W( Z* ^5 E0 v+ d6 i    CHAR16                *FileName;
. L( I  l, y7 _% l, Q; n        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
* D) T+ b0 Z1 B* O' ^    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
1 _9 U* @" H9 q: `" r! a    int         Bit0, Bit1, Key_Space;       
- S/ o1 y( ^5 X. ~
    FileName = L"NewFile.txt";
        Space_Key = L" ";
/ D' ]$ }- f+ v% f5 W       
7 W, u$ u  ^4 c+ v        BufferA = L"ABCD";
3 n( {: ?* F1 n! U+ e/ ~4 e        BufferB = L"EFGH";
        BufferC = L"IJKL";

- F4 x7 m! ~9 V9 W        Bit0 = 0xff;
        Bit1 = 0xfe;
5 O1 q2 Y& O+ l5 g1 W6 e' V       
' Q% u: n! N7 G6 ~0 q        Key_Space = 0x0020;
       
, U" ^% D  j% r, Y2 A    InitializeLib (ImageHandle, SystemTable);       
2 X# D0 s) a, `1 X& H/ K; m
    DestAddr = AllocatePool (BufferSize);       

    Print (L"Value of Bit0 is %x\n", Bit0);
    Print (L"Value of Bit1 is %x\n", Bit1);       

- @- o( {6 u6 R6 U6 v! d       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
8 b. M. R0 t% }; V3 H, j    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
/ R6 g* W# L) k2 ^7 m* ]( c; }        return EFI_LOAD_ERROR;
/ R0 h! m$ i) L. E    }
+ I9 ~5 X  _* ]- N- w; F
. ]% P* t9 J4 v  c' b& M    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
: t# T, _7 P2 A( w& @) w    if (EFI_ERROR(Status)) {
4 G; E- A  `0 g. [# L            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
: z2 J/ h7 ?# a+ }            return EFI_LOAD_ERROR;
0 y0 V: I: D. [/ |1 D# b$ B* Z/ M        }       
% ]3 Q' i% h6 U5 g8 n       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
( |* s# F1 j+ a    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
! T0 j& ~8 W; x& a2 O! X# Q        }
+ s' X" Z7 t: U
    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
3 W5 ~; B  P$ r        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 i2 D2 c# F# [0 S& M7 m    if (EFI_ERROR(Status)) {
; f  b$ n( `1 ?, V        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
7 q$ D! c6 Z; ~5 v. q    }
       
    Status = FileHandle05->SetPosition(FileHandle05, 0);       
    if (EFI_ERROR(Status)) {
6 l6 t( s1 H5 L4 g        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
" U+ O/ c0 y+ J1 R$ g& [" q1 r    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
( b3 V1 a+ t, W3 i2 w$ S; U: z        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
& h3 D7 F- w% x2 |. f- T        return EFI_SUCCESS;
# B7 u# K6 g6 a- m( L' t    }       
7 k' u- O& e1 E5 ?+ @1 K
    Status = FileHandle05->SetPosition(FileHandle05, 1);       
* h; i3 z" E' Q# _4 n' s3 T    if (EFI_ERROR(Status)) {
: B2 K. B  M: A3 f3 @6 k) ?        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
& {) B, ~9 M5 ^5 f        return EFI_SUCCESS;
4 P% E* Z2 E* }" C, A! I    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
# T9 a6 t2 i: x: y5 p. C3 @        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
& E5 a( K* y; L: [+ V        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
' |; t) m- r- w! T* _        return EFI_SUCCESS;
/ z* M8 p  j0 t2 k    }               
1 T6 L& N8 n/ T1 E# }       
% u+ q3 v- P8 B2 X! q) l' \    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);
        return EFI_SUCCESS;
8 M0 \$ W  W' s( q    }
" ?  R, T! V: x. a" p) }/ L       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
/ o; v8 ~9 g6 H9 Z! {% l        Print (L"File Buffersize is %x\n\n", BufferSize);
. F& F  S) k1 |9 d+ J& E% D    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
) i% H2 j' p& y/ V        return EFI_SUCCESS;
) l0 e$ m+ q3 [* N5 z/ J. S    }       
, R# m  ?# C7 |( w
7 l  Y) p  \( u! Q7 {7 Y; {//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
3 E- b  f' u* u( u' ~* r2 C    if (EFI_ERROR(Status)) {
. C) L, s4 V) C! x2 \# M        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
4 i$ K* ?$ n- \9 M; I0 [3 ~' X        return EFI_SUCCESS;
    }
) J! ]6 ^. \: ^# C' D; V8 o
    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
! V) S: `' l. n) L- R+ t. i  P    if (EFI_ERROR(Status)) {
( k7 i/ K2 Z$ p1 t$ S. j        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
0 G0 H+ ]; s4 F  @7 G: y# u- p    }       
        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

    Status = FileHandle05->SetPosition(FileHandle05, 10);       
    if (EFI_ERROR(Status)) {
; B& l# B4 F( j6 p! ^9 r. H/ V        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
# ^/ h, o/ |9 H# i" e- r& f        return EFI_SUCCESS;
8 @4 ~- A8 S1 U    }
       
  X) T' ]* v; M% n( U6 s+ ]    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;
    }
0 W; k0 f. v7 H
    Status = FileHandle05->SetPosition(FileHandle05, 12);       
+ b8 h. U) B: d8 H; @4 h    if (EFI_ERROR(Status)) {
' U5 v. o* z4 [, ^0 Y; _        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
, p% Y' X4 Y& @6 R    }
3 J. c8 H, E8 `3 K* e       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
4 H+ l1 @" V: j1 k6 Q3 o        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
9 O- x5 m: H+ g6 N! J9 b    }
3 s' Z7 O6 n, S9 w/ [) g
    Status = FileHandle05->SetPosition(FileHandle05, 14);       
- x6 c1 w; m6 \* T    if (EFI_ERROR(Status)) {
7 `9 d' ~; i# h* R% P5 H6 x' j! `        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
! Q9 Q' \, M  h8 h+ Z( M3 }        return EFI_SUCCESS;
2 a+ _/ u$ @4 @, a: Z: s; d5 G3 x    }
$ p' h2 ]5 ^8 C+ g- N' q6 d       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
5 V% q  D" O6 Z+ p  I* o( O7 E( \        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
  f! E  C& C! d, W& i/ {1 |$ y        return EFI_SUCCESS;
' E8 h1 T% z3 Q$ h    }

# w' U) C6 B  ~  w4 }) v$ i: A    Status = FileHandle05->SetPosition(FileHandle05, 16);       
4 |, c- V8 z, x3 c    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
; p" H% L% [' l; Q3 s! q* y4 u8 Y        return EFI_SUCCESS;
    }
       
    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;
+ H8 P2 g8 R+ t' s    }       

//---------------------------------------------------------------------------------------------------------------------------------------------------------------
  T$ M+ k, h2 ]) g
5 x9 w8 G4 X! p, l    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
; _& y1 F3 S* i: r        return EFI_SUCCESS;
) z" ~) i& R9 I$ t; z$ n) @    }
       
; U9 Y: f. F2 t! [+ [! s, R2 g; o, p    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
# j2 U1 D( R5 ~8 u8 x    if (EFI_ERROR(Status)) {
; S/ Z8 i, p7 @" r" k- g  J        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
2 D8 L3 s/ N; ~8 f( }; O1 e        return EFI_SUCCESS;
9 |; y% o% Z& o, O2 p  V    }       
$ \0 d/ g. _+ W' E, P' B" j       
    FreePool(DestAddr);       

" [5 N5 z6 T' j4 x/ ]    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
( }3 i. [: O/ W        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
3 N  ]" i+ }+ m' c8 W    }
       
! [# N$ E; L/ f: p# y. I    Print(L"File Name = %s has already been created.\n", FileName);
( R, G& [! j$ W' i
    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();
5 ]+ X, ?+ v8 K4 i! J' h
: g' @4 r6 \/ J5 ], j    return Status;
}

chenhy

//**********************************************************************
//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
1 H% P9 ^# |( l% w//**********************************************************************
7 t$ z( g! G- R, M1 C7 z
; w3 G1 a+ n3 [5 j: S1 e#include "efi.h"
#include "efilib.h"
5 g) W* j9 ]! o) S' ?0 X' }
#define        Not_Found_Me        0x00000000
$ D. K8 k- ^+ @#define        Found_Me        0x00000001

static EFI_STATUS WaitForKeyOrReset(VOID)
5 x, K% B: I2 ~9 d/ Y, |{
    EFI_STATUS          Status;
6 t# e1 P- @7 [% t    EFI_INPUT_KEY       key;
( O: d# f# j1 P8 e+ r% a$ v    UINTN               index;
   
0 N8 S  T4 i, h8 v    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
2 S  q7 n7 M" A( D+ [$ |! a        if (Status == EFI_NOT_READY)
% a1 m  l0 k$ G+ n2 ~            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
  ]( ~3 F1 [: T# Q            break;
' a# c9 g+ T$ J2 m2 E0 _5 D, C" S    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
2 J; o& A" }6 u/ {4 J1 Y; J  V    }
   
    return Status;
! Q: F  a2 S2 `7 F" q}
2 e9 `1 D- x! A- |1 O# D1 s; M
UINTN
' S# E, m. P! o. a# u. `: hAccessPciConfig (
8 `7 T8 w) i: U! P1 j3 @    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
{
        UINT8  i;
1 }: p1 X- i- {1 q        UINTN  PCI_ATTR[4];
# z. V4 f6 H7 f$ _, C- e, l        UINT8  PCI_REG[4], pci_offset=0;

+ p; E+ N8 S" n/ t+ w        //get device attr
        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
4 M5 e9 H0 x/ o% L  `8 d    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));       
          //print register value
' S7 x* g# Z# [: H! {          for (i = 0;i < 4;i++)
          {
        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
          }
8 U. x" Z% M2 c  f. U          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
& T5 [6 _4 \. P6 @) ^  Q) C4 E    }
1 ^9 E5 e9 s% i* d
        return Not_Found_Me;
/ R& |$ s9 k( m1 O3 ]}

EFI_STATUS
* `- t: S2 {- Z9 V) D& nInitializePciApplication (
, A" }) N9 m, `3 ?! Y, T& s    IN EFI_HANDLE           ImageHandle,
7 _) o- m* u( h    IN EFI_SYSTEM_TABLE     *SystemTable
    )
% \7 L4 _9 m* ^. O: J{
# B* F& C( K/ s1 m  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
1 B, g4 \1 o; E; O  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

    //
" {; D: F8 S% ^    // Initialize the Library.
    //
    InitializeLib (ImageHandle, SystemTable);
    //
    // Find the PCI driver
6 \) t0 _( }. v8 b    //
    Handle = NULL;
    BufferSize = 1;
$ a* x& \  P5 t& N" O    do
3 _' `  B% J6 ]/ n. n; z8 b    {
4 i& `4 I1 t6 {* u& _1 \0 e      if (Handle)
3 t) `- u) W$ g* M0 Z  L  @      {
: t' P% \) S$ t        FreePool (Handle);
      }
               
0 Q; g; v6 u" @1 }& ?' \      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
1 D% y4 `( \; W' Q
7 F. [* n. G" t/ o. W7 ]7 E    } while (Status == EFI_BUFFER_TOO_SMALL);
6 h$ F, ?0 }3 g4 g, t' ]+ u+ K    ASSERT (!EFI_ERROR(Status));
- e5 \4 Y/ P$ M3 k% [
$ I2 p0 t: ]! I2 T/ h( s- C8 ]    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
+ a/ i3 K; R3 r5 w3 Q$ ]    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
3 B+ l0 E1 x- K4 m3 ]0 X# ]' V) y    Result = Not_Found_Me;       
    do
    {       
. E- n: {1 B# M8 |* q        if(Index == NoHandles - 1)
1 q* C+ r; T5 S, t# W- k         Index=0;
          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
* f5 R, u% ?9 R) z; E, V- L! X        Print(L"THIS IS DEVICE %d \n",Index);
" x3 B) E) S: }' h6 C7 `& p: Z    if (!EFI_ERROR(Status))
4 t# l- V7 o! J/ d    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
- N3 x5 n) |5 Y& u& k2 g5 _//-          Print(L"Again now Result is %x \n",Result);          
% m( m" J9 F; r& ]    }
; [9 B: H: b# `& X& A    Index++;               
    } while(Result == Not_Found_Me);
       
        Print(L"Please hit any key to exit this image.");       
! c4 Y7 Y+ s" L0 t) l6 ]% l    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
4 c0 n7 B3 h5 W    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习