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

kaurohy

/*
* myfileio.c
) ^  z1 b1 z2 \- Z" _5 _' r * Apps
# {: x3 ^7 d8 R5 v, M1 | */

6 g* J: P6 w( C! p9 g2 N( S3 H% i8 J#include "efi.h"
! v! F* e( N3 ~* H#include "efilib.h"
2 @9 m2 Y* _, P! |5 m: x
#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
  N2 r/ g# Y  F, j% s6 O0 x
static EFI_STATUS WaitForKeyOrReset(VOID)
/ R" |0 w* z& |' K5 L8 H3 q{
* J. O# H/ n( w' b    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
1 \1 Y" O! h7 \    UINTN               index;
   
+ P& H. R  Z' c    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
, s4 ^% C, Z' H% B7 X8 \        if (Status == EFI_NOT_READY)
1 n4 d) C; I! f- V+ w9 L- y            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
3 D- r8 s6 R" T5 q9 q' G6 T        else
& [# `* h& I4 c: R# Z- Z4 P! I            break;
    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
3 d+ f' b1 v- d$ r6 J  [            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
0 }- B; d+ R, W( @! q6 O. E8 h   
+ e: g- e! K6 K5 u$ D) Y; @0 `6 w5 e    return Status;
- J4 M4 |/ c( @. w$ y1 z) C}
- E' R1 s; ?* \7 q% {+ N4 ?# w0 t6 g
EFI_STATUS
& Y" }0 Z' E# Z3 d- k+ e7 \EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
{
    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
5 k! q( _0 b2 P4 |    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
7 Y  r6 S' H( i) ~    CHAR16                *FileName;
6 F& H& l4 ]0 r  g$ D( |        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
, t$ n5 z( \) q! `$ j1 _5 B& O( |    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
3 K& n: H' m* o4 [' `2 q    int         Bit0, Bit1, Key_Space;       
) `/ o$ [2 [( i1 f4 s0 Q. c
+ K& W5 A6 R3 k% C* }4 e    FileName = L"NewFile.txt";
: S1 z0 u9 w8 |; V- B        Space_Key = L" ";
3 x5 w: O1 L& x3 Z! U       
2 J4 o! Q% a* r* c; k& k        BufferA = L"ABCD";
        BufferB = L"EFGH";
        BufferC = L"IJKL";
. _, E- q* B" t
- e- C/ R4 A' f        Bit0 = 0xff;
        Bit1 = 0xfe;
       
' r' f5 F% M# M$ B" M% M        Key_Space = 0x0020;
       
5 U$ C& u4 Y4 D5 I3 ~! s5 F    InitializeLib (ImageHandle, SystemTable);       

    DestAddr = AllocatePool (BufferSize);       
, s$ x/ [0 {& Y9 ]0 @+ l5 p
    Print (L"Value of Bit0 is %x\n", Bit0);
. J; G. F' o# E' ]' f! }    Print (L"Value of Bit1 is %x\n", Bit1);       

       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
4 @4 N2 X/ L7 E5 t, n        return EFI_LOAD_ERROR;
    }

    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;
9 L5 F; g8 d4 G% Q9 {" x8 Q2 R9 ^# b        }       
       
    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);
0 Z, o* m+ O1 O4 r            return EFI_LOAD_ERROR;
! ^) k, D. P. F0 [$ t- {+ e        }
/ ^; l8 J% c6 \7 m. ]* k6 m
    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
. V. @4 B5 q9 ?+ O8 ]4 o/ w* Q        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
- ]9 v! c3 g/ N5 F2 o1 u        return EFI_LOAD_ERROR;
    }               
               
    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
    if (EFI_ERROR(Status)) {
        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
  C: U* [% e: {% {/ S. F  s        return EFI_LOAD_ERROR;
    }
       
    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;
% L* V/ W) @9 }# B% i    }
7 Z- n' E0 j) C4 k7 L! l% z6 w! H       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
' ~! a, S/ M+ s* e    if (EFI_ERROR(Status)) {
# u. B( ]8 c1 T# O/ x$ W# d        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       

* J0 z. V# z! p6 T& \' B    Status = FileHandle05->SetPosition(FileHandle05, 1);       
* v2 U: z6 ~# b2 t* ]1 J- g    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
2 |+ N# |: r, m: O        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
, J1 }; o0 k) R- v( q3 s        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);
& _4 L& d4 v2 S0 `+ d! D% ?        return EFI_SUCCESS;
! g  X/ y' b& x% F# D: D    }               
       
, P! t) V* U# n' x    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;
' r# V4 {+ T( }' o    }
8 L$ v" |) \8 _6 ~) L0 q0 B' N       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
        Print (L"File Buffersize is %x\n\n", BufferSize);
4 z, b$ Q3 l8 x: n    if (EFI_ERROR(Status)) {
! h  w+ i# J) ~, \; e$ g        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       
3 @1 z. L1 R8 u
4 x: ]) {0 y: b//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
* N2 y# L+ ^1 f; D) h5 Y$ _2 ~    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
" B# w: n  C5 Y4 Z    Status = FileHandle05->SetPosition(FileHandle05, 2);       
1 n  {, g; y' ~2 N    if (EFI_ERROR(Status)) {
6 [7 r# `% n/ V3 ~$ r1 k        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
% h- f7 B  ~1 ], D" _        return EFI_SUCCESS;
% C3 h, N, @! T) |    }

