[原创]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);
. u; H( K  b, i8 o, ]    }% M' ^0 K: C$ U7 y$ \
   Print(L"\n");    9 A5 I% c! u/ v) \" z0 G- O
   ST->ConOut->SetAttribute (ST->ConOut, EFI_TEXT_ATTR (EFI_WHITE, EFI_BLACK));
4 j. `/ X! t  h; U- \: h) ?; P    return Found_Me;
6 F8 ^% i' u$ Z, R; y }2 Q, b4 ~( G7 d) F0 _9 @$ A

9 g* g! B( M' K" S( V return Not_Found_Me;
: m% Y7 j4 J% g& G# v}
+ t; p, Q0 ]5 m* |5 o- k2 J! g3 n; M3 W
EFI_STATUS+ [7 U7 G+ w" Y9 c
InitializePciApplication (
: V3 W2 ]- W8 v& Y0 ` IN EFI_HANDLE          ImageHandle,3 e# V; m! J5 ]$ A$ J
IN EFI_SYSTEM_TABLE    *SystemTable* C. S2 }, [6 V" G0 x
): T- v$ @" d5 B/ j9 L+ g; U
{2 |7 o3 A8 n) I( |6 s7 f9 @8 P
  EFI_STATUS             Status;
7 Y) K8 R! d6 ]) B3 U$ M  EFI_HANDLE             *Handle;
$ H( x! y2 s9 F+ a! C7 T4 e  UINTN                BufferSize, Result;0 g2 a0 C4 j/ ?- R* r+ z
  UINTN                NoHandles, Index;
5 d0 X! D2 I5 M  q3 n' e. l  EFI_PCI_IO_PROTOCOL    *PCI_DEVICE_TARGET;) h& E( y) I* }6 V; ^; u
! b9 z! @. ^! ]" H$ s' C
//1 a7 n2 K8 k* `0 l+ D+ G9 m
// Initialize the Library.% z1 E" h4 d# e& F! Z: i9 ]4 u
//. P2 w# N. R3 _2 ~
InitializeLib (ImageHandle, SystemTable);; g, M3 z0 T+ p7 x
//- A. _' W) x4 z  V( F2 j$ o' n
// Find the PCI driver
$ z1 M* E) n" p$ O/ a" G( i$ A //! ~/ B: k6 L3 x7 a5 W/ ?
Handle = NULL;% n, a0 L# R* E! _/ x0 F6 h& F3 g
BufferSize = 1;& I' d6 G2 x' ]2 B& v3 u
do
1 o) |1 ?( a6 ]) Z {( A( y; r, @3 H
   if (Handle)
' E2 ~- h% c5 X3 _! y- M# c    {
3 h8 I1 F1 Z9 k6 S       FreePool (Handle);* Q  [/ t4 x! q! {5 f4 w, m
   }' K% A6 a; s4 ]1 e. l& M

: m# s& p5 @/ B! D- J5 i5 l. G    Handle = AllocatePool (BufferSize);; ~" C% ]$ B' q
   Status = BS->LocateHandle (ByProtocol, &PciIoProtocol, NULL, &BufferSize, Handle);
+ A+ y  S0 x8 |/ j( R- Q! O3 K" n: R5 i$ ]# H
} while (Status == EFI_BUFFER_TOO_SMALL);) I+ V8 u% P- M7 Y6 B
ASSERT (!EFI_ERROR(Status));2 |7 x3 \9 v2 P
+ S: r. {! ?' y
NoHandles  = BufferSize / sizeof(EFI_HANDLE);
7 B7 @' }7 y- T) _. o$ z3 V, z2 j Index = 0;& s5 e: j+ Q1 E/ z( J, z
ST->ConOut->ClearScreen (ST->ConOut);. ]6 H+ v- p# k5 Z# Q. ?, o6 r# g
Result = Not_Found_Me;
  ]& [; k# v3 F/ o do# n6 m( E+ X0 z# g/ I3 `1 }/ v7 Y
{ 6 B% N$ Z' J4 h8 o
if(Index == NoHandles - 1)2 Q8 y7 @/ n) }4 |1 A1 i0 f* l
      Index=0;# t8 a" Y. W  [5 K  G/ `
      else
+ }' W2 |; v# { Status = BS->HandleProtocol (Handle[Index], &PciIoProtocol, (VOID*)&PCI_DEVICE_TARGET);8 m, }  E3 y1 i. @6 @0 h
Print(L"THIS IS DEVICE %d \n",Index);
' a$ C% K  Q" K2 ?! b1 W7 B if (!EFI_ERROR(Status))
! m( f& |3 C& P- ~1 n3 b9 a {
+ v- ]6 k4 c1 L0 ]  V- J    Result = AccessPciConfig(PCI_DEVICE_TARGET);+ ~9 z" {# f: ]+ L, j) ~) Y  n
//-    Print(L"Again now Result is %x \n",Result);
! s3 u) ?8 k( m7 I5 a5 W }3 Q% I- N* |% O2 q& Z
Index++; ( Y" j' ~, m  o' [% w! C; g. ~( I5 G
} while(Result == Not_Found_Me);
  R* I% M, f9 D/ D% S: l
" r7 f, E$ P% M% l Print(L"Please hit any key to exit this image."); ) X4 g8 a4 ~, Z3 `# `
WaitForSingleEvent (ST->ConIn->WaitForKey, 0);& H, \1 ~5 }* u$ }) X( |: V
+ k3 g0 {- m7 N4 p5 H
return EFI_SUCCESS;
9 \! b# e6 W$ Z$ E2 W}

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

感激不尽