[mirror_edk2.git] / MdeModulePkg / Universal / EbcDxe / EbcDebugger / EdbSupportFile.c

/** @file\r
\r
Copyright (c) 2007, Intel Corporation. All rights reserved.<BR>\r
This program and the accompanying materials\r
are licensed and made available under the terms and conditions of the BSD License\r
which accompanies this distribution.  The full text of the license may be found at\r
http://opensource.org/licenses/bsd-license.php\r
\r
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
\r
**/\r
\r
#include "Edb.h"\r
\r
/**\r
  Read a file.\r
\r
  @param  Vol             - File System Volume\r
  @param  FileName        - The file to be read.\r
  @param  BufferSize      - The file buffer size\r
  @param  Buffer          - The file buffer\r
\r
  @retval EFI_SUCCESS    - read file successfully\r
  @retval EFI_NOT_FOUND  - file not found\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
ReadFileFromVol (\r
  IN  EFI_SIMPLE_FILE_SYSTEM_PROTOCOL   *Vol,\r
  IN  CHAR16                      *FileName,\r
  OUT UINTN                       *BufferSize,\r
  OUT VOID                        **Buffer\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  EFI_FILE_HANDLE                   RootDir;\r
  EFI_FILE_HANDLE                   Handle;\r
  UINTN                             FileInfoSize;\r
  EFI_FILE_INFO                     *FileInfo;\r
  UINTN                             TempBufferSize;\r
  VOID                              *TempBuffer;\r
\r
  //\r
  // Open the root directory\r
  //\r
  Status = Vol->OpenVolume (Vol, &RootDir);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  // Open the file\r
  //\r
  Status = RootDir->Open (\r
                      RootDir,\r
                      &Handle,\r
                      FileName,\r
                      EFI_FILE_MODE_READ,\r
                      0\r
                      );\r
  if (EFI_ERROR (Status)) {\r
    RootDir->Close (RootDir);\r
    return Status;\r
  }\r
\r
  RootDir->Close (RootDir);\r
\r
  //\r
  // Get the file information\r
  //\r
  FileInfoSize = sizeof(EFI_FILE_INFO) + 1024;\r
\r
  FileInfo = AllocateZeroPool (FileInfoSize);\r
  if (FileInfo == NULL) {\r
    Handle->Close (Handle);\r
    return Status;\r
  }\r
\r
  Status = Handle->GetInfo (\r
                     Handle,\r
                     &gEfiFileInfoGuid,\r
                     &FileInfoSize,\r
                     FileInfo\r
                     );\r
  if (EFI_ERROR (Status)) {\r
    Handle->Close (Handle);\r
    gBS->FreePool (FileInfo);\r
    return Status;\r
  }\r
\r
  //\r
  // Allocate buffer for the file data. The last CHAR16 is for L'\0'\r
  //\r
  TempBufferSize = (UINTN) FileInfo->FileSize + sizeof(CHAR16);\r
  TempBuffer = AllocateZeroPool (TempBufferSize);\r
  if (TempBuffer == NULL) {\r
    Handle->Close (Handle);\r
    gBS->FreePool (FileInfo);\r
    return Status;\r
  }\r
\r
  gBS->FreePool (FileInfo);\r
\r
  //\r
  // Read the file data to the buffer\r
  //\r
  Status = Handle->Read (\r
                     Handle,\r
                     &TempBufferSize,\r
                     TempBuffer\r
                     );\r
  if (EFI_ERROR (Status)) {\r
    Handle->Close (Handle);\r
    gBS->FreePool (TempBuffer);\r
    return Status;\r
  }\r
\r
  Handle->Close (Handle);\r
\r
  *BufferSize = TempBufferSize;\r
  *Buffer     = TempBuffer;\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
\r
  Read a file.\r
  If ScanFs is FLASE, it will use DebuggerPrivate->Vol as default Fs.\r
  If ScanFs is TRUE, it will scan all FS and check the file.\r
  If there is only one file match the name, it will be read.\r
  If there is more than one file match the name, it will return Error.\r
\r
  @param  DebuggerPrivate - EBC Debugger private data structure\r
  @param  FileName        - The file to be read.\r
  @param  BufferSize      - The file buffer size\r
  @param  Buffer          - The file buffer\r
  @param  ScanFs          - Need Scan all FS\r
\r
  @retval EFI_SUCCESS    - read file successfully\r
  @retval EFI_NOT_FOUND  - file not found\r
  @retval EFI_NO_MAPPING - there is duplicated files found\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
ReadFileToBuffer (\r
  IN  EFI_DEBUGGER_PRIVATE_DATA   *DebuggerPrivate,\r
  IN  CHAR16                      *FileName,\r
  OUT UINTN                       *BufferSize,\r
  OUT VOID                        **Buffer,\r
  IN  BOOLEAN                     ScanFs\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  EFI_SIMPLE_FILE_SYSTEM_PROTOCOL   *Vol;\r
  UINTN                             TempBufferSize;\r
  VOID                              *TempBuffer;\r
  UINTN                             NoHandles;\r
  EFI_HANDLE                        *HandleBuffer;\r
  UINTN                             Index;\r
\r
  //\r
  // Check parameters\r
  //\r
  if ((FileName == NULL) || (Buffer == NULL)) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // not scan fs\r
  //\r
  if (!ScanFs) {\r
    if (DebuggerPrivate->Vol == NULL) {\r
      return EFI_INVALID_PARAMETER;\r
    }\r
    //\r
    // Read file directly from Vol\r
    //\r
    return ReadFileFromVol (DebuggerPrivate->Vol, FileName, BufferSize, Buffer);\r
  }\r
\r
  //\r
  // need scan fs\r
  //\r
\r
  //\r
  // Get all Vol handle\r
  //\r
  Status = gBS->LocateHandleBuffer (\r
                   ByProtocol,\r
                   &gEfiSimpleFileSystemProtocolGuid,\r
                   NULL,\r
                   &NoHandles,\r
                   &HandleBuffer\r
                   );\r
  if (EFI_ERROR (Status) && (NoHandles == 0)) {\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  //\r
  // Walk through each Vol\r
  //\r
  DebuggerPrivate->Vol = NULL;\r
  *BufferSize = 0;\r
  *Buffer     = NULL;\r
  for (Index = 0; Index < NoHandles; Index++) {\r
    Status = gBS->HandleProtocol (\r
                    HandleBuffer[Index],\r
                    &gEfiSimpleFileSystemProtocolGuid,\r
                    (VOID**) &Vol\r
                    );\r
    if (EFI_ERROR(Status)) {\r
      continue;\r
    }\r
\r
    Status = ReadFileFromVol (Vol, FileName, &TempBufferSize, &TempBuffer);\r
    if (!EFI_ERROR (Status)) {\r
      //\r
      // Read file OK, check duplication\r
      //\r
      if (DebuggerPrivate->Vol != NULL) {\r
        //\r
        // Find the duplicated file\r
        //\r
        gBS->FreePool (TempBuffer);\r
        gBS->FreePool (*Buffer);\r
        EDBPrint (L"Duplicated FileName found!\n");\r
        return EFI_NO_MAPPING;\r
      } else {\r
        //\r
        // Record value\r
        //\r
        DebuggerPrivate->Vol = Vol;\r
        *BufferSize = TempBufferSize;\r
        *Buffer     = TempBuffer;\r
      }\r
    }\r
  }\r
\r
  //\r
  // Scan Fs done\r
  //\r
  if (DebuggerPrivate->Vol == NULL) {\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  //\r
  // Done\r
  //\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
\r
  Get file name under this dir with index\r
\r
  @param  DebuggerPrivate - EBC Debugger private data structure\r
  @param  DirName         - The dir to be read.\r
  @param  FileName        - The file name pattern under this dir\r
  @param  Index           - The file index under this dir\r
\r
  @return File Name which match the pattern and index.\r
\r
**/\r
CHAR16 *\r
EFIAPI\r
GetFileNameUnderDir (\r
  IN  EFI_DEBUGGER_PRIVATE_DATA   *DebuggerPrivate,\r
  IN  CHAR16                      *DirName,\r
  IN  CHAR16                      *FileName,\r
  IN OUT UINTN                    *Index\r
  )\r
{\r
  EFI_STATUS                        Status;\r
  EFI_FILE_HANDLE                   RootDir;\r
  EFI_FILE_HANDLE                   Handle;\r
  UINTN                             FileInfoSize;\r
  EFI_FILE_INFO                     *FileInfo;\r
  EFI_SIMPLE_FILE_SYSTEM_PROTOCOL   *Vol;\r
  VOID                              *TempName;\r
  UINTN                             FileIndex;\r
\r
  if (DebuggerPrivate->Vol == NULL) {\r
    Status = gBS->LocateProtocol (\r
                    &gEfiSimpleFileSystemProtocolGuid,\r
                    NULL,\r
                    (VOID**) &DebuggerPrivate->Vol\r
                    );\r
    if (EFI_ERROR(Status)) {\r
      return NULL;\r
    }\r
  }\r
  Vol = DebuggerPrivate->Vol;\r
\r
  //\r
  // Open the root directory\r
  //\r
  Status = Vol->OpenVolume (Vol, &RootDir);\r
  if (EFI_ERROR (Status)) {\r
    return NULL;\r
  }\r
\r
  //\r
  // Open the file\r
  //\r
  Status = RootDir->Open (\r
                      RootDir,\r
                      &Handle,\r
                      DirName,\r
                      EFI_FILE_MODE_READ,\r
                      EFI_FILE_DIRECTORY\r
                      );\r
  if (EFI_ERROR (Status)) {\r
    RootDir->Close (RootDir);\r
    return NULL;\r
  }\r
  RootDir->Close (RootDir);\r
\r
  //\r
  // Set Dir Position\r
  //\r
  Status = Handle->SetPosition (Handle, 0);\r
  if (EFI_ERROR (Status)) {\r
    Handle->Close (Handle);\r
    return NULL;\r
  }\r
\r
  //\r
  // Get the file information\r
  //\r
  FileInfoSize = sizeof(EFI_FILE_INFO) + 1024;\r
\r
  FileInfo = AllocateZeroPool (FileInfoSize);\r
  if (FileInfo == NULL) {\r
    Handle->Close (Handle);\r
    return NULL;\r
  }\r
\r
  //\r
  // Walk through each file in the directory\r
  //\r
  FileIndex = 0;\r
  TempName = NULL;\r
  while (TRUE) {\r
    //\r
    // Read a file entry\r
    //\r
    FileInfoSize = sizeof(EFI_FILE_INFO) + 1024;\r
\r
    Status = Handle->Read (\r
                       Handle,\r
                       &FileInfoSize,\r
                       FileInfo\r
                       );\r
    if (EFI_ERROR (Status) || (FileInfoSize == 0)) {\r
      break;\r
    }\r
\r
    if ((FileInfo->Attribute & EFI_FILE_DIRECTORY) == 0) {\r
      //\r
      // This is a file\r
      //\r
\r
      //\r
      // Only deal with the EFI key file\r
      //\r
      if (!StrEndWith (FileInfo->FileName, FileName)) {\r
        continue;\r
      }\r
\r
      if (FileIndex == *Index) {\r
        TempName = StrDuplicate (FileInfo->FileName);\r
        *Index = *Index + 1;\r
        break;\r
      }\r
      FileIndex ++;\r
    }\r
  }\r
\r
  //\r
  // Free resources\r
  //\r
  gBS->FreePool (FileInfo);\r
  Handle->Close (Handle);\r
\r
  return TempName;\r
}\r
Commit	Line	Data
e8a5ac7c	1	/** @file\r
748edcd5	2	\r
e8a5ac7c DB	3	Copyright (c) 2007, Intel Corporation. All rights reserved.<BR>\r
e8a5ac7c DB	4	This program and the accompanying materials\r
748edcd5 PB	5	are licensed and made available under the terms and conditions of the BSD License\r
	6	which accompanies this distribution. The full text of the license may be found at\r
	7	http://opensource.org/licenses/bsd-license.php\r
	8	\r
	9	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	10	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	11	\r
748edcd5	12	\r
e8a5ac7c	13	**/\r
748edcd5	14	\r
e8a5ac7c	15	#include "Edb.h"\r
748edcd5	16	\r
e8a5ac7c DB	17	/**\r
e8a5ac7c DB	18	Read a file.\r
748edcd5	19	\r
e8a5ac7c DB	20	@param Vol - File System Volume\r
	21	@param FileName - The file to be read.\r
	22	@param BufferSize - The file buffer size\r
	23	@param Buffer - The file buffer\r
748edcd5	24	\r
e8a5ac7c DB	25	@retval EFI_SUCCESS - read file successfully\r
e8a5ac7c DB	26	@retval EFI_NOT_FOUND - file not found\r
748edcd5	27	\r
e8a5ac7c	28	**/\r
748edcd5 PB	29	EFI_STATUS\r
	30	EFIAPI\r
	31	ReadFileFromVol (\r
	32	IN EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *Vol,\r
	33	IN CHAR16 *FileName,\r
	34	OUT UINTN *BufferSize,\r
	35	OUT VOID **Buffer\r
	36	)\r
