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

kaurohy

/*
* myfileio.c
* Apps
*/
# p: @# y$ F' w
#include "efi.h"
#include "efilib.h"

#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE

static EFI_STATUS WaitForKeyOrReset(VOID)
* v9 _. ?1 o  g6 `' Y" M' |{
    EFI_STATUS          Status;
# j; T* F. f! i& C% D+ j' r    EFI_INPUT_KEY       key;
8 ^2 {2 m, d" C! j9 ~    UINTN               index;
   
6 |. j; ?: _, A( k: s* p    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
3 f7 Q. f7 H  P2 @5 v        else
( H& }$ `; H6 H! F+ L. m% k            break;
) g8 V* t7 z) t# Y    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
; F* s$ s# M+ j. E  t9 A   
    return Status;
}

( w1 _( w+ B  uEFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
- O7 J" Y' T/ l1 M6 k# F             IN EFI_SYSTEM_TABLE     *SystemTable)
{
& g2 V, y/ h, v! y1 y8 ]    EFI_STATUS                Status;
) d/ N+ _5 y" Y& x1 @4 Y    EFI_HANDLE                *DestAddr;       
' ~$ z5 w' A6 z% e- C) F9 v% f+ L5 T    EFI_LOADED_IMAGE        *FileHandle01;
    EFI_DEVICE_PATH        *FileHandle02;       
0 c. a& Q- L& d  o2 \% @- r& N/ N    EFI_FILE_IO_INTERFACE        *FileHandle03;
    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
