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

kaurohy

/*
* myfileio.c
6 \: ~% k' y5 f, V& C2 [ * Apps
3 u, X' I: a: l- w# ?! E0 m */

#include "efi.h"
) O9 F* Q$ X7 v5 P, L#include "efilib.h"
/ P( b% w3 r. D" k5 y+ a
#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

5 |, B& y0 V5 N; L, K" Kstatic EFI_STATUS WaitForKeyOrReset(VOID)
{
" R1 H8 [/ W: R1 \# C    EFI_STATUS          Status;
8 z5 Y& x# U+ T& x9 _    EFI_INPUT_KEY       key;
    UINTN               index;
   
, k) R! M" ~8 {# A7 B    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
5 t% f  g& [* B* t" n1 _  c( O" W        else
            break;
    }
" ~4 m" P' m4 V4 E& }# ^    if (!EFI_ERROR(Status)) {
- T8 n/ F& i8 D- k8 W        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
+ Z- k$ U8 R( Y5 f    }
3 G1 e# O2 L" E  `1 u. K- o  R$ d   
    return Status;
/ T5 s; w$ z5 O: a! {}
1 Y. S, i" a& T! B
EFI_STATUS
$ w$ ^& a/ o1 z! b3 X* CEFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
% I) S# N8 X3 @. n6 U{
    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
9 v2 g/ @  k. e9 C4 }' J    EFI_LOADED_IMAGE        *FileHandle01;
5 Q, [& m3 A+ B6 t5 @) r$ @    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
2 h# O8 Z5 e$ u8 i    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
% G: b! q" B' o5 L    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       
1 ~1 b4 F) h" `% \3 K
5 b% r: {+ Z4 W7 V  I3 k    FileName = L"NewFile.txt";
# k. I" y' S" _- h6 D; K2 f        Space_Key = L" ";
       
% [* R' [" i( E7 F: @5 ]        BufferA = L"ABCD";
        BufferB = L"EFGH";
        BufferC = L"IJKL";
# C" q% l9 O5 p4 ?" E6 c
        Bit0 = 0xff;
        Bit1 = 0xfe;
2 c6 t  ?' i4 N       
! e$ X2 {: j" J( Z9 S        Key_Space = 0x0020;
4 q8 A) ?) |' D) d& ?2 B- d       
8 ~$ D. S: \# ^( i    InitializeLib (ImageHandle, SystemTable);       

  `7 u2 h; u, y" }( k    DestAddr = AllocatePool (BufferSize);       
* q/ Q2 A; h1 g$ R' L4 G
    Print (L"Value of Bit0 is %x\n", Bit0);
    Print (L"Value of Bit1 is %x\n", Bit1);       
7 u  ]; l0 F5 c* C
. f8 L, e% _6 k4 ?' i0 r2 M       
. v- @" G: @3 w& x& I9 V1 e& D    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
. v3 |3 n( P) b+ B# @        return EFI_LOAD_ERROR;
    }
6 Q8 S  }5 ~( j
- t+ J5 g6 D9 j0 H/ v$ s    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
    if (EFI_ERROR(Status)) {
' z* Q+ h) e4 k  n" N0 E            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
5 O6 L% _8 u/ J) E            return EFI_LOAD_ERROR;
: m( t8 j! n1 `8 ?% d        }       
/ P5 A7 a& e3 s$ _, ~& C8 z6 d       
    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
# C& e6 O% h7 y, V' @    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
! m: K! Q/ c7 W- u2 t4 J# e! `        }

    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
: z" _6 n7 r! t% e4 y' t        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }               
               
