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

kaurohy

/*
* myfileio.c
* Apps
- R. k9 i5 Z0 L7 B4 j" Q */

#include "efi.h"
#include "efilib.h"
# X7 `" |$ @# j5 C0 J5 }* z4 M7 r
#define FILE_ATTRIB_CREATENEW         EFI_FILE_MODE_READ  | EFI_FILE_MODE_WRITE | EFI_FILE_MODE_CREATE
( ^5 \+ Q% O: Q7 B5 ^
static EFI_STATUS WaitForKeyOrReset(VOID)
$ Y3 |; _0 v# m2 ~! N- T& ^{
9 w/ T2 d+ T( [- y1 i* d6 ?$ T    EFI_STATUS          Status;
    EFI_INPUT_KEY       key;
    UINTN               index;
: z; V9 Y, l; T7 Z3 ]/ D   
    for(;;) {
9 r7 A3 C1 S# T: K        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
) b! Q+ }, k1 q6 y7 L) R1 X) r& k! g8 m        else
            break;
, ^% ?! k5 t1 R2 E: k7 V, G& u    }
" b+ _# \6 V! e1 a    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
' G5 o, C6 j" \  i( a% s/ z6 |, I. S            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
& w' P8 I/ [& r; \! u    }
   
    return Status;
}

EFI_STATUS
EFIAPI
+ Z# q- |# p2 b2 P( hMyfileioMain (IN EFI_HANDLE           ImageHandle,
3 c/ Z) n6 v" Z$ K: d! I* `             IN EFI_SYSTEM_TABLE     *SystemTable)
' h" ^0 P+ P. `( o5 o! N/ s{
    EFI_STATUS                Status;
    EFI_HANDLE                *DestAddr;       
    EFI_LOADED_IMAGE        *FileHandle01;