748edcd5 PB	37	{\r
	38	EFI_STATUS Status;\r
	39	EFI_FILE_HANDLE RootDir;\r
	40	EFI_FILE_HANDLE Handle;\r
	41	UINTN FileInfoSize;\r
	42	EFI_FILE_INFO *FileInfo;\r
	43	UINTN TempBufferSize;\r
	44	VOID *TempBuffer;\r
	45	\r
	46	//\r
	47	// Open the root directory\r
	48	//\r
	49	Status = Vol->OpenVolume (Vol, &RootDir);\r
	50	if (EFI_ERROR (Status)) {\r
	51	return Status;\r
	52	}\r
	53	\r
	54	//\r
	55	// Open the file\r
	56	//\r
	57	Status = RootDir->Open (\r
	58	RootDir,\r
	59	&Handle,\r
	60	FileName,\r
	61	EFI_FILE_MODE_READ,\r
	62	0\r
	63	);\r
	64	if (EFI_ERROR (Status)) {\r
	65	RootDir->Close (RootDir);\r
	66	return Status;\r
	67	}\r
	68	\r
	69	RootDir->Close (RootDir);\r
	70	\r
	71	//\r
	72	// Get the file information\r
	73	//\r
	74	FileInfoSize = sizeof(EFI_FILE_INFO) + 1024;\r
	75	\r
	76	FileInfo = AllocateZeroPool (FileInfoSize);\r
	77	if (FileInfo == NULL) {\r
	78	Handle->Close (Handle);\r
	79	return Status;\r
	80	}\r
	81	\r
	82	Status = Handle->GetInfo (\r
	83	Handle,\r
	84	&gEfiFileInfoGuid,\r
	85	&FileInfoSize,\r
	86	FileInfo\r
	87	);\r
	88	if (EFI_ERROR (Status)) {\r
	89	Handle->Close (Handle);\r
	90	gBS->FreePool (FileInfo);\r
	91	return Status;\r
	92	}\r
	93	\r
	94	//\r
	95	// Allocate buffer for the file data. The last CHAR16 is for L'\0'\r
	96	//\r
	97	TempBufferSize = (UINTN) FileInfo->FileSize + sizeof(CHAR16);\r
	98	TempBuffer = AllocateZeroPool (TempBufferSize);\r
	99	if (TempBuffer == NULL) {\r
	100	Handle->Close (Handle);\r
101	gBS->FreePool (FileInfo);\r
102	return Status;\r
103	}\r
104	\r
105	gBS->FreePool (FileInfo);\r
106	\r
107	//\r
108	// Read the file data to the buffer\r
109	//\r
110	Status = Handle->Read (\r
111	Handle,\r
112	&TempBufferSize,\r
113	TempBuffer\r
114	);\r
115	if (EFI_ERROR (Status)) {\r
116	Handle->Close (Handle);\r
117	gBS->FreePool (TempBuffer);\r
118	return Status;\r
119	}\r
120	\r
121	Handle->Close (Handle);\r
122	\r
123	*BufferSize = TempBufferSize;\r
124	*Buffer = TempBuffer;\r
125	return EFI_SUCCESS;\r
126	}\r
127	\r
e8a5ac7c DB	128	/**\r
	129	\r
	130	Read a file.\r
	131	If ScanFs is FLASE, it will use DebuggerPrivate->Vol as default Fs.\r
	132	If ScanFs is TRUE, it will scan all FS and check the file.\r
	133	If there is only one file match the name, it will be read.\r
	134	If there is more than one file match the name, it will return Error.\r
	135	\r
	136	@param DebuggerPrivate - EBC Debugger private data structure\r
	137	@param FileName - The file to be read.\r
	138	@param BufferSize - The file buffer size\r
	139	@param Buffer - The file buffer\r
	140	@param ScanFs - Need Scan all FS\r
	141	\r
	142	@retval EFI_SUCCESS - read file successfully\r
	143	@retval EFI_NOT_FOUND - file not found\r
	144	@retval EFI_NO_MAPPING - there is duplicated files found\r
	145	\r
	146	**/\r
