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

kaurohy · 发表于 2009-11-11 16:21:18

/*
* myfileio.c
* Apps
*/

#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)
{
EFI_STATUS       Status;
EFI_INPUT_KEY    key;
UINTN             index;

for(;;) {
      Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
      if (Status == EFI_NOT_READY)
         BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
      else
         break;
}
if (!EFI_ERROR(Status)) {
      if (key.ScanCode == SCAN_ESC)
         RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
}

return Status;
}

EFI_STATUS
EFIAPI
MyfileioMain (IN EFI_HANDLE          ImageHandle,
         IN EFI_SYSTEM_TABLE    *SystemTable)
{
EFI_STATUS Status;
EFI_HANDLE *DestAddr;
EFI_LOADED_IMAGE *FileHandle01;
EFI_DEVICE_PATH *FileHandle02;
EFI_FILE_IO_INTERFACE *FileHandle03;
EFI_FILE *FileHandle04;
EFI_FILE *FileHandle05;
CHAR16 *FileName;
CHAR16    *BufferA, *BufferB, *BufferC, *Space_Key;
UINTN BufferSize = 8, BufferSizeKeySpace = 2;
int       Bit0, Bit1, Key_Space;

FileName = L"NewFile.txt";
Space_Key = L" ";

BufferA = L"ABCD";
BufferB = L"EFGH";
BufferC = L"IJKL";

Bit0 = 0xff;
Bit1 = 0xfe;

Key_Space = 0x0020;

InitializeLib (ImageHandle, SystemTable);

DestAddr = AllocatePool (BufferSize);

Print (L"Value of Bit0 is %x\n", Bit0);
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;
}

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;
      }

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)) {
      Print (L"Could not open file, FileHandle05 err code is %x\n",Status);
      return EFI_LOAD_ERROR;
}

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;
}

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;
}

Status = FileHandle05->SetPosition(FileHandle05, 1);
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, &BufferSize, &Bit1);
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;
}

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);
      return EFI_SUCCESS;
}

Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferA);
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;
}

//--------------------------------------------------------------------------------------------------------------------------------------------------------------->>>
Print (L"Before Read FileHandle05 file Data of BufferB is %s\n\n",BufferB);
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);
      return EFI_SUCCESS;
}

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);
//---------------------------------------------------------------------------------------------------------------------------------------------------------------<<<

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;
}

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, 12);
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);
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);
if (EFI_ERROR(Status)) {
      Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
      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);
      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;
}

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

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);
      return EFI_SUCCESS;
}

Status = FileHandle05->Write(FileHandle05, &BufferSize, BufferC);
if (EFI_ERROR(Status)) {
      Print (L"Could not write file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

FreePool(DestAddr);

Status = FileHandle05->Close(FileHandle05);
if (EFI_ERROR(Status)) {
      Print (L"Could not close file, FileHandle05 err code is %x\n",Status);
      return EFI_SUCCESS;
}

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 · 发表于 2009-11-30 09:50:02

//**********************************************************************
//**********************************************************************
//** Demo code for PCI configuration space I/O access program **
//** By Lawrence 2009/11/20 @Taiwan**
//**********************************************************************
//**********************************************************************

#include "efi.h"
#include "efilib.h"

#define Not_Found_Me 0x00000000
#define Found_Me 0x00000001

static EFI_STATUS WaitForKeyOrReset(VOID)
{
EFI_STATUS       Status;
EFI_INPUT_KEY    key;
UINTN             index;

for(;;) {
      Status = ST->ConIn->ReadKeyStroke(ST->ConIn, &key);
      if (Status == EFI_NOT_READY)
         BS->WaitForEvent(1, &ST->ConIn->WaitForKey, &index);
      else
         break;
}
if (!EFI_ERROR(Status)) {
      if (key.ScanCode == SCAN_ESC)
         RT->ResetSystem(EfiResetCold, EFI_SUCCESS, 0, NULL);
}

return Status;
}

UINTN
AccessPciConfig (
IN  EFI_PCI_IO_PROTOCOL             *PCI_DEVICE_TARGET
)
{
UINT8  i;
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);
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));
   //print register value
   for (i = 0;i < 4;i++)
   {
      PCI_DEVICE_TARGET->Pci.Read (PCI_DEVICE_TARGET,0,pci_offset,4,PCI_REG);
Print(L"Register0x%d value = %02x \n",i, PCI_REG);9 k! L; M2 p# Q* L" ~; c) R
   }
