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

kaurohy

/*
1 E8 P' Y0 u6 o, W2 E * myfileio.c
* Apps
*/

8 x. L; R5 d2 G0 S1 n#include "efi.h"
, V- s  G3 l  g# J, Y3 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)
{
& C* E# F- r7 E: d* x; a3 p1 N    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
   
6 d. E0 V- {5 T$ q5 H$ \/ r7 U0 r* ]    for(;;) {
1 n. [3 U; y7 C& v. n) v        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
# v+ |2 ^! Y  r4 w1 d        else
+ @& M; q% K1 v            break;
+ R5 U& O$ Z* Z, L( d' C6 r    }
! h* V, P  G' k4 h    if (!EFI_ERROR(Status)) {
# S) C+ X+ C" b! z9 Z4 n( l  {        if (key.ScanCode == SCAN_ESC)
7 r9 ~6 a1 ^2 U% V/ {" X' `! L0 I/ A            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
    }
   
! e) F, S+ n* z6 O6 F/ `' n  X& S& s    return Status;
}

7 |+ ^/ g0 t; `EFI_STATUS
, v! c* ]8 V1 `3 c" s( p4 nEFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
+ X9 ^7 C  h" ]5 m3 M             IN EFI_SYSTEM_TABLE     *SystemTable)
- s2 X9 ?3 E' `: z. {7 ]2 I$ r6 v6 Z% s& j{
    EFI_STATUS                Status;
' _( }  z: C$ ~& Y  d    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
& z$ C4 e+ F% F    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
- E7 {) V+ D  g3 N; O" J4 y. I    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
; n" X% _# E1 \& z0 F3 W, H5 }    CHAR16                *FileName;
/ T$ r! K( L1 `* |( k" C5 ?# M$ r        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
  c) N! W( p: b    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       
6 j; c5 b: z7 s
    FileName = L"NewFile.txt";
7 V. ]; h5 \9 W4 q        Space_Key = L" ";
       
        BufferA = L"ABCD";
/ |+ t* ]5 @9 `        BufferB = L"EFGH";
. R# N; a- Q+ N$ i6 t1 P" C3 W        BufferC = L"IJKL";
( p+ Q8 v+ B0 V7 O" m
        Bit0 = 0xff;
        Bit1 = 0xfe;
       
        Key_Space = 0x0020;
       
  y1 x) ~: Q; R! Z8 @    InitializeLib (ImageHandle, SystemTable);       
, V5 Y6 N/ Q1 C" `( n& `
    DestAddr = AllocatePool (BufferSize);       

    Print (L"Value of Bit0 is %x\n", Bit0);
" {9 R( g  ], ^, Q4 h8 c" z4 J    Print (L"Value of Bit1 is %x\n", Bit1);       
3 ]1 D, \3 m4 E
       
    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
! C/ @1 `2 L- G1 d, u0 a- C& h/ \2 J    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
$ `  r' [  d" h! Z) S        return EFI_LOAD_ERROR;
    }

: |0 {4 o/ d, x/ H* M    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;
# s/ I% R" g" P7 ?. p        }       
( F$ M3 b. L3 e       
    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);
% e2 Q( {3 W2 ]* R9 e7 {& F1 ^, c            return EFI_LOAD_ERROR;
( M! S: P: V0 g7 U; R        }

    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);
    if (EFI_ERROR(Status)) {
8 J9 P4 d8 W' I' a/ \7 h1 F7 `/ c        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
    }
       
; y0 @' w; G+ e. i1 [; y( a, C    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);
! `* x9 O+ G, }- L2 n; ^4 d        return EFI_SUCCESS;
/ Q7 j: I) U3 k5 n5 X: F    }
       
    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);
        return EFI_SUCCESS;
