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

kaurohy

/*
* myfileio.c
6 \1 e% I+ n" w6 Q& e. y4 j& ?5 B * Apps
( y+ X2 r" }+ m8 R; c* G0 n2 V */
8 D) n: c; F# w8 w
#include "efi.h"
' ^; I! _" w# z1 M#include "efilib.h"

! Z' H% {+ ?3 X; \3 R#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
. r% S* N% R! {6 \6 h. B$ D- A
static EFI_STATUS WaitForKeyOrReset(VOID)
! [" W5 F! u  s# c% Z{
- O) S- R! W8 H; ^7 Y    EFI_STATUS          Status;
4 p/ p1 M4 k% @$ _    EFI_INPUT_KEY       key;
    UINTN               index;
1 I( o' ?: N: q% G   
    for(;;) {
8 G0 Z! |; }# x        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
- u, _  ^, x/ x0 C! F        if (Status == EFI_NOT_READY)
9 g# h+ o1 b9 S- ^/ u: U            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
        else
            break;
* M/ _' q0 n8 c    }
! \1 \, t& H- B! i8 B' v6 z6 w    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
0 i8 v. s! L- j% ~/ ]# B            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
3 M' W. z6 _! W) ]5 C1 \& \1 p) t3 d    }
   
$ l% r# B$ I: F  Q2 y2 b# u    return Status;
% R9 J2 @7 q4 ^0 T6 Z}

EFI_STATUS
& ?2 ?" u' F) V! {; O/ gEFIAPI
MyfileioMain (IN EFI_HANDLE           ImageHandle,
             IN EFI_SYSTEM_TABLE     *SystemTable)
{
    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
* i: e. [# @& ?5 X3 y3 S    EFI_LOADED_IMAGE        *FileHandle01;
2 c. D7 G4 `- W5 P1 ~    EFI_DEVICE_PATH        *FileHandle02;       
    EFI_FILE_IO_INTERFACE        *FileHandle03;
/ V8 z$ J9 L! f& [5 p8 a$ P    EFI_FILE                *FileHandle04;
  y; _1 |. j) s, l8 n7 @    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       
4 i8 R. K5 g& B2 H# G) A+ I+ u6 n
& m7 t: O- i- J    FileName = L"NewFile.txt";
        Space_Key = L" ";
8 i0 _! M2 g4 H: [       
        BufferA = L"ABCD";
* e9 o" y( G- T5 R        BufferB = L"EFGH";
        BufferC = L"IJKL";

3 e1 i- g6 ^6 s9 E2 u; J# {3 ^        Bit0 = 0xff;
9 a6 P/ e! S7 \" S        Bit1 = 0xfe;
       
        Key_Space = 0x0020;
$ G$ m$ ]+ t! m( s; Q5 I$ F: W       
# n% U5 J; M3 \    InitializeLib (ImageHandle, SystemTable);       

# q  J$ \1 b. [0 c+ q: e: c    DestAddr = AllocatePool (BufferSize);       

# x% J( q5 d6 ~8 Q8 e1 `- x0 n    Print (L"Value of Bit0 is %x\n", Bit0);
! b2 [7 m' L  X3 Q; c+ Z    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);
        return EFI_LOAD_ERROR;
2 t2 V! M  T9 i: F, J/ U  O# M$ M    }
' j$ e" a  w6 \$ {
    Status = BS->HandleProtocol(FileHandle01->DeviceHandle,&DevicePathProtocol,(VOID*)&FileHandle02);
    if (EFI_ERROR(Status)) {
4 O. R7 _# G& U            Print (L"Could not obtain Device Path, FileHandle02 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
' T  W) j( @5 X" |3 [4 y: _; J! x        }       
       
3 @, J9 ?0 ]% C7 s    Status = BS->HandleProtocol (FileHandle01->DeviceHandle,&FileSystemProtocol,(VOID*)&FileHandle03);
- H1 t9 O) |5 x0 |7 I- a    if (EFI_ERROR(Status)) {
! ^' z, q, }9 p, g            Print (L"Could not obtain File System, FileHandle03 err code is %x\n",Status);
            return EFI_LOAD_ERROR;
        }

    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;
; w7 c$ _+ C; W5 j8 L    }               
4 x9 G# @) p/ ~: S! D7 \               
7 g, A" O  W+ \- k: q% I    Status = FileHandle04->Open(FileHandle04, &FileHandle05, FileName, FILE_ATTRIB_CREATENEW, 0x20);
  x5 k8 y& v9 X3 d    if (EFI_ERROR(Status)) {
5 X( T1 g) z2 g4 g6 O' a, J        Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
7 }; s  b$ @" L/ `, v" m        return EFI_LOAD_ERROR;
    }
       
5 l5 T; ~# z; @) M7 C" j    Status = FileHandle05->SetPosition(FileHandle05, 0);       
. \6 n: N" ~3 ^9 J8 Y  `! a& A    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
5 I  u' `9 T2 P+ G& {# i    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
        Print (L"File Buffersize is %x\n\n", BufferSize);