748edcd5 PB	147	EFI_STATUS\r
	148	EFIAPI\r
	149	ReadFileToBuffer (\r
	150	IN EFI_DEBUGGER_PRIVATE_DATA *DebuggerPrivate,\r
	151	IN CHAR16 *FileName,\r
	152	OUT UINTN *BufferSize,\r
	153	OUT VOID **Buffer,\r
	154	IN BOOLEAN ScanFs\r
	155	)\r
748edcd5 PB	156	{\r
	157	EFI_STATUS Status;\r
	158	EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *Vol;\r
	159	UINTN TempBufferSize;\r
	160	VOID *TempBuffer;\r
	161	UINTN NoHandles;\r
	162	EFI_HANDLE *HandleBuffer;\r
	163	UINTN Index;\r
	164	\r
	165	//\r
	166	// Check parameters\r
	167	//\r
	168	if ((FileName == NULL) \|\| (Buffer == NULL)) {\r
	169	return EFI_INVALID_PARAMETER;\r
	170	}\r
	171	\r
	172	//\r
	173	// not scan fs\r
	174	//\r
	175	if (!ScanFs) {\r
	176	if (DebuggerPrivate->Vol == NULL) {\r
	177	return EFI_INVALID_PARAMETER;\r
	178	}\r
	179	//\r
	180	// Read file directly from Vol\r
	181	//\r
	182	return ReadFileFromVol (DebuggerPrivate->Vol, FileName, BufferSize, Buffer);\r
	183	}\r
	184	\r
	185	//\r
	186	// need scan fs\r
	187	//\r
	188	\r
	189	//\r
	190	// Get all Vol handle\r
	191	//\r
	192	Status = gBS->LocateHandleBuffer (\r
	193	ByProtocol,\r
	194	&gEfiSimpleFileSystemProtocolGuid,\r
	195	NULL,\r
	196	&NoHandles,\r
	197	&HandleBuffer\r
	198	);\r
	199	if (EFI_ERROR (Status) && (NoHandles == 0)) {\r
	200	return EFI_NOT_FOUND;\r
	201	}\r
	202	\r
	203	//\r
	204	// Walk through each Vol\r
	205	//\r
	206	DebuggerPrivate->Vol = NULL;\r
	207	*BufferSize = 0;\r
	208	*Buffer = NULL;\r
	209	for (Index = 0; Index < NoHandles; Index++) {\r
	210	Status = gBS->HandleProtocol (\r
	211	HandleBuffer[Index],\r
	212	&gEfiSimpleFileSystemProtocolGuid,\r
	213	(VOID**) &Vol\r
	214	);\r
	215	if (EFI_ERROR(Status)) {\r
	216	continue;\r
	217	}\r
	218	\r
	219	Status = ReadFileFromVol (Vol, FileName, &TempBufferSize, &TempBuffer);\r
220	if (!EFI_ERROR (Status)) {\r
221	//\r
222	// Read file OK, check duplication\r
223	//\r
224	if (DebuggerPrivate->Vol != NULL) {\r
225	//\r
226	// Find the duplicated file\r
227	//\r
228	gBS->FreePool (TempBuffer);\r
229	gBS->FreePool (*Buffer);\r
230	EDBPrint (L"Duplicated FileName found!\n");\r
231	return EFI_NO_MAPPING;\r
232	} else {\r
233	//\r
234	// Record value\r
235	//\r
236	DebuggerPrivate->Vol = Vol;\r
237	*BufferSize = TempBufferSize;\r
238	*Buffer = TempBuffer;\r
239	}\r
240	}\r
241	}\r
242	\r
243	//\r
244	// Scan Fs done\r
245	//\r
246	if (DebuggerPrivate->Vol == NULL) {\r
247	return EFI_NOT_FOUND;\r
248	}\r
249	\r
250	//\r
251	// Done\r
252	//\r
253	return EFI_SUCCESS;\r
254	}\r
255	\r
e8a5ac7c DB	256	/**\r
	257	\r
	258	Get file name under this dir with index\r
	259	\r
	260	@param DebuggerPrivate - EBC Debugger private data structure\r
	261	@param DirName - The dir to be read.\r
	262	@param FileName - The file name pattern under this dir\r
	263	@param Index - The file index under this dir\r
	264	\r
	265	@return File Name which match the pattern and index.\r
	266	\r
	267	**/\r