- K9 ^/ i$ K* S  V1 y    Print(L"\n");
0 g! |( n' B& e+ G3 L/ z5 w    ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));
! {0 ~! m( C8 H: @2 `5 `! U! {* ~    return Found_Me;- t' D7 i# s/ Q+ C1 x, R' y
}2 o) ~4 S7 y: d8 k* r3 L: M( R
) V9 k( W" M, D8 r  x' \4 G2 j. ~
return Not_Found_Me;5 M6 g# A% m# f
}
7 G  U& V3 I3 s, r2 C8 g  ?/ o; S8 P" a
EFI_STATUS
+ @8 j) E' \( b% x" JInitializePciApplication (
: ?/ F; J  T- w/ J IN EFI_HANDLE          ImageHandle,
+ d! ?/ x; P- G  r IN EFI_SYSTEM_TABLE    *SystemTable! @( C* b4 M* K8 O" f! D" q2 T- y
)
- a7 Q, o, f/ A9 s{" L1 K. }1 \7 n2 |( c5 g9 `
  EFI_STATUS             Status;) E; j7 K, ~' X2 \* x' A
  EFI_HANDLE             *Handle;6 @0 t) T6 |# w% T+ E+ }( {+ q0 Z
  UINTN                BufferSize, Result;1 i& G4 `4 j# m8 A: B# [* E1 P
  UINTN                NoHandles, Index;
) n1 N4 r. K! V. y  EFI_PCI_IO_PROTOCOL    *PCI_DEVICE_TARGET;* Y, q; {7 k% y% m4 Y' f% Z
0 X' l+ P) C# d4 Q
//$ W& C5 g, `; ?* [
// Initialize the Library.6 N& ~* W! T5 z8 \8 v
//: ~# N7 @+ k; ^6 p* c4 v
InitializeLib (ImageHandle, SystemTable);
. x& l. u7 t3 ~- d //. E  h0 M7 c0 L4 O0 [
// Find the PCI driver( e0 W) h4 G: G7 L1 g
//
# G: L( u9 c" E( B# |3 T Handle = NULL;
% B9 C( x* ~+ L8 G! P9 u1 p BufferSize = 1;
; X! f  Z7 U0 d7 G: t2 J do8 t! ]7 n1 _! X9 ?9 h: W
{
6 I( ~  }- [, s6 j    if (Handle)7 ]9 o1 e" w+ K" r0 K* p
   {# r4 i* m$ l$ n; F! @9 S3 \
      FreePool (Handle);
0 J6 x' P3 y9 `' r( O* ^    }
+ _6 F* }' a* a7 O 6 a  s% g; H; \0 A5 P+ p
   Handle = AllocatePool (BufferSize);" l8 s; Q* y# X5 P8 @
   Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);" M2 i+ V; G9 a2 p
! _) M5 k9 b) @
} while (Status == EFI_BUFFER_TOO_SMALL);
. |8 w1 Q# h. S8 P3 D1 ]# Q- T ASSERT (!EFI_ERROR(Status));
9 g2 _8 s. P3 K" w
$ C" N) L3 R2 C NoHandles  = BufferSize / sizeof(EFI_HANDLE);
/ M2 |4 Q$ _' L Index = 0;( o% t/ l% `4 Q: |
ST->ConOut->ClearScreen (ST->ConOut);5 ]; I2 W4 N1 j/ U' [
Result = Not_Found_Me; $ R4 }2 [$ {1 b6 ^/ m" r! _+ d
do, L' }$ b# c& n4 b, ^
{ 9 ^& E3 z$ [4 K1 e6 J# W" N
if(Index == NoHandles - 1)( ^$ h$ F& @* P! J
      Index=0;
; W1 \8 h. S# F. {& a% }) `4 x       else/ P$ H1 _7 z" u: c* ]+ w" ]
Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);' T6 [( G3 t0 H& L" |, [
Print(L"THIS IS DEVICE %d \n",Index);
6 J. F% E# b0 V' T' ^8 a- K if (!EFI_ERROR(Status))
, x( n% G' x1 t, Y1 C {
) F$ h1 p  u' F! `/ s7 E    Result = AccessPciConfig(PCI_DEVICE_TARGET);1 S2 Q2 x# H4 W' @. Z3 S
//-    Print(L"Again now Result is %x \n",Result);    # i6 L/ Q, t. M, `
}* H( v7 [7 O  l2 v6 F
Index++; ! d$ K# n5 d0 u
} while(Result == Not_Found_Me);
* A2 v8 d/ [. v. r
) b% [. P5 C: s- K# a Print(L"Please hit any key to exit this image."); % E" b1 d  Y' z( K, y( N
WaitForSingleEvent (ST->ConIn->WaitForKey, 0);; E9 c6 x- t5 ]

  {8 Q/ `" ?% P return EFI_SUCCESS;
4 o8 ]8 P, j9 E* \6 A3 T2 e}

xjf2088 · 发表于 2010-4-16 14:47:39

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

dhua · 发表于 2010-4-30 14:51:38

都是强人。向你们学习！

shaopy · 发表于 2010-8-23 12:52:26

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

蓝色永恒 · 发表于 2010-8-23 14:07:13

景仰一下，哈哈

zywwin · 发表于 2012-5-21 08:57:06

新手，学习

		自动登录	找回密码
密码			加入计匠网

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

我寫了一個 File io 的 EFI shell app

pci register access sample

感激不尽