4 L9 H3 k8 E0 R+ F8 ^  X$ _    }       
6 L3 i; S( I" W: M# C( ^
    Status = FileHandle05->SetPosition(FileHandle05, 1);       
    if (EFI_ERROR(Status)) {
2 t( q- P% L" r2 U3 a, S( C        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
0 |: H% R" D" {& _& l       
) R; C/ i% u& |% Q- @( s5 ~5 o    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
+ i5 G# z" X! S, C* F        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
# h: a5 b/ D# _' m& c9 N8 _1 V. A        return EFI_SUCCESS;
    }               
       
" d( y6 f% D3 _    Status = FileHandle05->SetPosition(FileHandle05, 2);       
0 ?, w; s, I- T/ k    if (EFI_ERROR(Status)) {
% F# G; V( D2 E# ]- I; b1 A; r        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
: S( T5 Y9 A) Z. r  }0 z+ L        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
        Print (L"File Buffersize is %x\n\n", BufferSize);
% O' A# g, O4 e8 u. s1 S# U$ M4 @* b    if (EFI_ERROR(Status)) {
' r" q7 \+ Q6 E$ O  j8 f/ h4 m        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
* H/ m2 F, b. j. P. `5 h        return EFI_SUCCESS;
: V* V# B; X5 [! i  m" J    }       

//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
6 f( z, [9 Q* j# q* s    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
+ p1 @! L: E( c% L    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
* r/ k: v1 X2 l" S        return EFI_SUCCESS;
7 k' ]6 j9 p$ H3 B- ]9 M7 u    }

    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
    if (EFI_ERROR(Status)) {
' Z+ f) y' j  V' r0 v        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
9 _' k9 M; u: O5 o! E/ p, @        return EFI_SUCCESS;
    }       
        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<
2 r6 M1 H% h( s3 Q$ F2 P2 [% E
5 `/ C& L/ Z& @0 W. U    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);
* ]4 Z) H1 `0 a3 Z        return EFI_SUCCESS;
    }
0 M' l" ]4 L, K) _7 Q6 ^       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
3 `# y7 G5 @$ h- F; q0 Q) ^    if (EFI_ERROR(Status)) {
# _* X4 Y( w8 f, K6 |. `/ ?        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
# J4 n0 w$ q9 t9 r7 L. k# X        return EFI_SUCCESS;
    }
1 }/ A; N2 h1 p3 P7 G+ R) K
6 G4 G# ~) {3 Y9 j    Status = FileHandle05->SetPosition(FileHandle05, 12);       
! ]) O! ]& \1 D5 g    if (EFI_ERROR(Status)) {
$ e* p) a. Q2 g        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
7 O0 Z% [$ i7 C& @) `7 D+ x        return EFI_SUCCESS;
1 X& t) D1 J& 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);
        return EFI_SUCCESS;
! ~& u  q$ `3 t( G' a    }

4 a. l# s3 E8 _# V  X+ Q: [) G; z    Status = FileHandle05->SetPosition(FileHandle05, 14);       
    if (EFI_ERROR(Status)) {
' u' m" f; J- p# z: x9 S3 t/ z0 ^        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
- A7 P$ ~9 P3 b* y7 s6 }& U8 [$ a       
    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;
    }
$ e5 E5 v. M/ x; U3 E3 W& T/ `$ _  X% }
- p$ h& ~0 X8 d! r  r- n    Status = FileHandle05->SetPosition(FileHandle05, 16);       
    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
" K4 x. r1 N1 Q        return EFI_SUCCESS;
0 K! H$ i. M3 k+ A    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
+ i4 O- B1 i! I- e; d+ Z1 U) X        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
8 J3 r- _, c% T( B    }       
+ ^+ h# V6 `; r  n
//---------------------------------------------------------------------------------------------------------------------------------------------------------------

' }2 e& V' [0 ^5 c6 Z- z9 I    Status = FileHandle05->SetPosition(FileHandle05, 18);       
$ _4 z, [% R, q( v* Y. A    if (EFI_ERROR(Status)) {
' W3 [. X, g2 @        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
& r2 a: q$ j1 K& h1 E        return EFI_SUCCESS;
    }
" L. X; b* R" ~9 x* h6 Z6 Z+ [       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
4 n$ `' x; q$ y    if (EFI_ERROR(Status)) {
; ^; f; }3 Q5 K% f- c! `) t        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
# `. s' v* I7 g        return EFI_SUCCESS;
    }       
) P8 t3 U5 m! ]+ G! G  @* {1 H8 r       
9 e( f+ e" h, a- H    FreePool(DestAddr);       

0 c6 j* t* G& O6 I    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
9 w; @$ H- k$ `* N" t% u! Y6 q        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
$ y% z! Q) K7 C# q    Print(L"File Name = %s has already been created.\n", FileName);

    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();

    return Status;
}

chenhy

//**********************************************************************
//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
- g" x! H" E! E! u7 E/ B# e//** By Lawrence 2009/11/20 @Taiwan**                          
//**********************************************************************
/ _% E8 X3 d' L' T: t//**********************************************************************
, k" M. u! R; }+ }0 j/ r
0 V9 n' Z4 ?2 Q3 s#include "efi.h"
#include "efilib.h"
# k+ S# m) i6 N
#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001
% M5 p7 A% F$ K4 w* J  w: ?) {
" G, Q$ w7 e' z) F8 ]static EFI_STATUS WaitForKeyOrReset(VOID)
4 N  `: ^9 T; `& v7 }8 q2 h  f{
6 \& O2 A$ s3 j  X8 d2 s2 w    EFI_STATUS          Status;
2 h! Y) @' O1 o7 Y. I1 ^( Q    EFI_INPUT_KEY       key;
5 T  |+ U5 M6 m" u% |% c    UINTN               index;
   
" w* U! _; n6 Z4 s5 J$ \: b    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
) S- H, m3 c  m$ Z. V5 Y3 d        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
5 ~* ~4 V( D. d$ t) M% \        else
            break;
    }
    if (!EFI_ERROR(Status)) {
4 ]" A0 d% Y1 B3 A; ?) e        if (key.ScanCode == SCAN_ESC)
9 F: \' N( [3 x% Z! S, Q3 r            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
2 Y6 I0 K7 M  f' j' g# K0 D    }
   
    return Status;
}

  d* U6 Z4 ?1 Q+ j% v* s0 LUINTN
: T( j0 n; o* AAccessPciConfig (
) h) w, O9 i8 I$ o; e- x$ N    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
  v5 d8 P8 N; M" r5 l' k    )
$ i* r3 I2 Z/ H- e: M& u9 U{
        UINT8  i;
& B+ k. x6 Q+ {' ?6 Y        UINTN  PCI_ATTR[4];
3 H: _) Y8 y% F$ G  J0 t# @8 F        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);       
0 j( g% o& S/ e' Q    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));       
+ g+ X' }9 k) e$ I# f$ j( B          //print register value
          for (i = 0;i < 4;i++)
$ k' m' t6 h9 r- s; @          {
* p2 I7 i/ s) K2 O- Q        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
  r. X* b: K& ~6 W) d% ~# {          }
( h) `( u, J* h! K) Q          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
0 G. q+ D1 y: r) q& H. V          return Found_Me;
( W+ ^! O+ I. l( i    }
3 s1 V! y) U1 {4 m3 b. j1 C% B4 ~7 W
, @- l/ X* j8 c* }- B        return Not_Found_Me;
}

4 I5 j& `% R9 ?) ^, ^2 {EFI_STATUS
InitializePciApplication (
; R+ U6 F2 [6 i! q$ d$ T7 y, u    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
    )
* t3 ]' a: X/ w0 b' t% I. i2 b% O{
  EFI_STATUS              Status;
  \' R1 }& n# s3 X  EFI_HANDLE              *Handle;
9 S6 X3 d7 T4 Q" y  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
9 N3 x9 o# ]: {2 V8 e. b8 H6 C  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
3 W5 `5 |2 ]: y, p
    //
    // Initialize the Library.
    //