# O0 V4 g$ q3 [" i% b1 v& N    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
0 D4 `! S- E. t    if (EFI_ERROR(Status)) {
2 b7 M+ p$ @' g0 e        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
! ^- J3 Y+ T: }# f1 n' ]8 i& ?5 b  @    }
       
" @: b6 O; h0 Y6 X    Status = FileHandle05->SetPosition(FileHandle05, 0);       
5 e3 k4 t# M( I: Y    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
- m4 V% @; }0 ]# d7 L* P  k9 p       
* |1 T; v. |. j% z) o4 ~( f* H    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        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);
# M5 ?* b5 u2 Q% `: e2 J# @6 e6 V0 G# h        return EFI_SUCCESS;
( z7 ~" r7 ~1 L: q    }       
; y* n# Y+ u9 ?4 N
    Status = FileHandle05->SetPosition(FileHandle05, 1);       
    if (EFI_ERROR(Status)) {
  y: J/ m: X" u0 }7 P3 H  f. R        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
/ e* ~/ {% k4 S9 d+ _0 N; f    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
! C& c1 d* o: z; q- c" f    if (EFI_ERROR(Status)) {
8 @. I1 f# t0 v" K) w        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }               
       
1 A8 R" l$ K1 K# ?    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);
$ Y* {8 a6 g* H- [        return EFI_SUCCESS;
    }
       
1 x& a; F7 s( h2 s( X    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
+ A) z' ~( b, X- f2 X8 s        Print (L"File Buffersize is %x\n\n", BufferSize);
% Q8 f: X; E& v2 n    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
% u& l7 k" ?* [, I8 D% D+ {        return EFI_SUCCESS;
    }       

//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
. \  O! P0 g/ a$ F1 q" k) K    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
) o# \+ k) g% y5 p$ ~+ v    Status = FileHandle05->SetPosition(FileHandle05, 2);       
& u9 V2 ~+ R9 a( v% |% I" M0 |& J    if (EFI_ERROR(Status)) {
( P/ U1 c& ?* z8 T9 L$ z5 x        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
5 ^0 o( A; b, ^. P        return EFI_SUCCESS;
    }

    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
" _6 M$ x, @/ W    if (EFI_ERROR(Status)) {
# C; F* X! N% r        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
% ^. X, e3 K8 C; c' i) N        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

' w4 D7 n/ K5 y  ]$ a' {& 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);
        return EFI_SUCCESS;
    }
' c/ F5 I  S& ?0 L       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
% R$ S. D6 L  s* Y# h; d        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
' \/ Q* l  f7 N; g$ B# j        return EFI_SUCCESS;
    }

4 Y7 w& y9 W/ P    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
1 M; }1 W+ p: x# ~        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
& Q9 Z! q+ {0 V, B. p. a3 e: v        return EFI_SUCCESS;
7 f( B0 A/ i' Z1 ~' k1 W    }
% n6 [8 G/ }" E  O       
/ A3 F7 @7 T0 C    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;
    }

    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;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
+ p5 H3 V. z3 q1 r    if (EFI_ERROR(Status)) {
5 J2 G* s& s. O# r! ^4 R        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
4 ]- u( I% @8 j/ P4 }) G  w        return EFI_SUCCESS;
4 F$ X. Q/ [+ z    }
- o4 W( S# s6 g' d/ B
6 K$ a- h9 X" Y6 R    Status = FileHandle05->SetPosition(FileHandle05, 16);       
1 Q; c; |0 U3 `    if (EFI_ERROR(Status)) {
. X+ J$ y5 P* z: p: B2 @7 z  K        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
& q$ A  U" u; ^; k    }
2 }! q" Y: p2 y/ 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);
8 W0 M: w. `9 |/ ~' g8 M        return EFI_SUCCESS;
    }       

6 W: D9 F5 C. n5 s5 [! M//---------------------------------------------------------------------------------------------------------------------------------------------------------------

    Status = FileHandle05->SetPosition(FileHandle05, 18);       
/ Q9 @9 c6 }: P8 s    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
; @9 p' _6 m1 A4 g( x+ U: l        return EFI_SUCCESS;
2 L; f/ L6 v& g* k5 r    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
9 m3 i' Q1 L. l" K6 p. S        return EFI_SUCCESS;
  A7 P$ G8 c1 ~0 R9 M/ I3 K    }       
       
; Q+ ^: U% \: Y( {    FreePool(DestAddr);       
( G# r# W* h; Y' a* ^* j
    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
* }& w: B2 D  [+ Y9 E+ X1 |, p# {        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
) P' i  p- k1 L8 w2 n2 G8 g  R- E    }
& t$ }; U- b; }3 J       
    Print(L"File Name = %s has already been created.\n", FileName);