$ `7 P0 Z4 O2 @5 `0 V    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
3 D$ m( O' _0 n1 w    int         Bit0, Bit1, Key_Space;       

2 q! e* ^: ?6 s" S    FileName = L"NewFile.txt";
        Space_Key = L" ";
9 Y+ C: h/ F7 W       
        BufferA = L"ABCD";
* s/ n: E' I. @$ Y9 n        BufferB = L"EFGH";
        BufferC = L"IJKL";
2 q( H8 C6 s4 `, u
; r% C! H$ F/ n' i        Bit0 = 0xff;
$ o% X  F0 b* H, M% W# d1 e+ L        Bit1 = 0xfe;
       
; V2 Q( O1 G/ A( w1 j        Key_Space = 0x0020;
5 _2 X- [4 M- n# R/ x' f# x$ ~* t       
+ \5 t3 \2 G1 q7 {- f4 C5 T% p    InitializeLib (ImageHandle, SystemTable);       

) q  s9 J6 T8 m. W    DestAddr = AllocatePool (BufferSize);       

    Print (L"Value of Bit0 is %x\n", Bit0);
    Print (L"Value of Bit1 is %x\n", Bit1);       
* {7 n' k0 w4 ^3 y
       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
  c% O$ z' N6 X2 m8 H' F    if (EFI_ERROR(Status)) {
) l- U/ g# G* v3 T$ F        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
# D- O! b; z6 m9 N        return EFI_LOAD_ERROR;
0 n: Z  G$ S" f. v% @2 x    }
6 r6 O8 b7 R; r/ W$ J' b
    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
    if (EFI_ERROR(Status)) {
0 u3 S/ g9 A1 S2 Z0 F7 f4 m  K            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
7 y* S3 c0 \  \0 O6 c% B            return EFI_LOAD_ERROR;
8 a, i9 G# @" L6 t" |        }       
       
) y7 Y4 @0 w0 G; w    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
  a6 K) e( D) L4 y3 O8 y    if (EFI_ERROR(Status)) {
            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
% j4 W' e4 J! d% ^- s- }  \8 w            return EFI_LOAD_ERROR;
        }
; x, u. k6 @# C- ?: E5 {- Y
0 K3 g6 e& v) F& M# w8 g8 V# w. Y    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
    if (EFI_ERROR(Status)) {
        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);
" }$ Z) J7 }# g5 v* q5 Y    if (EFI_ERROR(Status)) {
) V1 x9 h& c+ H, ^' r0 J        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
) A+ U* Q. ?  v- F* N        return EFI_LOAD_ERROR;
! L+ `" \: N6 l% s3 @- t- B    }
       
9 K2 j  n8 [0 g( e9 x* @* b    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;
, F# M& _- r5 v3 R1 |    }
2 R3 j$ W5 E. p       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
' T8 g1 _7 u2 `6 L; ^- ?    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
. N6 N: G: v# v- J7 g    }       

! G7 C+ c% _: w0 E! @. P. h% f    Status = FileHandle05->SetPosition(FileHandle05, 1);       
# j/ a+ s6 u$ J4 o( [. Q& B    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
  W+ X% L4 J+ W        return EFI_SUCCESS;
    }
       
9 {2 K( p5 D  t+ d$ p, N4 q    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
! C) q8 T2 W0 `6 _& ]% x9 U2 G        Print (L"File Buffersize is %x\n\n", BufferSize);
& S2 A- Y8 {0 t7 w2 v2 {    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
1 o/ m$ O$ O7 i! O: b  w5 M2 s) J. B        return EFI_SUCCESS;
2 s6 K  @; e6 A4 m( z2 D    }               
9 d4 C8 e. Z- d: c# a+ R       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
. e( p8 Q  Z9 U: U. {        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
        Print (L"File Buffersize is %x\n\n", BufferSize);
. S% N# Q, |2 {% q& x/ c. ~5 `    if (EFI_ERROR(Status)) {
  J, D0 ^; ]8 W- ]1 P: i0 ?) d        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
, s/ f- _  J$ l( G. `    }       
4 y% b0 x) R1 O$ m$ `9 H* |/ `
//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
1 J9 H9 H; D1 z$ i$ p    Status = FileHandle05->SetPosition(FileHandle05, 2);       
    if (EFI_ERROR(Status)) {
( [# B  z) d; C& G0 s- x        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }

5 x! J' [9 j- r9 w# g) d    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
    if (EFI_ERROR(Status)) {
/ x; z' C& b1 o5 x1 ^. s8 f" i        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
7 m) j2 S+ H$ A. W, s4 p  p    }       
        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
& _7 R) b! e& N1 f3 |* e
" I; n- P' I9 f3 o4 f' f    Status = FileHandle05->SetPosition(FileHandle05, 10);       
  C; Q$ ~" x! n! I) y1 n9 x4 \. W    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
! i# M- Q1 g; [! m        return EFI_SUCCESS;
4 M( h" K4 u0 Q; Y6 M+ k: L1 a    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
% q, W( |9 q; H$ K4 }& Q    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
; V: D( Z% Y, q: h. {/ b( a& J        return EFI_SUCCESS;
) G0 L" s# Z8 S    }
2 T1 d3 s7 N% ]* R
  V  S+ E3 E+ Q) T    Status = FileHandle05->SetPosition(FileHandle05, 12);       
( f2 a) y/ b' S- ]" [& r, T    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
" y0 N1 y5 J) l0 S5 Q  N3 ~: f        return EFI_SUCCESS;
    }
       
( \6 E$ ~: K9 N) i7 U4 X8 H    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
; w, \: V. J8 t: L' [9 C        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
" B6 R* I+ E' l3 W, i    }
# l3 N+ U/ ]( t- x: W
    Status = FileHandle05->SetPosition(FileHandle05, 14);       
3 h5 }: N* w4 K" q: K: J    if (EFI_ERROR(Status)) {
/ k, E. c) F8 |3 R        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
; ?' n) F+ ^9 J        return EFI_SUCCESS;
    }
       
( @2 P5 A3 ^3 a2 a6 d% G( e    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
; R0 w# }: e  F, n, E+ L- z9 q1 N& z    if (EFI_ERROR(Status)) {
# l. G3 ]2 ?4 o) G5 r        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
4 R) p6 J5 q5 y0 N        return EFI_SUCCESS;
    }

    Status = FileHandle05->SetPosition(FileHandle05, 16);       
    if (EFI_ERROR(Status)) {
9 m( `4 s# E' ]8 f        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
) y5 h5 E  V1 z  L; c9 Z3 _    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
; y5 k' a9 \9 `3 r7 a9 D% e; N& M    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
0 M, J" L2 ?3 j5 C# @- V4 N    }       

& k0 @; l7 d9 Z+ }9 u& d8 Y' Z//---------------------------------------------------------------------------------------------------------------------------------------------------------------
' n7 [7 b) R. f' d
    Status = FileHandle05->SetPosition(FileHandle05, 18);       
2 ?! R- J+ K  B" J& f    if (EFI_ERROR(Status)) {
* w* E2 f' _1 A% y1 Y! {        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
8 K/ a8 ^3 v/ p0 r" Q2 V        return EFI_SUCCESS;
    }
       