# ?3 q  g% D* L/ a) n- h    InitializeLib (ImageHandle, SystemTable);
: |8 ~- x% o' V: \7 H) z    //
    // Find the PCI driver
    //
3 I. Y  R8 M( J* C    Handle = NULL;
1 _1 r" k6 W4 @% ~    BufferSize = 1;
    do
1 x0 _5 g) n1 O/ x6 _    {
      if (Handle)
6 i/ ^5 a/ g; k3 ]: w      {
% p6 k. U* I) ^; h+ T( t        FreePool (Handle);
      }
               
5 I- X) G& p( }* O  L/ x6 S% u$ a      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
! }7 [9 r4 ^; c# z4 K
    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));

    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
- Q/ I2 G  S' r$ a6 n7 ?' V3 A    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
/ Y. b- F" D- O  ^+ \: {5 a5 S    Result = Not_Found_Me;       
- H& h- p- _) y# t! i9 w    do
3 R- H: ?" r- W# Z. N5 q0 I    {       
( J: f  Z6 I+ V$ @$ e3 A        if(Index == NoHandles - 1)
) U' z) F( @% O  o3 e& U0 w         Index=0;
          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
' r, p- L+ k3 B, t) C    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
  i& E, h7 @" ?8 r; M9 L5 Y+ d3 ]//-          Print(L"Again now Result is %x \n",Result);          
    }
    Index++;               
    } while(Result == Not_Found_Me);
       
# @" z' w* P; c9 H        Print(L"Please hit any key to exit this image.");       
8 j# E' t5 V5 v    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
: x' ~& P' Q6 P4 L! u% |0 E* [               
    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习