/ X4 R* T- Z) P5 o- g3 J    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();
+ ]3 w5 g% A& f  q
. y8 o7 T, m( \1 r3 L6 I    return Status;
! m8 f/ k6 E; c" T  p; @}

chenhy

//**********************************************************************
& I% `( q/ ]2 B//**********************************************************************
6 a# Y7 u: f, [% E8 U9 ]//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
- O. F# z( F$ b7 i0 ]//**********************************************************************
//**********************************************************************
/ v9 S) [) H' w5 T' M
! S9 E& {/ i# }) S  z" j) x. ^9 H#include "efi.h"
#include "efilib.h"
: o! H8 V* g# W7 E% f: v4 a
' C& L3 S% u8 o0 h7 {: {! g#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001
# f1 U- a( g3 ^4 h% R
/ G9 T$ E3 t5 @. N( ~' k( F" d' Cstatic EFI_STATUS WaitForKeyOrReset(VOID)
1 I7 b9 _1 {0 _- [{
    EFI_STATUS          Status;
" o+ m* F2 k3 E. x    EFI_INPUT_KEY       key;
    UINTN               index;
% n$ W8 T9 N; n4 g2 ^   
3 u% p/ U6 a1 ]) {5 K    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
5 p9 H5 [6 n7 o/ b- w: H6 Y        if (Status == EFI_NOT_READY)
8 G; j& ]6 @' m            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
8 z! O' Z" F0 m3 u/ v7 g            break;
3 O. L6 m6 P0 b    }
1 O! U  i, G& w( G% L" N0 e    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
% n, y; e! V* u5 @0 E   
6 B0 m( p  C4 X# o$ U4 ^    return Status;
5 d$ X: m, H) ~1 \5 U}
$ y5 j6 w9 J" O' t. ~5 f3 Y$ m
UINTN
! y; x7 Y8 ?% EAccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
{
        UINT8  i;
# W$ u3 S# n9 Q' R7 V  T# n0 d        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;

        //get device attr
        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
/ D, v1 M4 F1 W: c7 i, W    if ((PCI_ATTR[1] == 0x00) && (PCI_ATTR[2] == 0x1e) && (PCI_ATTR[3] == 0x00)){
; J7 s+ g' e6 Z3 S& k          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
          for (i = 0;i < 4;i++)
          {
        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
. ?0 @  s* K1 s+ u: D1 N6 b( @( H                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
4 m  s: [- [3 s3 l3 ]4 @( d          }
4 P7 N; C1 C. Y: ]/ j          Print(L"\n");          
; T" v$ X6 H" E$ L          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
    }

        return Not_Found_Me;
/ e6 D0 }- Q# N$ i# k% `6 m* p}

. [) ~5 y3 R. k! K4 i  X3 S: q4 C) hEFI_STATUS
InitializePciApplication (
$ \6 @9 O; C" X" m( L# m$ I    IN EFI_HANDLE           ImageHandle,
. v. M1 L4 q* O7 N. L* o    IN EFI_SYSTEM_TABLE     *SystemTable
) f2 @+ u: R7 I4 g    )
{
  EFI_STATUS              Status;
2 K( Y8 v- U; B* D/ H* L/ ^  EFI_HANDLE              *Handle;
3 l: F& v" \2 l  UINTN                   BufferSize, Result;
9 {$ P5 b* M0 U! A! ^  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
; i% Y9 K# B3 T+ g/ v4 r7 C* K+ s
    //
    // Initialize the Library.
" o. H- e" v1 a: l; P/ q    //
9 k0 p6 |! q7 s% H8 s9 z& E    InitializeLib (ImageHandle, SystemTable);
    //
4 i+ w3 C; o; w% a    // Find the PCI driver
6 Q5 b$ Z+ `  M    //
    Handle = NULL;
/ R& P" f& O, t    BufferSize = 1;
    do
    {
# c" a# u" s- [3 F  s      if (Handle)
, `# m0 J& J9 _. T) J      {
" D0 _* K6 m& Z$ K; i! [+ N4 Y        FreePool (Handle);
      }
               
5 {) [% x% P, m5 B9 @# b/ L      Handle = AllocatePool (BufferSize);
2 ~$ B3 S6 H( ^$ Y) {6 r      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);

% v) l, X9 d1 u1 {5 Y8 }# V6 y% [! X    } while (Status == EFI_BUFFER_TOO_SMALL);
: Q. |5 [% }0 t' J; m    ASSERT (!EFI_ERROR(Status));
; I2 w4 L! T( |0 X0 j9 o& U
    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
% U' D9 r4 J4 `0 [5 G( G- t2 B    Result = Not_Found_Me;       
8 K6 G, E4 R; x+ W    do
    {       
        if(Index == NoHandles - 1)
1 F) T& E; c5 S! Z* m* o' f( }4 x         Index=0;
9 w7 x$ Y2 n6 a* \: l" W1 m          else
% ^1 v2 _/ F3 W6 B    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
4 N( r7 _% N. F% y2 m7 U6 _% w        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
) [  [+ d$ a( N5 H          Result = AccessPciConfig(PCI_DEVICE_TARGET);
//-          Print(L"Again now Result is %x \n",Result);          
2 H! x* u  I. ]; H    }
    Index++;               
    } while(Result == Not_Found_Me);
       
        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
    return EFI_SUCCESS;
4 r) m/ H* m6 J( `7 A* o' |- m; K}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习