* m& S. T: B+ K/ {$ a. x% |' l) U1 Q    EFI_DEVICE_PATH        *FileHandle02;       
* z/ g! e  N, }/ o+ K4 k    EFI_FILE_IO_INTERFACE        *FileHandle03;
3 i2 \( i6 N6 M$ i    EFI_FILE                *FileHandle04;
    EFI_FILE                *FileHandle05;
    CHAR16                *FileName;
        CHAR16      *BufferA, *BufferB, *BufferC, *Space_Key;
    UINTN                BufferSize = 8, BufferSizeKeySpace = 2;
    int         Bit0, Bit1, Key_Space;       
. j; B! j, U2 o6 O( D' G5 C
    FileName = L"NewFile.txt";
9 [  o' W3 @& O/ }+ K4 ]        Space_Key = L" ";
       
        BufferA = L"ABCD";
        BufferB = L"EFGH";
        BufferC = L"IJKL";

        Bit0 = 0xff;
        Bit1 = 0xfe;
' N4 C  j/ [5 a( Z       
* `3 f( f2 l  Q9 H7 b  R        Key_Space = 0x0020;
       
, p+ ^. _3 m' X8 T; w# h2 x9 |7 A& ^    InitializeLib (ImageHandle, SystemTable);       
1 o8 |* w5 W2 f9 ~  a; R4 G0 @
* }9 s* K- w( g9 B) u5 q) m    DestAddr = AllocatePool (BufferSize);       

( p" C/ c4 E2 a$ p+ L4 x+ u    Print (L"Value of Bit0 is %x\n", Bit0);
/ f& L+ N& ?; z! l3 \    Print (L"Value of Bit1 is %x\n", Bit1);       
% _. A4 H# z/ Y  B- ]
8 c+ v3 ]; t/ b$ m       
8 p$ v" c5 r4 f! {4 ?    Status = BS->HandleProtocol(ImageHandle, &LoadedImageProtocol, (VOID*)&FileHandle01);
& T" l3 y6 b9 k    if (EFI_ERROR(Status)) {
        Print (L"Could not obtain Image, FileHandle01 err code is %x\n",Status);
        return EFI_LOAD_ERROR;
. T) a* v! i2 d) P    }
; G/ }' a$ M  o" U  ~
& P  a4 i& }: V. m) o$ e  o+ b    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;
        }       
       
    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);
            return EFI_LOAD_ERROR;
        }
2 B/ b: j; g( G! S; ]5 h
    Status = FileHandle03->OpenVolume(FileHandle03, &FileHandle04);
- n* |) I0 P" O) h+ k  `$ q    if (EFI_ERROR(Status)) {
        Print (L"Could not open volume, FileHandle04 err code is %x\n",Status);
5 ?9 V( [6 |& L; _        return EFI_LOAD_ERROR;
    }               
- g* e6 \( i' P+ k               
    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);
- J1 d4 u* r! N/ W        return EFI_LOAD_ERROR;
    }
  V: Q- S0 \% [+ e7 R4 s# L       
0 B, i7 T. q, A) g4 N& |8 i( ~    Status = FileHandle05->SetPosition(FileHandle05, 0);       
- S4 n9 i& d% O0 L$ I/ ~    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
. E2 E5 y. _. V6 J    }
       
! f1 V/ B9 A" f5 j! o    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit0);
) ]8 n$ a0 x$ M( l, N* f        Print (L"File Buffersize is %x\n\n", BufferSize);
7 Y5 R* {" F/ J+ q    if (EFI_ERROR(Status)) {
) {1 [- P% ]# P- o+ w1 [        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       
. T# ?$ r  S1 M1 ^) P2 H
# D. Y# _3 G& g6 ]( I2 i* {    Status = FileHandle05->SetPosition(FileHandle05, 1);       
) P, G& q! p1 Z' D/ b+ F1 R    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
2 b$ Y$ T" W9 a8 F        return EFI_SUCCESS;
. x% P& ]4 @% [' O/ M    }
- Z# D" P" F/ _2 F! Q9 O       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, &Bit1);
        Print (L"File Buffersize is %x\n\n", BufferSize);
    if (EFI_ERROR(Status)) {
) N$ A8 L9 _5 A. N# `        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
: k+ n5 M! u) n& M6 F8 [( {, ]" n        return EFI_SUCCESS;
2 {% e" C4 U+ H( ]5 d! D) s    }               
2 J7 K. B. V$ y0 |       
    Status = FileHandle05->SetPosition(FileHandle05, 2);       
5 S; B7 `8 h: S+ _, B" v  Y. a    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
6 ]2 N6 s+ \* z( W6 X. m: N* B        return EFI_SUCCESS;
4 i+ F7 V8 `+ \9 a    }
3 J- x: z4 P# m" m8 X9 g7 f, I       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
8 Q6 j% Q% [# F        Print (L"File Buffersize is %x\n\n", BufferSize);
4 m8 V# T" h6 a6 @( q4 J( q    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n\n",Status);
        return EFI_SUCCESS;
    }       
* {1 o! w+ T% i+ v
//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
7 G0 u" U2 D+ @    Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
- r8 U! _' T* G. B    Status = FileHandle05->SetPosition(FileHandle05, 2);       
7 q8 M) d7 |. ^* M( Z    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }
+ e: {6 Z9 F- f# b2 ^
    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;
    }       
        Print(L"1. After Read FileHandle05 file of Data of BufferSize is %d and Data of BufferB is %s\n\n", BufferSize, BufferB);
- W8 I' O) O" |' I" K//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

3 b. I; f" R. ^7 H0 f; A. x. i    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;
    }
* Z) n' e8 l  m       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
. Y, Y0 c3 n4 @% t; [( e  u5 X    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
% L. y& M. @% `+ {! o8 m' H        return EFI_SUCCESS;
    }

    Status = FileHandle05->SetPosition(FileHandle05, 12);       
    if (EFI_ERROR(Status)) {
' d0 ?3 g6 m7 k9 T* x2 H        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
+ I2 a' a2 B9 |5 m( r        return EFI_SUCCESS;
    }
/ j" H% _) k; g$ q1 x! [       
& b* G; X( N6 N    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
    if (EFI_ERROR(Status)) {
" T# }9 \: l; T  p; k, ~        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
7 ^7 F" C9 ?* _* h, W        return EFI_SUCCESS;
    }
* T6 o' A. D- F1 v
    Status = FileHandle05->SetPosition(FileHandle05, 14);       
5 \" ~+ }+ o- g    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
9 q7 V6 Y8 J$ e. y" k    }
       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
7 }) d) N4 I! y1 ]# r    if (EFI_ERROR(Status)) {
        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
" W/ B. v+ g0 j5 M% O, j- P2 U        return EFI_SUCCESS;
    }

    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);
; k% G6 h! p; o* k! Y& r; }        return EFI_SUCCESS;
    }
) b, r; h+ J9 v" b5 k. i5 s3 P       
    Status = FileHandle05->Write(FileHandle05, &BufferSizeKeySpace, Space_Key);
) q) p9 l5 o( y# U, N/ G$ h* A5 q/ z3 W    if (EFI_ERROR(Status)) {
0 j% I- j8 G. a7 Z        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
/ M. \' ?! a4 r7 g" ?# b9 v
//---------------------------------------------------------------------------------------------------------------------------------------------------------------
1 S; W) X" i, L1 B8 b
. k6 E- c6 N% U" }    Status = FileHandle05->SetPosition(FileHandle05, 18);       
    if (EFI_ERROR(Status)) {
        Print (L"Could not do SetPositioon to the file, FileHandle05 err code is %x\n",Status);
. ^+ ?+ X+ [4 r        return EFI_SUCCESS;
    }
0 T  d8 l3 x& f       
    Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
    if (EFI_ERROR(Status)) {
8 y4 G% r$ [- l$ {0 X* u        Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
        return EFI_SUCCESS;
    }       
9 e9 J8 V5 i8 t9 N; w       
8 P. T" @! h% O5 s4 y$ }; V" S    FreePool(DestAddr);       
) a9 }% s$ c5 I; l
    Status = FileHandle05->Close(FileHandle05);
    if (EFI_ERROR(Status)) {
: t# n& S# _0 f2 M  |        Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
2 T6 ]) W. m7 i  C1 B        return EFI_SUCCESS;
    }
       
    Print(L"File Name = %s has already been created.\n", FileName);
/ e. {6 F/ x  C% l& u# v0 V
3 t$ U+ N2 t$ Q6 [    Print(L"\nPress ESC to reboot or any other key to return control to the firmware.\n\n");
' m  `6 M0 A7 c    WaitForKeyOrReset();

. _9 ~$ e4 x  P( ~3 K- y, l& T$ A; l    return Status;
}

chenhy

//**********************************************************************
//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**                          
( G# f) V1 L) m//**********************************************************************
//**********************************************************************
1 J! ], \# D" B3 G$ U
- d0 e/ _' W. e#include "efi.h"
% T  I) j6 n4 L#include "efilib.h"

#define        Not_Found_Me        0x00000000
#define        Found_Me        0x00000001

static EFI_STATUS WaitForKeyOrReset(VOID)
  S- {# R- T- u7 ]; T; p; O{
    EFI_STATUS          Status;
: N. l1 W/ O( n% M6 j    EFI_INPUT_KEY       key;
    UINTN               index;
   
' D( \7 X! ~  w; W* L0 M3 q# F    for(;;) {
        Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
1 x9 v7 K9 s, b3 O% O$ h        if (Status == EFI_NOT_READY)
            BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
% M3 a& n4 b) ]8 K' e( K# q( u        else
            break;
    }
    if (!EFI_ERROR(Status)) {
        if (key.ScanCode == SCAN_ESC)
            RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
3 j; l1 h2 n5 q2 i; [    }
   
6 l1 ^7 h  S3 s0 j' p3 `/ ?    return Status;
}

3 n3 B1 p* H* W+ W8 IUINTN
: u. _$ `( T' V# |) oAccessPciConfig (
) [- N" @5 A% D2 ^    IN  EFI_PCI_IO_PROTOCOL              *PCI_DEVICE_TARGET
    )
{
: k  h' D* U2 b$ P) ]% [        UINT8  i;
        UINTN  PCI_ATTR[4];
( Z" {! t: O4 O        UINT8  PCI_REG[4], pci_offset=0;
. @8 d0 `' q* C4 m, v; V. b
        //get device attr
) n. y* R$ A# P" i- t1 M        PCI_DEVICE_TARGET->GetLocation(PCI_DEVICE_TARGET,PCI_ATTR,PCI_ATTR+1,PCI_ATTR+2,PCI_ATTR+3);       
* P% z6 U$ Q* C6 l1 S( [, G; ~4 t    if ((PCI_ATTR[1] == 0x00) && (PCI_ATTR[2] == 0x1e) && (PCI_ATTR[3] == 0x00)){
6 G) v; W. o8 A. ]8 V( O          Print (L"Device found at BUS = %02x         DEV= %02x        FUN= %02x  \n",PCI_ATTR[1],PCI_ATTR[2],PCI_ATTR[3]);
) B9 P( Q* }8 t/ J$ L          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_GREEN, EFI_BLACK));       
* X* a& Z8 y" R' s          //print register value
/ u! G" _3 {. t2 d- ^2 F3 h          for (i = 0;i < 4;i++)
. ]7 A' j0 r3 w  L: R/ ~8 {          {
& {, g$ T/ U) N        PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
$ f5 j5 p/ L* O; M: ]                Print(L"Register0x%d value = %02x \n",i, PCI_REG);
* U0 Z% [( t  i          }
          Print(L"\n");          
: f  z. C- M) q0 J* D          ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));                  
          return Found_Me;