; B  y7 w( M3 S+ S+ i1 l+ E    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;
% p. _6 Y' L2 [: a    }       
4 E+ C, u% ^7 L6 g# U        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
, s$ }0 d' u0 S& P4 I
0 k3 G/ ]2 A: b- J    Status = FileHandle05->SetPosition(FileHandle05, 10);       
% s% e6 h) t+ L# T    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
! u* H& s8 S6 n& f! N$ o        return EFI_SUCCESS;
( ]" L, e# Z( l    }
+ h) R, U8 e+ R1 @. g       
# C+ C" p: m/ p    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
' z" J9 o- }$ G- f3 O  V0 S    if (EFI_ERROR(Status)) {
! H  h% H$ J  }; W! o  d+ `        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
9 h3 I* M+ [/ @* ?- Z9 i  x( p' B    }

; G0 T6 z3 F$ V3 h4 d  n$ m6 ]. f    Status = FileHandle05->SetPosition(FileHandle05, 12);       
6 t; j" m" i! W) L6 c    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
3 M3 q1 Q( v! I* y+ I8 J       
' E3 u8 M: m( d- N1 N& o  p    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
4 O% @4 Z. `" p    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
% B, u0 |% u. M' p/ {; |    }
, |9 h& d2 b0 S/ T+ l+ H
# ^& l4 p  J, X; U1 g. _    Status = FileHandle05->SetPosition(FileHandle05, 14);       
7 _. b) J+ |0 z; ]% }! E    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
# }! C/ C0 O5 N/ y( J( Q; f        return EFI_SUCCESS;
  Z5 h/ Y+ D+ y8 U5 }4 T$ A    }
8 Z+ `3 o# [0 W0 z       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
2 V2 g" D6 C0 P7 b3 d( W        return EFI_SUCCESS;
3 d4 {# V. @4 O    }
' L; r& h* m1 h' o- m
    Status = FileHandle05->SetPosition(FileHandle05, 16);       
8 P. N. i. r, t( ]$ m    if (EFI_ERROR(Status)) {
: J) G2 Q8 I2 M$ g2 [7 Y        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
; u) z! D0 ?. z; S/ X1 A    }
" S8 V, _$ A4 T2 d6 g; O       
" ^# w+ q; ~4 F( q4 X3 O! K    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
6 Q, K: ]7 Q7 |7 O        return EFI_SUCCESS;
$ P5 G# r0 t, k. c* ?    }       

//---------------------------------------------------------------------------------------------------------------------------------------------------------------
  `* S0 D! R3 w6 |/ C
+ P8 i5 ^) f& @0 s# V; {5 j3 Y    Status = FileHandle05->SetPosition(FileHandle05, 18);       
# F9 j0 B1 a' t+ M9 o. ]+ S( _8 s    if (EFI_ERROR(Status)) {
* K+ i5 `% c+ Y% l! z4 R        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
6 F& P: m. S/ Z9 x       
9 A- Y6 G4 R/ j+ I" b5 V, B! [    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
3 N# Y' o  v5 @# X& O    }       
       
    FreePool(DestAddr);       
- H# Q2 O' c" ?- E: A+ `
    Status = FileHandle05->Close(FileHandle05);
" E- W% I1 [+ Y, b. d    if (EFI_ERROR(Status)) {
/ W# D6 B) Q* D2 R$ O4 N8 B        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
" ~1 Z' T7 i" D3 m        return EFI_SUCCESS;
% x' Q( Y6 D$ E; N& Z/ k5 o    }
) R) ]8 n& t7 ^8 q5 X0 f& Y       
: N4 {- ^3 a2 D* a8 A# @( S# j    Print(L"File Name = %s has already been created.\n", FileName);
& ~' W5 r+ ?. L+ H# {
! a) H8 e' o/ J' ^    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
! A2 `3 ?; i. S( n* s$ a' z) f    WaitForKeyOrReset();
# Y1 U& C$ L; ?3 s  _
    return Status;
) U* ~$ X: p5 `) h) E% v0 K}

chenhy