# U# p- k5 ^3 q. p* k    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
* n1 [+ N: z! ?$ p: X    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
       
    FreePool(DestAddr);       

7 ~8 h0 M; {: O0 T: @) T3 w    Status = FileHandle05->Close(FileHandle05);
3 F0 u/ o; i: `( V1 Y    if (EFI_ERROR(Status)) {
9 A- {" b( r; b% ^        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
' [' f) }* I) n    }
       
# p8 T7 {; R) ?" @- H    Print(L"File Name = %s has already been created.\n", FileName);
# g9 |# z( A* @, Z* ?' n
    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
* F* D) W( ~  i    WaitForKeyOrReset();

    return Status;
}

chenhy

//**********************************************************************
//**********************************************************************
! g: e: r3 e. P//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
6 R$ O5 E' M- x) a//**********************************************************************
! b5 b( R3 S' [. j) q, Z: }//**********************************************************************
. O8 {# d( }) L6 E0 U6 p
#include "efi.h"
#include "efilib.h"

7 _( V7 O( w% @#define        Not_Found_Me        0x00000000
3 y' H" ?  M& @) t$ c& v#define        Found_Me        0x00000001

) N. a, Z- J7 o" \7 B; K- ustatic EFI_STATUS WaitForKeyOrReset(VOID)
5 v  U3 A7 G3 {4 D+ K" b{
    EFI_STATUS          Status;
! J6 o4 j3 r9 S+ N  |    EFI_INPUT_KEY       key;
1 {) l# V; i6 C8 ^/ |. d% I    UINTN               index;
) e9 N8 K% q- x' [: T   
    for(;;) {
! J( o$ z1 \3 ~/ U& p        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
4 ~& S+ [2 R- |9 R& J2 M0 P' |        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
0 S$ c- q3 ^4 w  n            break;
    }
4 J. U8 o0 q4 C, I; D    if (!EFI_ERROR(Status)) {
6 x( Q. p& _* _9 w/ B/ R7 v6 P5 E& C        if (key.ScanCode == SCAN_ESC)
* v5 w% M5 N1 K( J" j            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
8 i6 F; D& a1 _( i6 X( J4 \( O    }
0 P; H9 L  C2 F$ D   
" Z. {! |, ~3 f1 t7 ?( D8 g    return Status;
}

6 L' |* T; X0 f, F+ o4 |% aUINTN
4 i, m" K! t7 k( ]! N+ TAccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
  }- V3 K( I4 v& S, v1 Y    )
& f/ A; j: N# r& ~0 O% H{
        UINT8  i;
* ?  U1 A( l! S2 g# |( ~  M9 N& K        UINTN  PCI_ATTR[4];
9 E, l3 n3 v1 }: i        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);       
    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]);
2 F7 l; W) A( p8 W  L3 @' Q          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
          //print register value
          for (i = 0;i < 4;i++)
2 D; {% |/ V% `+ c          {
( ^( b7 v$ o) L$ B* o! l+ s        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
          }
          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
2 Y( b  Y" p, a3 H0 d4 |    }
$ [6 P0 c; n- H+ b: A  i7 L
9 |0 X# {8 P# _* x, i        return Not_Found_Me;
7 ]) {6 q3 V$ c( n, [4 F}
- E. u2 d0 i% K6 u* X7 O1 J: e# F/ p3 B
& F& A% b" Q# G; \; YEFI_STATUS
) o# b3 ~/ Q; i9 @$ V& B4 m' vInitializePciApplication (
, ?2 X4 w" Y* W0 w6 o* j8 ^5 b    IN EFI_HANDLE           ImageHandle,
( p, U/ L5 R4 b! o, A) t    IN EFI_SYSTEM_TABLE     *SystemTable
    )
{
' E! T* h; i- t  EFI_STATUS              Status;
: t! g/ y9 K9 r# U& g  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
  D# a9 a8 `# t2 T/ I/ P  UINTN                   NoHandles, Index;
  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
3 S8 _1 v6 \" w' [% s$ X
$ i; y* c; `" Z    //
    // Initialize the Library.
    //
    InitializeLib (ImageHandle, SystemTable);
    //
    // Find the PCI driver
    //
+ b1 u8 m" Y; G: b6 E8 `' ]    Handle = NULL;
; U" Q2 W/ A+ o  k1 c0 Z+ m- i5 g    BufferSize = 1;
    do
9 B$ l- M# W2 O. U+ Z1 ?    {
* j# @7 m! T2 j5 b% S      if (Handle)
      {
        FreePool (Handle);
      }
8 s4 ~' M: V/ l" N& J, B               
9 J0 U* M. A6 Z  r- n" m0 D( }      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
, d$ |' [1 A9 a6 x% H
    } while (Status == EFI_BUFFER_TOO_SMALL);
7 h/ F2 Y6 P# f7 ]% n$ V/ ^4 f    ASSERT (!EFI_ERROR(Status));
( @6 c8 A0 s. S' x
+ K6 W- j" z4 t! s0 |, p! q7 \8 D    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
' `' W/ x% b2 A* g# {3 g3 n    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
, e, z/ C0 |3 \1 i7 t5 k7 D    do
$ d6 q5 ]- W2 d. @    {       
5 J5 G. P, K8 p. @# `5 T1 {        if(Index == NoHandles - 1)
         Index=0;
* T7 r0 i2 A& ~$ E. [          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
, D. I" ]3 G/ A% I0 u        Print(L"THIS IS DEVICE %d \n",Index);
* C( `" W4 e1 `- d    if (!EFI_ERROR(Status))
+ R+ r  f2 P$ h+ a8 f    {
) C0 X0 j7 `& k* h/ y          Result = AccessPciConfig(PCI_DEVICE_TARGET);
( p4 |0 D3 h% \! U) T//-          Print(L"Again now Result is %x \n",Result);          
    }
    Index++;               
8 w1 t6 E: S' J. _" P2 n    } while(Result == Not_Found_Me);
       
5 I4 c* `( b! s; M& A& x7 N        Print(L"Please hit any key to exit this image.");       
    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
    return EFI_SUCCESS;
: Y: ?! E2 ^/ y2 g2 ?4 ]) n& m}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习