. Z# U4 ]: [( H: ?) b( [4 a! N6 U    }
* d/ }& H; Y0 k; ?- q# V
        return Not_Found_Me;
3 h7 \5 W+ \1 ?}

EFI_STATUS
InitializePciApplication (
; B; s- U, n  p! y: [1 }    IN EFI_HANDLE           ImageHandle,
    IN EFI_SYSTEM_TABLE     *SystemTable
    )
{
  EFI_STATUS              Status;
  EFI_HANDLE              *Handle;
  UINTN                   BufferSize, Result;
  UINTN                   NoHandles, Index;
8 V$ a1 ?: h  `+ Q3 m* }  EFI_PCI_IO_PROTOCOL          *PCI_DEVICE_TARGET;

    //
    // Initialize the Library.
    //
" H( X' r* j9 y2 \$ m# ?( A  j    InitializeLib (ImageHandle, SystemTable);
8 d1 a8 W. j. ^# e* D$ i8 Y; e( W    //
    // Find the PCI driver
% m; x6 z, {% r  `( r# B7 N    //
5 s8 W4 f' [" q: `, _: H    Handle = NULL;
    BufferSize = 1;
    do
    {
      if (Handle)
3 f- r+ d" R( Z* q, y      {
1 l) `1 T: V) s; r4 a        FreePool (Handle);
( ^- P/ k4 _" j( Z+ G! e      }
: t& q! ~! E1 K- J3 h# _# m& x               
( }* N3 U9 _4 ?1 X      Handle = AllocatePool (BufferSize);
5 A8 j! Z' f9 ~; c- e      Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
9 {% b. x- x) u# J3 v: a" W- M
    } while (Status == EFI_BUFFER_TOO_SMALL);
: U' F9 R) \+ ]# w    ASSERT (!EFI_ERROR(Status));
/ a) n' W! P. O
( L3 w1 R! Q: @6 n) B$ Z2 c    NoHandles  = BufferSize / sizeof(EFI_HANDLE);
$ a/ n7 n7 J) v8 b% e    Index = 0;
    ST->ConOut->ClearScreen (ST->ConOut);
2 }: N' U5 n+ S0 a, K6 h    Result = Not_Found_Me;       
    do
    {       
& X. @8 K6 [7 z' {3 }  r9 A' A2 s        if(Index == NoHandles - 1)
( M8 u3 L+ ]7 n+ Y         Index=0;
          else
9 y$ }' g! o. i! J7 G    Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);
        Print(L"THIS IS DEVICE %d \n",Index);
    if (!EFI_ERROR(Status))
    {
; _7 n; h+ i$ a" n( G+ @- U; x          Result = AccessPciConfig(PCI_DEVICE_TARGET);
6 J% d; d( y% @; V5 b//-          Print(L"Again now Result is %x \n",Result);          
$ r0 c, @+ m( [) f- }0 t    }
    Index++;               
    } while(Result == Not_Found_Me);
0 F; Y3 y7 P2 T# s4 j0 l       
        Print(L"Please hit any key to exit this image.");       
! z2 p, N; x1 A$ y2 C; L    WaitForSingleEvent (ST->ConIn->WaitForKey, 0);
               
. J1 C0 F* [2 [3 Q    return EFI_SUCCESS;
}

xjf2088

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

dhua

都是强人。向你们学习！

shaopy

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

蓝色永恒

景仰一下，哈哈

zywwin

新手，学习