//**********************************************************************
# j3 T* \; L0 |" F$ m* I//**********************************************************************
! v5 ?/ N  k: S# i' Y& d//** Demo code for PCI configuration space I/O access program **
# w. ?) b# L/ j' [& p//** By Lawrence 2009/11/20 @Taiwan**                          
, G' D# y- J" i2 Y4 E//**********************************************************************
/ g: V( y% ]# o" x+ b//**********************************************************************
; |1 P  z  I. x  c9 y3 M
5 u7 Y- c6 ?1 g$ U: E" P2 h#include "efi.h"
2 W; g: e8 |) U/ ?5 g7 Z& \( s#include "efilib.h"
5 F0 B3 Y3 u0 H1 U% [, j
) l2 G2 V) C/ R* A0 I' A( V#define        Not_Found_Me        0x00000000
3 @" \( [, P& I5 M: a3 t#define        Found_Me        0x00000001
( n1 o3 |& f, {8 G: J3 U# B
) x# C* ]! T$ l5 g' astatic EFI_STATUS WaitForKeyOrReset(VOID)
{
3 u3 c6 A2 k, w" r    EFI_STATUS          Status;
/ m, h0 O2 B8 L$ Y6 x    EFI_INPUT_KEY       key;
    UINTN               index;
- q% T' C1 A# V' i& Y" \   
    for(;;) {
. Q0 O, O$ c5 _4 @# R0 g/ T        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
/ _7 _3 O1 V: R# g1 @: J            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
* c& J2 v4 P# i) q( w5 `7 ?            break;
6 |% U+ U$ A+ X. J    }
4 T' I& l. h7 m% D+ y$ x4 I    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
. l! M0 V, S4 C) w. r8 [* e    }
   
    return Status;
}
1 I1 K9 M4 B& Z8 }) l0 R
8 t1 c+ e5 I% c6 G" |UINTN
; A0 z- D# x$ c1 N5 o. YAccessPciConfig (
* A' C+ s. P# E( W    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
{
7 o9 D! d! R  L7 i! U( y8 ?" G        UINT8  i;
3 [  M8 T, F) F8 n2 y/ {6 G        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;

        //get device attr
. M0 G9 v& ]( {1 M# k7 J        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
4 T4 s! p$ o9 C, K: j" ]! ^    if ((PCI_ATTR[1] == 0x00) && (PCI_ATTR[2] == 0x1e) && (PCI_ATTR[3] == 0x00)){
( F8 ~! ~( `- |          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));       
9 a; V% S5 h: T$ _6 N) @8 v          //print register value
3 W: D6 c% ]( w' b5 ]/ D          for (i = 0;i < 4;i++)
9 F5 w: N1 z) I% e3 n* d0 P          {
1 j; E9 G8 ^) H+ V" J6 U; y        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
' E' K- `) D' Z) J          }
          Print(L"\n");          
$ r$ A1 S$ U; `  s1 a          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
: U( a# U, v, i3 Q4 [1 x, `8 Z          return Found_Me;
    }
( P! B  S, D) z
        return Not_Found_Me;
}

EFI_STATUS
6 ~& ~- Y! Z: q" ]) J0 KInitializePciApplication (
    IN EFI_HANDLE           ImageHandle,
7 A  f% ^# _2 k4 N$ x  X5 X    IN EFI_SYSTEM_TABLE     *SystemTable
( e8 c  t* K+ s; v+ v) J    )
{
  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
6 F+ A. u# a6 ~, I9 a  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
0 u, K8 z8 N$ E2 a+ m- c  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
* C: G* v1 L$ `* V4 j5 P/ W5 P
    //
! a6 ~" r% W5 h0 Y- t5 Q, H. _    // Initialize the Library.
    //
# H2 L2 H0 d1 t( u5 W    InitializeLib (ImageHandle, SystemTable);
    //
2 c: {3 c+ h+ L; L$ \: J    // Find the PCI driver
    //
4 H; `8 [' V! q/ M$ ?' z+ c    Handle = NULL;
    BufferSize = 1;
    do
    {
      if (Handle)
      {
        FreePool (Handle);
      }
) e$ ]' t; P1 u, p% `5 J4 X  p- S. q               
      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
) Q8 \; v; X$ V  A: Q3 M
    } while (Status == EFI_BUFFER_TOO_SMALL);
* d$ b; S$ R9 ?7 l: e    ASSERT (!EFI_ERROR(Status));

    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
& @: P6 Y0 t. E/ a6 r5 U/ S    Result = Not_Found_Me;       
    do
9 J3 y3 d0 W8 z* L! a1 }0 p, t    {       
# K$ n. c) C$ I3 V; L7 U8 Q        if(Index == NoHandles - 1)
, C- B) C7 g( R# P5 y         Index=0;
          else
4 Y9 R" e# t) Q, q9 a' z2 c% m3 l    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);          
    }
" f3 H2 v9 J/ z3 Q! x0 }    Index++;               
    } while(Result == Not_Found_Me);
       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
% Q" _8 J6 C+ J    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习