748edcd5 PB	268	CHAR16 *\r
	269	EFIAPI\r
	270	GetFileNameUnderDir (\r
	271	IN EFI_DEBUGGER_PRIVATE_DATA *DebuggerPrivate,\r
	272	IN CHAR16 *DirName,\r
	273	IN CHAR16 *FileName,\r
	274	IN OUT UINTN *Index\r
	275	)\r
748edcd5 PB	276	{\r
	277	EFI_STATUS Status;\r
	278	EFI_FILE_HANDLE RootDir;\r
	279	EFI_FILE_HANDLE Handle;\r
	280	UINTN FileInfoSize;\r
	281	EFI_FILE_INFO *FileInfo;\r
	282	EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *Vol;\r
	283	VOID *TempName;\r
	284	UINTN FileIndex;\r
	285	\r
	286	if (DebuggerPrivate->Vol == NULL) {\r
	287	Status = gBS->LocateProtocol (\r
	288	&gEfiSimpleFileSystemProtocolGuid,\r
	289	NULL,\r
	290	(VOID**) &DebuggerPrivate->Vol\r
	291	);\r
	292	if (EFI_ERROR(Status)) {\r
	293	return NULL;\r
	294	}\r
	295	}\r
	296	Vol = DebuggerPrivate->Vol;\r
	297	\r
	298	//\r
	299	// Open the root directory\r
	300	//\r
	301	Status = Vol->OpenVolume (Vol, &RootDir);\r
	302	if (EFI_ERROR (Status)) {\r
	303	return NULL;\r
	304	}\r
	305	\r
	306	//\r
	307	// Open the file\r
	308	//\r
	309	Status = RootDir->Open (\r
	310	RootDir,\r
	311	&Handle,\r
	312	DirName,\r
	313	EFI_FILE_MODE_READ,\r
	314	EFI_FILE_DIRECTORY\r
	315	);\r
	316	if (EFI_ERROR (Status)) {\r
	317	RootDir->Close (RootDir);\r
	318	return NULL;\r
	319	}\r
	320	RootDir->Close (RootDir);\r
	321	\r
	322	//\r
	323	// Set Dir Position\r
	324	//\r
	325	Status = Handle->SetPosition (Handle, 0);\r
	326	if (EFI_ERROR (Status)) {\r
	327	Handle->Close (Handle);\r
	328	return NULL;\r
	329	}\r
	330	\r
	331	//\r
	332	// Get the file information\r
	333	//\r
	334	FileInfoSize = sizeof(EFI_FILE_INFO) + 1024;\r
	335	\r
	336	FileInfo = AllocateZeroPool (FileInfoSize);\r
	337	if (FileInfo == NULL) {\r
	338	Handle->Close (Handle);\r
	339	return NULL;\r
340	}\r
341	\r
342	//\r
343	// Walk through each file in the directory\r
344	//\r
345	FileIndex = 0;\r
346	TempName = NULL;\r
347	while (TRUE) {\r
348	//\r
349	// Read a file entry\r
350	//\r
351	FileInfoSize = sizeof(EFI_FILE_INFO) + 1024;\r
352	\r
353	Status = Handle->Read (\r
354	Handle,\r
355	&FileInfoSize,\r
356	FileInfo\r
357	);\r
358	if (EFI_ERROR (Status) \|\| (FileInfoSize == 0)) {\r
359	break;\r
360	}\r
361	\r
362	if ((FileInfo->Attribute & EFI_FILE_DIRECTORY) == 0) {\r
363	//\r
364	// This is a file\r
365	//\r
366	\r
367	//\r
368	// Only deal with the EFI key file\r
369	//\r
370	if (!StrEndWith (FileInfo->FileName, FileName)) {\r
371	continue;\r
372	}\r
373	\r
374	if (FileIndex == *Index) {\r
375	TempName = StrDuplicate (FileInfo->FileName);\r
376	Index = Index + 1;\r
377	break;\r
378	}\r
379	FileIndex ++;\r
380	}\r
381	}\r
382	\r
383	//\r
384	// Free resources\r
385	//\r
386	gBS->FreePool (FileInfo);\r
387	Handle->Close (Handle);\r
388	\r
389	return TempName;\r
390	}\r