5 J; }( G, f" r5 Y& Q5 ]3 r" O; T    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
* ^  ^# b3 ]" U2 l. ?6 m    }       

# g% L5 j0 F- \7 w( }# L    Status = FileHandle05->SetPosition(FileHandle05, 1);       
    if (EFI_ERROR(Status)) {
! {. F& r6 b' I1 q3 V5 C$ n* O" o: p        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
- h0 N* o5 c6 K* X  x& {        return EFI_SUCCESS;
    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
1 }) {. m4 H- b! q+ N# L& x        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
1 D( G& C9 R, K7 c% j3 t        return EFI_SUCCESS;
: E* t, P1 o1 S9 v4 \: |! V$ |8 j    }               
3 Z' K) t1 O! _1 i       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
: g$ m+ i* [7 J6 Z: N' B    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
/ {2 s5 Q+ Y. j/ c* v, Y  {! W        return EFI_SUCCESS;
    }
       
: Z& w( A1 g! T4 R: y$ l6 v" c    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
# N2 X+ [5 w8 r# c, }; w        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);
1 n: ]1 k. P& f$ o" o$ |        return EFI_SUCCESS;
5 T4 E3 q. }! L" S" _    }       

, i" W, f  i5 k& z: a9 [//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
) H# w8 _/ \" B6 Q& Z2 u7 k    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
' P3 X4 T! {" v; U# L* J    Status = FileHandle05->SetPosition(FileHandle05, 2);       
! l8 S1 r/ W: @# N/ q    if (EFI_ERROR(Status)) {
8 b: r" Z7 ?% U2 ?- d2 j5 ^4 A        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
! a5 W9 t+ U0 C2 E- U& s! Y; O( ^- C        return EFI_SUCCESS;
    }

    Status = FileHandle05->Read(FileHandle05, &BufferSize, BufferB);
" I3 f8 Y. h. p    if (EFI_ERROR(Status)) {
  H% V% {- P' ]  z        Print (L"Could not read file, FileHandle05 err code is %x\n",Status);
        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);
9 Z$ `1 y# x& d//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

, L+ c- A9 f; ?# ^: z5 r2 c4 f    Status = FileHandle05->SetPosition(FileHandle05, 10);       
7 B3 v% C& f0 b# d! ~# u/ \    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
; {5 r; u0 N5 B9 O5 `    }
& F# D: T: u/ O( }; g4 s5 C       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
7 q) ]# Q7 g+ H* o& z& e; x    if (EFI_ERROR(Status)) {
5 j7 p8 w2 r+ I. j6 [' s' X        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
4 {/ `1 @! S, t8 P/ ~  a        return EFI_SUCCESS;
    }

1 c- ?, P$ Z. m7 s    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
$ E/ M2 v. }( N) X9 N        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
       
3 Q! c0 h# V2 i9 _& Q6 `- ~    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
; a+ r, P; m$ c    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
4 n/ F; Y% `& s- T* K    }
+ a, y( M- q2 ~. ~7 p# J& G
4 y( z& m; w% m- O+ H) C    Status = FileHandle05->SetPosition(FileHandle05, 14);       
- K) r) P, Z3 |: i& g  D" Y* l    if (EFI_ERROR(Status)) {
( @. _% W4 H  L/ m        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
0 B* V( c5 i# @$ S  ^% Z& K& y) A        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;
    }
0 W9 n- |9 S3 V" F9 m/ u$ C
$ Z& |" }: @+ Y  x. _/ x1 @    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);
        return EFI_SUCCESS;
    }
4 t0 i/ ~: v, t' C       
8 q' r4 z0 d, X, H    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
+ z  n' z8 B, B& ~4 c; |    if (EFI_ERROR(Status)) {
" @$ v' Z3 d$ _( O' U5 c, U        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
9 S' G1 `1 e1 D5 w9 A, ?( j    }       

//---------------------------------------------------------------------------------------------------------------------------------------------------------------

    Status = FileHandle05->SetPosition(FileHandle05, 18);       
* m0 \) P8 l$ ]; O1 N, F    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
1 A9 {  R  ~# _: c    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
2 @9 }: T/ M6 N2 c. c$ a        return EFI_SUCCESS;
  c3 H! F" D4 x) r    }       
       
, n. w3 T' `$ w* x: n% k3 w% A    FreePool(DestAddr);       
' C+ r2 b& _( V: C1 D- ?1 P
: E+ n3 ?" X+ r: @, D- e    Status = FileHandle05->Close(FileHandle05);
7 M! e) o  h# |1 {  D    if (EFI_ERROR(Status)) {
        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
8 h5 \- m3 {5 Q2 O% C        return EFI_SUCCESS;
    }
       
    Print(L"File Name = %s has already been created.\n", FileName);
, ~7 H$ M  s  `, p4 Y! U+ y8 e
8 ~+ j& p! \" K& i- K    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
    WaitForKeyOrReset();
  L$ E1 h# a# J, R
9 T1 e2 n2 Z2 x    return Status;
5 B% F: a0 j3 H3 {0 f3 ]8 d( n}

chenhy

//**********************************************************************
//**********************************************************************
! x  U# w/ j/ `$ u6 w, F9 Y//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
# Y3 \8 h7 f4 ^//**********************************************************************
//**********************************************************************

#include "efi.h"
" b/ I9 Z- J6 w% p" W#include "efilib.h"

#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001

static EFI_STATUS WaitForKeyOrReset(VOID)
/ o& n, `' w9 D0 L3 F  H# N3 q: M{
    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
7 ]6 N4 y  s; M7 ?    UINTN               index;
2 Z# P( h% {% o- ~1 H/ D+ ~   
    for(;;) {
0 y, {9 Q3 ?+ |5 E0 \' X        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
1 h$ c: L( t, ~/ G            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
' P7 |9 u% y# N) d+ R* S6 w        else
            break;
    }
    if (!EFI_ERROR(Status)) {
; C# v7 r$ d$ S2 \& {1 G: F6 n        if (key.ScanCode == SCAN_ESC)
. K4 m; r) d) Y: R" ]! T* H            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
3 e+ n( G; f- \$ E8 Y; u! c5 A    }
   
    return Status;
3 \7 N$ ?# k* T/ U& B! e0 d}
4 h& t* D$ o  U# j2 Y: \
UINTN
) \& p  \9 _* ?$ ~+ H7 n$ |AccessPciConfig (
    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
3 M, C( P) S+ `8 M$ `: J/ x    )
4 c" `0 e5 c7 I; _! a: [( F{
        UINT8  i;
        UINTN  PCI_ATTR[4];
        UINT8  PCI_REG[4], pci_offset=0;

        //get device attr
- z/ P* K6 v' }+ k5 K+ C" a        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)){
5 c% f* R- v. w1 }          Print (L"Device found at BUS = %02x         DEV= %02x        FUN= %02x  \n",PCI_ATTR[1],PCI_ATTR[2],PCI_ATTR[3]);
! X7 F6 J5 T; R# k) g8 g5 o          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
& M) N2 n- C/ J# Y          //print register value
          for (i = 0;i < 4;i++)
          {
        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
0 y' J+ ?0 M: r0 y; ~0 H* R& w8 S                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
& ~/ F6 U8 E3 C          }
          Print(L"\n");          
          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
    }
; Q* \* w3 L+ v
) S8 o" Q3 n( _9 f* z        return Not_Found_Me;
% N$ r3 c0 m1 z6 a, [}
# K* \) E2 R' O# s4 M$ P
EFI_STATUS
3 d# c% x8 Q0 a6 ?/ GInitializePciApplication (
/ i; Q5 I! ^1 w& z7 s    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
3 u3 q# M3 ^) A4 |    )
8 F0 g3 m! i) q- i& t2 Z1 m{
% s. ?- l# X' ]4 O  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
9 N. X. i3 w1 o  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;
: \: m6 i; G0 U; U7 r
; q5 X# J  t. `( q    //
2 X* ~9 {4 E  P% g; o& q" ?    // Initialize the Library.
: `' v/ W5 h8 i    //
    InitializeLib (ImageHandle, SystemTable);
. y; R  C3 ]) B. Y+ E. Q: M    //
4 x% w8 l+ v$ @% w    // Find the PCI driver
    //
    Handle = NULL;
7 f# e( L1 Y& G# n+ ?    BufferSize = 1;
    do
4 C3 Y2 m0 N  u3 W) O    {
* m! l% D- B- F      if (Handle)
$ W9 s1 n: c+ Q5 @* v/ q      {
7 S( s6 X" `, J4 L# o4 h/ b0 m        FreePool (Handle);
      }
               
      Handle = AllocatePool (BufferSize);
      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
& }0 e+ I4 B4 e/ R; M
    } while (Status == EFI_BUFFER_TOO_SMALL);
    ASSERT (!EFI_ERROR(Status));
1 f' E# j! H: f* z" j0 J1 O( u
    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
6 H2 P1 J) x  h" q0 T' C3 s. ?! m    Index = 0;
- Z' \2 V* s+ r& @1 G    ST->ConOut->ClearScreen (ST->ConOut);
    Result = Not_Found_Me;       
1 M8 d  |2 Q$ t. w% `% L4 m    do
- n. `/ Y  J  v, c4 a/ Y6 \    {       
7 G5 y& `9 F9 T9 v        if(Index == NoHandles - 1)
7 \, K+ b3 Q. {         Index=0;
$ [* e7 K9 _) I% i# c) v          else
    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
+ z; j9 N% ~. N! K4 }; |, r    {
          Result = AccessPciConfig(PCI_DEVICE_TARGET);
/ C: M) X! j( K8 i//-          Print(L"Again now Result is %x \n",Result);          
' e; @: m3 c, Q% Y5 E5 u2 `  m    }
! G# b1 ?3 k9 E" F* b# W& E    Index++;               
2 D7 }2 |% E0 l    } while(Result == Not_Found_Me);
, a: U' J4 d3 ?5 r6 Q       
        Print(L"Please hit any key to exit this image.");       
& C4 m$ [6 l  J, _2 D    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
6 I& m. o# G6 P7 x# [- n               
    return EFI_SUCCESS;
$ H/ S0 V" Z% h/ Q% B}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习